dlls/sti/sti_main.c

   1 /*
   2  * Copyright (C) 2002 Aric Stewart for CodeWeavers
   3  * Copyright (C) 2009 Damjan Jovanovic
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include <stdarg.h>
  21
  22 #define COBJMACROS
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "winreg.h"
  27 #include "winerror.h"
  28 #include "objbase.h"
  29 #include "initguid.h"
  30 #include "wia_lh.h"
  31 #include "sti.h"
  32
  33 #include "wine/debug.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(sti);
  36
  37 extern HRESULT WINAPI STI_DllGetClassObject(REFCLSID, REFIID, LPVOID *);
  38 extern BOOL WINAPI STI_DllMain(HINSTANCE, DWORD, LPVOID);
  39 extern HRESULT WINAPI STI_DllRegisterServer(void);
  40 extern HRESULT WINAPI STI_DllUnregisterServer(void);
  41
  42 typedef HRESULT (*fnCreateInstance)(REFIID riid, IUnknown *pUnkOuter, LPVOID *ppObj);
  43
  44 typedef struct
  45 {
  46     IClassFactory IClassFactory_iface;
  47     fnCreateInstance pfnCreateInstance;
  48 } sti_cf;
  49
  50 static inline sti_cf *impl_from_IClassFactory( IClassFactory *iface )
  51 {
  52     return CONTAINING_RECORD(iface, sti_cf, IClassFactory_iface);
  53 }
  54
  55 static HRESULT sti_create( REFIID riid, IUnknown *pUnkOuter, LPVOID *ppObj )
  56 {
  57     if (pUnkOuter != NULL && !IsEqualIID(riid, &IID_IUnknown))
  58         return CLASS_E_NOAGGREGATION;
  59
  60     if (IsEqualGUID(riid, &IID_IUnknown))
  61         return StiCreateInstanceW(GetCurrentProcess(), STI_VERSION_REAL | STI_VERSION_FLAG_UNICODE, (PSTIW*) ppObj, pUnkOuter);
  62     else if (IsEqualGUID(riid, &IID_IStillImageW))
  63         return StiCreateInstanceW(GetCurrentProcess(), STI_VERSION_REAL | STI_VERSION_FLAG_UNICODE, (PSTIW*) ppObj, NULL);
  64     else if (IsEqualGUID(riid, &IID_IStillImageA))
  65         return StiCreateInstanceA(GetCurrentProcess(), STI_VERSION_REAL, (PSTIA*) ppObj, NULL);
  66     else
  67     {
  68         FIXME("no interface %s\n", debugstr_guid(riid));
  69         return E_NOINTERFACE;
  70     }
  71 }
  72
  73 static HRESULT WINAPI sti_cf_QueryInterface( IClassFactory *iface, REFIID riid, LPVOID *ppobj )
  74 {
  75     if (IsEqualGUID(riid, &IID_IUnknown) ||
  76         IsEqualGUID(riid, &IID_IClassFactory))
  77     {
  78         IClassFactory_AddRef( iface );
  79         *ppobj = iface;
  80         return S_OK;
  81     }
  82     FIXME("interface %s not implemented\n", debugstr_guid(riid));
  83     return E_NOINTERFACE;
  84 }
  85
  86 static ULONG WINAPI sti_cf_AddRef( IClassFactory *iface )
  87 {
  88     return 2;
  89 }
  90
  91 static ULONG WINAPI sti_cf_Release( IClassFactory *iface )
  92 {
  93     return 1;
  94 }
  95
  96 static HRESULT WINAPI sti_cf_CreateInstance( IClassFactory *iface, LPUNKNOWN pOuter,
  97                                              REFIID riid, LPVOID *ppobj )
  98 {
  99     sti_cf *This = impl_from_IClassFactory( iface );
 100     HRESULT r;
 101     IUnknown *punk;
 102
 103     TRACE("%p %s %p\n", pOuter, debugstr_guid(riid), ppobj);
 104
 105     *ppobj = NULL;
 106
 107     r = This->pfnCreateInstance( riid, pOuter, (LPVOID *)&punk );
 108     if (FAILED(r))
 109         return r;
 110
 111     r = IUnknown_QueryInterface( punk, riid, ppobj );
 112     if (FAILED(r))
 113         return r;
 114
 115     IUnknown_Release( punk );
 116     return r;
 117 }
 118
 119 static HRESULT WINAPI sti_cf_LockServer( IClassFactory *iface, BOOL dolock )
 120 {
 121     FIXME("(%p)->(%d)\n", iface, dolock);
 122     return S_OK;
 123 }
 124
 125 static const struct IClassFactoryVtbl sti_cf_vtbl =
 126 {
 127     sti_cf_QueryInterface,
 128     sti_cf_AddRef,
 129     sti_cf_Release,
 130     sti_cf_CreateInstance,
 131     sti_cf_LockServer
 132 };
 133
 134 static sti_cf the_sti_cf = { { &sti_cf_vtbl }, sti_create };
 135
 136 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpvReserved)
 137 {
 138     TRACE("(0x%p, %ld, %p)\n",hInstDLL,fdwReason,lpvReserved);
 139
 140     return STI_DllMain(hInstDLL, fdwReason, lpvReserved);
 141 }
 142
 143 /******************************************************************************
 144  *           DllGetClassObject   (STI.@)
 145  */
 146 HRESULT WINAPI DllGetClassObject( REFCLSID rclsid, REFIID iid, LPVOID *ppv )
 147 {
 148     IClassFactory *cf = NULL;
 149
 150     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
 151
 152     if (IsEqualGUID( rclsid, &CLSID_Sti ))
 153     {
 154        cf = &the_sti_cf.IClassFactory_iface;
 155     }
 156
 157     if (cf)
 158         return IClassFactory_QueryInterface( cf, iid, ppv );
 159     return STI_DllGetClassObject( rclsid, iid, ppv );
 160 }
 161
 162 /***********************************************************************
 163  *           DllRegisterServer (STI.@)
 164  */
 165 HRESULT WINAPI DllRegisterServer(void)
 166 {
 167     return STI_DllRegisterServer();
 168 }
 169
 170 /***********************************************************************
 171  *           DllUnRegisterServer (STI.@)
 172  */
 173 HRESULT WINAPI DllUnregisterServer(void)
 174 {
 175     return STI_DllUnregisterServer();
 176 }