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