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