- Always query for the correct stub interface, otherwise we will be
[wine] / dlls / amstream / main.c
1 /*
2  *     MultiMedia Streams Base Functions (AMSTREAM.DLL)
3  *
4  * Copyright 2004 Christian Costa
5  *
6  * This file contains the (internal) driver registration functions,
7  * driver enumeration APIs and DirectDraw creation functions.
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with this library; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
22  */
23
24 #include <stdarg.h>
25 #include <string.h>
26
27 #define COBJMACROS
28 #define COM_NO_WINDOWS_H
29
30 #include "windef.h"
31 #include "winbase.h"
32 #include "winuser.h"
33 #include "winreg.h"
34 #include "winerror.h"
35
36 #include "ole2.h"
37 #include "uuids.h"
38
39 #include "amstream_private.h"
40 #include "amstream.h"
41
42 #include "wine/debug.h"
43
44 WINE_DEFAULT_DEBUG_CHANNEL(amstream);
45
46 static DWORD dll_ref = 0;
47
48 /* For the moment, do nothing here. */
49 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
50 {
51     switch(fdwReason) {
52         case DLL_PROCESS_ATTACH:
53             DisableThreadLibraryCalls(hInstDLL);
54             break;
55         case DLL_PROCESS_DETACH:
56             break;
57     }
58     return TRUE;
59 }
60
61 /******************************************************************************
62  * Multimedia Streams ClassFactory
63  */
64 typedef struct {
65     IClassFactory ITF_IClassFactory;
66
67     LONG ref;
68     HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
69 } IClassFactoryImpl;
70
71 struct object_creation_info
72 {
73     const CLSID *clsid;
74     HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
75 };
76
77 static const struct object_creation_info object_creation[] =
78 {
79     { &CLSID_AMMultiMediaStream, AM_create },
80 };
81
82 static HRESULT WINAPI
83 AMCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid,LPVOID *ppobj)
84 {
85     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
86
87     if (IsEqualGUID(riid, &IID_IUnknown)
88         || IsEqualGUID(riid, &IID_IClassFactory))
89     {
90         IClassFactory_AddRef(iface);
91         *ppobj = This;
92         return S_OK;
93     }
94
95     WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppobj);
96     return E_NOINTERFACE;
97 }
98
99 static ULONG WINAPI AMCF_AddRef(LPCLASSFACTORY iface)
100 {
101     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
102     return InterlockedIncrement(&This->ref);
103 }
104
105 static ULONG WINAPI AMCF_Release(LPCLASSFACTORY iface)
106 {
107     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
108
109     ULONG ref = InterlockedDecrement(&This->ref);
110
111     if (ref == 0)
112         HeapFree(GetProcessHeap(), 0, This);
113
114     return ref;
115 }
116
117
118 static HRESULT WINAPI AMCF_CreateInstance(LPCLASSFACTORY iface, LPUNKNOWN pOuter,
119                                           REFIID riid, LPVOID *ppobj)
120 {
121     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
122     HRESULT hres;
123     LPUNKNOWN punk;
124     
125     TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);
126
127
128     *ppobj = NULL;
129     hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk);
130     if (SUCCEEDED(hres)) {
131         hres = IUnknown_QueryInterface(punk, riid, ppobj);
132         IUnknown_Release(punk);
133     }
134     return hres;
135 }
136
137 static HRESULT WINAPI AMCF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
138 {
139     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
140     FIXME("(%p)->(%d),stub!\n",This,dolock);
141     return S_OK;
142 }
143
144 static const IClassFactoryVtbl DSCF_Vtbl =
145 {
146     AMCF_QueryInterface,
147     AMCF_AddRef,
148     AMCF_Release,
149     AMCF_CreateInstance,
150     AMCF_LockServer
151 };
152
153 /*******************************************************************************
154  * DllGetClassObject [AMSTREAM.@]
155  * Retrieves class object from a DLL object
156  *
157  * NOTES
158  *    Docs say returns STDAPI
159  *
160  * PARAMS
161  *    rclsid [I] CLSID for the class object
162  *    riid   [I] Reference to identifier of interface for class object
163  *    ppv    [O] Address of variable to receive interface pointer for riid
164  *
165  * RETURNS
166  *    Success: S_OK
167  *    Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
168  *             E_UNEXPECTED
169  */
170 DWORD WINAPI AMSTREAM_DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
171 {
172     int i;
173     IClassFactoryImpl *factory;
174     
175     TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
176     
177     if ( !IsEqualGUID( &IID_IClassFactory, riid )
178          && ! IsEqualGUID( &IID_IUnknown, riid) )
179         return E_NOINTERFACE;
180
181     for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
182     {
183         if (IsEqualGUID(object_creation[i].clsid, rclsid))
184             break;
185     }
186
187     if (i == sizeof(object_creation)/sizeof(object_creation[0]))
188     {
189         FIXME("%s: no class found.\n", debugstr_guid(rclsid));
190         return CLASS_E_CLASSNOTAVAILABLE;
191     }
192
193     factory = HeapAlloc(GetProcessHeap(), 0, sizeof(*factory));
194     if (factory == NULL) return E_OUTOFMEMORY;
195
196     factory->ITF_IClassFactory.lpVtbl = &DSCF_Vtbl;
197     factory->ref = 1;
198
199     factory->pfnCreateInstance = object_creation[i].pfnCreateInstance;
200
201     *ppv = &(factory->ITF_IClassFactory);
202     return S_OK;
203 }
204
205 /***********************************************************************
206  *              DllCanUnloadNow (AMSTREAM.@)
207  */
208 HRESULT WINAPI AMSTREAM_DllCanUnloadNow()
209 {
210     return dll_ref != 0 ? S_FALSE : S_OK;
211 }