shell32/tests: Fix a test failure on a default Windows 7 system.
[wine] / 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 "sti.h"
30
31 #include "wine/debug.h"
32
33 WINE_DEFAULT_DEBUG_CHANNEL(sti);
34
35 extern HRESULT WINAPI STI_DllGetClassObject(REFCLSID, REFIID, LPVOID *) DECLSPEC_HIDDEN;
36 extern BOOL WINAPI STI_DllMain(HINSTANCE, DWORD, LPVOID) DECLSPEC_HIDDEN;
37
38 typedef HRESULT (*fnCreateInstance)(REFIID riid, IUnknown *pUnkOuter, LPVOID *ppObj);
39
40 typedef struct
41 {
42     const struct IClassFactoryVtbl *vtbl;
43     fnCreateInstance pfnCreateInstance;
44 } sti_cf;
45
46 static inline sti_cf *impl_from_IClassFactory( IClassFactory *iface )
47 {
48     return (sti_cf *)((char *)iface - FIELD_OFFSET( sti_cf, vtbl ));
49 }
50
51 static HRESULT sti_create( REFIID riid, IUnknown *pUnkOuter, LPVOID *ppObj )
52 {
53     if (pUnkOuter != NULL && !IsEqualIID(riid, &IID_IUnknown))
54         return CLASS_E_NOAGGREGATION;
55
56     if (IsEqualGUID(riid, &IID_IUnknown))
57         return StiCreateInstanceW(GetCurrentProcess(), STI_VERSION_REAL | STI_VERSION_FLAG_UNICODE, (PSTIW*) ppObj, pUnkOuter);
58     else if (IsEqualGUID(riid, &IID_IStillImageW))
59         return StiCreateInstanceW(GetCurrentProcess(), STI_VERSION_REAL | STI_VERSION_FLAG_UNICODE, (PSTIW*) ppObj, NULL);
60     else if (IsEqualGUID(riid, &IID_IStillImageA))
61         return StiCreateInstanceA(GetCurrentProcess(), STI_VERSION_REAL, (PSTIA*) ppObj, NULL);
62     else
63     {
64         FIXME("no interface %s\n", debugstr_guid(riid));
65         return E_NOINTERFACE;
66     }
67 }
68
69 static HRESULT WINAPI sti_cf_QueryInterface( IClassFactory *iface, REFIID riid, LPVOID *ppobj )
70 {
71     if (IsEqualGUID(riid, &IID_IUnknown) ||
72         IsEqualGUID(riid, &IID_IClassFactory))
73     {
74         IClassFactory_AddRef( iface );
75         *ppobj = iface;
76         return S_OK;
77     }
78     FIXME("interface %s not implemented\n", debugstr_guid(riid));
79     return E_NOINTERFACE;
80 }
81
82 static ULONG WINAPI sti_cf_AddRef( IClassFactory *iface )
83 {
84     return 2;
85 }
86
87 static ULONG WINAPI sti_cf_Release( IClassFactory *iface )
88 {
89     return 1;
90 }
91
92 static HRESULT WINAPI sti_cf_CreateInstance( IClassFactory *iface, LPUNKNOWN pOuter,
93                                              REFIID riid, LPVOID *ppobj )
94 {
95     sti_cf *This = impl_from_IClassFactory( iface );
96     HRESULT r;
97     IUnknown *punk;
98
99     TRACE("%p %s %p\n", pOuter, debugstr_guid(riid), ppobj);
100
101     *ppobj = NULL;
102
103     r = This->pfnCreateInstance( riid, pOuter, (LPVOID *)&punk );
104     if (FAILED(r))
105         return r;
106
107     r = IUnknown_QueryInterface( punk, riid, ppobj );
108     if (FAILED(r))
109         return r;
110
111     IUnknown_Release( punk );
112     return r;
113 }
114
115 static HRESULT WINAPI sti_cf_LockServer( IClassFactory *iface, BOOL dolock )
116 {
117     FIXME("(%p)->(%d)\n", iface, dolock);
118     return S_OK;
119 }
120
121 static const struct IClassFactoryVtbl sti_cf_vtbl =
122 {
123     sti_cf_QueryInterface,
124     sti_cf_AddRef,
125     sti_cf_Release,
126     sti_cf_CreateInstance,
127     sti_cf_LockServer
128 };
129
130 static sti_cf the_sti_cf = { &sti_cf_vtbl, sti_create };
131
132 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpvReserved)
133 {
134     TRACE("(0x%p, %d, %p)\n",hInstDLL,fdwReason,lpvReserved);
135
136     if (fdwReason == DLL_WINE_PREATTACH)
137         return FALSE;
138     return STI_DllMain(hInstDLL, fdwReason, lpvReserved);
139 }
140
141 /******************************************************************************
142  *           DllGetClassObject   (STI.@)
143  */
144 HRESULT WINAPI DllGetClassObject( REFCLSID rclsid, REFIID iid, LPVOID *ppv )
145 {
146     IClassFactory *cf = NULL;
147
148     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
149
150     if (IsEqualGUID( rclsid, &CLSID_Sti ))
151     {
152        cf = (IClassFactory *)&the_sti_cf.vtbl;
153     }
154
155     if (cf)
156         return IClassFactory_QueryInterface( cf, iid, ppv );
157     return STI_DllGetClassObject( rclsid, iid, ppv );
158 }
159
160 /******************************************************************************
161  *           DllCanUnloadNow   (STI.@)
162  */
163 HRESULT WINAPI DllCanUnloadNow( void )
164 {
165     FIXME("\n");
166     return S_FALSE;
167 }