msdmo: Improve the prototypes Win64 compatibility by replacing long with LONG.
[wine.git] / dlls / msdmo / dmoreg.c
blobff0440af7ccec0889d328976dc8b55aa30e348fa
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 #include <stdarg.h>
22 #include "windef.h"
23 #include "winbase.h"
24 #include "winuser.h"
25 #include "winerror.h"
26 #include "winreg.h"
27 #include "objbase.h"
28 #include "wine/unicode.h"
29 #include "wine/debug.h"
30 #include "initguid.h"
31 #include "dmo.h"
33 WINE_DEFAULT_DEBUG_CHANNEL(msdmo);
36 static const WCHAR szDMORootKey[] =
38 'D','i','r','e','c','t','S','h','o','w','\\',
39 'M','e','d','i','a','O','b','j','e','c','t','s',0
40 };
42 static const WCHAR szDMOInputType[] =
44 'I','n','p','u','t','T','y','p','e','s',0
47 static const WCHAR szDMOOutputType[] =
49 'O','u','t','p','u','t','T','y','p','e','s',0
52 static const WCHAR szDMOKeyed[] =
54 'K','e','y','e','d',0
57 static const WCHAR szDMOCategories[] =
59 'C','a','t','e','g','o','r','i','e','s',0
62 static const WCHAR szGUIDFmt[] =
64 '%','0','8','X','-','%','0','4','X','-','%','0','4','X','-','%','0',
65 '2','X','%','0','2','X','-','%','0','2','X','%','0','2','X','%','0','2',
66 'X','%','0','2','X','%','0','2','X','%','0','2','X',0
69 static const WCHAR szCat3Fmt[] =
71 '%','s','\\','%','s','\\','%','s',0
74 static const WCHAR szCat2Fmt[] =
76 '%','s','\\','%','s',0
79 typedef struct
81 const IEnumDMOVtbl *lpVtbl;
82 LONG ref;
83 DWORD index;
84 const GUID* guidCategory;
85 DWORD dwFlags;
86 DWORD cInTypes;
87 DMO_PARTIAL_MEDIATYPE *pInTypes;
88 DWORD cOutTypes;
89 DMO_PARTIAL_MEDIATYPE *pOutTypes;
90 HKEY hkey;
91 } IEnumDMOImpl;
93 static const IEnumDMOVtbl edmovt;
95 static LPWSTR GUIDToString(LPWSTR lpwstr, REFGUID lpcguid)
97 wsprintfW(lpwstr, szGUIDFmt, lpcguid->Data1, lpcguid->Data2,
98 lpcguid->Data3, lpcguid->Data4[0], lpcguid->Data4[1],
99 lpcguid->Data4[2], lpcguid->Data4[3], lpcguid->Data4[4],
100 lpcguid->Data4[5], lpcguid->Data4[6], lpcguid->Data4[7]);
102 return lpwstr;
105 static BOOL IsMediaTypeEqual(const DMO_PARTIAL_MEDIATYPE* mt1, const DMO_PARTIAL_MEDIATYPE* mt2)
108 return (IsEqualCLSID(&mt1->type, &mt2->type) ||
109 IsEqualCLSID(&mt2->type, &GUID_NULL) ||
110 IsEqualCLSID(&mt1->type, &GUID_NULL)) &&
111 (IsEqualCLSID(&mt1->subtype, &mt2->subtype) ||
112 IsEqualCLSID(&mt2->subtype, &GUID_NULL) ||
113 IsEqualCLSID(&mt1->subtype, &GUID_NULL));
116 /***************************************************************
117 * DMORegister
119 * Register a DirectX Media Object.
121 HRESULT WINAPI DMORegister(
122 LPCWSTR szName,
123 REFCLSID clsidDMO,
124 REFGUID guidCategory,
125 DWORD dwFlags,
126 DWORD cInTypes,
127 const DMO_PARTIAL_MEDIATYPE *pInTypes,
128 DWORD cOutTypes,
129 const DMO_PARTIAL_MEDIATYPE *pOutTypes
132 WCHAR szguid[64];
133 HRESULT hres;
134 HKEY hrkey = 0;
135 HKEY hkey = 0;
136 HKEY hckey = 0;
137 HKEY hclskey = 0;
139 TRACE("%s\n", debugstr_w(szName));
141 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
142 if (ERROR_SUCCESS != hres)
143 goto lend;
145 /* Create clsidDMO key under MediaObjects */
146 hres = RegCreateKeyExW(hrkey, GUIDToString(szguid, clsidDMO), 0, NULL,
147 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, NULL);
148 if (ERROR_SUCCESS != hres)
149 goto lend;
151 /* Set default Name value */
152 hres = RegSetValueExW(hkey, NULL, 0, REG_SZ, (const BYTE*) szName,
153 (strlenW(szName) + 1)) * sizeof(WCHAR);
154 /* Set InputTypes */
155 hres = RegSetValueExW(hkey, szDMOInputType, 0, REG_BINARY,
156 (const BYTE*) pInTypes, cInTypes * sizeof(DMO_PARTIAL_MEDIATYPE));
157 /* Set OutputTypes */
158 hres = RegSetValueExW(hkey, szDMOOutputType, 0, REG_BINARY,
159 (const BYTE*) pOutTypes, cOutTypes * sizeof(DMO_PARTIAL_MEDIATYPE));
161 if (dwFlags & DMO_REGISTERF_IS_KEYED)
163 /* Create Keyed key */
164 hres = RegCreateKeyExW(hkey, szDMOKeyed, 0, NULL,
165 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hckey, NULL);
166 if (ERROR_SUCCESS != hres)
167 goto lend;
168 RegCloseKey(hckey);
171 /* Register the category */
172 hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
173 if (ERROR_SUCCESS != hres)
174 goto lend;
176 RegCloseKey(hkey);
178 hres = RegOpenKeyExW(hckey, GUIDToString(szguid, guidCategory), 0, KEY_WRITE, &hkey);
179 if (ERROR_SUCCESS != hres)
180 goto lend;
181 hres = RegCreateKeyExW(hkey, GUIDToString(szguid, clsidDMO), 0, NULL,
182 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hclskey, NULL);
183 if (ERROR_SUCCESS != hres)
184 goto lend;
186 lend:
187 if (hkey)
188 RegCloseKey(hkey);
189 if (hckey)
190 RegCloseKey(hckey);
191 if (hclskey)
192 RegCloseKey(hclskey);
193 if (hrkey)
194 RegCloseKey(hrkey);
196 TRACE(" hresult=0x%08x\n", hres);
197 return hres;
201 /***************************************************************
202 * DMOUnregister
204 * Unregister a DirectX Media Object.
206 HRESULT WINAPI DMOUnregister(REFCLSID clsidDMO, REFGUID guidCategory)
208 HRESULT hres;
209 WCHAR szguid[64];
210 HKEY hrkey = 0;
211 HKEY hckey = 0;
213 GUIDToString(szguid, clsidDMO);
215 TRACE("%s %p\n", debugstr_w(szguid), guidCategory);
217 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
218 if (ERROR_SUCCESS != hres)
219 goto lend;
221 hres = RegDeleteKeyW(hrkey, szguid);
222 if (ERROR_SUCCESS != hres)
223 goto lend;
225 hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
226 if (ERROR_SUCCESS != hres)
227 goto lend;
229 hres = RegDeleteKeyW(hckey, szguid);
230 if (ERROR_SUCCESS != hres)
231 goto lend;
233 lend:
234 if (hckey)
235 RegCloseKey(hckey);
236 if (hrkey)
237 RegCloseKey(hrkey);
239 return hres;
243 /***************************************************************
244 * DMOGetName
246 * Get DMP Name from the registry
248 HRESULT WINAPI DMOGetName(REFCLSID clsidDMO, WCHAR* szName)
250 WCHAR szguid[64];
251 HRESULT hres;
252 HKEY hrkey = 0;
253 HKEY hkey = 0;
254 DWORD count;
256 TRACE("%s\n", debugstr_guid(clsidDMO));
258 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey,
259 0, KEY_READ, &hrkey);
260 if (ERROR_SUCCESS != hres)
261 goto lend;
263 hres = RegOpenKeyExW(hrkey, GUIDToString(szguid, clsidDMO),
264 0, KEY_READ, &hkey);
265 if (ERROR_SUCCESS != hres)
266 goto lend;
268 count = 80 * sizeof(WCHAR); /* 80 by API definition */
269 hres = RegQueryValueExW(hkey, NULL, NULL, NULL,
270 (LPBYTE) szName, &count);
272 TRACE(" szName=%s\n", debugstr_w(szName));
273 lend:
274 if (hkey)
275 RegCloseKey(hrkey);
276 if (hkey)
277 RegCloseKey(hkey);
279 return hres;
283 /**************************************************************************
284 * IEnumDMO_Destructor
286 static BOOL IEnumDMO_Destructor(IEnumDMO* iface)
288 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
290 TRACE("%p\n", This);
292 if (This->hkey)
293 RegCloseKey(This->hkey);
295 HeapFree(GetProcessHeap(), 0, This->pInTypes);
296 HeapFree(GetProcessHeap(), 0, This->pOutTypes);
298 return TRUE;
302 /**************************************************************************
303 * IEnumDMO_Constructor
305 static IEnumDMO * IEnumDMO_Constructor(
306 REFGUID guidCategory,
307 DWORD dwFlags,
308 DWORD cInTypes,
309 const DMO_PARTIAL_MEDIATYPE *pInTypes,
310 DWORD cOutTypes,
311 const DMO_PARTIAL_MEDIATYPE *pOutTypes)
313 UINT size;
314 IEnumDMOImpl* lpedmo;
315 BOOL ret = FALSE;
317 lpedmo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IEnumDMOImpl));
319 if (lpedmo)
321 lpedmo->ref = 1;
322 lpedmo->lpVtbl = &edmovt;
323 lpedmo->index = -1;
324 lpedmo->guidCategory = guidCategory;
325 lpedmo->dwFlags = dwFlags;
327 size = cInTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
328 lpedmo->pInTypes = HeapAlloc(GetProcessHeap(), 0, size);
329 if (!lpedmo->pInTypes)
330 goto lerr;
331 memcpy(lpedmo->pInTypes, pInTypes, size);
332 lpedmo->cInTypes = cInTypes;
334 size = cOutTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
335 lpedmo->pOutTypes = HeapAlloc(GetProcessHeap(), 0, size);
336 if (!lpedmo->pOutTypes)
337 goto lerr;
338 memcpy(lpedmo->pOutTypes, pOutTypes, size);
339 lpedmo->cOutTypes = cOutTypes;
341 /* If not filtering by category enum from media objects root */
342 if (IsEqualGUID(guidCategory, &GUID_NULL))
344 if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey,
345 0, KEY_READ, &lpedmo->hkey))
346 ret = TRUE;
348 else
350 WCHAR szguid[64];
351 WCHAR szKey[MAX_PATH];
353 wsprintfW(szKey, szCat3Fmt, szDMORootKey, szDMOCategories,
354 GUIDToString(szguid, guidCategory));
355 if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey,
356 0, KEY_READ, &lpedmo->hkey))
357 ret = TRUE;
360 lerr:
361 if(!ret)
363 IEnumDMO_Destructor((IEnumDMO*)lpedmo);
364 HeapFree(GetProcessHeap(),0,lpedmo);
365 lpedmo = NULL;
369 TRACE("returning %p\n", lpedmo);
371 return (IEnumDMO*)lpedmo;
375 /******************************************************************************
376 * IEnumDMO_fnAddRef
378 static ULONG WINAPI IEnumDMO_fnAddRef(IEnumDMO * iface)
380 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
381 return InterlockedIncrement(&This->ref);
385 /**************************************************************************
386 * EnumDMO_QueryInterface
388 static HRESULT WINAPI IEnumDMO_fnQueryInterface(
389 IEnumDMO* iface,
390 REFIID riid,
391 LPVOID *ppvObj)
393 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
395 *ppvObj = NULL;
397 if(IsEqualIID(riid, &IID_IUnknown))
398 *ppvObj = This;
399 else if(IsEqualIID(riid, &IID_IEnumDMO))
400 *ppvObj = (IEnumDMO*)This;
402 if(*ppvObj)
404 IEnumDMO_fnAddRef((IEnumDMO*)*ppvObj);
405 return S_OK;
408 return E_NOINTERFACE;
412 /******************************************************************************
413 * IEnumDMO_fnRelease
415 static ULONG WINAPI IEnumDMO_fnRelease(IEnumDMO * iface)
417 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
418 ULONG refCount = InterlockedDecrement(&This->ref);
420 if (!refCount)
422 IEnumDMO_Destructor((IEnumDMO*)This);
423 HeapFree(GetProcessHeap(),0,This);
425 return refCount;
429 /******************************************************************************
430 * IEnumDMO_fnNext
432 static HRESULT WINAPI IEnumDMO_fnNext(
433 IEnumDMO * iface,
434 DWORD cItemsToFetch,
435 CLSID * pCLSID,
436 WCHAR ** Names,
437 DWORD * pcItemsFetched)
439 FILETIME ft;
440 HKEY hkey;
441 WCHAR szNextKey[MAX_PATH];
442 WCHAR szKey[MAX_PATH];
443 WCHAR szValue[MAX_PATH];
444 DWORD len;
445 UINT count = 0;
446 HRESULT hres = S_OK;
448 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
450 TRACE("%d\n", cItemsToFetch);
452 if (!pCLSID || !Names || !pcItemsFetched)
453 return E_POINTER;
455 while (count < cItemsToFetch)
457 This->index++;
459 hres = RegEnumKeyExW(This->hkey, This->index, szNextKey, &len, NULL, NULL, NULL, &ft);
460 if (hres != ERROR_SUCCESS)
461 break;
463 TRACE("found %s\n", debugstr_w(szNextKey));
465 if (This->dwFlags & DMO_REGISTERF_IS_KEYED)
467 wsprintfW(szKey, szCat3Fmt, szDMORootKey, szNextKey, szDMOKeyed);
468 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
469 if (ERROR_SUCCESS != hres)
470 continue;
471 RegCloseKey(hkey);
474 wsprintfW(szKey, szCat2Fmt, szDMORootKey, szNextKey);
475 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
477 if (This->pInTypes)
479 UINT i, j;
480 DWORD cInTypes;
481 DMO_PARTIAL_MEDIATYPE* pInTypes;
483 len = MAX_PATH * sizeof(WCHAR);
484 hres = RegQueryValueExW(hkey, szDMOInputType, NULL, NULL, (LPBYTE) szValue, &len);
485 if (ERROR_SUCCESS != hres)
487 RegCloseKey(hkey);
488 continue;
491 cInTypes = len / sizeof(DMO_PARTIAL_MEDIATYPE);
492 pInTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
494 for (i = 0; i < This->cInTypes; i++)
496 for (j = 0; j < cInTypes; j++)
498 if (IsMediaTypeEqual(&pInTypes[j], &This->pInTypes[i]))
499 break;
502 if (j >= cInTypes)
503 break;
506 if (i < This->cInTypes)
508 RegCloseKey(hkey);
509 continue;
513 if (This->pOutTypes)
515 UINT i, j;
516 DWORD cOutTypes;
517 DMO_PARTIAL_MEDIATYPE* pOutTypes;
519 len = MAX_PATH * sizeof(WCHAR);
520 hres = RegQueryValueExW(hkey, szDMOOutputType, NULL, NULL, (LPBYTE) szValue, &len);
521 if (ERROR_SUCCESS != hres)
523 RegCloseKey(hkey);
524 continue;
527 cOutTypes = len / sizeof(DMO_PARTIAL_MEDIATYPE);
528 pOutTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
530 for (i = 0; i < This->cOutTypes; i++)
532 for (j = 0; j < cOutTypes; j++)
534 if (IsMediaTypeEqual(&pOutTypes[j], &This->pOutTypes[i]))
535 break;
538 if (j >= cOutTypes)
539 break;
542 if (i < This->cOutTypes)
544 RegCloseKey(hkey);
545 continue;
549 /* Media object wasn't filtered so add it to return list */
550 Names[count] = NULL;
551 len = MAX_PATH * sizeof(WCHAR);
552 hres = RegQueryValueExW(hkey, NULL, NULL, NULL, (LPBYTE) szValue, &len);
553 if (ERROR_SUCCESS == hres)
555 Names[count] = HeapAlloc(GetProcessHeap(), 0, strlenW(szValue) + 1);
556 if (Names[count])
557 strcmpW(Names[count], szValue);
559 CLSIDFromString(szNextKey, &pCLSID[count]);
561 TRACE("found match %s %s\n", debugstr_w(szValue), debugstr_w(szNextKey));
562 RegCloseKey(hkey);
563 count++;
566 *pcItemsFetched = count;
567 if (*pcItemsFetched < cItemsToFetch)
568 hres = S_FALSE;
570 return hres;
574 /******************************************************************************
575 * IEnumDMO_fnSkip
577 static HRESULT WINAPI IEnumDMO_fnSkip(IEnumDMO * iface, DWORD cItemsToSkip)
579 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
581 This->index += cItemsToSkip;
583 return S_OK;
587 /******************************************************************************
588 * IEnumDMO_fnReset
590 static HRESULT WINAPI IEnumDMO_fnReset(IEnumDMO * iface)
592 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
594 This->index = -1;
596 return S_OK;
600 /******************************************************************************
601 * IEnumDMO_fnClone
603 static HRESULT WINAPI IEnumDMO_fnClone(IEnumDMO * iface, IEnumDMO **ppEnum)
605 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
607 FIXME("(%p)->() to (%p)->() E_NOTIMPL\n", This, ppEnum);
609 return E_NOTIMPL;
613 /***************************************************************
614 * DMOEnum
616 * Enumerate DirectX Media Objects in the registry.
618 HRESULT WINAPI DMOEnum(
619 REFGUID guidCategory,
620 DWORD dwFlags,
621 DWORD cInTypes,
622 const DMO_PARTIAL_MEDIATYPE *pInTypes,
623 DWORD cOutTypes,
624 const DMO_PARTIAL_MEDIATYPE *pOutTypes,
625 IEnumDMO **ppEnum)
627 HRESULT hres = E_FAIL;
629 TRACE("guidCategory=%p dwFlags=0x%08x cInTypes=%d cOutTypes=%d\n",
630 guidCategory, dwFlags, cInTypes, cOutTypes);
632 *ppEnum = IEnumDMO_Constructor(guidCategory, dwFlags, cInTypes,
633 pInTypes, cOutTypes, pOutTypes);
634 if (*ppEnum)
635 hres = S_OK;
637 return hres;
641 static const IEnumDMOVtbl edmovt =
643 IEnumDMO_fnQueryInterface,
644 IEnumDMO_fnAddRef,
645 IEnumDMO_fnRelease,
646 IEnumDMO_fnNext,
647 IEnumDMO_fnSkip,
648 IEnumDMO_fnReset,
649 IEnumDMO_fnClone,
653 HRESULT WINAPI DMOGetTypes(REFCLSID a, ULONG b, ULONG* c,
654 DMO_PARTIAL_MEDIATYPE* d, ULONG e,
655 ULONG* f, DMO_PARTIAL_MEDIATYPE* g)
657 FIXME("(%p,%u,%p,%p,%u,%p,%p),stub!\n",a,b,c,d,e,f,g);
659 return E_NOTIMPL;