setupapi: Fix buffer overflow in load_fake_dll.
[wine.git] / dlls / mshtml / main.c
blob8468f5c675999f0e9d5e39c3d0dd870d2ba7a7ee
1 /*
2 * MSHTML Class Factory
4 * Copyright 2002 Lionel Ulmer
5 * Copyright 2003 Mike McCormack
6 * Copyright 2005 Jacek Caban
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.
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.
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
23 #include <stdarg.h>
24 #include <stdio.h>
26 #define COBJMACROS
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"
39 #include "wine/debug.h"
41 #define INIT_GUID
42 #include "mshtml_private.h"
43 #include "resource.h"
45 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
47 HINSTANCE hInst;
48 DWORD mshtml_tls = TLS_OUT_OF_INDEXES;
50 static HINSTANCE shdoclc = NULL;
51 static HDC display_dc;
52 static WCHAR *status_strings[IDS_STATUS_LAST-IDS_STATUS_FIRST+1];
54 static void thread_detach(void)
56 thread_data_t *thread_data;
58 thread_data = get_thread_data(FALSE);
59 if(!thread_data)
60 return;
62 if(thread_data->thread_hwnd)
63 DestroyWindow(thread_data->thread_hwnd);
65 heap_free(thread_data);
68 static void free_strings(void)
70 int i;
71 for(i = 0; i < sizeof(status_strings)/sizeof(*status_strings); i++)
72 heap_free(status_strings[i]);
75 static void process_detach(void)
77 close_gecko();
78 release_typelib();
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);
87 free_strings();
90 void set_statustext(HTMLDocumentObj* doc, INT id, LPCWSTR arg)
92 int index = id - IDS_STATUS_FIRST;
93 WCHAR *p = status_strings[index];
94 DWORD len;
96 if(!doc->frame)
97 return;
99 if(!p) {
100 len = 255;
101 p = heap_alloc(len * sizeof(WCHAR));
102 len = LoadStringW(hInst, id, p, len) + 1;
103 p = heap_realloc(p, len * sizeof(WCHAR));
104 if(InterlockedCompareExchangePointer((void**)&status_strings[index], p, NULL)) {
105 heap_free(p);
106 p = status_strings[index];
110 if(arg) {
111 WCHAR *buf;
113 len = lstrlenW(p) + lstrlenW(arg) - 1;
114 buf = heap_alloc(len * sizeof(WCHAR));
116 snprintfW(buf, len, p, arg);
118 p = buf;
121 IOleInPlaceFrame_SetStatusText(doc->frame, p);
123 if(arg)
124 heap_free(p);
127 HRESULT do_query_service(IUnknown *unk, REFGUID guid_service, REFIID riid, void **ppv)
129 IServiceProvider *sp;
130 HRESULT hres;
132 hres = IUnknown_QueryInterface(unk, &IID_IServiceProvider, (void**)&sp);
133 if(FAILED(hres))
134 return hres;
136 hres = IServiceProvider_QueryService(sp, guid_service, riid, ppv);
137 IServiceProvider_Release(sp);
138 return hres;
141 HINSTANCE get_shdoclc(void)
143 static const WCHAR wszShdoclc[] =
144 {'s','h','d','o','c','l','c','.','d','l','l',0};
146 if(shdoclc)
147 return shdoclc;
149 return shdoclc = LoadLibraryExW(wszShdoclc, NULL, LOAD_LIBRARY_AS_DATAFILE);
152 HDC get_display_dc(void)
154 static const WCHAR displayW[] = {'D','I','S','P','L','A','Y',0};
156 if(!display_dc) {
157 HDC hdc;
159 hdc = CreateICW(displayW, NULL, NULL, NULL);
160 if(InterlockedCompareExchangePointer((void**)&display_dc, hdc, NULL))
161 DeleteObject(hdc);
164 return display_dc;
167 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
169 switch(fdwReason) {
170 case DLL_PROCESS_ATTACH:
171 hInst = hInstDLL;
172 break;
173 case DLL_PROCESS_DETACH:
174 process_detach();
175 break;
176 case DLL_THREAD_DETACH:
177 thread_detach();
178 break;
180 return TRUE;
183 /***********************************************************
184 * ClassFactory implementation
186 typedef HRESULT (*CreateInstanceFunc)(IUnknown*,REFIID,void**);
187 typedef struct {
188 IClassFactory IClassFactory_iface;
189 LONG ref;
190 CreateInstanceFunc fnCreateInstance;
191 } ClassFactory;
193 static inline ClassFactory *impl_from_IClassFactory(IClassFactory *iface)
195 return CONTAINING_RECORD(iface, ClassFactory, IClassFactory_iface);
198 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFGUID riid, void **ppvObject)
200 if(IsEqualGUID(&IID_IClassFactory, riid) || IsEqualGUID(&IID_IUnknown, riid)) {
201 IClassFactory_AddRef(iface);
202 *ppvObject = iface;
203 return S_OK;
206 WARN("not supported iid %s\n", debugstr_guid(riid));
207 *ppvObject = NULL;
208 return E_NOINTERFACE;
211 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
213 ClassFactory *This = impl_from_IClassFactory(iface);
214 ULONG ref = InterlockedIncrement(&This->ref);
215 TRACE("(%p) ref = %u\n", This, ref);
216 return ref;
219 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
221 ClassFactory *This = impl_from_IClassFactory(iface);
222 ULONG ref = InterlockedDecrement(&This->ref);
224 TRACE("(%p) ref = %u\n", This, ref);
226 if(!ref) {
227 heap_free(This);
230 return ref;
233 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter,
234 REFIID riid, void **ppvObject)
236 ClassFactory *This = impl_from_IClassFactory(iface);
237 return This->fnCreateInstance(pUnkOuter, riid, ppvObject);
240 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
242 TRACE("(%p)->(%x)\n", iface, dolock);
244 /* We never unload the DLL. See DllCanUnloadNow(). */
245 return S_OK;
248 static const IClassFactoryVtbl HTMLClassFactoryVtbl = {
249 ClassFactory_QueryInterface,
250 ClassFactory_AddRef,
251 ClassFactory_Release,
252 ClassFactory_CreateInstance,
253 ClassFactory_LockServer
256 static HRESULT ClassFactory_Create(REFIID riid, void **ppv, CreateInstanceFunc fnCreateInstance)
258 ClassFactory *ret = heap_alloc(sizeof(ClassFactory));
259 HRESULT hres;
261 ret->IClassFactory_iface.lpVtbl = &HTMLClassFactoryVtbl;
262 ret->ref = 0;
263 ret->fnCreateInstance = fnCreateInstance;
265 hres = IClassFactory_QueryInterface(&ret->IClassFactory_iface, riid, ppv);
266 if(FAILED(hres)) {
267 heap_free(ret);
268 *ppv = NULL;
270 return hres;
273 /******************************************************************
274 * DllGetClassObject (MSHTML.@)
276 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
278 if(IsEqualGUID(&CLSID_HTMLDocument, rclsid)) {
279 TRACE("(CLSID_HTMLDocument %s %p)\n", debugstr_guid(riid), ppv);
280 return ClassFactory_Create(riid, ppv, HTMLDocument_Create);
281 }else if(IsEqualGUID(&CLSID_AboutProtocol, rclsid)) {
282 TRACE("(CLSID_AboutProtocol %s %p)\n", debugstr_guid(riid), ppv);
283 return ProtocolFactory_Create(rclsid, riid, ppv);
284 }else if(IsEqualGUID(&CLSID_JSProtocol, rclsid)) {
285 TRACE("(CLSID_JSProtocol %s %p)\n", debugstr_guid(riid), ppv);
286 return ProtocolFactory_Create(rclsid, riid, ppv);
287 }else if(IsEqualGUID(&CLSID_MailtoProtocol, rclsid)) {
288 TRACE("(CLSID_MailtoProtocol %s %p)\n", debugstr_guid(riid), ppv);
289 return ProtocolFactory_Create(rclsid, riid, ppv);
290 }else if(IsEqualGUID(&CLSID_ResProtocol, rclsid)) {
291 TRACE("(CLSID_ResProtocol %s %p)\n", debugstr_guid(riid), ppv);
292 return ProtocolFactory_Create(rclsid, riid, ppv);
293 }else if(IsEqualGUID(&CLSID_SysimageProtocol, rclsid)) {
294 TRACE("(CLSID_SysimageProtocol %s %p)\n", debugstr_guid(riid), ppv);
295 return ProtocolFactory_Create(rclsid, riid, ppv);
296 }else if(IsEqualGUID(&CLSID_HTMLLoadOptions, rclsid)) {
297 TRACE("(CLSID_HTMLLoadOptions %s %p)\n", debugstr_guid(riid), ppv);
298 return ClassFactory_Create(riid, ppv, HTMLLoadOptions_Create);
301 FIXME("Unknown class %s\n", debugstr_guid(rclsid));
302 return CLASS_E_CLASSNOTAVAILABLE;
305 /******************************************************************
306 * DllCanUnloadNow (MSHTML.@)
308 HRESULT WINAPI DllCanUnloadNow(void)
310 TRACE("()\n");
311 /* The cost of keeping this DLL in memory is small. */
312 return S_FALSE;
315 /***********************************************************************
316 * RunHTMLApplication (MSHTML.@)
318 * Appears to have the same prototype as WinMain.
320 HRESULT WINAPI RunHTMLApplication( HINSTANCE hinst, HINSTANCE hPrevInst,
321 LPSTR szCmdLine, INT nCmdShow )
323 FIXME("%p %p %s %d\n", hinst, hPrevInst, debugstr_a(szCmdLine), nCmdShow );
324 return 0;
327 /***********************************************************************
328 * RNIGetCompatibleVersion (MSHTML.@)
330 DWORD WINAPI RNIGetCompatibleVersion(void)
332 TRACE("()\n");
333 return 0x20000;
336 /***********************************************************************
337 * DllInstall (MSHTML.@)
339 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
341 FIXME("stub %d %s: returning S_OK\n", bInstall, debugstr_w(cmdline));
342 return S_OK;
345 /***********************************************************************
346 * ShowHTMLDialog (MSHTML.@)
348 HRESULT WINAPI ShowHTMLDialog(HWND hwndParent, IMoniker *pMk, VARIANT *pvarArgIn,
349 WCHAR *pchOptions, VARIANT *pvarArgOut)
351 FIXME("(%p %p %p %s %p)\n", hwndParent, pMk, pvarArgIn, debugstr_w(pchOptions), pvarArgOut);
352 return E_NOTIMPL;
355 /***********************************************************************
356 * PrintHTML (MSHTML.@)
358 void WINAPI PrintHTML(HWND hwnd, HINSTANCE handle, LPCSTR cmdline, INT show)
360 FIXME("(%p %p %s %x)\n", hwnd, handle, debugstr_a(cmdline), show);
363 DEFINE_GUID(CLSID_CBackgroundPropertyPage, 0x3050F232, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
364 DEFINE_GUID(CLSID_CCDAnchorPropertyPage, 0x3050F1FC, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
365 DEFINE_GUID(CLSID_CCDGenericPropertyPage, 0x3050F17F, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
366 DEFINE_GUID(CLSID_CDwnBindInfo, 0x3050F3C2, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
367 DEFINE_GUID(CLSID_CHiFiUses, 0x5AAF51B3, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
368 DEFINE_GUID(CLSID_CHtmlComponentConstructor, 0x3050F4F8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
369 DEFINE_GUID(CLSID_CInlineStylePropertyPage, 0x3050F296, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
370 DEFINE_GUID(CLSID_CPeerHandler, 0x5AAF51B2, 0xB1F0, 0x11D1, 0xB6,0xAB, 0x00,0xA0,0xC9,0x08,0x33,0xE9);
371 DEFINE_GUID(CLSID_CRecalcEngine, 0x3050F499, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
372 DEFINE_GUID(CLSID_CSvrOMUses, 0x3050F4F0, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
373 DEFINE_GUID(CLSID_CrSource, 0x65014010, 0x9F62, 0x11D1, 0xA6,0x51, 0x00,0x60,0x08,0x11,0xD5,0xCE);
374 DEFINE_GUID(CLSID_ExternalFrameworkSite, 0x3050F163, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
375 DEFINE_GUID(CLSID_HTADocument, 0x3050F5C8, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
376 DEFINE_GUID(CLSID_HTMLPluginDocument, 0x25336921, 0x03F9, 0x11CF, 0x8F,0xD0, 0x00,0xAA,0x00,0x68,0x6F,0x13);
377 DEFINE_GUID(CLSID_HTMLPopup, 0x3050F667, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
378 DEFINE_GUID(CLSID_HTMLPopupDoc, 0x3050F67D, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
379 DEFINE_GUID(CLSID_HTMLServerDoc, 0x3050F4E7, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
380 DEFINE_GUID(CLSID_IImageDecodeFilter, 0x607FD4E8, 0x0A03, 0x11D1, 0xAB,0x1D, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
381 DEFINE_GUID(CLSID_IImgCtx, 0x3050F3D6, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
382 DEFINE_GUID(CLSID_IntDitherer, 0x05F6FE1A, 0xECEF, 0x11D0, 0xAA,0xE7, 0x00,0xC0,0x4F,0xC9,0xB3,0x04);
383 DEFINE_GUID(CLSID_MHTMLDocument, 0x3050F3D9, 0x98B5, 0x11CF, 0xBB,0x82, 0x00,0xAA,0x00,0xBD,0xCE,0x0B);
384 DEFINE_GUID(CLSID_TridentAPI, 0x429AF92C, 0xA51F, 0x11D2, 0x86,0x1E, 0x00,0xC0,0x4F,0xA3,0x5C,0x89);
386 #define INF_SET_ID(id) \
387 do \
389 static CHAR name[] = #id; \
391 pse[i].pszName = name; \
392 clsids[i++] = &id; \
393 } while (0)
395 #define INF_SET_CLSID(clsid) INF_SET_ID(CLSID_ ## clsid)
397 static HRESULT register_server(BOOL do_register)
399 HRESULT hres;
400 HMODULE hAdvpack;
401 HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
402 STRTABLEA strtable;
403 STRENTRYA pse[35];
404 static CLSID const *clsids[35];
405 unsigned int i = 0;
407 static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
409 TRACE("(%x)\n", do_register);
411 INF_SET_CLSID(AboutProtocol);
412 INF_SET_CLSID(CAnchorBrowsePropertyPage);
413 INF_SET_CLSID(CBackgroundPropertyPage);
414 INF_SET_CLSID(CCDAnchorPropertyPage);
415 INF_SET_CLSID(CCDGenericPropertyPage);
416 INF_SET_CLSID(CDocBrowsePropertyPage);
417 INF_SET_CLSID(CDwnBindInfo);
418 INF_SET_CLSID(CHiFiUses);
419 INF_SET_CLSID(CHtmlComponentConstructor);
420 INF_SET_CLSID(CImageBrowsePropertyPage);
421 INF_SET_CLSID(CInlineStylePropertyPage);
422 INF_SET_CLSID(CPeerHandler);
423 INF_SET_CLSID(CRecalcEngine);
424 INF_SET_CLSID(CSvrOMUses);
425 INF_SET_CLSID(CrSource);
426 INF_SET_CLSID(ExternalFrameworkSite);
427 INF_SET_CLSID(HTADocument);
428 INF_SET_CLSID(HTMLDocument);
429 INF_SET_CLSID(HTMLLoadOptions);
430 INF_SET_CLSID(HTMLPluginDocument);
431 INF_SET_CLSID(HTMLPopup);
432 INF_SET_CLSID(HTMLPopupDoc);
433 INF_SET_CLSID(HTMLServerDoc);
434 INF_SET_CLSID(HTMLWindowProxy);
435 INF_SET_CLSID(IImageDecodeFilter);
436 INF_SET_CLSID(IImgCtx);
437 INF_SET_CLSID(IntDitherer);
438 INF_SET_CLSID(JSProtocol);
439 INF_SET_CLSID(MHTMLDocument);
440 INF_SET_CLSID(MailtoProtocol);
441 INF_SET_CLSID(ResProtocol);
442 INF_SET_CLSID(Scriptlet);
443 INF_SET_CLSID(SysimageProtocol);
444 INF_SET_CLSID(TridentAPI);
445 INF_SET_ID(LIBID_MSHTML);
447 for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++) {
448 pse[i].pszValue = heap_alloc(39);
449 sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
450 clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
451 clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
452 clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
455 strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
456 strtable.pse = pse;
458 hAdvpack = LoadLibraryW(wszAdvpack);
459 pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
461 hres = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll", &strtable);
463 for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
464 heap_free(pse[i].pszValue);
466 if(FAILED(hres))
467 ERR("RegInstall failed: %08x\n", hres);
469 return hres;
472 #undef INF_SET_CLSID
473 #undef INF_SET_ID
475 /***********************************************************************
476 * DllRegisterServer (MSHTML.@)
478 HRESULT WINAPI DllRegisterServer(void)
480 HRESULT hres;
482 hres = __wine_register_resources( hInst );
483 if(SUCCEEDED(hres))
484 hres = register_server(TRUE);
485 if(SUCCEEDED(hres))
486 load_gecko();
488 return hres;
491 /***********************************************************************
492 * DllUnregisterServer (MSHTML.@)
494 HRESULT WINAPI DllUnregisterServer(void)
496 HRESULT hres = __wine_unregister_resources( hInst );
497 if(SUCCEEDED(hres)) hres = register_server(FALSE);
498 return hres;
501 const char *debugstr_variant(const VARIANT *v)
503 if(!v)
504 return "(null)";
506 switch(V_VT(v)) {
507 case VT_EMPTY:
508 return "{VT_EMPTY}";
509 case VT_NULL:
510 return "{VT_NULL}";
511 case VT_I2:
512 return wine_dbg_sprintf("{VT_I2: %d}", V_I2(v));
513 case VT_I4:
514 return wine_dbg_sprintf("{VT_I4: %d}", V_I4(v));
515 case VT_R8:
516 return wine_dbg_sprintf("{VT_R8: %lf}", V_R8(v));
517 case VT_BSTR:
518 return wine_dbg_sprintf("{VT_BSTR: %s}", debugstr_w(V_BSTR(v)));
519 case VT_DISPATCH:
520 return wine_dbg_sprintf("{VT_DISPATCH: %p}", V_DISPATCH(v));
521 case VT_ERROR:
522 return wine_dbg_sprintf("{VT_ERROR: %08x}", V_ERROR(v));
523 case VT_BOOL:
524 return wine_dbg_sprintf("{VT_BOOL: %x}", V_BOOL(v));
525 case VT_UINT:
526 return wine_dbg_sprintf("{VT_UINT: %u}", V_UINT(v));
527 default:
528 return wine_dbg_sprintf("{vt %d}", V_VT(v));