mshtml: Use a helper function for QueryService calls in hlink_frame_navigate.
[wine] / dlls / mshtml / main.c
1 /*
2  *    MSHTML Class Factory
3  *
4  * Copyright 2002 Lionel Ulmer
5  * Copyright 2003 Mike McCormack
6  * Copyright 2005 Jacek Caban
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  */
22
23 #include <stdarg.h>
24 #include <stdio.h>
25
26 #define COBJMACROS
27
28 #include "windef.h"
29 #include "winbase.h"
30 #include "winuser.h"
31 #include "winreg.h"
32 #include "ole2.h"
33 #include "advpub.h"
34 #include "shlwapi.h"
35 #include "optary.h"
36 #include "rpcproxy.h"
37 #include "shlguid.h"
38
39 #include "wine/debug.h"
40
41 #define INIT_GUID
42 #include "mshtml_private.h"
43 #include "resource.h"
44
45 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
46
47 HINSTANCE hInst;
48 DWORD mshtml_tls = TLS_OUT_OF_INDEXES;
49
50 static HINSTANCE shdoclc = NULL;
51 static HDC display_dc;
52 static LPWSTR status_strings[NUM_STATUS_STRINGS];
53
54 static void thread_detach(void)
55 {
56     thread_data_t *thread_data;
57
58     thread_data = get_thread_data(FALSE);
59     if(!thread_data)
60         return;
61
62     if(thread_data->thread_hwnd)
63         DestroyWindow(thread_data->thread_hwnd);
64
65     heap_free(thread_data);
66 }
67
68 static void free_strings(void)
69 {
70     int i;
71     for(i = 0; i < NUM_STATUS_STRINGS; i++)
72         heap_free(status_strings[i]);
73 }
74
75 static void process_detach(void)
76 {
77     close_gecko();
78     release_typelib();
79
80     if(shdoclc)
81         FreeLibrary(shdoclc);
82     if(mshtml_tls != TLS_OUT_OF_INDEXES)
83         TlsFree(mshtml_tls);
84     if(display_dc)
85         DeleteObject(display_dc);
86
87     free_strings();
88 }
89
90 void set_statustext(HTMLDocumentObj* doc, INT id, LPCWSTR arg)
91 {
92     int index = id - IDS_STATUS_DONE;
93     LPWSTR p = status_strings[index];
94
95     if(!doc->frame)
96         return;
97
98     if(!p)
99     {
100         DWORD len = 255;
101         p = heap_alloc(len * sizeof(WCHAR));
102         len = LoadStringW(hInst, id, p, len);
103         len++;
104         p = heap_realloc(p, len * sizeof(WCHAR));
105         if(InterlockedCompareExchangePointer((void**)&status_strings[index], p, NULL))
106         {
107             heap_free(p);
108             p = status_strings[index];
109         }
110     }
111
112     if(arg)
113     {
114         DWORD len = lstrlenW(p) + lstrlenW(arg) - 1;
115         LPWSTR buf = heap_alloc(len * sizeof(WCHAR));
116
117         snprintfW(buf, len, p, arg);
118
119         p = buf;
120     }
121
122     IOleInPlaceFrame_SetStatusText(doc->frame, p);
123
124     if(arg)
125         heap_free(p);
126 }
127
128 HRESULT do_query_service(IUnknown *unk, REFGUID guid_service, REFIID riid, void **ppv)
129 {
130     IServiceProvider *sp;
131     HRESULT hres;
132
133     hres = IUnknown_QueryInterface(unk, &IID_IServiceProvider, (void**)&sp);
134     if(FAILED(hres))
135         return hres;
136
137     hres = IServiceProvider_QueryService(sp, guid_service, riid, ppv);
138     IServiceProvider_Release(sp);
139     return hres;
140 }
141
142 HINSTANCE get_shdoclc(void)
143 {
144     static const WCHAR wszShdoclc[] =
145         {'s','h','d','o','c','l','c','.','d','l','l',0};
146
147     if(shdoclc)
148         return shdoclc;
149
150     return shdoclc = LoadLibraryExW(wszShdoclc, NULL, LOAD_LIBRARY_AS_DATAFILE);
151 }
152
153 HDC get_display_dc(void)
154 {
155     static const WCHAR displayW[] = {'D','I','S','P','L','A','Y',0};
156
157     if(!display_dc) {
158         HDC hdc;
159
160         hdc = CreateICW(displayW, NULL, NULL, NULL);
161         if(InterlockedCompareExchangePointer((void**)&display_dc, hdc, NULL))
162             DeleteObject(hdc);
163     }
164
165     return display_dc;
166 }
167
168 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
169 {
170     switch(fdwReason) {
171     case DLL_PROCESS_ATTACH:
172         hInst = hInstDLL;
173         break;
174     case DLL_PROCESS_DETACH:
175         process_detach();
176         break;
177     case DLL_THREAD_DETACH:
178         thread_detach();
179         break;
180     }
181     return TRUE;
182 }
183
184 /***********************************************************
185  *    ClassFactory implementation
186  */
187 typedef HRESULT (*CreateInstanceFunc)(IUnknown*,REFIID,void**);
188 typedef struct {
189     IClassFactory IClassFactory_iface;
190     LONG ref;
191     CreateInstanceFunc fnCreateInstance;
192 } ClassFactory;
193
194 static inline ClassFactory *impl_from_IClassFactory(IClassFactory *iface)
195 {
196     return CONTAINING_RECORD(iface, ClassFactory, IClassFactory_iface);
197 }
198
199 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFGUID riid, void **ppvObject)
200 {
201     if(IsEqualGUID(&IID_IClassFactory, riid) || IsEqualGUID(&IID_IUnknown, riid)) {
202         IClassFactory_AddRef(iface);
203         *ppvObject = iface;
204         return S_OK;
205     }
206
207     WARN("not supported iid %s\n", debugstr_guid(riid));
208     *ppvObject = NULL;
209     return E_NOINTERFACE;
210 }
211
212 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
213 {
214     ClassFactory *This = impl_from_IClassFactory(iface);
215     ULONG ref = InterlockedIncrement(&This->ref);
216     TRACE("(%p) ref = %u\n", This, ref);
217     return ref;
218 }
219
220 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
221 {
222     ClassFactory *This = impl_from_IClassFactory(iface);
223     ULONG ref = InterlockedDecrement(&This->ref);
224
225     TRACE("(%p) ref = %u\n", This, ref);
226
227     if(!ref) {
228         heap_free(This);
229     }
230
231     return ref;
232 }
233
234 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter,
235         REFIID riid, void **ppvObject)
236 {
237     ClassFactory *This = impl_from_IClassFactory(iface);
238     return This->fnCreateInstance(pUnkOuter, riid, ppvObject);
239 }
240
241 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
242 {
243     TRACE("(%p)->(%x)\n", iface, dolock);
244
245     /* We never unload the DLL. See DllCanUnloadNow(). */
246     return S_OK;
247 }
248
249 static const IClassFactoryVtbl HTMLClassFactoryVtbl = {
250     ClassFactory_QueryInterface,
251     ClassFactory_AddRef,
252     ClassFactory_Release,
253     ClassFactory_CreateInstance,
254     ClassFactory_LockServer
255 };
256
257 static HRESULT ClassFactory_Create(REFIID riid, void **ppv, CreateInstanceFunc fnCreateInstance)
258 {
259     ClassFactory *ret = heap_alloc(sizeof(ClassFactory));
260     HRESULT hres;
261
262     ret->IClassFactory_iface.lpVtbl = &HTMLClassFactoryVtbl;
263     ret->ref = 0;
264     ret->fnCreateInstance = fnCreateInstance;
265
266     hres = IClassFactory_QueryInterface(&ret->IClassFactory_iface, riid, ppv);
267     if(FAILED(hres)) {
268         heap_free(ret);
269         *ppv = NULL;
270     }
271     return hres;
272 }
273
274 /******************************************************************
275  *              DllGetClassObject (MSHTML.@)
276  */
277 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
278 {
279     if(IsEqualGUID(&CLSID_HTMLDocument, rclsid)) {
280         TRACE("(CLSID_HTMLDocument %s %p)\n", debugstr_guid(riid), ppv);
281         return ClassFactory_Create(riid, ppv, HTMLDocument_Create);
282     }else if(IsEqualGUID(&CLSID_AboutProtocol, rclsid)) {
283         TRACE("(CLSID_AboutProtocol %s %p)\n", debugstr_guid(riid), ppv);
284         return ProtocolFactory_Create(rclsid, riid, ppv);
285     }else if(IsEqualGUID(&CLSID_JSProtocol, rclsid)) {
286         TRACE("(CLSID_JSProtocol %s %p)\n", debugstr_guid(riid), ppv);
287         return ProtocolFactory_Create(rclsid, riid, ppv);
288     }else if(IsEqualGUID(&CLSID_MailtoProtocol, rclsid)) {
289         TRACE("(CLSID_MailtoProtocol %s %p)\n", debugstr_guid(riid), ppv);
290         return ProtocolFactory_Create(rclsid, riid, ppv);
291     }else if(IsEqualGUID(&CLSID_ResProtocol, rclsid)) {
292         TRACE("(CLSID_ResProtocol %s %p)\n", debugstr_guid(riid), ppv);
293         return ProtocolFactory_Create(rclsid, riid, ppv);
294     }else if(IsEqualGUID(&CLSID_SysimageProtocol, rclsid)) {
295         TRACE("(CLSID_SysimageProtocol %s %p)\n", debugstr_guid(riid), ppv);
296         return ProtocolFactory_Create(rclsid, riid, ppv);
297     }else if(IsEqualGUID(&CLSID_HTMLLoadOptions, rclsid)) {
298         TRACE("(CLSID_HTMLLoadOptions %s %p)\n", debugstr_guid(riid), ppv);
299         return ClassFactory_Create(riid, ppv, HTMLLoadOptions_Create);
300     }
301
302     FIXME("Unknown class %s\n", debugstr_guid(rclsid));
303     return CLASS_E_CLASSNOTAVAILABLE;
304 }
305
306 /******************************************************************
307  *              DllCanUnloadNow (MSHTML.@)
308  */
309 HRESULT WINAPI DllCanUnloadNow(void)
310 {
311     TRACE("()\n");
312     /* The cost of keeping this DLL in memory is small. */
313     return S_FALSE;
314 }
315
316 /***********************************************************************
317  *          RunHTMLApplication (MSHTML.@)
318  *
319  * Appears to have the same prototype as WinMain.
320  */
321 HRESULT WINAPI RunHTMLApplication( HINSTANCE hinst, HINSTANCE hPrevInst,
322                                LPSTR szCmdLine, INT nCmdShow )
323 {
324     FIXME("%p %p %s %d\n", hinst, hPrevInst, debugstr_a(szCmdLine), nCmdShow );
325     return 0;
326 }
327
328 /***********************************************************************
329  *          RNIGetCompatibleVersion (MSHTML.@)
330  */
331 DWORD WINAPI RNIGetCompatibleVersion(void)
332 {
333     TRACE("()\n");
334     return 0x20000;
335 }
336
337 /***********************************************************************
338  *          DllInstall (MSHTML.@)
339  */
340 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
341 {
342     FIXME("stub %d %s: returning S_OK\n", bInstall, debugstr_w(cmdline));
343     return S_OK;
344 }
345
346 /***********************************************************************
347  *          ShowHTMLDialog (MSHTML.@)
348  */
349 HRESULT WINAPI ShowHTMLDialog(HWND hwndParent, IMoniker *pMk, VARIANT *pvarArgIn,
350         WCHAR *pchOptions, VARIANT *pvarArgOut)
351 {
352     FIXME("(%p %p %p %s %p)\n", hwndParent, pMk, pvarArgIn, debugstr_w(pchOptions), pvarArgOut);
353     return E_NOTIMPL;
354 }
355
356 /***********************************************************************
357  *          PrintHTML (MSHTML.@)
358  */
359 void WINAPI PrintHTML(HWND hwnd, HINSTANCE handle, LPCSTR cmdline, INT show)
360 {
361     FIXME("(%p %p %s %x)\n", hwnd, handle, debugstr_a(cmdline), show);
362 }
363
364 DEFINE_GUID(CLSID_CBackgroundPropertyPage, 0x3050F232, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
365 DEFINE_GUID(CLSID_CCDAnchorPropertyPage, 0x3050F1FC, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
366 DEFINE_GUID(CLSID_CCDGenericPropertyPage, 0x3050F17F, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
367 DEFINE_GUID(CLSID_CDwnBindInfo, 0x3050F3C2, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
368 DEFINE_GUID(CLSID_CHiFiUses, 0x5AAF51B3, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
369 DEFINE_GUID(CLSID_CHtmlComponentConstructor, 0x3050F4F8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
370 DEFINE_GUID(CLSID_CInlineStylePropertyPage, 0x3050F296, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
371 DEFINE_GUID(CLSID_CPeerHandler, 0x5AAF51B2, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
372 DEFINE_GUID(CLSID_CRecalcEngine, 0x3050F499, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
373 DEFINE_GUID(CLSID_CSvrOMUses, 0x3050F4F0, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
374 DEFINE_GUID(CLSID_CrSource, 0x65014010, 0x9F62, 0x11D1, 0xA6,0x51, 0x00,0x60,0x08,0x11,0xD5,0xCE);
375 DEFINE_GUID(CLSID_ExternalFrameworkSite, 0x3050F163, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
376 DEFINE_GUID(CLSID_HTADocument, 0x3050F5C8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
377 DEFINE_GUID(CLSID_HTMLPluginDocument, 0x25336921, 0x03F9, 0x11CF, 0x8F,0xD0, 0x00,0xAA,0x00,0x68,0x6F,0x13);
378 DEFINE_GUID(CLSID_HTMLPopup, 0x3050F667, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
379 DEFINE_GUID(CLSID_HTMLPopupDoc, 0x3050F67D, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
380 DEFINE_GUID(CLSID_HTMLServerDoc, 0x3050F4E7, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
381 DEFINE_GUID(CLSID_IImageDecodeFilter, 0x607FD4E8, 0x0A03, 0x11D1, 0xAB,0x1D, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
382 DEFINE_GUID(CLSID_IImgCtx, 0x3050F3D6, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
383 DEFINE_GUID(CLSID_IntDitherer, 0x05F6FE1A, 0xECEF, 0x11D0, 0xAA,0xE7, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
384 DEFINE_GUID(CLSID_MHTMLDocument, 0x3050F3D9, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
385 DEFINE_GUID(CLSID_TridentAPI, 0x429AF92C, 0xA51F, 0x11D2, 0x86,0x1E, 0x00,0xC0,0x4F,0xA3,0x5C,0x89);
386
387 #define INF_SET_ID(id)            \
388     do                            \
389     {                             \
390         static CHAR name[] = #id; \
391                                   \
392         pse[i].pszName = name;    \
393         clsids[i++] = &id;        \
394     } while (0)
395
396 #define INF_SET_CLSID(clsid) INF_SET_ID(CLSID_ ## clsid)
397
398 static HRESULT register_server(BOOL do_register)
399 {
400     HRESULT hres;
401     HMODULE hAdvpack;
402     HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
403     STRTABLEA strtable;
404     STRENTRYA pse[35];
405     static CLSID const *clsids[35];
406     unsigned int i = 0;
407
408     static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
409
410     TRACE("(%x)\n", do_register);
411
412     INF_SET_CLSID(AboutProtocol);
413     INF_SET_CLSID(CAnchorBrowsePropertyPage);
414     INF_SET_CLSID(CBackgroundPropertyPage);
415     INF_SET_CLSID(CCDAnchorPropertyPage);
416     INF_SET_CLSID(CCDGenericPropertyPage);
417     INF_SET_CLSID(CDocBrowsePropertyPage);
418     INF_SET_CLSID(CDwnBindInfo);
419     INF_SET_CLSID(CHiFiUses);
420     INF_SET_CLSID(CHtmlComponentConstructor);
421     INF_SET_CLSID(CImageBrowsePropertyPage);
422     INF_SET_CLSID(CInlineStylePropertyPage);
423     INF_SET_CLSID(CPeerHandler);
424     INF_SET_CLSID(CRecalcEngine);
425     INF_SET_CLSID(CSvrOMUses);
426     INF_SET_CLSID(CrSource);
427     INF_SET_CLSID(ExternalFrameworkSite);
428     INF_SET_CLSID(HTADocument);
429     INF_SET_CLSID(HTMLDocument);
430     INF_SET_CLSID(HTMLLoadOptions);
431     INF_SET_CLSID(HTMLPluginDocument);
432     INF_SET_CLSID(HTMLPopup);
433     INF_SET_CLSID(HTMLPopupDoc);
434     INF_SET_CLSID(HTMLServerDoc);
435     INF_SET_CLSID(HTMLWindowProxy);
436     INF_SET_CLSID(IImageDecodeFilter);
437     INF_SET_CLSID(IImgCtx);
438     INF_SET_CLSID(IntDitherer);
439     INF_SET_CLSID(JSProtocol);
440     INF_SET_CLSID(MHTMLDocument);
441     INF_SET_CLSID(MailtoProtocol);
442     INF_SET_CLSID(ResProtocol);
443     INF_SET_CLSID(Scriptlet);
444     INF_SET_CLSID(SysimageProtocol);
445     INF_SET_CLSID(TridentAPI);
446     INF_SET_ID(LIBID_MSHTML);
447
448     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++) {
449         pse[i].pszValue = heap_alloc(39);
450         sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
451                 clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
452                 clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
453                 clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
454     }
455
456     strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
457     strtable.pse = pse;
458
459     hAdvpack = LoadLibraryW(wszAdvpack);
460     pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
461
462     hres = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll", &strtable);
463
464     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
465         heap_free(pse[i].pszValue);
466
467     if(FAILED(hres))
468         ERR("RegInstall failed: %08x\n", hres);
469
470     return hres;
471 }
472
473 #undef INF_SET_CLSID
474 #undef INF_SET_ID
475
476 /***********************************************************************
477  *          DllRegisterServer (MSHTML.@)
478  */
479 HRESULT WINAPI DllRegisterServer(void)
480 {
481     HRESULT hres;
482
483     hres = __wine_register_resources( hInst );
484     if(SUCCEEDED(hres))
485         hres = register_server(TRUE);
486     if(SUCCEEDED(hres))
487         load_gecko(FALSE);
488
489     return hres;
490 }
491
492 /***********************************************************************
493  *          DllUnregisterServer (MSHTML.@)
494  */
495 HRESULT WINAPI DllUnregisterServer(void)
496 {
497     HRESULT hres = __wine_unregister_resources( hInst );
498     if(SUCCEEDED(hres)) hres = register_server(FALSE);
499     return hres;
500 }
501
502 const char *debugstr_variant(const VARIANT *v)
503 {
504     if(!v)
505         return "(null)";
506
507     switch(V_VT(v)) {
508     case VT_EMPTY:
509         return "{VT_EMPTY}";
510     case VT_NULL:
511         return "{VT_NULL}";
512     case VT_I4:
513         return wine_dbg_sprintf("{VT_I4: %d}", V_I4(v));
514     case VT_R8:
515         return wine_dbg_sprintf("{VT_R8: %lf}", V_R8(v));
516     case VT_BSTR:
517         return wine_dbg_sprintf("{VT_BSTR: %s}", debugstr_w(V_BSTR(v)));
518     case VT_DISPATCH:
519         return wine_dbg_sprintf("{VT_DISPATCH: %p}", V_DISPATCH(v));
520     case VT_ERROR:
521         return wine_dbg_sprintf("{VT_ERROR: %08x}", V_ERROR(v));
522     case VT_BOOL:
523         return wine_dbg_sprintf("{VT_BOOL: %x}", V_BOOL(v));
524     case VT_UINT:
525         return wine_dbg_sprintf("{VT_UINT: %u}", V_UINT(v));
526     default:
527         return wine_dbg_sprintf("{vt %d}", V_VT(v));
528     }
529 }