dlls/scrrun/scrrun.c

   1 /*
   2  * Copyright 2011 Hans Leidekker 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 #define COBJMACROS
  19
  20 #include "config.h"
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "ole2.h"
  26 #include "rpcproxy.h"
  27
  28 #include <initguid.h>
  29 #include "scrrun.h"
  30 #include "scrrun_private.h"
  31
  32 #include "wine/debug.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(scrrun);
  35
  36 static HINSTANCE scrrun_instance;
  37
  38 typedef HRESULT (*fnCreateInstance)(LPVOID *ppObj);
  39
  40 static HRESULT WINAPI scrruncf_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppv )
  41 {
  42     *ppv = NULL;
  43
  44     if(IsEqualGUID(&IID_IUnknown, riid)) {
  45         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
  46         *ppv = iface;
  47     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
  48         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
  49         *ppv = iface;
  50     }
  51
  52     if(*ppv) {
  53         IUnknown_AddRef((IUnknown*)*ppv);
  54         return S_OK;
  55     }
  56
  57     WARN("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
  58     return E_NOINTERFACE;
  59 }
  60
  61 static ULONG WINAPI scrruncf_AddRef(IClassFactory *iface )
  62 {
  63     TRACE("(%p)\n", iface);
  64     return 2;
  65 }
  66
  67 static ULONG WINAPI scrruncf_Release(IClassFactory *iface )
  68 {
  69     TRACE("(%p)\n", iface);
  70     return 1;
  71 }
  72
  73 static HRESULT WINAPI scrruncf_LockServer(IClassFactory *iface, BOOL fLock)
  74 {
  75     TRACE("(%p)->(%x)\n", iface, fLock);
  76     return S_OK;
  77 }
  78
  79 static const struct IClassFactoryVtbl scrruncf_vtbl =
  80 {
  81     scrruncf_QueryInterface,
  82     scrruncf_AddRef,
  83     scrruncf_Release,
  84     FileSystem_CreateInstance,
  85     scrruncf_LockServer
  86 };
  87
  88 static const struct IClassFactoryVtbl dictcf_vtbl =
  89 {
  90     scrruncf_QueryInterface,
  91     scrruncf_AddRef,
  92     scrruncf_Release,
  93     Dictionary_CreateInstance,
  94     scrruncf_LockServer
  95 };
  96
  97 static IClassFactory FileSystemFactory = { &scrruncf_vtbl };
  98 static IClassFactory DictionaryFactory = { &dictcf_vtbl };
  99
 100 static ITypeLib *typelib;
 101 static ITypeInfo *typeinfos[LAST_tid];
 102
 103 static REFIID tid_ids[] = {
 104     &IID_NULL,
 105     &IID_IDictionary,
 106     &IID_IDrive,
 107     &IID_IDriveCollection,
 108     &IID_IFile,
 109     &IID_IFileCollection,
 110     &IID_IFileSystem3,
 111     &IID_IFolder,
 112     &IID_IFolderCollection,
 113     &IID_ITextStream
 114 };
 115
 116 static HRESULT load_typelib(void)
 117 {
 118     HRESULT hres;
 119     ITypeLib *tl;
 120
 121     hres = LoadRegTypeLib(&LIBID_Scripting, 1, 0, LOCALE_SYSTEM_DEFAULT, &tl);
 122     if(FAILED(hres)) {
 123         ERR("LoadRegTypeLib failed: %08x\n", hres);
 124         return hres;
 125     }
 126
 127     if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
 128         ITypeLib_Release(tl);
 129     return hres;
 130 }
 131
 132 HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
 133 {
 134     HRESULT hres;
 135
 136     if (!typelib)
 137         hres = load_typelib();
 138     if (!typelib)
 139         return hres;
 140
 141     if(!typeinfos[tid]) {
 142         ITypeInfo *ti;
 143
 144         hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &ti);
 145         if(FAILED(hres)) {
 146             ERR("GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_guid(tid_ids[tid]), hres);
 147             return hres;
 148         }
 149
 150         if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), ti, NULL))
 151             ITypeInfo_Release(ti);
 152     }
 153
 154     *typeinfo = typeinfos[tid];
 155     ITypeInfo_AddRef(typeinfos[tid]);
 156     return S_OK;
 157 }
 158
 159 static void release_typelib(void)
 160 {
 161     unsigned i;
 162
 163     if(!typelib)
 164         return;
 165
 166     for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
 167         if(typeinfos[i])
 168             ITypeInfo_Release(typeinfos[i]);
 169
 170     ITypeLib_Release(typelib);
 171 }
 172
 173 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
 174 {
 175     TRACE("%p, %u, %p\n", hinst, reason, reserved);
 176
 177     switch (reason)
 178     {
 179         case DLL_WINE_PREATTACH:
 180             return FALSE;    /* prefer native version */
 181         case DLL_PROCESS_ATTACH:
 182             DisableThreadLibraryCalls( hinst );
 183             scrrun_instance = hinst;
 184             break;
 185         case DLL_PROCESS_DETACH:
 186             if (reserved) break;
 187             release_typelib();
 188             break;
 189     }
 190     return TRUE;
 191 }
 192
 193 /***********************************************************************
 194  *      DllRegisterServer (scrrun.@)
 195  */
 196 HRESULT WINAPI DllRegisterServer(void)
 197 {
 198     TRACE("()\n");
 199     return __wine_register_resources(scrrun_instance);
 200 }
 201
 202 /***********************************************************************
 203  *      DllUnregisterServer (scrrun.@)
 204  */
 205 HRESULT WINAPI DllUnregisterServer(void)
 206 {
 207     TRACE("()\n");
 208     return __wine_unregister_resources(scrrun_instance);
 209 }
 210
 211 /***********************************************************************
 212  *      DllGetClassObject (scrrun.@)
 213  */
 214
 215 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
 216 {
 217     if(IsEqualGUID(&CLSID_FileSystemObject, rclsid)) {
 218         TRACE("(CLSID_FileSystemObject %s %p)\n", debugstr_guid(riid), ppv);
 219         return IClassFactory_QueryInterface(&FileSystemFactory, riid, ppv);
 220     }
 221     else if(IsEqualGUID(&CLSID_Dictionary, rclsid)) {
 222         TRACE("(CLSID_Dictionary %s %p)\n", debugstr_guid(riid), ppv);
 223         return IClassFactory_QueryInterface(&DictionaryFactory, riid, ppv);
 224     }
 225
 226     FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 227     return CLASS_E_CLASSNOTAVAILABLE;
 228 }
 229
 230 /***********************************************************************
 231  *      DllCanUnloadNow (scrrun.@)
 232  */
 233 HRESULT WINAPI DllCanUnloadNow(void)
 234 {
 235     return S_FALSE;
 236 }