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 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->is_webbrowser)
  79         return;
  80
  81     /* Don't notify if we were in about: page */
  82     if(doc->basedoc.window->uri) {
  83         DWORD scheme;
  84
  85         hres = IUri_GetScheme(doc->basedoc.window->uri, &scheme);
  86         if(SUCCEEDED(hres) && scheme == URL_SCHEME_ABOUT)
  87             return;
  88     }
  89
  90     hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&cmdtrg);
  91     if(SUCCEEDED(hres)) {
  92         VARIANT vin;
  93
  94         V_VT(&vin) = VT_I4;
  95         V_I4(&vin) = 0;
  96
  97         IOleCommandTarget_Exec(cmdtrg, &CGID_Explorer, CMDID_EXPLORER_UPDATEHISTORY, 0, &vin, NULL);
  98         IOleCommandTarget_Release(cmdtrg);
  99     }
 100 }
 101
 102 void set_current_uri(HTMLOuterWindow *window, IUri *uri)
 103 {
 104     if(window->uri) {
 105         IUri_Release(window->uri);
 106         window->uri = NULL;
 107     }
 108
 109     SysFreeString(window->url);
 110     window->url = NULL;
 111
 112     if(!uri)
 113         return;
 114
 115     IUri_AddRef(uri);
 116     window->uri = uri;
 117
 118     IUri_GetDisplayUri(uri, &window->url);
 119 }
 120
 121 void set_current_mon(HTMLOuterWindow *This, IMoniker *mon)
 122 {
 123     IUriContainer *uri_container;
 124     IUri *uri = NULL;
 125     HRESULT hres;
 126
 127     if(This->mon) {
 128         if(This->doc_obj)
 129             notify_travellog_update(This->doc_obj);
 130         IMoniker_Release(This->mon);
 131         This->mon = NULL;
 132     }
 133
 134     if(!mon)
 135         return;
 136
 137     IMoniker_AddRef(mon);
 138     This->mon = mon;
 139
 140     hres = IMoniker_QueryInterface(mon, &IID_IUriContainer, (void**)&uri_container);
 141     if(SUCCEEDED(hres)) {
 142         hres = IUriContainer_GetIUri(uri_container, &uri);
 143         IUriContainer_Release(uri_container);
 144         if(hres != S_OK) {
 145             WARN("GetIUri failed: %08x\n", hres);
 146             uri = NULL;
 147         }
 148     }
 149
 150     if(!uri) {
 151         WCHAR *url;
 152
 153         hres = IMoniker_GetDisplayName(mon, NULL, NULL, &url);
 154         if(SUCCEEDED(hres)) {
 155             hres = CreateUri(url, 0, 0, &uri);
 156             if(FAILED(hres)) {
 157                 WARN("CrateUri failed: %08x\n", hres);
 158                 set_current_uri(This, NULL);
 159                 This->url = SysAllocString(url);
 160                 CoTaskMemFree(url);
 161                 return;
 162             }
 163             CoTaskMemFree(url);
 164         }else {
 165             WARN("GetDisplayName failed: %08x\n", hres);
 166         }
 167     }
 168
 169     set_current_uri(This, uri);
 170     if(uri)
 171         IUri_Release(uri);
 172     set_script_mode(This, use_gecko_script(This) ? SCRIPTMODE_GECKO : SCRIPTMODE_ACTIVESCRIPT);
 173 }
 174
 175 HRESULT create_relative_uri(HTMLOuterWindow *window, const WCHAR *rel_uri, IUri **uri)
 176 {
 177     return window->uri
 178         ? CoInternetCombineUrlEx(window->uri, rel_uri, URL_ESCAPE_SPACES_ONLY|URL_DONT_ESCAPE_EXTRA_INFO, uri, 0)
 179         : CreateUri(rel_uri, 0, 0, uri);
 180 }
 181
 182 void set_download_state(HTMLDocumentObj *doc, int state)
 183 {
 184     if(doc->client) {
 185         IOleCommandTarget *olecmd;
 186         HRESULT hres;
 187
 188         hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 189         if(SUCCEEDED(hres)) {
 190             VARIANT var;
 191
 192             V_VT(&var) = VT_I4;
 193             V_I4(&var) = state;
 194
 195             IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETDOWNLOADSTATE,
 196                     OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
 197             IOleCommandTarget_Release(olecmd);
 198         }
 199     }
 200
 201     doc->download_state = state;
 202 }
 203
 204 static void set_progress_proc(task_t *_task)
 205 {
 206     docobj_task_t *task = (docobj_task_t*)_task;
 207     IOleCommandTarget *olecmd = NULL;
 208     HTMLDocumentObj *doc = task->doc;
 209     HRESULT hres;
 210
 211     TRACE("(%p)\n", doc);
 212
 213     if(doc->client)
 214         IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 215
 216     if(olecmd) {
 217         VARIANT progress_max, progress;
 218
 219         V_VT(&progress_max) = VT_I4;
 220         V_I4(&progress_max) = 0; /* FIXME */
 221         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSMAX, OLECMDEXECOPT_DONTPROMPTUSER,
 222                                &progress_max, NULL);
 223
 224         V_VT(&progress) = VT_I4;
 225         V_I4(&progress) = 0; /* FIXME */
 226         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS, OLECMDEXECOPT_DONTPROMPTUSER,
 227                                &progress, NULL);
 228         IOleCommandTarget_Release(olecmd);
 229     }
 230
 231     if(doc->usermode == EDITMODE && doc->hostui) {
 232         DOCHOSTUIINFO hostinfo;
 233
 234         memset(&hostinfo, 0, sizeof(DOCHOSTUIINFO));
 235         hostinfo.cbSize = sizeof(DOCHOSTUIINFO);
 236         hres = IDocHostUIHandler_GetHostInfo(doc->hostui, &hostinfo);
 237         if(SUCCEEDED(hres))
 238             /* FIXME: use hostinfo */
 239             TRACE("hostinfo = {%u %08x %08x %s %s}\n",
 240                     hostinfo.cbSize, hostinfo.dwFlags, hostinfo.dwDoubleClick,
 241                     debugstr_w(hostinfo.pchHostCss), debugstr_w(hostinfo.pchHostNS));
 242     }
 243 }
 244
 245 static void set_downloading_proc(task_t *_task)
 246 {
 247     download_proc_task_t *task = (download_proc_task_t*)_task;
 248     HTMLDocumentObj *doc = task->doc;
 249     HRESULT hres;
 250
 251     TRACE("(%p)\n", doc);
 252
 253     set_statustext(doc, IDS_STATUS_DOWNLOADINGFROM, task->url);
 254
 255     if(task->set_download)
 256         set_download_state(doc, 1);
 257
 258     if(!doc->client)
 259         return;
 260
 261     if(doc->view_sink)
 262         IAdviseSink_OnViewChange(doc->view_sink, DVASPECT_CONTENT, -1);
 263
 264     if(doc->hostui) {
 265         IDropTarget *drop_target = NULL;
 266
 267         hres = IDocHostUIHandler_GetDropTarget(doc->hostui, NULL /* FIXME */, &drop_target);
 268         if(SUCCEEDED(hres) && drop_target) {
 269             FIXME("Use IDropTarget\n");
 270             IDropTarget_Release(drop_target);
 271         }
 272     }
 273 }
 274
 275 static void set_downloading_task_destr(task_t *_task)
 276 {
 277     download_proc_task_t *task = (download_proc_task_t*)_task;
 278
 279     CoTaskMemFree(task->url);
 280     heap_free(task);
 281 }
 282
 283 void prepare_for_binding(HTMLDocument *This, IMoniker *mon, BOOL navigated_binding)
 284 {
 285     HRESULT hres;
 286
 287     if(This->doc_obj->client) {
 288         VARIANT silent, offline;
 289
 290         hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_SILENT, &silent);
 291         if(SUCCEEDED(hres)) {
 292             if(V_VT(&silent) != VT_BOOL)
 293                 WARN("silent = %s\n", debugstr_variant(&silent));
 294             else if(V_BOOL(&silent))
 295                 FIXME("silent == true\n");
 296         }
 297
 298         hres = get_client_disp_property(This->doc_obj->client,
 299                 DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &offline);
 300         if(SUCCEEDED(hres)) {
 301             if(V_VT(&offline) != VT_BOOL)
 302                 WARN("offline = %s\n", debugstr_variant(&offline));
 303             else if(V_BOOL(&offline))
 304                 FIXME("offline == true\n");
 305         }
 306     }
 307
 308     if(This->window->mon) {
 309         update_doc(This, UPDATE_TITLE|UPDATE_UI);
 310     }else {
 311         update_doc(This, UPDATE_TITLE);
 312         set_current_mon(This->window, mon);
 313     }
 314
 315     if(This->doc_obj->client) {
 316         IOleCommandTarget *cmdtrg = NULL;
 317
 318         hres = IOleClientSite_QueryInterface(This->doc_obj->client, &IID_IOleCommandTarget,
 319                 (void**)&cmdtrg);
 320         if(SUCCEEDED(hres)) {
 321             VARIANT var, out;
 322
 323             if(!navigated_binding) {
 324                 V_VT(&var) = VT_I4;
 325                 V_I4(&var) = 0;
 326                 IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
 327             }else {
 328                 V_VT(&var) = VT_UNKNOWN;
 329                 V_UNKNOWN(&var) = (IUnknown*)&This->window->base.IHTMLWindow2_iface;
 330                 V_VT(&out) = VT_EMPTY;
 331                 hres = IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 63, 0, &var, &out);
 332                 if(SUCCEEDED(hres))
 333                     VariantClear(&out);
 334             }
 335
 336             IOleCommandTarget_Release(cmdtrg);
 337         }
 338     }
 339 }
 340
 341 HRESULT set_moniker(HTMLDocument *This, IMoniker *mon, IBindCtx *pibc, nsChannelBSC *async_bsc, BOOL set_download)
 342 {
 343     download_proc_task_t *download_task;
 344     nsChannelBSC *bscallback;
 345     nsWineURI *nsuri;
 346     LPOLESTR url;
 347     HRESULT hres;
 348
 349     hres = IMoniker_GetDisplayName(mon, pibc, NULL, &url);
 350     if(FAILED(hres)) {
 351         WARN("GetDiaplayName failed: %08x\n", hres);
 352         return hres;
 353     }
 354
 355     TRACE("got url: %s\n", debugstr_w(url));
 356
 357     set_ready_state(This->window, READYSTATE_LOADING);
 358
 359     hres = create_doc_uri(This->window, url, &nsuri);
 360     if(SUCCEEDED(hres)) {
 361         if(async_bsc)
 362             bscallback = async_bsc;
 363         else
 364             hres = create_channelbsc(mon, NULL, NULL, 0, TRUE, &bscallback);
 365     }
 366
 367     if(SUCCEEDED(hres)) {
 368         remove_target_tasks(This->task_magic);
 369         abort_window_bindings(This->window->base.inner_window);
 370
 371         hres = load_nsuri(This->window, nsuri, bscallback, 0/*LOAD_INITIAL_DOCUMENT_URI*/);
 372         nsISupports_Release((nsISupports*)nsuri); /* FIXME */
 373         if(SUCCEEDED(hres))
 374             hres = create_pending_window(This->window, bscallback);
 375         if(bscallback != async_bsc)
 376             IBindStatusCallback_Release(&bscallback->bsc.IBindStatusCallback_iface);
 377     }
 378
 379     if(FAILED(hres)) {
 380         CoTaskMemFree(url);
 381         return hres;
 382     }
 383
 384     HTMLDocument_LockContainer(This->doc_obj, TRUE);
 385
 386     if(This->doc_obj->frame) {
 387         docobj_task_t *task;
 388
 389         task = heap_alloc(sizeof(docobj_task_t));
 390         task->doc = This->doc_obj;
 391         hres = push_task(&task->header, set_progress_proc, NULL, This->doc_obj->basedoc.task_magic);
 392         if(FAILED(hres)) {
 393             CoTaskMemFree(url);
 394             return hres;
 395         }
 396     }
 397
 398     download_task = heap_alloc(sizeof(download_proc_task_t));
 399     download_task->doc = This->doc_obj;
 400     download_task->set_download = set_download;
 401     download_task->url = url;
 402     return push_task(&download_task->header, set_downloading_proc, set_downloading_task_destr, This->doc_obj->basedoc.task_magic);
 403 }
 404
 405 void set_ready_state(HTMLOuterWindow *window, READYSTATE readystate)
 406 {
 407     window->readystate = readystate;
 408
 409     if(window->doc_obj && window->doc_obj->basedoc.window == window)
 410         call_property_onchanged(&window->doc_obj->basedoc.cp_propnotif, DISPID_READYSTATE);
 411
 412     fire_event(window->base.inner_window->doc, EVENTID_READYSTATECHANGE, FALSE, window->base.inner_window->doc->node.nsnode, NULL);
 413
 414     if(window->frame_element)
 415         fire_event(window->frame_element->element.node.doc, EVENTID_READYSTATECHANGE,
 416                    TRUE, window->frame_element->element.node.nsnode, NULL);
 417 }
 418
 419 static HRESULT get_doc_string(HTMLDocumentNode *This, char **str)
 420 {
 421     nsIDOMNode *nsnode;
 422     LPCWSTR strw;
 423     nsAString nsstr;
 424     nsresult nsres;
 425     HRESULT hres;
 426
 427     if(!This->nsdoc) {
 428         WARN("NULL nsdoc\n");
 429         return E_UNEXPECTED;
 430     }
 431
 432     nsres = nsIDOMHTMLDocument_QueryInterface(This->nsdoc, &IID_nsIDOMNode, (void**)&nsnode);
 433     if(NS_FAILED(nsres)) {
 434         ERR("Could not get nsIDOMNode failed: %08x\n", nsres);
 435         return E_FAIL;
 436     }
 437
 438     nsAString_Init(&nsstr, NULL);
 439     hres = nsnode_to_nsstring(nsnode, &nsstr);
 440     nsIDOMNode_Release(nsnode);
 441     if(FAILED(hres)) {
 442         nsAString_Finish(&nsstr);
 443         return hres;
 444     }
 445
 446     nsAString_GetData(&nsstr, &strw);
 447     TRACE("%s\n", debugstr_w(strw));
 448
 449     *str = heap_strdupWtoA(strw);
 450
 451     nsAString_Finish(&nsstr);
 452
 453     if(!*str)
 454         return E_OUTOFMEMORY;
 455     return S_OK;
 456 }
 457
 458
 459 /**********************************************************
 460  * IPersistMoniker implementation
 461  */
 462
 463 static inline HTMLDocument *impl_from_IPersistMoniker(IPersistMoniker *iface)
 464 {
 465     return CONTAINING_RECORD(iface, HTMLDocument, IPersistMoniker_iface);
 466 }
 467
 468 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid, void **ppv)
 469 {
 470     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 471     return htmldoc_query_interface(This, riid, ppv);
 472 }
 473
 474 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
 475 {
 476     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 477     return htmldoc_addref(This);
 478 }
 479
 480 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
 481 {
 482     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 483     return htmldoc_release(This);
 484 }
 485
 486 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
 487 {
 488     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 489     return IPersistFile_GetClassID(&This->IPersistFile_iface, pClassID);
 490 }
 491
 492 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
 493 {
 494     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 495
 496     TRACE("(%p)\n", This);
 497
 498     return IPersistStreamInit_IsDirty(&This->IPersistStreamInit_iface);
 499 }
 500
 501 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
 502         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
 503 {
 504     HTMLDocument *This = impl_from_IPersistMoniker(iface);
 505     HRESULT hres;
 506
 507     TRACE("(%p)->(%x %p %p %08x)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
 508
 509     if(pibc) {
 510         IUnknown *unk = NULL;
 511
 512         /* FIXME:
 513          * Use params:
 514          * "__PrecreatedObject"
 515          * "BIND_CONTEXT_PARAM"
 516          * "__HTMLLOADOPTIONS"
 517          * "__DWNBINDINFO"
 518          * "URL Context"
 519          * "_ITransData_Object_"
 520          * "_EnumFORMATETC_"
 521          */
 522
 523         hres = IBindCtx_GetObjectParam(pibc, (LPOLESTR)SZ_HTML_CLIENTSITE_OBJECTPARAM, &unk);
 524         if(SUCCEEDED(hres) && unk) {
 525             IOleClientSite *client = NULL;
 526
 527             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
 528             if(SUCCEEDED(hres)) {
 529                 TRACE("Got client site %p\n", client);
 530                 IOleObject_SetClientSite(&This->IOleObject_iface, client);
 531                 IOleClientSite_Release(client);
 532             }
 533
 534             IUnknown_Release(unk);
 535         }
 536     }
 537
 538     prepare_for_binding(This, pimkName, FALSE);
 539     call_docview_84(This->doc_obj);
 540     hres = set_moniker(This, pimkName, pibc, NULL, TRUE);
 541     if(FAILED(hres))
 542         return hres;
 543
 544     return start_binding(This->window->pending_window, (BSCallback*)This->window->pending_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 IPersistFile_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->pending_window, 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->pending_window, 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 IPersistFile_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 }