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