trunk/Src/wptKeyserverDlg.cpp

/* wptKeyserverDlg.cpp - Keyserver dialog
 *      Copyright (C) 2000-2005 Timo Schulz
 *      Copyright (C) 2005 g10 Code GmbH
 *
 * This file is part of WinPT.
 *
 * WinPT is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * WinPT is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with WinPT; if not, write to the Free Software Foundation, 
 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
 */
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <windows.h>
#include <commctrl.h>
#include <malloc.h>

#include "resource.h"
#include "wptKeyserver.h"
#include "wptErrors.h"
#include "wptTypes.h"
#include "wptCommonCtl.h"
#include "wptNLS.h"
#include "wptW32API.h"
#include "wptVersion.h"
#include "wptGPG.h"
#include "wptKeyManager.h"
#include "wptContext.h" /* for passphrase_s */
#include "wptDlgs.h"
#include "wptUTF8.h"


#define MAX_KEYSIZE 70000

char* get_reg_entry_keyserver (const char *);
int set_reg_entry_keyserver (const char *, const char *);


/* Print out keyserver error and a possible Winsock error. */
static void
hkp_err_box (HWND dlg, const char *host, WORD port, int rc)
{
    const char *err = kserver_strerror ();

    log_box (_("Keyserver"), MB_ERR, "%s:%d: %s", 
             host, port, winpt_strerror (rc));
    if (err)
        msg_box (dlg, err, wsock_strerror (), MB_ERR);
}


/* Send the extracted key with keyid @pattern to the 
   keyserver @kserver (port @port).
   Return value: 0 on success. */
int
hkp_send_key (HWND dlg, const char *kserver, WORD port, const char *pattern)
{
    gpgme_ctx_t ctx;
    gpgme_data_t keydata;
    gpgme_error_t ec;
    char *rawkey = NULL;
    char msg[384];
    size_t n;
    
    ec = gpgme_new (&ctx);
    if (ec)
        BUG (NULL);
    gpgme_set_armor (ctx, 1);
    ec = gpgme_data_new (&keydata);
    if (ec)
        BUG (NULL);
    ec = gpgme_op_export (ctx, pattern, 0, keydata);
    if (ec) {
        msg_box (dlg, gpgme_strerror (ec), _("Export"), MB_ERR);
        goto leave;
    }
    rawkey = gpgme_data_release_and_get_mem (keydata, &n);
    ec = kserver_sendkey (kserver, port, rawkey, n);
    if (ec) {
        hkp_err_box (dlg, kserver, port, ec);
        goto leave;
    }
    
    _snprintf (msg, sizeof (msg) -1, _("Key '%s' successfully sent"), pattern);
    status_box (dlg, msg, _("GnuPG status"));
    
leave:
    gpgme_release (ctx);
    if (rawkey)
        gpgme_free (rawkey);
    return ec? -1 : 0;
}


/* Show all received keys from the keyserver. */
static void
show_imported_keys (gpgme_import_result_t res)
{
    gpgme_import_status_t t;
    gpgme_key_t key;
    gpgme_data_t msg;
    gpgme_ctx_t ctx;
    gpgme_error_t err;
    const char *s;
    char *p, *uid;
    size_t n=0;

    if (!res)
        return;

    err = gpgme_data_new (&msg);
    if (err)
        BUG (NULL);
    err = gpgme_new (&ctx);
    if (err)
        BUG (NULL);
    if (res->considered > 1) {
        s = _("WARNING: multiple keys matched request.\n\n");
        gpgme_data_write (msg, s, strlen (s));
    }

    if (res->unchanged == res->considered)
        s = _("Key(s) successfully received but nothing was changed.");
    else
        s = _("Key(s) sucessfully received and imported.");
    gpgme_data_write (msg, s, strlen (s));
    gpgme_data_write (msg, "\n\n", 2);

    for (t=res->imports; t; t = t->next) {
        if (!gpgme_get_key (ctx, t->fpr, &key, 0)) {
            s = uid = utf8_to_wincp2 (key->uids->uid);
            gpgme_data_write (msg, s, strlen (s));
            gpgme_data_write (msg, "\n", 1);
            gpgme_key_release (key);
            free (uid);
        }
    }
    gpgme_data_write (msg, "\0", 1);
    p = gpgme_data_release_and_get_mem (msg, &n);
    if (p != NULL) {
        msg_box (NULL, p, _("Imported Keys"), MB_INFO);
        gpgme_free (p);
    }
    gpgme_release (ctx);
}


/* Receive a key from the keyserver @kserver (port @port)
   with the pattern @pattern.
   Return value: 0 on success. */
static int
keyserver_recv_key (HWND dlg, const char *kserver, WORD port,
                    const char *pattern, int proto, int flags,
                    char **r_fpr)
{
    gpgme_ctx_t ctx;
    gpgme_data_t keydata;
    gpgme_error_t ec;
    gpgme_import_result_t import_res = NULL;    
    char *rawkey = NULL;
    int rc;
    
    /* XXX: implement dynamic buffers. */
    rawkey = new char[MAX_KEYSIZE];
    if (!rawkey)
        BUG (0);
    memset (rawkey, 0, MAX_KEYSIZE);
    if (proto == KSPROTO_LDAP) {
        rc = ldap_recvkey (kserver, pattern, rawkey, MAX_KEYSIZE-1);
        if (rc) {
            msg_box (dlg, _("LDAP key import failed.\n"
                            "Please make sure you have an online connection"
                            " and gpgkeys_ldap.exe is installed"),
                            _("Keyserver"), MB_ERR);
            free_if_alloc (rawkey);
            return rc;
        }
    }
    else if (proto == KSPROTO_FINGER) {
        rc = finger_recvkey (kserver, pattern, rawkey, MAX_KEYSIZE-1);
        if (rc) {
            log_box (_("Keyserver"), MB_ERR, 
                     _("Finger key import failed: %s\n"), winpt_strerror (rc));
            free_if_alloc (rawkey);
            return rc;
        }
    }
    else if ((rc = kserver_recvkey (kserver, port, 
                                    kserver_check_keyid (pattern),
                                    rawkey, MAX_KEYSIZE-1))) {
        hkp_err_box (dlg, kserver, port, rc);
        free_if_alloc (rawkey);
        return rc;
    }
    else {
        if (!strstr (rawkey, "BEGIN PGP PUBLIC KEY BLOCK")) {
            msg_box (dlg, _("This is not a valid OpenPGP key."), 
                     _("Keyserver"), MB_ERR);
            goto leave;
        }
        ec = gpgme_new (&ctx);
        if (ec)
            BUG (NULL);
        gpgme_data_new_from_mem (&keydata, rawkey, strlen (rawkey), 1);
        rc = gpgme_op_import (ctx, keydata);
        if (rc) {
            msg_box (dlg, gpgme_strerror ((gpgme_error_t)rc), _("Import"), MB_ERR);
            goto leave;
        }
        import_res = gpgme_op_import_result (ctx);
        if (import_res && r_fpr)
            *r_fpr = m_strdup (import_res->imports->fpr);
    }
    
    /* if we use the refresh mode, a lot of keys will be fetched and thus only
       a summarize at the end is presented and not for each key. */
    if (!(flags & KM_KS_REFRESH)) {
        show_imported_keys (import_res);
        if (import_res && import_res->unchanged == import_res->considered) {
            rc = WPTERR_GENERAL; /* no keys updated. */
            goto leave;
        }
    }

leave:
    free_if_alloc (rawkey);
    gpgme_release (ctx);
    gpgme_data_release (keydata);

    return rc;
}


/* Alias for keyserver_recv_key but without the ability to return 
   the fingerprint of the key. */
int
hkp_recv_key (HWND dlg, const char *kserver, WORD port,
              const char *pattern, int proto, int flags)
{
    return keyserver_recv_key (dlg, kserver, port, pattern, 
                               proto, flags, NULL);
}

int
hkp_recv_key2 (HWND dlg, const char *kserver, WORD port,
               const char *pattern, int proto, char **r_fpr)
{
    return keyserver_recv_key (dlg, kserver, port, pattern, proto, 0,
                               r_fpr);
}


#define my_iskeychar(a) (((a) >='0' && (a) <= '9' ) || ((a) >= 'A' && (a) <= 'F'))

static int
check_pattern (const char *pattern)
{
    int rc = 1;
    
    /* Whitespace are not allowed! */
    if (strchr( pattern, ' ')) {
        rc = WPTERR_GENERAL;
        goto leave;
    }
    
    if (((strstr (pattern, "0x")) && (strlen (pattern) == 10)) ||
         (strstr(pattern, "0x")) && (strlen (pattern)  == 18)) {
        rc = 0; /* Either long or short keyid */
        goto leave;
    }
    
    if( (( my_iskeychar( pattern[0] )) && ( strlen( pattern )  == 8 ) )
         || (my_iskeychar(pattern[0])) && ( strlen( pattern ) == 16) ) {
        rc = 0;
        goto leave;
    }
    
    if( ( strchr( pattern, '@' ) ) && ( strlen( pattern ) >= 3 ) ) {
        rc = 0;
        goto leave;
    }
    
leave:
    return rc;
} /* check_pattern */


static void
set_proxy (HWND dlg)
{
    char proxy[256];
    int port = 0;
    
    strcpy (proxy, "HTTP proxy: ");
    if (kserver_get_proxy (&port)) {
        char t[128];
        const char *http = kserver_get_proxy (&port);
        _snprintf (t, sizeof (t) - 1, "\"%s:%d\"", http, port);
        strcat (proxy, t);
    }
    else
        strcat (proxy, "none");
    SetDlgItemText (dlg, IDC_KEYSERVER_PROXY, proxy);
} /* set_proxy */


static int inline
kserver_get_pos (listview_ctrl_t lv)
{
    if (listview_count_items (lv, 0) == 1)
        return 0;
    return listview_get_curr_pos (lv);
}


static u16 inline
kserver_get_port (listview_ctrl_t lv)
{
    char buf[16];

    listview_get_item_text (lv, kserver_get_pos (lv), 3, buf, 15);
    return (u16)strtoul (buf, NULL, 10);
}

static void
load_default_ks (listview_ctrl_t lv)
{
    char * p, buf[192];
    int i;

    p = get_reg_entry_keyserver ("Default");
    if (!p)
        return;
    for( i = 0; i < listview_count_items( lv, 0 ); i++ ) {
        listview_get_item_text( lv, i, 0, buf, sizeof (buf)-1 );
        if( !strncmp( p, buf, strlen( p ) ) ) {
            listview_add_sub_item( lv, i, 2, "x" );
            break;
        }
    }
    free_if_alloc (p);
} /* load_default_ks */


static int
save_default_ks (listview_ctrl_t lv)
{
    char buf[192], port[32];
    int idx, i;   

    idx = listview_get_curr_pos( lv );
    if( idx == -1 ) {
        msg_box( NULL, _("Please select one of the servers."), _("Keyserver"), MB_ERR );
        return -1;
    }
    listview_get_item_text( lv, idx, 1, buf, sizeof (buf)-1 );
    if( *buf != 'H' ) {
        msg_box( NULL, _("Only HTTP keyserver can be used."), _("Keyserver"), MB_ERR );
        return -1;
    }
    for (i = 0; i < listview_count_items( lv, 0 ); i++)
        listview_add_sub_item (lv, i, 2, "");
    listview_add_sub_item (lv, idx, 2, "x");
    listview_get_item_text (lv, idx, 0, buf, sizeof (buf)-1);
    set_reg_entry_keyserver ("Default", buf);
    i = kserver_get_port (lv);
    sprintf (port, "%d", i);
    set_reg_entry_keyserver ("Default_Port", port);
    keyserver_set_default (buf, (u16)i);
    return 0;
} /* save_default_ks */


int
keyserver_list_build (listview_ctrl_t *r_lv, HWND hwnd)
{
    struct listview_column_s keyserver[] = {
        {0, 160, (char *)_("DNS Name")},        
        {1,  52, (char *)_("Protocol")},
        {2,  46, (char *)_("Default")},
        {3,  46, (char *)_("Port")},
        {0,   0, NULL}
    };
    HICON ico[1];
    listview_ctrl_t lv;
    char buf[32];
    int j;

    ico[0] = LoadIcon (glob_hinst, (LPCTSTR)IDI_COMPUTER);
    listview_new (&lv);    
    lv->ctrl = hwnd;
    for (j=0; keyserver[j].fieldname; j++)
        listview_add_column (lv, &keyserver[j]);
    listview_set_image_list (lv, ico, 1);
    for (j = 0; j<MAX_KEYSERVERS; j++) {
        if (!server[j].used)
            continue;
        listview_add_item_image (lv, " ", 0);
        listview_add_sub_item (lv, 0, 0, server[j].name);
        switch (server[j].proto) {
        case KSPROTO_HTTP:
            listview_add_sub_item( lv, 0, 1, "H" ); break;
        case KSPROTO_LDAP:
            listview_add_sub_item( lv, 0, 1, "L" ); break;
        case KSPROTO_FINGER:
            listview_add_sub_item( lv, 0, 1, "F" ); break;
        }
        sprintf (buf, "%d", server[j].port);
        listview_add_sub_item (lv, 0, 3, buf);
    }
    load_default_ks (lv);
    if (listview_count_items (lv, 0) == 0) {
        listview_add_item (lv, "");
        listview_add_sub_item (lv, 0, 0, DEF_HKP_KEYSERVER);
        listview_add_sub_item (lv, 0, 1, "H");
    }
    listview_set_ext_style (lv);
    *r_lv = lv;
    return 0;
}


/* Dialog box procedure to access keyservers. */
BOOL CALLBACK
keyserver_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static listview_ctrl_t lv = NULL;
    static int lv_idx = 0;
    keyserver_ctx ksc;
    char kserver[128], pattern[128];
    char proto[16], *fpr = NULL;
    int rc, proto_nr = 0;
    
    switch (msg) {
    case WM_INITDIALOG:
        SetWindowText (dlg, _("Keyserver Access"));
        SetDlgItemText (dlg, IDC_KEYSERVER_RECV, _("&Receive"));
        SetDlgItemText (dlg, IDC_KEYSERVER_SEND, 
                        _("Send key (default is receiving)"));
        SetDlgItemText (dlg, IDC_KEYSERVER_INFO,
                        _("Please enter the key ID or email address you search for"));
        SetDlgItemText (dlg, IDC_KEYSERVER_INDEX, _("&Search"));
        SetDlgItemText (dlg, IDC_KEYSERVER_PROXSETT, _("C&hange"));
        SetDlgItemText (dlg, IDC_KEYSERVER_DEFAULT, _("Set &default"));
        SetDlgItemText (dlg, IDCANCEL, _("&Close"));

        set_proxy (dlg);
        keyserver_list_build (&lv, GetDlgItem (dlg, IDC_KEYSERVER_LIST));
        center_window (dlg, NULL);
        SetForegroundWindow (dlg);
        return TRUE;
        
    case WM_NOTIFY:
        NMHDR *notify;
        notify = (NMHDR *)lparam;
        if (notify && notify->code == NM_CLICK
            && notify->idFrom == IDC_KEYSERVER_LIST)
            lv_idx = listview_get_curr_pos (lv);
        return TRUE;
        
    case WM_DESTROY:
        if (lv) {
            listview_release (lv);
            lv = NULL;
        }
        lv_idx = 0;
        return FALSE;
        
    case WM_SYSCOMMAND:
        if (LOWORD (wparam) == SC_CLOSE)
            EndDialog (dlg, TRUE);
        return FALSE;
        
    case WM_COMMAND:
        switch (LOWORD (wparam)) {
        case IDC_KEYSERVER_PROXSETT:
            dialog_box_param (glob_hinst, (LPCTSTR)IDD_WINPT_KEYSERVER_PROXY, 
                              dlg, keyserver_proxy_dlg_proc, NULL,
                              _("Proxy Settings"), IDS_WINPT_KEYSERVER_PROXY);
            set_proxy (dlg);
            return TRUE;
            
        case IDC_KEYSERVER_INDEX:
            if (!lv_idx) {
                lv_idx = kserver_get_pos (lv);
                if (lv_idx == -1) {
                    msg_box (dlg, _("Please select one of the keyservers."), 
                             _("Keyserver"), MB_INFO);
                    return FALSE;
                }
            }
            listview_get_item_text (lv, lv_idx, 1, proto, sizeof (proto)-1);
            if (*proto == 'L') {
                msg_box (dlg, _("This is not implemented yet!"), 
                         _("Keyserver"), MB_ERR);
                return FALSE;
            }
            listview_get_item_text (lv, lv_idx, 0, kserver, sizeof (kserver)-1);
            if (!GetDlgItemText (dlg, IDC_KEYSERVER_SEARCH, pattern, sizeof (pattern)-1)) {
                msg_box (dlg, _("Please enter the search pattern."), 
                         _("Keyserver"), MB_INFO);
                return FALSE;
            }
            ksc.name = kserver;
            ksc.pattern = pattern;
            ksc.port = kserver_get_port (lv);
            DialogBoxParam (glob_hinst, (LPCSTR)IDD_WINPT_HKPSEARCH, dlg,
                            hkpsearch_dlg_proc, (LPARAM) &ksc);
            return TRUE;
            
        case IDC_KEYSERVER_RECV:
            memset (&kserver, 0, sizeof (kserver));
            if (!lv_idx) {
                lv_idx = kserver_get_pos (lv);
                if (lv_idx == -1) {
                    msg_box (dlg, _("Please select one of the keyservers."), 
                            _("Keyserver"), MB_INFO);
                    return FALSE;
                }
            }
            listview_get_item_text( lv, lv_idx, 1, proto, sizeof (proto)-1 );
            proto_nr = KSPROTO_HTTP;
            if (*proto == 'L')
                proto_nr = KSPROTO_LDAP;
            else if (*proto == 'F')
                proto_nr = KSPROTO_FINGER;
            listview_get_item_text (lv, lv_idx, 0, kserver, sizeof (kserver)-1);
            if( !GetDlgItemText( dlg, IDC_KEYSERVER_SEARCH, pattern, sizeof (pattern)-1)) {
                msg_box (dlg, _("Please enter the search pattern."),
                         _("Keyserver"), MB_INFO);
                return FALSE;
            }
            if (proto_nr == KSPROTO_LDAP && strchr (pattern, '@')) {
                msg_box (dlg, _("Only keyids are allowed."), _("LDAP Keyserver"), MB_INFO);
                return FALSE;
            }
            else if (proto_nr == KSPROTO_FINGER) {
                if (strchr (pattern, '@') || strchr (pattern, ' ')) {
                    msg_box (dlg, _("Only enter the name of the user."), 
                             _("FINGER Keyserver"), MB_INFO);
                    return FALSE;
                }
            }
            else if (check_pattern (pattern)) {
                msg_box (dlg, _("Only email addresses or keyids are allowed."),
                        _("HKP Keyserver"), MB_INFO);
                return FALSE;
            }
            rc = keyserver_recv_key (dlg, kserver, kserver_get_port (lv),
                                     pattern, proto_nr, 0,
                                     &fpr);
            if (!rc && fpr != NULL) {
                keycache_update (0, fpr);
                free_if_alloc (fpr);
            }
            return TRUE;

        case IDC_KEYSERVER_DEFAULT:
            save_default_ks (lv);
            break;
            
        case IDCANCEL:
            EndDialog (dlg, FALSE);
            return FALSE;
        }
        break;
    }
    
    return FALSE;
}
1	/* wptKeyserverDlg.cpp - Keyserver dialog
2	* Copyright (C) 2000-2005 Timo Schulz
3	* Copyright (C) 2005 g10 Code GmbH
4	*
5	* This file is part of WinPT.
6	*
7	* WinPT is free software; you can redistribute it and/or modify
8	* it under the terms of the GNU General Public License as published by
9	* the Free Software Foundation; either version 2 of the License, or
10	* (at your option) any later version.
11	*
12	* WinPT is distributed in the hope that it will be useful,
13	* but WITHOUT ANY WARRANTY; without even the implied warranty of
14	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15	* GNU General Public License for more details.
16	*
17	* You should have received a copy of the GNU General Public License
18	* along with WinPT; if not, write to the Free Software Foundation,
19	* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
20	*/
21	#ifdef HAVE_CONFIG_H
22	#include <config.h>
23	#endif
24
25	#include <windows.h>
26	#include <commctrl.h>
27	#include <malloc.h>
28
29	#include "resource.h"
30	#include "wptKeyserver.h"
31	#include "wptErrors.h"
32	#include "wptTypes.h"
33	#include "wptCommonCtl.h"
34	#include "wptNLS.h"
35	#include "wptW32API.h"
36	#include "wptVersion.h"
37	#include "wptGPG.h"
38	#include "wptKeyManager.h"
39	#include "wptContext.h" /* for passphrase_s */
40	#include "wptDlgs.h"
41	#include "wptUTF8.h"
42
43
44	#define MAX_KEYSIZE 70000
45
46	char* get_reg_entry_keyserver (const char *);
47	int set_reg_entry_keyserver (const char , const char );
48
49
50	/* Print out keyserver error and a possible Winsock error. */
51	static void
52	hkp_err_box (HWND dlg, const char *host, WORD port, int rc)
53	{
54	const char *err = kserver_strerror ();
55
56	log_box (_("Keyserver"), MB_ERR, "%s:%d: %s",
57	host, port, winpt_strerror (rc));
58	if (err)
59	msg_box (dlg, err, wsock_strerror (), MB_ERR);
60	}
61
62
63	/* Send the extracted key with keyid @pattern to the
64	keyserver @kserver (port @port).
65	Return value: 0 on success. */
66	int
67	hkp_send_key (HWND dlg, const char kserver, WORD port, const char pattern)
68	{
69	gpgme_ctx_t ctx;
70	gpgme_data_t keydata;
71	gpgme_error_t ec;
72	char *rawkey = NULL;
73	char msg[384];
74	size_t n;
75
76	ec = gpgme_new (&ctx);
77	if (ec)
78	BUG (NULL);
79	gpgme_set_armor (ctx, 1);
80	ec = gpgme_data_new (&keydata);
81	if (ec)
82	BUG (NULL);
83	ec = gpgme_op_export (ctx, pattern, 0, keydata);
84	if (ec) {
85	msg_box (dlg, gpgme_strerror (ec), _("Export"), MB_ERR);
86	goto leave;
87	}
88	rawkey = gpgme_data_release_and_get_mem (keydata, &n);
89	ec = kserver_sendkey (kserver, port, rawkey, n);
90	if (ec) {
91	hkp_err_box (dlg, kserver, port, ec);
92	goto leave;
93	}
94
95	_snprintf (msg, sizeof (msg) -1, _("Key '%s' successfully sent"), pattern);
96	status_box (dlg, msg, _("GnuPG status"));
97
98	leave:
99	gpgme_release (ctx);
100	if (rawkey)
101	gpgme_free (rawkey);
102	return ec? -1 : 0;
103	}
104
105
106
107	/* Show all received keys from the keyserver. */
108	static void
109	show_imported_keys (gpgme_import_result_t res)
110	{
111	gpgme_import_status_t t;
112	gpgme_key_t key;
113	gpgme_data_t msg;
114	gpgme_ctx_t ctx;
115	gpgme_error_t err;
116	const char *s;
117	char p, uid;
118	size_t n=0;
119
120	if (!res)
121	return;
122
123	err = gpgme_data_new (&msg);
124	if (err)
125	BUG (NULL);
126	err = gpgme_new (&ctx);
127	if (err)
128	BUG (NULL);
129	if (res->considered > 1) {
130	s = _("WARNING: multiple keys matched request.\n\n");
131	gpgme_data_write (msg, s, strlen (s));
132	}
133
134	if (res->unchanged == res->considered)
135	s = _("Key(s) successfully received but nothing was changed.");
136	else
137	s = _("Key(s) sucessfully received and imported.");
138	gpgme_data_write (msg, s, strlen (s));
139	gpgme_data_write (msg, "\n\n", 2);
140
141	for (t=res->imports; t; t = t->next) {
142	if (!gpgme_get_key (ctx, t->fpr, &key, 0)) {
143	s = uid = utf8_to_wincp2 (key->uids->uid);
144	gpgme_data_write (msg, s, strlen (s));
145	gpgme_data_write (msg, "\n", 1);
146	gpgme_key_release (key);
147	free (uid);
148	}
149	}
150	gpgme_data_write (msg, "\0", 1);
151	p = gpgme_data_release_and_get_mem (msg, &n);
152	if (p != NULL) {
153	msg_box (NULL, p, _("Imported Keys"), MB_INFO);
154	gpgme_free (p);
155	}
156	gpgme_release (ctx);
157	}
158
159
160	/* Receive a key from the keyserver @kserver (port @port)
161	with the pattern @pattern.
162	Return value: 0 on success. */
163	static int
164	keyserver_recv_key (HWND dlg, const char *kserver, WORD port,
165	const char *pattern, int proto, int flags,
166	char **r_fpr)
167	{
168	gpgme_ctx_t ctx;
169	gpgme_data_t keydata;
170	gpgme_error_t ec;
171	gpgme_import_result_t import_res = NULL;
172	char *rawkey = NULL;
173	int rc;
174
175	/* XXX: implement dynamic buffers. */
176	rawkey = new char[MAX_KEYSIZE];
177	if (!rawkey)
178	BUG (0);
179	memset (rawkey, 0, MAX_KEYSIZE);
180	if (proto == KSPROTO_LDAP) {
181	rc = ldap_recvkey (kserver, pattern, rawkey, MAX_KEYSIZE-1);
182	if (rc) {
183	msg_box (dlg, _("LDAP key import failed.\n"
184	"Please make sure you have an online connection"
185	" and gpgkeys_ldap.exe is installed"),
186	_("Keyserver"), MB_ERR);
187	free_if_alloc (rawkey);
188	return rc;
189	}
190	}
191	else if (proto == KSPROTO_FINGER) {
192	rc = finger_recvkey (kserver, pattern, rawkey, MAX_KEYSIZE-1);
193	if (rc) {
194	log_box (_("Keyserver"), MB_ERR,
195	_("Finger key import failed: %s\n"), winpt_strerror (rc));
196	free_if_alloc (rawkey);
197	return rc;
198	}
199	}
200	else if ((rc = kserver_recvkey (kserver, port,
201	kserver_check_keyid (pattern),
202	rawkey, MAX_KEYSIZE-1))) {
203	hkp_err_box (dlg, kserver, port, rc);
204	free_if_alloc (rawkey);
205	return rc;
206	}
207	else {
208	if (!strstr (rawkey, "BEGIN PGP PUBLIC KEY BLOCK")) {
209	msg_box (dlg, _("This is not a valid OpenPGP key."),
210	_("Keyserver"), MB_ERR);
211	goto leave;
212	}
213	ec = gpgme_new (&ctx);
214	if (ec)
215	BUG (NULL);
216	gpgme_data_new_from_mem (&keydata, rawkey, strlen (rawkey), 1);
217	rc = gpgme_op_import (ctx, keydata);
218	if (rc) {
219	msg_box (dlg, gpgme_strerror ((gpgme_error_t)rc), _("Import"), MB_ERR);
220	goto leave;
221	}
222	import_res = gpgme_op_import_result (ctx);
223	if (import_res && r_fpr)
224	*r_fpr = m_strdup (import_res->imports->fpr);
225	}
226
227	/* if we use the refresh mode, a lot of keys will be fetched and thus only
228	a summarize at the end is presented and not for each key. */
229	if (!(flags & KM_KS_REFRESH)) {
230	show_imported_keys (import_res);
231	if (import_res && import_res->unchanged == import_res->considered) {
232	rc = WPTERR_GENERAL; /* no keys updated. */
233	goto leave;
234	}
235	}
236
237	leave:
238	free_if_alloc (rawkey);
239	gpgme_release (ctx);
240	gpgme_data_release (keydata);
241
242	return rc;
243	}
244
245
246	/* Alias for keyserver_recv_key but without the ability to return
247	the fingerprint of the key. */
248	int
249	hkp_recv_key (HWND dlg, const char *kserver, WORD port,
250	const char *pattern, int proto, int flags)
251	{
252	return keyserver_recv_key (dlg, kserver, port, pattern,
253	proto, flags, NULL);
254	}
255
256	int
257	hkp_recv_key2 (HWND dlg, const char *kserver, WORD port,
258	const char pattern, int proto, char *r_fpr)
259	{
260	return keyserver_recv_key (dlg, kserver, port, pattern, proto, 0,
261	r_fpr);
262	}
263
264
265	#define my_iskeychar(a) (((a) >='0' && (a) <= '9' ) \|\| ((a) >= 'A' && (a) <= 'F'))
266
267	static int
268	check_pattern (const char *pattern)
269	{
270	int rc = 1;
271
272	/* Whitespace are not allowed! */
273	if (strchr( pattern, ' ')) {
274	rc = WPTERR_GENERAL;
275	goto leave;
276	}
277
278	if (((strstr (pattern, "0x")) && (strlen (pattern) == 10)) \|\|
279	(strstr(pattern, "0x")) && (strlen (pattern) == 18)) {
280	rc = 0; /* Either long or short keyid */
281	goto leave;
282	}
283
284	if( (( my_iskeychar( pattern[0] )) && ( strlen( pattern ) == 8 ) )
285	\|\| (my_iskeychar(pattern[0])) && ( strlen( pattern ) == 16) ) {
286	rc = 0;
287	goto leave;
288	}
289
290	if( ( strchr( pattern, '@' ) ) && ( strlen( pattern ) >= 3 ) ) {
291	rc = 0;
292	goto leave;
293	}
294
295	leave:
296	return rc;
297	} /* check_pattern */
298
299
300	static void
301	set_proxy (HWND dlg)
302	{
303	char proxy[256];
304	int port = 0;
305
306	strcpy (proxy, "HTTP proxy: ");
307	if (kserver_get_proxy (&port)) {
308	char t[128];
309	const char *http = kserver_get_proxy (&port);
310	_snprintf (t, sizeof (t) - 1, "\"%s:%d\"", http, port);
311	strcat (proxy, t);
312	}
313	else
314	strcat (proxy, "none");
315	SetDlgItemText (dlg, IDC_KEYSERVER_PROXY, proxy);
316	} /* set_proxy */
317
318
319	static int inline
320	kserver_get_pos (listview_ctrl_t lv)
321	{
322	if (listview_count_items (lv, 0) == 1)
323	return 0;
324	return listview_get_curr_pos (lv);
325	}
326
327
328	static u16 inline
329	kserver_get_port (listview_ctrl_t lv)
330	{
331	char buf[16];
332
333	listview_get_item_text (lv, kserver_get_pos (lv), 3, buf, 15);
334	return (u16)strtoul (buf, NULL, 10);
335	}
336
337	static void
338	load_default_ks (listview_ctrl_t lv)
339	{
340	char * p, buf[192];
341	int i;
342
343	p = get_reg_entry_keyserver ("Default");
344	if (!p)
345	return;
346	for( i = 0; i < listview_count_items( lv, 0 ); i++ ) {
347	listview_get_item_text( lv, i, 0, buf, sizeof (buf)-1 );
348	if( !strncmp( p, buf, strlen( p ) ) ) {
349	listview_add_sub_item( lv, i, 2, "x" );
350	break;
351	}
352	}
353	free_if_alloc (p);
354	} /* load_default_ks */
355
356
357	static int
358	save_default_ks (listview_ctrl_t lv)
359	{
360	char buf[192], port[32];
361	int idx, i;
362
363	idx = listview_get_curr_pos( lv );
364	if( idx == -1 ) {
365	msg_box( NULL, _("Please select one of the servers."), _("Keyserver"), MB_ERR );
366	return -1;
367	}
368	listview_get_item_text( lv, idx, 1, buf, sizeof (buf)-1 );
369	if( *buf != 'H' ) {
370	msg_box( NULL, _("Only HTTP keyserver can be used."), _("Keyserver"), MB_ERR );
371	return -1;
372	}
373	for (i = 0; i < listview_count_items( lv, 0 ); i++)
374	listview_add_sub_item (lv, i, 2, "");
375	listview_add_sub_item (lv, idx, 2, "x");
376	listview_get_item_text (lv, idx, 0, buf, sizeof (buf)-1);
377	set_reg_entry_keyserver ("Default", buf);
378	i = kserver_get_port (lv);
379	sprintf (port, "%d", i);
380	set_reg_entry_keyserver ("Default_Port", port);
381	keyserver_set_default (buf, (u16)i);
382	return 0;
383	} /* save_default_ks */
384
385
386	int
387	keyserver_list_build (listview_ctrl_t *r_lv, HWND hwnd)
388	{
389	struct listview_column_s keyserver[] = {
390	{0, 160, (char *)_("DNS Name")},
391	{1, 52, (char *)_("Protocol")},
392	{2, 46, (char *)_("Default")},
393	{3, 46, (char *)_("Port")},
394	{0, 0, NULL}
395	};
396	HICON ico[1];
397	listview_ctrl_t lv;
398	char buf[32];
399	int j;
400
401	ico[0] = LoadIcon (glob_hinst, (LPCTSTR)IDI_COMPUTER);
402	listview_new (&lv);
403	lv->ctrl = hwnd;
404	for (j=0; keyserver[j].fieldname; j++)
405	listview_add_column (lv, &keyserver[j]);
406	listview_set_image_list (lv, ico, 1);
407	for (j = 0; j<MAX_KEYSERVERS; j++) {
408	if (!server[j].used)
409	continue;
410	listview_add_item_image (lv, " ", 0);
411	listview_add_sub_item (lv, 0, 0, server[j].name);
412	switch (server[j].proto) {
413	case KSPROTO_HTTP:
414	listview_add_sub_item( lv, 0, 1, "H" ); break;
415	case KSPROTO_LDAP:
416	listview_add_sub_item( lv, 0, 1, "L" ); break;
417	case KSPROTO_FINGER:
418	listview_add_sub_item( lv, 0, 1, "F" ); break;
419	}
420	sprintf (buf, "%d", server[j].port);
421	listview_add_sub_item (lv, 0, 3, buf);
422	}
423	load_default_ks (lv);
424	if (listview_count_items (lv, 0) == 0) {
425	listview_add_item (lv, "");
426	listview_add_sub_item (lv, 0, 0, DEF_HKP_KEYSERVER);
427	listview_add_sub_item (lv, 0, 1, "H");
428	}
429	listview_set_ext_style (lv);
430	*r_lv = lv;
431	return 0;
432	}
433
434
435	/* Dialog box procedure to access keyservers. */
436	BOOL CALLBACK
437	keyserver_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
438	{
439	static listview_ctrl_t lv = NULL;
440	static int lv_idx = 0;
441	keyserver_ctx ksc;
442	char kserver[128], pattern[128];
443	char proto[16], *fpr = NULL;
444	int rc, proto_nr = 0;
445
446	switch (msg) {
447	case WM_INITDIALOG:
448	SetWindowText (dlg, _("Keyserver Access"));
449	SetDlgItemText (dlg, IDC_KEYSERVER_RECV, _("&Receive"));
450	SetDlgItemText (dlg, IDC_KEYSERVER_SEND,
451	_("Send key (default is receiving)"));
452	SetDlgItemText (dlg, IDC_KEYSERVER_INFO,
453	_("Please enter the key ID or email address you search for"));
454	SetDlgItemText (dlg, IDC_KEYSERVER_INDEX, _("&Search"));
455	SetDlgItemText (dlg, IDC_KEYSERVER_PROXSETT, _("C&hange"));
456	SetDlgItemText (dlg, IDC_KEYSERVER_DEFAULT, _("Set &default"));
457	SetDlgItemText (dlg, IDCANCEL, _("&Close"));
458
459	set_proxy (dlg);
460	keyserver_list_build (&lv, GetDlgItem (dlg, IDC_KEYSERVER_LIST));
461	center_window (dlg, NULL);
462	SetForegroundWindow (dlg);
463	return TRUE;
464
465	case WM_NOTIFY:
466	NMHDR *notify;
467	notify = (NMHDR *)lparam;
468	if (notify && notify->code == NM_CLICK
469	&& notify->idFrom == IDC_KEYSERVER_LIST)
470	lv_idx = listview_get_curr_pos (lv);
471	return TRUE;
472
473	case WM_DESTROY:
474	if (lv) {
475	listview_release (lv);
476	lv = NULL;
477	}
478	lv_idx = 0;
479	return FALSE;
480
481	case WM_SYSCOMMAND:
482	if (LOWORD (wparam) == SC_CLOSE)
483	EndDialog (dlg, TRUE);
484	return FALSE;
485
486	case WM_COMMAND:
487	switch (LOWORD (wparam)) {
488	case IDC_KEYSERVER_PROXSETT:
489	dialog_box_param (glob_hinst, (LPCTSTR)IDD_WINPT_KEYSERVER_PROXY,
490	dlg, keyserver_proxy_dlg_proc, NULL,
491	_("Proxy Settings"), IDS_WINPT_KEYSERVER_PROXY);
492	set_proxy (dlg);
493	return TRUE;
494
495	case IDC_KEYSERVER_INDEX:
496	if (!lv_idx) {
497	lv_idx = kserver_get_pos (lv);
498	if (lv_idx == -1) {
499	msg_box (dlg, _("Please select one of the keyservers."),
500	_("Keyserver"), MB_INFO);
501	return FALSE;
502	}
503	}
504	listview_get_item_text (lv, lv_idx, 1, proto, sizeof (proto)-1);
505	if (*proto == 'L') {
506	msg_box (dlg, _("This is not implemented yet!"),
507	_("Keyserver"), MB_ERR);
508	return FALSE;
509	}
510	listview_get_item_text (lv, lv_idx, 0, kserver, sizeof (kserver)-1);
511	if (!GetDlgItemText (dlg, IDC_KEYSERVER_SEARCH, pattern, sizeof (pattern)-1)) {
512	msg_box (dlg, _("Please enter the search pattern."),
513	_("Keyserver"), MB_INFO);
514	return FALSE;
515	}
516	ksc.name = kserver;
517	ksc.pattern = pattern;
518	ksc.port = kserver_get_port (lv);
519	DialogBoxParam (glob_hinst, (LPCSTR)IDD_WINPT_HKPSEARCH, dlg,
520	hkpsearch_dlg_proc, (LPARAM) &ksc);
521	return TRUE;
522
523	case IDC_KEYSERVER_RECV:
524	memset (&kserver, 0, sizeof (kserver));
525	if (!lv_idx) {
526	lv_idx = kserver_get_pos (lv);
527	if (lv_idx == -1) {
528	msg_box (dlg, _("Please select one of the keyservers."),
529	_("Keyserver"), MB_INFO);
530	return FALSE;
531	}
532	}
533	listview_get_item_text( lv, lv_idx, 1, proto, sizeof (proto)-1 );
534	proto_nr = KSPROTO_HTTP;
535	if (*proto == 'L')
536	proto_nr = KSPROTO_LDAP;
537	else if (*proto == 'F')
538	proto_nr = KSPROTO_FINGER;
539	listview_get_item_text (lv, lv_idx, 0, kserver, sizeof (kserver)-1);
540	if( !GetDlgItemText( dlg, IDC_KEYSERVER_SEARCH, pattern, sizeof (pattern)-1)) {
541	msg_box (dlg, _("Please enter the search pattern."),
542	_("Keyserver"), MB_INFO);
543	return FALSE;
544	}
545	if (proto_nr == KSPROTO_LDAP && strchr (pattern, '@')) {
546	msg_box (dlg, _("Only keyids are allowed."), _("LDAP Keyserver"), MB_INFO);
547	return FALSE;
548	}
549	else if (proto_nr == KSPROTO_FINGER) {
550	if (strchr (pattern, '@') \|\| strchr (pattern, ' ')) {
551	msg_box (dlg, _("Only enter the name of the user."),
552	_("FINGER Keyserver"), MB_INFO);
553	return FALSE;
554	}
555	}
556	else if (check_pattern (pattern)) {
557	msg_box (dlg, _("Only email addresses or keyids are allowed."),
558	_("HKP Keyserver"), MB_INFO);
559	return FALSE;
560	}
561	rc = keyserver_recv_key (dlg, kserver, kserver_get_port (lv),
562	pattern, proto_nr, 0,
563	&fpr);
564	if (!rc && fpr != NULL) {
565	keycache_update (0, fpr);
566	free_if_alloc (fpr);
567	}
568	return TRUE;
569
570	case IDC_KEYSERVER_DEFAULT:
571	save_default_ks (lv);
572	break;
573
574	case IDCANCEL:
575	EndDialog (dlg, FALSE);
576	return FALSE;
577	}
578	break;
579	}
580
581	return FALSE;
582	}