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