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