msxml3: Add a function to initialize xmlDoc->_private data.
[wine] / dlls / mshtml / task.c
1 /*
2  * Copyright 2006 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 #include <stdio.h>
23
24 #define COBJMACROS
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "ole2.h"
30
31 #include "wine/debug.h"
32
33 #include "mshtml_private.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
36
37 #define WM_PROCESSTASK 0x8008
38 #define TIMER_ID 0x3000
39
40 typedef struct {
41     HTMLDocument *doc;
42     DWORD id;
43     DWORD time;
44     DWORD interval;
45     IDispatch *disp;
46
47     struct list entry;
48 } task_timer_t;
49
50 void push_task(task_t *task, task_proc_t proc, LONG magic)
51 {
52     thread_data_t *thread_data = get_thread_data(TRUE);
53
54     task->target_magic = magic;
55     task->proc = proc;
56     task->next = NULL;
57
58     if(thread_data->task_queue_tail)
59         thread_data->task_queue_tail->next = task;
60     else
61         thread_data->task_queue_head = task;
62
63     thread_data->task_queue_tail = task;
64
65     PostMessageW(thread_data->thread_hwnd, WM_PROCESSTASK, 0, 0);
66 }
67
68 static task_t *pop_task(void)
69 {
70     thread_data_t *thread_data = get_thread_data(TRUE);
71     task_t *task = thread_data->task_queue_head;
72
73     if(!task)
74         return NULL;
75
76     thread_data->task_queue_head = task->next;
77     if(!thread_data->task_queue_head)
78         thread_data->task_queue_tail = NULL;
79
80     return task;
81 }
82
83 static void release_task_timer(HWND thread_hwnd, task_timer_t *timer)
84 {
85     list_remove(&timer->entry);
86
87     IDispatch_Release(timer->disp);
88
89     heap_free(timer);
90 }
91
92 void remove_target_tasks(LONG target)
93 {
94     thread_data_t *thread_data = get_thread_data(FALSE);
95     struct list *liter, *ltmp;
96     task_timer_t *timer;
97     task_t *iter, *tmp;
98
99     if(!thread_data)
100         return;
101
102     LIST_FOR_EACH_SAFE(liter, ltmp, &thread_data->timer_list) {
103         timer = LIST_ENTRY(liter, task_timer_t, entry);
104         if(timer->doc->task_magic == target)
105             release_task_timer(thread_data->thread_hwnd, timer);
106     }
107
108     if(!list_empty(&thread_data->timer_list)) {
109         timer = LIST_ENTRY(list_head(&thread_data->timer_list), task_timer_t, entry);
110         SetTimer(thread_data->thread_hwnd, TIMER_ID, timer->time - GetTickCount(), NULL);
111     }
112
113     while(thread_data->task_queue_head
114           && thread_data->task_queue_head->target_magic == target)
115         pop_task();
116
117     for(iter = thread_data->task_queue_head; iter; iter = iter->next) {
118         while(iter->next && iter->next->target_magic == target) {
119             tmp = iter->next;
120             iter->next = tmp->next;
121             heap_free(tmp);
122         }
123
124         if(!iter->next)
125             thread_data->task_queue_tail = iter;
126     }
127 }
128
129 LONG get_task_target_magic(void)
130 {
131     static LONG magic = 0x10000000;
132     return InterlockedIncrement(&magic);
133 }
134
135 static BOOL queue_timer(thread_data_t *thread_data, task_timer_t *timer)
136 {
137     task_timer_t *iter;
138
139     list_remove(&timer->entry);
140
141     if(list_empty(&thread_data->timer_list)
142        || LIST_ENTRY(list_head(&thread_data->timer_list), task_timer_t, entry)->time > timer->time) {
143
144         list_add_head(&thread_data->timer_list, &timer->entry);
145         return TRUE;
146     }
147
148     LIST_FOR_EACH_ENTRY(iter, &thread_data->timer_list, task_timer_t, entry) {
149         if(iter->time > timer->time) {
150             list_add_tail(&iter->entry, &timer->entry);
151             return FALSE;
152         }
153     }
154
155     list_add_tail(&thread_data->timer_list, &timer->entry);
156     return FALSE;
157 }
158
159 DWORD set_task_timer(HTMLDocument *doc, DWORD msec, BOOL interval, IDispatch *disp)
160 {
161     thread_data_t *thread_data = get_thread_data(TRUE);
162     task_timer_t *timer;
163     DWORD tc = GetTickCount();
164
165     static DWORD id_cnt = 0x20000000;
166
167     timer = heap_alloc(sizeof(task_timer_t));
168     timer->id = id_cnt++;
169     timer->doc = doc;
170     timer->time = tc + msec;
171     timer->interval = interval ? msec : 0;
172     list_init(&timer->entry);
173
174     IDispatch_AddRef(disp);
175     timer->disp = disp;
176
177     if(queue_timer(thread_data, timer))
178         SetTimer(thread_data->thread_hwnd, TIMER_ID, msec, NULL);
179
180     return timer->id;
181 }
182
183 HRESULT clear_task_timer(HTMLDocument *doc, BOOL interval, DWORD id)
184 {
185     thread_data_t *thread_data = get_thread_data(FALSE);
186     task_timer_t *iter;
187
188     if(!thread_data)
189         return S_OK;
190
191     LIST_FOR_EACH_ENTRY(iter, &thread_data->timer_list, task_timer_t, entry) {
192         if(iter->id == id && iter->doc == doc && (iter->interval == 0) == !interval) {
193             release_task_timer(thread_data->thread_hwnd, iter);
194             return S_OK;
195         }
196     }
197
198     WARN("timet not found\n");
199     return S_OK;
200 }
201
202 static void call_timer_disp(IDispatch *disp)
203 {
204     DISPPARAMS dp = {NULL, NULL, 0, 0};
205     EXCEPINFO ei;
206     VARIANT res;
207     HRESULT hres;
208
209     V_VT(&res) = VT_EMPTY;
210     memset(&ei, 0, sizeof(ei));
211
212     TRACE(">>>\n");
213     hres = IDispatch_Invoke(disp, DISPID_VALUE, &IID_NULL, 0, DISPATCH_METHOD, &dp, &res, &ei, NULL);
214     if(hres == S_OK)
215         TRACE("<<<\n");
216     else
217         WARN("<<< %08x\n", hres);
218
219     VariantClear(&res);
220 }
221
222 static LRESULT process_timer(void)
223 {
224     thread_data_t *thread_data = get_thread_data(TRUE);
225     IDispatch *disp;
226     DWORD tc;
227     task_timer_t *timer;
228
229     TRACE("\n");
230
231     while(!list_empty(&thread_data->timer_list)) {
232         timer = LIST_ENTRY(list_head(&thread_data->timer_list), task_timer_t, entry);
233
234         tc = GetTickCount();
235         if(timer->time > tc) {
236             SetTimer(thread_data->thread_hwnd, TIMER_ID, timer->time-tc, NULL);
237             return 0;
238         }
239
240         disp = timer->disp;
241         IDispatch_AddRef(disp);
242
243         if(timer->interval) {
244             timer->time += timer->interval;
245             queue_timer(thread_data, timer);
246         }else {
247             release_task_timer(thread_data->thread_hwnd, timer);
248         }
249
250         call_timer_disp(disp);
251
252         IDispatch_Release(disp);
253     }
254
255     KillTimer(thread_data->thread_hwnd, TIMER_ID);
256     return 0;
257 }
258
259 static LRESULT WINAPI hidden_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
260 {
261     switch(msg) {
262     case WM_PROCESSTASK:
263         while(1) {
264             task_t *task = pop_task();
265             if(!task)
266                 break;
267
268             task->proc(task);
269             heap_free(task);
270         }
271
272         return 0;
273     case WM_TIMER:
274         return process_timer();
275     }
276
277     if(msg > WM_USER)
278         FIXME("(%p %d %lx %lx)\n", hwnd, msg, wParam, lParam);
279
280     return DefWindowProcW(hwnd, msg, wParam, lParam);
281 }
282
283 static HWND create_thread_hwnd(void)
284 {
285     static ATOM hidden_wnd_class = 0;
286     static const WCHAR wszInternetExplorer_Hidden[] = {'I','n','t','e','r','n','e','t',
287             ' ','E','x','p','l','o','r','e','r','_','H','i','d','d','e','n',0};
288
289     if(!hidden_wnd_class) {
290         WNDCLASSEXW wndclass = {
291             sizeof(WNDCLASSEXW), 0,
292             hidden_proc,
293             0, 0, hInst, NULL, NULL, NULL, NULL,
294             wszInternetExplorer_Hidden,
295             NULL
296         };
297
298         hidden_wnd_class = RegisterClassExW(&wndclass);
299     }
300
301     return CreateWindowExW(0, wszInternetExplorer_Hidden, NULL, WS_POPUP,
302                            0, 0, 0, 0, NULL, NULL, hInst, NULL);
303 }
304
305 HWND get_thread_hwnd(void)
306 {
307     thread_data_t *thread_data = get_thread_data(TRUE);
308
309     if(!thread_data->thread_hwnd)
310         thread_data->thread_hwnd = create_thread_hwnd();
311
312     return thread_data->thread_hwnd;
313 }
314
315 thread_data_t *get_thread_data(BOOL create)
316 {
317     thread_data_t *thread_data;
318
319     if(mshtml_tls == TLS_OUT_OF_INDEXES) {
320         DWORD tls;
321
322         if(!create)
323             return NULL;
324
325         tls = TlsAlloc();
326         if(tls == TLS_OUT_OF_INDEXES)
327             return NULL;
328
329         tls = InterlockedCompareExchange((LONG*)&mshtml_tls, tls, TLS_OUT_OF_INDEXES);
330         if(tls != mshtml_tls)
331             TlsFree(tls);
332     }
333
334     thread_data = TlsGetValue(mshtml_tls);
335     if(!thread_data && create) {
336         thread_data = heap_alloc_zero(sizeof(thread_data_t));
337         TlsSetValue(mshtml_tls, thread_data);
338         list_init(&thread_data->timer_list);
339     }
340
341     return thread_data;
342 }