devenum: Do not fail with a IBindCtx (with tests).
[wine/multimedia.git] / dlls / devenum / mediacatenum.c
blob24d236d5c2c780e4a072c2695a6029e8b9cea537
1 /*
2 * IEnumMoniker implementation for DEVENUM.dll
4 * Copyright (C) 2002 Robert Shearman
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 * NOTES ON THIS FILE:
21 * - Implements IEnumMoniker interface which enumerates through moniker
22 * objects created from HKEY_CLASSES/CLSID/{DEVICE_CLSID}/Instance
25 #include "devenum_private.h"
26 #include "oleauto.h"
27 #include "ocidl.h"
29 #include "wine/debug.h"
31 WINE_DEFAULT_DEBUG_CHANNEL(devenum);
33 typedef struct
35 IEnumMoniker IEnumMoniker_iface;
36 LONG ref;
37 DWORD index;
38 HKEY hkey;
39 } EnumMonikerImpl;
41 typedef struct
43 IPropertyBag IPropertyBag_iface;
44 LONG ref;
45 HKEY hkey;
46 } RegPropBagImpl;
49 static inline RegPropBagImpl *impl_from_IPropertyBag(IPropertyBag *iface)
51 return CONTAINING_RECORD(iface, RegPropBagImpl, IPropertyBag_iface);
54 static HRESULT WINAPI DEVENUM_IPropertyBag_QueryInterface(
55 LPPROPERTYBAG iface,
56 REFIID riid,
57 LPVOID *ppvObj)
59 RegPropBagImpl *This = impl_from_IPropertyBag(iface);
61 TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), ppvObj);
63 if (This == NULL || ppvObj == NULL) return E_POINTER;
65 if (IsEqualGUID(riid, &IID_IUnknown) ||
66 IsEqualGUID(riid, &IID_IPropertyBag))
68 *ppvObj = iface;
69 IPropertyBag_AddRef(iface);
70 return S_OK;
73 FIXME("- no interface IID: %s\n", debugstr_guid(riid));
74 return E_NOINTERFACE;
77 /**********************************************************************
78 * DEVENUM_IPropertyBag_AddRef (also IUnknown)
80 static ULONG WINAPI DEVENUM_IPropertyBag_AddRef(LPPROPERTYBAG iface)
82 RegPropBagImpl *This = impl_from_IPropertyBag(iface);
84 TRACE("(%p)->() AddRef from %d\n", iface, This->ref);
86 return InterlockedIncrement(&This->ref);
89 /**********************************************************************
90 * DEVENUM_IPropertyBag_Release (also IUnknown)
92 static ULONG WINAPI DEVENUM_IPropertyBag_Release(LPPROPERTYBAG iface)
94 RegPropBagImpl *This = impl_from_IPropertyBag(iface);
95 ULONG ref;
97 TRACE("(%p)->() ReleaseThis->ref from %d\n", iface, This->ref);
99 ref = InterlockedDecrement(&This->ref);
100 if (ref == 0) {
101 RegCloseKey(This->hkey);
102 CoTaskMemFree(This);
103 DEVENUM_UnlockModule();
105 return ref;
108 static HRESULT WINAPI DEVENUM_IPropertyBag_Read(
109 LPPROPERTYBAG iface,
110 LPCOLESTR pszPropName,
111 VARIANT* pVar,
112 IErrorLog* pErrorLog)
114 LPVOID pData = NULL;
115 DWORD received;
116 DWORD type = 0;
117 RegPropBagImpl *This = impl_from_IPropertyBag(iface);
118 HRESULT res = S_OK;
119 LONG reswin32;
121 TRACE("(%p)->(%s, %p, %p)\n", This, debugstr_w(pszPropName), pVar, pErrorLog);
123 if (!pszPropName || !pVar)
124 return E_POINTER;
126 reswin32 = RegQueryValueExW(This->hkey, pszPropName, NULL, NULL, NULL, &received);
127 res = HRESULT_FROM_WIN32(reswin32);
129 if (SUCCEEDED(res))
131 pData = HeapAlloc(GetProcessHeap(), 0, received);
133 /* work around a GCC bug that occurs here unless we use the reswin32 variable as well */
134 reswin32 = RegQueryValueExW(This->hkey, pszPropName, NULL, &type, pData, &received);
135 res = HRESULT_FROM_WIN32(reswin32);
138 if (SUCCEEDED(res))
140 res = E_INVALIDARG; /* assume we cannot coerce into right type */
142 TRACE("Read %d bytes (%s)\n", received, type == REG_SZ ? debugstr_w(pData) : "binary data");
144 switch (type)
146 case REG_SZ:
147 switch (V_VT(pVar))
149 case VT_LPWSTR:
150 V_UNION(pVar, bstrVal) = CoTaskMemAlloc(received);
151 memcpy(V_UNION(pVar, bstrVal), pData, received);
152 res = S_OK;
153 break;
154 case VT_EMPTY:
155 V_VT(pVar) = VT_BSTR;
156 /* fall through */
157 case VT_BSTR:
158 V_UNION(pVar, bstrVal) = SysAllocStringLen(pData, received/sizeof(WCHAR) - 1);
159 res = S_OK;
160 break;
162 break;
163 case REG_DWORD:
164 TRACE("REG_DWORD: %x\n", *(DWORD *)pData);
165 switch (V_VT(pVar))
167 case VT_EMPTY:
168 V_VT(pVar) = VT_I4;
169 /* fall through */
170 case VT_I4:
171 case VT_UI4:
172 V_UNION(pVar, ulVal) = *(DWORD *)pData;
173 res = S_OK;
174 break;
176 break;
177 case REG_BINARY:
179 SAFEARRAYBOUND bound;
180 void * pArrayElements;
181 bound.lLbound = 0;
182 bound.cElements = received;
183 TRACE("REG_BINARY: len = %d\n", received);
184 switch (V_VT(pVar))
186 case VT_EMPTY:
187 V_VT(pVar) = VT_ARRAY | VT_UI1;
188 /* fall through */
189 case VT_ARRAY | VT_UI1:
190 if (!(V_UNION(pVar, parray) = SafeArrayCreate(VT_UI1, 1, &bound)))
191 res = E_OUTOFMEMORY;
192 else
193 res = S_OK;
194 break;
197 if (res == E_INVALIDARG)
198 break;
200 res = SafeArrayAccessData(V_UNION(pVar, parray), &pArrayElements);
201 if (FAILED(res))
202 break;
204 CopyMemory(pArrayElements, pData, received);
205 res = SafeArrayUnaccessData(V_UNION(pVar, parray));
206 break;
209 if (res == E_INVALIDARG)
210 FIXME("Variant type %x not supported for regtype %x\n", V_VT(pVar), type);
213 HeapFree(GetProcessHeap(), 0, pData);
215 TRACE("<- %x\n", res);
216 return res;
219 static HRESULT WINAPI DEVENUM_IPropertyBag_Write(
220 LPPROPERTYBAG iface,
221 LPCOLESTR pszPropName,
222 VARIANT* pVar)
224 RegPropBagImpl *This = impl_from_IPropertyBag(iface);
225 LPVOID lpData = NULL;
226 DWORD cbData = 0;
227 DWORD dwType = 0;
228 HRESULT res = S_OK;
230 TRACE("(%p)->(%s, %p)\n", This, debugstr_w(pszPropName), pVar);
232 switch (V_VT(pVar))
234 case VT_BSTR:
235 TRACE("writing %s\n", debugstr_w(V_UNION(pVar, bstrVal)));
236 lpData = V_UNION(pVar, bstrVal);
237 dwType = REG_SZ;
238 cbData = (lstrlenW(V_UNION(pVar, bstrVal)) + 1) * sizeof(WCHAR);
239 break;
240 case VT_I4:
241 case VT_UI4:
242 TRACE("writing %u\n", V_UNION(pVar, ulVal));
243 lpData = &V_UNION(pVar, ulVal);
244 dwType = REG_DWORD;
245 cbData = sizeof(DWORD);
246 break;
247 case VT_ARRAY | VT_UI1:
249 LONG lUbound = 0;
250 LONG lLbound = 0;
251 dwType = REG_BINARY;
252 res = SafeArrayGetLBound(V_UNION(pVar, parray), 1, &lLbound);
253 res = SafeArrayGetUBound(V_UNION(pVar, parray), 1, &lUbound);
254 cbData = (lUbound - lLbound + 1) /* * sizeof(BYTE)*/;
255 TRACE("cbData: %d\n", cbData);
256 res = SafeArrayAccessData(V_UNION(pVar, parray), &lpData);
257 break;
259 default:
260 FIXME("Variant type %d not handled\n", V_VT(pVar));
261 return E_FAIL;
264 if (RegSetValueExW(This->hkey,
265 pszPropName, 0,
266 dwType, lpData, cbData) != ERROR_SUCCESS)
267 res = E_FAIL;
269 if (V_VT(pVar) & VT_ARRAY)
270 res = SafeArrayUnaccessData(V_UNION(pVar, parray));
272 return res;
275 static const IPropertyBagVtbl IPropertyBag_Vtbl =
277 DEVENUM_IPropertyBag_QueryInterface,
278 DEVENUM_IPropertyBag_AddRef,
279 DEVENUM_IPropertyBag_Release,
280 DEVENUM_IPropertyBag_Read,
281 DEVENUM_IPropertyBag_Write
284 static HRESULT DEVENUM_IPropertyBag_Construct(HANDLE hkey, IPropertyBag **ppBag)
286 RegPropBagImpl * rpb = CoTaskMemAlloc(sizeof(RegPropBagImpl));
287 if (!rpb)
288 return E_OUTOFMEMORY;
289 rpb->IPropertyBag_iface.lpVtbl = &IPropertyBag_Vtbl;
290 rpb->ref = 1;
291 rpb->hkey = hkey;
292 *ppBag = &rpb->IPropertyBag_iface;
293 DEVENUM_LockModule();
294 return S_OK;
298 static inline MediaCatMoniker *impl_from_IMoniker(IMoniker *iface)
300 return CONTAINING_RECORD(iface, MediaCatMoniker, IMoniker_iface);
303 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_QueryInterface(IMoniker *iface, REFIID riid,
304 void **ppv)
306 TRACE("\n\tIID:\t%s\n",debugstr_guid(riid));
308 if (!ppv)
309 return E_POINTER;
311 if (IsEqualGUID(riid, &IID_IUnknown) ||
312 IsEqualGUID(riid, &IID_IPersist) ||
313 IsEqualGUID(riid, &IID_IPersistStream) ||
314 IsEqualGUID(riid, &IID_IMoniker))
316 *ppv = iface;
317 IMoniker_AddRef(iface);
318 return S_OK;
321 FIXME("- no interface IID: %s\n", debugstr_guid(riid));
322 *ppv = NULL;
323 return E_NOINTERFACE;
326 static ULONG WINAPI DEVENUM_IMediaCatMoniker_AddRef(IMoniker *iface)
328 MediaCatMoniker *This = impl_from_IMoniker(iface);
329 ULONG ref = InterlockedIncrement(&This->ref);
331 TRACE("(%p) ref=%d\n", This, ref);
333 return ref;
336 static ULONG WINAPI DEVENUM_IMediaCatMoniker_Release(IMoniker *iface)
338 MediaCatMoniker *This = impl_from_IMoniker(iface);
339 ULONG ref = InterlockedDecrement(&This->ref);
341 TRACE("(%p) ref=%d\n", This, ref);
343 if (ref == 0) {
344 RegCloseKey(This->hkey);
345 CoTaskMemFree(This);
346 DEVENUM_UnlockModule();
348 return ref;
351 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetClassID(IMoniker *iface, CLSID *pClassID)
353 MediaCatMoniker *This = impl_from_IMoniker(iface);
355 FIXME("(%p)->(%p): stub\n", This, pClassID);
357 if (pClassID == NULL)
358 return E_POINTER;
360 return E_NOTIMPL;
363 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsDirty(IMoniker *iface)
365 FIXME("(%p)->(): stub\n", iface);
367 return S_FALSE;
370 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Load(IMoniker *iface, IStream *pStm)
372 FIXME("(%p)->(%p): stub\n", iface, pStm);
374 return E_NOTIMPL;
377 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Save(IMoniker *iface, IStream *pStm, BOOL fClearDirty)
379 FIXME("(%p)->(%p, %s): stub\n", iface, pStm, fClearDirty ? "true" : "false");
381 return STG_E_CANTSAVE;
384 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetSizeMax(IMoniker *iface, ULARGE_INTEGER *pcbSize)
386 FIXME("(%p)->(%p): stub\n", iface, pcbSize);
388 ZeroMemory(pcbSize, sizeof(*pcbSize));
390 return S_OK;
393 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_BindToObject(IMoniker *iface, IBindCtx *pbc,
394 IMoniker *pmkToLeft, REFIID riidResult, void **ppvResult)
396 MediaCatMoniker *This = impl_from_IMoniker(iface);
397 IUnknown * pObj = NULL;
398 IPropertyBag * pProp = NULL;
399 CLSID clsID;
400 VARIANT var;
401 HRESULT res = E_FAIL;
403 TRACE("(%p)->(%p, %p, %s, %p)\n", This, pbc, pmkToLeft, debugstr_guid(riidResult), ppvResult);
405 VariantInit(&var);
406 *ppvResult = NULL;
408 if(pmkToLeft==NULL)
410 /* first activation of this class */
411 LPVOID pvptr;
412 res=IMoniker_BindToStorage(iface, NULL, NULL, &IID_IPropertyBag, &pvptr);
413 pProp = pvptr;
414 if (SUCCEEDED(res))
416 V_VT(&var) = VT_LPWSTR;
417 res = IPropertyBag_Read(pProp, clsid_keyname, &var, NULL);
419 if (SUCCEEDED(res))
421 res = CLSIDFromString(V_UNION(&var,bstrVal), &clsID);
422 CoTaskMemFree(V_UNION(&var, bstrVal));
424 if (SUCCEEDED(res))
426 res=CoCreateInstance(&clsID,NULL,CLSCTX_ALL,&IID_IUnknown,&pvptr);
427 pObj = pvptr;
431 if (pObj!=NULL)
433 /* get the requested interface from the loaded class */
434 res = S_OK;
435 if (pProp) {
436 HRESULT res2;
437 LPVOID ppv = NULL;
438 res2 = IUnknown_QueryInterface(pObj, &IID_IPersistPropertyBag, &ppv);
439 if (SUCCEEDED(res2)) {
440 res = IPersistPropertyBag_Load((IPersistPropertyBag *) ppv, pProp, NULL);
441 IPersistPropertyBag_Release((IPersistPropertyBag *) ppv);
444 if (SUCCEEDED(res))
445 res= IUnknown_QueryInterface(pObj,riidResult,ppvResult);
446 IUnknown_Release(pObj);
449 if (pProp)
451 IPropertyBag_Release(pProp);
454 TRACE("<- 0x%x\n", res);
456 return res;
459 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_BindToStorage(IMoniker *iface, IBindCtx *pbc,
460 IMoniker *pmkToLeft, REFIID riid, void **ppvObj)
462 MediaCatMoniker *This = impl_from_IMoniker(iface);
464 TRACE("(%p)->(%p, %p, %s, %p)\n", This, pbc, pmkToLeft, debugstr_guid(riid), ppvObj);
466 *ppvObj = NULL;
468 if (pmkToLeft)
469 return MK_E_NOSTORAGE;
471 if (pbc != NULL)
473 static DWORD reported;
474 if (!reported)
476 FIXME("ignoring IBindCtx %p\n", pbc);
477 reported++;
481 if (IsEqualGUID(riid, &IID_IPropertyBag))
483 HANDLE hkey;
484 DuplicateHandle(GetCurrentProcess(), This->hkey, GetCurrentProcess(), &hkey, 0, 0, DUPLICATE_SAME_ACCESS);
485 return DEVENUM_IPropertyBag_Construct(hkey, (IPropertyBag**)ppvObj);
488 return MK_E_NOSTORAGE;
491 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Reduce(IMoniker *iface, IBindCtx *pbc,
492 DWORD dwReduceHowFar, IMoniker **ppmkToLeft, IMoniker **ppmkReduced)
494 TRACE("(%p)->(%p, %d, %p, %p)\n", iface, pbc, dwReduceHowFar, ppmkToLeft, ppmkReduced);
496 if (ppmkToLeft)
497 *ppmkToLeft = NULL;
498 *ppmkReduced = iface;
500 return MK_S_REDUCED_TO_SELF;
503 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_ComposeWith(IMoniker *iface, IMoniker *pmkRight,
504 BOOL fOnlyIfNotGeneric, IMoniker **ppmkComposite)
506 FIXME("(%p)->(%p, %s, %p): stub\n", iface, pmkRight, fOnlyIfNotGeneric ? "true" : "false", ppmkComposite);
508 /* FIXME: use CreateGenericComposite? */
509 *ppmkComposite = NULL;
511 return E_NOTIMPL;
514 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Enum(IMoniker *iface, BOOL fForward,
515 IEnumMoniker **ppenumMoniker)
517 FIXME("(%p)->(%s, %p): stub\n", iface, fForward ? "true" : "false", ppenumMoniker);
519 *ppenumMoniker = NULL;
521 return S_OK;
524 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsEqual(IMoniker *iface, IMoniker *pmkOtherMoniker)
526 FIXME("(%p)->(%p): stub\n", iface, pmkOtherMoniker);
528 return E_NOTIMPL;
531 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Hash(IMoniker *iface, DWORD *pdwHash)
533 TRACE("(%p)->(%p)\n", iface, pdwHash);
535 *pdwHash = 0;
537 return S_OK;
540 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsRunning(IMoniker *iface, IBindCtx *pbc,
541 IMoniker *pmkToLeft, IMoniker *pmkNewlyRunning)
543 FIXME("(%p)->(%p, %p, %p): stub\n", iface, pbc, pmkToLeft, pmkNewlyRunning);
545 return S_FALSE;
548 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetTimeOfLastChange(IMoniker *iface, IBindCtx *pbc,
549 IMoniker *pmkToLeft, FILETIME *pFileTime)
551 TRACE("(%p)->(%p, %p, %p)\n", iface, pbc, pmkToLeft, pFileTime);
553 pFileTime->dwLowDateTime = 0xFFFFFFFF;
554 pFileTime->dwHighDateTime = 0x7FFFFFFF;
556 return MK_E_UNAVAILABLE;
559 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Inverse(IMoniker *iface, IMoniker **ppmk)
561 TRACE("(%p)->(%p)\n", iface, ppmk);
563 *ppmk = NULL;
565 return MK_E_NOINVERSE;
568 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_CommonPrefixWith(IMoniker *iface,
569 IMoniker *pmkOtherMoniker, IMoniker **ppmkPrefix)
571 TRACE("(%p)->(%p, %p)\n", iface, pmkOtherMoniker, ppmkPrefix);
573 *ppmkPrefix = NULL;
575 return MK_E_NOPREFIX;
578 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_RelativePathTo(IMoniker *iface, IMoniker *pmkOther,
579 IMoniker **ppmkRelPath)
581 TRACE("(%p)->(%p, %p)\n", iface, pmkOther, ppmkRelPath);
583 *ppmkRelPath = pmkOther;
585 return MK_S_HIM;
588 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetDisplayName(IMoniker *iface, IBindCtx *pbc,
589 IMoniker *pmkToLeft, LPOLESTR *ppszDisplayName)
591 MediaCatMoniker *This = impl_from_IMoniker(iface);
592 WCHAR wszBuffer[MAX_PATH];
593 static const WCHAR wszFriendlyName[] = {'F','r','i','e','n','d','l','y','N','a','m','e',0};
594 LONG received = sizeof(wszFriendlyName);
596 TRACE("(%p)->(%p, %p, %p)\n", iface, pbc, pmkToLeft, ppszDisplayName);
598 *ppszDisplayName = NULL;
600 /* FIXME: should this be the weird stuff we have to parse in IParseDisplayName? */
601 if (RegQueryValueW(This->hkey, wszFriendlyName, wszBuffer, &received) == ERROR_SUCCESS)
603 *ppszDisplayName = CoTaskMemAlloc(received);
604 strcpyW(*ppszDisplayName, wszBuffer);
605 return S_OK;
608 return E_FAIL;
611 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_ParseDisplayName(IMoniker *iface, IBindCtx *pbc,
612 IMoniker *pmkToLeft, LPOLESTR pszDisplayName, ULONG *pchEaten, IMoniker **ppmkOut)
614 FIXME("(%p)->(%p, %p, %s, %p, %p)\n", iface, pbc, pmkToLeft, debugstr_w(pszDisplayName), pchEaten, ppmkOut);
616 *pchEaten = 0;
617 *ppmkOut = NULL;
619 return MK_E_SYNTAX;
622 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsSystemMoniker(IMoniker *iface, DWORD *pdwMksys)
624 TRACE("(%p)->(%p)\n", iface, pdwMksys);
626 return S_FALSE;
629 static const IMonikerVtbl IMoniker_Vtbl =
631 DEVENUM_IMediaCatMoniker_QueryInterface,
632 DEVENUM_IMediaCatMoniker_AddRef,
633 DEVENUM_IMediaCatMoniker_Release,
634 DEVENUM_IMediaCatMoniker_GetClassID,
635 DEVENUM_IMediaCatMoniker_IsDirty,
636 DEVENUM_IMediaCatMoniker_Load,
637 DEVENUM_IMediaCatMoniker_Save,
638 DEVENUM_IMediaCatMoniker_GetSizeMax,
639 DEVENUM_IMediaCatMoniker_BindToObject,
640 DEVENUM_IMediaCatMoniker_BindToStorage,
641 DEVENUM_IMediaCatMoniker_Reduce,
642 DEVENUM_IMediaCatMoniker_ComposeWith,
643 DEVENUM_IMediaCatMoniker_Enum,
644 DEVENUM_IMediaCatMoniker_IsEqual,
645 DEVENUM_IMediaCatMoniker_Hash,
646 DEVENUM_IMediaCatMoniker_IsRunning,
647 DEVENUM_IMediaCatMoniker_GetTimeOfLastChange,
648 DEVENUM_IMediaCatMoniker_Inverse,
649 DEVENUM_IMediaCatMoniker_CommonPrefixWith,
650 DEVENUM_IMediaCatMoniker_RelativePathTo,
651 DEVENUM_IMediaCatMoniker_GetDisplayName,
652 DEVENUM_IMediaCatMoniker_ParseDisplayName,
653 DEVENUM_IMediaCatMoniker_IsSystemMoniker
656 MediaCatMoniker * DEVENUM_IMediaCatMoniker_Construct(void)
658 MediaCatMoniker * pMoniker = NULL;
659 pMoniker = CoTaskMemAlloc(sizeof(MediaCatMoniker));
660 if (!pMoniker)
661 return NULL;
663 pMoniker->IMoniker_iface.lpVtbl = &IMoniker_Vtbl;
664 pMoniker->ref = 0;
665 pMoniker->hkey = NULL;
667 DEVENUM_IMediaCatMoniker_AddRef(&pMoniker->IMoniker_iface);
669 DEVENUM_LockModule();
671 return pMoniker;
674 static inline EnumMonikerImpl *impl_from_IEnumMoniker(IEnumMoniker *iface)
676 return CONTAINING_RECORD(iface, EnumMonikerImpl, IEnumMoniker_iface);
679 static HRESULT WINAPI DEVENUM_IEnumMoniker_QueryInterface(IEnumMoniker *iface, REFIID riid,
680 void **ppv)
682 TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), ppv);
684 if (!ppv)
685 return E_POINTER;
687 if (IsEqualGUID(riid, &IID_IUnknown) ||
688 IsEqualGUID(riid, &IID_IEnumMoniker))
690 *ppv = iface;
691 IEnumMoniker_AddRef(iface);
692 return S_OK;
695 FIXME("- no interface IID: %s\n", debugstr_guid(riid));
696 *ppv = NULL;
697 return E_NOINTERFACE;
700 static ULONG WINAPI DEVENUM_IEnumMoniker_AddRef(IEnumMoniker *iface)
702 EnumMonikerImpl *This = impl_from_IEnumMoniker(iface);
703 ULONG ref = InterlockedIncrement(&This->ref);
705 TRACE("(%p) ref=%d\n", This, ref);
707 return ref;
710 static ULONG WINAPI DEVENUM_IEnumMoniker_Release(IEnumMoniker *iface)
712 EnumMonikerImpl *This = impl_from_IEnumMoniker(iface);
713 ULONG ref = InterlockedDecrement(&This->ref);
715 TRACE("(%p) ref=%d\n", This, ref);
717 if (!ref)
719 RegCloseKey(This->hkey);
720 CoTaskMemFree(This);
721 DEVENUM_UnlockModule();
722 return 0;
724 return ref;
727 static HRESULT WINAPI DEVENUM_IEnumMoniker_Next(IEnumMoniker *iface, ULONG celt, IMoniker **rgelt,
728 ULONG *pceltFetched)
730 EnumMonikerImpl *This = impl_from_IEnumMoniker(iface);
731 WCHAR buffer[MAX_PATH + 1];
732 LONG res;
733 ULONG fetched = 0;
734 MediaCatMoniker * pMoniker;
736 TRACE("(%p)->(%d, %p, %p)\n", iface, celt, rgelt, pceltFetched);
738 while (fetched < celt)
740 res = RegEnumKeyW(This->hkey, This->index, buffer, sizeof(buffer) / sizeof(WCHAR));
741 if (res != ERROR_SUCCESS)
743 break;
745 pMoniker = DEVENUM_IMediaCatMoniker_Construct();
746 if (!pMoniker)
747 return E_OUTOFMEMORY;
749 if (RegOpenKeyW(This->hkey, buffer, &pMoniker->hkey) != ERROR_SUCCESS)
751 IMoniker_Release(&pMoniker->IMoniker_iface);
752 break;
754 rgelt[fetched] = &pMoniker->IMoniker_iface;
755 fetched++;
758 This->index += fetched;
760 TRACE("-- fetched %d\n", fetched);
762 if (pceltFetched)
763 *pceltFetched = fetched;
765 if (fetched != celt)
766 return S_FALSE;
767 else
768 return S_OK;
771 static HRESULT WINAPI DEVENUM_IEnumMoniker_Skip(IEnumMoniker *iface, ULONG celt)
773 EnumMonikerImpl *This = impl_from_IEnumMoniker(iface);
774 DWORD subKeys;
776 TRACE("(%p)->(%d)\n", iface, celt);
778 /* Before incrementing, check if there are any more values to run through.
779 Some programs use the Skip() function to get the number of devices */
780 if(RegQueryInfoKeyW(This->hkey, NULL, NULL, NULL, &subKeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
782 return S_FALSE;
784 if((This->index + celt) >= subKeys)
786 return S_FALSE;
789 This->index += celt;
791 return S_OK;
794 static HRESULT WINAPI DEVENUM_IEnumMoniker_Reset(IEnumMoniker *iface)
796 EnumMonikerImpl *This = impl_from_IEnumMoniker(iface);
798 TRACE("(%p)->()\n", iface);
800 This->index = 0;
802 return S_OK;
805 static HRESULT WINAPI DEVENUM_IEnumMoniker_Clone(IEnumMoniker *iface, IEnumMoniker **ppenum)
807 FIXME("(%p)->(%p): stub\n", iface, ppenum);
809 return E_NOTIMPL;
812 /**********************************************************************
813 * IEnumMoniker_Vtbl
815 static const IEnumMonikerVtbl IEnumMoniker_Vtbl =
817 DEVENUM_IEnumMoniker_QueryInterface,
818 DEVENUM_IEnumMoniker_AddRef,
819 DEVENUM_IEnumMoniker_Release,
820 DEVENUM_IEnumMoniker_Next,
821 DEVENUM_IEnumMoniker_Skip,
822 DEVENUM_IEnumMoniker_Reset,
823 DEVENUM_IEnumMoniker_Clone
826 HRESULT DEVENUM_IEnumMoniker_Construct(HKEY hkey, IEnumMoniker ** ppEnumMoniker)
828 EnumMonikerImpl * pEnumMoniker = CoTaskMemAlloc(sizeof(EnumMonikerImpl));
829 if (!pEnumMoniker)
830 return E_OUTOFMEMORY;
832 pEnumMoniker->IEnumMoniker_iface.lpVtbl = &IEnumMoniker_Vtbl;
833 pEnumMoniker->ref = 1;
834 pEnumMoniker->index = 0;
835 pEnumMoniker->hkey = hkey;
837 *ppEnumMoniker = &pEnumMoniker->IEnumMoniker_iface;
839 DEVENUM_LockModule();
841 return S_OK;