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