Only return SHDOCVW_ClassFactory if asked for CLSID_WebBrowser.
[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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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
34 #include "wine/debug.h"
35
36 #include "mshtml_private.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
39
40 struct BindStatusCallback {
41     const IBindStatusCallbackVtbl *lpBindStatusCallbackVtbl;
42
43     LONG ref;
44
45     HTMLDocument *doc;
46     IBinding *binding;
47     IStream *stream;
48     LPOLESTR url;
49 };
50
51 #define STATUSCLB_THIS(iface) DEFINE_THIS(BindStatusCallback, BindStatusCallback, iface)
52
53 static HRESULT WINAPI BindStatusCallback_QueryInterface(IBindStatusCallback *iface,
54         REFIID riid, void **ppv)
55 {
56     BindStatusCallback *This = STATUSCLB_THIS(iface);
57
58     *ppv = NULL;
59     if(IsEqualGUID(&IID_IUnknown, riid)) {
60         TRACE("(%p)->(IID_IUnknown, %p)\n", This, ppv);
61         *ppv = STATUSCLB(This);
62     }else if(IsEqualGUID(&IID_IBindStatusCallback, riid)) {
63         TRACE("(%p)->(IID_IBindStatusCallback, %p)\n", This, ppv);
64         *ppv = STATUSCLB(This);
65     }
66
67     if(*ppv) {
68         IBindStatusCallback_AddRef(STATUSCLB(This));
69         return S_OK;
70     }
71
72     TRACE("Unsupported riid = %s\n", debugstr_guid(riid));
73     return E_NOINTERFACE;
74 }
75
76 static ULONG WINAPI BindStatusCallback_AddRef(IBindStatusCallback *iface)
77 {
78     BindStatusCallback *This = STATUSCLB_THIS(iface);
79     LONG ref = InterlockedIncrement(&This->ref);
80
81     TRACE("(%p) ref = %ld\n", This, ref);
82
83     return ref;
84 }
85
86 static ULONG WINAPI BindStatusCallback_Release(IBindStatusCallback *iface)
87 {
88     BindStatusCallback *This = STATUSCLB_THIS(iface);
89     LONG ref = InterlockedDecrement(&This->ref);
90
91     TRACE("(%p) ref = %ld\n", This, ref);
92
93     if(!ref) {
94         if(This->doc->status_callback == This)
95             This->doc->status_callback = NULL;
96         IHTMLDocument2_Release(HTMLDOC(This->doc));
97         if(This->stream)
98             IStream_Release(This->stream);
99         CoTaskMemFree(This->url);
100         HeapFree(GetProcessHeap(), 0, This);
101     }
102
103     return ref;
104 }
105
106 static HRESULT WINAPI BindStatusCallback_OnStartBinding(IBindStatusCallback *iface,
107         DWORD dwReserved, IBinding *pbind)
108 {
109     BindStatusCallback *This = STATUSCLB_THIS(iface);
110
111     TRACE("(%p)->(%ld %p)\n", This, dwReserved, pbind);
112
113     This->binding = pbind;
114     IBinding_AddRef(pbind);
115
116     if(This->doc->nscontainer && This->doc->nscontainer->stream) {
117         nsACString *strTextHtml;
118         nsresult nsres;
119         nsIURI *uri = get_nsIURI(This->url);
120
121         strTextHtml = nsACString_Create();
122         /* FIXME: Set it correctly */
123         nsACString_SetData(strTextHtml, "text/html");
124
125         nsres = nsIWebBrowserStream_OpenStream(This->doc->nscontainer->stream, uri, strTextHtml);
126         if(NS_FAILED(nsres))
127             ERR("OpenStream failed: %08lx\n", nsres);
128
129         nsIURI_Release(uri);
130     }
131
132     return S_OK;
133 }
134
135 static HRESULT WINAPI BindStatusCallback_GetPriority(IBindStatusCallback *iface, LONG *pnPriority)
136 {
137     BindStatusCallback *This = STATUSCLB_THIS(iface);
138     FIXME("(%p)->(%p)\n", This, pnPriority);
139     return E_NOTIMPL;
140 }
141
142 static HRESULT WINAPI BindStatusCallback_OnLowResource(IBindStatusCallback *iface, DWORD reserved)
143 {
144     BindStatusCallback *This = STATUSCLB_THIS(iface);
145     FIXME("(%p)->(%ld)\n", This, reserved);
146     return E_NOTIMPL;
147 }
148
149 static HRESULT WINAPI BindStatusCallback_OnProgress(IBindStatusCallback *iface, ULONG ulProgress,
150         ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
151 {
152     BindStatusCallback *This = STATUSCLB_THIS(iface);
153     TRACE("%p)->(%lu %lu %lu %s)\n", This, ulProgress, ulProgressMax, ulStatusCode,
154             debugstr_w(szStatusText));
155     return S_OK;
156 }
157
158 static HRESULT WINAPI BindStatusCallback_OnStopBinding(IBindStatusCallback *iface,
159         HRESULT hresult, LPCWSTR szError)
160 {
161     BindStatusCallback *This = STATUSCLB_THIS(iface);
162
163     TRACE("(%p)->(%08lx %s)\n", This, hresult, debugstr_w(szError));
164
165     if(This->doc->nscontainer && This->doc->nscontainer->stream)
166         nsIWebBrowserStream_CloseStream(This->doc->nscontainer->stream);
167
168     IBinding_Release(This->binding);
169     This->binding = NULL;
170     return S_OK;
171 }
172
173 static HRESULT WINAPI BindStatusCallback_GetBindInfo(IBindStatusCallback *iface,
174         DWORD *grfBINDF, BINDINFO *pbindinfo)
175 {
176     BindStatusCallback *This = STATUSCLB_THIS(iface);
177     DWORD size;
178
179     TRACE("(%p)->(%p %p)\n", This, grfBINDF, pbindinfo);
180
181     *grfBINDF = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE | BINDF_PULLDATA;
182     size = pbindinfo->cbSize;
183     memset(pbindinfo, 0, size);
184     pbindinfo->cbSize = size;
185
186     return S_OK;
187 }
188
189 static HRESULT WINAPI BindStatusCallback_OnDataAvailable(IBindStatusCallback *iface,
190         DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
191 {
192     BindStatusCallback *This = STATUSCLB_THIS(iface);
193
194     TRACE("(%p)->(%08lx %ld %p %p)\n", This, grfBSCF, dwSize, pformatetc, pstgmed);
195
196     if(!This->stream) {
197         This->stream = pstgmed->u.pstm;
198         IStream_AddRef(This->stream);
199     }
200
201     if(This->doc->nscontainer && This->doc->nscontainer->stream) {
202         BYTE buf[1024];
203         DWORD size;
204         HRESULT hres;
205
206         do {
207             size = sizeof(buf);
208             hres = IStream_Read(This->stream, buf, size, &size);
209             nsIWebBrowserStream_AppendToStream(This->doc->nscontainer->stream, buf, size);
210         }while(hres == S_OK);
211     }
212
213     return S_OK;
214 }
215
216 static HRESULT WINAPI BindStatusCallback_OnObjectAvailable(IBindStatusCallback *iface,
217         REFIID riid, IUnknown *punk)
218 {
219     BindStatusCallback *This = STATUSCLB_THIS(iface);
220     FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), punk);
221     return E_NOTIMPL;
222 }
223
224 static const IBindStatusCallbackVtbl BindStatusCallbackVtbl = {
225     BindStatusCallback_QueryInterface,
226     BindStatusCallback_AddRef,
227     BindStatusCallback_Release,
228     BindStatusCallback_OnStartBinding,
229     BindStatusCallback_GetPriority,
230     BindStatusCallback_OnLowResource,
231     BindStatusCallback_OnProgress,
232     BindStatusCallback_OnStopBinding,
233     BindStatusCallback_GetBindInfo,
234     BindStatusCallback_OnDataAvailable,
235     BindStatusCallback_OnObjectAvailable
236 };
237
238 static BindStatusCallback *BindStatusCallback_Create(HTMLDocument *doc, LPOLESTR url)
239 {
240     BindStatusCallback *ret = HeapAlloc(GetProcessHeap(), 0, sizeof(BindStatusCallback));
241     ret->lpBindStatusCallbackVtbl = &BindStatusCallbackVtbl;
242     ret->ref = 0;
243     ret->url = url;
244     ret->doc = doc;
245     ret->stream = NULL;
246     IHTMLDocument2_AddRef(HTMLDOC(doc));
247     return ret;
248 }
249
250 /**********************************************************
251  * IPersistMoniker implementation
252  */
253
254 #define PERSISTMON_THIS(iface) DEFINE_THIS(HTMLDocument, PersistMoniker, iface)
255
256 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid,
257                                                             void **ppvObject)
258 {
259     HTMLDocument *This = PERSISTMON_THIS(iface);
260     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
261 }
262
263 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
264 {
265     HTMLDocument *This = PERSISTMON_THIS(iface);
266     return IHTMLDocument2_AddRef(HTMLDOC(This));
267 }
268
269 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
270 {
271     HTMLDocument *This = PERSISTMON_THIS(iface);
272     return IHTMLDocument2_Release(HTMLDOC(This));
273 }
274
275 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
276 {
277     HTMLDocument *This = PERSISTMON_THIS(iface);
278     return IPersist_GetClassID(PERSIST(This), pClassID);
279 }
280
281 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
282 {
283     HTMLDocument *This = PERSISTMON_THIS(iface);
284     FIXME("(%p)\n", This);
285     return E_NOTIMPL;
286 }
287
288 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
289         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
290 {
291     HTMLDocument *This = PERSISTMON_THIS(iface);
292     IBindCtx *pbind;
293     BindStatusCallback *callback;
294     IStream *str = NULL;
295     LPOLESTR url;
296     HRESULT hres;
297     nsresult nsres;
298
299     TRACE("(%p)->(%x %p %p %08lx)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
300
301     if(pibc) {
302         IUnknown *unk = NULL;
303
304         static WCHAR wszClientSiteParam[] = {'{','d','4','d','b','6','8','5','0','-',
305             '5','3','8','5','-','1','1','d','0','-','8','9','e','9','-','0','0','a',
306             '0','c','9','0','a','9','0','a','c','}',0};
307
308         /* FIXME:
309          * Use params:
310          * "__PrecreatedObject"
311          * "BIND_CONTEXT_PARAM"
312          * "__HTMLLOADOPTIONS"
313          * "__DWNBINDINFO"
314          * "URL Context"
315          * "CBinding Context"
316          * "_ITransData_Object_"
317          * "_EnumFORMATETC_"
318          */
319
320         IBindCtx_GetObjectParam(pibc, wszClientSiteParam, &unk);
321         if(unk) {
322             IOleClientSite *client = NULL;
323
324             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
325             if(SUCCEEDED(hres)) {
326                 TRACE("Got client site %p\n", client);
327                 IOleObject_SetClientSite(OLEOBJ(This), client);
328                 IOleClientSite_Release(client);
329             }
330
331             IUnknown_Release(unk);
332         }
333     }
334
335     HTMLDocument_LockContainer(This, TRUE);
336     
337     hres = IMoniker_GetDisplayName(pimkName, pibc, NULL, &url);
338     if(FAILED(hres)) {
339         WARN("GetDiaplayName failed: %08lx\n", hres);
340         return hres;
341     }
342
343     TRACE("got url: %s\n", debugstr_w(url));
344
345     if(This->client) {
346         IOleCommandTarget *cmdtrg = NULL;
347
348         hres = IOleClientSite_QueryInterface(This->client, &IID_IOleCommandTarget,
349                 (void**)&cmdtrg);
350         if(SUCCEEDED(hres)) {
351             VARIANT var;
352
353             V_VT(&var) = VT_I4;
354             V_I4(&var) = 0;
355             IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
356         }
357     }
358
359     if(This->nscontainer && !This->nscontainer->stream) {
360         /*
361          * This is a workaround for older Gecko that doesn't support nsIWebBrowserStream.
362          * It uses Gecko's LoadURI instead of IMoniker's BindToStorage. Should we improve
363          * it (to do so we'd have to use not frozen interfaces)?
364          */
365         LPOLESTR old_url = This->nscontainer->url;
366
367         This->nscontainer->url = url;
368         nsres = nsIWebNavigation_LoadURI(This->nscontainer->navigation, url,
369                 LOAD_FLAGS_NONE, NULL, NULL, NULL);
370         if(NS_SUCCEEDED(nsres)) {
371             CoTaskMemFree(old_url);
372             return S_OK;
373         }else {
374             WARN("LoadURI failed: %08lx\n", nsres);
375             This->nscontainer->url = old_url;
376         }
377     }    
378
379     /* FIXME: Use grfMode */
380
381     if(fFullyAvailable)
382         FIXME("not supported fFullyAvailable\n");
383
384     if(This->status_callback && This->status_callback->binding)
385         IBinding_Abort(This->status_callback->binding);
386
387     callback = This->status_callback = BindStatusCallback_Create(This, url);
388
389     if(pibc) {
390         pbind = pibc;
391         RegisterBindStatusCallback(pbind, STATUSCLB(callback), NULL, 0);
392     }else {
393         CreateAsyncBindCtx(0, STATUSCLB(callback), NULL, &pbind);
394     }
395
396     hres = IMoniker_BindToStorage(pimkName, pbind, NULL, &IID_IStream, (void**)&str);
397
398     if(!pibc)
399         IBindCtx_Release(pbind);
400     if(str)
401         IStream_Release(str);
402     if(FAILED(hres)) {
403         WARN("BindToStorage failed: %08lx\n", hres);
404         return hres;
405     }
406
407     return S_OK;
408 }
409
410 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
411         LPBC pbc, BOOL fRemember)
412 {
413     HTMLDocument *This = PERSISTMON_THIS(iface);
414     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
415     return E_NOTIMPL;
416 }
417
418 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
419 {
420     HTMLDocument *This = PERSISTMON_THIS(iface);
421     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
422     return E_NOTIMPL;
423 }
424
425 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
426 {
427     HTMLDocument *This = PERSISTMON_THIS(iface);
428     FIXME("(%p)->(%p)\n", This, ppimkName);
429     return E_NOTIMPL;
430 }
431
432 static const IPersistMonikerVtbl PersistMonikerVtbl = {
433     PersistMoniker_QueryInterface,
434     PersistMoniker_AddRef,
435     PersistMoniker_Release,
436     PersistMoniker_GetClassID,
437     PersistMoniker_IsDirty,
438     PersistMoniker_Load,
439     PersistMoniker_Save,
440     PersistMoniker_SaveCompleted,
441     PersistMoniker_GetCurMoniker
442 };
443
444 /**********************************************************
445  * IMonikerProp implementation
446  */
447
448 #define MONPROP_THIS(iface) DEFINE_THIS(HTMLDocument, MonikerProp, iface)
449
450 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppvObject)
451 {
452     HTMLDocument *This = MONPROP_THIS(iface);
453     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
454 }
455
456 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
457 {
458     HTMLDocument *This = MONPROP_THIS(iface);
459     return IHTMLDocument2_AddRef(HTMLDOC(This));
460 }
461
462 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
463 {
464     HTMLDocument *This = MONPROP_THIS(iface);
465     return IHTMLDocument_Release(HTMLDOC(This));
466 }
467
468 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
469 {
470     HTMLDocument *This = MONPROP_THIS(iface);
471     FIXME("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
472     return E_NOTIMPL;
473 }
474
475 static const IMonikerPropVtbl MonikerPropVtbl = {
476     MonikerProp_QueryInterface,
477     MonikerProp_AddRef,
478     MonikerProp_Release,
479     MonikerProp_PutProperty
480 };
481
482 /**********************************************************
483  * IPersistFile implementation
484  */
485
486 #define PERSISTFILE_THIS(iface) DEFINE_THIS(HTMLDocument, PersistFile, iface)
487
488 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppvObject)
489 {
490     HTMLDocument *This = PERSISTFILE_THIS(iface);
491     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
492 }
493
494 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
495 {
496     HTMLDocument *This = PERSISTFILE_THIS(iface);
497     return IHTMLDocument2_AddRef(HTMLDOC(This));
498 }
499
500 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
501 {
502     HTMLDocument *This = PERSISTFILE_THIS(iface);
503     return IHTMLDocument2_Release(HTMLDOC(This));
504 }
505
506 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
507 {
508     HTMLDocument *This = PERSISTFILE_THIS(iface);
509
510     TRACE("(%p)->(%p)\n", This, pClassID);
511
512     if(!pClassID)
513         return E_INVALIDARG;
514
515     memcpy(pClassID, &CLSID_HTMLDocument, sizeof(CLSID));
516     return S_OK;
517 }
518
519 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
520 {
521     HTMLDocument *This = PERSISTFILE_THIS(iface);
522     FIXME("(%p)\n", This);
523     return E_NOTIMPL;
524 }
525
526 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
527 {
528     HTMLDocument *This = PERSISTFILE_THIS(iface);
529     FIXME("(%p)->(%s %08lx)\n", This, debugstr_w(pszFileName), dwMode);
530     return E_NOTIMPL;
531 }
532
533 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
534 {
535     HTMLDocument *This = PERSISTFILE_THIS(iface);
536     FIXME("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
537     return E_NOTIMPL;
538 }
539
540 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
541 {
542     HTMLDocument *This = PERSISTFILE_THIS(iface);
543     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
544     return E_NOTIMPL;
545 }
546
547 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
548 {
549     HTMLDocument *This = PERSISTFILE_THIS(iface);
550     FIXME("(%p)->(%p)\n", This, pszFileName);
551     return E_NOTIMPL;
552 }
553
554 static const IPersistFileVtbl PersistFileVtbl = {
555     PersistFile_QueryInterface,
556     PersistFile_AddRef,
557     PersistFile_Release,
558     PersistFile_GetClassID,
559     PersistFile_IsDirty,
560     PersistFile_Load,
561     PersistFile_Save,
562     PersistFile_SaveCompleted,
563     PersistFile_GetCurFile
564 };
565
566 BOOL HTMLDocument_OnLoad(HTMLDocument *This, LPCWSTR url)
567 {
568     IOleCommandTarget *cmdtrg = NULL;
569     HRESULT hres;
570
571     TRACE("(%p)->(%s)\n", This, debugstr_w(url));
572
573     hres = IOleClientSite_QueryInterface(This->client, &IID_IOleCommandTarget, (void**)&cmdtrg);
574     if(SUCCEEDED(hres)) {
575         VARIANT varUrl, varRes;
576
577         V_VT(&varUrl) = VT_BSTR;
578         V_BSTR(&varUrl) = SysAllocString(url);
579         V_VT(&varRes) = VT_BOOL;
580
581         hres = IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 67, 0, &varUrl, &varRes);
582         SysFreeString(V_BSTR(&varUrl));
583
584         if(SUCCEEDED(hres) && !V_BOOL(&varRes)) {
585             TRACE("got VARIANT_FALSE, do not load\n");
586             IOleCommandTarget_Release(cmdtrg);
587             return FALSE;
588         }
589     }
590
591     return TRUE;
592 }
593
594 void HTMLDocument_Persist_Init(HTMLDocument *This)
595 {
596     This->lpPersistMonikerVtbl = &PersistMonikerVtbl;
597     This->lpPersistFileVtbl = &PersistFileVtbl;
598     This->lpMonikerPropVtbl = &MonikerPropVtbl;
599
600     This->status_callback = NULL;
601 }