shlwapi: Fix memory leak in test.
[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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22  */
23
24 #include <stdarg.h>
25 #include <string.h>
26
27 #define COBJMACROS
28
29 #include "windef.h"
30 #include "winbase.h"
31 #include "winuser.h"
32 #include "winerror.h"
33
34 #include "ole2.h"
35
36 #include "amstream_private.h"
37 #include "amstream.h"
38
39 #include "wine/debug.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(amstream);
42
43 static DWORD dll_ref = 0;
44
45 /* For the moment, do nothing here. */
46 BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv)
47 {
48     switch(fdwReason) {
49         case DLL_PROCESS_ATTACH:
50             DisableThreadLibraryCalls(hInstDLL);
51             break;
52         case DLL_PROCESS_DETACH:
53             break;
54     }
55     return TRUE;
56 }
57
58 /******************************************************************************
59  * Multimedia Streams ClassFactory
60  */
61 typedef struct {
62     IClassFactory ITF_IClassFactory;
63
64     LONG ref;
65     HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
66 } IClassFactoryImpl;
67
68 struct object_creation_info
69 {
70     const CLSID *clsid;
71     HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
72 };
73
74 static const struct object_creation_info object_creation[] =
75 {
76     { &CLSID_AMMultiMediaStream, AM_create },
77 };
78
79 static HRESULT WINAPI
80 AMCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid,LPVOID *ppobj)
81 {
82     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
83
84     if (IsEqualGUID(riid, &IID_IUnknown)
85         || IsEqualGUID(riid, &IID_IClassFactory))
86     {
87         IClassFactory_AddRef(iface);
88         *ppobj = This;
89         return S_OK;
90     }
91
92     WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppobj);
93     return E_NOINTERFACE;
94 }
95
96 static ULONG WINAPI AMCF_AddRef(LPCLASSFACTORY iface)
97 {
98     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
99     return InterlockedIncrement(&This->ref);
100 }
101
102 static ULONG WINAPI AMCF_Release(LPCLASSFACTORY iface)
103 {
104     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
105
106     ULONG ref = InterlockedDecrement(&This->ref);
107
108     if (ref == 0)
109         HeapFree(GetProcessHeap(), 0, This);
110
111     return ref;
112 }
113
114
115 static HRESULT WINAPI AMCF_CreateInstance(LPCLASSFACTORY iface, LPUNKNOWN pOuter,
116                                           REFIID riid, LPVOID *ppobj)
117 {
118     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
119     HRESULT hres;
120     LPUNKNOWN punk;
121     
122     TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);
123
124     *ppobj = NULL;
125     hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk);
126     if (SUCCEEDED(hres)) {
127         hres = IUnknown_QueryInterface(punk, riid, ppobj);
128         IUnknown_Release(punk);
129     }
130     return hres;
131 }
132
133 static HRESULT WINAPI AMCF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
134 {
135     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
136     FIXME("(%p)->(%d),stub!\n",This,dolock);
137     return S_OK;
138 }
139
140 static const IClassFactoryVtbl DSCF_Vtbl =
141 {
142     AMCF_QueryInterface,
143     AMCF_AddRef,
144     AMCF_Release,
145     AMCF_CreateInstance,
146     AMCF_LockServer
147 };
148
149 /*******************************************************************************
150  * DllGetClassObject [AMSTREAM.@]
151  * Retrieves class object from a DLL object
152  *
153  * NOTES
154  *    Docs say returns STDAPI
155  *
156  * PARAMS
157  *    rclsid [I] CLSID for the class object
158  *    riid   [I] Reference to identifier of interface for class object
159  *    ppv    [O] Address of variable to receive interface pointer for riid
160  *
161  * RETURNS
162  *    Success: S_OK
163  *    Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
164  *             E_UNEXPECTED
165  */
166 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
167 {
168     unsigned int i;
169     IClassFactoryImpl *factory;
170     
171     TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
172     
173     if ( !IsEqualGUID( &IID_IClassFactory, riid )
174          && ! IsEqualGUID( &IID_IUnknown, riid) )
175         return E_NOINTERFACE;
176
177     for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
178     {
179         if (IsEqualGUID(object_creation[i].clsid, rclsid))
180             break;
181     }
182
183     if (i == sizeof(object_creation)/sizeof(object_creation[0]))
184     {
185         FIXME("%s: no class found.\n", debugstr_guid(rclsid));
186         return CLASS_E_CLASSNOTAVAILABLE;
187     }
188
189     factory = HeapAlloc(GetProcessHeap(), 0, sizeof(*factory));
190     if (factory == NULL) return E_OUTOFMEMORY;
191
192     factory->ITF_IClassFactory.lpVtbl = &DSCF_Vtbl;
193     factory->ref = 1;
194
195     factory->pfnCreateInstance = object_creation[i].pfnCreateInstance;
196
197     *ppv = &(factory->ITF_IClassFactory);
198     return S_OK;
199 }
200
201 /***********************************************************************
202  *              DllCanUnloadNow (AMSTREAM.@)
203  */
204 HRESULT WINAPI DllCanUnloadNow(void)
205 {
206     return dll_ref != 0 ? S_FALSE : S_OK;
207 }