Make it more explicit how we deal with WS_DISABLED.
[wine] / dlls / msdmo / dmoreg.c
1 /*
2  * Copyright (C) 2003 Michael Günnewig
3  * Copyright (C) 2003 CodeWeavers Inc. (Ulrich Czekalla)
4  *
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.
9  *
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.
14  *
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
18  */
19
20 #define COM_NO_WINDOWS_H
21 #include <stdarg.h>
22
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"
30
31 #include "wine/unicode.h"
32 #include "wine/debug.h"
33
34 WINE_DEFAULT_DEBUG_CHANNEL(msdmo);
35
36
37 static const WCHAR szDMORootKey[] = 
38 {
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 }; 
42
43 static const WCHAR szDMOInputType[] =
44 {
45     'I','n','p','u','t','T','y','p','e','s',0
46 };
47
48 static const WCHAR szDMOOutputType[] =
49 {
50     'O','u','t','p','u','t','T','y','p','e','s',0
51 };
52
53 static const WCHAR szDMOKeyed[] =
54 {
55     'K','e','y','e','d',0
56 };
57
58 static const WCHAR szDMOCategories[] =
59 {
60     'C','a','t','e','g','o','r','i','e','s',0
61 };
62
63 static const WCHAR szGUIDFmt[] =
64 {
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
68 };
69
70 static const WCHAR szCat3Fmt[] =
71 {
72     '%','s','\\','%','s','\\','%','s',0
73 };
74
75 static const WCHAR szCat2Fmt[] =
76 {
77     '%','s','\\','%','s',0
78 };
79
80 typedef struct
81 {
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;
93
94 const GUID IID_IEnumDMO = { 0x2c3cd98a, 0x2bfa, 0x4a53,
95     { 0x9c, 0x27, 0x52, 0x49, 0xba, 0x64, 0xba, 0x0f}};
96
97 static struct IEnumDMOVtbl edmovt;
98
99 static LPWSTR GUIDToString(LPWSTR lpwstr, REFGUID lpcguid)
100 {
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]);
105
106     return lpwstr;
107 }
108
109 static BOOL IsMediaTypeEqual(DMO_PARTIAL_MEDIATYPE* mt1, DMO_PARTIAL_MEDIATYPE* mt2)
110 {
111
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));
118 }
119
120 /***************************************************************
121  * DMORegister
122  *
123  * Register a DirectX Media Object.
124  */
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
134 )
135 {
136     WCHAR szguid[64];
137     HRESULT hres;
138     HKEY hrkey = 0;
139     HKEY hkey = 0;
140     HKEY hckey = 0;
141     HKEY hclskey = 0;
142
143     TRACE("%s\n", debugstr_w(szName));
144
145     hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
146     if (ERROR_SUCCESS != hres)
147         goto lend;
148
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;
154
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));
164
165     if (dwFlags & DMO_REGISTERF_IS_KEYED)
166     {
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);
173     }
174
175     /* Register the category */
176     hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
177     if (ERROR_SUCCESS != hres)
178         goto lend;
179
180     RegCloseKey(hkey);
181
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;
189
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);
199
200     TRACE(" hresult=0x%08lx\n", hres);
201     return hres;
202 }
203
204
205 /***************************************************************
206  * DMOUnregister
207  *
208  * Unregister a DirectX Media Object.
209  */
210 HRESULT WINAPI DMOUnregister(REFCLSID clsidDMO, REFGUID guidCategory)
211 {
212     HRESULT hres;
213     WCHAR szguid[64];
214     HKEY hrkey = 0;
215     HKEY hckey = 0;
216
217     GUIDToString(szguid, clsidDMO);
218
219     TRACE("%s %p\n", debugstr_w(szguid), guidCategory);
220
221     hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 0, KEY_WRITE, &hrkey);
222     if (ERROR_SUCCESS != hres)
223         goto lend;
224
225     hres = RegDeleteKeyW(hrkey, szguid);
226     if (ERROR_SUCCESS != hres)
227         goto lend;
228
229     hres = RegOpenKeyExW(hrkey, szDMOCategories, 0, KEY_WRITE, &hckey);
230     if (ERROR_SUCCESS != hres)
231         goto lend;
232
233     hres = RegDeleteKeyW(hckey, szguid);
234     if (ERROR_SUCCESS != hres)
235         goto lend;
236
237 lend:
238     if (hckey)
239         RegCloseKey(hckey);
240     if (hrkey)
241         RegCloseKey(hrkey);
242
243     return hres;
244 }
245
246
247 /***************************************************************
248  * DMOGetName
249  *
250  * Get DMP Name from the registry
251  */
252 HRESULT WINAPI DMOGetName(REFCLSID clsidDMO, WCHAR* szName)
253 {
254     WCHAR szguid[64];
255     HRESULT hres;
256     HKEY hrkey = 0;
257     HKEY hkey = 0;
258     DWORD count;
259
260     TRACE("%s\n", debugstr_guid(clsidDMO));
261
262     hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 
263         0, KEY_READ, &hrkey);
264     if (ERROR_SUCCESS != hres)
265         goto lend;
266
267     hres = RegOpenKeyExW(hrkey, GUIDToString(szguid, clsidDMO),
268         0, KEY_READ, &hkey);
269     if (ERROR_SUCCESS != hres)
270         goto lend;
271
272     count = 80 * sizeof(WCHAR); /* 80 by API definition */
273     hres = RegQueryValueExW(hkey, NULL, NULL, NULL, 
274         (LPBYTE) szName, &count); 
275
276     TRACE(" szName=%s\n", debugstr_w(szName));
277 lend:
278     if (hkey)
279         RegCloseKey(hrkey);
280     if (hkey)
281         RegCloseKey(hkey);
282
283     return hres;
284 }
285
286
287 /**************************************************************************
288 *   IEnumDMO_Destructor
289 */
290 static BOOL IEnumDMO_Destructor(IEnumDMO* iface)
291 {
292     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
293
294     TRACE("%p\n", This);
295
296     if (This->hkey)
297         RegCloseKey(This->hkey);
298
299     HeapFree(GetProcessHeap(), 0, This->pInTypes);
300     HeapFree(GetProcessHeap(), 0, This->pOutTypes);
301
302     return TRUE;
303 }
304
305
306 /**************************************************************************
307  *  IEnumDMO_Constructor
308  */
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)
316 {
317     UINT size;
318     IEnumDMOImpl* lpedmo;
319     BOOL ret = FALSE;
320
321     lpedmo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IEnumDMOImpl));
322
323     if (lpedmo)
324     {
325         lpedmo->ref = 1;
326         lpedmo->lpVtbl = &edmovt;
327         lpedmo->index = -1;
328         lpedmo->guidCategory = guidCategory;
329         lpedmo->dwFlags = dwFlags;
330
331         size = cInTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
332         lpedmo->pInTypes = HeapAlloc(GetProcessHeap(), 0, size);
333         if (!lpedmo->pInTypes)
334             goto lerr;
335         memcpy(lpedmo->pInTypes, pInTypes, size);
336         lpedmo->cInTypes = cInTypes;
337
338         size = cOutTypes * sizeof(DMO_PARTIAL_MEDIATYPE);
339         lpedmo->pOutTypes = HeapAlloc(GetProcessHeap(), 0, size);
340         if (!lpedmo->pOutTypes)
341             goto lerr;
342         memcpy(lpedmo->pOutTypes, pOutTypes, size);
343         lpedmo->cOutTypes = cOutTypes;
344
345         /* If not filtering by category enum from media objects root */
346         if (IsEqualGUID(guidCategory, &GUID_NULL))
347         {
348             if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szDMORootKey, 
349                 0, KEY_READ, &lpedmo->hkey))
350                 ret = TRUE;
351         }
352         else
353         {
354             WCHAR szguid[64];
355             WCHAR szKey[MAX_PATH];
356
357             wsprintfW(szKey, szCat3Fmt, szDMORootKey, szDMOCategories, 
358                 GUIDToString(szguid, guidCategory));
359             if (ERROR_SUCCESS == RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 
360                 0, KEY_READ, &lpedmo->hkey))
361                 ret = TRUE;
362         }
363
364 lerr:
365         if(!ret)
366         {
367             IEnumDMO_Destructor((IEnumDMO*)lpedmo);
368             HeapFree(GetProcessHeap(),0,lpedmo);
369             lpedmo = NULL;
370         }
371     }
372
373     TRACE("returning %p\n", lpedmo);
374
375     return (IEnumDMO*)lpedmo;
376 }
377
378
379 /******************************************************************************
380  * IEnumDMO_fnAddRef
381  */
382 static ULONG WINAPI IEnumDMO_fnAddRef(IEnumDMO * iface)
383 {
384     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
385     return InterlockedIncrement(&This->ref);
386 }
387
388
389 /**************************************************************************
390  *  EnumDMO_QueryInterface
391  */
392 static HRESULT WINAPI IEnumDMO_fnQueryInterface(
393     IEnumDMO* iface,
394     REFIID riid,
395     LPVOID *ppvObj)
396 {
397     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
398
399     *ppvObj = NULL;
400
401     if(IsEqualIID(riid, &IID_IUnknown))
402         *ppvObj = This;
403     else if(IsEqualIID(riid, &IID_IEnumDMO))
404         *ppvObj = (IEnumDMO*)This;
405
406     if(*ppvObj)
407     {
408         IEnumDMO_fnAddRef((IEnumDMO*)*ppvObj);
409         return S_OK;
410     }
411
412     return E_NOINTERFACE;
413 }
414
415
416 /******************************************************************************
417  * IEnumDMO_fnRelease
418  */
419 static ULONG WINAPI IEnumDMO_fnRelease(IEnumDMO * iface)
420 {
421     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
422     ULONG refCount = InterlockedDecrement(&This->ref);
423
424     if (!refCount)
425     {
426         IEnumDMO_Destructor((IEnumDMO*)This);
427         HeapFree(GetProcessHeap(),0,This);
428     }
429     return refCount;
430 }
431
432
433 /******************************************************************************
434  * IEnumDMO_fnNext
435  */
436 static HRESULT WINAPI IEnumDMO_fnNext(
437     IEnumDMO * iface, 
438     DWORD cItemsToFetch,
439     CLSID * pCLSID,
440     WCHAR ** Names,
441     DWORD * pcItemsFetched)
442 {
443     FILETIME ft;
444     HKEY hkey;
445     WCHAR szNextKey[MAX_PATH];
446     WCHAR szKey[MAX_PATH];
447     WCHAR szValue[MAX_PATH];
448     DWORD len;
449     UINT count = 0;
450     HRESULT hres = S_OK;
451
452     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
453
454     TRACE("%ld\n", cItemsToFetch);
455
456     if (!pCLSID || !Names || !pcItemsFetched)
457         return E_POINTER;
458
459     while (count < cItemsToFetch)
460     {
461         This->index++;
462
463         hres = RegEnumKeyExW(This->hkey, This->index, szNextKey, &len, NULL, NULL, NULL, &ft);
464         if (hres != ERROR_SUCCESS)
465             break;
466
467         TRACE("found %s\n", debugstr_w(szNextKey));
468
469         if (This->dwFlags & DMO_REGISTERF_IS_KEYED)
470         {
471             wsprintfW(szKey, szCat3Fmt, szDMORootKey, szNextKey, szDMOKeyed);
472             hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
473             if (ERROR_SUCCESS != hres)
474                 continue;
475             RegCloseKey(hkey);
476         }
477
478         wsprintfW(szKey, szCat2Fmt, szDMORootKey, szNextKey);
479         hres = RegOpenKeyExW(HKEY_CLASSES_ROOT, szKey, 0, KEY_READ, &hkey);
480
481         if (This->pInTypes)
482         {
483             UINT i, j;
484             DWORD cInTypes;
485             DMO_PARTIAL_MEDIATYPE* pInTypes;
486
487             len = MAX_PATH * sizeof(WCHAR);
488             hres = RegQueryValueExW(hkey, szDMOInputType, NULL, NULL, (LPBYTE) szValue, &len);
489             if (ERROR_SUCCESS != hres)
490             {
491                 RegCloseKey(hkey);
492                 continue;
493             }
494
495             cInTypes = len / sizeof(DMO_PARTIAL_MEDIATYPE);
496             pInTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
497
498             for (i = 0; i < This->cInTypes; i++)
499             {
500                 for (j = 0; j < cInTypes; j++) 
501                 {
502                     if (IsMediaTypeEqual(&pInTypes[j], &This->pInTypes[i]))
503                         break;
504                 }
505
506                 if (j >= cInTypes)
507                     break;
508             }
509
510             if (i < This->cInTypes)
511             {
512                 RegCloseKey(hkey);
513                 continue;
514             }
515         }
516
517         if (This->pOutTypes)
518         {
519             UINT i, j;
520             DWORD cOutTypes;
521             DMO_PARTIAL_MEDIATYPE* pOutTypes;
522
523             len = MAX_PATH * sizeof(WCHAR);
524             hres = RegQueryValueExW(hkey, szDMOOutputType, NULL, NULL, (LPBYTE) szValue, &len);
525             if (ERROR_SUCCESS != hres)
526             {
527                 RegCloseKey(hkey);
528                 continue;
529             }
530
531             cOutTypes = len / sizeof(DMO_PARTIAL_MEDIATYPE);
532             pOutTypes = (DMO_PARTIAL_MEDIATYPE*) szValue;
533
534             for (i = 0; i < This->cOutTypes; i++)
535             {
536                 for (j = 0; j < cOutTypes; j++) 
537                 {
538                     if (IsMediaTypeEqual(&pOutTypes[j], &This->pOutTypes[i]))
539                         break;
540                 }
541
542                 if (j >= cOutTypes)
543                     break;
544             }
545
546             if (i < This->cOutTypes)
547             {
548                 RegCloseKey(hkey);
549                 continue;
550             }
551         }
552
553         /* Media object wasn't filtered so add it to return list */
554         Names[count] = NULL;
555         len = MAX_PATH * sizeof(WCHAR);
556         hres = RegQueryValueExW(hkey, NULL, NULL, NULL, (LPBYTE) szValue, &len); 
557         if (ERROR_SUCCESS == hres)
558         {
559             Names[count] = HeapAlloc(GetProcessHeap(), 0, strlenW(szValue) + 1);
560             if (Names[count])
561                 strcmpW(Names[count], szValue);
562         }
563         CLSIDFromString(szNextKey, &pCLSID[count]);
564
565         TRACE("found match %s %s\n", debugstr_w(szValue), debugstr_w(szNextKey));
566         RegCloseKey(hkey);
567         count++;
568     }
569
570     *pcItemsFetched = count;
571     if (*pcItemsFetched < cItemsToFetch)
572         hres = S_FALSE;
573
574     return hres;
575 }
576  
577
578 /******************************************************************************
579  * IEnumDMO_fnSkip
580  */
581 static HRESULT WINAPI IEnumDMO_fnSkip(IEnumDMO * iface, DWORD cItemsToSkip)
582 {
583     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
584
585     This->index += cItemsToSkip;
586
587     return S_OK;
588 }
589
590
591 /******************************************************************************
592  * IEnumDMO_fnReset
593  */
594 static HRESULT WINAPI IEnumDMO_fnReset(IEnumDMO * iface)
595 {
596     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
597
598     This->index = -1;
599
600     return S_OK;
601 }
602  
603
604 /******************************************************************************
605  * IEnumDMO_fnClone
606  */
607 static HRESULT WINAPI IEnumDMO_fnClone(IEnumDMO * iface, IEnumDMO **ppEnum)
608 {
609     IEnumDMOImpl *This = (IEnumDMOImpl *)iface;
610
611     FIXME("(%p)->() to (%p)->() E_NOTIMPL\n", This, ppEnum);
612
613   return E_NOTIMPL;
614 }
615
616
617 /***************************************************************
618  * DMOEnum
619  *
620  * Enumerate DirectX Media Objects in the registry.
621  */
622 HRESULT WINAPI DMOEnum(
623     REFGUID guidCategory,
624     DWORD dwFlags,
625     DWORD cInTypes,
626     const DMO_PARTIAL_MEDIATYPE *pInTypes,
627     DWORD cOutTypes,
628     const DMO_PARTIAL_MEDIATYPE *pOutTypes,
629     IEnumDMO **ppEnum)
630 {
631     HRESULT hres = E_FAIL;
632
633     TRACE("guidCategory=%p dwFlags=0x%08lx cInTypes=%ld cOutTypes=%ld\n",
634         guidCategory, dwFlags, cInTypes, cOutTypes);
635
636     *ppEnum = IEnumDMO_Constructor(guidCategory, dwFlags, cInTypes,
637         pInTypes, cOutTypes, pOutTypes);
638     if (*ppEnum)
639         hres = S_OK;
640
641     return hres;
642 }
643
644
645 static IEnumDMOVtbl edmovt =
646 {
647         IEnumDMO_fnQueryInterface,
648         IEnumDMO_fnAddRef,
649         IEnumDMO_fnRelease,
650         IEnumDMO_fnNext,
651         IEnumDMO_fnSkip,
652         IEnumDMO_fnReset,
653         IEnumDMO_fnClone,
654 };
655
656
657 HRESULT WINAPI DMOGetTypes(REFCLSID a, unsigned long b, unsigned long* c,
658                            DMO_PARTIAL_MEDIATYPE* d, unsigned long e,
659                            unsigned long* f, DMO_PARTIAL_MEDIATYPE* g)
660 {
661   FIXME("(%p,%lu,%p,%p,%lu,%p,%p),stub!\n",a,b,c,d,e,f,g);
662
663   return E_NOTIMPL;
664 }