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