trunk/src/OEProc.c

/* OEProc.c - OE window procedure
 *      Copyright (C) 2001, 2002, 2003, 2006 Timo Schulz
 *
 * This file is part of GPGOE.
 *
 * GPGOE is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 * 
 * GPGOE 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 Lesser General Public License
 * along with GPGOE; 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 <stdio.h>
#include <commctrl.h>

#include "gpgme.h"
#include "GPGOE.h"

/* FIXME: I'm not sure if this still works with the Vista version of Express */

/* Outlook V6.x command IDs. */ 
#define ID_OE_ENCRYPT       40260
#define ID_OE_SIGN          40299
#define ID_OE_SEND          40411
#define ID_OE_PREVMSG       40145
#define ID_OE_NEXTMSG       40146
#define ID_OE_ATTACH        40237

/* Outlook select charset IDs. */
#define ID_OE_CP_UTF8           42540
#define ID_OE_CP_WEUROPE_ISO    42544
#define ID_OE_CP_WEUROPE_WINCP  42545
#define ID_OE_CP_MEUROPE_ISO    42525
#define ID_OE_CP_MEUROPE_WINCP  42526
#define ID_OE_CP_LATIN9         42535

/* Global plugin structure. */
plugin_ctx_t oe_plug;

/* We show a warning for the attachment problem, but only once. */
static int attach_warn_shown = 0;

/* One time initialisation finished?. */
static int plugin_preload_done = 0;

/* Event to indicate if the plugin is active or not. */
HANDLE plugin_active = NULL;


/* Display a warning that the attachments of the mail will be
   sent in cleartext. */
static void
show_attachment_warn (HWND hwnd)
{
    if (attach_warn_shown)
        return;

    MessageBox (hwnd, _("GPGOE is unable to secure attachments.\r\n"
                        "As a result the data attached to this mail is NOT encrypted."),
                _("GPG Plug-in Warning"), MB_ICONWARNING|MB_OK);
    attach_warn_shown = 1;
}


/* Display a warning that the mail will be sent in clear. */
static void
show_plaintext_warn (HWND hwnd, int rc)
{
    const char *fmt, *s;
    char *p;

    fmt = _("WARNING: This message will be sent in cleartext.\r\n\r\n"
            "Error description: %s.");
    s = gpgme_strerror (rc);
    p = xcalloc (1, strlen (fmt)+ strlen (s)+2);
    sprintf (p, fmt, s);
    MessageBox (hwnd, p, _("GPG Plug-in Warning"), MB_ICONWARNING|MB_OK);
    free_if_alloc (p);
}


/* Reset the plugin state. */
static void
plugin_reset (plugin_ctx_t ctx)
{
    if (!ctx)
        return;
    free_if_alloc (ctx->to);
    free_if_alloc (ctx->cc);
    free_if_alloc (ctx->bcc);
    ctx->sign = ctx->encrypt = 0;
    memset (ctx->errbuf, 0, sizeof (ctx->errbuf));
}


/* Release plugin context. */
static void
plugin_release (plugin_ctx_t ctx)
{
    if (!ctx)
        return;
    plugin_reset (ctx);
    free_if_alloc (ctx);
}


/* Create new plugin context. */
static int
plugin_new (plugin_ctx_t *r_ctx)
{    
    plugin_ctx_t ctx;

    *r_ctx = NULL;
    ctx = xcalloc (1, sizeof *ctx);
    *r_ctx = ctx;
    return 0;
}


/* Extract the email from the given edit control @edit
   and return it as a string. */
static char*
get_address_text (HWND main, HWND edit)
{
    AttachThreadInput (GetCurrentThreadId (),
                      GetWindowThreadProcessId (main, NULL),
                      TRUE);
        
    SetFocus (edit);

    AttachThreadInput (GetCurrentThreadId (),
                      GetWindowThreadProcessId (main, NULL),
                      FALSE);

    SendMessage (main, WM_COMMAND, MAKEWPARAM (ID_OE_SELECTALL, 0), 0);
    SendMessage (main, WM_COMMAND, MAKEWPARAM (ID_OE_COPY, 0), 0);
    
    Sleep (200);

    return get_clip_text (NULL);;
}


/* Try to figure out if the message is from the inbox or the outbox folder. */
int
window_is_inbox (HWND to_hwnd)
{
    char wclass[200];

    if (GetDlgCtrlID (to_hwnd) == 1028 &&
        GetClassName (to_hwnd, wclass, sizeof (wclass)-1) > 0
        && !strcmp (wclass, "RichEdit20W"))
        return 1;
    return 0;
}


/* Check if the attach listview control contains at least one entry.
   Return the amount of attachments. */
int
mail_count_attachments (plugin_ctx_t ctx)
{
    if (!ctx->attach)
        return 0;
    return ListView_GetItemCount (ctx->attach);
}


/* Initialize the plugin context:
   Find all windows the plugin need to handle messages.
   The text of the recipients fields is only used when the message comes
   from the outbox, because the inbox fields are readonly. */
int
plugin_init (plugin_ctx_t ctx, HWND main, int is_inbox)
{
    /* store original clipboard text. */
    free_if_alloc (ctx->orig_text);
    ctx->orig_text = get_clip_text (NULL);

    ctx->sign = 0;
    ctx->encrypt = 0;

    ctx->main_wnd = main;
    ctx->addr_wnd = FindWindowEx (main, NULL, "OE_Envelope", NULL);
    ctx->to_wnd = FindWindowEx (ctx->addr_wnd, NULL, "RichEdit20W", NULL);
    ctx->cc_wnd = FindWindowEx (ctx->addr_wnd, ctx->to_wnd, "RichEdit20W", NULL);
    ctx->bcc_wnd = FindWindowEx (ctx->addr_wnd, ctx->cc_wnd, "RichEdit20W", NULL);

    ctx->to = NULL;
    ctx->cc = NULL;
    ctx->bcc = NULL;

    ctx->attach = FindWindowEx (ctx->addr_wnd, NULL, "SysListView32", NULL);

    if (!is_inbox && !window_is_inbox (ctx->to_wnd)) {
        ctx->to = get_address_text (ctx->main_wnd, ctx->to_wnd);
        ctx->cc = get_address_text (ctx->main_wnd, ctx->cc_wnd);
        ctx->bcc = get_address_text (ctx->main_wnd, ctx->bcc_wnd);
    }

    return 0;
}


/* Clear the clipboard contents. */
static void
clear_clipboard (void)
{
    OpenClipboard (NULL);
    EmptyClipboard ();
    CloseClipboard ();
}


/* Restore the original clipboard data from the plugin context. */
static void
restore_clipboard (plugin_ctx_t ctx)
{
    if (!ctx->orig_text)
        return;
    set_clip_text (NULL, ctx->orig_text, strlen (ctx->orig_text));
    free_if_alloc (ctx->orig_text);
}


/* Subclass dialog procedure for the outlook message window. */
LRESULT CALLBACK
oe_proc (HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
{    
    static int initialized = 0;
    static int encr_msg = 0;
    static int sign_msg = 0;
    int rc = 0;

    if (initialized == 0)
        initialized = SetTimer (hwnd, 1, 1000, NULL);

    switch (msg) {
    case WM_CREATE:
        if (!plugin_preload_done) {
            /* We need to load this lib here otherwise the richedit 
               control would not be displayed. */
            LoadLibrary ("RichEd32.Dll");
            setup_gettext();
            plugin_active = CreateEvent (NULL, TRUE, FALSE, NULL);
            plugin_preload_done = 1;
        }
        initialized = 0;
        plugin_new (&oe_plug);
        break;

    case WM_DESTROY:
        ResetEvent (plugin_active);
        plugin_release (oe_plug);
        encr_msg = sign_msg = 0;
        break;

    case WM_TIMER:      
        KillTimer (hwnd, initialized);
        plugin_reset (oe_plug);
        plugin_init (oe_plug, hwnd, 1);
        if (window_is_inbox (oe_plug->to_wnd)) {
            clear_clipboard ();
            rc = oe_handle_mail (oe_plug);
            if (rc)
                show_error (hwnd, _("GPG Plug-in Error"), MB_ICONERROR|MB_OK,
                            _("decrypt/verify: %s\n%s"), 
                            gpgme_strerror (rc), oe_plug->errbuf);
        }
        break;   

    case WM_COMMAND:
        switch (LOWORD (wparam)) {
        case ID_OE_PREVMSG:
        case ID_OE_NEXTMSG:
            SetTimer (hwnd, 1, 1000, NULL);
            break;

        case ID_OE_CP_UTF8:
            oe_plug->use_utf8 = 1;
            break;

        case ID_OE_CP_WEUROPE_ISO:
        case ID_OE_CP_WEUROPE_WINCP:
        case ID_OE_CP_MEUROPE_ISO:
        case ID_OE_CP_MEUROPE_WINCP:
        case ID_OE_CP_LATIN9:
            oe_plug->use_utf8 = 0;
            break;

        case ID_OE_ATTACH:
            break;

        case ID_OE_ENCRYPT:
            encr_msg ^= 1;
            break;

        case ID_OE_SIGN:
            sign_msg ^= 1;
            break;

        case ID_OE_SEND:
            if (encr_msg || sign_msg) {
                plugin_init (oe_plug, hwnd, 0);
                if (encr_msg) {
                    SendMessage (hwnd, WM_COMMAND, MAKEWPARAM (ID_OE_ENCRYPT, 0), 0);
                    oe_plug->encrypt = 1;
                }
                if (sign_msg) {
                    SendMessage (hwnd, WM_COMMAND, MAKEWPARAM (ID_OE_SIGN, 0), 0);
                    oe_plug->sign = 1;
                }
                rc = oe_handle_mail (oe_plug);
                if (rc)
                    show_plaintext_warn (hwnd, rc);
                restore_clipboard (oe_plug);
                if (mail_count_attachments (oe_plug) > 0)
                    show_attachment_warn (hwnd);
            }
            break;
        }
        break;
    }

    return CallWindowProc (oe_proc_old, hwnd, msg, wparam, lparam);
}
1	twoaday	16	/* OEProc.c - OE window procedure
2	twoaday	1	* Copyright (C) 2001, 2002, 2003, 2006 Timo Schulz
3			*
4			* This file is part of GPGOE.
5			*
6			* GPGOE is free software; you can redistribute it and/or modify
7			* it under the terms of the GNU Lesser General Public License as published by
8			* the Free Software Foundation; either version 2.1 of the License, or
9			* (at your option) any later version.
10			*
11			* GPGOE 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 Lesser General Public License
17			* along with GPGOE; if not, write to the Free Software Foundation,
18			* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19			*/
20
21			#ifdef HAVE_CONFIG_H
22			#include <config.h>
23			#endif
24			#include <windows.h>
25			#include <stdio.h>
26	twoaday	11	#include <commctrl.h>
27
28	twoaday	1	#include "gpgme.h"
29			#include "GPGOE.h"
30
31	twoaday	23	/* FIXME: I'm not sure if this still works with the Vista version of Express */
32
33	twoaday	12	/* Outlook V6.x command IDs. */
34			#define ID_OE_ENCRYPT 40260
35			#define ID_OE_SIGN 40299
36			#define ID_OE_SEND 40411
37			#define ID_OE_PREVMSG 40145
38			#define ID_OE_NEXTMSG 40146
39	twoaday	16	#define ID_OE_ATTACH 40237
40	twoaday	11
41	twoaday	12	/* Outlook select charset IDs. */
42			#define ID_OE_CP_UTF8 42540
43			#define ID_OE_CP_WEUROPE_ISO 42544
44			#define ID_OE_CP_WEUROPE_WINCP 42545
45			#define ID_OE_CP_MEUROPE_ISO 42525
46			#define ID_OE_CP_MEUROPE_WINCP 42526
47			#define ID_OE_CP_LATIN9 42535
48
49	twoaday	1	/* Global plugin structure. */
50	twoaday	16	plugin_ctx_t oe_plug;
51	twoaday	1
52	twoaday	11	/* We show a warning for the attachment problem, but only once. */
53			static int attach_warn_shown = 0;
54	twoaday	1
55	twoaday	12	/* One time initialisation finished?. */
56			static int plugin_preload_done = 0;
57	twoaday	11
58	twoaday	12	/* Event to indicate if the plugin is active or not. */
59			HANDLE plugin_active = NULL;
60
61
62	twoaday	11	/* Display a warning that the attachments of the mail will be
63			sent in cleartext. */
64			static void
65			show_attachment_warn (HWND hwnd)
66			{
67			if (attach_warn_shown)
68			return;
69
70			MessageBox (hwnd, _("GPGOE is unable to secure attachments.\r\n"
71			"As a result the data attached to this mail is NOT encrypted."),
72			_("GPG Plug-in Warning"), MB_ICONWARNING\|MB_OK);
73			attach_warn_shown = 1;
74			}
75
76
77	twoaday	1	/* Display a warning that the mail will be sent in clear. */
78			static void
79			show_plaintext_warn (HWND hwnd, int rc)
80			{
81			const char fmt, s;
82			char *p;
83
84	twoaday	11	fmt = _("WARNING: This message will be sent in cleartext.\r\n\r\n"
85			"Error description: %s.");
86	twoaday	1	s = gpgme_strerror (rc);
87			p = xcalloc (1, strlen (fmt)+ strlen (s)+2);
88			sprintf (p, fmt, s);
89			MessageBox (hwnd, p, _("GPG Plug-in Warning"), MB_ICONWARNING\|MB_OK);
90			free_if_alloc (p);
91			}
92
93
94	twoaday	12	/* Reset the plugin state. */
95	twoaday	1	static void
96			plugin_reset (plugin_ctx_t ctx)
97			{
98			if (!ctx)
99			return;
100			free_if_alloc (ctx->to);
101			free_if_alloc (ctx->cc);
102			free_if_alloc (ctx->bcc);
103			ctx->sign = ctx->encrypt = 0;
104			memset (ctx->errbuf, 0, sizeof (ctx->errbuf));
105			}
106
107
108	twoaday	12	/* Release plugin context. */
109	twoaday	1	static void
110			plugin_release (plugin_ctx_t ctx)
111			{
112			if (!ctx)
113			return;
114			plugin_reset (ctx);
115			free_if_alloc (ctx);
116			}
117
118
119	twoaday	12	/* Create new plugin context. */
120	twoaday	1	static int
121			plugin_new (plugin_ctx_t *r_ctx)
122			{
123			plugin_ctx_t ctx;
124
125			*r_ctx = NULL;
126			ctx = xcalloc (1, sizeof *ctx);
127			*r_ctx = ctx;
128			return 0;
129			}
130
131
132			/* Extract the email from the given edit control @edit
133			and return it as a string. */
134			static char*
135			get_address_text (HWND main, HWND edit)
136			{
137			AttachThreadInput (GetCurrentThreadId (),
138			GetWindowThreadProcessId (main, NULL),
139			TRUE);
140
141			SetFocus (edit);
142
143			AttachThreadInput (GetCurrentThreadId (),
144			GetWindowThreadProcessId (main, NULL),
145			FALSE);
146
147	twoaday	12	SendMessage (main, WM_COMMAND, MAKEWPARAM (ID_OE_SELECTALL, 0), 0);
148			SendMessage (main, WM_COMMAND, MAKEWPARAM (ID_OE_COPY, 0), 0);
149	twoaday	1
150			Sleep (200);
151
152			return get_clip_text (NULL);;
153			}
154
155
156	twoaday	16	/* Try to figure out if the message is from the inbox or the outbox folder. */
157	twoaday	1	int
158			window_is_inbox (HWND to_hwnd)
159			{
160	twoaday	16	char wclass[200];
161
162			if (GetDlgCtrlID (to_hwnd) == 1028 &&
163			GetClassName (to_hwnd, wclass, sizeof (wclass)-1) > 0
164			&& !strcmp (wclass, "RichEdit20W"))
165	twoaday	1	return 1;
166			return 0;
167			}
168
169
170	twoaday	11	/* Check if the attach listview control contains at least one entry.
171	twoaday	12	Return the amount of attachments. */
172	twoaday	11	int
173	twoaday	12	mail_count_attachments (plugin_ctx_t ctx)
174	twoaday	11	{
175	twoaday	12	if (!ctx->attach)
176			return 0;
177			return ListView_GetItemCount (ctx->attach);
178	twoaday	11	}
179
180
181	twoaday	1	/* Initialize the plugin context:
182			Find all windows the plugin need to handle messages.
183			The text of the recipients fields is only used when the message comes
184	twoaday	16	from the outbox, because the inbox fields are readonly. */
185	twoaday	1	int
186			plugin_init (plugin_ctx_t ctx, HWND main, int is_inbox)
187			{
188	twoaday	11	/* store original clipboard text. */
189			free_if_alloc (ctx->orig_text);
190			ctx->orig_text = get_clip_text (NULL);
191
192	twoaday	1	ctx->sign = 0;
193			ctx->encrypt = 0;
194
195			ctx->main_wnd = main;
196			ctx->addr_wnd = FindWindowEx (main, NULL, "OE_Envelope", NULL);
197	twoaday	16	ctx->to_wnd = FindWindowEx (ctx->addr_wnd, NULL, "RichEdit20W", NULL);
198			ctx->cc_wnd = FindWindowEx (ctx->addr_wnd, ctx->to_wnd, "RichEdit20W", NULL);
199			ctx->bcc_wnd = FindWindowEx (ctx->addr_wnd, ctx->cc_wnd, "RichEdit20W", NULL);
200
201	twoaday	1	ctx->to = NULL;
202			ctx->cc = NULL;
203			ctx->bcc = NULL;
204
205	twoaday	12	ctx->attach = FindWindowEx (ctx->addr_wnd, NULL, "SysListView32", NULL);
206
207	twoaday	1	if (!is_inbox && !window_is_inbox (ctx->to_wnd)) {
208			ctx->to = get_address_text (ctx->main_wnd, ctx->to_wnd);
209			ctx->cc = get_address_text (ctx->main_wnd, ctx->cc_wnd);
210			ctx->bcc = get_address_text (ctx->main_wnd, ctx->bcc_wnd);
211			}
212
213			return 0;
214			}
215
216
217	twoaday	11	/* Clear the clipboard contents. */
218	twoaday	1	static void
219			clear_clipboard (void)
220			{
221			OpenClipboard (NULL);
222			EmptyClipboard ();
223			CloseClipboard ();
224			}
225
226
227	twoaday	12	/* Restore the original clipboard data from the plugin context. */
228	twoaday	11	static void
229			restore_clipboard (plugin_ctx_t ctx)
230			{
231			if (!ctx->orig_text)
232			return;
233			set_clip_text (NULL, ctx->orig_text, strlen (ctx->orig_text));
234			free_if_alloc (ctx->orig_text);
235			}
236
237
238	twoaday	1	/* Subclass dialog procedure for the outlook message window. */
239			LRESULT CALLBACK
240			oe_proc (HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
241			{
242	twoaday	4	static int initialized = 0;
243	twoaday	1	static int encr_msg = 0;
244			static int sign_msg = 0;
245			int rc = 0;
246
247			if (initialized == 0)
248			initialized = SetTimer (hwnd, 1, 1000, NULL);
249
250			switch (msg) {
251	twoaday	11	case WM_CREATE:
252	twoaday	12	if (!plugin_preload_done) {
253	twoaday	23	/* We need to load this lib here otherwise the richedit
254	twoaday	12	control would not be displayed. */
255			LoadLibrary ("RichEd32.Dll");
256			setup_gettext();
257	twoaday	16	plugin_active = CreateEvent (NULL, TRUE, FALSE, NULL);
258	twoaday	12	plugin_preload_done = 1;
259			}
260	twoaday	1	initialized = 0;
261			plugin_new (&oe_plug);
262			break;
263
264			case WM_DESTROY:
265	twoaday	12	ResetEvent (plugin_active);
266	twoaday	1	plugin_release (oe_plug);
267	twoaday	11	encr_msg = sign_msg = 0;
268	twoaday	1	break;
269
270	twoaday	11	case WM_TIMER:
271	twoaday	1	KillTimer (hwnd, initialized);
272			plugin_reset (oe_plug);
273			plugin_init (oe_plug, hwnd, 1);
274			if (window_is_inbox (oe_plug->to_wnd)) {
275	twoaday	11	clear_clipboard ();
276	twoaday	1	rc = oe_handle_mail (oe_plug);
277			if (rc)
278			show_error (hwnd, _("GPG Plug-in Error"), MB_ICONERROR\|MB_OK,
279			_("decrypt/verify: %s\n%s"),
280			gpgme_strerror (rc), oe_plug->errbuf);
281			}
282	twoaday	12	break;
283	twoaday	1
284	twoaday	12	case WM_COMMAND:
285			switch (LOWORD (wparam)) {
286			case ID_OE_PREVMSG:
287			case ID_OE_NEXTMSG:
288			SetTimer (hwnd, 1, 1000, NULL);
289			break;
290	twoaday	1
291	twoaday	12	case ID_OE_CP_UTF8:
292			oe_plug->use_utf8 = 1;
293			break;
294	twoaday	1
295	twoaday	12	case ID_OE_CP_WEUROPE_ISO:
296			case ID_OE_CP_WEUROPE_WINCP:
297			case ID_OE_CP_MEUROPE_ISO:
298			case ID_OE_CP_MEUROPE_WINCP:
299			case ID_OE_CP_LATIN9:
300			oe_plug->use_utf8 = 0;
301			break;
302	twoaday	1
303	twoaday	16	case ID_OE_ATTACH:
304			break;
305
306	twoaday	12	case ID_OE_ENCRYPT:
307			encr_msg ^= 1;
308			break;
309
310			case ID_OE_SIGN:
311			sign_msg ^= 1;
312			break;
313
314			case ID_OE_SEND:
315			if (encr_msg \|\| sign_msg) {
316			plugin_init (oe_plug, hwnd, 0);
317			if (encr_msg) {
318			SendMessage (hwnd, WM_COMMAND, MAKEWPARAM (ID_OE_ENCRYPT, 0), 0);
319			oe_plug->encrypt = 1;
320			}
321			if (sign_msg) {
322			SendMessage (hwnd, WM_COMMAND, MAKEWPARAM (ID_OE_SIGN, 0), 0);
323			oe_plug->sign = 1;
324			}
325			rc = oe_handle_mail (oe_plug);
326			if (rc)
327			show_plaintext_warn (hwnd, rc);
328			restore_clipboard (oe_plug);
329			if (mail_count_attachments (oe_plug) > 0)
330			show_attachment_warn (hwnd);
331			}
332			break;
333			}
334			break;
335	twoaday	1	}
336
337			return CallWindowProc (oe_proc_old, hwnd, msg, wparam, lparam);
338			}