d3drm: Fix QueryInterface forwarding.
[wine] / dlls / mshtml / oleobj.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
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "ole2.h"
30 #include "shlguid.h"
31 #include "shdeprecated.h"
32 #include "mshtmdid.h"
33 #include "idispids.h"
34
35 #include "wine/debug.h"
36
37 #include "mshtml_private.h"
38
39 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
40
41 #define DOCHOST_DOCCANNAVIGATE  0
42
43 /**********************************************************
44  * IOleObject implementation
45  */
46
47 static inline HTMLDocument *impl_from_IOleObject(IOleObject *iface)
48 {
49     return CONTAINING_RECORD(iface, HTMLDocument, IOleObject_iface);
50 }
51
52 static HRESULT WINAPI OleObject_QueryInterface(IOleObject *iface, REFIID riid, void **ppv)
53 {
54     HTMLDocument *This = impl_from_IOleObject(iface);
55     return htmldoc_query_interface(This, riid, ppv);
56 }
57
58 static ULONG WINAPI OleObject_AddRef(IOleObject *iface)
59 {
60     HTMLDocument *This = impl_from_IOleObject(iface);
61     return htmldoc_addref(This);
62 }
63
64 static ULONG WINAPI OleObject_Release(IOleObject *iface)
65 {
66     HTMLDocument *This = impl_from_IOleObject(iface);
67     return htmldoc_release(This);
68 }
69
70 static void update_hostinfo(HTMLDocumentObj *This, DOCHOSTUIINFO *hostinfo)
71 {
72     nsIScrollable *scrollable;
73     nsresult nsres;
74
75     if(!This->nscontainer)
76         return;
77
78     nsres = nsIWebBrowser_QueryInterface(This->nscontainer->webbrowser, &IID_nsIScrollable, (void**)&scrollable);
79     if(NS_SUCCEEDED(nsres)) {
80         nsres = nsIScrollable_SetDefaultScrollbarPreferences(scrollable, ScrollOrientation_Y,
81                 (hostinfo->dwFlags & DOCHOSTUIFLAG_SCROLL_NO) ? Scrollbar_Never : Scrollbar_Always);
82         if(NS_FAILED(nsres))
83             ERR("Could not set default Y scrollbar prefs: %08x\n", nsres);
84
85         nsres = nsIScrollable_SetDefaultScrollbarPreferences(scrollable, ScrollOrientation_X,
86                 hostinfo->dwFlags & DOCHOSTUIFLAG_SCROLL_NO ? Scrollbar_Never : Scrollbar_Auto);
87         if(NS_FAILED(nsres))
88             ERR("Could not set default X scrollbar prefs: %08x\n", nsres);
89
90         nsIScrollable_Release(scrollable);
91     }else {
92         ERR("Could not get nsIScrollable: %08x\n", nsres);
93     }
94 }
95
96 static HRESULT WINAPI OleObject_SetClientSite(IOleObject *iface, IOleClientSite *pClientSite)
97 {
98     HTMLDocument *This = impl_from_IOleObject(iface);
99     IOleCommandTarget *cmdtrg = NULL;
100     IOleWindow *ole_window;
101     BOOL hostui_setup;
102     VARIANT silent;
103     HWND hwnd;
104     HRESULT hres;
105
106     TRACE("(%p)->(%p)\n", This, pClientSite);
107
108     if(pClientSite == This->doc_obj->client)
109         return S_OK;
110
111     if(This->doc_obj->client) {
112         IOleClientSite_Release(This->doc_obj->client);
113         This->doc_obj->client = NULL;
114         This->doc_obj->usermode = UNKNOWN_USERMODE;
115     }
116
117     if(This->doc_obj->hostui && !This->doc_obj->custom_hostui) {
118         IDocHostUIHandler_Release(This->doc_obj->hostui);
119         This->doc_obj->hostui = NULL;
120     }
121
122     if(This->doc_obj->doc_object_service) {
123         IDocObjectService_Release(This->doc_obj->doc_object_service);
124         This->doc_obj->doc_object_service = NULL;
125     }
126
127     memset(&This->doc_obj->hostinfo, 0, sizeof(DOCHOSTUIINFO));
128     This->doc_obj->is_webbrowser = FALSE;
129
130     if(!pClientSite)
131         return S_OK;
132
133     IOleClientSite_AddRef(pClientSite);
134     This->doc_obj->client = pClientSite;
135
136     hostui_setup = This->doc_obj->hostui_setup;
137
138     if(!This->doc_obj->hostui) {
139         IDocHostUIHandler *uihandler;
140
141         This->doc_obj->custom_hostui = FALSE;
142
143         hres = IOleObject_QueryInterface(pClientSite, &IID_IDocHostUIHandler, (void**)&uihandler);
144         if(SUCCEEDED(hres))
145             This->doc_obj->hostui = uihandler;
146     }
147
148     if(This->doc_obj->hostui) {
149         DOCHOSTUIINFO hostinfo;
150         LPOLESTR key_path = NULL, override_key_path = NULL;
151         IDocHostUIHandler2 *uihandler2;
152
153         memset(&hostinfo, 0, sizeof(DOCHOSTUIINFO));
154         hostinfo.cbSize = sizeof(DOCHOSTUIINFO);
155         hres = IDocHostUIHandler_GetHostInfo(This->doc_obj->hostui, &hostinfo);
156         if(SUCCEEDED(hres)) {
157             TRACE("hostinfo = {%u %08x %08x %s %s}\n",
158                     hostinfo.cbSize, hostinfo.dwFlags, hostinfo.dwDoubleClick,
159                     debugstr_w(hostinfo.pchHostCss), debugstr_w(hostinfo.pchHostNS));
160             update_hostinfo(This->doc_obj, &hostinfo);
161             This->doc_obj->hostinfo = hostinfo;
162         }
163
164         if(!hostui_setup) {
165             hres = IDocHostUIHandler_GetOptionKeyPath(This->doc_obj->hostui, &key_path, 0);
166             if(hres == S_OK && key_path) {
167                 if(key_path[0]) {
168                     /* FIXME: use key_path */
169                     TRACE("key_path = %s\n", debugstr_w(key_path));
170                 }
171                 CoTaskMemFree(key_path);
172             }
173
174             hres = IDocHostUIHandler_QueryInterface(This->doc_obj->hostui, &IID_IDocHostUIHandler2,
175                     (void**)&uihandler2);
176             if(SUCCEEDED(hres)) {
177                 hres = IDocHostUIHandler2_GetOverrideKeyPath(uihandler2, &override_key_path, 0);
178                 if(hres == S_OK && override_key_path && override_key_path[0]) {
179                     if(override_key_path[0]) {
180                         /*FIXME: use override_key_path */
181                         TRACE("override_key_path = %s\n", debugstr_w(override_key_path));
182                     }
183                     CoTaskMemFree(override_key_path);
184                 }
185                 IDocHostUIHandler2_Release(uihandler2);
186             }
187
188             This->doc_obj->hostui_setup = TRUE;
189         }
190     }
191
192     /* Native calls here GetWindow. What is it for?
193      * We don't have anything to do with it here (yet). */
194     hres = IOleClientSite_QueryInterface(pClientSite, &IID_IOleWindow, (void**)&ole_window);
195     if(SUCCEEDED(hres)) {
196         IOleWindow_GetWindow(ole_window, &hwnd);
197         IOleWindow_Release(ole_window);
198     }
199
200     hres = IOleClientSite_QueryInterface(pClientSite, &IID_IOleCommandTarget, (void**)&cmdtrg);
201     if(SUCCEEDED(hres)) {
202         VARIANT var;
203         OLECMD cmd = {OLECMDID_SETPROGRESSTEXT, 0};
204
205         if(!hostui_setup) {
206             IDocObjectService *doc_object_service;
207             IBrowserService *browser_service;
208             IWebBrowser2 *wb;
209
210             V_VT(&var) = VT_UNKNOWN;
211             V_UNKNOWN(&var) = (IUnknown*)&This->window->IHTMLWindow2_iface;
212             IOleCommandTarget_Exec(cmdtrg, &CGID_DocHostCmdPriv, DOCHOST_DOCCANNAVIGATE, 0, &var, NULL);
213
214             hres = do_query_service((IUnknown*)pClientSite, &IID_IShellBrowser,
215                     &IID_IBrowserService, (void**)&browser_service);
216             if(SUCCEEDED(hres)) {
217                 hres = IBrowserService_QueryInterface(browser_service,
218                         &IID_IDocObjectService, (void**)&doc_object_service);
219                 if(SUCCEEDED(hres)) {
220                     This->doc_obj->doc_object_service = doc_object_service;
221
222                     /*
223                      * Some embedding routines, esp. in regards to use of IDocObjectService, differ if
224                      * embedder supports IWebBrowserApp.
225                      */
226                     hres = do_query_service((IUnknown*)pClientSite, &IID_IWebBrowserApp, &IID_IWebBrowser2, (void**)&wb);
227                     if(SUCCEEDED(hres)) {
228                         This->doc_obj->is_webbrowser = TRUE;
229                         IWebBrowser2_Release(wb);
230                     }
231
232                     IBrowserService_Release(browser_service);
233                 }
234             }
235         }
236
237         IOleCommandTarget_QueryStatus(cmdtrg, NULL, 1, &cmd, NULL);
238
239         V_VT(&var) = VT_I4;
240         V_I4(&var) = 0;
241         IOleCommandTarget_Exec(cmdtrg, NULL, OLECMDID_SETPROGRESSMAX,
242                 OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
243         IOleCommandTarget_Exec(cmdtrg, NULL, OLECMDID_SETPROGRESSPOS, 
244                 OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
245
246         IOleCommandTarget_Release(cmdtrg);
247     }
248
249     if(This->doc_obj->usermode == UNKNOWN_USERMODE)
250         IOleControl_OnAmbientPropertyChange(&This->IOleControl_iface, DISPID_AMBIENT_USERMODE);
251
252     IOleControl_OnAmbientPropertyChange(&This->IOleControl_iface,
253             DISPID_AMBIENT_OFFLINEIFNOTCONNECTED);
254
255     hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_SILENT, &silent);
256     if(SUCCEEDED(hres)) {
257         if(V_VT(&silent) != VT_BOOL)
258             WARN("V_VT(silent) = %d\n", V_VT(&silent));
259         else if(V_BOOL(&silent))
260             FIXME("silent == true\n");
261     }
262
263     IOleControl_OnAmbientPropertyChange(&This->IOleControl_iface, DISPID_AMBIENT_USERAGENT);
264     IOleControl_OnAmbientPropertyChange(&This->IOleControl_iface, DISPID_AMBIENT_PALETTE);
265
266     return S_OK;
267 }
268
269 static HRESULT WINAPI OleObject_GetClientSite(IOleObject *iface, IOleClientSite **ppClientSite)
270 {
271     HTMLDocument *This = impl_from_IOleObject(iface);
272
273     TRACE("(%p)->(%p)\n", This, ppClientSite);
274
275     if(!ppClientSite)
276         return E_INVALIDARG;
277
278     if(This->doc_obj->client)
279         IOleClientSite_AddRef(This->doc_obj->client);
280     *ppClientSite = This->doc_obj->client;
281
282     return S_OK;
283 }
284
285 static HRESULT WINAPI OleObject_SetHostNames(IOleObject *iface, LPCOLESTR szContainerApp, LPCOLESTR szContainerObj)
286 {
287     HTMLDocument *This = impl_from_IOleObject(iface);
288     FIXME("(%p)->(%s %s)\n", This, debugstr_w(szContainerApp), debugstr_w(szContainerObj));
289     return E_NOTIMPL;
290 }
291
292 static HRESULT WINAPI OleObject_Close(IOleObject *iface, DWORD dwSaveOption)
293 {
294     HTMLDocument *This = impl_from_IOleObject(iface);
295
296     TRACE("(%p)->(%08x)\n", This, dwSaveOption);
297
298     if(dwSaveOption == OLECLOSE_PROMPTSAVE)
299         FIXME("OLECLOSE_PROMPTSAVE not implemented\n");
300
301     if(This->doc_obj->in_place_active)
302         IOleInPlaceObjectWindowless_InPlaceDeactivate(&This->IOleInPlaceObjectWindowless_iface);
303
304     HTMLDocument_LockContainer(This->doc_obj, FALSE);
305
306     if(This->advise_holder)
307         IOleAdviseHolder_SendOnClose(This->advise_holder);
308     
309     return S_OK;
310 }
311
312 static HRESULT WINAPI OleObject_SetMoniker(IOleObject *iface, DWORD dwWhichMoniker, IMoniker *pmk)
313 {
314     HTMLDocument *This = impl_from_IOleObject(iface);
315     FIXME("(%p %d %p)->()\n", This, dwWhichMoniker, pmk);
316     return E_NOTIMPL;
317 }
318
319 static HRESULT WINAPI OleObject_GetMoniker(IOleObject *iface, DWORD dwAssign, DWORD dwWhichMoniker, IMoniker **ppmk)
320 {
321     HTMLDocument *This = impl_from_IOleObject(iface);
322     FIXME("(%p)->(%d %d %p)\n", This, dwAssign, dwWhichMoniker, ppmk);
323     return E_NOTIMPL;
324 }
325
326 static HRESULT WINAPI OleObject_InitFromData(IOleObject *iface, IDataObject *pDataObject, BOOL fCreation,
327                                         DWORD dwReserved)
328 {
329     HTMLDocument *This = impl_from_IOleObject(iface);
330     FIXME("(%p)->(%p %x %d)\n", This, pDataObject, fCreation, dwReserved);
331     return E_NOTIMPL;
332 }
333
334 static HRESULT WINAPI OleObject_GetClipboardData(IOleObject *iface, DWORD dwReserved, IDataObject **ppDataObject)
335 {
336     HTMLDocument *This = impl_from_IOleObject(iface);
337     FIXME("(%p)->(%d %p)\n", This, dwReserved, ppDataObject);
338     return E_NOTIMPL;
339 }
340
341 static HRESULT WINAPI OleObject_DoVerb(IOleObject *iface, LONG iVerb, LPMSG lpmsg, IOleClientSite *pActiveSite,
342                                         LONG lindex, HWND hwndParent, LPCRECT lprcPosRect)
343 {
344     HTMLDocument *This = impl_from_IOleObject(iface);
345     IOleDocumentSite *pDocSite;
346     HRESULT hres;
347
348     TRACE("(%p)->(%d %p %p %d %p %p)\n", This, iVerb, lpmsg, pActiveSite, lindex, hwndParent, lprcPosRect);
349
350     if(iVerb != OLEIVERB_SHOW && iVerb != OLEIVERB_UIACTIVATE && iVerb != OLEIVERB_INPLACEACTIVATE) { 
351         FIXME("iVerb = %d not supported\n", iVerb);
352         return E_NOTIMPL;
353     }
354
355     if(!pActiveSite)
356         pActiveSite = This->doc_obj->client;
357
358     hres = IOleClientSite_QueryInterface(pActiveSite, &IID_IOleDocumentSite, (void**)&pDocSite);
359     if(SUCCEEDED(hres)) {
360         HTMLDocument_LockContainer(This->doc_obj, TRUE);
361
362         /* FIXME: Create new IOleDocumentView. See CreateView for more info. */
363         hres = IOleDocumentSite_ActivateMe(pDocSite, &This->IOleDocumentView_iface);
364         IOleDocumentSite_Release(pDocSite);
365     }else {
366         hres = IOleDocumentView_UIActivate(&This->IOleDocumentView_iface, TRUE);
367         if(SUCCEEDED(hres)) {
368             if(lprcPosRect) {
369                 RECT rect; /* We need to pass rect as not const pointer */
370                 rect = *lprcPosRect;
371                 IOleDocumentView_SetRect(&This->IOleDocumentView_iface, &rect);
372             }
373             IOleDocumentView_Show(&This->IOleDocumentView_iface, TRUE);
374         }
375     }
376
377     return hres;
378 }
379
380 static HRESULT WINAPI OleObject_EnumVerbs(IOleObject *iface, IEnumOLEVERB **ppEnumOleVerb)
381 {
382     HTMLDocument *This = impl_from_IOleObject(iface);
383     FIXME("(%p)->(%p)\n", This, ppEnumOleVerb);
384     return E_NOTIMPL;
385 }
386
387 static HRESULT WINAPI OleObject_Update(IOleObject *iface)
388 {
389     HTMLDocument *This = impl_from_IOleObject(iface);
390     FIXME("(%p)\n", This);
391     return E_NOTIMPL;
392 }
393
394 static HRESULT WINAPI OleObject_IsUpToDate(IOleObject *iface)
395 {
396     HTMLDocument *This = impl_from_IOleObject(iface);
397     FIXME("(%p)\n", This);
398     return E_NOTIMPL;
399 }
400
401 static HRESULT WINAPI OleObject_GetUserClassID(IOleObject *iface, CLSID *pClsid)
402 {
403     HTMLDocument *This = impl_from_IOleObject(iface);
404
405     TRACE("(%p)->(%p)\n", This, pClsid);
406
407     if(!pClsid)
408         return E_INVALIDARG;
409
410     *pClsid = CLSID_HTMLDocument;
411     return S_OK;
412 }
413
414 static HRESULT WINAPI OleObject_GetUserType(IOleObject *iface, DWORD dwFormOfType, LPOLESTR *pszUserType)
415 {
416     HTMLDocument *This = impl_from_IOleObject(iface);
417     FIXME("(%p)->(%d %p)\n", This, dwFormOfType, pszUserType);
418     return E_NOTIMPL;
419 }
420
421 static HRESULT WINAPI OleObject_SetExtent(IOleObject *iface, DWORD dwDrawAspect, SIZEL *psizel)
422 {
423     HTMLDocument *This = impl_from_IOleObject(iface);
424     FIXME("(%p)->(%d %p)\n", This, dwDrawAspect, psizel);
425     return E_NOTIMPL;
426 }
427
428 static HRESULT WINAPI OleObject_GetExtent(IOleObject *iface, DWORD dwDrawAspect, SIZEL *psizel)
429 {
430     HTMLDocument *This = impl_from_IOleObject(iface);
431     FIXME("(%p)->(%d %p)\n", This, dwDrawAspect, psizel);
432     return E_NOTIMPL;
433 }
434
435 static HRESULT WINAPI OleObject_Advise(IOleObject *iface, IAdviseSink *pAdvSink, DWORD *pdwConnection)
436 {
437     HTMLDocument *This = impl_from_IOleObject(iface);
438     TRACE("(%p)->(%p %p)\n", This, pAdvSink, pdwConnection);
439
440     if(!pdwConnection)
441         return E_INVALIDARG;
442
443     if(!pAdvSink) {
444         *pdwConnection = 0;
445         return E_INVALIDARG;
446     }
447
448     if(!This->advise_holder) {
449         CreateOleAdviseHolder(&This->advise_holder);
450         if(!This->advise_holder)
451             return E_OUTOFMEMORY;
452     }
453
454     return IOleAdviseHolder_Advise(This->advise_holder, pAdvSink, pdwConnection);
455 }
456
457 static HRESULT WINAPI OleObject_Unadvise(IOleObject *iface, DWORD dwConnection)
458 {
459     HTMLDocument *This = impl_from_IOleObject(iface);
460     TRACE("(%p)->(%d)\n", This, dwConnection);
461
462     if(!This->advise_holder)
463         return OLE_E_NOCONNECTION;
464
465     return IOleAdviseHolder_Unadvise(This->advise_holder, dwConnection);
466 }
467
468 static HRESULT WINAPI OleObject_EnumAdvise(IOleObject *iface, IEnumSTATDATA **ppenumAdvise)
469 {
470     HTMLDocument *This = impl_from_IOleObject(iface);
471
472     if(!This->advise_holder) {
473         *ppenumAdvise = NULL;
474         return S_OK;
475     }
476
477     return IOleAdviseHolder_EnumAdvise(This->advise_holder, ppenumAdvise);
478 }
479
480 static HRESULT WINAPI OleObject_GetMiscStatus(IOleObject *iface, DWORD dwAspect, DWORD *pdwStatus)
481 {
482     HTMLDocument *This = impl_from_IOleObject(iface);
483     FIXME("(%p)->(%d %p)\n", This, dwAspect, pdwStatus);
484     return E_NOTIMPL;
485 }
486
487 static HRESULT WINAPI OleObject_SetColorScheme(IOleObject *iface, LOGPALETTE *pLogpal)
488 {
489     HTMLDocument *This = impl_from_IOleObject(iface);
490     FIXME("(%p)->(%p)\n", This, pLogpal);
491     return E_NOTIMPL;
492 }
493
494 static const IOleObjectVtbl OleObjectVtbl = {
495     OleObject_QueryInterface,
496     OleObject_AddRef,
497     OleObject_Release,
498     OleObject_SetClientSite,
499     OleObject_GetClientSite,
500     OleObject_SetHostNames,
501     OleObject_Close,
502     OleObject_SetMoniker,
503     OleObject_GetMoniker,
504     OleObject_InitFromData,
505     OleObject_GetClipboardData,
506     OleObject_DoVerb,
507     OleObject_EnumVerbs,
508     OleObject_Update,
509     OleObject_IsUpToDate,
510     OleObject_GetUserClassID,
511     OleObject_GetUserType,
512     OleObject_SetExtent,
513     OleObject_GetExtent,
514     OleObject_Advise,
515     OleObject_Unadvise,
516     OleObject_EnumAdvise,
517     OleObject_GetMiscStatus,
518     OleObject_SetColorScheme
519 };
520
521 /**********************************************************
522  * IOleDocument implementation
523  */
524
525 static inline HTMLDocument *impl_from_IOleDocument(IOleDocument *iface)
526 {
527     return CONTAINING_RECORD(iface, HTMLDocument, IOleDocument_iface);
528 }
529
530 static HRESULT WINAPI OleDocument_QueryInterface(IOleDocument *iface, REFIID riid, void **ppv)
531 {
532     HTMLDocument *This = impl_from_IOleDocument(iface);
533     return htmldoc_query_interface(This, riid, ppv);
534 }
535
536 static ULONG WINAPI OleDocument_AddRef(IOleDocument *iface)
537 {
538     HTMLDocument *This = impl_from_IOleDocument(iface);
539     return htmldoc_addref(This);
540 }
541
542 static ULONG WINAPI OleDocument_Release(IOleDocument *iface)
543 {
544     HTMLDocument *This = impl_from_IOleDocument(iface);
545     return htmldoc_release(This);
546 }
547
548 static HRESULT WINAPI OleDocument_CreateView(IOleDocument *iface, IOleInPlaceSite *pIPSite, IStream *pstm,
549                                    DWORD dwReserved, IOleDocumentView **ppView)
550 {
551     HTMLDocument *This = impl_from_IOleDocument(iface);
552     HRESULT hres;
553
554     TRACE("(%p)->(%p %p %d %p)\n", This, pIPSite, pstm, dwReserved, ppView);
555
556     if(!ppView)
557         return E_INVALIDARG;
558
559     /* FIXME:
560      * Windows implementation creates new IOleDocumentView when function is called for the
561      * first time and returns E_FAIL when it is called for the second time, but it doesn't matter
562      * if the application uses returned interfaces, passed to ActivateMe or returned by
563      * QueryInterface, so there is no reason to create new interface. This needs more testing.
564      */
565
566     if(pIPSite) {
567         hres = IOleDocumentView_SetInPlaceSite(&This->IOleDocumentView_iface, pIPSite);
568         if(FAILED(hres))
569             return hres;
570     }
571
572     if(pstm)
573         FIXME("pstm is not supported\n");
574
575     IOleDocumentView_AddRef(&This->IOleDocumentView_iface);
576     *ppView = &This->IOleDocumentView_iface;
577     return S_OK;
578 }
579
580 static HRESULT WINAPI OleDocument_GetDocMiscStatus(IOleDocument *iface, DWORD *pdwStatus)
581 {
582     HTMLDocument *This = impl_from_IOleDocument(iface);
583     FIXME("(%p)->(%p)\n", This, pdwStatus);
584     return E_NOTIMPL;
585 }
586
587 static HRESULT WINAPI OleDocument_EnumViews(IOleDocument *iface, IEnumOleDocumentViews **ppEnum,
588                                    IOleDocumentView **ppView)
589 {
590     HTMLDocument *This = impl_from_IOleDocument(iface);
591     FIXME("(%p)->(%p %p)\n", This, ppEnum, ppView);
592     return E_NOTIMPL;
593 }
594
595 static const IOleDocumentVtbl OleDocumentVtbl = {
596     OleDocument_QueryInterface,
597     OleDocument_AddRef,
598     OleDocument_Release,
599     OleDocument_CreateView,
600     OleDocument_GetDocMiscStatus,
601     OleDocument_EnumViews
602 };
603
604 /**********************************************************
605  * IOleControl implementation
606  */
607
608 static inline HTMLDocument *impl_from_IOleControl(IOleControl *iface)
609 {
610     return CONTAINING_RECORD(iface, HTMLDocument, IOleControl_iface);
611 }
612
613 static HRESULT WINAPI OleControl_QueryInterface(IOleControl *iface, REFIID riid, void **ppv)
614 {
615     HTMLDocument *This = impl_from_IOleControl(iface);
616     return htmldoc_query_interface(This, riid, ppv);
617 }
618
619 static ULONG WINAPI OleControl_AddRef(IOleControl *iface)
620 {
621     HTMLDocument *This = impl_from_IOleControl(iface);
622     return htmldoc_addref(This);
623 }
624
625 static ULONG WINAPI OleControl_Release(IOleControl *iface)
626 {
627     HTMLDocument *This = impl_from_IOleControl(iface);
628     return htmldoc_release(This);
629 }
630
631 static HRESULT WINAPI OleControl_GetControlInfo(IOleControl *iface, CONTROLINFO *pCI)
632 {
633     HTMLDocument *This = impl_from_IOleControl(iface);
634     FIXME("(%p)->(%p)\n", This, pCI);
635     return E_NOTIMPL;
636 }
637
638 static HRESULT WINAPI OleControl_OnMnemonic(IOleControl *iface, MSG *pMsg)
639 {
640     HTMLDocument *This = impl_from_IOleControl(iface);
641     FIXME("(%p)->(%p)\n", This, pMsg);
642     return E_NOTIMPL;
643 }
644
645 HRESULT get_client_disp_property(IOleClientSite *client, DISPID dispid, VARIANT *res)
646 {
647     IDispatch *disp = NULL;
648     DISPPARAMS dispparams = {NULL, 0};
649     UINT err;
650     HRESULT hres;
651
652     hres = IOleClientSite_QueryInterface(client, &IID_IDispatch, (void**)&disp);
653     if(FAILED(hres)) {
654         TRACE("Could not get IDispatch\n");
655         return hres;
656     }
657
658     VariantInit(res);
659
660     hres = IDispatch_Invoke(disp, dispid, &IID_NULL, LOCALE_SYSTEM_DEFAULT,
661             DISPATCH_PROPERTYGET, &dispparams, res, NULL, &err);
662
663     IDispatch_Release(disp);
664
665     return hres;
666 }
667
668 static HRESULT on_change_dlcontrol(HTMLDocument *This)
669 {
670     VARIANT res;
671     HRESULT hres;
672     
673     hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_DLCONTROL, &res);
674     if(SUCCEEDED(hres))
675         FIXME("unsupported dlcontrol %08x\n", V_I4(&res));
676
677     return S_OK;
678 }
679
680 static HRESULT WINAPI OleControl_OnAmbientPropertyChange(IOleControl *iface, DISPID dispID)
681 {
682     HTMLDocument *This = impl_from_IOleControl(iface);
683     IOleClientSite *client;
684     VARIANT res;
685     HRESULT hres;
686
687     client = This->doc_obj->client;
688     if(!client) {
689         TRACE("client = NULL\n");
690         return S_OK;
691     }
692
693     switch(dispID) {
694     case DISPID_AMBIENT_USERMODE:
695         TRACE("(%p)->(DISPID_AMBIENT_USERMODE)\n", This);
696         hres = get_client_disp_property(client, DISPID_AMBIENT_USERMODE, &res);
697         if(FAILED(hres))
698             return S_OK;
699
700         if(V_VT(&res) == VT_BOOL) {
701             if(V_BOOL(&res)) {
702                 This->doc_obj->usermode = BROWSEMODE;
703             }else {
704                 FIXME("edit mode is not supported\n");
705                 This->doc_obj->usermode = EDITMODE;
706             }
707         }else {
708             FIXME("V_VT(res)=%d\n", V_VT(&res));
709         }
710         return S_OK;
711     case DISPID_AMBIENT_DLCONTROL:
712         TRACE("(%p)->(DISPID_AMBIENT_DLCONTROL)\n", This);
713         return on_change_dlcontrol(This);
714     case DISPID_AMBIENT_OFFLINEIFNOTCONNECTED:
715         TRACE("(%p)->(DISPID_AMBIENT_OFFLINEIFNOTCONNECTED)\n", This);
716         on_change_dlcontrol(This);
717         hres = get_client_disp_property(client, DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &res);
718         if(FAILED(hres))
719             return S_OK;
720
721         if(V_VT(&res) == VT_BOOL) {
722             if(V_BOOL(&res)) {
723                 FIXME("offline connection is not supported\n");
724                 hres = E_FAIL;
725             }
726         }else {
727             FIXME("V_VT(res)=%d\n", V_VT(&res));
728         }
729         return S_OK;
730     case DISPID_AMBIENT_SILENT:
731         TRACE("(%p)->(DISPID_AMBIENT_SILENT)\n", This);
732         on_change_dlcontrol(This);
733         hres = get_client_disp_property(client, DISPID_AMBIENT_SILENT, &res);
734         if(FAILED(hres))
735             return S_OK;
736
737         if(V_VT(&res) == VT_BOOL) {
738             if(V_BOOL(&res)) {
739                 FIXME("silent mode is not supported\n");
740                 hres = E_FAIL;
741             }
742         }else {
743             FIXME("V_VT(res)=%d\n", V_VT(&res));
744         }
745         return S_OK;
746     case DISPID_AMBIENT_USERAGENT:
747         TRACE("(%p)->(DISPID_AMBIENT_USERAGENT)\n", This);
748         hres = get_client_disp_property(client, DISPID_AMBIENT_USERAGENT, &res);
749         if(FAILED(hres))
750             return S_OK;
751
752         FIXME("not supported AMBIENT_USERAGENT\n");
753         hres = E_FAIL;
754         return S_OK;
755     case DISPID_AMBIENT_PALETTE:
756         TRACE("(%p)->(DISPID_AMBIENT_PALETTE)\n", This);
757         hres = get_client_disp_property(client, DISPID_AMBIENT_PALETTE, &res);
758         if(FAILED(hres))
759             return S_OK;
760
761         FIXME("not supported AMBIENT_PALETTE\n");
762         hres = E_FAIL;
763         return S_OK;
764     }
765
766     FIXME("(%p) unsupported dispID=%d\n", This, dispID);
767     return E_FAIL;
768 }
769
770 static HRESULT WINAPI OleControl_FreezeEvents(IOleControl *iface, BOOL bFreeze)
771 {
772     HTMLDocument *This = impl_from_IOleControl(iface);
773     FIXME("(%p)->(%x)\n", This, bFreeze);
774     return E_NOTIMPL;
775 }
776
777 static const IOleControlVtbl OleControlVtbl = {
778     OleControl_QueryInterface,
779     OleControl_AddRef,
780     OleControl_Release,
781     OleControl_GetControlInfo,
782     OleControl_OnMnemonic,
783     OleControl_OnAmbientPropertyChange,
784     OleControl_FreezeEvents
785 };
786
787 /**********************************************************
788  * IObjectWithSite implementation
789  */
790
791 static inline HTMLDocument *impl_from_IObjectWithSite(IObjectWithSite *iface)
792 {
793     return CONTAINING_RECORD(iface, HTMLDocument, IObjectWithSite_iface);
794 }
795
796 static HRESULT WINAPI ObjectWithSite_QueryInterface(IObjectWithSite *iface, REFIID riid, void **ppv)
797 {
798     HTMLDocument *This = impl_from_IObjectWithSite(iface);
799     return htmldoc_query_interface(This, riid, ppv);
800 }
801
802 static ULONG WINAPI ObjectWithSite_AddRef(IObjectWithSite *iface)
803 {
804     HTMLDocument *This = impl_from_IObjectWithSite(iface);
805     return htmldoc_addref(This);
806 }
807
808 static ULONG WINAPI ObjectWithSite_Release(IObjectWithSite *iface)
809 {
810     HTMLDocument *This = impl_from_IObjectWithSite(iface);
811     return htmldoc_release(This);
812 }
813
814 static HRESULT WINAPI ObjectWithSite_SetSite(IObjectWithSite *iface, IUnknown *pUnkSite)
815 {
816     HTMLDocument *This = impl_from_IObjectWithSite(iface);
817     FIXME("(%p)->(%p)\n", This, pUnkSite);
818     return E_NOTIMPL;
819 }
820
821 static HRESULT WINAPI ObjectWithSite_GetSite(IObjectWithSite* iface, REFIID riid, PVOID *ppvSite)
822 {
823     HTMLDocument *This = impl_from_IObjectWithSite(iface);
824     FIXME("(%p)->(%p)\n", This, ppvSite);
825     return E_NOTIMPL;
826 }
827
828 static const IObjectWithSiteVtbl ObjectWithSiteVtbl = {
829     ObjectWithSite_QueryInterface,
830     ObjectWithSite_AddRef,
831     ObjectWithSite_Release,
832     ObjectWithSite_SetSite,
833     ObjectWithSite_GetSite
834 };
835
836 /**********************************************************
837  * IOleContainer implementation
838  */
839
840 static inline HTMLDocument *impl_from_IOleContainer(IOleContainer *iface)
841 {
842     return CONTAINING_RECORD(iface, HTMLDocument, IOleContainer_iface);
843 }
844
845 static HRESULT WINAPI OleContainer_QueryInterface(IOleContainer *iface, REFIID riid, void **ppv)
846 {
847     HTMLDocument *This = impl_from_IOleContainer(iface);
848     return htmldoc_query_interface(This, riid, ppv);
849 }
850
851 static ULONG WINAPI OleContainer_AddRef(IOleContainer *iface)
852 {
853     HTMLDocument *This = impl_from_IOleContainer(iface);
854     return htmldoc_addref(This);
855 }
856
857 static ULONG WINAPI OleContainer_Release(IOleContainer *iface)
858 {
859     HTMLDocument *This = impl_from_IOleContainer(iface);
860     return htmldoc_release(This);
861 }
862
863 static HRESULT WINAPI OleContainer_ParseDisplayName(IOleContainer *iface, IBindCtx *pbc, LPOLESTR pszDisplayName,
864         ULONG *pchEaten, IMoniker **ppmkOut)
865 {
866     HTMLDocument *This = impl_from_IOleContainer(iface);
867     FIXME("(%p)->(%p %s %p %p)\n", This, pbc, debugstr_w(pszDisplayName), pchEaten, ppmkOut);
868     return E_NOTIMPL;
869 }
870
871 static HRESULT WINAPI OleContainer_EnumObjects(IOleContainer *iface, DWORD grfFlags, IEnumUnknown **ppenum)
872 {
873     HTMLDocument *This = impl_from_IOleContainer(iface);
874     FIXME("(%p)->(%x %p)\n", This, grfFlags, ppenum);
875     return E_NOTIMPL;
876 }
877
878 static HRESULT WINAPI OleContainer_LockContainer(IOleContainer *iface, BOOL fLock)
879 {
880     HTMLDocument *This = impl_from_IOleContainer(iface);
881     FIXME("(%p)->(%x)\n", This, fLock);
882     return E_NOTIMPL;
883 }
884
885 static const IOleContainerVtbl OleContainerVtbl = {
886     OleContainer_QueryInterface,
887     OleContainer_AddRef,
888     OleContainer_Release,
889     OleContainer_ParseDisplayName,
890     OleContainer_EnumObjects,
891     OleContainer_LockContainer
892 };
893
894 /**********************************************************
895  * IObjectSafety implementation
896  */
897
898 static inline HTMLDocument *impl_from_IObjectSafety(IObjectSafety *iface)
899 {
900     return CONTAINING_RECORD(iface, HTMLDocument, IObjectSafety_iface);
901 }
902
903 static HRESULT WINAPI ObjectSafety_QueryInterface(IObjectSafety *iface, REFIID riid, void **ppv)
904 {
905     HTMLDocument *This = impl_from_IObjectSafety(iface);
906     return htmldoc_query_interface(This, riid, ppv);
907 }
908
909 static ULONG WINAPI ObjectSafety_AddRef(IObjectSafety *iface)
910 {
911     HTMLDocument *This = impl_from_IObjectSafety(iface);
912     return htmldoc_addref(This);
913 }
914
915 static ULONG WINAPI ObjectSafety_Release(IObjectSafety *iface)
916 {
917     HTMLDocument *This = impl_from_IObjectSafety(iface);
918     return htmldoc_release(This);
919 }
920
921 static HRESULT WINAPI ObjectSafety_GetInterfaceSafetyOptions(IObjectSafety *iface,
922         REFIID riid, DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
923 {
924     HTMLDocument *This = impl_from_IObjectSafety(iface);
925     FIXME("(%p)->(%s %p %p)\n", This, debugstr_guid(riid), pdwSupportedOptions, pdwEnabledOptions);
926     return E_NOTIMPL;
927 }
928
929 static HRESULT WINAPI ObjectSafety_SetInterfaceSafetyOptions(IObjectSafety *iface,
930         REFIID riid, DWORD dwOptionSetMask, DWORD dwEnabledOptions)
931 {
932     HTMLDocument *This = impl_from_IObjectSafety(iface);
933     FIXME("(%p)->(%s %x %x)\n", This, debugstr_guid(riid), dwOptionSetMask, dwEnabledOptions);
934
935     if(IsEqualGUID(&IID_IPersistMoniker, riid) &&
936             dwOptionSetMask==INTERFACESAFE_FOR_UNTRUSTED_DATA &&
937             dwEnabledOptions==INTERFACESAFE_FOR_UNTRUSTED_DATA)
938         return S_OK;
939
940     return E_NOTIMPL;
941 }
942
943 static const IObjectSafetyVtbl ObjectSafetyVtbl = {
944     ObjectSafety_QueryInterface,
945     ObjectSafety_AddRef,
946     ObjectSafety_Release,
947     ObjectSafety_GetInterfaceSafetyOptions,
948     ObjectSafety_SetInterfaceSafetyOptions
949 };
950
951 void HTMLDocument_LockContainer(HTMLDocumentObj *This, BOOL fLock)
952 {
953     IOleContainer *container;
954     HRESULT hres;
955
956     if(!This->client || This->container_locked == fLock)
957         return;
958
959     hres = IOleClientSite_GetContainer(This->client, &container);
960     if(SUCCEEDED(hres)) {
961         IOleContainer_LockContainer(container, fLock);
962         This->container_locked = fLock;
963         IOleContainer_Release(container);
964     }
965 }
966
967 void HTMLDocument_OleObj_Init(HTMLDocument *This)
968 {
969     This->IOleObject_iface.lpVtbl = &OleObjectVtbl;
970     This->IOleDocument_iface.lpVtbl = &OleDocumentVtbl;
971     This->IOleControl_iface.lpVtbl = &OleControlVtbl;
972     This->IObjectWithSite_iface.lpVtbl = &ObjectWithSiteVtbl;
973     This->IOleContainer_iface.lpVtbl = &OleContainerVtbl;
974     This->IObjectSafety_iface.lpVtbl = &ObjectSafetyVtbl;
975 }