setupapi/tests: Add tests to show that SetupIterateCabinet handles callback exception...
[wine] / dlls / mshtml / nsevents.c
1 /*
2  * Copyright 2007-2008 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "config.h"
20
21 #include <stdarg.h>
22
23 #define COBJMACROS
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winuser.h"
28 #include "ole2.h"
29 #include "mshtmcid.h"
30 #include "shlguid.h"
31
32 #include "wine/debug.h"
33 #include "wine/unicode.h"
34
35 #include "mshtml_private.h"
36 #include "htmlevent.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
39
40 typedef struct {
41     const nsIDOMEventListenerVtbl      *lpDOMEventListenerVtbl;
42     nsDocumentEventListener *This;
43 } nsEventListener;
44
45 struct nsDocumentEventListener {
46     nsEventListener blur_listener;
47     nsEventListener focus_listener;
48     nsEventListener keypress_listener;
49     nsEventListener load_listener;
50     nsEventListener htmlevent_listener;
51
52     LONG ref;
53
54     HTMLDocumentNode *doc;
55 };
56
57 static LONG release_listener(nsDocumentEventListener *This)
58 {
59     LONG ref = InterlockedDecrement(&This->ref);
60
61     TRACE("(%p) ref=%d\n", This, ref);
62
63     if(!ref)
64         heap_free(This);
65
66     return ref;
67 }
68
69 #define NSEVENTLIST_THIS(iface) DEFINE_THIS(nsEventListener, DOMEventListener, iface)
70
71 static nsresult NSAPI nsDOMEventListener_QueryInterface(nsIDOMEventListener *iface,
72                                                         nsIIDRef riid, nsQIResult result)
73 {
74     nsEventListener *This = NSEVENTLIST_THIS(iface);
75
76     *result = NULL;
77
78     if(IsEqualGUID(&IID_nsISupports, riid)) {
79         TRACE("(%p)->(IID_nsISupports, %p)\n", This, result);
80         *result = NSEVENTLIST(This);
81     }else if(IsEqualGUID(&IID_nsIDOMEventListener, riid)) {
82         TRACE("(%p)->(IID_nsIDOMEventListener %p)\n", This, result);
83         *result = NSEVENTLIST(This);
84     }
85
86     if(*result) {
87         nsIWebBrowserChrome_AddRef(NSEVENTLIST(This));
88         return NS_OK;
89     }
90
91     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), result);
92     return NS_NOINTERFACE;
93 }
94
95 static nsrefcnt NSAPI nsDOMEventListener_AddRef(nsIDOMEventListener *iface)
96 {
97     nsDocumentEventListener *This = NSEVENTLIST_THIS(iface)->This;
98     LONG ref = InterlockedIncrement(&This->ref);
99
100     TRACE("(%p) ref=%d\n", This, ref);
101
102     return ref;
103 }
104
105 static nsrefcnt NSAPI nsDOMEventListener_Release(nsIDOMEventListener *iface)
106 {
107     nsDocumentEventListener *This = NSEVENTLIST_THIS(iface)->This;
108
109     return release_listener(This);
110 }
111
112 static BOOL is_doc_child_focus(NSContainer *nscontainer)
113 {
114     HWND hwnd;
115
116     for(hwnd = GetFocus(); hwnd && hwnd != nscontainer->hwnd; hwnd = GetParent(hwnd));
117
118     return hwnd != NULL;
119 }
120
121 static nsresult NSAPI handle_blur(nsIDOMEventListener *iface, nsIDOMEvent *event)
122 {
123     HTMLDocumentNode *doc = NSEVENTLIST_THIS(iface)->This->doc;
124     HTMLDocumentObj *doc_obj;
125
126     TRACE("(%p)\n", doc);
127
128     if(!doc || !doc->basedoc.doc_obj)
129         return NS_ERROR_FAILURE;
130     doc_obj = doc->basedoc.doc_obj;
131
132     if(doc_obj->focus && !is_doc_child_focus(doc_obj->nscontainer)) {
133         doc_obj->focus = FALSE;
134         notif_focus(doc_obj);
135     }
136
137     return NS_OK;
138 }
139
140 static nsresult NSAPI handle_focus(nsIDOMEventListener *iface, nsIDOMEvent *event)
141 {
142     HTMLDocumentNode *doc = NSEVENTLIST_THIS(iface)->This->doc;
143     HTMLDocumentObj *doc_obj;
144
145     TRACE("(%p)\n", doc);
146
147     if(!doc)
148         return NS_ERROR_FAILURE;
149     doc_obj = doc->basedoc.doc_obj;
150
151     if(!doc_obj->focus) {
152         doc_obj->focus = TRUE;
153         notif_focus(doc_obj);
154     }
155
156     return NS_OK;
157 }
158
159 static nsresult NSAPI handle_keypress(nsIDOMEventListener *iface,
160         nsIDOMEvent *event)
161 {
162     HTMLDocumentNode *doc = NSEVENTLIST_THIS(iface)->This->doc;
163     HTMLDocumentObj *doc_obj;
164
165     if(!doc)
166         return NS_ERROR_FAILURE;
167     doc_obj = doc->basedoc.doc_obj;
168
169     TRACE("(%p)->(%p)\n", doc, event);
170
171     update_doc(&doc_obj->basedoc, UPDATE_UI);
172     if(doc_obj->usermode == EDITMODE)
173         handle_edit_event(&doc_obj->basedoc, event);
174
175     return NS_OK;
176 }
177
178 static void handle_docobj_load(HTMLDocumentObj *doc)
179 {
180     IOleCommandTarget *olecmd = NULL;
181     HRESULT hres;
182
183     if(!doc->client)
184         return;
185
186     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
187     if(SUCCEEDED(hres)) {
188         if(doc->download_state) {
189             VARIANT state, progress;
190
191             V_VT(&progress) = VT_I4;
192             V_I4(&progress) = 0;
193             IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS,
194                     OLECMDEXECOPT_DONTPROMPTUSER, &progress, NULL);
195
196             V_VT(&state) = VT_I4;
197             V_I4(&state) = 0;
198             IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETDOWNLOADSTATE,
199                     OLECMDEXECOPT_DONTPROMPTUSER, &state, NULL);
200         }
201
202         IOleCommandTarget_Exec(olecmd, &CGID_ShellDocView, 103, 0, NULL, NULL);
203         IOleCommandTarget_Exec(olecmd, &CGID_MSHTML, IDM_PARSECOMPLETE, 0, NULL, NULL);
204         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_HTTPEQUIV_DONE, 0, NULL, NULL);
205
206         IOleCommandTarget_Release(olecmd);
207     }
208     doc->download_state = 0;
209 }
210
211 static nsresult NSAPI handle_load(nsIDOMEventListener *iface, nsIDOMEvent *event)
212 {
213     HTMLDocumentNode *doc = NSEVENTLIST_THIS(iface)->This->doc;
214     HTMLDocumentObj *doc_obj;
215     nsIDOMHTMLElement *nsbody = NULL;
216
217     TRACE("(%p)\n", doc);
218
219     if(!doc || !doc->basedoc.window)
220         return NS_ERROR_FAILURE;
221     doc_obj = doc->basedoc.doc_obj;
222
223     connect_scripts(doc->basedoc.window);
224
225     if(doc_obj->nscontainer->editor_controller) {
226         nsIController_Release(doc_obj->nscontainer->editor_controller);
227         doc_obj->nscontainer->editor_controller = NULL;
228     }
229
230     if(doc_obj->usermode == EDITMODE)
231         handle_edit_load(&doc_obj->basedoc);
232
233     if(doc == doc_obj->basedoc.doc_node)
234         handle_docobj_load(doc_obj);
235
236     set_ready_state(doc->basedoc.window, READYSTATE_COMPLETE);
237
238     if(doc == doc_obj->basedoc.doc_node) {
239         if(doc_obj->view_sink)
240             IAdviseSink_OnViewChange(doc_obj->view_sink, DVASPECT_CONTENT, -1);
241
242         if(doc_obj->frame) {
243             static const WCHAR wszDone[] = {'D','o','n','e',0};
244             IOleInPlaceFrame_SetStatusText(doc_obj->frame, wszDone);
245         }
246
247         update_title(doc_obj);
248     }
249
250     if(!doc->nsdoc) {
251         ERR("NULL nsdoc\n");
252         return NS_ERROR_FAILURE;
253     }
254
255     nsIDOMHTMLDocument_GetBody(doc->nsdoc, &nsbody);
256     if(nsbody) {
257         fire_event(doc, EVENTID_LOAD, TRUE, (nsIDOMNode*)nsbody, event);
258         nsIDOMHTMLElement_Release(nsbody);
259     }
260
261     return NS_OK;
262 }
263
264 static nsresult NSAPI handle_htmlevent(nsIDOMEventListener *iface, nsIDOMEvent *event)
265 {
266     HTMLDocumentNode *doc = NSEVENTLIST_THIS(iface)->This->doc;
267     const PRUnichar *type;
268     nsIDOMEventTarget *event_target;
269     nsIDOMNode *nsnode;
270     nsAString type_str;
271     eventid_t eid;
272     nsresult nsres;
273
274     TRACE("\n");
275
276     nsAString_Init(&type_str, NULL);
277     nsIDOMEvent_GetType(event, &type_str);
278     nsAString_GetData(&type_str, &type);
279     eid = str_to_eid(type);
280     nsAString_Finish(&type_str);
281
282     nsres = nsIDOMEvent_GetTarget(event, &event_target);
283     if(NS_FAILED(nsres) || !event_target) {
284         ERR("GetEventTarget failed: %08x\n", nsres);
285         return NS_OK;
286     }
287
288     nsres = nsIDOMEventTarget_QueryInterface(event_target, &IID_nsIDOMNode, (void**)&nsnode);
289     nsIDOMEventTarget_Release(event_target);
290     if(NS_FAILED(nsres)) {
291         ERR("Could not get nsIDOMNode: %08x\n", nsres);
292         return NS_OK;
293     }
294
295     fire_event(doc, eid, TRUE, nsnode, event);
296
297     nsIDOMNode_Release(nsnode);
298
299     return NS_OK;
300 }
301
302 #undef NSEVENTLIST_THIS
303
304 #define EVENTLISTENER_VTBL(handler) \
305     { \
306         nsDOMEventListener_QueryInterface, \
307         nsDOMEventListener_AddRef, \
308         nsDOMEventListener_Release, \
309         handler, \
310     }
311
312 static const nsIDOMEventListenerVtbl blur_vtbl =      EVENTLISTENER_VTBL(handle_blur);
313 static const nsIDOMEventListenerVtbl focus_vtbl =     EVENTLISTENER_VTBL(handle_focus);
314 static const nsIDOMEventListenerVtbl keypress_vtbl =  EVENTLISTENER_VTBL(handle_keypress);
315 static const nsIDOMEventListenerVtbl load_vtbl =      EVENTLISTENER_VTBL(handle_load);
316 static const nsIDOMEventListenerVtbl htmlevent_vtbl = EVENTLISTENER_VTBL(handle_htmlevent);
317
318 static void init_event(nsIDOMEventTarget *target, const PRUnichar *type,
319         nsIDOMEventListener *listener, BOOL capture)
320 {
321     nsAString type_str;
322     nsresult nsres;
323
324     nsAString_InitDepend(&type_str, type);
325     nsres = nsIDOMEventTarget_AddEventListener(target, &type_str, listener, capture);
326     nsAString_Finish(&type_str);
327     if(NS_FAILED(nsres))
328         ERR("AddEventTarget failed: %08x\n", nsres);
329
330 }
331
332 static void init_listener(nsEventListener *This, nsDocumentEventListener *listener,
333         const nsIDOMEventListenerVtbl *vtbl)
334 {
335     This->lpDOMEventListenerVtbl = vtbl;
336     This->This = listener;
337 }
338
339 void add_nsevent_listener(HTMLDocumentNode *doc, nsIDOMNode *nsnode, LPCWSTR type)
340 {
341     nsIDOMEventTarget *target;
342     nsresult nsres;
343
344     if(nsnode)
345         nsres = nsIDOMNode_QueryInterface(nsnode, &IID_nsIDOMEventTarget, (void**)&target);
346     else
347         nsres = nsIDOMWindow_QueryInterface(doc->basedoc.window->nswindow, &IID_nsIDOMEventTarget, (void**)&target);
348     if(NS_FAILED(nsres)) {
349         ERR("Could not get nsIDOMEventTarget interface: %08x\n", nsres);
350         return;
351     }
352
353     init_event(target, type, NSEVENTLIST(&doc->nsevent_listener->htmlevent_listener), TRUE);
354     nsIDOMEventTarget_Release(target);
355 }
356
357 void release_nsevents(HTMLDocumentNode *doc)
358 {
359     if(doc->nsevent_listener) {
360         doc->nsevent_listener->doc = NULL;
361         release_listener(doc->nsevent_listener);
362         doc->nsevent_listener = NULL;
363     }
364 }
365
366 void init_nsevents(HTMLDocumentNode *doc)
367 {
368     nsDocumentEventListener *listener;
369     nsIDOMEventTarget *target;
370     nsresult nsres;
371
372     static const PRUnichar wsz_blur[]      = {'b','l','u','r',0};
373     static const PRUnichar wsz_focus[]     = {'f','o','c','u','s',0};
374     static const PRUnichar wsz_keypress[]  = {'k','e','y','p','r','e','s','s',0};
375     static const PRUnichar wsz_load[]      = {'l','o','a','d',0};
376
377     listener = heap_alloc(sizeof(nsDocumentEventListener));
378     if(!listener)
379         return;
380
381     listener->ref = 1;
382     listener->doc = doc;
383
384     init_listener(&listener->blur_listener,        listener, &blur_vtbl);
385     init_listener(&listener->focus_listener,       listener, &focus_vtbl);
386     init_listener(&listener->keypress_listener,    listener, &keypress_vtbl);
387     init_listener(&listener->load_listener,        listener, &load_vtbl);
388     init_listener(&listener->htmlevent_listener,   listener, &htmlevent_vtbl);
389
390     doc->nsevent_listener = listener;
391
392     nsres = nsIDOMWindow_QueryInterface(doc->basedoc.window->nswindow, &IID_nsIDOMEventTarget, (void**)&target);
393     if(NS_FAILED(nsres)) {
394         ERR("Could not get nsIDOMEventTarget interface: %08x\n", nsres);
395         return;
396     }
397
398     init_event(target, wsz_blur,       NSEVENTLIST(&listener->blur_listener),        TRUE);
399     init_event(target, wsz_focus,      NSEVENTLIST(&listener->focus_listener),       TRUE);
400     init_event(target, wsz_keypress,   NSEVENTLIST(&listener->keypress_listener),    FALSE);
401     init_event(target, wsz_load,       NSEVENTLIST(&listener->load_listener),        TRUE);
402
403     nsIDOMEventTarget_Release(target);
404 }