trunk/Src/wptGPGME.cpp

/* wptGPGME.cpp - WinPT GPGME interface
 *      Copyright (C) 2001-2005 Timo Schulz
 *
 * 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 
 */

#include <sys/types.h>
#include <windows.h>

#include "../resource.h"
#include "wptNLS.h"
#include "wptGPG.h"
#include "wptErrors.h"
#include "wptTypes.h"
#include "wptW32API.h"
#include "wptVersion.h"
#include "wptCommonCtl.h"
#include "wptContext.h"
#include "wptRegistry.h"
#include "wptDlgs.h"

#include "openpgp.h"

BOOL CALLBACK  keycache_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam);
void progress_cleanup (progress_filter_s * pfx);

static gpgme_keycache_t pub = NULL;
static gpgme_keycache_t sec = NULL;
static unsigned int reload = 0;
static char *gpg_secring = NULL;

void
keycache_reload (HWND dlg) 
{       
    refresh_cache_s rcs;

    memset (&rcs, 0, sizeof rcs);
    rcs.kr_reload = rcs.kr_update = 1;
    rcs.tr_update = 0;
    DialogBoxParam (glob_hinst, (LPCSTR)IDD_WINPT_KEYCACHE, dlg,                                        
                    keycache_dlg_proc, (LPARAM)&rcs);
} /* keycache_reload */

void
keycache_release (void)
{
    int n = gpgme_keycache_count (pub);
    char tmpbuf[64];

    /* XXX: update the value when the cache has changed. */
    sprintf (tmpbuf, "%d", n);
    set_reg_key (HKEY_CURRENT_USER, "Software\\WinPT", "nKeys", tmpbuf);

    if (pub) {
        gpgme_keycache_release (pub); 
        pub = NULL;
    }
    if (sec) {
        gpgme_keycache_release (sec);
        sec = NULL;
    }
} /* keycache_release */


gpgme_error_t
keycache_update (int is_sec, const char *keyid)
{
    gpgme_keycache_t ctx = pub;
    gpgme_error_t err;

    if (is_sec)
        ctx = sec;
    err =  gpgme_keycache_update_key (ctx, is_sec, pub, keyid);
    if (is_sec)
        gpgme_keycache_prepare_single (ctx, keyid, NULL, gpg_secring);
    return err;
}

/* XXX: cache_keyring_names must be called then the GPG homedir changes! */

gpgme_error_t
keycache_init (const char *pubring, const char * secring)
{
    struct progress_filter_s pfx;
    gpgme_error_t err;
    int val = 0;
    char * p;

    if (secring != NULL) {
        free_if_alloc (gpg_secring);
        gpg_secring = get_gnupg_keyring (0, NO_STRICT);
    }

    if (reload) {
        keycache_release ();
        reload = 0;
    }
    p = get_reg_entry (HKEY_CURRENT_USER, "Software\\WinPT", "nKeys");
    if (p && *p != ' ') {
        val = atoi (p);
        free_if_alloc (p);
        memset (&pfx, 0, sizeof (pfx));
    }

    err = gpgme_keycache_new (&pub);
    if (err)
        return err;
    if (val != 0)
        gpgme_keycache_set_cb (pub, progress_callback, &pfx, val);
    err = gpgme_keycache_new (&sec);
    if (!err)
        err = gpgme_keycache_init (pub, NULL, 0);
    if (!err)
        err = gpgme_keycache_init( sec, NULL, 1 );
    if( !err && pubring && *pubring )
        err = gpgme_keycache_prepare( pub, pubring, NULL );
    if( !err && secring && * secring )
        err = gpgme_keycache_prepare( sec, NULL, secring );
    if (!err)
        gpgme_keycache_sync (pub, sec);
    if (val != 0)
        progress_cleanup (&pfx);
    return err;
} /* keycache_init */


void
keycache_set_reload( int yes )
{
    reload = yes;
} /* keycache_set_reload */


int
keycache_get_reload( void )
{
    return reload;
} /* keycache_get_reload */


gpgme_keycache_t
keycache_get_ctx (int is_pub)
{
    return is_pub? pub : sec;
} /* keycache_get_ctx */


static int
get_key (const char *keyid, gpgme_key_t *r_key, struct keycache_s **c, int secret)
{
    gpgme_keycache_t cache;
    gpgme_error_t err;
    int mode = secret? KEYCACHE_PRV : KEYCACHE_PUB;

    if( !keyid )
        return WPTERR_GENERAL;
    if( r_key )
        *r_key = NULL;
    cache = keycache_get_ctx (mode);
    if( !cache )
        BUG( NULL );
    if (!c)
        err = gpgme_keycache_find_key( cache, keyid, 0, r_key );
    else
        err = gpgme_keycache_find_key2 (cache, keyid, 0, r_key, c);
    if( err )
        return WPTERR_GENERAL;
    return 0;
} /* get_key */


static int
get_key2( const char * keyid, gpgme_key_t * r_key, int secret )
{
    gpg_iobuf_t inp;
    char * p;
    int rc;

    p = get_gnupg_keyring (1, !NO_STRICT);
    if( !p )
        BUG( NULL );

    inp = gpg_iobuf_open( p );
    if( !inp ) {
        const char *s = winpt_strerror( WPTERR_FILE_OPEN );
        log_box( _("WinPT Error"), 0, "%s: %s", p, s );
        free_if_alloc( p );
        return NULL;
    } 
    gpg_iobuf_ioctl( inp, 3, 1, NULL ); /* disable cache */
    /*rc = gpgme_getkey_bykeyid( inp, keyid, r_key ); XXX */
    rc=0;

    gpg_iobuf_close( inp );
    free_if_alloc( p );
    return rc;
} /* get_key2 */


int
winpt_get_pubkey (const char *keyid, winpt_key_s *k)
{
    int rc;
    
    rc = get_key (keyid, &k->ctx, &k->ext, 0);
    if (rc)
        return rc;
    k->is_v3 = k->ctx->subkeys->pubkey_algo == GPGME_PK_RSA && strlen (k->ctx->subkeys->fpr) == 32;
    k->is_protected = k->ext->gloflags.is_protected;
    k->keyid = k->ctx->subkeys->keyid;
    k->uid = k->ctx->uids->uid;
    return rc;
}


int
winpt_get_seckey (const char *keyid, winpt_key_s *k)
{
    int rc;
    rc = get_key (keyid, &k->ctx, &k->ext, 1);
    if (rc)
        return rc;
    k->is_v3 = k->ctx->subkeys->pubkey_algo == GPGME_PK_RSA && strlen (k->ctx->subkeys->fpr) == 32;
    k->is_protected = k->ext->gloflags.is_protected;
    k->keyid = k->ctx->subkeys->keyid;
    k->uid = k->ctx->uids->uid;
    return rc;
}


int
get_pubkey (const char *keyid, gpgme_key_t *ret_key)
{
    int rc = 0;

    if (pub && sec)
        rc = get_key (keyid, ret_key, NULL, 0);
    else
        rc = get_key2 (keyid, ret_key, 0);
    return rc;
} /* get_pubkey */


int
get_seckey( const char *keyid, gpgme_key_t *ret_skey )
{
    int rc = 0;

    if( pub && sec )
        rc = get_key( keyid, ret_skey, NULL, 1 );
    else
        rc = get_key2( keyid, ret_skey, 1 );
    return rc;
} /* get_seckey */


int
count_insecure_elgkeys (void)
{
    gpgme_keycache_t pc;
    gpgme_key_t key;
    int n=0;

    pc = keycache_get_ctx (1);
    if (!pc)
        BUG (0);
    while (!gpgme_keycache_next_key (pc, 0, &key)) {
        if (key->subkeys->pubkey_algo == GPGME_PK_ELG)
            n++;
    }
    gpgme_keycache_rewind (pc);
    return n;
}
1	/* wptGPGME.cpp - WinPT GPGME interface
2	* Copyright (C) 2001-2005 Timo Schulz
3	*
4	* This file is part of WinPT.
5	*
6	* WinPT is free software; you can redistribute it and/or
7	* modify it under the terms of the GNU General Public License
8	* as published by the Free Software Foundation; either version 2
9	* of the License, or (at your option) any later version.
10	*
11	* WinPT is distributed in the hope that it will be useful,
12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14	* General Public License for more details.
15	*
16	* You should have received a copy of the GNU General Public License
17	* along with WinPT; if not, write to the Free Software Foundation,
18	* Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19	*/
20
21	#include <sys/types.h>
22	#include <windows.h>
23
24	#include "../resource.h"
25	#include "wptNLS.h"
26	#include "wptGPG.h"
27	#include "wptErrors.h"
28	#include "wptTypes.h"
29	#include "wptW32API.h"
30	#include "wptVersion.h"
31	#include "wptCommonCtl.h"
32	#include "wptContext.h"
33	#include "wptRegistry.h"
34	#include "wptDlgs.h"
35
36	#include "openpgp.h"
37
38	BOOL CALLBACK keycache_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam);
39	void progress_cleanup (progress_filter_s * pfx);
40
41	static gpgme_keycache_t pub = NULL;
42	static gpgme_keycache_t sec = NULL;
43	static unsigned int reload = 0;
44	static char *gpg_secring = NULL;
45
46	void
47	keycache_reload (HWND dlg)
48	{
49	refresh_cache_s rcs;
50
51	memset (&rcs, 0, sizeof rcs);
52	rcs.kr_reload = rcs.kr_update = 1;
53	rcs.tr_update = 0;
54	DialogBoxParam (glob_hinst, (LPCSTR)IDD_WINPT_KEYCACHE, dlg,
55	keycache_dlg_proc, (LPARAM)&rcs);
56	} /* keycache_reload */
57
58	void
59	keycache_release (void)
60	{
61	int n = gpgme_keycache_count (pub);
62	char tmpbuf[64];
63
64	/* XXX: update the value when the cache has changed. */
65	sprintf (tmpbuf, "%d", n);
66	set_reg_key (HKEY_CURRENT_USER, "Software\\WinPT", "nKeys", tmpbuf);
67
68	if (pub) {
69	gpgme_keycache_release (pub);
70	pub = NULL;
71	}
72	if (sec) {
73	gpgme_keycache_release (sec);
74	sec = NULL;
75	}
76	} /* keycache_release */
77
78
79	gpgme_error_t
80	keycache_update (int is_sec, const char *keyid)
81	{
82	gpgme_keycache_t ctx = pub;
83	gpgme_error_t err;
84
85	if (is_sec)
86	ctx = sec;
87	err = gpgme_keycache_update_key (ctx, is_sec, pub, keyid);
88	if (is_sec)
89	gpgme_keycache_prepare_single (ctx, keyid, NULL, gpg_secring);
90	return err;
91	}
92
93	/* XXX: cache_keyring_names must be called then the GPG homedir changes! */
94
95	gpgme_error_t
96	keycache_init (const char pubring, const char secring)
97	{
98	struct progress_filter_s pfx;
99	gpgme_error_t err;
100	int val = 0;
101	char * p;
102
103	if (secring != NULL) {
104	free_if_alloc (gpg_secring);
105	gpg_secring = get_gnupg_keyring (0, NO_STRICT);
106	}
107
108	if (reload) {
109	keycache_release ();
110	reload = 0;
111	}
112	p = get_reg_entry (HKEY_CURRENT_USER, "Software\\WinPT", "nKeys");
113	if (p && *p != ' ') {
114	val = atoi (p);
115	free_if_alloc (p);
116	memset (&pfx, 0, sizeof (pfx));
117	}
118
119	err = gpgme_keycache_new (&pub);
120	if (err)
121	return err;
122	if (val != 0)
123	gpgme_keycache_set_cb (pub, progress_callback, &pfx, val);
124	err = gpgme_keycache_new (&sec);
125	if (!err)
126	err = gpgme_keycache_init (pub, NULL, 0);
127	if (!err)
128	err = gpgme_keycache_init( sec, NULL, 1 );
129	if( !err && pubring && *pubring )
130	err = gpgme_keycache_prepare( pub, pubring, NULL );
131	if( !err && secring && * secring )
132	err = gpgme_keycache_prepare( sec, NULL, secring );
133	if (!err)
134	gpgme_keycache_sync (pub, sec);
135	if (val != 0)
136	progress_cleanup (&pfx);
137	return err;
138	} /* keycache_init */
139
140
141	void
142	keycache_set_reload( int yes )
143	{
144	reload = yes;
145	} /* keycache_set_reload */
146
147
148	int
149	keycache_get_reload( void )
150	{
151	return reload;
152	} /* keycache_get_reload */
153
154
155	gpgme_keycache_t
156	keycache_get_ctx (int is_pub)
157	{
158	return is_pub? pub : sec;
159	} /* keycache_get_ctx */
160
161
162	static int
163	get_key (const char keyid, gpgme_key_t r_key, struct keycache_s **c, int secret)
164	{
165	gpgme_keycache_t cache;
166	gpgme_error_t err;
167	int mode = secret? KEYCACHE_PRV : KEYCACHE_PUB;
168
169	if( !keyid )
170	return WPTERR_GENERAL;
171	if( r_key )
172	*r_key = NULL;
173	cache = keycache_get_ctx (mode);
174	if( !cache )
175	BUG( NULL );
176	if (!c)
177	err = gpgme_keycache_find_key( cache, keyid, 0, r_key );
178	else
179	err = gpgme_keycache_find_key2 (cache, keyid, 0, r_key, c);
180	if( err )
181	return WPTERR_GENERAL;
182	return 0;
183	} /* get_key */
184
185
186	static int
187	get_key2( const char * keyid, gpgme_key_t * r_key, int secret )
188	{
189	gpg_iobuf_t inp;
190	char * p;
191	int rc;
192
193	p = get_gnupg_keyring (1, !NO_STRICT);
194	if( !p )
195	BUG( NULL );
196
197	inp = gpg_iobuf_open( p );
198	if( !inp ) {
199	const char *s = winpt_strerror( WPTERR_FILE_OPEN );
200	log_box( _("WinPT Error"), 0, "%s: %s", p, s );
201	free_if_alloc( p );
202	return NULL;
203	}
204	gpg_iobuf_ioctl( inp, 3, 1, NULL ); /* disable cache */
205	/rc = gpgme_getkey_bykeyid( inp, keyid, r_key ); XXX /
206	rc=0;
207
208	gpg_iobuf_close( inp );
209	free_if_alloc( p );
210	return rc;
211	} /* get_key2 */
212
213
214	int
215	winpt_get_pubkey (const char keyid, winpt_key_s k)
216	{
217	int rc;
218
219	rc = get_key (keyid, &k->ctx, &k->ext, 0);
220	if (rc)
221	return rc;
222	k->is_v3 = k->ctx->subkeys->pubkey_algo == GPGME_PK_RSA && strlen (k->ctx->subkeys->fpr) == 32;
223	k->is_protected = k->ext->gloflags.is_protected;
224	k->keyid = k->ctx->subkeys->keyid;
225	k->uid = k->ctx->uids->uid;
226	return rc;
227	}
228
229
230	int
231	winpt_get_seckey (const char keyid, winpt_key_s k)
232	{
233	int rc;
234	rc = get_key (keyid, &k->ctx, &k->ext, 1);
235	if (rc)
236	return rc;
237	k->is_v3 = k->ctx->subkeys->pubkey_algo == GPGME_PK_RSA && strlen (k->ctx->subkeys->fpr) == 32;
238	k->is_protected = k->ext->gloflags.is_protected;
239	k->keyid = k->ctx->subkeys->keyid;
240	k->uid = k->ctx->uids->uid;
241	return rc;
242	}
243
244
245	int
246	get_pubkey (const char keyid, gpgme_key_t ret_key)
247	{
248	int rc = 0;
249
250	if (pub && sec)
251	rc = get_key (keyid, ret_key, NULL, 0);
252	else
253	rc = get_key2 (keyid, ret_key, 0);
254	return rc;
255	} /* get_pubkey */
256
257
258	int
259	get_seckey( const char keyid, gpgme_key_t ret_skey )
260	{
261	int rc = 0;
262
263	if( pub && sec )
264	rc = get_key( keyid, ret_skey, NULL, 1 );
265	else
266	rc = get_key2( keyid, ret_skey, 1 );
267	return rc;
268	} /* get_seckey */
269
270
271	int
272	count_insecure_elgkeys (void)
273	{
274	gpgme_keycache_t pc;
275	gpgme_key_t key;
276	int n=0;
277
278	pc = keycache_get_ctx (1);
279	if (!pc)
280	BUG (0);
281	while (!gpgme_keycache_next_key (pc, 0, &key)) {
282	if (key->subkeys->pubkey_algo == GPGME_PK_ELG)
283	n++;
284	}
285	gpgme_keycache_rewind (pc);
286	return n;
287	}