mshtml: Split IPersistMoniker::Load.
[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 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 static HRESULT set_moniker(HTMLDocument *This, IMoniker *mon, IBindCtx *pibc)
117 {
118     BSCallback *bscallback;
119     LPOLESTR url = NULL;
120     task_t *task;
121     HRESULT hres;
122     nsresult nsres;
123
124     if(pibc) {
125         IUnknown *unk = NULL;
126
127         /* FIXME:
128          * Use params:
129          * "__PrecreatedObject"
130          * "BIND_CONTEXT_PARAM"
131          * "__HTMLLOADOPTIONS"
132          * "__DWNBINDINFO"
133          * "URL Context"
134          * "CBinding Context"
135          * "_ITransData_Object_"
136          * "_EnumFORMATETC_"
137          */
138
139         IBindCtx_GetObjectParam(pibc, (LPOLESTR)SZ_HTML_CLIENTSITE_OBJECTPARAM, &unk);
140         if(unk) {
141             IOleClientSite *client = NULL;
142
143             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
144             if(SUCCEEDED(hres)) {
145                 TRACE("Got client site %p\n", client);
146                 IOleObject_SetClientSite(OLEOBJ(This), client);
147                 IOleClientSite_Release(client);
148             }
149
150             IUnknown_Release(unk);
151         }
152     }
153
154     This->readystate = READYSTATE_LOADING;
155     call_property_onchanged(This->cp_propnotif, DISPID_READYSTATE);
156
157     HTMLDocument_LockContainer(This, TRUE);
158     
159     hres = IMoniker_GetDisplayName(mon, pibc, NULL, &url);
160     if(FAILED(hres)) {
161         WARN("GetDiaplayName failed: %08x\n", hres);
162         return hres;
163     }
164
165     TRACE("got url: %s\n", debugstr_w(url));
166
167     if(This->client) {
168         IOleCommandTarget *cmdtrg = NULL;
169
170         hres = IOleClientSite_QueryInterface(This->client, &IID_IOleCommandTarget,
171                 (void**)&cmdtrg);
172         if(SUCCEEDED(hres)) {
173             VARIANT var;
174
175             V_VT(&var) = VT_I4;
176             V_I4(&var) = 0;
177             IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
178
179             IOleCommandTarget_Release(cmdtrg);
180         }
181     }
182
183     if(This->client) {
184         VARIANT silent, offline;
185
186         hres = get_client_disp_property(This->client, DISPID_AMBIENT_SILENT, &silent);
187         if(SUCCEEDED(hres)) {
188             if(V_VT(&silent) != VT_BOOL)
189                 WARN("V_VT(silent) = %d\n", V_VT(&silent));
190             else if(V_BOOL(&silent))
191                 FIXME("silent == true\n");
192         }
193
194         hres = get_client_disp_property(This->client,
195                 DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &offline);
196         if(SUCCEEDED(hres)) {
197             if(V_VT(&silent) != VT_BOOL)
198                 WARN("V_VT(offline) = %d\n", V_VT(&silent));
199             else if(V_BOOL(&silent))
200                 FIXME("offline == true\n");
201         }
202     }
203
204     bscallback = create_bscallback(mon);
205
206     if(This->frame) {
207         task = mshtml_alloc(sizeof(task_t));
208
209         task->doc = This;
210         task->task_id = TASK_SETPROGRESS;
211         task->next = NULL;
212
213         push_task(task);
214     }
215
216     task = mshtml_alloc(sizeof(task_t));
217
218     task->doc = This;
219     task->task_id = TASK_SETDOWNLOADSTATE;
220     task->next = NULL;
221
222     push_task(task);
223
224     if(This->nscontainer) {
225         nsIInputStream *post_data_stream = get_post_data_stream(pibc);
226
227         This->nscontainer->bscallback = bscallback;
228         nsres = nsIWebNavigation_LoadURI(This->nscontainer->navigation, url,
229                 LOAD_FLAGS_NONE, NULL, post_data_stream, NULL);
230         This->nscontainer->bscallback = NULL;
231
232         if(post_data_stream)
233             nsIInputStream_Release(post_data_stream);
234
235         if(NS_SUCCEEDED(nsres)) {
236             /* FIXME: don't return here (URL Moniker needs to be good enough) */
237
238             IBindStatusCallback_Release(STATUSCLB(bscallback));
239             CoTaskMemFree(url);
240             return S_OK;
241         }else if(nsres != WINE_NS_LOAD_FROM_MONIKER) {
242             WARN("LoadURI failed: %08x\n", nsres);
243         }
244     }
245
246     set_document_bscallback(This, bscallback);
247     IBindStatusCallback_Release(STATUSCLB(bscallback));
248     CoTaskMemFree(url);
249
250     return S_OK;
251 }
252
253 static HRESULT get_doc_string(HTMLDocument *This, char **str, DWORD *len)
254 {
255     nsIDOMDocument *nsdoc;
256     nsIDOMNode *nsnode;
257     LPCWSTR strw;
258     nsAString nsstr;
259     nsresult nsres;
260
261     if(!This->nscontainer) {
262         WARN("no nscontainer, returning NULL\n");
263         return S_OK;
264     }
265
266     nsres = nsIWebNavigation_GetDocument(This->nscontainer->navigation, &nsdoc);
267     if(NS_FAILED(nsres)) {
268         ERR("GetDocument failed: %08x\n", nsres);
269         return E_FAIL;
270     }
271
272     nsres = nsIDOMDocument_QueryInterface(nsdoc, &IID_nsIDOMNode, (void**)&nsnode);
273     nsIDOMDocument_Release(nsdoc);
274     if(NS_FAILED(nsres)) {
275         ERR("Could not get nsIDOMNode failed: %08x\n", nsres);
276         return E_FAIL;
277     }
278
279     nsAString_Init(&nsstr, NULL);
280     nsnode_to_nsstring(nsnode, &nsstr);
281     nsIDOMNode_Release(nsnode);
282
283     nsAString_GetData(&nsstr, &strw, NULL);
284     TRACE("%s\n", debugstr_w(strw));
285
286     *len = WideCharToMultiByte(CP_ACP, 0, strw, -1, NULL, 0, NULL, NULL);
287     *str = mshtml_alloc(*len);
288     WideCharToMultiByte(CP_ACP, 0, strw, -1, *str, *len, NULL, NULL);
289
290     nsAString_Finish(&nsstr);
291
292     return S_OK;
293 }
294
295
296 /**********************************************************
297  * IPersistMoniker implementation
298  */
299
300 #define PERSISTMON_THIS(iface) DEFINE_THIS(HTMLDocument, PersistMoniker, iface)
301
302 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid,
303                                                             void **ppvObject)
304 {
305     HTMLDocument *This = PERSISTMON_THIS(iface);
306     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
307 }
308
309 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
310 {
311     HTMLDocument *This = PERSISTMON_THIS(iface);
312     return IHTMLDocument2_AddRef(HTMLDOC(This));
313 }
314
315 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
316 {
317     HTMLDocument *This = PERSISTMON_THIS(iface);
318     return IHTMLDocument2_Release(HTMLDOC(This));
319 }
320
321 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
322 {
323     HTMLDocument *This = PERSISTMON_THIS(iface);
324     return IPersist_GetClassID(PERSIST(This), pClassID);
325 }
326
327 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
328 {
329     HTMLDocument *This = PERSISTMON_THIS(iface);
330     FIXME("(%p)\n", This);
331     return E_NOTIMPL;
332 }
333
334 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
335         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
336 {
337     HTMLDocument *This = PERSISTMON_THIS(iface);
338     HRESULT hres;
339
340     TRACE("(%p)->(%x %p %p %08x)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
341
342     hres = set_moniker(This, pimkName, pibc);
343     if(FAILED(hres))
344         return hres;
345
346     return start_binding(This->bscallback);
347 }
348
349 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
350         LPBC pbc, BOOL fRemember)
351 {
352     HTMLDocument *This = PERSISTMON_THIS(iface);
353     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
354     return E_NOTIMPL;
355 }
356
357 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
358 {
359     HTMLDocument *This = PERSISTMON_THIS(iface);
360     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
361     return E_NOTIMPL;
362 }
363
364 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
365 {
366     HTMLDocument *This = PERSISTMON_THIS(iface);
367     FIXME("(%p)->(%p)\n", This, ppimkName);
368     return E_NOTIMPL;
369 }
370
371 static const IPersistMonikerVtbl PersistMonikerVtbl = {
372     PersistMoniker_QueryInterface,
373     PersistMoniker_AddRef,
374     PersistMoniker_Release,
375     PersistMoniker_GetClassID,
376     PersistMoniker_IsDirty,
377     PersistMoniker_Load,
378     PersistMoniker_Save,
379     PersistMoniker_SaveCompleted,
380     PersistMoniker_GetCurMoniker
381 };
382
383 /**********************************************************
384  * IMonikerProp implementation
385  */
386
387 #define MONPROP_THIS(iface) DEFINE_THIS(HTMLDocument, MonikerProp, iface)
388
389 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppvObject)
390 {
391     HTMLDocument *This = MONPROP_THIS(iface);
392     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
393 }
394
395 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
396 {
397     HTMLDocument *This = MONPROP_THIS(iface);
398     return IHTMLDocument2_AddRef(HTMLDOC(This));
399 }
400
401 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
402 {
403     HTMLDocument *This = MONPROP_THIS(iface);
404     return IHTMLDocument_Release(HTMLDOC(This));
405 }
406
407 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
408 {
409     HTMLDocument *This = MONPROP_THIS(iface);
410     FIXME("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
411     return E_NOTIMPL;
412 }
413
414 static const IMonikerPropVtbl MonikerPropVtbl = {
415     MonikerProp_QueryInterface,
416     MonikerProp_AddRef,
417     MonikerProp_Release,
418     MonikerProp_PutProperty
419 };
420
421 /**********************************************************
422  * IPersistFile implementation
423  */
424
425 #define PERSISTFILE_THIS(iface) DEFINE_THIS(HTMLDocument, PersistFile, iface)
426
427 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppvObject)
428 {
429     HTMLDocument *This = PERSISTFILE_THIS(iface);
430     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
431 }
432
433 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
434 {
435     HTMLDocument *This = PERSISTFILE_THIS(iface);
436     return IHTMLDocument2_AddRef(HTMLDOC(This));
437 }
438
439 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
440 {
441     HTMLDocument *This = PERSISTFILE_THIS(iface);
442     return IHTMLDocument2_Release(HTMLDOC(This));
443 }
444
445 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
446 {
447     HTMLDocument *This = PERSISTFILE_THIS(iface);
448
449     TRACE("(%p)->(%p)\n", This, pClassID);
450
451     if(!pClassID)
452         return E_INVALIDARG;
453
454     memcpy(pClassID, &CLSID_HTMLDocument, sizeof(CLSID));
455     return S_OK;
456 }
457
458 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
459 {
460     HTMLDocument *This = PERSISTFILE_THIS(iface);
461     FIXME("(%p)\n", This);
462     return E_NOTIMPL;
463 }
464
465 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
466 {
467     HTMLDocument *This = PERSISTFILE_THIS(iface);
468     FIXME("(%p)->(%s %08x)\n", This, debugstr_w(pszFileName), dwMode);
469     return E_NOTIMPL;
470 }
471
472 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
473 {
474     HTMLDocument *This = PERSISTFILE_THIS(iface);
475     char *str;
476     DWORD len, written=0;
477     HANDLE file;
478     HRESULT hres;
479
480     TRACE("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
481
482     file = CreateFileW(pszFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
483                        FILE_ATTRIBUTE_NORMAL, NULL);
484     if(file == INVALID_HANDLE_VALUE) {
485         WARN("Could not create file: %u\n", GetLastError());
486         return E_FAIL;
487     }
488
489     hres = get_doc_string(This, &str, &len);
490     if(FAILED(hres))
491         return hres;
492
493     WriteFile(file, str, len, &written, NULL);
494     CloseHandle(file);
495     return S_OK;
496 }
497
498 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
499 {
500     HTMLDocument *This = PERSISTFILE_THIS(iface);
501     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
502     return E_NOTIMPL;
503 }
504
505 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
506 {
507     HTMLDocument *This = PERSISTFILE_THIS(iface);
508     FIXME("(%p)->(%p)\n", This, pszFileName);
509     return E_NOTIMPL;
510 }
511
512 static const IPersistFileVtbl PersistFileVtbl = {
513     PersistFile_QueryInterface,
514     PersistFile_AddRef,
515     PersistFile_Release,
516     PersistFile_GetClassID,
517     PersistFile_IsDirty,
518     PersistFile_Load,
519     PersistFile_Save,
520     PersistFile_SaveCompleted,
521     PersistFile_GetCurFile
522 };
523
524 #define PERSTRINIT_THIS(iface) DEFINE_THIS(HTMLDocument, PersistStreamInit, iface)
525
526 static HRESULT WINAPI PersistStreamInit_QueryInterface(IPersistStreamInit *iface,
527                                                        REFIID riid, void **ppv)
528 {
529     HTMLDocument *This = PERSTRINIT_THIS(iface);
530     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppv);
531 }
532
533 static ULONG WINAPI PersistStreamInit_AddRef(IPersistStreamInit *iface)
534 {
535     HTMLDocument *This = PERSTRINIT_THIS(iface);
536     return IHTMLDocument2_AddRef(HTMLDOC(This));
537 }
538
539 static ULONG WINAPI PersistStreamInit_Release(IPersistStreamInit *iface)
540 {
541     HTMLDocument *This = PERSTRINIT_THIS(iface);
542     return IHTMLDocument2_Release(HTMLDOC(This));
543 }
544
545 static HRESULT WINAPI PersistStreamInit_GetClassID(IPersistStreamInit *iface, CLSID *pClassID)
546 {
547     HTMLDocument *This = PERSTRINIT_THIS(iface);
548     return IPersist_GetClassID(PERSIST(This), pClassID);
549 }
550
551 static HRESULT WINAPI PersistStreamInit_IsDirty(IPersistStreamInit *iface)
552 {
553     HTMLDocument *This = PERSTRINIT_THIS(iface);
554     FIXME("(%p)\n", This);
555     return E_NOTIMPL;
556 }
557
558 static HRESULT WINAPI PersistStreamInit_Load(IPersistStreamInit *iface, LPSTREAM pStm)
559 {
560     HTMLDocument *This = PERSTRINIT_THIS(iface);
561     FIXME("(%p)->(%p)\n", This, pStm);
562     return E_NOTIMPL;
563 }
564
565 static HRESULT WINAPI PersistStreamInit_Save(IPersistStreamInit *iface, LPSTREAM pStm,
566                                              BOOL fClearDirty)
567 {
568     HTMLDocument *This = PERSTRINIT_THIS(iface);
569     char *str;
570     DWORD len, written=0;
571     HRESULT hres;
572
573     WARN("(%p)->(%p %x) needs more work\n", This, pStm, fClearDirty);
574
575     hres = get_doc_string(This, &str, &len);
576     if(FAILED(hres))
577         return hres;
578
579
580     hres = IStream_Write(pStm, str, len, &written);
581     if(FAILED(hres))
582         FIXME("Write failed: %08x\n", hres);
583
584     mshtml_free(str);
585     return S_OK;
586 }
587
588 static HRESULT WINAPI PersistStreamInit_GetSizeMax(IPersistStreamInit *iface,
589                                                    ULARGE_INTEGER *pcbSize)
590 {
591     HTMLDocument *This = PERSTRINIT_THIS(iface);
592     FIXME("(%p)->(%p)\n", This, pcbSize);
593     return E_NOTIMPL;
594 }
595
596 static HRESULT WINAPI PersistStreamInit_InitNew(IPersistStreamInit *iface)
597 {
598     HTMLDocument *This = PERSTRINIT_THIS(iface);
599     FIXME("(%p)\n", This);
600     return E_NOTIMPL;
601 }
602
603 #undef PERSTRINIT_THIS
604
605 static const IPersistStreamInitVtbl PersistStreamInitVtbl = {
606     PersistStreamInit_QueryInterface,
607     PersistStreamInit_AddRef,
608     PersistStreamInit_Release,
609     PersistStreamInit_GetClassID,
610     PersistStreamInit_IsDirty,
611     PersistStreamInit_Load,
612     PersistStreamInit_Save,
613     PersistStreamInit_GetSizeMax,
614     PersistStreamInit_InitNew
615 };
616
617 void HTMLDocument_Persist_Init(HTMLDocument *This)
618 {
619     This->lpPersistMonikerVtbl = &PersistMonikerVtbl;
620     This->lpPersistFileVtbl = &PersistFileVtbl;
621     This->lpMonikerPropVtbl = &MonikerPropVtbl;
622     This->lpPersistStreamInitVtbl = &PersistStreamInitVtbl;
623
624     This->bscallback = NULL;
625 }