trunk/Src/wptClipSignEncDlg.cpp

/* wptSignEncDlg.cpp - Sign & encrypt dialog
 *      Copyright (C) 2000-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 <windows.h>
#include <commctrl.h>

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


/* Encrypt the clipboard data with the recipients from @rset and
   additionally sign the data before @signer as the keyID.
   Return value: 0 on success. */
gpgme_error_t
gpg_clip_sign_encrypt (struct passphrase_cb_s *cb, const char *signer, 
                       gpgme_key_t *rset, int opts)
{
    gpgme_error_t err;
    gpgme_data_t plain = NULL;
    gpgme_data_t ciph = NULL;
    gpgme_key_t key = NULL;
    gpgme_ctx_t ctx;
 
    if (!signer)
        return gpg_error (GPG_ERR_INV_ARG);
    if (get_pubkey (signer, &key))
        return gpg_error (GPG_ERR_NO_PUBKEY);
    
    err = gpgme_new (&ctx);
    if (err)
        return err;

    /* Update the gpgme context manually. */
    cb->gpg = ctx;

    gpgme_set_armor (ctx, 1);
    
    err = gpg_data_new_from_clipboard (&plain, 0);
    if (err)
        goto leave;   
    err = gpgme_data_new (&ciph);
    if (err)
        goto leave;
    err = gpgme_signers_add (ctx, key);
    if (err)
        goto leave;

    err = gpgme_op_encrypt_sign (ctx, rset, 
                                 opts? GPGME_ENCRYPT_ALWAYS_TRUST : (gpgme_encrypt_flags_t)0,
                                 plain, ciph);
    if (err)
        goto leave;

    gpg_data_release_and_set_clipboard (ciph, 1);
    ciph = NULL;

leave:
    if (plain)
        gpgme_data_release (plain);
    if (ciph)
        gpgme_data_release (ciph);
    gpgme_key_release (key);
    gpgme_release (ctx);
    return err;
}


/* Dialog procedure for clipboard sign + encrypt. */
BOOL CALLBACK 
clip_signenc_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
{
    static listview_ctrl_t lv = NULL;
    static keylist_t list = NULL;
    static int enable = 0;
    gpgme_keycache_t kc;
    gpgme_key_t *rset;
    gpgme_error_t err;
    passphrase_cb_s pwd;
    char * signer = NULL;
    int force_trust = 0, opt = 0;
    int kmode = reg_prefs.keylist_mode? KEYLIST_ENCRYPT_MIN : KEYLIST_ENCRYPT;
    int n;
    
    switch( msg ) {
    case WM_INITDIALOG:
        enable = 0;
    #ifndef LANG_DE
        SetWindowText( dlg, _("Sign & Encrypt") );
    #endif
        kc = keycache_get_ctx( KEYCACHE_PUB );
        if( !kc )
            BUG( NULL );
        lv = keylist_load( GetDlgItem( dlg, IDC_SIGNENC_KEYLIST ), 
                           kc, NULL, kmode, KEY_SORT_USERID);
        seclist_init( dlg, IDC_SIGNENC_SECLIST, KEYLIST_FLAG_SHORT, &list );
        center_window (dlg, NULL);
        set_active_window (dlg);
        EnableWindow (GetDlgItem (dlg, IDC_SIGNENC_SECLIST), FALSE);
        SetDlgItemText (dlg, IDC_SIGNENC_SELKEY, _("Select key for signing"));
        SetDlgItemText (dlg, IDC_SIGNENC_SECLISTINF, _("Signing key:"));
        SetForegroundWindow (dlg);
        return TRUE;
        
    case WM_DESTROY:
        seclist_destroy (&list);
        reset_active_window ();
        if (lv) {
            keylist_delete( lv );
            lv = NULL;
        }
        return FALSE;
        
    case WM_NOTIFY:
        NMHDR * notify;
        notify = (NMHDR *)lparam;
        if( notify && notify->code == NM_DBLCLK
            && notify->idFrom == IDC_SIGNENC_KEYLIST )
            PostMessage( dlg, WM_COMMAND, MAKEWPARAM(IDOK, 0), NULL );
        return TRUE;
        
    case WM_SYSCOMMAND:
        if( LOWORD (wparam) == SC_CLOSE )
            EndDialog( dlg, TRUE );
        return FALSE;
        
    case WM_COMMAND:
        /* fixme: the enable seems to have a sync problem */
        if( HIWORD( wparam ) == BN_CLICKED 
            && LOWORD( wparam ) == IDC_SIGNENC_SELKEY ) {
            enable ^= 1;
            EnableWindow( GetDlgItem( dlg, IDC_SIGNENC_SECLIST ), enable? TRUE : FALSE );
        }
        switch( LOWORD( wparam ) ) {
        case IDOK:
            rset = keylist_get_recipients( lv, &force_trust, &n );
            if (!n) {
                msg_box( dlg, _("You must select at least one key."), _("Sign & Encrypt"), MB_ERR );
                return FALSE;
            }
            if( IsDlgButtonChecked( dlg, IDC_SIGNENC_SELKEY ) ) {
                gpgme_key_t key;
                const char * s;

                if( seclist_select_key( dlg, IDC_SIGNENC_SECLIST, &key ) ) {
                    msg_box( dlg, _("No key was selected."), _("Signing"), MB_ERR );
                    return FALSE;
                }
                s = key->subkeys->keyid;
                if (s)
                    signer = m_strdup(s+8);
            }
            else {
                signer = get_gnupg_default_key ();
                if (!signer) {
                    msg_box (dlg, _("Could not get default key."), _("Signing"), MB_ERR);
                    return FALSE;
                }
            }
                
            set_gpg_passphrase_cb (&pwd, NULL, GPG_CMD_SIGN, dlg, _("Sign & Encrypt"));
            err = gpg_clip_sign_encrypt (&pwd, signer, rset, force_trust);
            release_gpg_passphrase_cb (&pwd);
            free (rset);
            free_if_alloc (signer);
            if (err == gpg_error (GPG_ERR_BAD_PASSPHRASE))
                agent_del_cache (pwd.keyid);
            if (err) {
                msg_box (dlg, gpgme_strerror (err), _("Sign & Encrypt"), MB_ERR );
                return FALSE;
            }
            show_msg (dlg, 1500, _("GnuPG Status: Finished"));
            EndDialog (dlg, TRUE);
            return TRUE;
            
        case IDCANCEL:
            EndDialog( dlg, FALSE );
            return FALSE;
        }
        break;
    }
    
    return FALSE;
}
1	twoaday	2	/* wptSignEncDlg.cpp - Sign & encrypt dialog
2	twoaday	12	* Copyright (C) 2000-2005 Timo Schulz
3	twoaday	2	*
4			* This file is part of WinPT.
5			*
6			* WinPT is free software; you can redistribute it and/or modify
7			* it under the terms of the GNU General Public License as published by
8			* the Free Software Foundation; either version 2 of the License, or
9			* (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
14			* GNU 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 <windows.h>
22			#include <commctrl.h>
23
24			#include "../resource.h"
25			#include "wptErrors.h"
26			#include "wptAgent.h"
27			#include "wptGPG.h"
28			#include "wptCommonCtl.h"
29			#include "wptKeylist.h"
30			#include "wptTypes.h"
31			#include "wptNLS.h"
32			#include "wptContext.h" /* for passphrase_s */
33			#include "wptDlgs.h"
34			#include "wptW32API.h"
35			#include "wptKeylist.h"
36			#include "wptVersion.h"
37			#include "wptGPG.h"
38			#include "wptRegistry.h"
39			#include "wptUTF8.h"
40
41
42	twoaday	23	/* Encrypt the clipboard data with the recipients from @rset and
43			additionally sign the data before @signer as the keyID.
44			Return value: 0 on success. */
45			gpgme_error_t
46			gpg_clip_sign_encrypt (struct passphrase_cb_s cb, const char signer,
47			gpgme_key_t *rset, int opts)
48			{
49			gpgme_error_t err;
50			gpgme_data_t plain = NULL;
51			gpgme_data_t ciph = NULL;
52			gpgme_key_t key = NULL;
53			gpgme_ctx_t ctx;
54
55			if (!signer)
56			return gpg_error (GPG_ERR_INV_ARG);
57			if (get_pubkey (signer, &key))
58			return gpg_error (GPG_ERR_NO_PUBKEY);
59
60			err = gpgme_new (&ctx);
61			if (err)
62			return err;
63
64			/* Update the gpgme context manually. */
65			cb->gpg = ctx;
66
67			gpgme_set_armor (ctx, 1);
68
69			err = gpg_data_new_from_clipboard (&plain, 0);
70			if (err)
71			goto leave;
72			err = gpgme_data_new (&ciph);
73			if (err)
74			goto leave;
75			err = gpgme_signers_add (ctx, key);
76			if (err)
77			goto leave;
78
79			err = gpgme_op_encrypt_sign (ctx, rset,
80			opts? GPGME_ENCRYPT_ALWAYS_TRUST : (gpgme_encrypt_flags_t)0,
81			plain, ciph);
82			if (err)
83			goto leave;
84
85			gpg_data_release_and_set_clipboard (ciph, 1);
86			ciph = NULL;
87
88			leave:
89			if (plain)
90			gpgme_data_release (plain);
91			if (ciph)
92			gpgme_data_release (ciph);
93			gpgme_key_release (key);
94			gpgme_release (ctx);
95			return err;
96			}
97
98
99			/* Dialog procedure for clipboard sign + encrypt. */
100	twoaday	2	BOOL CALLBACK
101	twoaday	12	clip_signenc_dlg_proc (HWND dlg, UINT msg, WPARAM wparam, LPARAM lparam)
102	twoaday	2	{
103			static listview_ctrl_t lv = NULL;
104			static keylist_t list = NULL;
105			static int enable = 0;
106			gpgme_keycache_t kc;
107	twoaday	23	gpgme_key_t *rset;
108	twoaday	2	gpgme_error_t err;
109			passphrase_cb_s pwd;
110			char * signer = NULL;
111			int force_trust = 0, opt = 0;
112			int kmode = reg_prefs.keylist_mode? KEYLIST_ENCRYPT_MIN : KEYLIST_ENCRYPT;
113	twoaday	23	int n;
114	twoaday	2
115			switch( msg ) {
116			case WM_INITDIALOG:
117	twoaday	12	enable = 0;
118	twoaday	23	#ifndef LANG_DE
119	twoaday	2	SetWindowText( dlg, _("Sign & Encrypt") );
120	twoaday	23	#endif
121	twoaday	2	kc = keycache_get_ctx( KEYCACHE_PUB );
122			if( !kc )
123			BUG( NULL );
124			lv = keylist_load( GetDlgItem( dlg, IDC_SIGNENC_KEYLIST ),
125	twoaday	23	kc, NULL, kmode, KEY_SORT_USERID);
126	twoaday	2	seclist_init( dlg, IDC_SIGNENC_SECLIST, KEYLIST_FLAG_SHORT, &list );
127	twoaday	23	center_window (dlg, NULL);
128	twoaday	12	set_active_window (dlg);
129			EnableWindow (GetDlgItem (dlg, IDC_SIGNENC_SECLIST), FALSE);
130	twoaday	2	SetDlgItemText (dlg, IDC_SIGNENC_SELKEY, _("Select key for signing"));
131			SetDlgItemText (dlg, IDC_SIGNENC_SECLISTINF, _("Signing key:"));
132	twoaday	12	SetForegroundWindow (dlg);
133	twoaday	2	return TRUE;
134
135			case WM_DESTROY:
136	twoaday	23	seclist_destroy (&list);
137			reset_active_window ();
138			if (lv) {
139	twoaday	2	keylist_delete( lv );
140			lv = NULL;
141			}
142			return FALSE;
143
144			case WM_NOTIFY:
145			NMHDR * notify;
146			notify = (NMHDR *)lparam;
147			if( notify && notify->code == NM_DBLCLK
148			&& notify->idFrom == IDC_SIGNENC_KEYLIST )
149			PostMessage( dlg, WM_COMMAND, MAKEWPARAM(IDOK, 0), NULL );
150			return TRUE;
151
152			case WM_SYSCOMMAND:
153			if( LOWORD (wparam) == SC_CLOSE )
154			EndDialog( dlg, TRUE );
155			return FALSE;
156
157			case WM_COMMAND:
158			/* fixme: the enable seems to have a sync problem */
159			if( HIWORD( wparam ) == BN_CLICKED
160			&& LOWORD( wparam ) == IDC_SIGNENC_SELKEY ) {
161			enable ^= 1;
162			EnableWindow( GetDlgItem( dlg, IDC_SIGNENC_SECLIST ), enable? TRUE : FALSE );
163			}
164			switch( LOWORD( wparam ) ) {
165			case IDOK:
166	twoaday	23	rset = keylist_get_recipients( lv, &force_trust, &n );
167			if (!n) {
168	twoaday	2	msg_box( dlg, _("You must select at least one key."), _("Sign & Encrypt"), MB_ERR );
169			return FALSE;
170			}
171			if( IsDlgButtonChecked( dlg, IDC_SIGNENC_SELKEY ) ) {
172			gpgme_key_t key;
173			const char * s;
174
175			if( seclist_select_key( dlg, IDC_SIGNENC_SECLIST, &key ) ) {
176			msg_box( dlg, _("No key was selected."), _("Signing"), MB_ERR );
177			return FALSE;
178			}
179	twoaday	23	s = key->subkeys->keyid;
180			if (s)
181			signer = m_strdup(s+8);
182	twoaday	2	}
183			else {
184	twoaday	23	signer = get_gnupg_default_key ();
185			if (!signer) {
186			msg_box (dlg, _("Could not get default key."), _("Signing"), MB_ERR);
187	twoaday	2	return FALSE;
188			}
189			}
190	twoaday	23
191			set_gpg_passphrase_cb (&pwd, NULL, GPG_CMD_SIGN, dlg, _("Sign & Encrypt"));
192			err = gpg_clip_sign_encrypt (&pwd, signer, rset, force_trust);
193			release_gpg_passphrase_cb (&pwd);
194			free (rset);
195			free_if_alloc (signer);
196			if (err == gpg_error (GPG_ERR_BAD_PASSPHRASE))
197			agent_del_cache (pwd.keyid);
198			if (err) {
199			msg_box (dlg, gpgme_strerror (err), _("Sign & Encrypt"), MB_ERR );
200	twoaday	2	return FALSE;
201			}
202	twoaday	23	show_msg (dlg, 1500, _("GnuPG Status: Finished"));
203			EndDialog (dlg, TRUE);
204	twoaday	2	return TRUE;
205
206			case IDCANCEL:
207			EndDialog( dlg, FALSE );
208			return FALSE;
209			}
210			break;
211			}
212
213			return FALSE;
214	twoaday	23	}