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