dlls/mshtml/persist.c

   1 /*
   2  * Copyright 2005 Jacek Caban
   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 "shlguid.h"
  31 #include "idispids.h"
  32 #include "mimeole.h"
  33
  34 #define NO_SHLWAPI_REG
  35 #include "shlwapi.h"
  36
  37 #include "wine/debug.h"
  38
  39 #include "mshtml_private.h"
  40 #include "htmlscript.h"
  41 #include "htmlevent.h"
  42 #include "binding.h"
  43 #include "resource.h"
  44
  45 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  46
  47 /* Undocumented notification, see tests */
  48 #define CMDID_EXPLORER_UPDATEHISTORY 38
  49
  50 static const WCHAR about_blankW[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
  51
  52 typedef struct {
  53     task_t header;
  54     HTMLDocumentObj *doc;
  55     BOOL set_download;
  56     LPOLESTR url;
  57 } download_proc_task_t;
  58
  59 static BOOL use_gecko_script(HTMLOuterWindow *window)
  60 {
  61     DWORD zone;
  62     HRESULT hres;
  63
  64     hres = IInternetSecurityManager_MapUrlToZone(window->secmgr, window->url, &zone, 0);
  65     if(FAILED(hres)) {
  66         WARN("Could not map %s to zone: %08x\n", debugstr_w(window->url), hres);
  67         return TRUE;
  68     }
  69
  70     TRACE("zone %d\n", zone);
  71     return zone == URLZONE_UNTRUSTED;
  72 }
  73
  74 static void notify_travellog_update(HTMLDocumentObj *doc)
  75 {
  76     IOleCommandTarget *cmdtrg;
  77     HRESULT hres;
  78
  79     if(!doc->webbrowser)
  80         return;
  81
  82     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&cmdtrg);
  83     if(SUCCEEDED(hres)) {
  84         VARIANT vin;
  85
  86         V_VT(&vin) = VT_I4;
  87         V_I4(&vin) = 0;
  88
  89         IOleCommandTarget_Exec(cmdtrg, &CGID_Explorer, CMDID_EXPLORER_UPDATEHISTORY, 0, &vin, NULL);
  90         IOleCommandTarget_Release(cmdtrg);
  91     }
  92 }
  93
  94 void set_current_uri(HTMLOuterWindow *window, IUri *uri)
  95 {
  96     if(window->uri) {
  97         IUri_Release(window->uri);
  98         window->uri = NULL;
  99     }
 100
 101     if(window->uri_nofrag) {
 102         IUri_Release(window->uri_nofrag);
 103         window->uri_nofrag = NULL;
 104     }
 105
 106     SysFreeString(window->url);
 107     window->url = NULL;
 108
 109     if(!uri)
 110         return;
 111
 112     IUri_AddRef(uri);
 113     window->uri = uri;
 114
 115     window->uri_nofrag = get_uri_nofrag(uri);
 116     if(!window->uri_nofrag) {
 117         FIXME("get_uri_nofrag failed\n");
 118         IUri_AddRef(uri);
 119         window->uri_nofrag = uri;
 120     }
 121
 122     IUri_GetDisplayUri(uri, &window->url);
 123 }
 124
 125 void set_current_mon(HTMLOuterWindow *This, IMoniker *mon, DWORD flags)
 126 {
 127     IUriContainer *uri_container;
 128     IUri *uri = NULL;
 129     HRESULT hres;
 130
 131     if(This->mon) {
 132         if(This->doc_obj && !(flags & (BINDING_REPLACE|BINDING_REFRESH))) {
 133             if(This == This->doc_obj->basedoc.window)
 134                 notify_travellog_update(This->doc_obj);
 135             else
 136                 TRACE("Skipping travellog update for frame navigation.\n");
 137         }
 138         IMoniker_Release(This->mon);
 139         This->mon = NULL;
 140     }
 141
 142     This->load_flags = flags;
 143     if(!mon)
 144         return;
 145
 146     IMoniker_AddRef(mon);
 147     This->mon = mon;
 148
 149     hres = IMoniker_QueryInterface(mon, &IID_IUriContainer, (void**)&uri_container);
 150     if(SUCCEEDED(hres)) {
 151         hres = IUriContainer_GetIUri(uri_container, &uri);
 152         IUriContainer_Release(uri_container);
 153         if(hres != S_OK) {
 154             WARN("GetIUri failed: %08x\n", hres);
 155             uri = NULL;
 156         }
 157     }
 158
 159     if(!uri) {
 160         WCHAR *url;
 161
 162         hres = IMoniker_GetDisplayName(mon, NULL, NULL, &url);
 163         if(SUCCEEDED(hres)) {
 164             hres = create_uri(url, 0, &uri);
 165             if(FAILED(hres)) {
 166                 WARN("CreateUri failed: %08x\n", hres);
 167                 set_current_uri(This, NULL);
 168                 This->url = SysAllocString(url);
 169                 CoTaskMemFree(url);
 170                 return;
 171             }
 172             CoTaskMemFree(url);
 173         }else {
 174             WARN("GetDisplayName failed: %08x\n", hres);
 175         }
 176     }
 177
 178     set_current_uri(This, uri);
 179     if(uri)
 180         IUri_Release(uri);
 181     set_script_mode(This, use_gecko_script(This) ? SCRIPTMODE_GECKO : SCRIPTMODE_ACTIVESCRIPT);
 182 }
 183
 184 HRESULT create_uri(const WCHAR *uri_str, DWORD flags, IUri **uri)
 185 {
 186     return CreateUri(uri_str, flags | Uri_CREATE_ALLOW_IMPLICIT_FILE_SCHEME, 0, uri);
 187 }
 188
 189 HRESULT create_relative_uri(HTMLOuterWindow *window, const WCHAR *rel_uri, IUri **uri)
 190 {
 191     return window->uri
 192         ? CoInternetCombineUrlEx(window->uri, rel_uri, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO, uri, 0)
 193         : create_uri(rel_uri, 0, uri);
 194 }
 195
 196 void set_download_state(HTMLDocumentObj *doc, int state)
 197 {
 198     if(doc->client) {
 199         IOleCommandTarget *olecmd;
 200         HRESULT hres;
 201
 202         hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 203         if(SUCCEEDED(hres)) {
 204             VARIANT var;
 205
 206             V_VT(&var) = VT_I4;
 207             V_I4(&var) = state;
 208
 209             IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETDOWNLOADSTATE,
 210                     OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
 211             IOleCommandTarget_Release(olecmd);
 212         }
 213     }
 214
 215     doc->download_state = state;
 216 }
 217
 218 static void set_progress_proc(task_t *_task)
 219 {
 220     docobj_task_t *task = (docobj_task_t*)_task;
 221     IOleCommandTarget *olecmd = NULL;
 222     HTMLDocumentObj *doc = task->doc;
 223     HRESULT hres;
 224
 225     TRACE("(%p)\n", doc);
 226
 227     if(doc->client)
 228         IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 229
 230     if(olecmd) {
 231         VARIANT progress_max, progress;
 232
 233         V_VT(&progress_max) = VT_I4;
 234         V_I4(&progress_max) = 0; /* FIXME */
 235         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSMAX, OLECMDEXECOPT_DONTPROMPTUSER,
 236                                &progress_max, NULL);
 237
 238         V_VT(&progress) = VT_I4;
 239         V_I4(&progress) = 0; /* FIXME */
 240         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS, OLECMDEXECOPT_DONTPROMPTUSER,
 241                                &progress, NULL);
 242         IOleCommandTarget_Release(olecmd);
 243     }
 244
 245     if(doc->usermode == EDITMODE && doc->hostui) {
 246         DOCHOSTUIINFO hostinfo;
 247
 248         memset(&hostinfo, 0, sizeof(DOCHOSTUIINFO));
 249         hostinfo.cbSize = sizeof(DOCHOSTUIINFO);
 250         hres = IDocHostUIHandler_GetHostInfo(doc->hostui, &hostinfo);
 251         if(SUCCEEDED(hres))
 252             /* FIXME: use hostinfo */
 253             TRACE("hostinfo = {%u %08x %08x %s %s}\n",
 254                     hostinfo.cbSize, hostinfo.dwFlags, hostinfo.dwDoubleClick,
 255                     debugstr_w(hostinfo.pchHostCss), debugstr_w(hostinfo.pchHostNS));
 256     }
 257 }
 258
 259 static void set_downloading_proc(task_t *_task)
 260 {
 261     download_proc_task_t *task = (download_proc_task_t*)_task;
 262     HTMLDocumentObj *doc = task->doc;
 263     HRESULT hres;
 264
 265     TRACE("(%p)\n", doc);
 266
 267     set_statustext(doc, IDS_STATUS_DOWNLOADINGFROM, task->url);
 268
 269     if(task->set_download)
 270         set_download_state(doc, 1);
 271
 272     if(!doc->client)
 273         return;
 274
 275     if(doc->view_sink)
 276         IAdviseSink_OnViewChange(doc->view_sink, DVASPECT_CONTENT, -1);
 277
 278     if(doc->hostui) {
 279         IDropTarget *drop_target = NULL;
 280
 281         hres = IDocHostUIHandler_GetDropTarget(doc->hostui, NULL /* FIXME */, &drop_target);
 282         if(SUCCEEDED(hres) && drop_target) {
 283             FIXME("Use IDropTarget\n");
 284             IDropTarget_Release(drop_target);
 285         }
 286     }
 287 }
 288
 289 static void set_downloading_task_destr(task_t *_task)
 290 {
 291     download_proc_task_t *task = (download_proc_task_t*)_task;
 292
 293     CoTaskMemFree(task->url);
 294     heap_free(task);
 295 }
 296
 297 void prepare_for_binding(HTMLDocument *This, IMoniker *mon, DWORD flags)
 298 {
 299     HRESULT hres;
 300
 301     if(This->doc_obj->client) {
 302         VARIANT silent, offline;
 303
 304         hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_SILENT, &silent);
 305         if(SUCCEEDED(hres)) {
 306             if(V_VT(&silent) != VT_BOOL)
 307                 WARN("silent = %s\n", debugstr_variant(&silent));
 308             else if(V_BOOL(&silent))
 309                 FIXME("silent == true\n");
 310         }
 311
 312         hres = get_client_disp_property(This->doc_obj->client,
 313                 DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &offline);
 314         if(SUCCEEDED(hres)) {
 315             if(V_VT(&offline) != VT_BOOL)
 316                 WARN("offline = %s\n", debugstr_variant(&offline));
 317             else if(V_BOOL(&offline))
 318                 FIXME("offline == true\n");
 319         }
 320     }
 321
 322     if(This->window->mon) {
 323         update_doc(This, UPDATE_TITLE|UPDATE_UI);
 324     }else {
 325         update_doc(This, UPDATE_TITLE);
 326         set_current_mon(This->window, mon, flags);
 327     }
 328
 329     if(This->doc_obj->client) {
 330         IOleCommandTarget *cmdtrg = NULL;
 331
 332         hres = IOleClientSite_QueryInterface(This->doc_obj->client, &IID_IOleCommandTarget,
 333                 (void**)&cmdtrg);
 334         if(SUCCEEDED(hres)) {
 335             VARIANT var, out;
 336
 337             if(flags & BINDING_NAVIGATED) {
 338                 V_VT(&var) = VT_UNKNOWN;
 339                 V_UNKNOWN(&var) = (IUnknown*)&This->window->base.IHTMLWindow2_iface;
 340                 V_VT(&out) = VT_EMPTY;
 341                 hres = IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 63, 0, &var, &out);
 342                 if(SUCCEEDED(hres))
 343                     VariantClear(&out);
 344             }else if(!(flags & BINDING_FROMHIST)) {
 345                 V_VT(&var) = VT_I4;
 346                 V_I4(&var) = 0;
 347                 IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
 348             }
 349
 350             IOleCommandTarget_Release(cmdtrg);
 351         }
 352     }
 353 }
 354
 355 HRESULT set_moniker(HTMLOuterWindow *window, IMoniker *mon, IUri *nav_uri, IBindCtx *pibc, nsChannelBSC *async_bsc,
 356         BOOL set_download)
 357 {
 358     download_proc_task_t *download_task;
 359     HTMLDocumentObj *doc_obj = NULL;
 360     nsChannelBSC *bscallback;
 361     nsWineURI *nsuri;
 362     LPOLESTR url;
 363     IUri *uri;
 364     HRESULT hres;
 365
 366     if(window->doc_obj && window->doc_obj->basedoc.window == window)
 367         doc_obj = window->doc_obj;
 368
 369     hres = IMoniker_GetDisplayName(mon, pibc, NULL, &url);
 370     if(FAILED(hres)) {
 371         WARN("GetDiaplayName failed: %08x\n", hres);
 372         return hres;
 373     }
 374
 375     if(nav_uri) {
 376         uri = nav_uri;
 377     }else {
 378         hres = create_uri(url, 0, &uri);
 379         if(FAILED(hres)) {
 380             CoTaskMemFree(url);
 381             return hres;
 382         }
 383     }
 384
 385     TRACE("got url: %s\n", debugstr_w(url));
 386
 387     set_ready_state(window, READYSTATE_LOADING);
 388
 389     hres = create_doc_uri(window, uri, &nsuri);
 390     if(!nav_uri)
 391         IUri_Release(uri);
 392     if(SUCCEEDED(hres)) {
 393         if(async_bsc)
 394             bscallback = async_bsc;
 395         else
 396             hres = create_channelbsc(mon, NULL, NULL, 0, TRUE, &bscallback);
 397     }
 398
 399     if(SUCCEEDED(hres)) {
 400         if(window->base.inner_window->doc)
 401             remove_target_tasks(window->base.inner_window->task_magic);
 402         abort_window_bindings(window->base.inner_window);
 403
 404         hres = load_nsuri(window, nsuri, NULL, bscallback, LOAD_FLAGS_BYPASS_CACHE);
 405         nsISupports_Release((nsISupports*)nsuri); /* FIXME */
 406         if(SUCCEEDED(hres)) {
 407             hres = create_pending_window(window, bscallback);
 408             TRACE("pending window for %p %p %p\n", window, bscallback, window->pending_window);
 409         }
 410         if(bscallback != async_bsc)
 411             IBindStatusCallback_Release(&bscallback->bsc.IBindStatusCallback_iface);
 412     }
 413
 414     if(FAILED(hres)) {
 415         CoTaskMemFree(url);
 416         return hres;
 417     }
 418
 419     if(doc_obj) {
 420         HTMLDocument_LockContainer(doc_obj, TRUE);
 421
 422         if(doc_obj->frame) {
 423             docobj_task_t *task;
 424
 425             task = heap_alloc(sizeof(docobj_task_t));
 426             task->doc = doc_obj;
 427             hres = push_task(&task->header, set_progress_proc, NULL, doc_obj->basedoc.task_magic);
 428             if(FAILED(hres)) {
 429                 CoTaskMemFree(url);
 430                 return hres;
 431             }
 432         }
 433
 434         download_task = heap_alloc(sizeof(download_proc_task_t));
 435         download_task->doc = doc_obj;
 436         download_task->set_download = set_download;
 437         download_task->url = url;
 438         return push_task(&download_task->header, set_downloading_proc, set_downloading_task_destr, doc_obj->basedoc.task_magic);
 439     }
 440
 441     return S_OK;
 442 }
 443
 444 static void notif_readystate(HTMLOuterWindow *window)
 445 {
 446     window->readystate_pending = FALSE;
 447
 448     if(window->doc_obj && window->doc_obj->basedoc.window == window)
 449         call_property_onchanged(&window->doc_obj->basedoc.cp_container, DISPID_READYSTATE);
 450
 451     fire_event(window->base.inner_window->doc, EVENTID_READYSTATECHANGE, FALSE,
 452             &window->base.inner_window->doc->node, NULL, NULL);
 453
 454     if(window->frame_element)
 455         fire_event(window->frame_element->element.node.doc, EVENTID_READYSTATECHANGE,
 456                    TRUE, &window->frame_element->element.node, NULL, NULL);
 457 }
 458
 459 typedef struct {
 460     task_t header;
 461     HTMLOuterWindow *window;
 462 } readystate_task_t;
 463
 464 static void notif_readystate_proc(task_t *_task)
 465 {
 466     readystate_task_t *task = (readystate_task_t*)_task;
 467     notif_readystate(task->window);
 468 }
 469
 470 static void notif_readystate_destr(task_t *_task)
 471 {
 472     readystate_task_t *task = (readystate_task_t*)_task;
 473     IHTMLWindow2_Release(&task->window->base.IHTMLWindow2_iface);
 474 }
 475
 476 void set_ready_state(HTMLOuterWindow *window, READYSTATE readystate)
 477 {
 478     READYSTATE prev_state = window->readystate;
 479
 480     window->readystate = readystate;
 481
 482     if(window->readystate_locked) {
 483         readystate_task_t *task;
 484         HRESULT hres;
 485
 486         if(window->readystate_pending || prev_state == readystate)
 487             return;
 488
 489         task = heap_alloc(sizeof(*task));
 490         if(!task)
 491             return;
 492
 493         IHTMLWindow2_AddRef(&window->base.IHTMLWindow2_iface);
 494         task->window = window;
 495
 496         hres = push_task(&task->header, notif_readystate_proc, notif_readystate_destr, window->task_magic);
 497         if(SUCCEEDED(hres))
 498             window->readystate_pending = TRUE;
 499         return;
 500     }
 501
 502     notif_readystate(window);
 503 }
 504
 505 static HRESULT get_doc_string(HTMLDocumentNode *This, char **str)
 506 {
 507     nsIDOMNode *nsnode;
 508     LPCWSTR strw;
 509     nsAString nsstr;
 510     nsresult nsres;
 511     HRESULT hres;
 512
 513     if(!This->nsdoc) {
 514         WARN("NULL nsdoc\n");
 515         return E_UNEXPECTED;
 516     }
 517
 518     nsres = nsIDOMHTMLDocument_QueryInterface(This->nsdoc, &IID_nsIDOMNode, (void**)&nsnode);
 519     if(NS_FAILED(nsres)) {
 520         ERR("Could not get nsIDOMNode failed: %08x\n", nsres);
 521         return E_FAIL;
 522     }
 523
 524     nsAString_Init(&nsstr, NULL);
 525     hres = nsnode_to_nsstring(nsnode, &nsstr);
 526     nsIDOMNode_Release(nsnode);
 527     if(FAILED(hres)) {
 528         nsAString_Finish(&nsstr);
 529         return hres;
 530     }
 531
 532     nsAString_GetData(&nsstr, &strw);
 533     TRACE("%s\n", debugstr_w(strw));
 534
 535     *str = heap_strdupWtoA(strw);
 536
 537     nsAString_Finish(&nsstr);
 538
 539     if(!*str)
 540         return E_OUTOFMEMORY;
 541     return S_OK;
 542 }
 543
 544
 545 /**********************************************************
 546  * IPersistMoniker implementation
 547  */
 548
 549 static inline HTMLDocument *impl_from_IPersistMoniker(IPersistMoniker *iface)
 550 {
 551     return CONTAINING_RECORD(iface, HTMLDocument, IPersistMoniker_iface);
 552 }
 553
 554 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid, void **ppv)
 555 {
 556     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 557     return htmldoc_query_interface(This, riid, ppv);
 558 }
 559
 560 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
 561 {
 562     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 563     return htmldoc_addref(This);
 564 }
 565
 566 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
 567 {
 568     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 569     return htmldoc_release(This);
 570 }
 571
 572 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
 573 {
 574     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 575     return IPersistFile_GetClassID(&This->IPersistFile_iface, pClassID);
 576 }
 577
 578 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
 579 {
 580     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 581
 582     TRACE("(%p)\n", This);
 583
 584     return IPersistStreamInit_IsDirty(&This->IPersistStreamInit_iface);
 585 }
 586
 587 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
 588         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
 589 {
 590     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 591     IMoniker *mon;
 592     HRESULT hres;
 593
 594     TRACE("(%p)->(%x %p %p %08x)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
 595
 596     if(pibc) {
 597         IUnknown *unk = NULL;
 598
 599         /* FIXME:
 600          * Use params:
 601          * "__PrecreatedObject"
 602          * "BIND_CONTEXT_PARAM"
 603          * "__HTMLLOADOPTIONS"
 604          * "__DWNBINDINFO"
 605          * "URL Context"
 606          * "_ITransData_Object_"
 607          * "_EnumFORMATETC_"
 608          */
 609
 610         hres = IBindCtx_GetObjectParam(pibc, (LPOLESTR)SZ_HTML_CLIENTSITE_OBJECTPARAM, &unk);
 611         if(SUCCEEDED(hres) && unk) {
 612             IOleClientSite *client = NULL;
 613
 614             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
 615             if(SUCCEEDED(hres)) {
 616                 TRACE("Got client site %p\n", client);
 617                 IOleObject_SetClientSite(&This->IOleObject_iface, client);
 618                 IOleClientSite_Release(client);
 619             }
 620
 621             IUnknown_Release(unk);
 622         }
 623     }
 624
 625     if(This->doc_obj->is_mhtml) {
 626         IUnknown *unk;
 627
 628         hres = MimeOleObjectFromMoniker(0, pimkName, pibc, &IID_IUnknown, (void**)&unk, &mon);
 629         if(FAILED(hres))
 630             return hres;
 631         IUnknown_Release(unk);
 632         pibc = NULL;
 633     }else {
 634         IMoniker_AddRef(mon = pimkName);
 635     }
 636
 637     prepare_for_binding(This, mon, FALSE);
 638     call_docview_84(This->doc_obj);
 639     hres = set_moniker(This->window, mon, NULL, pibc, NULL, TRUE);
 640     IMoniker_Release(mon);
 641     if(FAILED(hres))
 642         return hres;
 643
 644     return start_binding(This->window->pending_window, (BSCallback*)This->window->pending_window->bscallback, pibc);
 645 }
 646
 647 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
 648         LPBC pbc, BOOL fRemember)
 649 {
 650     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 651     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
 652     return E_NOTIMPL;
 653 }
 654
 655 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
 656 {
 657     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 658     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
 659     return E_NOTIMPL;
 660 }
 661
 662 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
 663 {
 664     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 665
 666     TRACE("(%p)->(%p)\n", This, ppimkName);
 667
 668     if(!This->window || !This->window->mon)
 669         return E_UNEXPECTED;
 670
 671     IMoniker_AddRef(This->window->mon);
 672     *ppimkName = This->window->mon;
 673     return S_OK;
 674 }
 675
 676 static const IPersistMonikerVtbl PersistMonikerVtbl = {
 677     PersistMoniker_QueryInterface,
 678     PersistMoniker_AddRef,
 679     PersistMoniker_Release,
 680     PersistMoniker_GetClassID,
 681     PersistMoniker_IsDirty,
 682     PersistMoniker_Load,
 683     PersistMoniker_Save,
 684     PersistMoniker_SaveCompleted,
 685     PersistMoniker_GetCurMoniker
 686 };
 687
 688 /**********************************************************
 689  * IMonikerProp implementation
 690  */
 691
 692 static inline HTMLDocument *impl_from_IMonikerProp(IMonikerProp *iface)
 693 {
 694     return CONTAINING_RECORD(iface, HTMLDocument, IMonikerProp_iface);
 695 }
 696
 697 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppv)
 698 {
 699     HTMLDocument *This = impl_from_IMonikerProp(iface);
 700     return htmldoc_query_interface(This, riid, ppv);
 701 }
 702
 703 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
 704 {
 705     HTMLDocument *This = impl_from_IMonikerProp(iface);
 706     return htmldoc_addref(This);
 707 }
 708
 709 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
 710 {
 711     HTMLDocument *This = impl_from_IMonikerProp(iface);
 712     return htmldoc_release(This);
 713 }
 714
 715 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
 716 {
 717     HTMLDocument *This = impl_from_IMonikerProp(iface);
 718
 719     TRACE("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
 720
 721     switch(mkp) {
 722     case MIMETYPEPROP:
 723         heap_free(This->doc_obj->mime);
 724         This->doc_obj->mime = heap_strdupW(val);
 725         break;
 726
 727     case CLASSIDPROP:
 728         break;
 729
 730     default:
 731         FIXME("mkp %d\n", mkp);
 732         return E_NOTIMPL;
 733     }
 734
 735     return S_OK;
 736 }
 737
 738 static const IMonikerPropVtbl MonikerPropVtbl = {
 739     MonikerProp_QueryInterface,
 740     MonikerProp_AddRef,
 741     MonikerProp_Release,
 742     MonikerProp_PutProperty
 743 };
 744
 745 /**********************************************************
 746  * IPersistFile implementation
 747  */
 748
 749 static inline HTMLDocument *impl_from_IPersistFile(IPersistFile *iface)
 750 {
 751     return CONTAINING_RECORD(iface, HTMLDocument, IPersistFile_iface);
 752 }
 753
 754 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppv)
 755 {
 756     HTMLDocument *This = impl_from_IPersistFile(iface);
 757     return htmldoc_query_interface(This, riid, ppv);
 758 }
 759
 760 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
 761 {
 762     HTMLDocument *This = impl_from_IPersistFile(iface);
 763     return htmldoc_addref(This);
 764 }
 765
 766 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
 767 {
 768     HTMLDocument *This = impl_from_IPersistFile(iface);
 769     return htmldoc_release(This);
 770 }
 771
 772 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
 773 {
 774     HTMLDocument *This = impl_from_IPersistFile(iface);
 775
 776     TRACE("(%p)->(%p)\n", This, pClassID);
 777
 778     if(!pClassID)
 779         return E_INVALIDARG;
 780
 781     *pClassID = CLSID_HTMLDocument;
 782     return S_OK;
 783 }
 784
 785 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
 786 {
 787     HTMLDocument *This = impl_from_IPersistFile(iface);
 788
 789     TRACE("(%p)\n", This);
 790
 791     return IPersistStreamInit_IsDirty(&This->IPersistStreamInit_iface);
 792 }
 793
 794 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
 795 {
 796     HTMLDocument *This = impl_from_IPersistFile(iface);
 797     FIXME("(%p)->(%s %08x)\n", This, debugstr_w(pszFileName), dwMode);
 798     return E_NOTIMPL;
 799 }
 800
 801 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
 802 {
 803     HTMLDocument *This = impl_from_IPersistFile(iface);
 804     char *str;
 805     DWORD written=0;
 806     HANDLE file;
 807     HRESULT hres;
 808
 809     TRACE("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
 810
 811     file = CreateFileW(pszFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
 812                        FILE_ATTRIBUTE_NORMAL, NULL);
 813     if(file == INVALID_HANDLE_VALUE) {
 814         WARN("Could not create file: %u\n", GetLastError());
 815         return E_FAIL;
 816     }
 817
 818     hres = get_doc_string(This->doc_node, &str);
 819     if(SUCCEEDED(hres))
 820         WriteFile(file, str, strlen(str), &written, NULL);
 821
 822     CloseHandle(file);
 823     return hres;
 824 }
 825
 826 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
 827 {
 828     HTMLDocument *This = impl_from_IPersistFile(iface);
 829     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
 830     return E_NOTIMPL;
 831 }
 832
 833 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
 834 {
 835     HTMLDocument *This = impl_from_IPersistFile(iface);
 836     FIXME("(%p)->(%p)\n", This, pszFileName);
 837     return E_NOTIMPL;
 838 }
 839
 840 static const IPersistFileVtbl PersistFileVtbl = {
 841     PersistFile_QueryInterface,
 842     PersistFile_AddRef,
 843     PersistFile_Release,
 844     PersistFile_GetClassID,
 845     PersistFile_IsDirty,
 846     PersistFile_Load,
 847     PersistFile_Save,
 848     PersistFile_SaveCompleted,
 849     PersistFile_GetCurFile
 850 };
 851
 852 static inline HTMLDocument *impl_from_IPersistStreamInit(IPersistStreamInit *iface)
 853 {
 854     return CONTAINING_RECORD(iface, HTMLDocument, IPersistStreamInit_iface);
 855 }
 856
 857 static HRESULT WINAPI PersistStreamInit_QueryInterface(IPersistStreamInit *iface,
 858                                                        REFIID riid, void **ppv)
 859 {
 860     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 861     return htmldoc_query_interface(This, riid, ppv);
 862 }
 863
 864 static ULONG WINAPI PersistStreamInit_AddRef(IPersistStreamInit *iface)
 865 {
 866     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 867     return htmldoc_addref(This);
 868 }
 869
 870 static ULONG WINAPI PersistStreamInit_Release(IPersistStreamInit *iface)
 871 {
 872     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 873     return htmldoc_release(This);
 874 }
 875
 876 static HRESULT WINAPI PersistStreamInit_GetClassID(IPersistStreamInit *iface, CLSID *pClassID)
 877 {
 878     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 879     return IPersistFile_GetClassID(&This->IPersistFile_iface, pClassID);
 880 }
 881
 882 static HRESULT WINAPI PersistStreamInit_IsDirty(IPersistStreamInit *iface)
 883 {
 884     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 885
 886     TRACE("(%p)\n", This);
 887
 888     if(This->doc_obj->usermode == EDITMODE)
 889         return editor_is_dirty(This);
 890
 891     return S_FALSE;
 892 }
 893
 894 static HRESULT WINAPI PersistStreamInit_Load(IPersistStreamInit *iface, IStream *pStm)
 895 {
 896     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 897     IMoniker *mon;
 898     HRESULT hres;
 899
 900     TRACE("(%p)->(%p)\n", This, pStm);
 901
 902     hres = CreateURLMoniker(NULL, about_blankW, &mon);
 903     if(FAILED(hres)) {
 904         WARN("CreateURLMoniker failed: %08x\n", hres);
 905         return hres;
 906     }
 907
 908     prepare_for_binding(This, mon, FALSE);
 909     hres = set_moniker(This->window, mon, NULL, NULL, NULL, TRUE);
 910     if(FAILED(hres))
 911         return hres;
 912
 913     hres = channelbsc_load_stream(This->window->pending_window, mon, pStm);
 914     IMoniker_Release(mon);
 915     return hres;
 916 }
 917
 918 static HRESULT WINAPI PersistStreamInit_Save(IPersistStreamInit *iface, IStream *pStm,
 919                                              BOOL fClearDirty)
 920 {
 921     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 922     char *str;
 923     DWORD written=0;
 924     HRESULT hres;
 925
 926     TRACE("(%p)->(%p %x)\n", This, pStm, fClearDirty);
 927
 928     hres = get_doc_string(This->doc_node, &str);
 929     if(FAILED(hres))
 930         return hres;
 931
 932     hres = IStream_Write(pStm, str, strlen(str), &written);
 933     if(FAILED(hres))
 934         FIXME("Write failed: %08x\n", hres);
 935
 936     heap_free(str);
 937
 938     if(fClearDirty)
 939         set_dirty(This, VARIANT_FALSE);
 940
 941     return S_OK;
 942 }
 943
 944 static HRESULT WINAPI PersistStreamInit_GetSizeMax(IPersistStreamInit *iface,
 945                                                    ULARGE_INTEGER *pcbSize)
 946 {
 947     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 948     FIXME("(%p)->(%p)\n", This, pcbSize);
 949     return E_NOTIMPL;
 950 }
 951
 952 static HRESULT WINAPI PersistStreamInit_InitNew(IPersistStreamInit *iface)
 953 {
 954     HTMLDocument *This = impl_from_IPersistStreamInit(iface);
 955     IMoniker *mon;
 956     HRESULT hres;
 957
 958     TRACE("(%p)\n", This);
 959
 960     hres = CreateURLMoniker(NULL, about_blankW, &mon);
 961     if(FAILED(hres)) {
 962         WARN("CreateURLMoniker failed: %08x\n", hres);
 963         return hres;
 964     }
 965
 966     prepare_for_binding(This, mon, FALSE);
 967     hres = set_moniker(This->window, mon, NULL, NULL, NULL, FALSE);
 968     if(FAILED(hres))
 969         return hres;
 970
 971     hres = channelbsc_load_stream(This->window->pending_window, mon, NULL);
 972     IMoniker_Release(mon);
 973     return hres;
 974 }
 975
 976 static const IPersistStreamInitVtbl PersistStreamInitVtbl = {
 977     PersistStreamInit_QueryInterface,
 978     PersistStreamInit_AddRef,
 979     PersistStreamInit_Release,
 980     PersistStreamInit_GetClassID,
 981     PersistStreamInit_IsDirty,
 982     PersistStreamInit_Load,
 983     PersistStreamInit_Save,
 984     PersistStreamInit_GetSizeMax,
 985     PersistStreamInit_InitNew
 986 };
 987
 988 /**********************************************************
 989  * IPersistHistory implementation
 990  */
 991
 992 static inline HTMLDocument *impl_from_IPersistHistory(IPersistHistory *iface)
 993 {
 994     return CONTAINING_RECORD(iface, HTMLDocument, IPersistHistory_iface);
 995 }
 996
 997 static HRESULT WINAPI PersistHistory_QueryInterface(IPersistHistory *iface, REFIID riid, void **ppv)
 998 {
 999     HTMLDocument *This = impl_from_IPersistHistory(iface);
1000     return htmldoc_query_interface(This, riid, ppv);
1001 }
1002
1003 static ULONG WINAPI PersistHistory_AddRef(IPersistHistory *iface)
1004 {
1005     HTMLDocument *This = impl_from_IPersistHistory(iface);
1006     return htmldoc_addref(This);
1007 }
1008
1009 static ULONG WINAPI PersistHistory_Release(IPersistHistory *iface)
1010 {
1011     HTMLDocument *This = impl_from_IPersistHistory(iface);
1012     return htmldoc_release(This);
1013 }
1014
1015 static HRESULT WINAPI PersistHistory_GetClassID(IPersistHistory *iface, CLSID *pClassID)
1016 {
1017     HTMLDocument *This = impl_from_IPersistHistory(iface);
1018     return IPersistFile_GetClassID(&This->IPersistFile_iface, pClassID);
1019 }
1020
1021 static HRESULT WINAPI PersistHistory_LoadHistory(IPersistHistory *iface, IStream *pStream, IBindCtx *pbc)
1022 {
1023     HTMLDocument *This = impl_from_IPersistHistory(iface);
1024     ULONG str_len, read;
1025     WCHAR *uri_str;
1026     IUri *uri;
1027     HRESULT hres;
1028
1029     TRACE("(%p)->(%p %p)\n", This, pStream, pbc);
1030
1031     if(!This->window) {
1032         FIXME("No current window\n");
1033         return E_UNEXPECTED;
1034     }
1035
1036     if(pbc)
1037         FIXME("pbc not supported\n");
1038
1039     if(This->doc_obj->client) {
1040         IOleCommandTarget *cmdtrg = NULL;
1041
1042         hres = IOleClientSite_QueryInterface(This->doc_obj->client, &IID_IOleCommandTarget,
1043                 (void**)&cmdtrg);
1044         if(SUCCEEDED(hres)) {
1045             IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 138, 0, NULL, NULL);
1046             IOleCommandTarget_Release(cmdtrg);
1047         }
1048     }
1049
1050     hres = IStream_Read(pStream, &str_len, sizeof(str_len), &read);
1051     if(FAILED(hres))
1052         return hres;
1053     if(read != sizeof(str_len))
1054         return E_FAIL;
1055
1056     uri_str = heap_alloc((str_len+1)*sizeof(WCHAR));
1057     if(!uri_str)
1058         return E_OUTOFMEMORY;
1059
1060     hres = IStream_Read(pStream, uri_str, str_len*sizeof(WCHAR), &read);
1061     if(SUCCEEDED(hres) && read != str_len*sizeof(WCHAR))
1062         hres = E_FAIL;
1063     if(SUCCEEDED(hres)) {
1064         uri_str[str_len] = 0;
1065         hres = create_uri(uri_str, 0, &uri);
1066     }
1067     heap_free(uri_str);
1068     if(FAILED(hres))
1069         return hres;
1070
1071     hres = load_uri(This->window, uri, BINDING_FROMHIST);
1072     IUri_Release(uri);
1073     return hres;
1074 }
1075
1076 static HRESULT WINAPI PersistHistory_SaveHistory(IPersistHistory *iface, IStream *pStream)
1077 {
1078     HTMLDocument *This = impl_from_IPersistHistory(iface);
1079     ULONG len, written;
1080     BSTR display_uri;
1081     HRESULT hres;
1082
1083     TRACE("(%p)->(%p)\n", This, pStream);
1084
1085     if(!This->window || !This->window->uri) {
1086         FIXME("No current URI\n");
1087         return E_FAIL;
1088     }
1089
1090     /* NOTE: The format we store is *not* compatible with native MSHTML. We currently
1091      * store only URI of the page (as a length followed by a string) */
1092     hres = IUri_GetDisplayUri(This->window->uri, &display_uri);
1093     if(FAILED(hres))
1094         return hres;
1095
1096     len = SysStringLen(display_uri);
1097     hres = IStream_Write(pStream, &len, sizeof(len), &written);
1098     if(SUCCEEDED(hres))
1099         hres = IStream_Write(pStream, display_uri, len*sizeof(WCHAR), &written);
1100     SysFreeString(display_uri);
1101     return hres;
1102 }
1103
1104 static HRESULT WINAPI PersistHistory_SetPositionCookie(IPersistHistory *iface, DWORD dwPositioncookie)
1105 {
1106     HTMLDocument *This = impl_from_IPersistHistory(iface);
1107     FIXME("(%p)->(%x)\n", This, dwPositioncookie);
1108     return E_NOTIMPL;
1109 }
1110
1111 static HRESULT WINAPI PersistHistory_GetPositionCookie(IPersistHistory *iface, DWORD *pdwPositioncookie)
1112 {
1113     HTMLDocument *This = impl_from_IPersistHistory(iface);
1114     FIXME("(%p)->(%p)\n", This, pdwPositioncookie);
1115     return E_NOTIMPL;
1116 }
1117
1118 static const IPersistHistoryVtbl PersistHistoryVtbl = {
1119     PersistHistory_QueryInterface,
1120     PersistHistory_AddRef,
1121     PersistHistory_Release,
1122     PersistHistory_GetClassID,
1123     PersistHistory_LoadHistory,
1124     PersistHistory_SaveHistory,
1125     PersistHistory_SetPositionCookie,
1126     PersistHistory_GetPositionCookie
1127 };
1128
1129 void HTMLDocument_Persist_Init(HTMLDocument *This)
1130 {
1131     This->IPersistMoniker_iface.lpVtbl = &PersistMonikerVtbl;
1132     This->IPersistFile_iface.lpVtbl = &PersistFileVtbl;
1133     This->IMonikerProp_iface.lpVtbl = &MonikerPropVtbl;
1134     This->IPersistStreamInit_iface.lpVtbl = &PersistStreamInitVtbl;
1135     This->IPersistHistory_iface.lpVtbl = &PersistHistoryVtbl;
1136 }