dlls/wshom.ocx/wshom_main.c

   1 /*
   2  * Copyright 2011 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "wshom_private.h"
  20
  21 #include "initguid.h"
  22 #include "wshom.h"
  23 #include "rpcproxy.h"
  24
  25 #include "wine/debug.h"
  26
  27 WINE_DEFAULT_DEBUG_CHANNEL(wshom);
  28
  29 static inline struct provideclassinfo *impl_from_IProvideClassInfo(IProvideClassInfo *iface)
  30 {
  31     return CONTAINING_RECORD(iface, struct provideclassinfo, IProvideClassInfo_iface);
  32 }
  33
  34 static ITypeLib *typelib;
  35 static ITypeInfo *typeinfos[LAST_tid];
  36
  37 static REFIID tid_ids[] = {
  38     &IID_NULL,
  39     &IID_IWshCollection,
  40     &IID_IWshEnvironment,
  41     &IID_IWshExec,
  42     &IID_IWshShell3,
  43     &IID_IWshShortcut,
  44     &IID_IWshNetwork2
  45 };
  46
  47 static HRESULT load_typelib(void)
  48 {
  49     HRESULT hres;
  50     ITypeLib *tl;
  51
  52     if(typelib)
  53         return S_OK;
  54
  55     hres = LoadRegTypeLib(&LIBID_IWshRuntimeLibrary, 1, 0, LOCALE_SYSTEM_DEFAULT, &tl);
  56     if(FAILED(hres)) {
  57         ERR("LoadRegTypeLib failed: %#lx.\n", hres);
  58         return hres;
  59     }
  60
  61     if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
  62         ITypeLib_Release(tl);
  63     return hres;
  64 }
  65
  66 static HRESULT get_typeinfo_of_guid(const GUID *guid, ITypeInfo **tinfo)
  67 {
  68     HRESULT hres;
  69
  70     if(FAILED(hres = load_typelib()))
  71         return hres;
  72
  73     return ITypeLib_GetTypeInfoOfGuid(typelib, guid, tinfo);
  74 }
  75
  76 HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
  77 {
  78     HRESULT hres;
  79
  80     if (FAILED(hres = load_typelib()))
  81         return hres;
  82
  83     if(!typeinfos[tid]) {
  84         ITypeInfo *ti;
  85
  86         hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &ti);
  87         if(FAILED(hres)) {
  88             ERR("GetTypeInfoOfGuid(%s) failed: %#lx.\n", debugstr_guid(tid_ids[tid]), hres);
  89             return hres;
  90         }
  91
  92         if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), ti, NULL))
  93             ITypeInfo_Release(ti);
  94     }
  95
  96     *typeinfo = typeinfos[tid];
  97     ITypeInfo_AddRef(*typeinfo);
  98     return S_OK;
  99 }
 100
 101 static
 102 void release_typelib(void)
 103 {
 104     unsigned i;
 105
 106     if(!typelib)
 107         return;
 108
 109     for(i = 0; i < ARRAY_SIZE(typeinfos); i++)
 110         if(typeinfos[i])
 111             ITypeInfo_Release(typeinfos[i]);
 112
 113     ITypeLib_Release(typelib);
 114 }
 115
 116 static HRESULT WINAPI provideclassinfo_QueryInterface(IProvideClassInfo *iface, REFIID riid, void **obj)
 117 {
 118     struct provideclassinfo *This = impl_from_IProvideClassInfo(iface);
 119
 120     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
 121
 122     if (IsEqualIID(riid, &IID_IProvideClassInfo)) {
 123         *obj = iface;
 124         IProvideClassInfo_AddRef(iface);
 125         return S_OK;
 126     }
 127     else
 128         return IUnknown_QueryInterface(This->outer, riid, obj);
 129 }
 130
 131 static ULONG WINAPI provideclassinfo_AddRef(IProvideClassInfo *iface)
 132 {
 133     struct provideclassinfo *This = impl_from_IProvideClassInfo(iface);
 134     return IUnknown_AddRef(This->outer);
 135 }
 136
 137 static ULONG WINAPI provideclassinfo_Release(IProvideClassInfo *iface)
 138 {
 139     struct provideclassinfo *This = impl_from_IProvideClassInfo(iface);
 140     return IUnknown_Release(This->outer);
 141 }
 142
 143 static HRESULT WINAPI provideclassinfo_GetClassInfo(IProvideClassInfo *iface, ITypeInfo **ti)
 144 {
 145     struct provideclassinfo *This = impl_from_IProvideClassInfo(iface);
 146
 147     TRACE("(%p)->(%p)\n", This, ti);
 148
 149     return get_typeinfo_of_guid(This->guid, ti);
 150 }
 151
 152 static const IProvideClassInfoVtbl provideclassinfovtbl = {
 153     provideclassinfo_QueryInterface,
 154     provideclassinfo_AddRef,
 155     provideclassinfo_Release,
 156     provideclassinfo_GetClassInfo
 157 };
 158
 159 void init_classinfo(const GUID *guid, IUnknown *outer, struct provideclassinfo *classinfo)
 160 {
 161     classinfo->IProvideClassInfo_iface.lpVtbl = &provideclassinfovtbl;
 162     classinfo->outer = outer;
 163     classinfo->guid = guid;
 164 }
 165
 166 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 167 {
 168     *ppv = NULL;
 169
 170     if(IsEqualGUID(&IID_IUnknown, riid)) {
 171         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
 172         *ppv = iface;
 173     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
 174         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
 175         *ppv = iface;
 176     }
 177
 178     if(*ppv) {
 179         IUnknown_AddRef((IUnknown*)*ppv);
 180         return S_OK;
 181     }
 182
 183     WARN("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
 184     return E_NOINTERFACE;
 185 }
 186
 187 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 188 {
 189     TRACE("(%p)\n", iface);
 190     return 2;
 191 }
 192
 193 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 194 {
 195     TRACE("(%p)\n", iface);
 196     return 1;
 197 }
 198
 199 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
 200 {
 201     TRACE("(%p)->(%x)\n", iface, fLock);
 202     return S_OK;
 203 }
 204
 205 static const IClassFactoryVtbl WshShellFactoryVtbl = {
 206     ClassFactory_QueryInterface,
 207     ClassFactory_AddRef,
 208     ClassFactory_Release,
 209     WshShellFactory_CreateInstance,
 210     ClassFactory_LockServer
 211 };
 212
 213 static const IClassFactoryVtbl WshNetworkFactoryVtbl = {
 214     ClassFactory_QueryInterface,
 215     ClassFactory_AddRef,
 216     ClassFactory_Release,
 217     WshNetworkFactory_CreateInstance,
 218     ClassFactory_LockServer
 219 };
 220
 221 static IClassFactory WshShellFactory = { &WshShellFactoryVtbl };
 222 static IClassFactory WshNetworkFactory = { &WshNetworkFactoryVtbl };
 223
 224 /******************************************************************
 225  *              DllMain (wshom.ocx.@)
 226  */
 227 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
 228 {
 229     TRACE("%p, %ld, %p.\n", hInstDLL, fdwReason, lpv);
 230
 231     switch(fdwReason)
 232     {
 233     case DLL_PROCESS_ATTACH:
 234         DisableThreadLibraryCalls(hInstDLL);
 235         break;
 236     case DLL_PROCESS_DETACH:
 237         if (lpv) break;
 238         release_typelib();
 239         break;
 240     }
 241
 242     return TRUE;
 243 }
 244
 245 /***********************************************************************
 246  *              DllGetClassObject       (wshom.ocx.@)
 247  */
 248 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 249 {
 250     if(IsEqualGUID(&CLSID_WshShell, rclsid)) {
 251         TRACE("(CLSID_WshShell %s %p)\n", debugstr_guid(riid), ppv);
 252         return IClassFactory_QueryInterface(&WshShellFactory, riid, ppv);
 253     }
 254     else if(IsEqualGUID(&CLSID_WshNetwork, rclsid)) {
 255         TRACE("(CLSID_WshNetwork %s %p)\n", debugstr_guid(riid), ppv);
 256         return IClassFactory_QueryInterface(&WshNetworkFactory, riid, ppv);
 257     }
 258
 259     FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 260     return CLASS_E_CLASSNOTAVAILABLE;
 261 }