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