mshtml: Write aCount bytes in nsIInputStream::ReadSegments.
[wine] / 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     nsIDOMNode *nsnode;
339     LPCWSTR strw;
340     nsAString nsstr;
341     nsresult nsres;
342
343     if(!This->nsdoc) {
344         WARN("NULL nsdoc\n");
345         return E_UNEXPECTED;
346     }
347
348     nsres = nsIDOMHTMLDocument_QueryInterface(This->nsdoc, &IID_nsIDOMNode, (void**)&nsnode);
349     if(NS_FAILED(nsres)) {
350         ERR("Could not get nsIDOMNode failed: %08x\n", nsres);
351         return E_FAIL;
352     }
353
354     nsAString_Init(&nsstr, NULL);
355     nsnode_to_nsstring(nsnode, &nsstr);
356     nsIDOMNode_Release(nsnode);
357
358     nsAString_GetData(&nsstr, &strw);
359     TRACE("%s\n", debugstr_w(strw));
360
361     *len = WideCharToMultiByte(CP_ACP, 0, strw, -1, NULL, 0, NULL, NULL);
362     *str = heap_alloc(*len);
363     WideCharToMultiByte(CP_ACP, 0, strw, -1, *str, *len, NULL, NULL);
364
365     nsAString_Finish(&nsstr);
366
367     return S_OK;
368 }
369
370
371 /**********************************************************
372  * IPersistMoniker implementation
373  */
374
375 #define PERSISTMON_THIS(iface) DEFINE_THIS(HTMLDocument, PersistMoniker, iface)
376
377 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid,
378                                                             void **ppvObject)
379 {
380     HTMLDocument *This = PERSISTMON_THIS(iface);
381     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
382 }
383
384 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
385 {
386     HTMLDocument *This = PERSISTMON_THIS(iface);
387     return IHTMLDocument2_AddRef(HTMLDOC(This));
388 }
389
390 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
391 {
392     HTMLDocument *This = PERSISTMON_THIS(iface);
393     return IHTMLDocument2_Release(HTMLDOC(This));
394 }
395
396 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
397 {
398     HTMLDocument *This = PERSISTMON_THIS(iface);
399     return IPersist_GetClassID(PERSIST(This), pClassID);
400 }
401
402 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
403 {
404     HTMLDocument *This = PERSISTMON_THIS(iface);
405
406     TRACE("(%p)\n", This);
407
408     return IPersistStreamInit_IsDirty(PERSTRINIT(This));
409 }
410
411 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
412         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
413 {
414     HTMLDocument *This = PERSISTMON_THIS(iface);
415     BOOL bind_complete = FALSE;
416     HRESULT hres;
417
418     TRACE("(%p)->(%x %p %p %08x)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
419
420     hres = set_moniker(This, pimkName, pibc, &bind_complete);
421     if(FAILED(hres))
422         return hres;
423
424     if(!bind_complete)
425         return start_binding(This, (BSCallback*)This->bscallback, pibc);
426
427     return S_OK;
428 }
429
430 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
431         LPBC pbc, BOOL fRemember)
432 {
433     HTMLDocument *This = PERSISTMON_THIS(iface);
434     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
435     return E_NOTIMPL;
436 }
437
438 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
439 {
440     HTMLDocument *This = PERSISTMON_THIS(iface);
441     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
442     return E_NOTIMPL;
443 }
444
445 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
446 {
447     HTMLDocument *This = PERSISTMON_THIS(iface);
448
449     TRACE("(%p)->(%p)\n", This, ppimkName);
450
451     if(!This->mon)
452         return E_UNEXPECTED;
453
454     IMoniker_AddRef(This->mon);
455     *ppimkName = This->mon;
456     return S_OK;
457 }
458
459 static const IPersistMonikerVtbl PersistMonikerVtbl = {
460     PersistMoniker_QueryInterface,
461     PersistMoniker_AddRef,
462     PersistMoniker_Release,
463     PersistMoniker_GetClassID,
464     PersistMoniker_IsDirty,
465     PersistMoniker_Load,
466     PersistMoniker_Save,
467     PersistMoniker_SaveCompleted,
468     PersistMoniker_GetCurMoniker
469 };
470
471 /**********************************************************
472  * IMonikerProp implementation
473  */
474
475 #define MONPROP_THIS(iface) DEFINE_THIS(HTMLDocument, MonikerProp, iface)
476
477 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppvObject)
478 {
479     HTMLDocument *This = MONPROP_THIS(iface);
480     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
481 }
482
483 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
484 {
485     HTMLDocument *This = MONPROP_THIS(iface);
486     return IHTMLDocument2_AddRef(HTMLDOC(This));
487 }
488
489 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
490 {
491     HTMLDocument *This = MONPROP_THIS(iface);
492     return IHTMLDocument_Release(HTMLDOC(This));
493 }
494
495 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
496 {
497     HTMLDocument *This = MONPROP_THIS(iface);
498
499     TRACE("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
500
501     switch(mkp) {
502     case MIMETYPEPROP:
503         heap_free(This->mime);
504         This->mime = heap_strdupW(val);
505         break;
506
507     case CLASSIDPROP:
508         break;
509
510     default:
511         FIXME("mkp %d\n", mkp);
512         return E_NOTIMPL;
513     }
514
515     return S_OK;
516 }
517
518 static const IMonikerPropVtbl MonikerPropVtbl = {
519     MonikerProp_QueryInterface,
520     MonikerProp_AddRef,
521     MonikerProp_Release,
522     MonikerProp_PutProperty
523 };
524
525 /**********************************************************
526  * IPersistFile implementation
527  */
528
529 #define PERSISTFILE_THIS(iface) DEFINE_THIS(HTMLDocument, PersistFile, iface)
530
531 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppvObject)
532 {
533     HTMLDocument *This = PERSISTFILE_THIS(iface);
534     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
535 }
536
537 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
538 {
539     HTMLDocument *This = PERSISTFILE_THIS(iface);
540     return IHTMLDocument2_AddRef(HTMLDOC(This));
541 }
542
543 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
544 {
545     HTMLDocument *This = PERSISTFILE_THIS(iface);
546     return IHTMLDocument2_Release(HTMLDOC(This));
547 }
548
549 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
550 {
551     HTMLDocument *This = PERSISTFILE_THIS(iface);
552
553     TRACE("(%p)->(%p)\n", This, pClassID);
554
555     if(!pClassID)
556         return E_INVALIDARG;
557
558     *pClassID = CLSID_HTMLDocument;
559     return S_OK;
560 }
561
562 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
563 {
564     HTMLDocument *This = PERSISTFILE_THIS(iface);
565
566     TRACE("(%p)\n", This);
567
568     return IPersistStreamInit_IsDirty(PERSTRINIT(This));
569 }
570
571 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
572 {
573     HTMLDocument *This = PERSISTFILE_THIS(iface);
574     FIXME("(%p)->(%s %08x)\n", This, debugstr_w(pszFileName), dwMode);
575     return E_NOTIMPL;
576 }
577
578 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
579 {
580     HTMLDocument *This = PERSISTFILE_THIS(iface);
581     char *str;
582     DWORD len, written=0;
583     HANDLE file;
584     HRESULT hres;
585
586     TRACE("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
587
588     file = CreateFileW(pszFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
589                        FILE_ATTRIBUTE_NORMAL, NULL);
590     if(file == INVALID_HANDLE_VALUE) {
591         WARN("Could not create file: %u\n", GetLastError());
592         return E_FAIL;
593     }
594
595     hres = get_doc_string(This, &str, &len);
596     if(SUCCEEDED(hres))
597         WriteFile(file, str, len, &written, NULL);
598
599     CloseHandle(file);
600     return hres;
601 }
602
603 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
604 {
605     HTMLDocument *This = PERSISTFILE_THIS(iface);
606     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
607     return E_NOTIMPL;
608 }
609
610 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
611 {
612     HTMLDocument *This = PERSISTFILE_THIS(iface);
613     FIXME("(%p)->(%p)\n", This, pszFileName);
614     return E_NOTIMPL;
615 }
616
617 static const IPersistFileVtbl PersistFileVtbl = {
618     PersistFile_QueryInterface,
619     PersistFile_AddRef,
620     PersistFile_Release,
621     PersistFile_GetClassID,
622     PersistFile_IsDirty,
623     PersistFile_Load,
624     PersistFile_Save,
625     PersistFile_SaveCompleted,
626     PersistFile_GetCurFile
627 };
628
629 #define PERSTRINIT_THIS(iface) DEFINE_THIS(HTMLDocument, PersistStreamInit, iface)
630
631 static HRESULT WINAPI PersistStreamInit_QueryInterface(IPersistStreamInit *iface,
632                                                        REFIID riid, void **ppv)
633 {
634     HTMLDocument *This = PERSTRINIT_THIS(iface);
635     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppv);
636 }
637
638 static ULONG WINAPI PersistStreamInit_AddRef(IPersistStreamInit *iface)
639 {
640     HTMLDocument *This = PERSTRINIT_THIS(iface);
641     return IHTMLDocument2_AddRef(HTMLDOC(This));
642 }
643
644 static ULONG WINAPI PersistStreamInit_Release(IPersistStreamInit *iface)
645 {
646     HTMLDocument *This = PERSTRINIT_THIS(iface);
647     return IHTMLDocument2_Release(HTMLDOC(This));
648 }
649
650 static HRESULT WINAPI PersistStreamInit_GetClassID(IPersistStreamInit *iface, CLSID *pClassID)
651 {
652     HTMLDocument *This = PERSTRINIT_THIS(iface);
653     return IPersist_GetClassID(PERSIST(This), pClassID);
654 }
655
656 static HRESULT WINAPI PersistStreamInit_IsDirty(IPersistStreamInit *iface)
657 {
658     HTMLDocument *This = PERSTRINIT_THIS(iface);
659
660     TRACE("(%p)\n", This);
661
662     if(This->usermode == EDITMODE)
663         return editor_is_dirty(This);
664
665     return S_FALSE;
666 }
667
668 static HRESULT WINAPI PersistStreamInit_Load(IPersistStreamInit *iface, LPSTREAM pStm)
669 {
670     HTMLDocument *This = PERSTRINIT_THIS(iface);
671     IMoniker *mon;
672     HRESULT hres;
673
674     static const WCHAR about_blankW[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
675
676     TRACE("(%p)->(%p)\n", This, pStm);
677
678     hres = CreateURLMoniker(NULL, about_blankW, &mon);
679     if(FAILED(hres)) {
680         WARN("CreateURLMoniker failed: %08x\n", hres);
681         return hres;
682     }
683
684     hres = set_moniker(This, mon, NULL, NULL);
685     IMoniker_Release(mon);
686     if(FAILED(hres))
687         return hres;
688
689     return channelbsc_load_stream(This->bscallback, pStm);
690 }
691
692 static HRESULT WINAPI PersistStreamInit_Save(IPersistStreamInit *iface, LPSTREAM pStm,
693                                              BOOL fClearDirty)
694 {
695     HTMLDocument *This = PERSTRINIT_THIS(iface);
696     char *str;
697     DWORD len, written=0;
698     HRESULT hres;
699
700     TRACE("(%p)->(%p %x)\n", This, pStm, fClearDirty);
701
702     hres = get_doc_string(This, &str, &len);
703     if(FAILED(hres))
704         return hres;
705
706     hres = IStream_Write(pStm, str, len, &written);
707     if(FAILED(hres))
708         FIXME("Write failed: %08x\n", hres);
709
710     heap_free(str);
711
712     if(fClearDirty)
713         set_dirty(This, VARIANT_FALSE);
714
715     return S_OK;
716 }
717
718 static HRESULT WINAPI PersistStreamInit_GetSizeMax(IPersistStreamInit *iface,
719                                                    ULARGE_INTEGER *pcbSize)
720 {
721     HTMLDocument *This = PERSTRINIT_THIS(iface);
722     FIXME("(%p)->(%p)\n", This, pcbSize);
723     return E_NOTIMPL;
724 }
725
726 static HRESULT WINAPI PersistStreamInit_InitNew(IPersistStreamInit *iface)
727 {
728     HTMLDocument *This = PERSTRINIT_THIS(iface);
729     FIXME("(%p)\n", This);
730     return E_NOTIMPL;
731 }
732
733 #undef PERSTRINIT_THIS
734
735 static const IPersistStreamInitVtbl PersistStreamInitVtbl = {
736     PersistStreamInit_QueryInterface,
737     PersistStreamInit_AddRef,
738     PersistStreamInit_Release,
739     PersistStreamInit_GetClassID,
740     PersistStreamInit_IsDirty,
741     PersistStreamInit_Load,
742     PersistStreamInit_Save,
743     PersistStreamInit_GetSizeMax,
744     PersistStreamInit_InitNew
745 };
746
747 void HTMLDocument_Persist_Init(HTMLDocument *This)
748 {
749     This->lpPersistMonikerVtbl = &PersistMonikerVtbl;
750     This->lpPersistFileVtbl = &PersistFileVtbl;
751     This->lpMonikerPropVtbl = &MonikerPropVtbl;
752     This->lpPersistStreamInitVtbl = &PersistStreamInitVtbl;
753
754     This->bscallback = NULL;
755     This->mon = NULL;
756     This->url = NULL;
757     This->mime = NULL;
758 }