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