dlls/shdocvw/factory.c

   1 /*
   2  * Implementation of class factory for IE Web Browser
   3  *
   4  * Copyright 2001 John R. Sheets (for CodeWeavers)
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <string.h>
  22 #include <stdio.h>
  23
  24 #include "shdocvw.h"
  25 #include "winreg.h"
  26 #include "advpub.h"
  27 #include "rpcproxy.h"
  28 #include "isguids.h"
  29
  30 #include "winver.h"
  31
  32 #include "wine/debug.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(shdocvw);
  35
  36 /**********************************************************************
  37  * Implement the WebBrowser class factory
  38  */
  39
  40 typedef struct
  41 {
  42     /* IUnknown fields */
  43     IClassFactory IClassFactory_iface;
  44     HRESULT (*cf)(LPUNKNOWN, REFIID, LPVOID *);
  45     LONG ref;
  46 } IClassFactoryImpl;
  47
  48 static inline IClassFactoryImpl *impl_from_IClassFactory(IClassFactory *iface)
  49 {
  50     return CONTAINING_RECORD(iface, IClassFactoryImpl, IClassFactory_iface);
  51 }
  52
  53
  54 /**********************************************************************
  55  * WBCF_QueryInterface (IUnknown)
  56  */
  57 static HRESULT WINAPI WBCF_QueryInterface(LPCLASSFACTORY iface,
  58                                           REFIID riid, LPVOID *ppobj)
  59 {
  60     TRACE("(%s %p)\n", debugstr_guid(riid), ppobj);
  61
  62     if (!ppobj)
  63         return E_POINTER;
  64
  65     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
  66         *ppobj = iface;
  67         IClassFactory_AddRef(iface);
  68         return S_OK;
  69     }
  70
  71     WARN("Not supported interface %s\n", debugstr_guid(riid));
  72
  73     *ppobj = NULL;
  74     return E_NOINTERFACE;
  75 }
  76
  77 /************************************************************************
  78  * WBCF_AddRef (IUnknown)
  79  */
  80 static ULONG WINAPI WBCF_AddRef(LPCLASSFACTORY iface)
  81 {
  82     SHDOCVW_LockModule();
  83
  84     return 2; /* non-heap based object */
  85 }
  86
  87 /************************************************************************
  88  * WBCF_Release (IUnknown)
  89  */
  90 static ULONG WINAPI WBCF_Release(LPCLASSFACTORY iface)
  91 {
  92     SHDOCVW_UnlockModule();
  93
  94     return 1; /* non-heap based object */
  95 }
  96
  97 /************************************************************************
  98  * WBCF_CreateInstance (IClassFactory)
  99  */
 100 static HRESULT WINAPI WBCF_CreateInstance(LPCLASSFACTORY iface, LPUNKNOWN pOuter,
 101                                           REFIID riid, LPVOID *ppobj)
 102 {
 103     IClassFactoryImpl *This = (IClassFactoryImpl *) iface;
 104     return This->cf(pOuter, riid, ppobj);
 105 }
 106
 107 /************************************************************************
 108  * WBCF_LockServer (IClassFactory)
 109  */
 110 static HRESULT WINAPI WBCF_LockServer(LPCLASSFACTORY iface, BOOL dolock)
 111 {
 112     TRACE("(%d)\n", dolock);
 113
 114     if (dolock)
 115         SHDOCVW_LockModule();
 116     else
 117         SHDOCVW_UnlockModule();
 118
 119     return S_OK;
 120 }
 121
 122 static const IClassFactoryVtbl WBCF_Vtbl =
 123 {
 124     WBCF_QueryInterface,
 125     WBCF_AddRef,
 126     WBCF_Release,
 127     WBCF_CreateInstance,
 128     WBCF_LockServer
 129 };
 130
 131 static HRESULT get_ieframe_object(REFCLSID rclsid, REFIID riid, void **ppv)
 132 {
 133     HINSTANCE ieframe_instance;
 134
 135     static HRESULT (WINAPI *ieframe_DllGetClassObject)(REFCLSID,REFIID,void**);
 136
 137     if(!ieframe_DllGetClassObject) {
 138         ieframe_instance = get_ieframe_instance();
 139         if(!ieframe_instance)
 140             return CLASS_E_CLASSNOTAVAILABLE;
 141
 142         ieframe_DllGetClassObject = (void*)GetProcAddress(ieframe_instance, "DllGetClassObject");
 143         if(!ieframe_DllGetClassObject)
 144             return CLASS_E_CLASSNOTAVAILABLE;
 145     }
 146
 147     return ieframe_DllGetClassObject(rclsid, riid, ppv);
 148 }
 149
 150 /*************************************************************************
 151  *              DllGetClassObject (SHDOCVW.@)
 152  */
 153 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
 154 {
 155     static IClassFactoryImpl WB1ClassFactory = {{&WBCF_Vtbl}, WebBrowserV1_Create};
 156     static IClassFactoryImpl WB2ClassFactory = {{&WBCF_Vtbl}, WebBrowserV2_Create};
 157
 158     TRACE("\n");
 159
 160     if(IsEqualGUID(&CLSID_WebBrowser, rclsid))
 161         return IClassFactory_QueryInterface(&WB2ClassFactory.IClassFactory_iface, riid, ppv);
 162
 163     if(IsEqualGUID(&CLSID_WebBrowser_V1, rclsid))
 164         return IClassFactory_QueryInterface(&WB1ClassFactory.IClassFactory_iface, riid, ppv);
 165
 166     if(IsEqualGUID(&CLSID_InternetShortcut, rclsid)
 167        || IsEqualGUID(&CLSID_CUrlHistory, rclsid)
 168        || IsEqualGUID(&CLSID_TaskbarList, rclsid))
 169         return get_ieframe_object(rclsid, riid, ppv);
 170
 171     /* As a last resort, figure if the CLSID belongs to a 'Shell Instance Object' */
 172     return SHDOCVW_GetShellInstanceObjectClassObject(rclsid, riid, ppv);
 173 }
 174
 175 HRESULT register_class_object(BOOL do_reg)
 176 {
 177     HRESULT hres;
 178
 179     static DWORD cookie;
 180     static IClassFactoryImpl IEClassFactory = {{&WBCF_Vtbl}, InternetExplorer_Create};
 181
 182     if(do_reg) {
 183         hres = CoRegisterClassObject(&CLSID_InternetExplorer,
 184                                      (IUnknown*)&IEClassFactory.IClassFactory_iface, CLSCTX_SERVER,
 185                                      REGCLS_MULTIPLEUSE|REGCLS_SUSPENDED, &cookie);
 186         if (FAILED(hres)) {
 187             ERR("failed to register object %08x\n", hres);
 188             return hres;
 189         }
 190
 191         hres = CoResumeClassObjects();
 192         if(SUCCEEDED(hres))
 193             return hres;
 194
 195         ERR("failed to resume object %08x\n", hres);
 196     }
 197
 198     return CoRevokeClassObject(cookie);
 199 }
 200
 201 static HRESULT reg_install(LPCSTR section, STRTABLEA *strtable)
 202 {
 203     HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
 204     HMODULE hadvpack;
 205     HRESULT hres;
 206
 207     static const WCHAR advpackW[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
 208
 209     hadvpack = LoadLibraryW(advpackW);
 210     pRegInstall = (void *)GetProcAddress(hadvpack, "RegInstall");
 211
 212     hres = pRegInstall(shdocvw_hinstance, section, strtable);
 213
 214     FreeLibrary(hadvpack);
 215     return hres;
 216 }
 217
 218 #define INF_SET_CLSID(clsid)                  \
 219     do                                        \
 220     {                                         \
 221         static CHAR name[] = "CLSID_" #clsid; \
 222                                               \
 223         pse[i].pszName = name;                \
 224         clsids[i++] = &CLSID_ ## clsid;       \
 225     } while (0)
 226
 227 static HRESULT register_server(BOOL doregister)
 228 {
 229     STRTABLEA strtable;
 230     STRENTRYA pse[3];
 231     static CLSID const *clsids[3];
 232     unsigned int i = 0;
 233     HRESULT hres;
 234
 235     INF_SET_CLSID(Internet);
 236     INF_SET_CLSID(InternetExplorer);
 237     INF_SET_CLSID(InternetShortcut);
 238
 239     for(i = 0; i < sizeof(pse)/sizeof(pse[0]); i++) {
 240         pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, 39);
 241         sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
 242                 clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
 243                 clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
 244                 clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
 245     }
 246
 247     strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
 248     strtable.pse = pse;
 249
 250     hres = reg_install(doregister ? "RegisterDll" : "UnregisterDll", &strtable);
 251
 252     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
 253         HeapFree(GetProcessHeap(), 0, pse[i].pszValue);
 254
 255     return hres;
 256 }
 257
 258 #undef INF_SET_CLSID
 259
 260 /***********************************************************************
 261  *          DllRegisterServer (shdocvw.@)
 262  */
 263 HRESULT WINAPI DllRegisterServer(void)
 264 {
 265     HRESULT hres;
 266
 267     hres = __wine_register_resources( shdocvw_hinstance, NULL );
 268     if(FAILED(hres))
 269         return hres;
 270
 271     return register_server(TRUE);
 272 }
 273
 274 /***********************************************************************
 275  *          DllUnregisterServer (shdocvw.@)
 276  */
 277 HRESULT WINAPI DllUnregisterServer(void)
 278 {
 279     HRESULT hres;
 280
 281     hres = register_server(FALSE);
 282     if(FAILED(hres))
 283         return hres;
 284
 285     return __wine_unregister_resources( shdocvw_hinstance, NULL );
 286 }
 287
 288 static BOOL check_native_ie(void)
 289 {
 290     static const WCHAR cszPath[] = {'b','r','o','w','s','e','u','i','.','d','l','l',0};
 291     DWORD handle,size;
 292     BOOL ret = TRUE;
 293
 294     size = GetFileVersionInfoSizeW(cszPath,&handle);
 295     if (size)
 296     {
 297         LPVOID buf;
 298         LPWSTR lpFileDescription;
 299         UINT dwBytes;
 300         static const WCHAR cszFD[] = {'\\','S','t','r','i','n','g','F','i','l','e','I','n','f','o','\\','0','4','0','9','0','4','e','4','\\','F','i','l','e','D','e','s','c','r','i','p','t','i','o','n',0};
 301         static const WCHAR cszWine[] = {'W','i','n','e',0};
 302
 303         buf = HeapAlloc(GetProcessHeap(),0,size);
 304         GetFileVersionInfoW(cszPath,0,size,buf);
 305
 306         if (VerQueryValueW(buf, cszFD, (LPVOID*)&lpFileDescription, &dwBytes) &&
 307             strstrW(lpFileDescription,cszWine))
 308                 ret = FALSE;
 309
 310         HeapFree(GetProcessHeap(), 0, buf);
 311     }
 312
 313     return ret;
 314 }
 315
 316 DWORD register_iexplore(BOOL doregister)
 317 {
 318     HRESULT hres;
 319     if (check_native_ie())
 320     {
 321         TRACE("Native IE detected, not doing registration\n");
 322         return S_OK;
 323     }
 324     hres = reg_install(doregister ? "RegisterIE" : "UnregisterIE", NULL);
 325     return FAILED(hres);
 326 }