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 #include "mshtmcid.h"
  31 #include "shlguid.h"
  32
  33 #include "wine/debug.h"
  34
  35 #include "mshtml_private.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  38
  39 #define WM_PROCESSTASK 0x8008
  40 #define TIMER_ID 0x3000
  41
  42 typedef struct {
  43     HTMLDocument *doc;
  44     DWORD id;
  45     DWORD time;
  46     DWORD interval;
  47     IDispatch *disp;
  48
  49     struct list entry;
  50 } task_timer_t;
  51
  52 void push_task(task_t *task)
  53 {
  54     thread_data_t *thread_data = get_thread_data(TRUE);
  55
  56     if(thread_data->task_queue_tail)
  57         thread_data->task_queue_tail->next = task;
  58     else
  59         thread_data->task_queue_head = task;
  60
  61     thread_data->task_queue_tail = task;
  62
  63     PostMessageW(thread_data->thread_hwnd, WM_PROCESSTASK, 0, 0);
  64 }
  65
  66 static task_t *pop_task(void)
  67 {
  68     thread_data_t *thread_data = get_thread_data(TRUE);
  69     task_t *task = thread_data->task_queue_head;
  70
  71     if(!task)
  72         return NULL;
  73
  74     thread_data->task_queue_head = task->next;
  75     if(!thread_data->task_queue_head)
  76         thread_data->task_queue_tail = NULL;
  77
  78     return task;
  79 }
  80
  81 static void release_task_timer(HWND thread_hwnd, task_timer_t *timer)
  82 {
  83     list_remove(&timer->entry);
  84
  85     IDispatch_Release(timer->disp);
  86
  87     heap_free(timer);
  88 }
  89
  90 void remove_doc_tasks(const HTMLDocument *doc)
  91 {
  92     thread_data_t *thread_data = get_thread_data(FALSE);
  93     struct list *liter, *ltmp;
  94     task_timer_t *timer;
  95     task_t *iter, *tmp;
  96
  97     LIST_FOR_EACH_SAFE(liter, ltmp, &thread_data->timer_list) {
  98         timer = LIST_ENTRY(liter, task_timer_t, entry);
  99         if(timer->doc == doc)
 100             release_task_timer(thread_data->thread_hwnd, timer);
 101     }
 102
 103     if(!list_empty(&thread_data->timer_list)) {
 104         timer = LIST_ENTRY(list_head(&thread_data->timer_list), task_timer_t, entry);
 105         SetTimer(thread_data->thread_hwnd, TIMER_ID, timer->time - GetTickCount(), NULL);
 106     }
 107
 108     if(!thread_data)
 109         return;
 110
 111     while(thread_data->task_queue_head
 112           && thread_data->task_queue_head->doc == doc)
 113         pop_task();
 114
 115     for(iter = thread_data->task_queue_head; iter; iter = iter->next) {
 116         while(iter->next && iter->next->doc == doc) {
 117             tmp = iter->next;
 118             iter->next = tmp->next;
 119             heap_free(tmp);
 120         }
 121
 122         if(!iter->next)
 123             thread_data->task_queue_tail = iter;
 124     }
 125 }
 126
 127 static BOOL queue_timer(thread_data_t *thread_data, task_timer_t *timer)
 128 {
 129     task_timer_t *iter;
 130
 131     list_remove(&timer->entry);
 132
 133     if(list_empty(&thread_data->timer_list)
 134        || LIST_ENTRY(list_head(&thread_data->timer_list), task_timer_t, entry)->time > timer->time) {
 135
 136         list_add_head(&thread_data->timer_list, &timer->entry);
 137         return TRUE;
 138     }
 139
 140     LIST_FOR_EACH_ENTRY(iter, &thread_data->timer_list, task_timer_t, entry) {
 141         if(iter->time > timer->time) {
 142             list_add_tail(&iter->entry, &timer->entry);
 143             return FALSE;
 144         }
 145     }
 146
 147     list_add_tail(&thread_data->timer_list, &timer->entry);
 148     return FALSE;
 149 }
 150
 151 DWORD set_task_timer(HTMLDocument *doc, DWORD msec, BOOL interval, IDispatch *disp)
 152 {
 153     thread_data_t *thread_data = get_thread_data(TRUE);
 154     task_timer_t *timer;
 155     DWORD tc = GetTickCount();
 156
 157     static DWORD id_cnt = 0x20000000;
 158
 159     timer = heap_alloc(sizeof(task_timer_t));
 160     timer->id = id_cnt++;
 161     timer->doc = doc;
 162     timer->time = tc + msec;
 163     timer->interval = interval ? msec : 0;
 164     list_init(&timer->entry);
 165
 166     IDispatch_AddRef(disp);
 167     timer->disp = disp;
 168
 169     if(queue_timer(thread_data, timer))
 170         SetTimer(thread_data->thread_hwnd, TIMER_ID, msec, NULL);
 171
 172     return timer->id;
 173 }
 174
 175 HRESULT clear_task_timer(HTMLDocument *doc, BOOL interval, DWORD id)
 176 {
 177     thread_data_t *thread_data = get_thread_data(FALSE);
 178     task_timer_t *iter;
 179
 180     if(!thread_data)
 181         return S_OK;
 182
 183     LIST_FOR_EACH_ENTRY(iter, &thread_data->timer_list, task_timer_t, entry) {
 184         if(iter->id == id && iter->doc == doc && (iter->interval == 0) == !interval) {
 185             release_task_timer(thread_data->thread_hwnd, iter);
 186             return S_OK;
 187         }
 188     }
 189
 190     WARN("timet not found\n");
 191     return S_OK;
 192 }
 193
 194 static void set_downloading(HTMLDocument *doc)
 195 {
 196     IOleCommandTarget *olecmd;
 197     HRESULT hres;
 198
 199     TRACE("(%p)\n", doc);
 200
 201     if(doc->frame)
 202         IOleInPlaceFrame_SetStatusText(doc->frame, NULL /* FIXME */);
 203
 204     if(!doc->client)
 205         return;
 206
 207     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 208     if(SUCCEEDED(hres)) {
 209         VARIANT var;
 210
 211         V_VT(&var) = VT_I4;
 212         V_I4(&var) = 1;
 213
 214         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETDOWNLOADSTATE, OLECMDEXECOPT_DONTPROMPTUSER,
 215                                &var, NULL);
 216         IOleCommandTarget_Release(olecmd);
 217     }
 218
 219     if(doc->hostui) {
 220         IDropTarget *drop_target = NULL;
 221
 222         hres = IDocHostUIHandler_GetDropTarget(doc->hostui, NULL /* FIXME */, &drop_target);
 223         if(drop_target) {
 224             FIXME("Use IDropTarget\n");
 225             IDropTarget_Release(drop_target);
 226         }
 227     }
 228 }
 229
 230 /* Calls undocumented 69 cmd of CGID_Explorer */
 231 static void call_explorer_69(HTMLDocument *doc)
 232 {
 233     IOleCommandTarget *olecmd;
 234     VARIANT var;
 235     HRESULT hres;
 236
 237     if(!doc->client)
 238         return;
 239
 240     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 241     if(FAILED(hres))
 242         return;
 243
 244     VariantInit(&var);
 245     hres = IOleCommandTarget_Exec(olecmd, &CGID_Explorer, 69, 0, NULL, &var);
 246     IOleCommandTarget_Release(olecmd);
 247     if(SUCCEEDED(hres) && V_VT(&var) != VT_NULL)
 248         FIXME("handle result\n");
 249 }
 250
 251 static void set_parsecomplete(HTMLDocument *doc)
 252 {
 253     IOleCommandTarget *olecmd = NULL;
 254
 255     TRACE("(%p)\n", doc);
 256
 257     if(doc->usermode == EDITMODE)
 258         init_editor(doc);
 259
 260     call_explorer_69(doc);
 261     call_property_onchanged(&doc->cp_propnotif, 1005);
 262     call_explorer_69(doc);
 263
 264     doc->readystate = READYSTATE_INTERACTIVE;
 265     call_property_onchanged(&doc->cp_propnotif, DISPID_READYSTATE);
 266
 267     if(doc->client)
 268         IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 269
 270     if(olecmd) {
 271         VARIANT state, progress;
 272
 273         V_VT(&progress) = VT_I4;
 274         V_I4(&progress) = 0;
 275         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS, OLECMDEXECOPT_DONTPROMPTUSER,
 276                                &progress, NULL);
 277
 278         V_VT(&state) = VT_I4;
 279         V_I4(&state) = 0;
 280         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETDOWNLOADSTATE, OLECMDEXECOPT_DONTPROMPTUSER,
 281                                &state, NULL);
 282
 283         IOleCommandTarget_Exec(olecmd, &CGID_ShellDocView, 103, 0, NULL, NULL);
 284         IOleCommandTarget_Exec(olecmd, &CGID_MSHTML, IDM_PARSECOMPLETE, 0, NULL, NULL);
 285         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_HTTPEQUIV_DONE, 0, NULL, NULL);
 286
 287         IOleCommandTarget_Release(olecmd);
 288     }
 289
 290     doc->readystate = READYSTATE_COMPLETE;
 291     call_property_onchanged(&doc->cp_propnotif, DISPID_READYSTATE);
 292
 293     if(doc->frame) {
 294         static const WCHAR wszDone[] = {'D','o','n','e',0};
 295         IOleInPlaceFrame_SetStatusText(doc->frame, wszDone);
 296     }
 297
 298     update_title(doc);
 299 }
 300
 301 static void set_progress(HTMLDocument *doc)
 302 {
 303     IOleCommandTarget *olecmd = NULL;
 304     HRESULT hres;
 305
 306     TRACE("(%p)\n", doc);
 307
 308     if(doc->client)
 309         IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 310
 311     if(olecmd) {
 312         VARIANT progress_max, progress;
 313
 314         V_VT(&progress_max) = VT_I4;
 315         V_I4(&progress_max) = 0; /* FIXME */
 316         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSMAX, OLECMDEXECOPT_DONTPROMPTUSER,
 317                                &progress_max, NULL);
 318
 319         V_VT(&progress) = VT_I4;
 320         V_I4(&progress) = 0; /* FIXME */
 321         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS, OLECMDEXECOPT_DONTPROMPTUSER,
 322                                &progress, NULL);
 323     }
 324
 325     if(doc->usermode == EDITMODE && doc->hostui) {
 326         DOCHOSTUIINFO hostinfo;
 327
 328         memset(&hostinfo, 0, sizeof(DOCHOSTUIINFO));
 329         hostinfo.cbSize = sizeof(DOCHOSTUIINFO);
 330         hres = IDocHostUIHandler_GetHostInfo(doc->hostui, &hostinfo);
 331         if(SUCCEEDED(hres))
 332             /* FIXME: use hostinfo */
 333             TRACE("hostinfo = {%u %08x %08x %s %s}\n",
 334                     hostinfo.cbSize, hostinfo.dwFlags, hostinfo.dwDoubleClick,
 335                     debugstr_w(hostinfo.pchHostCss), debugstr_w(hostinfo.pchHostNS));
 336     }
 337 }
 338
 339 static void task_start_binding(HTMLDocument *doc, BSCallback *bscallback)
 340 {
 341     if(doc)
 342         start_binding(doc, bscallback, NULL);
 343     IUnknown_Release((IUnknown*)bscallback);
 344 }
 345
 346 static void process_task(task_t *task)
 347 {
 348     switch(task->task_id) {
 349     case TASK_SETDOWNLOADSTATE:
 350         set_downloading(task->doc);
 351         break;
 352     case TASK_PARSECOMPLETE:
 353         set_parsecomplete(task->doc);
 354         break;
 355     case TASK_SETPROGRESS:
 356         set_progress(task->doc);
 357         break;
 358     case TASK_START_BINDING:
 359         task_start_binding(task->doc, (BSCallback*)task->bscallback);
 360         break;
 361     default:
 362         ERR("Wrong task_id %d\n", task->task_id);
 363     }
 364 }
 365
 366 static LRESULT process_timer(void)
 367 {
 368     thread_data_t *thread_data = get_thread_data(TRUE);
 369     HTMLDocument *doc;
 370     IDispatch *disp;
 371     DWORD tc;
 372     task_timer_t *timer;
 373
 374     TRACE("\n");
 375
 376     while(!list_empty(&thread_data->timer_list)) {
 377         timer = LIST_ENTRY(list_head(&thread_data->timer_list), task_timer_t, entry);
 378
 379         tc = GetTickCount();
 380         if(timer->time > tc) {
 381             SetTimer(thread_data->thread_hwnd, TIMER_ID, timer->time-tc, NULL);
 382             return 0;
 383         }
 384
 385         doc = timer->doc;
 386         disp = timer->disp;
 387         IDispatch_AddRef(disp);
 388
 389         if(timer->interval) {
 390             timer->time += timer->interval;
 391             queue_timer(thread_data, timer);
 392         }else {
 393             release_task_timer(thread_data->thread_hwnd, timer);
 394         }
 395
 396         call_disp_func(doc, disp);
 397
 398         IDispatch_Release(disp);
 399     }
 400
 401     KillTimer(thread_data->thread_hwnd, TIMER_ID);
 402     return 0;
 403 }
 404
 405 static LRESULT WINAPI hidden_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
 406 {
 407     switch(msg) {
 408     case WM_PROCESSTASK:
 409         while(1) {
 410             task_t *task = pop_task();
 411             if(!task)
 412                 break;
 413
 414             process_task(task);
 415             heap_free(task);
 416         }
 417
 418         return 0;
 419     case WM_TIMER:
 420         return process_timer();
 421     }
 422
 423     if(msg > WM_USER)
 424         FIXME("(%p %d %lx %lx)\n", hwnd, msg, wParam, lParam);
 425
 426     return DefWindowProcW(hwnd, msg, wParam, lParam);
 427 }
 428
 429 static HWND create_thread_hwnd(void)
 430 {
 431     static ATOM hidden_wnd_class = 0;
 432     static const WCHAR wszInternetExplorer_Hidden[] = {'I','n','t','e','r','n','e','t',
 433             ' ','E','x','p','l','o','r','e','r','_','H','i','d','d','e','n',0};
 434
 435     if(!hidden_wnd_class) {
 436         WNDCLASSEXW wndclass = {
 437             sizeof(WNDCLASSEXW), 0,
 438             hidden_proc,
 439             0, 0, hInst, NULL, NULL, NULL, NULL,
 440             wszInternetExplorer_Hidden,
 441             NULL
 442         };
 443
 444         hidden_wnd_class = RegisterClassExW(&wndclass);
 445     }
 446
 447     return CreateWindowExW(0, wszInternetExplorer_Hidden, NULL, WS_POPUP,
 448                            0, 0, 0, 0, NULL, NULL, hInst, NULL);
 449 }
 450
 451 HWND get_thread_hwnd(void)
 452 {
 453     thread_data_t *thread_data = get_thread_data(TRUE);
 454
 455     if(!thread_data->thread_hwnd)
 456         thread_data->thread_hwnd = create_thread_hwnd();
 457
 458     return thread_data->thread_hwnd;
 459 }
 460
 461 thread_data_t *get_thread_data(BOOL create)
 462 {
 463     thread_data_t *thread_data;
 464
 465     if(!mshtml_tls) {
 466         if(create)
 467             mshtml_tls = TlsAlloc();
 468         else
 469             return NULL;
 470     }
 471
 472     thread_data = TlsGetValue(mshtml_tls);
 473     if(!thread_data && create) {
 474         thread_data = heap_alloc_zero(sizeof(thread_data_t));
 475         TlsSetValue(mshtml_tls, thread_data);
 476         list_init(&thread_data->timer_list);
 477     }
 478
 479     return thread_data;
 480 }