kernel32/tests: Prevent memory leak.
[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 "config.h"
24
25 #include <stdarg.h>
26 #include <stdio.h>
27
28 #define COBJMACROS
29
30 #include "windef.h"
31 #include "winbase.h"
32 #include "winuser.h"
33 #include "winnls.h"
34 #include "winreg.h"
35 #include "ole2.h"
36 #include "advpub.h"
37 #include "shlwapi.h"
38 #include "optary.h"
39
40 #include "wine/unicode.h"
41 #include "wine/debug.h"
42
43 #define INIT_GUID
44 #include "mshtml_private.h"
45
46 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
47
48 HINSTANCE hInst;
49 LONG module_ref = 0;
50 DWORD mshtml_tls = 0;
51
52 static HINSTANCE shdoclc = NULL;
53
54 static void thread_detach(void)
55 {
56     thread_data_t *thread_data = get_thread_data(FALSE);
57
58     if(!thread_data)
59         return;
60
61     if(thread_data->thread_hwnd)
62         DestroyWindow(thread_data->thread_hwnd);
63
64     mshtml_free(thread_data);
65 }
66
67 HINSTANCE get_shdoclc(void)
68 {
69     static const WCHAR wszShdoclc[] =
70         {'s','h','d','o','c','l','c','.','d','l','l',0};
71
72     if(shdoclc)
73         return shdoclc;
74
75     return shdoclc = LoadLibraryExW(wszShdoclc, NULL, LOAD_LIBRARY_AS_DATAFILE);
76 }
77
78 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
79 {
80     switch(fdwReason) {
81     case DLL_PROCESS_ATTACH:
82         hInst = hInstDLL;
83         break;
84     case DLL_PROCESS_DETACH:
85         close_gecko();
86         if(shdoclc)
87             FreeLibrary(shdoclc);
88         if(mshtml_tls)
89             TlsFree(mshtml_tls);
90         break;
91     case DLL_THREAD_DETACH:
92         thread_detach();
93         break;
94     }
95     return TRUE;
96 }
97
98 /***********************************************************
99  *    ClassFactory implementation
100  */
101 typedef HRESULT (*CreateInstanceFunc)(IUnknown*,REFIID,void**);
102 typedef struct {
103     const IClassFactoryVtbl *lpVtbl;
104     LONG ref;
105     CreateInstanceFunc fnCreateInstance;
106 } ClassFactory;
107
108 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFGUID riid, void **ppvObject)
109 {
110     if(IsEqualGUID(&IID_IClassFactory, riid) || IsEqualGUID(&IID_IUnknown, riid)) {
111         IClassFactory_AddRef(iface);
112         *ppvObject = iface;
113         return S_OK;
114     }
115
116     WARN("not supported iid %s\n", debugstr_guid(riid));
117     *ppvObject = NULL;
118     return E_NOINTERFACE;
119 }
120
121 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
122 {
123     ClassFactory *This = (ClassFactory*)iface;
124     ULONG ref = InterlockedIncrement(&This->ref);
125     TRACE("(%p) ref = %u\n", This, ref);
126     return ref;
127 }
128
129 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
130 {
131     ClassFactory *This = (ClassFactory*)iface;
132     ULONG ref = InterlockedDecrement(&This->ref);
133
134     TRACE("(%p) ref = %u\n", This, ref);
135
136     if(!ref) {
137         mshtml_free(This);
138         UNLOCK_MODULE();
139     }
140
141     return ref;
142 }
143
144 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter,
145         REFIID riid, void **ppvObject)
146 {
147     ClassFactory *This = (ClassFactory*)iface;
148     return This->fnCreateInstance(pUnkOuter, riid, ppvObject);
149 }
150
151 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
152 {
153     TRACE("(%p)->(%x)\n", iface, dolock);
154
155     if(dolock)
156         LOCK_MODULE();
157     else
158         UNLOCK_MODULE();
159
160     return S_OK;
161 }
162
163 static const IClassFactoryVtbl HTMLClassFactoryVtbl = {
164     ClassFactory_QueryInterface,
165     ClassFactory_AddRef,
166     ClassFactory_Release,
167     ClassFactory_CreateInstance,
168     ClassFactory_LockServer
169 };
170
171 static HRESULT ClassFactory_Create(REFIID riid, void **ppv, CreateInstanceFunc fnCreateInstance)
172 {
173     ClassFactory *ret = mshtml_alloc(sizeof(ClassFactory));
174     HRESULT hres;
175
176     ret->lpVtbl = &HTMLClassFactoryVtbl;
177     ret->ref = 0;
178     ret->fnCreateInstance = fnCreateInstance;
179
180     hres = IClassFactory_QueryInterface((IClassFactory*)ret, riid, ppv);
181     if(SUCCEEDED(hres)) {
182         LOCK_MODULE();
183     }else {
184         mshtml_free(ret);
185         *ppv = NULL;
186     }
187     return hres;
188 }
189
190 /******************************************************************
191  *              DllGetClassObject (MSHTML.@)
192  */
193 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
194 {
195     if(IsEqualGUID(&CLSID_HTMLDocument, rclsid)) {
196         TRACE("(CLSID_HTMLDocument %s %p)\n", debugstr_guid(riid), ppv);
197         return ClassFactory_Create(riid, ppv, HTMLDocument_Create);
198     }else if(IsEqualGUID(&CLSID_AboutProtocol, rclsid)) {
199         TRACE("(CLSID_AboutProtocol %s %p)\n", debugstr_guid(riid), ppv);
200         return ProtocolFactory_Create(rclsid, riid, ppv);
201     }else if(IsEqualGUID(&CLSID_JSProtocol, rclsid)) {
202         TRACE("(CLSID_JSProtocol %s %p)\n", debugstr_guid(riid), ppv);
203         return ProtocolFactory_Create(rclsid, riid, ppv);
204     }else if(IsEqualGUID(&CLSID_MailtoProtocol, rclsid)) {
205         TRACE("(CLSID_MailtoProtocol %s %p)\n", debugstr_guid(riid), ppv);
206         return ProtocolFactory_Create(rclsid, riid, ppv);
207     }else if(IsEqualGUID(&CLSID_ResProtocol, rclsid)) {
208         TRACE("(CLSID_ResProtocol %s %p)\n", debugstr_guid(riid), ppv);
209         return ProtocolFactory_Create(rclsid, riid, ppv);
210     }else if(IsEqualGUID(&CLSID_SysimageProtocol, rclsid)) {
211         TRACE("(CLSID_SysimageProtocol %s %p)\n", debugstr_guid(riid), ppv);
212         return ProtocolFactory_Create(rclsid, riid, ppv);
213     }else if(IsEqualGUID(&CLSID_HTMLLoadOptions, rclsid)) {
214         TRACE("(CLSID_HTMLLoadOptions %s %p)\n", debugstr_guid(riid), ppv);
215         return ClassFactory_Create(riid, ppv, HTMLLoadOptions_Create);
216     }
217
218     FIXME("Unknown class %s\n", debugstr_guid(rclsid));
219     return CLASS_E_CLASSNOTAVAILABLE;
220 }
221
222 /******************************************************************
223  *              DllCanUnloadNow (MSHTML.@)
224  */
225 HRESULT WINAPI DllCanUnloadNow(void)
226 {
227     TRACE("() ref=%d\n", module_ref);
228     return module_ref ? S_FALSE : S_OK;
229 }
230
231 /***********************************************************************
232  *          RunHTMLApplication (MSHTML.@)
233  *
234  * Appears to have the same prototype as WinMain.
235  */
236 HRESULT WINAPI RunHTMLApplication( HINSTANCE hinst, HINSTANCE hPrevInst,
237                                LPSTR szCmdLine, INT nCmdShow )
238 {
239     FIXME("%p %p %s %d\n", hinst, hPrevInst, debugstr_a(szCmdLine), nCmdShow );
240     return 0;
241 }
242
243 /***********************************************************************
244  *          RNIGetCompatibleVersion (MSHTML.@)
245  */
246 DWORD WINAPI RNIGetCompatibleVersion(void)
247 {
248     TRACE("()\n");
249     return 0x20000;
250 }
251
252 /***********************************************************************
253  *          DllInstall (MSHTML.@)
254  */
255 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
256 {
257     FIXME("stub %d %s: returning S_OK\n", bInstall, debugstr_w(cmdline));
258     return S_OK;
259 }
260
261 /***********************************************************************
262  *          ShowHTMLDialog (MSHTML.@)
263  */
264 HRESULT WINAPI ShowHTMLDialog(HWND hwndParent, IMoniker *pMk, VARIANT *pvarArgIn,
265         WCHAR *pchOptions, VARIANT *pvarArgOut)
266 {
267     FIXME("(%p %p %p %s %p)\n", hwndParent, pMk, pvarArgIn, debugstr_w(pchOptions), pvarArgOut);
268     return E_NOTIMPL;
269 }
270
271 DEFINE_GUID(CLSID_CAnchorBrowsePropertyPage, 0x3050F3BB, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
272 DEFINE_GUID(CLSID_CBackgroundPropertyPage, 0x3050F232, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
273 DEFINE_GUID(CLSID_CCDAnchorPropertyPage, 0x3050F1FC, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
274 DEFINE_GUID(CLSID_CCDGenericPropertyPage, 0x3050F17F, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
275 DEFINE_GUID(CLSID_CDocBrowsePropertyPage, 0x3050F3B4, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
276 DEFINE_GUID(CLSID_CDwnBindInfo, 0x3050F3C2, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
277 DEFINE_GUID(CLSID_CHiFiUses, 0x5AAF51B3, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
278 DEFINE_GUID(CLSID_CHtmlComponentConstructor, 0x3050F4F8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
279 DEFINE_GUID(CLSID_CImageBrowsePropertyPage, 0x3050F3B3, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
280 DEFINE_GUID(CLSID_CInlineStylePropertyPage, 0x3050F296, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
281 DEFINE_GUID(CLSID_CPeerHandler, 0x5AAF51B2, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
282 DEFINE_GUID(CLSID_CRecalcEngine, 0x3050F499, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
283 DEFINE_GUID(CLSID_CSvrOMUses, 0x3050F4F0, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
284 DEFINE_GUID(CLSID_CrSource, 0x65014010, 0x9F62, 0x11D1, 0xA6,0x51, 0x00,0x60,0x08,0x11,0xD5,0xCE);
285 DEFINE_GUID(CLSID_ExternalFrameworkSite, 0x3050F163, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
286 DEFINE_GUID(CLSID_HTADocument, 0x3050F5C8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
287 DEFINE_GUID(CLSID_HTMLPluginDocument, 0x25336921, 0x03F9, 0x11CF, 0x8F,0xD0, 0x00,0xAA,0x00,0x68,0x6F,0x13);
288 DEFINE_GUID(CLSID_HTMLPopup, 0x3050F667, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
289 DEFINE_GUID(CLSID_HTMLPopupDoc, 0x3050F67D, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
290 DEFINE_GUID(CLSID_HTMLServerDoc, 0x3050F4E7, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
291 DEFINE_GUID(CLSID_HTMLWindowProxy, 0x3050F391, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
292 DEFINE_GUID(CLSID_IImageDecodeFilter, 0x607FD4E8, 0x0A03, 0x11D1, 0xAB,0x1D, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
293 DEFINE_GUID(CLSID_IImgCtx, 0x3050F3D6, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
294 DEFINE_GUID(CLSID_IntDitherer, 0x05F6FE1A, 0xECEF, 0x11D0, 0xAA,0xE7, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
295 DEFINE_GUID(CLSID_MHTMLDocument, 0x3050F3D9, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
296 DEFINE_GUID(CLSID_Scriptlet, 0xAE24FDAE, 0x03C6, 0x11D1, 0x8B,0x76, 0x00,0x80,0xC7,0x44,0xF3,0x89);
297 DEFINE_GUID(CLSID_TridentAPI, 0x429AF92C, 0xA51F, 0x11D2, 0x86,0x1E, 0x00,0xC0,0x4F,0xA3,0x5C,0x89);
298
299 #define INF_SET_ID(id)            \
300     do                            \
301     {                             \
302         static CHAR name[] = #id; \
303                                   \
304         pse[i].pszName = name;    \
305         clsids[i++] = &id;        \
306     } while (0)
307
308 #define INF_SET_CLSID(clsid) INF_SET_ID(CLSID_ ## clsid)
309
310 static HRESULT register_server(BOOL do_register)
311 {
312     HRESULT hres;
313     HMODULE hAdvpack;
314     typeof(RegInstallA) *pRegInstall;
315     STRTABLEA strtable;
316     STRENTRYA pse[35];
317     static CLSID const *clsids[35];
318     int i = 0;
319
320     static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
321
322     TRACE("(%x)\n", do_register);
323
324     INF_SET_CLSID(AboutProtocol);
325     INF_SET_CLSID(CAnchorBrowsePropertyPage);
326     INF_SET_CLSID(CBackgroundPropertyPage);
327     INF_SET_CLSID(CCDAnchorPropertyPage);
328     INF_SET_CLSID(CCDGenericPropertyPage);
329     INF_SET_CLSID(CDocBrowsePropertyPage);
330     INF_SET_CLSID(CDwnBindInfo);
331     INF_SET_CLSID(CHiFiUses);
332     INF_SET_CLSID(CHtmlComponentConstructor);
333     INF_SET_CLSID(CImageBrowsePropertyPage);
334     INF_SET_CLSID(CInlineStylePropertyPage);
335     INF_SET_CLSID(CPeerHandler);
336     INF_SET_CLSID(CRecalcEngine);
337     INF_SET_CLSID(CSvrOMUses);
338     INF_SET_CLSID(CrSource);
339     INF_SET_CLSID(ExternalFrameworkSite);
340     INF_SET_CLSID(HTADocument);
341     INF_SET_CLSID(HTMLDocument);
342     INF_SET_CLSID(HTMLLoadOptions);
343     INF_SET_CLSID(HTMLPluginDocument);
344     INF_SET_CLSID(HTMLPopup);
345     INF_SET_CLSID(HTMLPopupDoc);
346     INF_SET_CLSID(HTMLServerDoc);
347     INF_SET_CLSID(HTMLWindowProxy);
348     INF_SET_CLSID(IImageDecodeFilter);
349     INF_SET_CLSID(IImgCtx);
350     INF_SET_CLSID(IntDitherer);
351     INF_SET_CLSID(JSProtocol);
352     INF_SET_CLSID(MHTMLDocument);
353     INF_SET_CLSID(MailtoProtocol);
354     INF_SET_CLSID(ResProtocol);
355     INF_SET_CLSID(Scriptlet);
356     INF_SET_CLSID(SysimageProtocol);
357     INF_SET_CLSID(TridentAPI);
358     INF_SET_ID(LIBID_MSHTML);
359
360     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++) {
361         pse[i].pszValue = mshtml_alloc(39);
362         sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
363                 clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
364                 clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
365                 clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
366     }
367
368     strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
369     strtable.pse = pse;
370
371     hAdvpack = LoadLibraryW(wszAdvpack);
372     pRegInstall = (typeof(RegInstallA)*)GetProcAddress(hAdvpack, "RegInstall");
373
374     hres = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll", &strtable);
375
376     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
377         mshtml_free(pse[i].pszValue);
378
379     return hres;
380 }
381
382 #undef INF_SET_CLSID
383 #undef INF_SET_ID
384
385 /***********************************************************************
386  *          DllRegisterServer (MSHTML.@)
387  */
388 HRESULT WINAPI DllRegisterServer(void)
389 {
390     return register_server(TRUE);
391 }
392
393 /***********************************************************************
394  *          DllUnregisterServer (MSHTML.@)
395  */
396 HRESULT WINAPI DllUnregisterServer(void)
397 {
398     return register_server(FALSE);
399 }