1 /* 2 * Copyright 2002 Michael Günnewig 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 19 #include <stdarg.h> 20 21 #define COBJMACROS 22 23 #include "windef.h" 24 #include "winbase.h" 25 #include "wingdi.h" 26 #include "winuser.h" 27 #include "winerror.h" 28 #include "ole2.h" 29 #include "rpcproxy.h" 30 31 #include "initguid.h" 32 #include "vfw.h" 33 #include "avifile_private.h" 34 35 #include "wine/debug.h" 36 37 WINE_DEFAULT_DEBUG_CHANNEL(avifile); 38 39 HMODULE AVIFILE_hModule = NULL; 40 41 static BOOL AVIFILE_bLocked; 42 static UINT AVIFILE_uUseCount; 43 44 typedef struct 45 { 46 IClassFactory IClassFactory_iface; 47 LONG ref; 48 CLSID clsid; 49 } IClassFactoryImpl; 50 51 static inline IClassFactoryImpl *impl_from_IClassFactory(IClassFactory *iface) 52 { 53 return CONTAINING_RECORD(iface, IClassFactoryImpl, IClassFactory_iface); 54 } 55 56 static HRESULT WINAPI IClassFactory_fnQueryInterface(IClassFactory *iface, REFIID riid, 57 void **ppobj) 58 { 59 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(riid), ppobj); 60 61 if ((IsEqualGUID(&IID_IUnknown, riid)) || 62 (IsEqualGUID(&IID_IClassFactory, riid))) { 63 *ppobj = iface; 64 IClassFactory_AddRef(iface); 65 return S_OK; 66 } 67 68 return E_NOINTERFACE; 69 } 70 71 static ULONG WINAPI IClassFactory_fnAddRef(IClassFactory *iface) 72 { 73 IClassFactoryImpl *This = impl_from_IClassFactory(iface); 74 ULONG ref = InterlockedIncrement(&This->ref); 75 76 TRACE("(%p) ref = %u\n", This, ref); 77 return ref; 78 } 79 80 static ULONG WINAPI IClassFactory_fnRelease(IClassFactory *iface) 81 { 82 IClassFactoryImpl *This = impl_from_IClassFactory(iface); 83 ULONG ref = InterlockedDecrement(&This->ref); 84 85 TRACE("(%p) ref = %u\n", This, ref); 86 87 if(!ref) 88 HeapFree(GetProcessHeap(), 0, This); 89 90 return ref; 91 } 92 93 static HRESULT WINAPI IClassFactory_fnCreateInstance(IClassFactory *iface, IUnknown *pOuter, 94 REFIID riid, void **ppobj) 95 { 96 IClassFactoryImpl *This = impl_from_IClassFactory(iface); 97 98 TRACE("(%p,%p,%s,%p)\n", iface, pOuter, debugstr_guid(riid), 99 ppobj); 100 101 if (!ppobj) 102 return E_INVALIDARG; 103 *ppobj = NULL; 104 105 if (pOuter && !IsEqualGUID(&IID_IUnknown, riid)) 106 return E_INVALIDARG; 107 108 if (IsEqualGUID(&CLSID_AVIFile, &This->clsid)) 109 return AVIFILE_CreateAVIFile(pOuter, riid, ppobj); 110 if (IsEqualGUID(&CLSID_WAVFile, &This->clsid)) 111 return AVIFILE_CreateWAVFile(pOuter, riid, ppobj); 112 113 if (pOuter) 114 return CLASS_E_NOAGGREGATION; 115 116 if (IsEqualGUID(&CLSID_ICMStream, &This->clsid)) 117 return AVIFILE_CreateICMStream(riid,ppobj); 118 if (IsEqualGUID(&CLSID_ACMStream, &This->clsid)) 119 return AVIFILE_CreateACMStream(riid,ppobj); 120 121 return E_NOINTERFACE; 122 } 123 124 static HRESULT WINAPI IClassFactory_fnLockServer(IClassFactory *iface, BOOL dolock) 125 { 126 TRACE("(%p,%d)\n",iface,dolock); 127 128 AVIFILE_bLocked = dolock; 129 130 return S_OK; 131 } 132 133 static const IClassFactoryVtbl iclassfact = { 134 IClassFactory_fnQueryInterface, 135 IClassFactory_fnAddRef, 136 IClassFactory_fnRelease, 137 IClassFactory_fnCreateInstance, 138 IClassFactory_fnLockServer 139 }; 140 141 static HRESULT AVIFILE_CreateClassFactory(const CLSID *clsid, const IID *riid, void **ppv) 142 { 143 IClassFactoryImpl *cf; 144 HRESULT hr; 145 146 *ppv = NULL; 147 148 cf = HeapAlloc(GetProcessHeap(), 0, sizeof(*cf)); 149 if (!cf) 150 return E_OUTOFMEMORY; 151 152 cf->IClassFactory_iface.lpVtbl = &iclassfact; 153 cf->ref = 1; 154 cf->clsid = *clsid; 155 156 hr = IClassFactory_QueryInterface(&cf->IClassFactory_iface, riid, ppv); 157 IClassFactory_Release(&cf->IClassFactory_iface); 158 159 return hr; 160 } 161 162 LPCWSTR AVIFILE_BasenameW(LPCWSTR szPath) 163 { 164 #define SLASH(w) ((w) == '/' || (w) == '\\') 165 166 LPCWSTR szCur; 167 168 for (szCur = szPath + lstrlenW(szPath); 169 szCur > szPath && !SLASH(*szCur) && *szCur != ':';) 170 szCur--; 171 172 if (szCur == szPath) 173 return szCur; 174 else 175 return szCur + 1; 176 177 #undef SLASH 178 } 179 180 /*********************************************************************** 181 * DllGetClassObject (AVIFIL32.@) 182 */ 183 HRESULT WINAPI DllGetClassObject(REFCLSID pclsid, REFIID piid, LPVOID *ppv) 184 { 185 HRESULT hr; 186 187 TRACE("(%s,%s,%p)\n", debugstr_guid(pclsid), debugstr_guid(piid), ppv); 188 189 if (pclsid == NULL || piid == NULL || ppv == NULL) 190 return E_FAIL; 191 192 hr = AVIFILE_CreateClassFactory(pclsid,piid,ppv); 193 if (SUCCEEDED(hr)) 194 return hr; 195 196 return avifil32_DllGetClassObject(pclsid,piid,ppv); 197 } 198 199 /***************************************************************************** 200 * DllCanUnloadNow (AVIFIL32.@) 201 */ 202 HRESULT WINAPI DllCanUnloadNow(void) 203 { 204 return ((AVIFILE_bLocked || AVIFILE_uUseCount) ? S_FALSE : S_OK); 205 } 206 207 /***************************************************************************** 208 * DllMain [AVIFIL32.init] 209 */ 210 BOOL WINAPI DllMain(HINSTANCE hInstDll, DWORD fdwReason, LPVOID lpvReserved) 211 { 212 TRACE("(%p,%d,%p)\n", hInstDll, fdwReason, lpvReserved); 213 214 switch (fdwReason) { 215 case DLL_PROCESS_ATTACH: 216 DisableThreadLibraryCalls(hInstDll); 217 AVIFILE_hModule = hInstDll; 218 break; 219 }; 220 221 return TRUE; 222 } 223 224 /*********************************************************************** 225 * DllRegisterServer (AVIFIL32.@) 226 */ 227 HRESULT WINAPI DllRegisterServer(void) 228 { 229 return __wine_register_resources( AVIFILE_hModule ); 230 } 231 232 /*********************************************************************** 233 * DllUnregisterServer (AVIFIL32.@) 234 */ 235 HRESULT WINAPI DllUnregisterServer(void) 236 { 237 return __wine_unregister_resources( AVIFILE_hModule ); 238 } 239