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;