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