shell32/tests: Fix a test failure on a default Windows 7 system.
[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 "shlguid.h"
37
38 #include "wine/unicode.h"
39 #include "wine/debug.h"
40
41 #define INIT_GUID
42 #include "mshtml_private.h"
43
44 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
45
46 HINSTANCE hInst;
47 DWORD mshtml_tls = TLS_OUT_OF_INDEXES;
48
49 static HINSTANCE shdoclc = NULL;
50 static HDC display_dc;
51
52 static void thread_detach(void)
53 {
54     thread_data_t *thread_data;
55
56     thread_data = get_thread_data(FALSE);
57     if(!thread_data)
58         return;
59
60     if(thread_data->thread_hwnd)
61         DestroyWindow(thread_data->thread_hwnd);
62
63     heap_free(thread_data);
64 }
65
66 static void process_detach(void)
67 {
68     close_gecko();
69     release_typelib();
70
71     if(shdoclc)
72         FreeLibrary(shdoclc);
73     if(mshtml_tls != TLS_OUT_OF_INDEXES)
74         TlsFree(mshtml_tls);
75     if(display_dc)
76         DeleteObject(display_dc);
77 }
78
79 HINSTANCE get_shdoclc(void)
80 {
81     static const WCHAR wszShdoclc[] =
82         {'s','h','d','o','c','l','c','.','d','l','l',0};
83
84     if(shdoclc)
85         return shdoclc;
86
87     return shdoclc = LoadLibraryExW(wszShdoclc, NULL, LOAD_LIBRARY_AS_DATAFILE);
88 }
89
90 HDC get_display_dc(void)
91 {
92     static const WCHAR displayW[] = {'D','I','S','P','L','A','Y',0};
93
94     if(!display_dc) {
95         HDC hdc;
96
97         hdc = CreateICW(displayW, NULL, NULL, NULL);
98         if(InterlockedCompareExchangePointer((void**)&display_dc, hdc, NULL))
99             DeleteObject(hdc);
100     }
101
102     return display_dc;
103 }
104
105 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
106 {
107     switch(fdwReason) {
108     case DLL_PROCESS_ATTACH:
109         hInst = hInstDLL;
110         break;
111     case DLL_PROCESS_DETACH:
112         process_detach();
113         break;
114     case DLL_THREAD_DETACH:
115         thread_detach();
116         break;
117     }
118     return TRUE;
119 }
120
121 /***********************************************************
122  *    ClassFactory implementation
123  */
124 typedef HRESULT (*CreateInstanceFunc)(IUnknown*,REFIID,void**);
125 typedef struct {
126     const IClassFactoryVtbl *lpVtbl;
127     LONG ref;
128     CreateInstanceFunc fnCreateInstance;
129 } ClassFactory;
130
131 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFGUID riid, void **ppvObject)
132 {
133     if(IsEqualGUID(&IID_IClassFactory, riid) || IsEqualGUID(&IID_IUnknown, riid)) {
134         IClassFactory_AddRef(iface);
135         *ppvObject = iface;
136         return S_OK;
137     }
138
139     WARN("not supported iid %s\n", debugstr_guid(riid));
140     *ppvObject = NULL;
141     return E_NOINTERFACE;
142 }
143
144 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
145 {
146     ClassFactory *This = (ClassFactory*)iface;
147     ULONG ref = InterlockedIncrement(&This->ref);
148     TRACE("(%p) ref = %u\n", This, ref);
149     return ref;
150 }
151
152 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
153 {
154     ClassFactory *This = (ClassFactory*)iface;
155     ULONG ref = InterlockedDecrement(&This->ref);
156
157     TRACE("(%p) ref = %u\n", This, ref);
158
159     if(!ref) {
160         heap_free(This);
161     }
162
163     return ref;
164 }
165
166 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter,
167         REFIID riid, void **ppvObject)
168 {
169     ClassFactory *This = (ClassFactory*)iface;
170     return This->fnCreateInstance(pUnkOuter, riid, ppvObject);
171 }
172
173 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
174 {
175     TRACE("(%p)->(%x)\n", iface, dolock);
176
177     /* We never unload the DLL. See DllCanUnloadNow(). */
178     return S_OK;
179 }
180
181 static const IClassFactoryVtbl HTMLClassFactoryVtbl = {
182     ClassFactory_QueryInterface,
183     ClassFactory_AddRef,
184     ClassFactory_Release,
185     ClassFactory_CreateInstance,
186     ClassFactory_LockServer
187 };
188
189 static HRESULT ClassFactory_Create(REFIID riid, void **ppv, CreateInstanceFunc fnCreateInstance)
190 {
191     ClassFactory *ret = heap_alloc(sizeof(ClassFactory));
192     HRESULT hres;
193
194     ret->lpVtbl = &HTMLClassFactoryVtbl;
195     ret->ref = 0;
196     ret->fnCreateInstance = fnCreateInstance;
197
198     hres = IClassFactory_QueryInterface((IClassFactory*)ret, riid, ppv);
199     if(FAILED(hres)) {
200         heap_free(ret);
201         *ppv = NULL;
202     }
203     return hres;
204 }
205
206 /******************************************************************
207  *              DllGetClassObject (MSHTML.@)
208  */
209 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
210 {
211     if(IsEqualGUID(&CLSID_HTMLDocument, rclsid)) {
212         TRACE("(CLSID_HTMLDocument %s %p)\n", debugstr_guid(riid), ppv);
213         return ClassFactory_Create(riid, ppv, HTMLDocument_Create);
214     }else if(IsEqualGUID(&CLSID_AboutProtocol, rclsid)) {
215         TRACE("(CLSID_AboutProtocol %s %p)\n", debugstr_guid(riid), ppv);
216         return ProtocolFactory_Create(rclsid, riid, ppv);
217     }else if(IsEqualGUID(&CLSID_JSProtocol, rclsid)) {
218         TRACE("(CLSID_JSProtocol %s %p)\n", debugstr_guid(riid), ppv);
219         return ProtocolFactory_Create(rclsid, riid, ppv);
220     }else if(IsEqualGUID(&CLSID_MailtoProtocol, rclsid)) {
221         TRACE("(CLSID_MailtoProtocol %s %p)\n", debugstr_guid(riid), ppv);
222         return ProtocolFactory_Create(rclsid, riid, ppv);
223     }else if(IsEqualGUID(&CLSID_ResProtocol, rclsid)) {
224         TRACE("(CLSID_ResProtocol %s %p)\n", debugstr_guid(riid), ppv);
225         return ProtocolFactory_Create(rclsid, riid, ppv);
226     }else if(IsEqualGUID(&CLSID_SysimageProtocol, rclsid)) {
227         TRACE("(CLSID_SysimageProtocol %s %p)\n", debugstr_guid(riid), ppv);
228         return ProtocolFactory_Create(rclsid, riid, ppv);
229     }else if(IsEqualGUID(&CLSID_HTMLLoadOptions, rclsid)) {
230         TRACE("(CLSID_HTMLLoadOptions %s %p)\n", debugstr_guid(riid), ppv);
231         return ClassFactory_Create(riid, ppv, HTMLLoadOptions_Create);
232     }
233
234     FIXME("Unknown class %s\n", debugstr_guid(rclsid));
235     return CLASS_E_CLASSNOTAVAILABLE;
236 }
237
238 /******************************************************************
239  *              DllCanUnloadNow (MSHTML.@)
240  */
241 HRESULT WINAPI DllCanUnloadNow(void)
242 {
243     TRACE("()\n");
244     /* The cost of keeping this DLL in memory is small. */
245     return S_FALSE;
246 }
247
248 /***********************************************************************
249  *          RunHTMLApplication (MSHTML.@)
250  *
251  * Appears to have the same prototype as WinMain.
252  */
253 HRESULT WINAPI RunHTMLApplication( HINSTANCE hinst, HINSTANCE hPrevInst,
254                                LPSTR szCmdLine, INT nCmdShow )
255 {
256     FIXME("%p %p %s %d\n", hinst, hPrevInst, debugstr_a(szCmdLine), nCmdShow );
257     return 0;
258 }
259
260 /***********************************************************************
261  *          RNIGetCompatibleVersion (MSHTML.@)
262  */
263 DWORD WINAPI RNIGetCompatibleVersion(void)
264 {
265     TRACE("()\n");
266     return 0x20000;
267 }
268
269 /***********************************************************************
270  *          DllInstall (MSHTML.@)
271  */
272 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
273 {
274     FIXME("stub %d %s: returning S_OK\n", bInstall, debugstr_w(cmdline));
275     return S_OK;
276 }
277
278 /***********************************************************************
279  *          ShowHTMLDialog (MSHTML.@)
280  */
281 HRESULT WINAPI ShowHTMLDialog(HWND hwndParent, IMoniker *pMk, VARIANT *pvarArgIn,
282         WCHAR *pchOptions, VARIANT *pvarArgOut)
283 {
284     FIXME("(%p %p %p %s %p)\n", hwndParent, pMk, pvarArgIn, debugstr_w(pchOptions), pvarArgOut);
285     return E_NOTIMPL;
286 }
287
288 DEFINE_GUID(CLSID_CBackgroundPropertyPage, 0x3050F232, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
289 DEFINE_GUID(CLSID_CCDAnchorPropertyPage, 0x3050F1FC, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
290 DEFINE_GUID(CLSID_CCDGenericPropertyPage, 0x3050F17F, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
291 DEFINE_GUID(CLSID_CDwnBindInfo, 0x3050F3C2, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
292 DEFINE_GUID(CLSID_CHiFiUses, 0x5AAF51B3, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
293 DEFINE_GUID(CLSID_CHtmlComponentConstructor, 0x3050F4F8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
294 DEFINE_GUID(CLSID_CInlineStylePropertyPage, 0x3050F296, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
295 DEFINE_GUID(CLSID_CPeerHandler, 0x5AAF51B2, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
296 DEFINE_GUID(CLSID_CRecalcEngine, 0x3050F499, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
297 DEFINE_GUID(CLSID_CSvrOMUses, 0x3050F4F0, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
298 DEFINE_GUID(CLSID_CrSource, 0x65014010, 0x9F62, 0x11D1, 0xA6,0x51, 0x00,0x60,0x08,0x11,0xD5,0xCE);
299 DEFINE_GUID(CLSID_ExternalFrameworkSite, 0x3050F163, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
300 DEFINE_GUID(CLSID_HTADocument, 0x3050F5C8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
301 DEFINE_GUID(CLSID_HTMLPluginDocument, 0x25336921, 0x03F9, 0x11CF, 0x8F,0xD0, 0x00,0xAA,0x00,0x68,0x6F,0x13);
302 DEFINE_GUID(CLSID_HTMLPopup, 0x3050F667, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
303 DEFINE_GUID(CLSID_HTMLPopupDoc, 0x3050F67D, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
304 DEFINE_GUID(CLSID_HTMLServerDoc, 0x3050F4E7, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
305 DEFINE_GUID(CLSID_IImageDecodeFilter, 0x607FD4E8, 0x0A03, 0x11D1, 0xAB,0x1D, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
306 DEFINE_GUID(CLSID_IImgCtx, 0x3050F3D6, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
307 DEFINE_GUID(CLSID_IntDitherer, 0x05F6FE1A, 0xECEF, 0x11D0, 0xAA,0xE7, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
308 DEFINE_GUID(CLSID_MHTMLDocument, 0x3050F3D9, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
309 DEFINE_GUID(CLSID_Scriptlet, 0xAE24FDAE, 0x03C6, 0x11D1, 0x8B,0x76, 0x00,0x80,0xC7,0x44,0xF3,0x89);
310 DEFINE_GUID(CLSID_TridentAPI, 0x429AF92C, 0xA51F, 0x11D2, 0x86,0x1E, 0x00,0xC0,0x4F,0xA3,0x5C,0x89);
311
312 #define INF_SET_ID(id)            \
313     do                            \
314     {                             \
315         static CHAR name[] = #id; \
316                                   \
317         pse[i].pszName = name;    \
318         clsids[i++] = &id;        \
319     } while (0)
320
321 #define INF_SET_CLSID(clsid) INF_SET_ID(CLSID_ ## clsid)
322
323 static HRESULT register_server(BOOL do_register)
324 {
325     HRESULT hres;
326     HMODULE hAdvpack;
327     HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
328     STRTABLEA strtable;
329     STRENTRYA pse[35];
330     static CLSID const *clsids[35];
331     unsigned int i = 0;
332
333     static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
334
335     TRACE("(%x)\n", do_register);
336
337     INF_SET_CLSID(AboutProtocol);
338     INF_SET_CLSID(CAnchorBrowsePropertyPage);
339     INF_SET_CLSID(CBackgroundPropertyPage);
340     INF_SET_CLSID(CCDAnchorPropertyPage);
341     INF_SET_CLSID(CCDGenericPropertyPage);
342     INF_SET_CLSID(CDocBrowsePropertyPage);
343     INF_SET_CLSID(CDwnBindInfo);
344     INF_SET_CLSID(CHiFiUses);
345     INF_SET_CLSID(CHtmlComponentConstructor);
346     INF_SET_CLSID(CImageBrowsePropertyPage);
347     INF_SET_CLSID(CInlineStylePropertyPage);
348     INF_SET_CLSID(CPeerHandler);
349     INF_SET_CLSID(CRecalcEngine);
350     INF_SET_CLSID(CSvrOMUses);
351     INF_SET_CLSID(CrSource);
352     INF_SET_CLSID(ExternalFrameworkSite);
353     INF_SET_CLSID(HTADocument);
354     INF_SET_CLSID(HTMLDocument);
355     INF_SET_CLSID(HTMLLoadOptions);
356     INF_SET_CLSID(HTMLPluginDocument);
357     INF_SET_CLSID(HTMLPopup);
358     INF_SET_CLSID(HTMLPopupDoc);
359     INF_SET_CLSID(HTMLServerDoc);
360     INF_SET_CLSID(HTMLWindowProxy);
361     INF_SET_CLSID(IImageDecodeFilter);
362     INF_SET_CLSID(IImgCtx);
363     INF_SET_CLSID(IntDitherer);
364     INF_SET_CLSID(JSProtocol);
365     INF_SET_CLSID(MHTMLDocument);
366     INF_SET_CLSID(MailtoProtocol);
367     INF_SET_CLSID(ResProtocol);
368     INF_SET_CLSID(Scriptlet);
369     INF_SET_CLSID(SysimageProtocol);
370     INF_SET_CLSID(TridentAPI);
371     INF_SET_ID(LIBID_MSHTML);
372
373     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++) {
374         pse[i].pszValue = heap_alloc(39);
375         sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
376                 clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
377                 clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
378                 clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
379     }
380
381     strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
382     strtable.pse = pse;
383
384     hAdvpack = LoadLibraryW(wszAdvpack);
385     pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
386
387     hres = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll", &strtable);
388
389     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
390         heap_free(pse[i].pszValue);
391
392     if(FAILED(hres)) {
393         ERR("RegInstall failed: %08x\n", hres);
394         return hres;
395     }
396
397     if(do_register) {
398         ITypeLib *typelib;
399
400         static const WCHAR wszMSHTML[] = {'m','s','h','t','m','l','.','t','l','b',0};
401
402         hres = LoadTypeLibEx(wszMSHTML, REGKIND_REGISTER, &typelib);
403         if(SUCCEEDED(hres))
404             ITypeLib_Release(typelib);
405     }else {
406         hres = UnRegisterTypeLib(&LIBID_MSHTML, 4, 0, LOCALE_SYSTEM_DEFAULT, SYS_WIN32);
407     }
408
409     if(FAILED(hres))
410         ERR("typelib registration failed: %08x\n", hres);
411
412     return hres;
413 }
414
415 #undef INF_SET_CLSID
416 #undef INF_SET_ID
417
418 /***********************************************************************
419  *          DllRegisterServer (MSHTML.@)
420  */
421 HRESULT WINAPI DllRegisterServer(void)
422 {
423     HRESULT hres;
424
425     hres = register_server(TRUE);
426     if(SUCCEEDED(hres))
427         load_gecko(FALSE);
428
429     return hres;
430 }
431
432 /***********************************************************************
433  *          DllUnregisterServer (MSHTML.@)
434  */
435 HRESULT WINAPI DllUnregisterServer(void)
436 {
437     return register_server(FALSE);
438 }
439
440 const char *debugstr_variant(const VARIANT *v)
441 {
442     switch(V_VT(v)) {
443     case VT_EMPTY:
444         return wine_dbg_sprintf("{VT_EMPTY}");
445     case VT_NULL:
446         return wine_dbg_sprintf("{VT_NULL}");
447     case VT_I4:
448         return wine_dbg_sprintf("{VT_I4: %d}", V_I4(v));
449     case VT_R8:
450         return wine_dbg_sprintf("{VT_R8: %lf}", V_R8(v));
451     case VT_BSTR:
452         return wine_dbg_sprintf("{VT_BSTR: %s}", debugstr_w(V_BSTR(v)));
453     case VT_DISPATCH:
454         return wine_dbg_sprintf("{VT_DISPATCH: %p}", V_DISPATCH(v));
455     case VT_BOOL:
456         return wine_dbg_sprintf("{VT_BOOL: %x}", V_BOOL(v));
457     case VT_UINT:
458         return wine_dbg_sprintf("{VT_UINT: %u}", V_UINT(v));
459     default:
460         return wine_dbg_sprintf("{vt %d}", V_VT(v));
461     }
462 }