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
28 #include "wine/unicode.h"
29 #include "wine/debug.h"
33 WINE_DEFAULT_DEBUG_CHANNEL(msdmo);
35 #define MSDMO_MAJOR_VERSION 6
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
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[] =
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 static const WCHAR szToGuidFmt[] =
88 const IEnumDMOVtbl *lpVtbl;
91 const GUID* guidCategory;
94 DMO_PARTIAL_MEDIATYPE *pInTypes;
96 DMO_PARTIAL_MEDIATYPE *pOutTypes;
100 static HRESULT read_types(HKEY root, LPCWSTR key, ULONG *supplied, ULONG requested, DMO_PARTIAL_MEDIATYPE* types);
102 static const IEnumDMOVtbl edmovt;
104 static LPWSTR GUIDToString(LPWSTR lpwstr, REFGUID lpcguid)
106 wsprintfW(lpwstr, szGUIDFmt, lpcguid->Data1, lpcguid->Data2,
107 lpcguid->Data3, lpcguid->Data4[0], lpcguid->Data4[1],
108 lpcguid->Data4[2], lpcguid->Data4[3], lpcguid->Data4[4],
109 lpcguid->Data4[5], lpcguid->Data4[6], lpcguid->Data4[7]);
114 static BOOL IsMediaTypeEqual(const DMO_PARTIAL_MEDIATYPE* mt1, const DMO_PARTIAL_MEDIATYPE* mt2)
117 return (IsEqualCLSID(&mt1->type, &mt2->type) ||
118 IsEqualCLSID(&mt2->type, &GUID_NULL) ||
119 IsEqualCLSID(&mt1->type, &GUID_NULL)) &&
120 (IsEqualCLSID(&mt1->subtype, &mt2->subtype) ||
121 IsEqualCLSID(&mt2->subtype, &GUID_NULL) ||
122 IsEqualCLSID(&mt1->subtype, &GUID_NULL));
125 static HRESULT write_types(HKEY hkey, LPCWSTR name, const DMO_PARTIAL_MEDIATYPE* types, DWORD count)
128 if (MSDMO_MAJOR_VERSION > 5)
130 hres = RegSetValueExW(hkey, name, 0, REG_BINARY, (const BYTE*) types,
131 count* sizeof(DMO_PARTIAL_MEDIATYPE));
135 HKEY skey1,skey2,skey3;
139 hres = RegCreateKeyExW(hkey, name, 0, NULL, REG_OPTION_NON_VOLATILE,
140 KEY_WRITE, NULL, &skey1, NULL);
141 while (index < count)
143 GUIDToString(szGuidKey,&types[index].type);
144 hres = RegCreateKeyExW(skey1, szGuidKey, 0, NULL,
145 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &skey2, NULL);
146 GUIDToString(szGuidKey,&types[index].subtype);
147 hres = RegCreateKeyExW(skey2, szGuidKey, 0, NULL,
148 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &skey3, NULL);
159 /***************************************************************
160 * DMORegister (MSDMO.@)
162 * Register a DirectX Media Object.
164 HRESULT WINAPI DMORegister(
167 REFGUID guidCategory,
170 const DMO_PARTIAL_MEDIATYPE *pInTypes,
172 const DMO_PARTIAL_MEDIATYPE *pOutTypes
182 TRACE("%s\n", debugstr_w(szName));
184 hres = RegCreateKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, NULL,
185 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hrkey, NULL);
186 if (ERROR_SUCCESS != hres)
189 /* Create clsidDMO key under MediaObjects */
190 hres = RegCreateKeyExW(hrkey, GUIDToString(szguid, clsidDMO), 0, NULL,
191 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, NULL);
192 if (ERROR_SUCCESS != hres)
195 /* Set default Name value */
196 hres = RegSetValueExW(hkey, NULL, 0, REG_SZ, (const BYTE*) szName,
197 (strlenW(szName) + 1) * sizeof(WCHAR));
200 hres = write_types(hkey, szDMOInputType, pInTypes, cInTypes);
202 /* Set OutputTypes */
203 hres = write_types(hkey, szDMOOutputType, pOutTypes, cOutTypes);
205 if (dwFlags & DMO_REGISTERF_IS_KEYED)
207 /* Create Keyed key */
208 hres = RegCreateKeyExW(hkey, szDMOKeyed, 0, NULL,
209 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hckey, NULL);
210 if (ERROR_SUCCESS != hres)
215 /* Register the category */
216 hres = RegCreateKeyExW(hrkey, szDMOCategories, 0, NULL,
217 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hckey, NULL);
218 if (ERROR_SUCCESS != hres)
223 hres = RegCreateKeyExW(hckey, GUIDToString(szguid, guidCategory), 0, NULL,
224 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, NULL);
225 if (ERROR_SUCCESS != hres)
227 hres = RegCreateKeyExW(hkey, GUIDToString(szguid, clsidDMO), 0, NULL,
228 REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hclskey, NULL);
229 if (ERROR_SUCCESS != hres)
238 RegCloseKey(hclskey);
242 TRACE(" hresult=0x%08x\n", hres);
247 /***************************************************************
248 * DMOUnregister (MSDMO.@)
250 * Unregister a DirectX Media Object.
252 HRESULT WINAPI DMOUnregister(REFCLSID clsidDMO, REFGUID guidCategory)
259 GUIDToString(szguid, clsidDMO);
261 TRACE("%s %p\n", debugstr_w(szguid), guidCategory);
263 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
264 if (ERROR_SUCCESS != hres)
267 hres = RegDeleteKeyW(hrkey, szguid);
268 if (ERROR_SUCCESS != hres)
271 hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
272 if (ERROR_SUCCESS != hres)
275 hres = RegDeleteKeyW(hckey, szguid);
276 if (ERROR_SUCCESS != hres)
289 /***************************************************************
290 * DMOGetName (MSDMO.@)
292 * Get DMP Name from the registry
294 HRESULT WINAPI DMOGetName(REFCLSID clsidDMO, WCHAR szName[80])
302 TRACE("%s\n", debugstr_guid(clsidDMO));
304 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey,
305 0, KEY_READ, &hrkey);
306 if (ERROR_SUCCESS != hres)
309 hres = RegOpenKeyExW(hrkey, GUIDToString(szguid, clsidDMO),
311 if (ERROR_SUCCESS != hres)
314 count = sizeof(szName);
315 hres = RegQueryValueExW(hkey, NULL, NULL, NULL,
316 (LPBYTE) szName, &count);
318 TRACE(" szName=%s\n", debugstr_w(szName));
329 /**************************************************************************
330 * IEnumDMO_Destructor
332 static BOOL IEnumDMO_Destructor(IEnumDMO* iface)
334 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
339 RegCloseKey(This->hkey);
341 HeapFree(GetProcessHeap(), 0, This->pInTypes);
342 HeapFree(GetProcessHeap(), 0, This->pOutTypes);
348 /**************************************************************************
349 * IEnumDMO_Constructor
351 static IEnumDMO * IEnumDMO_Constructor(
352 REFGUID guidCategory,
355 const DMO_PARTIAL_MEDIATYPE *pInTypes,
357 const DMO_PARTIAL_MEDIATYPE *pOutTypes)
360 IEnumDMOImpl* lpedmo;
363 lpedmo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IEnumDMOImpl));
368 lpedmo->lpVtbl = &edmovt;
370 lpedmo->guidCategory = guidCategory;
371 lpedmo->dwFlags = dwFlags;
375 size = cInTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
376 lpedmo->pInTypes = HeapAlloc(GetProcessHeap(), 0, size);
377 if (!lpedmo->pInTypes)
379 memcpy(lpedmo->pInTypes, pInTypes, size);
380 lpedmo->cInTypes = cInTypes;
385 size = cOutTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
386 lpedmo->pOutTypes = HeapAlloc(GetProcessHeap(), 0, size);
387 if (!lpedmo->pOutTypes)
389 memcpy(lpedmo->pOutTypes, pOutTypes, size);
390 lpedmo->cOutTypes = cOutTypes;
393 /* If not filtering by category enum from media objects root */
394 if (IsEqualGUID(guidCategory, &GUID_NULL))
396 if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey,
397 0, KEY_READ, &lpedmo->hkey))
403 WCHAR szKey[MAX_PATH];
405 wsprintfW(szKey, szCat3Fmt, szDMORootKey, szDMOCategories,
406 GUIDToString(szguid, guidCategory));
407 if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey,
408 0, KEY_READ, &lpedmo->hkey))
415 IEnumDMO_Destructor((IEnumDMO*)lpedmo);
416 HeapFree(GetProcessHeap(),0,lpedmo);
421 TRACE("returning %p\n", lpedmo);
423 return (IEnumDMO*)lpedmo;
427 /******************************************************************************
430 static ULONG WINAPI IEnumDMO_fnAddRef(IEnumDMO * iface)
432 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
433 return InterlockedIncrement(&This->ref);
437 /**************************************************************************
438 * EnumDMO_QueryInterface
440 static HRESULT WINAPI IEnumDMO_fnQueryInterface(
445 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
449 if(IsEqualIID(riid, &IID_IUnknown))
451 else if(IsEqualIID(riid, &IID_IEnumDMO))
452 *ppvObj = (IEnumDMO*)This;
456 IEnumDMO_fnAddRef((IEnumDMO*)*ppvObj);
460 return E_NOINTERFACE;
464 /******************************************************************************
467 static ULONG WINAPI IEnumDMO_fnRelease(IEnumDMO * iface)
469 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
470 ULONG refCount = InterlockedDecrement(&This->ref);
474 IEnumDMO_Destructor((IEnumDMO*)This);
475 HeapFree(GetProcessHeap(),0,This);
481 /******************************************************************************
484 static HRESULT WINAPI IEnumDMO_fnNext(
489 DWORD * pcItemsFetched)
493 WCHAR szNextKey[MAX_PATH];
495 WCHAR szKey[MAX_PATH];
496 WCHAR szValue[MAX_PATH];
501 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
503 TRACE("--> (%p) %d %p %p %p\n", iface, cItemsToFetch, pCLSID, Names, pcItemsFetched);
505 if (!pCLSID || !Names || !pcItemsFetched)
508 while (count < cItemsToFetch)
513 hres = RegEnumKeyExW(This->hkey, This->index, szNextKey, &len, NULL, NULL, NULL, &ft);
514 if (hres != ERROR_SUCCESS)
517 TRACE("found %s\n", debugstr_w(szNextKey));
519 if (!(This->dwFlags & DMO_ENUMF_INCLUDE_KEYED))
521 wsprintfW(szKey, szCat3Fmt, szDMORootKey, szNextKey, szDMOKeyed);
522 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
523 if (ERROR_SUCCESS == hres)
526 /* Skip Keyed entries */
531 wsprintfW(szKey, szCat2Fmt, szDMORootKey, szNextKey);
532 hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
538 DMO_PARTIAL_MEDIATYPE* pInTypes;
540 hres = read_types(hkey, szDMOInputType, &cInTypes,
541 sizeof(szValue)/sizeof(DMO_PARTIAL_MEDIATYPE),
542 (DMO_PARTIAL_MEDIATYPE*)szValue);
544 if (ERROR_SUCCESS != hres)
550 pInTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
552 for (i = 0; i < This->cInTypes; i++)
554 for (j = 0; j < cInTypes; j++)
556 if (IsMediaTypeEqual(&pInTypes[j], &This->pInTypes[i]))
564 if (i < This->cInTypes)
575 DMO_PARTIAL_MEDIATYPE* pOutTypes;
577 hres = read_types(hkey, szDMOOutputType, &cOutTypes,
578 sizeof(szValue)/sizeof(DMO_PARTIAL_MEDIATYPE),
579 (DMO_PARTIAL_MEDIATYPE*)szValue);
581 if (ERROR_SUCCESS != hres)
587 pOutTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
589 for (i = 0; i < This->cOutTypes; i++)
591 for (j = 0; j < cOutTypes; j++)
593 if (IsMediaTypeEqual(&pOutTypes[j], &This->pOutTypes[i]))
601 if (i < This->cOutTypes)
608 /* Media object wasn't filtered so add it to return list */
610 len = MAX_PATH * sizeof(WCHAR);
611 hres = RegQueryValueExW(hkey, NULL, NULL, NULL, (LPBYTE) szValue, &len);
612 if (ERROR_SUCCESS == hres)
614 Names[count] = HeapAlloc(GetProcessHeap(), 0, strlenW(szValue) + 1);
616 strcmpW(Names[count], szValue);
618 wsprintfW(szGuidKey,szToGuidFmt,szNextKey);
619 CLSIDFromString(szGuidKey, &pCLSID[count]);
621 TRACE("found match %s %s\n", debugstr_w(szValue), debugstr_w(szNextKey));
626 *pcItemsFetched = count;
627 if (*pcItemsFetched < cItemsToFetch)
630 TRACE("<-- %i found\n",count);
635 /******************************************************************************
638 static HRESULT WINAPI IEnumDMO_fnSkip(IEnumDMO * iface, DWORD cItemsToSkip)
640 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
642 This->index += cItemsToSkip;
648 /******************************************************************************
651 static HRESULT WINAPI IEnumDMO_fnReset(IEnumDMO * iface)
653 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
661 /******************************************************************************
664 static HRESULT WINAPI IEnumDMO_fnClone(IEnumDMO * iface, IEnumDMO **ppEnum)
666 IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
668 FIXME("(%p)->() to (%p)->() E_NOTIMPL\n", This, ppEnum);
674 /***************************************************************
677 * Enumerate DirectX Media Objects in the registry.
679 HRESULT WINAPI DMOEnum(
680 REFGUID guidCategory,
683 const DMO_PARTIAL_MEDIATYPE *pInTypes,
685 const DMO_PARTIAL_MEDIATYPE *pOutTypes,
688 HRESULT hres = E_FAIL;
690 TRACE("guidCategory=%p dwFlags=0x%08x cInTypes=%d cOutTypes=%d\n",
691 guidCategory, dwFlags, cInTypes, cOutTypes);
693 *ppEnum = IEnumDMO_Constructor(guidCategory, dwFlags, cInTypes,
694 pInTypes, cOutTypes, pOutTypes);
702 static const IEnumDMOVtbl edmovt =
704 IEnumDMO_fnQueryInterface,
714 HRESULT read_types(HKEY root, LPCWSTR key, ULONG *supplied, ULONG requested, DMO_PARTIAL_MEDIATYPE* types )
717 if (MSDMO_MAJOR_VERSION > 5)
720 len = requested * sizeof(DMO_PARTIAL_MEDIATYPE);
721 ret = RegQueryValueExW(root, key, NULL, NULL, (LPBYTE) types, &len);
722 *supplied = len / sizeof(DMO_PARTIAL_MEDIATYPE);
730 if (ERROR_SUCCESS == RegOpenKeyExW(root, key, 0, KEY_READ, &hkey))
733 WCHAR szNextKey[MAX_PATH];
735 LONG rc = ERROR_SUCCESS;
737 while (rc == ERROR_SUCCESS)
739 len = MAX_PATH * sizeof(WCHAR);
740 rc = RegEnumKeyExW(hkey, index, szNextKey, &len, NULL, NULL, NULL, NULL);
741 if (rc == ERROR_SUCCESS)
745 LONG rcs = ERROR_SUCCESS;
746 WCHAR szSubKey[MAX_PATH];
748 RegOpenKeyExW(hkey, szNextKey, 0, KEY_READ, &subk);
749 while (rcs == ERROR_SUCCESS)
751 len = MAX_PATH * sizeof(WCHAR);
752 rcs = RegEnumKeyExW(subk, sub_index, szSubKey, &len, NULL, NULL, NULL, NULL);
753 if (rcs == ERROR_SUCCESS)
755 if (*supplied >= requested)
759 rc = ERROR_MORE_DATA;
760 rcs = ERROR_MORE_DATA;
764 wsprintfW(szGuidKey,szToGuidFmt,szNextKey);
765 CLSIDFromString(szGuidKey, &types[*supplied].type);
766 wsprintfW(szGuidKey,szToGuidFmt,szSubKey);
767 CLSIDFromString(szGuidKey, &types[*supplied].subtype);
768 TRACE("Adding type %s subtype %s at index %i\n",
769 debugstr_guid(&types[*supplied].type),
770 debugstr_guid(&types[*supplied].subtype),
785 /***************************************************************
786 * DMOGetTypes (MSDMO.@)
788 HRESULT WINAPI DMOGetTypes(REFCLSID clsidDMO,
789 ULONG ulInputTypesRequested,
790 ULONG* pulInputTypesSupplied,
791 DMO_PARTIAL_MEDIATYPE* pInputTypes,
792 ULONG ulOutputTypesRequested,
793 ULONG* pulOutputTypesSupplied,
794 DMO_PARTIAL_MEDIATYPE* pOutputTypes)
800 TRACE ("(%s,%u,%p,%p,%u,%p,%p),stub!\n", debugstr_guid(clsidDMO),
801 ulInputTypesRequested, pulInputTypesSupplied, pInputTypes,
802 ulOutputTypesRequested, pulOutputTypesSupplied, pOutputTypes);
804 if (ERROR_SUCCESS != RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0,
808 if (ERROR_SUCCESS != RegOpenKeyExW(root,GUIDToString(szguid,clsidDMO) , 0,
815 if (ulInputTypesRequested > 0)
817 ret = read_types(hkey, szDMOInputType, pulInputTypesSupplied, ulInputTypesRequested, pInputTypes );
820 *pulInputTypesSupplied = 0;
822 if (ulOutputTypesRequested > 0)
825 ret2 = read_types(hkey, szDMOOutputType, pulOutputTypesSupplied, ulOutputTypesRequested, pOutputTypes );
831 *pulOutputTypesSupplied = 0;