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 HINSTANCE wshom_instance;
  30
  31 static ITypeLib *typelib;
  32 static ITypeInfo *typeinfos[LAST_tid];
  33
  34 static REFIID tid_ids[] = {
  35     &IID_NULL,
  36     &IID_IWshCollection,
  37     &IID_IWshEnvironment,
  38     &IID_IWshShell3,
  39     &IID_IWshShortcut
  40 };
  41
  42 static HRESULT load_typelib(void)
  43 {
  44     HRESULT hres;
  45     ITypeLib *tl;
  46
  47     hres = LoadRegTypeLib(&LIBID_IWshRuntimeLibrary, 1, 0, LOCALE_SYSTEM_DEFAULT, &tl);
  48     if(FAILED(hres)) {
  49         ERR("LoadRegTypeLib failed: %08x\n", hres);
  50         return hres;
  51     }
  52
  53     if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
  54         ITypeLib_Release(tl);
  55     return hres;
  56 }
  57
  58 HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
  59 {
  60     HRESULT hres;
  61
  62     if (!typelib)
  63         hres = load_typelib();
  64     if (!typelib)
  65         return hres;
  66
  67     if(!typeinfos[tid]) {
  68         ITypeInfo *ti;
  69
  70         hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &ti);
  71         if(FAILED(hres)) {
  72             ERR("GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_guid(tid_ids[tid]), hres);
  73             return hres;
  74         }
  75
  76         if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), ti, NULL))
  77             ITypeInfo_Release(ti);
  78     }
  79
  80     *typeinfo = typeinfos[tid];
  81     ITypeInfo_AddRef(*typeinfo);
  82     return S_OK;
  83 }
  84
  85 static
  86 void release_typelib(void)
  87 {
  88     unsigned i;
  89
  90     if(!typelib)
  91         return;
  92
  93     for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
  94         if(typeinfos[i])
  95             ITypeInfo_Release(typeinfos[i]);
  96
  97     ITypeLib_Release(typelib);
  98 }
  99
 100 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 101 {
 102     *ppv = NULL;
 103
 104     if(IsEqualGUID(&IID_IUnknown, riid)) {
 105         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
 106         *ppv = iface;
 107     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
 108         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
 109         *ppv = iface;
 110     }
 111
 112     if(*ppv) {
 113         IUnknown_AddRef((IUnknown*)*ppv);
 114         return S_OK;
 115     }
 116
 117     FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
 118     return E_NOINTERFACE;
 119 }
 120
 121 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 122 {
 123     TRACE("(%p)\n", iface);
 124     return 2;
 125 }
 126
 127 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 128 {
 129     TRACE("(%p)\n", iface);
 130     return 1;
 131 }
 132
 133 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
 134 {
 135     TRACE("(%p)->(%x)\n", iface, fLock);
 136     return S_OK;
 137 }
 138
 139 static const IClassFactoryVtbl WshShellFactoryVtbl = {
 140     ClassFactory_QueryInterface,
 141     ClassFactory_AddRef,
 142     ClassFactory_Release,
 143     WshShellFactory_CreateInstance,
 144     ClassFactory_LockServer
 145 };
 146
 147 static IClassFactory WshShellFactory = { &WshShellFactoryVtbl };
 148
 149 /******************************************************************
 150  *              DllMain (wshom.ocx.@)
 151  */
 152 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
 153 {
 154     TRACE("(%p %d %p)\n", hInstDLL, fdwReason, lpv);
 155
 156     switch(fdwReason)
 157     {
 158     case DLL_WINE_PREATTACH:
 159         return FALSE;  /* prefer native version */
 160     case DLL_PROCESS_ATTACH:
 161         wshom_instance = hInstDLL;
 162         DisableThreadLibraryCalls(wshom_instance);
 163         break;
 164     case DLL_PROCESS_DETACH:
 165         if (lpv) break;
 166         release_typelib();
 167         break;
 168     }
 169
 170     return TRUE;
 171 }
 172
 173 /***********************************************************************
 174  *              DllGetClassObject       (wshom.ocx.@)
 175  */
 176 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 177 {
 178     if(IsEqualGUID(&CLSID_WshShell, rclsid)) {
 179         TRACE("(CLSID_WshShell %s %p)\n", debugstr_guid(riid), ppv);
 180         return IClassFactory_QueryInterface(&WshShellFactory, riid, ppv);
 181     }
 182
 183     FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 184     return CLASS_E_CLASSNOTAVAILABLE;
 185 }
 186
 187 /***********************************************************************
 188  *          DllCanUnloadNow (wshom.ocx.@)
 189  */
 190 HRESULT WINAPI DllCanUnloadNow(void)
 191 {
 192     return S_FALSE;
 193 }
 194
 195 /***********************************************************************
 196  *          DllRegisterServer (wshom.ocx.@)
 197  */
 198 HRESULT WINAPI DllRegisterServer(void)
 199 {
 200     TRACE("()\n");
 201     return __wine_register_resources(wshom_instance);
 202 }
 203
 204 /***********************************************************************
 205  *          DllUnregisterServer (wshom.ocx.@)
 206  */
 207 HRESULT WINAPI DllUnregisterServer(void)
 208 {
 209     TRACE("()\n");
 210     return __wine_unregister_resources(wshom_instance);
 211 }