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 "wine/debug.h"
  23 #include "shdocvw.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(shdocvw);
  26
  27 /**********************************************************************
  28  * Implement the WebBrowser class factory
  29  *
  30  * (Based on implementation in ddraw/main.c)
  31  */
  32
  33 #define FACTORY(x) ((IClassFactory*) &(x)->lpClassFactoryVtbl)
  34
  35 typedef struct
  36 {
  37     /* IUnknown fields */
  38     const IClassFactoryVtbl *lpClassFactoryVtbl;
  39     HRESULT (*cf)(LPUNKNOWN, REFIID, LPVOID *);
  40     LONG ref;
  41 } IClassFactoryImpl;
  42
  43
  44 /**********************************************************************
  45  * WBCF_QueryInterface (IUnknown)
  46  */
  47 static HRESULT WINAPI WBCF_QueryInterface(LPCLASSFACTORY iface,
  48                                           REFIID riid, LPVOID *ppobj)
  49 {
  50     TRACE("(%s %p)\n", debugstr_guid(riid), ppobj);
  51
  52     if (!ppobj)
  53         return E_POINTER;
  54
  55     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
  56         *ppobj = iface;
  57         return S_OK;
  58     }
  59
  60     WARN("Not supported interface %s\n", debugstr_guid(riid));
  61
  62     *ppobj = NULL;
  63     return E_NOINTERFACE;
  64 }
  65
  66 /************************************************************************
  67  * WBCF_AddRef (IUnknown)
  68  */
  69 static ULONG WINAPI WBCF_AddRef(LPCLASSFACTORY iface)
  70 {
  71     SHDOCVW_LockModule();
  72
  73     return 2; /* non-heap based object */
  74 }
  75
  76 /************************************************************************
  77  * WBCF_Release (IUnknown)
  78  */
  79 static ULONG WINAPI WBCF_Release(LPCLASSFACTORY iface)
  80 {
  81     SHDOCVW_UnlockModule();
  82
  83     return 1; /* non-heap based object */
  84 }
  85
  86 /************************************************************************
  87  * WBCF_CreateInstance (IClassFactory)
  88  */
  89 static HRESULT WINAPI WBCF_CreateInstance(LPCLASSFACTORY iface, LPUNKNOWN pOuter,
  90                                           REFIID riid, LPVOID *ppobj)
  91 {
  92     IClassFactoryImpl *This = (IClassFactoryImpl *) iface;
  93     return This->cf(pOuter, riid, ppobj);
  94 }
  95
  96 /************************************************************************
  97  * WBCF_LockServer (IClassFactory)
  98  */
  99 static HRESULT WINAPI WBCF_LockServer(LPCLASSFACTORY iface, BOOL dolock)
 100 {
 101     TRACE("(%d)\n", dolock);
 102
 103     if (dolock)
 104         SHDOCVW_LockModule();
 105     else
 106         SHDOCVW_UnlockModule();
 107
 108     return S_OK;
 109 }
 110
 111 static const IClassFactoryVtbl WBCF_Vtbl =
 112 {
 113     WBCF_QueryInterface,
 114     WBCF_AddRef,
 115     WBCF_Release,
 116     WBCF_CreateInstance,
 117     WBCF_LockServer
 118 };
 119
 120 /*************************************************************************
 121  *              DllGetClassObject (SHDOCVW.@)
 122  */
 123 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
 124 {
 125     static IClassFactoryImpl WB1ClassFactory = {&WBCF_Vtbl, WebBrowserV1_Create};
 126     static IClassFactoryImpl WB2ClassFactory = {&WBCF_Vtbl, WebBrowserV2_Create};
 127     static IClassFactoryImpl CUHClassFactory = {&WBCF_Vtbl, CUrlHistory_Create};
 128
 129     TRACE("\n");
 130
 131     if(IsEqualGUID(&CLSID_WebBrowser, rclsid))
 132         return IClassFactory_QueryInterface(FACTORY(&WB2ClassFactory), riid, ppv);
 133
 134     if(IsEqualGUID(&CLSID_WebBrowser_V1, rclsid))
 135         return IClassFactory_QueryInterface(FACTORY(&WB1ClassFactory), riid, ppv);
 136
 137     if(IsEqualGUID(&CLSID_CUrlHistory, rclsid))
 138         return IClassFactory_QueryInterface(FACTORY(&CUHClassFactory), riid, ppv);
 139
 140     /* As a last resort, figure if the CLSID belongs to a 'Shell Instance Object' */
 141     return SHDOCVW_GetShellInstanceObjectClassObject(rclsid, riid, ppv);
 142 }
 143
 144 HRESULT register_class_object(BOOL do_reg)
 145 {
 146     HRESULT hres;
 147
 148     static DWORD cookie;
 149     static IClassFactoryImpl IEClassFactory = {&WBCF_Vtbl, InternetExplorer_Create};
 150
 151     if(do_reg) {
 152         hres = CoRegisterClassObject(&CLSID_InternetExplorer, (IUnknown*)FACTORY(&IEClassFactory),
 153                                      CLSCTX_SERVER, REGCLS_MULTIPLEUSE|REGCLS_SUSPENDED, &cookie);
 154         if (FAILED(hres)) {
 155             ERR("failed to register object %08x\n", hres);
 156             return hres;
 157         }
 158
 159         hres = CoResumeClassObjects();
 160         if(SUCCEEDED(hres))
 161             return hres;
 162
 163         ERR("failed to resume object %08x\n", hres);
 164     }
 165
 166     return CoRevokeClassObject(cookie);
 167 }