dlls/ieframe/dochost.c

   1 /*
   2  * Copyright 2005-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 "ieframe.h"
  20
  21 #include "exdispid.h"
  22 #include "mshtml.h"
  23 #include "perhist.h"
  24 #include "initguid.h"
  25
  26 #include "wine/debug.h"
  27
  28 WINE_DEFAULT_DEBUG_CHANNEL(ieframe);
  29
  30 DEFINE_OLEGUID(CGID_DocHostCmdPriv, 0x000214D4L, 0, 0);
  31
  32 #define DOCHOST_DOCCANNAVIGATE  0
  33
  34 /* Undocumented notification, see mshtml tests */
  35 #define CMDID_EXPLORER_UPDATEHISTORY 38
  36
  37 static ATOM doc_view_atom = 0;
  38
  39 void push_dochost_task(DocHost *This, task_header_t *task, task_proc_t proc, task_destr_t destr, BOOL send)
  40 {
  41     BOOL is_empty;
  42
  43     task->proc = proc;
  44     task->destr = destr;
  45
  46     is_empty = list_empty(&This->task_queue);
  47     list_add_tail(&This->task_queue, &task->entry);
  48
  49     if(send)
  50         SendMessageW(This->frame_hwnd, WM_DOCHOSTTASK, 0, 0);
  51     else if(is_empty)
  52         PostMessageW(This->frame_hwnd, WM_DOCHOSTTASK, 0, 0);
  53 }
  54
  55 LRESULT process_dochost_tasks(DocHost *This)
  56 {
  57     task_header_t *task;
  58
  59     while(!list_empty(&This->task_queue)) {
  60         task = LIST_ENTRY(This->task_queue.next, task_header_t, entry);
  61         list_remove(&task->entry);
  62
  63         task->proc(This, task);
  64         task->destr(task);
  65     }
  66
  67     return 0;
  68 }
  69
  70 void abort_dochost_tasks(DocHost *This, task_proc_t proc)
  71 {
  72     task_header_t *task, *cursor;
  73
  74     LIST_FOR_EACH_ENTRY_SAFE(task, cursor, &This->task_queue, task_header_t, entry) {
  75         if(proc && proc != task->proc)
  76             continue;
  77
  78         list_remove(&task->entry);
  79         task->destr(task);
  80     }
  81 }
  82
  83 static void notif_complete(DocHost *This, DISPID dispid)
  84 {
  85     DISPPARAMS dispparams;
  86     VARIANTARG params[2];
  87     VARIANT url;
  88
  89     dispparams.cArgs = 2;
  90     dispparams.cNamedArgs = 0;
  91     dispparams.rgdispidNamedArgs = NULL;
  92     dispparams.rgvarg = params;
  93
  94     V_VT(params) = (VT_BYREF|VT_VARIANT);
  95     V_BYREF(params) = &url;
  96
  97     V_VT(params+1) = VT_DISPATCH;
  98     V_DISPATCH(params+1) = (IDispatch*)This->wb;
  99
 100     V_VT(&url) = VT_BSTR;
 101     V_BSTR(&url) = SysAllocString(This->url);
 102
 103     TRACE("%d >>>\n", dispid);
 104     call_sink(This->cps.wbe2, dispid, &dispparams);
 105     TRACE("%d <<<\n", dispid);
 106
 107     SysFreeString(V_BSTR(&url));
 108     This->busy = VARIANT_FALSE;
 109 }
 110
 111 static void object_available(DocHost *This)
 112 {
 113     IHlinkTarget *hlink;
 114     HRESULT hres;
 115
 116     TRACE("(%p)\n", This);
 117
 118     if(!This->document) {
 119         WARN("document == NULL\n");
 120         return;
 121     }
 122
 123     hres = IUnknown_QueryInterface(This->document, &IID_IHlinkTarget, (void**)&hlink);
 124     if(SUCCEEDED(hres)) {
 125         hres = IHlinkTarget_Navigate(hlink, 0, NULL);
 126         IHlinkTarget_Release(hlink);
 127         if(FAILED(hres))
 128             FIXME("Navigate failed\n");
 129     }else {
 130         IOleObject *ole_object;
 131         RECT rect;
 132
 133         TRACE("No IHlink iface\n");
 134
 135         hres = IUnknown_QueryInterface(This->document, &IID_IOleObject, (void**)&ole_object);
 136         if(FAILED(hres)) {
 137             FIXME("Could not get IOleObject iface: %08x\n", hres);
 138             return;
 139         }
 140
 141         GetClientRect(This->hwnd, &rect);
 142         hres = IOleObject_DoVerb(ole_object, OLEIVERB_SHOW, NULL, &This->IOleClientSite_iface, -1, This->hwnd, &rect);
 143         IOleObject_Release(ole_object);
 144         if(FAILED(hres))
 145             FIXME("DoVerb failed: %08x\n", hres);
 146     }
 147 }
 148
 149 static HRESULT get_doc_ready_state(DocHost *This, READYSTATE *ret)
 150 {
 151     DISPPARAMS dp = {NULL,NULL,0,0};
 152     IDispatch *disp;
 153     EXCEPINFO ei;
 154     VARIANT var;
 155     HRESULT hres;
 156
 157     hres = IUnknown_QueryInterface(This->document, &IID_IDispatch, (void**)&disp);
 158     if(FAILED(hres))
 159         return hres;
 160
 161     hres = IDispatch_Invoke(disp, DISPID_READYSTATE, &IID_NULL, LOCALE_SYSTEM_DEFAULT, DISPATCH_PROPERTYGET,
 162             &dp, &var, &ei, NULL);
 163     IDispatch_Release(disp);
 164     if(FAILED(hres)) {
 165         WARN("Invoke(DISPID_READYSTATE failed: %08x\n", hres);
 166         return hres;
 167     }
 168
 169     if(V_VT(&var) != VT_I4) {
 170         WARN("V_VT(var) = %d\n", V_VT(&var));
 171         VariantClear(&var);
 172         return E_FAIL;
 173     }
 174
 175     *ret = V_I4(&var);
 176     return S_OK;
 177 }
 178
 179 static void advise_prop_notif(DocHost *This, BOOL set)
 180 {
 181     IConnectionPointContainer *cp_container;
 182     IConnectionPoint *cp;
 183     HRESULT hres;
 184
 185     hres = IUnknown_QueryInterface(This->document, &IID_IConnectionPointContainer, (void**)&cp_container);
 186     if(FAILED(hres))
 187         return;
 188
 189     hres = IConnectionPointContainer_FindConnectionPoint(cp_container, &IID_IPropertyNotifySink, &cp);
 190     IConnectionPointContainer_Release(cp_container);
 191     if(FAILED(hres))
 192         return;
 193
 194     if(set)
 195         hres = IConnectionPoint_Advise(cp, (IUnknown*)&This->IPropertyNotifySink_iface, &This->prop_notif_cookie);
 196     else
 197         hres = IConnectionPoint_Unadvise(cp, This->prop_notif_cookie);
 198     IConnectionPoint_Release(cp);
 199
 200     if(SUCCEEDED(hres))
 201         This->is_prop_notif = set;
 202 }
 203
 204 void set_doc_state(DocHost *This, READYSTATE doc_state)
 205 {
 206     This->doc_state = doc_state;
 207     if(doc_state > This->ready_state)
 208         This->ready_state = doc_state;
 209 }
 210
 211 static void update_ready_state(DocHost *This, READYSTATE ready_state)
 212 {
 213     if(ready_state > READYSTATE_LOADING && This->travellog.loading_pos != -1) {
 214         WARN("histupdate not notified\n");
 215         This->travellog.position = This->travellog.loading_pos;
 216         This->travellog.loading_pos = -1;
 217     }
 218
 219     if(ready_state > READYSTATE_LOADING && This->doc_state <= READYSTATE_LOADING && !This->browser_service /* FIXME */)
 220         notif_complete(This, DISPID_NAVIGATECOMPLETE2);
 221
 222     if(ready_state == READYSTATE_COMPLETE && This->doc_state < READYSTATE_COMPLETE) {
 223         set_doc_state(This, READYSTATE_COMPLETE);
 224         if(!This->browser_service) /* FIXME: Not fully correct */
 225             notif_complete(This, DISPID_DOCUMENTCOMPLETE);
 226     }else {
 227         set_doc_state(This, ready_state);
 228     }
 229 }
 230
 231 typedef struct {
 232     task_header_t header;
 233     IUnknown *doc;
 234     READYSTATE ready_state;
 235 } ready_state_task_t;
 236
 237 static void ready_state_task_destr(task_header_t *_task)
 238 {
 239     ready_state_task_t *task = (ready_state_task_t*)_task;
 240
 241     IUnknown_Release(task->doc);
 242     heap_free(task);
 243 }
 244
 245 static void ready_state_proc(DocHost *This, task_header_t *_task)
 246 {
 247     ready_state_task_t *task = (ready_state_task_t*)_task;
 248
 249     if(task->doc == This->document)
 250         update_ready_state(This, task->ready_state);
 251 }
 252
 253 static void push_ready_state_task(DocHost *This, READYSTATE ready_state)
 254 {
 255     ready_state_task_t *task = heap_alloc(sizeof(ready_state_task_t));
 256
 257     IUnknown_AddRef(This->document);
 258     task->doc = This->document;
 259     task->ready_state = ready_state;
 260
 261     push_dochost_task(This, &task->header, ready_state_proc, ready_state_task_destr, FALSE);
 262 }
 263
 264 static void object_available_task_destr(task_header_t *task)
 265 {
 266     heap_free(task);
 267 }
 268
 269 static void object_available_proc(DocHost *This, task_header_t *task)
 270 {
 271     object_available(This);
 272 }
 273
 274 HRESULT dochost_object_available(DocHost *This, IUnknown *doc)
 275 {
 276     READYSTATE ready_state;
 277     task_header_t *task;
 278     IOleObject *oleobj;
 279     HRESULT hres;
 280
 281     IUnknown_AddRef(doc);
 282     This->document = doc;
 283
 284     hres = IUnknown_QueryInterface(doc, &IID_IOleObject, (void**)&oleobj);
 285     if(SUCCEEDED(hres)) {
 286         CLSID clsid;
 287
 288         hres = IOleObject_GetUserClassID(oleobj, &clsid);
 289         if(SUCCEEDED(hres))
 290             TRACE("Got clsid %s\n",
 291                   IsEqualGUID(&clsid, &CLSID_HTMLDocument) ? "CLSID_HTMLDocument" : debugstr_guid(&clsid));
 292
 293         hres = IOleObject_SetClientSite(oleobj, &This->IOleClientSite_iface);
 294         if(FAILED(hres))
 295             FIXME("SetClientSite failed: %08x\n", hres);
 296
 297         IOleObject_Release(oleobj);
 298     }else {
 299         FIXME("Could not get IOleObject iface: %08x\n", hres);
 300     }
 301
 302     /* FIXME: Call SetAdvise */
 303
 304     task = heap_alloc(sizeof(*task));
 305     push_dochost_task(This, task, object_available_proc, object_available_task_destr, FALSE);
 306
 307     hres = get_doc_ready_state(This, &ready_state);
 308     if(SUCCEEDED(hres)) {
 309         if(ready_state == READYSTATE_COMPLETE)
 310             push_ready_state_task(This, READYSTATE_COMPLETE);
 311         if(ready_state != READYSTATE_COMPLETE || This->doc_navigate)
 312             advise_prop_notif(This, TRUE);
 313     }else if(!This->doc_navigate) {
 314         /* If we can't get document's ready state, there is not much we can do.
 315          * Assume that document is complete at this point. */
 316         push_ready_state_task(This, READYSTATE_COMPLETE);
 317     }
 318
 319     return S_OK;
 320 }
 321
 322 static LRESULT resize_document(DocHost *This, LONG width, LONG height)
 323 {
 324     RECT rect = {0, 0, width, height};
 325
 326     TRACE("(%p)->(%d %d)\n", This, width, height);
 327
 328     if(This->view)
 329         IOleDocumentView_SetRect(This->view, &rect);
 330
 331     return 0;
 332 }
 333
 334 static LRESULT WINAPI doc_view_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
 335 {
 336     DocHost *This;
 337
 338     static const WCHAR wszTHIS[] = {'T','H','I','S',0};
 339
 340     if(msg == WM_CREATE) {
 341         This = *(DocHost**)lParam;
 342         SetPropW(hwnd, wszTHIS, This);
 343     }else {
 344         This = GetPropW(hwnd, wszTHIS);
 345     }
 346
 347     switch(msg) {
 348     case WM_SIZE:
 349         return resize_document(This, LOWORD(lParam), HIWORD(lParam));
 350     }
 351
 352     return DefWindowProcW(hwnd, msg, wParam, lParam);
 353 }
 354
 355 static void free_travellog_entry(travellog_entry_t *entry)
 356 {
 357     if(entry->stream)
 358         IStream_Release(entry->stream);
 359     heap_free(entry->url);
 360 }
 361
 362 static IStream *get_travellog_stream(DocHost *This)
 363 {
 364     IPersistHistory *persist_history;
 365     IStream *stream;
 366     HRESULT hres;
 367
 368     hres = IUnknown_QueryInterface(This->document, &IID_IPersistHistory, (void**)&persist_history);
 369     if(FAILED(hres))
 370         return NULL;
 371
 372     hres = CreateStreamOnHGlobal(NULL, TRUE, &stream);
 373     if(SUCCEEDED(hres))
 374         hres = IPersistHistory_SaveHistory(persist_history, stream);
 375     IPersistHistory_Release(persist_history);
 376     if(FAILED(hres)) {
 377         IStream_Release(stream);
 378         return NULL;
 379     }
 380
 381     return stream;
 382 }
 383
 384 static void dump_travellog(DocHost *This)
 385 {
 386     unsigned i;
 387
 388     for(i=0; i < This->travellog.length; i++)
 389         TRACE("%d: %s %s\n", i, i == This->travellog.position ? "=>" : "  ", debugstr_w(This->travellog.log[i].url));
 390     if(i == This->travellog.position)
 391         TRACE("%d: =>\n", i);
 392 }
 393
 394 static void update_travellog(DocHost *This)
 395 {
 396     travellog_entry_t *new_entry;
 397
 398     if(This->travellog.loading_pos == -1) {
 399         /* Clear forward history. */
 400         if(!This->travellog.log) {
 401             This->travellog.log = heap_alloc(4 * sizeof(*This->travellog.log));
 402             if(!This->travellog.log)
 403                 return;
 404
 405             This->travellog.size = 4;
 406         }else if(This->travellog.size < This->travellog.position+1) {
 407             travellog_entry_t *new_travellog;
 408
 409             new_travellog = heap_realloc(This->travellog.log, This->travellog.size*2*sizeof(*This->travellog.log));
 410             if(!new_travellog)
 411                 return;
 412
 413             This->travellog.log = new_travellog;
 414             This->travellog.size *= 2;
 415         }
 416
 417         while(This->travellog.length > This->travellog.position)
 418             free_travellog_entry(This->travellog.log + --This->travellog.length);
 419     }
 420
 421     new_entry = This->travellog.log + This->travellog.position;
 422
 423     new_entry->url = heap_strdupW(This->url);
 424     TRACE("Adding %s at %d\n", debugstr_w(This->url), This->travellog.position);
 425     if(!new_entry->url)
 426         return;
 427
 428     new_entry->stream = get_travellog_stream(This);
 429
 430     if(This->travellog.loading_pos == -1) {
 431         This->travellog.position++;
 432     }else {
 433          This->travellog.position = This->travellog.loading_pos;
 434          This->travellog.loading_pos = -1;
 435     }
 436     if(This->travellog.position > This->travellog.length)
 437         This->travellog.length = This->travellog.position;
 438
 439     dump_travellog(This);
 440 }
 441
 442 void create_doc_view_hwnd(DocHost *This)
 443 {
 444     RECT rect;
 445
 446     static const WCHAR wszShell_DocObject_View[] =
 447         {'S','h','e','l','l',' ','D','o','c','O','b','j','e','c','t',' ','V','i','e','w',0};
 448
 449     if(!doc_view_atom) {
 450         static WNDCLASSEXW wndclass = {
 451             sizeof(wndclass),
 452             CS_PARENTDC,
 453             doc_view_proc,
 454             0, 0 /* native uses 4*/, NULL, NULL, NULL,
 455             (HBRUSH)(COLOR_WINDOW + 1), NULL,
 456             wszShell_DocObject_View,
 457             NULL
 458         };
 459
 460         wndclass.hInstance = ieframe_instance;
 461
 462         doc_view_atom = RegisterClassExW(&wndclass);
 463     }
 464
 465     This->container_vtbl->GetDocObjRect(This, &rect);
 466     This->hwnd = CreateWindowExW(0, wszShell_DocObject_View,
 467          wszShell_DocObject_View,
 468          WS_CHILD | WS_VISIBLE | WS_CLIPSIBLINGS | WS_CLIPCHILDREN | WS_TABSTOP,
 469          rect.left, rect.top, rect.right, rect.bottom, This->frame_hwnd,
 470          NULL, ieframe_instance, This);
 471 }
 472
 473 void deactivate_document(DocHost *This)
 474 {
 475     IOleInPlaceObjectWindowless *winobj;
 476     IOleObject *oleobj = NULL;
 477     IHlinkTarget *hlink = NULL;
 478     HRESULT hres;
 479
 480     if(!This->document) return;
 481
 482     if(This->doc_navigate) {
 483         IUnknown_Release(This->doc_navigate);
 484         This->doc_navigate = NULL;
 485     }
 486
 487     if(This->is_prop_notif)
 488         advise_prop_notif(This, FALSE);
 489
 490     if(This->view)
 491         IOleDocumentView_UIActivate(This->view, FALSE);
 492
 493     hres = IUnknown_QueryInterface(This->document, &IID_IOleInPlaceObjectWindowless,
 494                                    (void**)&winobj);
 495     if(SUCCEEDED(hres)) {
 496         IOleInPlaceObjectWindowless_InPlaceDeactivate(winobj);
 497         IOleInPlaceObjectWindowless_Release(winobj);
 498     }
 499
 500     if(This->view) {
 501         IOleDocumentView_Show(This->view, FALSE);
 502         IOleDocumentView_CloseView(This->view, 0);
 503         IOleDocumentView_SetInPlaceSite(This->view, NULL);
 504         IOleDocumentView_Release(This->view);
 505         This->view = NULL;
 506     }
 507
 508     hres = IUnknown_QueryInterface(This->document, &IID_IOleObject, (void**)&oleobj);
 509     if(SUCCEEDED(hres))
 510         IOleObject_Close(oleobj, OLECLOSE_NOSAVE);
 511
 512     hres = IUnknown_QueryInterface(This->document, &IID_IHlinkTarget, (void**)&hlink);
 513     if(SUCCEEDED(hres)) {
 514         IHlinkTarget_SetBrowseContext(hlink, NULL);
 515         IHlinkTarget_Release(hlink);
 516     }
 517
 518     if(oleobj) {
 519         IOleClientSite *client_site = NULL;
 520
 521         IOleObject_GetClientSite(oleobj, &client_site);
 522         if(client_site) {
 523             if(client_site == &This->IOleClientSite_iface)
 524                 IOleObject_SetClientSite(oleobj, NULL);
 525             IOleClientSite_Release(client_site);
 526         }
 527
 528         IOleObject_Release(oleobj);
 529     }
 530
 531     IUnknown_Release(This->document);
 532     This->document = NULL;
 533 }
 534
 535 HRESULT refresh_document(DocHost *This)
 536 {
 537     IOleCommandTarget *cmdtrg;
 538     VARIANT vin, vout;
 539     HRESULT hres;
 540
 541     if(!This->document) {
 542         FIXME("no document\n");
 543         return E_FAIL;
 544     }
 545
 546     hres = IUnknown_QueryInterface(This->document, &IID_IOleCommandTarget, (void**)&cmdtrg);
 547     if(FAILED(hres))
 548         return hres;
 549
 550     V_VT(&vin) = VT_EMPTY;
 551     V_VT(&vout) = VT_EMPTY;
 552     hres = IOleCommandTarget_Exec(cmdtrg, NULL, OLECMDID_REFRESH, OLECMDEXECOPT_PROMPTUSER, &vin, &vout);
 553     IOleCommandTarget_Release(cmdtrg);
 554     if(FAILED(hres))
 555         return hres;
 556
 557     VariantClear(&vout);
 558     return S_OK;
 559 }
 560
 561 void release_dochost_client(DocHost *This)
 562 {
 563     if(This->hwnd) {
 564         DestroyWindow(This->hwnd);
 565         This->hwnd = NULL;
 566     }
 567
 568     if(This->hostui) {
 569         IDocHostUIHandler_Release(This->hostui);
 570         This->hostui = NULL;
 571     }
 572
 573     if(This->client_disp) {
 574         IDispatch_Release(This->client_disp);
 575         This->client_disp = NULL;
 576     }
 577
 578     if(This->frame) {
 579         IOleInPlaceFrame_Release(This->frame);
 580         This->frame = NULL;
 581     }
 582 }
 583
 584 static inline DocHost *impl_from_IOleCommandTarget(IOleCommandTarget *iface)
 585 {
 586     return CONTAINING_RECORD(iface, DocHost, IOleCommandTarget_iface);
 587 }
 588
 589 static HRESULT WINAPI ClOleCommandTarget_QueryInterface(IOleCommandTarget *iface,
 590         REFIID riid, void **ppv)
 591 {
 592     DocHost *This = impl_from_IOleCommandTarget(iface);
 593     return IOleClientSite_QueryInterface(&This->IOleClientSite_iface, riid, ppv);
 594 }
 595
 596 static ULONG WINAPI ClOleCommandTarget_AddRef(IOleCommandTarget *iface)
 597 {
 598     DocHost *This = impl_from_IOleCommandTarget(iface);
 599     return IOleClientSite_AddRef(&This->IOleClientSite_iface);
 600 }
 601
 602 static ULONG WINAPI ClOleCommandTarget_Release(IOleCommandTarget *iface)
 603 {
 604     DocHost *This = impl_from_IOleCommandTarget(iface);
 605     return IOleClientSite_Release(&This->IOleClientSite_iface);
 606 }
 607
 608 static HRESULT WINAPI ClOleCommandTarget_QueryStatus(IOleCommandTarget *iface,
 609         const GUID *pguidCmdGroup, ULONG cCmds, OLECMD prgCmds[], OLECMDTEXT *pCmdText)
 610 {
 611     DocHost *This = impl_from_IOleCommandTarget(iface);
 612     ULONG i= 0;
 613     FIXME("(%p)->(%s %u %p %p)\n", This, debugstr_guid(pguidCmdGroup), cCmds, prgCmds,
 614           pCmdText);
 615     while (prgCmds && (cCmds > i)) {
 616         FIXME("command_%u: %u, 0x%x\n", i, prgCmds[i].cmdID, prgCmds[i].cmdf);
 617         i++;
 618     }
 619     return E_NOTIMPL;
 620 }
 621
 622 static HRESULT WINAPI ClOleCommandTarget_Exec(IOleCommandTarget *iface,
 623         const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANT *pvaIn,
 624         VARIANT *pvaOut)
 625 {
 626     DocHost *This = impl_from_IOleCommandTarget(iface);
 627
 628     TRACE("(%p)->(%s %d %d %s %s)\n", This, debugstr_guid(pguidCmdGroup), nCmdID, nCmdexecopt,
 629             debugstr_variant(pvaIn), debugstr_variant(pvaOut));
 630
 631     if(!pguidCmdGroup) {
 632         switch(nCmdID) {
 633         case OLECMDID_UPDATECOMMANDS:
 634         case OLECMDID_SETDOWNLOADSTATE:
 635             return This->container_vtbl->exec(This, pguidCmdGroup, nCmdID, nCmdexecopt, pvaIn, pvaOut);
 636         default:
 637             FIXME("Unimplemented cmdid %d\n", nCmdID);
 638             return E_NOTIMPL;
 639         }
 640     }
 641
 642     if(IsEqualGUID(pguidCmdGroup, &CGID_DocHostCmdPriv)) {
 643         switch(nCmdID) {
 644         case DOCHOST_DOCCANNAVIGATE:
 645             if(!pvaIn || V_VT(pvaIn) != VT_UNKNOWN)
 646                 return E_INVALIDARG;
 647
 648             if(This->doc_navigate)
 649                 IUnknown_Release(This->doc_navigate);
 650             IUnknown_AddRef(V_UNKNOWN(pvaIn));
 651             This->doc_navigate = V_UNKNOWN(pvaIn);
 652             return S_OK;
 653
 654         case 1: {
 655             IHTMLWindow2 *win2;
 656             SAFEARRAY *sa = V_ARRAY(pvaIn);
 657             VARIANT status_code, url, htmlwindow;
 658             LONG ind;
 659             HRESULT hres;
 660
 661             if(V_VT(pvaIn) != VT_ARRAY || !sa || (SafeArrayGetDim(sa) != 1))
 662                 return E_INVALIDARG;
 663
 664             ind = 0;
 665             hres = SafeArrayGetElement(sa, &ind, &status_code);
 666             if(FAILED(hres) || V_VT(&status_code)!=VT_I4)
 667                 return E_INVALIDARG;
 668
 669             ind = 1;
 670             hres = SafeArrayGetElement(sa, &ind, &url);
 671             if(FAILED(hres) || V_VT(&url)!=VT_BSTR)
 672                 return E_INVALIDARG;
 673
 674             ind = 3;
 675             hres = SafeArrayGetElement(sa, &ind, &htmlwindow);
 676             if(FAILED(hres) || V_VT(&htmlwindow)!=VT_UNKNOWN || !V_UNKNOWN(&htmlwindow))
 677                 return E_INVALIDARG;
 678
 679             hres = IUnknown_QueryInterface(V_UNKNOWN(&htmlwindow), &IID_IHTMLWindow2, (void**)&win2);
 680             if(FAILED(hres))
 681                 return E_INVALIDARG;
 682
 683             handle_navigation_error(This, V_I4(&status_code), V_BSTR(&url), win2);
 684             IHTMLWindow2_Release(win2);
 685             return S_OK;
 686         }
 687
 688         default:
 689             FIXME("unsupported command %d of CGID_DocHostCmdPriv\n", nCmdID);
 690             return E_NOTIMPL;
 691         }
 692     }
 693
 694     if(IsEqualGUID(pguidCmdGroup, &CGID_Explorer)) {
 695         switch(nCmdID) {
 696         case CMDID_EXPLORER_UPDATEHISTORY:
 697             update_travellog(This);
 698             return S_OK;
 699
 700         default:
 701             FIXME("Unimplemented cmdid %d of CGID_Explorer\n", nCmdID);
 702             return E_NOTIMPL;
 703         }
 704     }
 705
 706     if(IsEqualGUID(pguidCmdGroup, &CGID_ShellDocView)) {
 707         switch(nCmdID) {
 708         default:
 709             FIXME("Unimplemented cmdid %d of CGID_ShellDocView\n", nCmdID);
 710             return E_NOTIMPL;
 711         }
 712     }
 713
 714     if(IsEqualGUID(&CGID_DocHostCommandHandler, pguidCmdGroup))
 715         return This->container_vtbl->exec(This, pguidCmdGroup, nCmdID, nCmdexecopt, pvaIn, pvaOut);
 716
 717     FIXME("Unimplemented cmdid %d of group %s\n", nCmdID, debugstr_guid(pguidCmdGroup));
 718     return E_NOTIMPL;
 719 }
 720
 721 static const IOleCommandTargetVtbl OleCommandTargetVtbl = {
 722     ClOleCommandTarget_QueryInterface,
 723     ClOleCommandTarget_AddRef,
 724     ClOleCommandTarget_Release,
 725     ClOleCommandTarget_QueryStatus,
 726     ClOleCommandTarget_Exec
 727 };
 728
 729 static inline DocHost *impl_from_IDocHostUIHandler2(IDocHostUIHandler2 *iface)
 730 {
 731     return CONTAINING_RECORD(iface, DocHost, IDocHostUIHandler2_iface);
 732 }
 733
 734 static HRESULT WINAPI DocHostUIHandler_QueryInterface(IDocHostUIHandler2 *iface,
 735                                                       REFIID riid, void **ppv)
 736 {
 737     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 738     return IOleClientSite_QueryInterface(&This->IOleClientSite_iface, riid, ppv);
 739 }
 740
 741 static ULONG WINAPI DocHostUIHandler_AddRef(IDocHostUIHandler2 *iface)
 742 {
 743     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 744     return IOleClientSite_AddRef(&This->IOleClientSite_iface);
 745 }
 746
 747 static ULONG WINAPI DocHostUIHandler_Release(IDocHostUIHandler2 *iface)
 748 {
 749     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 750     return IOleClientSite_Release(&This->IOleClientSite_iface);
 751 }
 752
 753 static HRESULT WINAPI DocHostUIHandler_ShowContextMenu(IDocHostUIHandler2 *iface,
 754          DWORD dwID, POINT *ppt, IUnknown *pcmdtReserved, IDispatch *pdispReserved)
 755 {
 756     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 757     HRESULT hres;
 758
 759     TRACE("(%p)->(%d %p %p %p)\n", This, dwID, ppt, pcmdtReserved, pdispReserved);
 760
 761     if(This->hostui) {
 762         hres = IDocHostUIHandler_ShowContextMenu(This->hostui, dwID, ppt, pcmdtReserved,
 763                                                  pdispReserved);
 764         if(hres == S_OK)
 765             return S_OK;
 766     }
 767
 768     FIXME("default action not implemented\n");
 769     return E_NOTIMPL;
 770 }
 771
 772 static HRESULT WINAPI DocHostUIHandler_GetHostInfo(IDocHostUIHandler2 *iface,
 773         DOCHOSTUIINFO *pInfo)
 774 {
 775     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 776     HRESULT hres;
 777
 778     TRACE("(%p)->(%p)\n", This, pInfo);
 779
 780     if(This->hostui) {
 781         hres = IDocHostUIHandler_GetHostInfo(This->hostui, pInfo);
 782         if(SUCCEEDED(hres))
 783             return hres;
 784     }
 785
 786     pInfo->dwFlags = DOCHOSTUIFLAG_DISABLE_HELP_MENU | DOCHOSTUIFLAG_OPENNEWWIN
 787         | DOCHOSTUIFLAG_URL_ENCODING_ENABLE_UTF8 | DOCHOSTUIFLAG_ENABLE_INPLACE_NAVIGATION
 788         | DOCHOSTUIFLAG_IME_ENABLE_RECONVERSION;
 789     return S_OK;
 790 }
 791
 792 static HRESULT WINAPI DocHostUIHandler_ShowUI(IDocHostUIHandler2 *iface, DWORD dwID,
 793         IOleInPlaceActiveObject *pActiveObject, IOleCommandTarget *pCommandTarget,
 794         IOleInPlaceFrame *pFrame, IOleInPlaceUIWindow *pDoc)
 795 {
 796     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 797     FIXME("(%p)->(%d %p %p %p %p)\n", This, dwID, pActiveObject, pCommandTarget,
 798           pFrame, pDoc);
 799     return E_NOTIMPL;
 800 }
 801
 802 static HRESULT WINAPI DocHostUIHandler_HideUI(IDocHostUIHandler2 *iface)
 803 {
 804     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 805     FIXME("(%p)\n", This);
 806     return E_NOTIMPL;
 807 }
 808
 809 static HRESULT WINAPI DocHostUIHandler_UpdateUI(IDocHostUIHandler2 *iface)
 810 {
 811     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 812
 813     TRACE("(%p)\n", This);
 814
 815     if(!This->hostui)
 816         return S_FALSE;
 817
 818     return IDocHostUIHandler_UpdateUI(This->hostui);
 819 }
 820
 821 static HRESULT WINAPI DocHostUIHandler_EnableModeless(IDocHostUIHandler2 *iface,
 822                                                       BOOL fEnable)
 823 {
 824     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 825     FIXME("(%p)->(%x)\n", This, fEnable);
 826     return E_NOTIMPL;
 827 }
 828
 829 static HRESULT WINAPI DocHostUIHandler_OnDocWindowActivate(IDocHostUIHandler2 *iface,
 830                                                            BOOL fActivate)
 831 {
 832     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 833     FIXME("(%p)->(%x)\n", This, fActivate);
 834     return E_NOTIMPL;
 835 }
 836
 837 static HRESULT WINAPI DocHostUIHandler_OnFrameWindowActivate(IDocHostUIHandler2 *iface,
 838                                                              BOOL fActivate)
 839 {
 840     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 841     FIXME("(%p)->(%x)\n", This, fActivate);
 842     return E_NOTIMPL;
 843 }
 844
 845 static HRESULT WINAPI DocHostUIHandler_ResizeBorder(IDocHostUIHandler2 *iface,
 846         LPCRECT prcBorder, IOleInPlaceUIWindow *pUIWindow, BOOL fRameWindow)
 847 {
 848     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 849     FIXME("(%p)->(%p %p %X)\n", This, prcBorder, pUIWindow, fRameWindow);
 850     return E_NOTIMPL;
 851 }
 852
 853 static HRESULT WINAPI DocHostUIHandler_TranslateAccelerator(IDocHostUIHandler2 *iface,
 854         LPMSG lpMsg, const GUID *pguidCmdGroup, DWORD nCmdID)
 855 {
 856     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 857     HRESULT hr = S_FALSE;
 858     TRACE("(%p)->(%p %p %d)\n", This, lpMsg, pguidCmdGroup, nCmdID);
 859
 860     if(This->hostui)
 861         hr = IDocHostUIHandler_TranslateAccelerator(This->hostui, lpMsg, pguidCmdGroup, nCmdID);
 862
 863     return hr;
 864 }
 865
 866 static HRESULT WINAPI DocHostUIHandler_GetOptionKeyPath(IDocHostUIHandler2 *iface,
 867         LPOLESTR *pchKey, DWORD dw)
 868 {
 869     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 870
 871     TRACE("(%p)->(%p %d)\n", This, pchKey, dw);
 872
 873     if(This->hostui)
 874         return IDocHostUIHandler_GetOptionKeyPath(This->hostui, pchKey, dw);
 875
 876     return S_OK;
 877 }
 878
 879 static HRESULT WINAPI DocHostUIHandler_GetDropTarget(IDocHostUIHandler2 *iface,
 880         IDropTarget *pDropTarget, IDropTarget **ppDropTarget)
 881 {
 882     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 883     FIXME("(%p)\n", This);
 884     return E_NOTIMPL;
 885 }
 886
 887 static HRESULT WINAPI DocHostUIHandler_GetExternal(IDocHostUIHandler2 *iface,
 888         IDispatch **ppDispatch)
 889 {
 890     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 891
 892     TRACE("(%p)->(%p)\n", This, ppDispatch);
 893
 894     if(This->hostui)
 895         return IDocHostUIHandler_GetExternal(This->hostui, ppDispatch);
 896
 897     if(!This->shell_ui_helper) {
 898         HRESULT hres;
 899
 900         hres = create_shell_ui_helper(&This->shell_ui_helper);
 901         if(FAILED(hres))
 902             return hres;
 903     }
 904
 905     *ppDispatch = (IDispatch*)This->shell_ui_helper;
 906     IDispatch_AddRef(*ppDispatch);
 907     return S_OK;
 908 }
 909
 910 static HRESULT WINAPI DocHostUIHandler_TranslateUrl(IDocHostUIHandler2 *iface,
 911         DWORD dwTranslate, OLECHAR *pchURLIn, OLECHAR **ppchURLOut)
 912 {
 913     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 914
 915     TRACE("(%p)->(%d %s %p)\n", This, dwTranslate, debugstr_w(pchURLIn), ppchURLOut);
 916
 917     if(This->hostui)
 918         return IDocHostUIHandler_TranslateUrl(This->hostui, dwTranslate,
 919                                               pchURLIn, ppchURLOut);
 920
 921     return S_FALSE;
 922 }
 923
 924 static HRESULT WINAPI DocHostUIHandler_FilterDataObject(IDocHostUIHandler2 *iface,
 925         IDataObject *pDO, IDataObject **ppDORet)
 926 {
 927     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 928     FIXME("(%p)->(%p %p)\n", This, pDO, ppDORet);
 929     return E_NOTIMPL;
 930 }
 931
 932 static HRESULT WINAPI DocHostUIHandler_GetOverrideKeyPath(IDocHostUIHandler2 *iface,
 933         LPOLESTR *pchKey, DWORD dw)
 934 {
 935     DocHost *This = impl_from_IDocHostUIHandler2(iface);
 936     IDocHostUIHandler2 *handler;
 937     HRESULT hres;
 938
 939     TRACE("(%p)->(%p %d)\n", This, pchKey, dw);
 940
 941     if(!This->hostui)
 942         return S_OK;
 943
 944     hres = IDocHostUIHandler_QueryInterface(This->hostui, &IID_IDocHostUIHandler2,
 945                                             (void**)&handler);
 946     if(SUCCEEDED(hres)) {
 947         hres = IDocHostUIHandler2_GetOverrideKeyPath(handler, pchKey, dw);
 948         IDocHostUIHandler2_Release(handler);
 949         return hres;
 950     }
 951
 952     return S_OK;
 953 }
 954
 955 static const IDocHostUIHandler2Vtbl DocHostUIHandler2Vtbl = {
 956     DocHostUIHandler_QueryInterface,
 957     DocHostUIHandler_AddRef,
 958     DocHostUIHandler_Release,
 959     DocHostUIHandler_ShowContextMenu,
 960     DocHostUIHandler_GetHostInfo,
 961     DocHostUIHandler_ShowUI,
 962     DocHostUIHandler_HideUI,
 963     DocHostUIHandler_UpdateUI,
 964     DocHostUIHandler_EnableModeless,
 965     DocHostUIHandler_OnDocWindowActivate,
 966     DocHostUIHandler_OnFrameWindowActivate,
 967     DocHostUIHandler_ResizeBorder,
 968     DocHostUIHandler_TranslateAccelerator,
 969     DocHostUIHandler_GetOptionKeyPath,
 970     DocHostUIHandler_GetDropTarget,
 971     DocHostUIHandler_GetExternal,
 972     DocHostUIHandler_TranslateUrl,
 973     DocHostUIHandler_FilterDataObject,
 974     DocHostUIHandler_GetOverrideKeyPath
 975 };
 976
 977 static inline DocHost *impl_from_IPropertyNotifySink(IPropertyNotifySink *iface)
 978 {
 979     return CONTAINING_RECORD(iface, DocHost, IPropertyNotifySink_iface);
 980 }
 981
 982 static HRESULT WINAPI PropertyNotifySink_QueryInterface(IPropertyNotifySink *iface,
 983         REFIID riid, void **ppv)
 984 {
 985     DocHost *This = impl_from_IPropertyNotifySink(iface);
 986     return IOleClientSite_QueryInterface(&This->IOleClientSite_iface, riid, ppv);
 987 }
 988
 989 static ULONG WINAPI PropertyNotifySink_AddRef(IPropertyNotifySink *iface)
 990 {
 991     DocHost *This = impl_from_IPropertyNotifySink(iface);
 992     return IOleClientSite_AddRef(&This->IOleClientSite_iface);
 993 }
 994
 995 static ULONG WINAPI PropertyNotifySink_Release(IPropertyNotifySink *iface)
 996 {
 997     DocHost *This = impl_from_IPropertyNotifySink(iface);
 998     return IOleClientSite_Release(&This->IOleClientSite_iface);
 999 }
1000
1001 static HRESULT WINAPI PropertyNotifySink_OnChanged(IPropertyNotifySink *iface, DISPID dispID)
1002 {
1003     DocHost *This = impl_from_IPropertyNotifySink(iface);
1004
1005     TRACE("(%p)->(%d)\n", This, dispID);
1006
1007     switch(dispID) {
1008     case DISPID_READYSTATE: {
1009         READYSTATE ready_state;
1010         HRESULT hres;
1011
1012         hres = get_doc_ready_state(This, &ready_state);
1013         if(FAILED(hres))
1014             return hres;
1015
1016         if(ready_state == READYSTATE_COMPLETE && !This->doc_navigate)
1017             advise_prop_notif(This, FALSE);
1018
1019         update_ready_state(This, ready_state);
1020         break;
1021     }
1022     default:
1023         FIXME("unimplemented dispid %d\n", dispID);
1024         return E_NOTIMPL;
1025     }
1026
1027     return S_OK;
1028 }
1029
1030 static HRESULT WINAPI PropertyNotifySink_OnRequestEdit(IPropertyNotifySink *iface, DISPID dispID)
1031 {
1032     DocHost *This = impl_from_IPropertyNotifySink(iface);
1033     FIXME("(%p)->(%d)\n", This, dispID);
1034     return E_NOTIMPL;
1035 }
1036
1037 static const IPropertyNotifySinkVtbl PropertyNotifySinkVtbl = {
1038     PropertyNotifySink_QueryInterface,
1039     PropertyNotifySink_AddRef,
1040     PropertyNotifySink_Release,
1041     PropertyNotifySink_OnChanged,
1042     PropertyNotifySink_OnRequestEdit
1043 };
1044
1045 void DocHost_Init(DocHost *This, IWebBrowser2 *wb, const IDocHostContainerVtbl* container)
1046 {
1047     This->IDocHostUIHandler2_iface.lpVtbl  = &DocHostUIHandler2Vtbl;
1048     This->IOleCommandTarget_iface.lpVtbl   = &OleCommandTargetVtbl;
1049     This->IPropertyNotifySink_iface.lpVtbl = &PropertyNotifySinkVtbl;
1050
1051     This->wb = wb;
1052     This->container_vtbl = container;
1053
1054     This->ready_state = READYSTATE_UNINITIALIZED;
1055     list_init(&This->task_queue);
1056
1057     This->travellog.loading_pos = -1;
1058
1059     DocHost_ClientSite_Init(This);
1060     DocHost_Frame_Init(This);
1061
1062     ConnectionPointContainer_Init(&This->cps, (IUnknown*)wb);
1063     IEHTMLWindow_Init(This);
1064     NewWindowManager_Init(This);
1065 }
1066
1067 void DocHost_Release(DocHost *This)
1068 {
1069     if(This->shell_ui_helper)
1070         IShellUIHelper2_Release(This->shell_ui_helper);
1071
1072     abort_dochost_tasks(This, NULL);
1073     release_dochost_client(This);
1074     DocHost_ClientSite_Release(This);
1075
1076     ConnectionPointContainer_Destroy(&This->cps);
1077
1078     while(This->travellog.length)
1079         free_travellog_entry(This->travellog.log + --This->travellog.length);
1080     heap_free(This->travellog.log);
1081
1082     heap_free(This->url);
1083 }