Implemented DllCanUnloadNow.
[wine/multimedia.git] / dlls / msdmo / dmoreg.c
blob51d8430b59e48ddb1d4645ed562ce0cd21d59829
1 /*
2 * Copyright (C) 2003 Michael Günnewig
3 * Copyright (C) 2003 CodeWeavers Inc. (Ulrich Czekalla)
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.
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.
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 #define COM_NO_WINDOWS_H
21 #include <stdarg.h>
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winuser.h"
26 #include "winerror.h"
27 #include "winreg.h"
28 #include "objbase.h"
29 #include "dmo.h"
31 #include "wine/unicode.h"
32 #include "wine/debug.h"
34 WINE_DEFAULT_DEBUG_CHANNEL(msdmo);
37 static const WCHAR szDMORootKey[] =
39 'D','i','r','e','c','t','S','h','o','w','\\',
40 'M','e','d','i','a','O','b','j','e','c','t','s',0
41 };
43 static const WCHAR szDMOInputType[] =
45 'I','n','p','u','t','T','y','p','e','s',0
48 static const WCHAR szDMOOutputType[] =
50 'O','u','t','p','u','t','T','y','p','e','s',0
53 static const WCHAR szDMOKeyed[] =
55 'K','e','y','e','d',0
58 static const WCHAR szDMOCategories[] =
60 'C','a','t','e','g','o','r','i','e','s',0
63 static const WCHAR szGUIDFmt[] =
65 '%','0','8','X','-','%','0','4','X','-','%','0','4','X','-','%','0',
66 '2','X','%','0','2','X','-','%','0','2','X','%','0','2','X','%','0','2',
67 'X','%','0','2','X','%','0','2','X','%','0','2','X',0
70 static const WCHAR szCat3Fmt[] =
72 '%','s','\\','%','s','\\','%','s',0
75 static const WCHAR szCat2Fmt[] =
77 '%','s','\\','%','s',0
80 typedef struct
82 IEnumDMOVtbl *lpVtbl;
83 DWORD ref;
84 DWORD index;
85 const GUID* guidCategory;
86 DWORD dwFlags;
87 DWORD cInTypes;
88 DMO_PARTIAL_MEDIATYPE *pInTypes;
89 DWORD cOutTypes;
90 DMO_PARTIAL_MEDIATYPE *pOutTypes;
91 HKEY hkey;
92 } IEnumDMOImpl;
94 const GUID IID_IEnumDMO = { 0x2c3cd98a, 0x2bfa, 0x4a53,
95 { 0x9c, 0x27, 0x52, 0x49, 0xba, 0x64, 0xba, 0x0f}};
97 static struct IEnumDMOVtbl edmovt;
99 static LPWSTR GUIDToString(LPWSTR lpwstr, REFGUID lpcguid)
101 wsprintfW(lpwstr, szGUIDFmt, lpcguid->Data1, lpcguid->Data2,
102 lpcguid->Data3, lpcguid->Data4[0], lpcguid->Data4[1],
103 lpcguid->Data4[2], lpcguid->Data4[3], lpcguid->Data4[4],
104 lpcguid->Data4[5], lpcguid->Data4[6], lpcguid->Data4[7]);
106 return lpwstr;
109 static BOOL IsMediaTypeEqual(DMO_PARTIAL_MEDIATYPE* mt1, DMO_PARTIAL_MEDIATYPE* mt2)
112 return (IsEqualCLSID(&mt1->type, &mt2->type) ||
113 IsEqualCLSID(&mt2->type, &GUID_NULL) ||
114 IsEqualCLSID(&mt1->type, &GUID_NULL)) &&
115 (IsEqualCLSID(&mt1->subtype, &mt2->subtype) ||
116 IsEqualCLSID(&mt2->subtype, &GUID_NULL) ||
117 IsEqualCLSID(&mt1->subtype, &GUID_NULL));
120 /***************************************************************
121 * DMORegister
123 * Register a DirectX Media Object.
125 HRESULT WINAPI DMORegister(
126 LPCWSTR szName,
127 REFCLSID clsidDMO,
128 REFGUID guidCategory,
129 DWORD dwFlags,
130 DWORD cInTypes,
131 const DMO_PARTIAL_MEDIATYPE *pInTypes,
132 DWORD cOutTypes,
133 const DMO_PARTIAL_MEDIATYPE *pOutTypes
136 WCHAR szguid[64];
137 HRESULT hres;
138 HKEY hrkey = 0;
139 HKEY hkey = 0;
140 HKEY hckey = 0;
141 HKEY hclskey = 0;
143 TRACE("%s\n", debugstr_w(szName));
145 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
146 if (ERROR_SUCCESS != hres)
147 goto lend;
149 /* Create clsidDMO key under MediaObjects */
150 hres = RegCreateKeyExW(hrkey, GUIDToString(szguid, clsidDMO), 0, NULL,
151 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, NULL);
152 if (ERROR_SUCCESS != hres)
153 goto lend;
155 /* Set default Name value */
156 hres = RegSetValueExW(hkey, NULL, 0, REG_SZ, (const BYTE*) szName,
157 (strlenW(szName) + 1)) * sizeof(WCHAR);
158 /* Set InputTypes */
159 hres = RegSetValueExW(hkey, szDMOInputType, 0, REG_BINARY,
160 (const BYTE*) pInTypes, cInTypes * sizeof(DMO_PARTIAL_MEDIATYPE));
161 /* Set OutputTypes */
162 hres = RegSetValueExW(hkey, szDMOOutputType, 0, REG_BINARY,
163 (const BYTE*) pOutTypes, cOutTypes * sizeof(DMO_PARTIAL_MEDIATYPE));
165 if (dwFlags & DMO_REGISTERF_IS_KEYED)
167 /* Create Keyed key */
168 hres = RegCreateKeyExW(hkey, szDMOKeyed, 0, NULL,
169 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hckey, NULL);
170 if (ERROR_SUCCESS != hres)
171 goto lend;
172 RegCloseKey(hckey);
175 /* Register the category */
176 hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
177 if (ERROR_SUCCESS != hres)
178 goto lend;
180 RegCloseKey(hkey);
182 hres = RegOpenKeyExW(hckey, GUIDToString(szguid, guidCategory), 0, KEY_WRITE, &hkey);
183 if (ERROR_SUCCESS != hres)
184 goto lend;
185 hres = RegCreateKeyExW(hkey, GUIDToString(szguid, clsidDMO), 0, NULL,
186 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hclskey, NULL);
187 if (ERROR_SUCCESS != hres)
188 goto lend;
190 lend:
191 if (hkey)
192 RegCloseKey(hkey);
193 if (hckey)
194 RegCloseKey(hckey);
195 if (hclskey)
196 RegCloseKey(hclskey);
197 if (hrkey)
198 RegCloseKey(hrkey);
200 TRACE(" hresult=0x%08lx\n", hres);
201 return hres;
205 /***************************************************************
206 * DMOUnregister
208 * Unregister a DirectX Media Object.
210 HRESULT WINAPI DMOUnregister(REFCLSID clsidDMO, REFGUID guidCategory)
212 HRESULT hres;
213 WCHAR szguid[64];
214 HKEY hrkey = 0;
215 HKEY hckey = 0;
217 GUIDToString(szguid, clsidDMO);
219 TRACE("%s %p\n", debugstr_w(szguid), guidCategory);
221 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
222 if (ERROR_SUCCESS != hres)
223 goto lend;
225 hres = RegDeleteKeyW(hrkey, szguid);
226 if (ERROR_SUCCESS != hres)
227 goto lend;
229 hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
230 if (ERROR_SUCCESS != hres)
231 goto lend;
233 hres = RegDeleteKeyW(hckey, szguid);
234 if (ERROR_SUCCESS != hres)
235 goto lend;
237 lend:
238 if (hckey)
239 RegCloseKey(hckey);
240 if (hrkey)
241 RegCloseKey(hrkey);
243 return hres;
247 /***************************************************************
248 * DMOGetName
250 * Get DMP Name from the registry
252 HRESULT WINAPI DMOGetName(REFCLSID clsidDMO, WCHAR* szName)
254 WCHAR szguid[64];
255 HRESULT hres;
256 HKEY hrkey = 0;
257 HKEY hkey = 0;
258 DWORD count;
260 TRACE("%s\n", debugstr_guid(clsidDMO));
262 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey,
263 0, KEY_READ, &hrkey);
264 if (ERROR_SUCCESS != hres)
265 goto lend;
267 hres = RegOpenKeyExW(hrkey, GUIDToString(szguid, clsidDMO),
268 0, KEY_READ, &hkey);
269 if (ERROR_SUCCESS != hres)
270 goto lend;
272 count = 80 * sizeof(WCHAR); /* 80 by API definition */
273 hres = RegQueryValueExW(hkey, NULL, NULL, NULL,
274 (LPBYTE) szName, &count);
276 TRACE(" szName=%s\n", debugstr_w(szName));
277 lend:
278 if (hkey)
279 RegCloseKey(hrkey);
280 if (hkey)
281 RegCloseKey(hkey);
283 return hres;
287 /**************************************************************************
288 * IEnumDMO_Destructor
290 static BOOL IEnumDMO_Destructor(IEnumDMO* iface)
292 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
294 TRACE("%p\n", This);
296 if (This->hkey)
297 RegCloseKey(This->hkey);
299 HeapFree(GetProcessHeap(), 0, This->pInTypes);
300 HeapFree(GetProcessHeap(), 0, This->pOutTypes);
302 return TRUE;
306 /**************************************************************************
307 * IEnumDMO_Constructor
309 IEnumDMO * IEnumDMO_Constructor(
310 REFGUID guidCategory,
311 DWORD dwFlags,
312 DWORD cInTypes,
313 const DMO_PARTIAL_MEDIATYPE *pInTypes,
314 DWORD cOutTypes,
315 const DMO_PARTIAL_MEDIATYPE *pOutTypes)
317 UINT size;
318 IEnumDMOImpl* lpedmo;
319 BOOL ret = FALSE;
321 lpedmo = (IEnumDMOImpl*)HeapAlloc(GetProcessHeap(),
322 HEAP_ZERO_MEMORY, sizeof(IEnumDMOImpl));
324 if (lpedmo)
326 lpedmo->ref = 1;
327 lpedmo->lpVtbl = &edmovt;
328 lpedmo->index = -1;
329 lpedmo->guidCategory = guidCategory;
330 lpedmo->dwFlags = dwFlags;
332 size = cInTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
333 lpedmo->pInTypes = HeapAlloc(GetProcessHeap(), 0, size);
334 if (!lpedmo->pInTypes)
335 goto lerr;
336 memcpy(lpedmo->pInTypes, pInTypes, size);
337 lpedmo->cInTypes = cInTypes;
339 size = cOutTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
340 lpedmo->pOutTypes = HeapAlloc(GetProcessHeap(), 0, size);
341 if (!lpedmo->pOutTypes)
342 goto lerr;
343 memcpy(lpedmo->pOutTypes, pOutTypes, size);
344 lpedmo->cOutTypes = cOutTypes;
346 /* If not filtering by category enum from media objects root */
347 if (IsEqualGUID(guidCategory, &GUID_NULL))
349 if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey,
350 0, KEY_READ, &lpedmo->hkey))
351 ret = TRUE;
353 else
355 WCHAR szguid[64];
356 WCHAR szKey[MAX_PATH];
358 wsprintfW(szKey, szCat3Fmt, szDMORootKey, szDMOCategories,
359 GUIDToString(szguid, guidCategory));
360 if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey,
361 0, KEY_READ, &lpedmo->hkey))
362 ret = TRUE;
365 lerr:
366 if(!ret)
368 IEnumDMO_Destructor((IEnumDMO*)lpedmo);
369 HeapFree(GetProcessHeap(),0,lpedmo);
370 lpedmo = NULL;
374 TRACE("returning %p\n", lpedmo);
376 return (IEnumDMO*)lpedmo;
380 /******************************************************************************
381 * IEnumDMO_fnAddRef
383 static ULONG WINAPI IEnumDMO_fnAddRef(IEnumDMO * iface)
385 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
386 return ++(This->ref);
390 /**************************************************************************
391 * EnumDMO_QueryInterface
393 static HRESULT WINAPI IEnumDMO_fnQueryInterface(
394 IEnumDMO* iface,
395 REFIID riid,
396 LPVOID *ppvObj)
398 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
400 *ppvObj = NULL;
402 if(IsEqualIID(riid, &IID_IUnknown))
403 *ppvObj = This;
404 else if(IsEqualIID(riid, &IID_IEnumDMO))
405 *ppvObj = (IEnumDMO*)This;
407 if(*ppvObj)
409 IEnumDMO_fnAddRef((IEnumDMO*)*ppvObj);
410 return S_OK;
413 return E_NOINTERFACE;
417 /******************************************************************************
418 * IEnumDMO_fnRelease
420 static ULONG WINAPI IEnumDMO_fnRelease(IEnumDMO * iface)
422 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
424 if (!--(This->ref))
426 IEnumDMO_Destructor((IEnumDMO*)This);
427 HeapFree(GetProcessHeap(),0,This);
428 return 0;
431 return This->ref;
435 /******************************************************************************
436 * IEnumDMO_fnNext
438 static HRESULT WINAPI IEnumDMO_fnNext(
439 IEnumDMO * iface,
440 DWORD cItemsToFetch,
441 CLSID * pCLSID,
442 WCHAR ** Names,
443 DWORD * pcItemsFetched)
445 FILETIME ft;
446 HKEY hkey;
447 WCHAR szNextKey[MAX_PATH];
448 WCHAR szKey[MAX_PATH];
449 WCHAR szValue[MAX_PATH];
450 DWORD len;
451 UINT count = 0;
452 HRESULT hres = S_OK;
454 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
456 TRACE("%ld\n", cItemsToFetch);
458 if (!pCLSID || !Names || !pcItemsFetched)
459 return E_POINTER;
461 while (count < cItemsToFetch)
463 This->index++;
465 hres = RegEnumKeyExW(This->hkey, This->index, szNextKey, &len, NULL, NULL, NULL, &ft);
466 if (hres != ERROR_SUCCESS)
467 break;
469 TRACE("found %s\n", debugstr_w(szNextKey));
471 if (This->dwFlags & DMO_REGISTERF_IS_KEYED)
473 wsprintfW(szKey, szCat3Fmt, szDMORootKey, szNextKey, szDMOKeyed);
474 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
475 if (ERROR_SUCCESS != hres)
476 continue;
477 RegCloseKey(hkey);
480 wsprintfW(szKey, szCat2Fmt, szDMORootKey, szNextKey);
481 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
483 if (This->pInTypes)
485 UINT i, j;
486 DWORD cInTypes;
487 DMO_PARTIAL_MEDIATYPE* pInTypes;
489 len = MAX_PATH * sizeof(WCHAR);
490 hres = RegQueryValueExW(hkey, szDMOInputType, NULL, NULL, (LPBYTE) szValue, &len);
491 if (ERROR_SUCCESS != hres)
493 RegCloseKey(hkey);
494 continue;
497 cInTypes = len / sizeof(DMO_PARTIAL_MEDIATYPE);
498 pInTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
500 for (i = 0; i < This->cInTypes; i++)
502 for (j = 0; j < cInTypes; j++)
504 if (IsMediaTypeEqual(&pInTypes[j], &This->pInTypes[i]))
505 break;
508 if (j >= cInTypes)
509 break;
512 if (i < This->cInTypes)
514 RegCloseKey(hkey);
515 continue;
519 if (This->pOutTypes)
521 UINT i, j;
522 DWORD cOutTypes;
523 DMO_PARTIAL_MEDIATYPE* pOutTypes;
525 len = MAX_PATH * sizeof(WCHAR);
526 hres = RegQueryValueExW(hkey, szDMOOutputType, NULL, NULL, (LPBYTE) szValue, &len);
527 if (ERROR_SUCCESS != hres)
529 RegCloseKey(hkey);
530 continue;
533 cOutTypes = len / sizeof(DMO_PARTIAL_MEDIATYPE);
534 pOutTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
536 for (i = 0; i < This->cOutTypes; i++)
538 for (j = 0; j < cOutTypes; j++)
540 if (IsMediaTypeEqual(&pOutTypes[j], &This->pOutTypes[i]))
541 break;
544 if (j >= cOutTypes)
545 break;
548 if (i < This->cOutTypes)
550 RegCloseKey(hkey);
551 continue;
555 /* Media object wasn't filtered so add it to return list */
556 Names[count] = NULL;
557 len = MAX_PATH * sizeof(WCHAR);
558 hres = RegQueryValueExW(hkey, NULL, NULL, NULL, (LPBYTE) szValue, &len);
559 if (ERROR_SUCCESS == hres)
561 Names[count] = HeapAlloc(GetProcessHeap(), 0, strlenW(szValue) + 1);
562 if (Names[count])
563 strcmpW(Names[count], szValue);
565 CLSIDFromString(szNextKey, &pCLSID[count]);
567 TRACE("found match %s %s\n", debugstr_w(szValue), debugstr_w(szNextKey));
568 RegCloseKey(hkey);
569 count++;
572 *pcItemsFetched = count;
573 if (*pcItemsFetched < cItemsToFetch)
574 hres = S_FALSE;
576 return hres;
580 /******************************************************************************
581 * IEnumDMO_fnSkip
583 static HRESULT WINAPI IEnumDMO_fnSkip(IEnumDMO * iface, DWORD cItemsToSkip)
585 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
587 This->index += cItemsToSkip;
589 return S_OK;
593 /******************************************************************************
594 * IEnumDMO_fnReset
596 static HRESULT WINAPI IEnumDMO_fnReset(IEnumDMO * iface)
598 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
600 This->index = -1;
602 return S_OK;
606 /******************************************************************************
607 * IEnumDMO_fnClone
609 static HRESULT WINAPI IEnumDMO_fnClone(IEnumDMO * iface, IEnumDMO **ppEnum)
611 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
613 FIXME("(%p)->() to (%p)->() E_NOTIMPL\n", This, ppEnum);
615 return E_NOTIMPL;
619 /***************************************************************
620 * DMOEnum
622 * Enumerate DirectX Media Objects in the registry.
624 HRESULT WINAPI DMOEnum(
625 REFGUID guidCategory,
626 DWORD dwFlags,
627 DWORD cInTypes,
628 const DMO_PARTIAL_MEDIATYPE *pInTypes,
629 DWORD cOutTypes,
630 const DMO_PARTIAL_MEDIATYPE *pOutTypes,
631 IEnumDMO **ppEnum)
633 HRESULT hres = E_FAIL;
635 TRACE("guidCategory=%p dwFlags=0x%08lx cInTypes=%ld cOutTypes=%ld\n",
636 guidCategory, dwFlags, cInTypes, cOutTypes);
638 *ppEnum = IEnumDMO_Constructor(guidCategory, dwFlags, cInTypes,
639 pInTypes, cOutTypes, pOutTypes);
640 if (*ppEnum)
641 hres = S_OK;
643 return hres;
647 static IEnumDMOVtbl edmovt =
649 IEnumDMO_fnQueryInterface,
650 IEnumDMO_fnAddRef,
651 IEnumDMO_fnRelease,
652 IEnumDMO_fnNext,
653 IEnumDMO_fnSkip,
654 IEnumDMO_fnReset,
655 IEnumDMO_fnClone,
659 HRESULT WINAPI DMOGetTypes(REFCLSID a, unsigned long b, unsigned long* c,
660 DMO_PARTIAL_MEDIATYPE* d, unsigned long e,
661 unsigned long* f, DMO_PARTIAL_MEDIATYPE* g)
663 FIXME("(%p,%lu,%p,%p,%lu,%p,%p),stub!\n",a,b,c,d,e,f,g);
665 return E_NOTIMPL;