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