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 #include "wine/debug.h"
  36 #include "wine/unicode.h"
  37
  38 #include "mshtml_private.h"
  39
  40 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  41
  42 #define USER_AGENT "User-Agent:"
  43 #define CONTENT_TYPE "Content-Type:"
  44
  45 static int fix_headers(char *buf, DWORD post_len)
  46 {
  47     char *ptr = buf, *ptr2;
  48
  49     while(*ptr && (ptr[0] != '\r' || ptr[1] != '\n')) {
  50         for(ptr2=ptr+1; *ptr2 && (ptr2[0] != '\r' || ptr2[1] != '\n'); ptr2++);
  51
  52         if(*ptr2)
  53             ptr2 += 2;
  54
  55         if(!strncasecmp(ptr, USER_AGENT, sizeof(USER_AGENT)-1)) {
  56             FIXME("Ignoring User-Agent header\n");
  57             memmove(ptr, ptr2, strlen(ptr2)+1);
  58         }else if(!post_len && !strncasecmp(ptr, CONTENT_TYPE, sizeof(CONTENT_TYPE)-1)) {
  59             TRACE("Ignoring Content-Type header\n");
  60             memmove(ptr, ptr2, strlen(ptr2)+1);
  61         }else {
  62             ptr = ptr2;
  63         }
  64     }
  65
  66     *ptr = 0;
  67     return ptr-buf;
  68 }
  69
  70 static nsIInputStream *get_post_data_stream(IBindCtx *bctx)
  71 {
  72     nsIInputStream *ret = NULL;
  73     IUnknown *unk;
  74     IBindStatusCallback *callback;
  75     IServiceProvider *service_provider;
  76     BINDINFO bindinfo;
  77     DWORD bindf = 0;
  78     DWORD post_len = 0, headers_len = 0;
  79     LPWSTR headers = NULL;
  80     WCHAR emptystr[] = {0};
  81     char *data;
  82     HRESULT hres;
  83
  84     static WCHAR _BSCB_Holder_[] =
  85         {'_','B','S','C','B','_','H','o','l','d','e','r','_',0};
  86
  87
  88     /* FIXME: This should be done in URLMoniker */
  89     if(!bctx)
  90         return NULL;
  91
  92     hres = IBindCtx_GetObjectParam(bctx, _BSCB_Holder_, &unk);
  93     if(FAILED(hres))
  94         return NULL;
  95
  96     hres = IUnknown_QueryInterface(unk, &IID_IBindStatusCallback, (void**)&callback);
  97     if(FAILED(hres)) {
  98         IUnknown_Release(unk);
  99         return NULL;
 100     }
 101
 102     hres = IUnknown_QueryInterface(unk, &IID_IServiceProvider, (void**)&service_provider);
 103     IUnknown_Release(unk);
 104     if(SUCCEEDED(hres)) {
 105         IHttpNegotiate *http_negotiate;
 106
 107         hres = IServiceProvider_QueryService(service_provider, &IID_IHttpNegotiate, &IID_IHttpNegotiate,
 108                                              (void**)&http_negotiate);
 109         if(SUCCEEDED(hres)) {
 110             hres = IHttpNegotiate_BeginningTransaction(http_negotiate, emptystr,
 111                                                        emptystr, 0, &headers);
 112             IHttpNegotiate_Release(http_negotiate);
 113
 114             if(SUCCEEDED(hres) && headers)
 115                 headers_len = WideCharToMultiByte(CP_ACP, 0, headers, -1, NULL, 0, NULL, NULL);
 116         }
 117
 118         IServiceProvider_Release(service_provider);
 119     }
 120
 121     memset(&bindinfo, 0, sizeof(bindinfo));
 122     bindinfo.cbSize = sizeof(bindinfo);
 123
 124     hres = IBindStatusCallback_GetBindInfo(callback, &bindf, &bindinfo);
 125
 126     if(SUCCEEDED(hres) && bindinfo.dwBindVerb == BINDVERB_POST)
 127         post_len = bindinfo.cbstgmedData;
 128
 129     if(headers_len || post_len) {
 130         int len = 0;
 131
 132         static const char content_length[] = "Content-Length: %u\r\n\r\n";
 133
 134         data = heap_alloc(headers_len+post_len+sizeof(content_length)+8);
 135
 136         if(headers_len) {
 137             WideCharToMultiByte(CP_ACP, 0, headers, -1, data, headers_len, NULL, NULL);
 138             len = fix_headers(data, post_len);
 139         }
 140
 141         if(post_len) {
 142             sprintf(data+len, content_length, post_len);
 143             len += strlen(data+len);
 144
 145             memcpy(data+len, bindinfo.stgmedData.u.hGlobal, post_len);
 146         }
 147
 148         TRACE("data = %s\n", debugstr_an(data, len+post_len));
 149
 150         if(len)
 151             ret = create_nsstream(data, len+post_len);
 152     }
 153
 154     CoTaskMemFree(headers);
 155     ReleaseBindInfo(&bindinfo);
 156     IBindStatusCallback_Release(callback);
 157
 158     return ret;
 159 }
 160
 161 static BOOL use_gecko_script(LPCWSTR url)
 162 {
 163     static const WCHAR fileW[] = {'f','i','l','e',':'};
 164     return strncmpiW(fileW, url, sizeof(fileW)/sizeof(WCHAR));
 165 }
 166
 167 void set_current_mon(HTMLDocument *This, IMoniker *mon)
 168 {
 169     HRESULT hres;
 170
 171     if(This->mon) {
 172         IMoniker_Release(This->mon);
 173         This->mon = NULL;
 174     }
 175
 176     if(This->url) {
 177         CoTaskMemFree(This->url);
 178         This->url = NULL;
 179     }
 180
 181     if(!mon)
 182         return;
 183
 184     IMoniker_AddRef(mon);
 185     This->mon = mon;
 186
 187     hres = IMoniker_GetDisplayName(mon, NULL, NULL, &This->url);
 188     if(FAILED(hres))
 189         WARN("GetDisplayName failed: %08x\n", hres);
 190
 191     set_script_mode(This, use_gecko_script(This->url) ? SCRIPTMODE_GECKO : SCRIPTMODE_ACTIVESCRIPT);
 192 }
 193
 194 static HRESULT set_moniker(HTMLDocument *This, IMoniker *mon, IBindCtx *pibc, BOOL *bind_complete)
 195 {
 196     nsChannelBSC *bscallback;
 197     LPOLESTR url = NULL;
 198     task_t *task;
 199     HRESULT hres;
 200     nsresult nsres;
 201
 202     if(pibc) {
 203         IUnknown *unk = NULL;
 204
 205         /* FIXME:
 206          * Use params:
 207          * "__PrecreatedObject"
 208          * "BIND_CONTEXT_PARAM"
 209          * "__HTMLLOADOPTIONS"
 210          * "__DWNBINDINFO"
 211          * "URL Context"
 212          * "CBinding Context"
 213          * "_ITransData_Object_"
 214          * "_EnumFORMATETC_"
 215          */
 216
 217         IBindCtx_GetObjectParam(pibc, (LPOLESTR)SZ_HTML_CLIENTSITE_OBJECTPARAM, &unk);
 218         if(unk) {
 219             IOleClientSite *client = NULL;
 220
 221             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
 222             if(SUCCEEDED(hres)) {
 223                 TRACE("Got client site %p\n", client);
 224                 IOleObject_SetClientSite(OLEOBJ(This), client);
 225                 IOleClientSite_Release(client);
 226             }
 227
 228             IUnknown_Release(unk);
 229         }
 230     }
 231
 232     This->readystate = READYSTATE_LOADING;
 233     call_property_onchanged(&This->cp_propnotif, DISPID_READYSTATE);
 234     update_doc(This, UPDATE_TITLE);
 235
 236     HTMLDocument_LockContainer(This, TRUE);
 237
 238     hres = IMoniker_GetDisplayName(mon, pibc, NULL, &url);
 239     if(FAILED(hres)) {
 240         WARN("GetDiaplayName failed: %08x\n", hres);
 241         return hres;
 242     }
 243
 244     TRACE("got url: %s\n", debugstr_w(url));
 245
 246     set_current_mon(This, mon);
 247
 248     if(This->client) {
 249         VARIANT silent, offline;
 250         IOleCommandTarget *cmdtrg = NULL;
 251
 252         hres = get_client_disp_property(This->client, DISPID_AMBIENT_SILENT, &silent);
 253         if(SUCCEEDED(hres)) {
 254             if(V_VT(&silent) != VT_BOOL)
 255                 WARN("V_VT(silent) = %d\n", V_VT(&silent));
 256             else if(V_BOOL(&silent))
 257                 FIXME("silent == true\n");
 258         }
 259
 260         hres = get_client_disp_property(This->client,
 261                 DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &offline);
 262         if(SUCCEEDED(hres)) {
 263             if(V_VT(&silent) != VT_BOOL)
 264                 WARN("V_VT(offline) = %d\n", V_VT(&silent));
 265             else if(V_BOOL(&silent))
 266                 FIXME("offline == true\n");
 267         }
 268
 269         hres = IOleClientSite_QueryInterface(This->client, &IID_IOleCommandTarget,
 270                 (void**)&cmdtrg);
 271         if(SUCCEEDED(hres)) {
 272             VARIANT var;
 273
 274             V_VT(&var) = VT_I4;
 275             V_I4(&var) = 0;
 276             IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
 277
 278             IOleCommandTarget_Release(cmdtrg);
 279         }
 280     }
 281
 282     bscallback = create_channelbsc(mon);
 283
 284     if(This->frame) {
 285         task = heap_alloc(sizeof(task_t));
 286
 287         task->doc = This;
 288         task->task_id = TASK_SETPROGRESS;
 289         task->next = NULL;
 290
 291         push_task(task);
 292     }
 293
 294     task = heap_alloc(sizeof(task_t));
 295
 296     task->doc = This;
 297     task->task_id = TASK_SETDOWNLOADSTATE;
 298     task->next = NULL;
 299
 300     push_task(task);
 301
 302     if(This->nscontainer) {
 303         nsIInputStream *post_data_stream = get_post_data_stream(pibc);
 304
 305         This->nscontainer->bscallback = bscallback;
 306         nsres = nsIWebNavigation_LoadURI(This->nscontainer->navigation, url,
 307                 LOAD_FLAGS_NONE, NULL, post_data_stream, NULL);
 308         This->nscontainer->bscallback = NULL;
 309
 310         if(post_data_stream)
 311             nsIInputStream_Release(post_data_stream);
 312
 313         if(NS_SUCCEEDED(nsres)) {
 314             /* FIXME: don't return here (URL Moniker needs to be good enough) */
 315
 316             IUnknown_Release((IUnknown*)bscallback);
 317             CoTaskMemFree(url);
 318
 319             if(bind_complete)
 320                 *bind_complete = TRUE;
 321             return S_OK;
 322         }else if(nsres != WINE_NS_LOAD_FROM_MONIKER) {
 323             WARN("LoadURI failed: %08x\n", nsres);
 324         }
 325     }
 326
 327     set_document_bscallback(This, bscallback);
 328     IUnknown_Release((IUnknown*)bscallback);
 329     CoTaskMemFree(url);
 330
 331     if(bind_complete)
 332         *bind_complete = FALSE;
 333     return S_OK;
 334 }
 335
 336 static HRESULT get_doc_string(HTMLDocument *This, char **str, DWORD *len)
 337 {
 338     nsIDOMDocument *nsdoc;
 339     nsIDOMNode *nsnode;
 340     LPCWSTR strw;
 341     nsAString nsstr;
 342     nsresult nsres;
 343
 344     if(!This->nscontainer) {
 345         WARN("no nscontainer, returning NULL\n");
 346         return S_OK;
 347     }
 348
 349     nsres = nsIWebNavigation_GetDocument(This->nscontainer->navigation, &nsdoc);
 350     if(NS_FAILED(nsres)) {
 351         ERR("GetDocument failed: %08x\n", nsres);
 352         return E_FAIL;
 353     }
 354
 355     nsres = nsIDOMDocument_QueryInterface(nsdoc, &IID_nsIDOMNode, (void**)&nsnode);
 356     nsIDOMDocument_Release(nsdoc);
 357     if(NS_FAILED(nsres)) {
 358         ERR("Could not get nsIDOMNode failed: %08x\n", nsres);
 359         return E_FAIL;
 360     }
 361
 362     nsAString_Init(&nsstr, NULL);
 363     nsnode_to_nsstring(nsnode, &nsstr);
 364     nsIDOMNode_Release(nsnode);
 365
 366     nsAString_GetData(&nsstr, &strw);
 367     TRACE("%s\n", debugstr_w(strw));
 368
 369     *len = WideCharToMultiByte(CP_ACP, 0, strw, -1, NULL, 0, NULL, NULL);
 370     *str = heap_alloc(*len);
 371     WideCharToMultiByte(CP_ACP, 0, strw, -1, *str, *len, NULL, NULL);
 372
 373     nsAString_Finish(&nsstr);
 374
 375     return S_OK;
 376 }
 377
 378
 379 /**********************************************************
 380  * IPersistMoniker implementation
 381  */
 382
 383 #define PERSISTMON_THIS(iface) DEFINE_THIS(HTMLDocument, PersistMoniker, iface)
 384
 385 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid,
 386                                                             void **ppvObject)
 387 {
 388     HTMLDocument *This = PERSISTMON_THIS(iface);
 389     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
 390 }
 391
 392 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
 393 {
 394     HTMLDocument *This = PERSISTMON_THIS(iface);
 395     return IHTMLDocument2_AddRef(HTMLDOC(This));
 396 }
 397
 398 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
 399 {
 400     HTMLDocument *This = PERSISTMON_THIS(iface);
 401     return IHTMLDocument2_Release(HTMLDOC(This));
 402 }
 403
 404 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
 405 {
 406     HTMLDocument *This = PERSISTMON_THIS(iface);
 407     return IPersist_GetClassID(PERSIST(This), pClassID);
 408 }
 409
 410 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
 411 {
 412     HTMLDocument *This = PERSISTMON_THIS(iface);
 413
 414     TRACE("(%p)\n", This);
 415
 416     return IPersistStreamInit_IsDirty(PERSTRINIT(This));
 417 }
 418
 419 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
 420         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
 421 {
 422     HTMLDocument *This = PERSISTMON_THIS(iface);
 423     BOOL bind_complete = FALSE;
 424     HRESULT hres;
 425
 426     TRACE("(%p)->(%x %p %p %08x)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
 427
 428     hres = set_moniker(This, pimkName, pibc, &bind_complete);
 429     if(FAILED(hres))
 430         return hres;
 431
 432     if(!bind_complete)
 433         return start_binding(This, (BSCallback*)This->bscallback, pibc);
 434
 435     return S_OK;
 436 }
 437
 438 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
 439         LPBC pbc, BOOL fRemember)
 440 {
 441     HTMLDocument *This = PERSISTMON_THIS(iface);
 442     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
 443     return E_NOTIMPL;
 444 }
 445
 446 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
 447 {
 448     HTMLDocument *This = PERSISTMON_THIS(iface);
 449     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
 450     return E_NOTIMPL;
 451 }
 452
 453 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
 454 {
 455     HTMLDocument *This = PERSISTMON_THIS(iface);
 456
 457     TRACE("(%p)->(%p)\n", This, ppimkName);
 458
 459     if(!This->mon)
 460         return E_UNEXPECTED;
 461
 462     IMoniker_AddRef(This->mon);
 463     *ppimkName = This->mon;
 464     return S_OK;
 465 }
 466
 467 static const IPersistMonikerVtbl PersistMonikerVtbl = {
 468     PersistMoniker_QueryInterface,
 469     PersistMoniker_AddRef,
 470     PersistMoniker_Release,
 471     PersistMoniker_GetClassID,
 472     PersistMoniker_IsDirty,
 473     PersistMoniker_Load,
 474     PersistMoniker_Save,
 475     PersistMoniker_SaveCompleted,
 476     PersistMoniker_GetCurMoniker
 477 };
 478
 479 /**********************************************************
 480  * IMonikerProp implementation
 481  */
 482
 483 #define MONPROP_THIS(iface) DEFINE_THIS(HTMLDocument, MonikerProp, iface)
 484
 485 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppvObject)
 486 {
 487     HTMLDocument *This = MONPROP_THIS(iface);
 488     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
 489 }
 490
 491 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
 492 {
 493     HTMLDocument *This = MONPROP_THIS(iface);
 494     return IHTMLDocument2_AddRef(HTMLDOC(This));
 495 }
 496
 497 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
 498 {
 499     HTMLDocument *This = MONPROP_THIS(iface);
 500     return IHTMLDocument_Release(HTMLDOC(This));
 501 }
 502
 503 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
 504 {
 505     HTMLDocument *This = MONPROP_THIS(iface);
 506
 507     TRACE("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
 508
 509     switch(mkp) {
 510     case MIMETYPEPROP:
 511         heap_free(This->mime);
 512         This->mime = heap_strdupW(val);
 513         break;
 514
 515     case CLASSIDPROP:
 516         break;
 517
 518     default:
 519         FIXME("mkp %d\n", mkp);
 520         return E_NOTIMPL;
 521     }
 522
 523     return S_OK;
 524 }
 525
 526 static const IMonikerPropVtbl MonikerPropVtbl = {
 527     MonikerProp_QueryInterface,
 528     MonikerProp_AddRef,
 529     MonikerProp_Release,
 530     MonikerProp_PutProperty
 531 };
 532
 533 /**********************************************************
 534  * IPersistFile implementation
 535  */
 536
 537 #define PERSISTFILE_THIS(iface) DEFINE_THIS(HTMLDocument, PersistFile, iface)
 538
 539 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppvObject)
 540 {
 541     HTMLDocument *This = PERSISTFILE_THIS(iface);
 542     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
 543 }
 544
 545 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
 546 {
 547     HTMLDocument *This = PERSISTFILE_THIS(iface);
 548     return IHTMLDocument2_AddRef(HTMLDOC(This));
 549 }
 550
 551 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
 552 {
 553     HTMLDocument *This = PERSISTFILE_THIS(iface);
 554     return IHTMLDocument2_Release(HTMLDOC(This));
 555 }
 556
 557 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
 558 {
 559     HTMLDocument *This = PERSISTFILE_THIS(iface);
 560
 561     TRACE("(%p)->(%p)\n", This, pClassID);
 562
 563     if(!pClassID)
 564         return E_INVALIDARG;
 565
 566     *pClassID = CLSID_HTMLDocument;
 567     return S_OK;
 568 }
 569
 570 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
 571 {
 572     HTMLDocument *This = PERSISTFILE_THIS(iface);
 573
 574     TRACE("(%p)\n", This);
 575
 576     return IPersistStreamInit_IsDirty(PERSTRINIT(This));
 577 }
 578
 579 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
 580 {
 581     HTMLDocument *This = PERSISTFILE_THIS(iface);
 582     FIXME("(%p)->(%s %08x)\n", This, debugstr_w(pszFileName), dwMode);
 583     return E_NOTIMPL;
 584 }
 585
 586 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
 587 {
 588     HTMLDocument *This = PERSISTFILE_THIS(iface);
 589     char *str;
 590     DWORD len, written=0;
 591     HANDLE file;
 592     HRESULT hres;
 593
 594     TRACE("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
 595
 596     file = CreateFileW(pszFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
 597                        FILE_ATTRIBUTE_NORMAL, NULL);
 598     if(file == INVALID_HANDLE_VALUE) {
 599         WARN("Could not create file: %u\n", GetLastError());
 600         return E_FAIL;
 601     }
 602
 603     hres = get_doc_string(This, &str, &len);
 604     if(FAILED(hres))
 605         return hres;
 606
 607     WriteFile(file, str, len, &written, NULL);
 608     CloseHandle(file);
 609     return S_OK;
 610 }
 611
 612 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
 613 {
 614     HTMLDocument *This = PERSISTFILE_THIS(iface);
 615     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
 616     return E_NOTIMPL;
 617 }
 618
 619 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
 620 {
 621     HTMLDocument *This = PERSISTFILE_THIS(iface);
 622     FIXME("(%p)->(%p)\n", This, pszFileName);
 623     return E_NOTIMPL;
 624 }
 625
 626 static const IPersistFileVtbl PersistFileVtbl = {
 627     PersistFile_QueryInterface,
 628     PersistFile_AddRef,
 629     PersistFile_Release,
 630     PersistFile_GetClassID,
 631     PersistFile_IsDirty,
 632     PersistFile_Load,
 633     PersistFile_Save,
 634     PersistFile_SaveCompleted,
 635     PersistFile_GetCurFile
 636 };
 637
 638 #define PERSTRINIT_THIS(iface) DEFINE_THIS(HTMLDocument, PersistStreamInit, iface)
 639
 640 static HRESULT WINAPI PersistStreamInit_QueryInterface(IPersistStreamInit *iface,
 641                                                        REFIID riid, void **ppv)
 642 {
 643     HTMLDocument *This = PERSTRINIT_THIS(iface);
 644     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppv);
 645 }
 646
 647 static ULONG WINAPI PersistStreamInit_AddRef(IPersistStreamInit *iface)
 648 {
 649     HTMLDocument *This = PERSTRINIT_THIS(iface);
 650     return IHTMLDocument2_AddRef(HTMLDOC(This));
 651 }
 652
 653 static ULONG WINAPI PersistStreamInit_Release(IPersistStreamInit *iface)
 654 {
 655     HTMLDocument *This = PERSTRINIT_THIS(iface);
 656     return IHTMLDocument2_Release(HTMLDOC(This));
 657 }
 658
 659 static HRESULT WINAPI PersistStreamInit_GetClassID(IPersistStreamInit *iface, CLSID *pClassID)
 660 {
 661     HTMLDocument *This = PERSTRINIT_THIS(iface);
 662     return IPersist_GetClassID(PERSIST(This), pClassID);
 663 }
 664
 665 static HRESULT WINAPI PersistStreamInit_IsDirty(IPersistStreamInit *iface)
 666 {
 667     HTMLDocument *This = PERSTRINIT_THIS(iface);
 668
 669     TRACE("(%p)\n", This);
 670
 671     if(This->usermode == EDITMODE)
 672         return editor_is_dirty(This);
 673
 674     return S_FALSE;
 675 }
 676
 677 static HRESULT WINAPI PersistStreamInit_Load(IPersistStreamInit *iface, LPSTREAM pStm)
 678 {
 679     HTMLDocument *This = PERSTRINIT_THIS(iface);
 680     IMoniker *mon;
 681     HRESULT hres;
 682
 683     static const WCHAR about_blankW[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
 684
 685     TRACE("(%p)->(%p)\n", This, pStm);
 686
 687     hres = CreateURLMoniker(NULL, about_blankW, &mon);
 688     if(FAILED(hres)) {
 689         WARN("CreateURLMoniker failed: %08x\n", hres);
 690         return hres;
 691     }
 692
 693     hres = set_moniker(This, mon, NULL, NULL);
 694     IMoniker_Release(mon);
 695     if(FAILED(hres))
 696         return hres;
 697
 698     return channelbsc_load_stream(This->bscallback, pStm);
 699 }
 700
 701 static HRESULT WINAPI PersistStreamInit_Save(IPersistStreamInit *iface, LPSTREAM pStm,
 702                                              BOOL fClearDirty)
 703 {
 704     HTMLDocument *This = PERSTRINIT_THIS(iface);
 705     char *str;
 706     DWORD len, written=0;
 707     HRESULT hres;
 708
 709     TRACE("(%p)->(%p %x)\n", This, pStm, fClearDirty);
 710
 711     hres = get_doc_string(This, &str, &len);
 712     if(FAILED(hres))
 713         return hres;
 714
 715     hres = IStream_Write(pStm, str, len, &written);
 716     if(FAILED(hres))
 717         FIXME("Write failed: %08x\n", hres);
 718
 719     heap_free(str);
 720
 721     if(fClearDirty)
 722         set_dirty(This, VARIANT_FALSE);
 723
 724     return S_OK;
 725 }
 726
 727 static HRESULT WINAPI PersistStreamInit_GetSizeMax(IPersistStreamInit *iface,
 728                                                    ULARGE_INTEGER *pcbSize)
 729 {
 730     HTMLDocument *This = PERSTRINIT_THIS(iface);
 731     FIXME("(%p)->(%p)\n", This, pcbSize);
 732     return E_NOTIMPL;
 733 }
 734
 735 static HRESULT WINAPI PersistStreamInit_InitNew(IPersistStreamInit *iface)
 736 {
 737     HTMLDocument *This = PERSTRINIT_THIS(iface);
 738     FIXME("(%p)\n", This);
 739     return E_NOTIMPL;
 740 }
 741
 742 #undef PERSTRINIT_THIS
 743
 744 static const IPersistStreamInitVtbl PersistStreamInitVtbl = {
 745     PersistStreamInit_QueryInterface,
 746     PersistStreamInit_AddRef,
 747     PersistStreamInit_Release,
 748     PersistStreamInit_GetClassID,
 749     PersistStreamInit_IsDirty,
 750     PersistStreamInit_Load,
 751     PersistStreamInit_Save,
 752     PersistStreamInit_GetSizeMax,
 753     PersistStreamInit_InitNew
 754 };
 755
 756 void HTMLDocument_Persist_Init(HTMLDocument *This)
 757 {
 758     This->lpPersistMonikerVtbl = &PersistMonikerVtbl;
 759     This->lpPersistFileVtbl = &PersistFileVtbl;
 760     This->lpMonikerPropVtbl = &MonikerPropVtbl;
 761     This->lpPersistStreamInitVtbl = &PersistStreamInitVtbl;
 762
 763     This->bscallback = NULL;
 764     This->mon = NULL;
 765     This->url = NULL;
 766     This->mime = NULL;
 767 }