dlls/devenum/mediacatenum.c

   1 /*
   2  *      IEnumMoniker implementation for DEVENUM.dll
   3  *
   4  * Copyright (C) 2002 Robert Shearman
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  *
  20  * NOTES ON THIS FILE:
  21  * - Implements IEnumMoniker interface which enumerates through moniker
  22  *   objects created from HKEY_CLASSES/CLSID/{DEVICE_CLSID}/Instance
  23  */
  24
  25 #include "devenum_private.h"
  26 #include "oleauto.h"
  27 #include "ocidl.h"
  28
  29 #include "wine/debug.h"
  30
  31 WINE_DEFAULT_DEBUG_CHANNEL(devenum);
  32
  33 static ULONG WINAPI DEVENUM_IEnumMoniker_AddRef(LPENUMMONIKER iface);
  34 static ULONG WINAPI DEVENUM_IMediaCatMoniker_AddRef(LPMONIKER iface);
  35 static ULONG WINAPI DEVENUM_IPropertyBag_AddRef(LPPROPERTYBAG iface);
  36
  37 typedef struct
  38 {
  39     const IPropertyBagVtbl *lpVtbl;
  40     LONG ref;
  41     HKEY hkey;
  42 } RegPropBagImpl;
  43
  44
  45 static HRESULT WINAPI DEVENUM_IPropertyBag_QueryInterface(
  46     LPPROPERTYBAG iface,
  47     REFIID riid,
  48     LPVOID *ppvObj)
  49 {
  50     RegPropBagImpl *This = (RegPropBagImpl *)iface;
  51
  52     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), ppvObj);
  53
  54     if (This == NULL || ppvObj == NULL) return E_POINTER;
  55
  56     if (IsEqualGUID(riid, &IID_IUnknown) ||
  57         IsEqualGUID(riid, &IID_IPropertyBag))
  58     {
  59         *ppvObj = iface;
  60         DEVENUM_IPropertyBag_AddRef(iface);
  61         return S_OK;
  62     }
  63
  64     FIXME("- no interface\n\tIID:\t%s\n", debugstr_guid(riid));
  65     return E_NOINTERFACE;
  66 }
  67
  68 /**********************************************************************
  69  * DEVENUM_IPropertyBag_AddRef (also IUnknown)
  70  */
  71 static ULONG WINAPI DEVENUM_IPropertyBag_AddRef(LPPROPERTYBAG iface)
  72 {
  73     RegPropBagImpl *This = (RegPropBagImpl *)iface;
  74
  75     TRACE("(%p)->() AddRef from %d\n", iface, This->ref);
  76
  77     return InterlockedIncrement(&This->ref);
  78 }
  79
  80 /**********************************************************************
  81  * DEVENUM_IPropertyBag_Release (also IUnknown)
  82  */
  83 static ULONG WINAPI DEVENUM_IPropertyBag_Release(LPPROPERTYBAG iface)
  84 {
  85     RegPropBagImpl *This = (RegPropBagImpl *)iface;
  86     ULONG ref;
  87
  88     TRACE("(%p)->() ReleaseThis->ref from %d\n", iface, This->ref);
  89
  90     ref = InterlockedDecrement(&This->ref);
  91     if (ref == 0) {
  92         RegCloseKey(This->hkey);
  93         CoTaskMemFree(This);
  94         DEVENUM_UnlockModule();
  95     }
  96     return ref;
  97 }
  98
  99 static HRESULT WINAPI DEVENUM_IPropertyBag_Read(
 100     LPPROPERTYBAG iface,
 101     LPCOLESTR pszPropName,
 102     VARIANT* pVar,
 103     IErrorLog* pErrorLog)
 104 {
 105     LPVOID pData = NULL;
 106     DWORD received;
 107     DWORD type = 0;
 108     RegPropBagImpl *This = (RegPropBagImpl *)iface;
 109     HRESULT res = S_OK;
 110     LONG reswin32;
 111
 112     TRACE("(%p)->(%s, %p, %p)\n", This, debugstr_w(pszPropName), pVar, pErrorLog);
 113
 114     if (!pszPropName || !pVar)
 115         return E_POINTER;
 116
 117     reswin32 = RegQueryValueExW(This->hkey, pszPropName, NULL, NULL, NULL, &received);
 118     res = HRESULT_FROM_WIN32(reswin32);
 119
 120     if (SUCCEEDED(res))
 121     {
 122         pData = HeapAlloc(GetProcessHeap(), 0, received);
 123
 124         /* work around a GCC bug that occurs here unless we use the reswin32 variable as well */
 125         reswin32 = RegQueryValueExW(This->hkey, pszPropName, NULL, &type, pData, &received);
 126         res = HRESULT_FROM_WIN32(reswin32);
 127     }
 128
 129     if (SUCCEEDED(res))
 130     {
 131         res = E_INVALIDARG; /* assume we cannot coerce into right type */
 132
 133         TRACE("Read %d bytes (%s)\n", received, type == REG_SZ ? debugstr_w(pData) : "binary data");
 134
 135         switch (type)
 136         {
 137         case REG_SZ:
 138             switch (V_VT(pVar))
 139             {
 140             case VT_LPWSTR:
 141                 V_UNION(pVar, bstrVal) = CoTaskMemAlloc(received);
 142                 memcpy(V_UNION(pVar, bstrVal), pData, received);
 143                 res = S_OK;
 144                 break;
 145             case VT_EMPTY:
 146                 V_VT(pVar) = VT_BSTR;
 147             /* fall through */
 148             case VT_BSTR:
 149                 V_UNION(pVar, bstrVal) = SysAllocStringLen(pData, received/sizeof(WCHAR) - 1);
 150                 res = S_OK;
 151                 break;
 152             }
 153             break;
 154         case REG_DWORD:
 155             TRACE("REG_DWORD: %x\n", *(DWORD *)pData);
 156             switch (V_VT(pVar))
 157             {
 158             case VT_EMPTY:
 159                 V_VT(pVar) = VT_I4;
 160                 /* fall through */
 161             case VT_I4:
 162             case VT_UI4:
 163                 V_UNION(pVar, ulVal) = *(DWORD *)pData;
 164                 res = S_OK;
 165                 break;
 166             }
 167             break;
 168         case REG_BINARY:
 169             {
 170                 SAFEARRAYBOUND bound;
 171                 void * pArrayElements;
 172                 bound.lLbound = 0;
 173                 bound.cElements = received;
 174                 TRACE("REG_BINARY: len = %d\n", received);
 175                 switch (V_VT(pVar))
 176                 {
 177                 case VT_EMPTY:
 178                     V_VT(pVar) = VT_ARRAY | VT_UI1;
 179                     /* fall through */
 180                 case VT_ARRAY | VT_UI1:
 181                     if (!(V_UNION(pVar, parray) = SafeArrayCreate(VT_UI1, 1, &bound)))
 182                         res = E_OUTOFMEMORY;
 183                     else
 184                         res = S_OK;
 185                     break;
 186                 }
 187
 188                 if (res == E_INVALIDARG)
 189                     break;
 190
 191                 res = SafeArrayAccessData(V_UNION(pVar, parray), &pArrayElements);
 192                 if (FAILED(res))
 193                     break;
 194
 195                 CopyMemory(pArrayElements, pData, received);
 196                 res = SafeArrayUnaccessData(V_UNION(pVar, parray));
 197                 break;
 198             }
 199         }
 200         if (res == E_INVALIDARG)
 201             FIXME("Variant type %x not supported for regtype %x\n", V_VT(pVar), type);
 202     }
 203
 204     HeapFree(GetProcessHeap(), 0, pData);
 205
 206     TRACE("<- %x\n", res);
 207     return res;
 208 }
 209
 210 static HRESULT WINAPI DEVENUM_IPropertyBag_Write(
 211     LPPROPERTYBAG iface,
 212     LPCOLESTR pszPropName,
 213     VARIANT* pVar)
 214 {
 215     RegPropBagImpl *This = (RegPropBagImpl *)iface;
 216     LPVOID lpData = NULL;
 217     DWORD cbData = 0;
 218     DWORD dwType = 0;
 219     HRESULT res = S_OK;
 220
 221     TRACE("(%p)->(%s, %p)\n", This, debugstr_w(pszPropName), pVar);
 222
 223     switch (V_VT(pVar))
 224     {
 225     case VT_BSTR:
 226         TRACE("writing %s\n", debugstr_w(V_UNION(pVar, bstrVal)));
 227         lpData = V_UNION(pVar, bstrVal);
 228         dwType = REG_SZ;
 229         cbData = (lstrlenW(V_UNION(pVar, bstrVal)) + 1) * sizeof(WCHAR);
 230         break;
 231     case VT_I4:
 232     case VT_UI4:
 233         TRACE("writing %u\n", V_UNION(pVar, ulVal));
 234         lpData = &V_UNION(pVar, ulVal);
 235         dwType = REG_DWORD;
 236         cbData = sizeof(DWORD);
 237         break;
 238     case VT_ARRAY | VT_UI1:
 239     {
 240         LONG lUbound = 0;
 241         LONG lLbound = 0;
 242         dwType = REG_BINARY;
 243         res = SafeArrayGetLBound(V_UNION(pVar, parray), 1, &lLbound);
 244         res = SafeArrayGetUBound(V_UNION(pVar, parray), 1, &lUbound);
 245         cbData = (lUbound - lLbound + 1) /* * sizeof(BYTE)*/;
 246         TRACE("cbData: %d\n", cbData);
 247         res = SafeArrayAccessData(V_UNION(pVar, parray), &lpData);
 248         break;
 249     }
 250     default:
 251         FIXME("Variant type %d not handled\n", V_VT(pVar));
 252         return E_FAIL;
 253     }
 254
 255     if (RegSetValueExW(This->hkey,
 256                        pszPropName, 0,
 257                        dwType, lpData, cbData) != ERROR_SUCCESS)
 258         res = E_FAIL;
 259
 260     if (V_VT(pVar) & VT_ARRAY)
 261         res = SafeArrayUnaccessData(V_UNION(pVar, parray));
 262
 263     return res;
 264 }
 265
 266 static const IPropertyBagVtbl IPropertyBag_Vtbl =
 267 {
 268     DEVENUM_IPropertyBag_QueryInterface,
 269     DEVENUM_IPropertyBag_AddRef,
 270     DEVENUM_IPropertyBag_Release,
 271     DEVENUM_IPropertyBag_Read,
 272     DEVENUM_IPropertyBag_Write
 273 };
 274
 275 static HRESULT DEVENUM_IPropertyBag_Construct(HANDLE hkey, IPropertyBag **ppBag)
 276 {
 277     RegPropBagImpl * rpb = CoTaskMemAlloc(sizeof(RegPropBagImpl));
 278     if (!rpb)
 279         return E_OUTOFMEMORY;
 280     rpb->lpVtbl = &IPropertyBag_Vtbl;
 281     rpb->ref = 1;
 282     rpb->hkey = hkey;
 283     *ppBag = (IPropertyBag*)rpb;
 284     DEVENUM_LockModule();
 285     return S_OK;
 286 }
 287
 288
 289 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_QueryInterface(
 290     LPMONIKER iface,
 291     REFIID riid,
 292     LPVOID *ppvObj)
 293 {
 294     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 295     TRACE("\n\tIID:\t%s\n",debugstr_guid(riid));
 296
 297     if (This == NULL || ppvObj == NULL) return E_POINTER;
 298
 299     *ppvObj = NULL;
 300
 301     if (IsEqualGUID(riid, &IID_IUnknown) ||
 302         IsEqualGUID(riid, &IID_IPersist) ||
 303         IsEqualGUID(riid, &IID_IPersistStream) ||
 304         IsEqualGUID(riid, &IID_IMoniker))
 305     {
 306         *ppvObj = iface;
 307         DEVENUM_IMediaCatMoniker_AddRef(iface);
 308         return S_OK;
 309     }
 310
 311     FIXME("- no interface\n\tIID:\t%s\n", debugstr_guid(riid));
 312     return E_NOINTERFACE;
 313 }
 314
 315 /**********************************************************************
 316  * DEVENUM_IMediaCatMoniker_AddRef (also IUnknown)
 317  */
 318 static ULONG WINAPI DEVENUM_IMediaCatMoniker_AddRef(LPMONIKER iface)
 319 {
 320     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 321     TRACE("\n");
 322
 323     return InterlockedIncrement(&This->ref);
 324 }
 325
 326 /**********************************************************************
 327  * DEVENUM_IMediaCatMoniker_Release (also IUnknown)
 328  */
 329 static ULONG WINAPI DEVENUM_IMediaCatMoniker_Release(LPMONIKER iface)
 330 {
 331     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 332     ULONG ref;
 333     TRACE("\n");
 334
 335     ref = InterlockedDecrement(&This->ref);
 336     if (ref == 0) {
 337         RegCloseKey(This->hkey);
 338         CoTaskMemFree(This);
 339         DEVENUM_UnlockModule();
 340     }
 341     return ref;
 342 }
 343
 344 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetClassID(
 345     LPMONIKER iface,
 346     CLSID* pClassID)
 347 {
 348     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 349     FIXME("(%p)->(%p): stub\n", This, pClassID);
 350
 351     if (pClassID == NULL)
 352         return E_POINTER;
 353
 354     return E_NOTIMPL;
 355 }
 356
 357 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsDirty(LPMONIKER iface)
 358 {
 359     FIXME("(%p)->(): stub\n", iface);
 360
 361     return S_FALSE;
 362 }
 363
 364 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Load(LPMONIKER iface, IStream* pStm)
 365 {
 366     FIXME("(%p)->(%p): stub\n", iface, pStm);
 367
 368     return E_NOTIMPL;
 369 }
 370
 371 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Save(LPMONIKER iface, IStream* pStm, BOOL fClearDirty)
 372 {
 373     FIXME("(%p)->(%p, %s): stub\n", iface, pStm, fClearDirty ? "true" : "false");
 374
 375     return STG_E_CANTSAVE;
 376 }
 377
 378 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetSizeMax(
 379     LPMONIKER iface,
 380     ULARGE_INTEGER* pcbSize)
 381 {
 382     FIXME("(%p)->(%p): stub\n", iface, pcbSize);
 383
 384     ZeroMemory(pcbSize, sizeof(*pcbSize));
 385
 386     return S_OK;
 387 }
 388
 389 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_BindToObject(
 390     LPMONIKER iface,
 391     IBindCtx* pbc,
 392     IMoniker* pmkToLeft,
 393     REFIID riidResult,
 394     void** ppvResult)
 395 {
 396     IUnknown * pObj = NULL;
 397     IPropertyBag * pProp = NULL;
 398     CLSID clsID;
 399     VARIANT var;
 400     HRESULT res = E_FAIL;
 401
 402     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 403
 404     VariantInit(&var);
 405
 406     TRACE("(%p)->(%p, %p, %s, %p)\n", This, pbc, pmkToLeft, debugstr_guid(riidResult), ppvResult);
 407
 408     *ppvResult = NULL;
 409
 410     if(pmkToLeft==NULL)
 411     {
 412             /* first activation of this class */
 413             LPVOID pvptr;
 414             res=IMoniker_BindToStorage(iface, NULL, NULL, &IID_IPropertyBag, &pvptr);
 415             pProp = pvptr;
 416             if (SUCCEEDED(res))
 417             {
 418                 V_VT(&var) = VT_LPWSTR;
 419                 res = IPropertyBag_Read(pProp, clsid_keyname, &var, NULL);
 420             }
 421             if (SUCCEEDED(res))
 422             {
 423                 res = CLSIDFromString(V_UNION(&var,bstrVal), &clsID);
 424                 CoTaskMemFree(V_UNION(&var, bstrVal));
 425             }
 426             if (SUCCEEDED(res))
 427             {
 428                 res=CoCreateInstance(&clsID,NULL,CLSCTX_ALL,&IID_IUnknown,&pvptr);
 429                 pObj = pvptr;
 430             }
 431     }
 432
 433     if (pObj!=NULL)
 434     {
 435         /* get the requested interface from the loaded class */
 436         res = S_OK;
 437         if (pProp) {
 438            HRESULT res2;
 439            LPVOID ppv = NULL;
 440            res2 = IUnknown_QueryInterface(pObj, &IID_IPersistPropertyBag, &ppv);
 441            if (SUCCEEDED(res2)) {
 442               res = IPersistPropertyBag_Load((IPersistPropertyBag *) ppv, pProp, NULL);
 443               IPersistPropertyBag_Release((IPersistPropertyBag *) ppv);
 444            }
 445         }
 446         if (SUCCEEDED(res))
 447            res= IUnknown_QueryInterface(pObj,riidResult,ppvResult);
 448         IUnknown_Release(pObj);
 449     }
 450
 451     if (pProp)
 452     {
 453         IPropertyBag_Release(pProp);
 454     }
 455
 456     TRACE("<- 0x%x\n", res);
 457
 458     return res;
 459 }
 460
 461 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_BindToStorage(
 462     LPMONIKER iface,
 463     IBindCtx* pbc,
 464     IMoniker* pmkToLeft,
 465     REFIID riid,
 466     void** ppvObj)
 467 {
 468     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 469     TRACE("(%p)->(%p, %p, %s, %p)\n", This, pbc, pmkToLeft, debugstr_guid(riid), ppvObj);
 470
 471     *ppvObj = NULL;
 472
 473     if (pbc || pmkToLeft)
 474         return MK_E_NOSTORAGE;
 475
 476     if (IsEqualGUID(riid, &IID_IPropertyBag))
 477     {
 478         HANDLE hkey;
 479         DuplicateHandle(GetCurrentProcess(), This->hkey, GetCurrentProcess(), &hkey, 0, 0, DUPLICATE_SAME_ACCESS);
 480         return DEVENUM_IPropertyBag_Construct(hkey, (IPropertyBag**)ppvObj);
 481     }
 482
 483     return MK_E_NOSTORAGE;
 484 }
 485
 486 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Reduce(
 487     LPMONIKER iface,
 488     IBindCtx* pbc,
 489     DWORD dwReduceHowFar,
 490     IMoniker** ppmkToLeft,
 491     IMoniker** ppmkReduced)
 492 {
 493     TRACE("(%p)->(%p, %d, %p, %p)\n", iface, pbc, dwReduceHowFar, ppmkToLeft, ppmkReduced);
 494
 495     if (ppmkToLeft)
 496         *ppmkToLeft = NULL;
 497     *ppmkReduced = iface;
 498
 499     return MK_S_REDUCED_TO_SELF;
 500 }
 501
 502 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_ComposeWith(
 503     LPMONIKER iface,
 504     IMoniker* pmkRight,
 505     BOOL fOnlyIfNotGeneric,
 506     IMoniker** ppmkComposite)
 507 {
 508     FIXME("(%p)->(%p, %s, %p): stub\n", iface, pmkRight, fOnlyIfNotGeneric ? "true" : "false", ppmkComposite);
 509
 510     /* FIXME: use CreateGenericComposite? */
 511     *ppmkComposite = NULL;
 512
 513     return E_NOTIMPL;
 514 }
 515
 516 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Enum(
 517     LPMONIKER iface,
 518     BOOL fForward,
 519     IEnumMoniker** ppenumMoniker)
 520 {
 521     FIXME("(%p)->(%s, %p): stub\n", iface, fForward ? "true" : "false", ppenumMoniker);
 522
 523     *ppenumMoniker = NULL;
 524
 525     return S_OK;
 526 }
 527
 528 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsEqual(
 529     LPMONIKER iface,
 530     IMoniker* pmkOtherMoniker)
 531 {
 532     FIXME("(%p)->(%p): stub\n", iface, pmkOtherMoniker);
 533
 534     return E_NOTIMPL;
 535 }
 536
 537 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Hash(
 538     LPMONIKER iface,
 539     DWORD* pdwHash)
 540 {
 541     TRACE("(%p)->(%p)\n", iface, pdwHash);
 542
 543     *pdwHash = 0;
 544
 545     return S_OK;
 546 }
 547
 548 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsRunning(
 549     LPMONIKER iface,
 550     IBindCtx* pbc,
 551     IMoniker* pmkToLeft,
 552     IMoniker* pmkNewlyRunning)
 553 {
 554     FIXME("(%p)->(%p, %p, %p): stub\n", iface, pbc, pmkToLeft, pmkNewlyRunning);
 555
 556     return S_FALSE;
 557 }
 558
 559 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetTimeOfLastChange(
 560     LPMONIKER iface,
 561     IBindCtx* pbc,
 562     IMoniker* pmkToLeft,
 563     FILETIME* pFileTime)
 564 {
 565     TRACE("(%p)->(%p, %p, %p)\n", iface, pbc, pmkToLeft, pFileTime);
 566
 567     pFileTime->dwLowDateTime = 0xFFFFFFFF;
 568     pFileTime->dwHighDateTime = 0x7FFFFFFF;
 569
 570     return MK_E_UNAVAILABLE;
 571 }
 572
 573 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_Inverse(
 574     LPMONIKER iface,
 575     IMoniker** ppmk)
 576 {
 577     TRACE("(%p)->(%p)\n", iface, ppmk);
 578
 579     *ppmk = NULL;
 580
 581     return MK_E_NOINVERSE;
 582 }
 583
 584 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_CommonPrefixWith(
 585     LPMONIKER iface,
 586     IMoniker* pmkOtherMoniker,
 587     IMoniker** ppmkPrefix)
 588 {
 589     TRACE("(%p)->(%p, %p)\n", iface, pmkOtherMoniker, ppmkPrefix);
 590
 591     *ppmkPrefix = NULL;
 592
 593     return MK_E_NOPREFIX;
 594 }
 595
 596 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_RelativePathTo(
 597     LPMONIKER iface,
 598     IMoniker* pmkOther,
 599     IMoniker** ppmkRelPath)
 600 {
 601     TRACE("(%p)->(%p, %p)\n", iface, pmkOther, ppmkRelPath);
 602
 603     *ppmkRelPath = pmkOther;
 604
 605     return MK_S_HIM;
 606 }
 607
 608 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_GetDisplayName(
 609     LPMONIKER iface,
 610     IBindCtx* pbc,
 611     IMoniker* pmkToLeft,
 612     LPOLESTR* ppszDisplayName)
 613 {
 614     MediaCatMoniker *This = (MediaCatMoniker *)iface;
 615     WCHAR wszBuffer[MAX_PATH];
 616     static const WCHAR wszFriendlyName[] = {'F','r','i','e','n','d','l','y','N','a','m','e',0};
 617     LONG received = sizeof(wszFriendlyName);
 618
 619     TRACE("(%p)->(%p, %p, %p)\n", iface, pbc, pmkToLeft, ppszDisplayName);
 620
 621     *ppszDisplayName = NULL;
 622
 623     /* FIXME: should this be the weird stuff we have to parse in IParseDisplayName? */
 624     if (RegQueryValueW(This->hkey, wszFriendlyName, wszBuffer, &received) == ERROR_SUCCESS)
 625     {
 626         *ppszDisplayName = CoTaskMemAlloc(received);
 627         strcpyW(*ppszDisplayName, wszBuffer);
 628         return S_OK;
 629     }
 630
 631     return E_FAIL;
 632 }
 633
 634 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_ParseDisplayName(
 635     LPMONIKER iface,
 636     IBindCtx* pbc,
 637     IMoniker* pmkToLeft,
 638     LPOLESTR pszDisplayName,
 639     ULONG* pchEaten,
 640     IMoniker** ppmkOut)
 641 {
 642     FIXME("(%p)->(%p, %p, %s, %p, %p)\n", iface, pbc, pmkToLeft, debugstr_w(pszDisplayName), pchEaten, ppmkOut);
 643
 644     *pchEaten = 0;
 645     *ppmkOut = NULL;
 646
 647     return MK_E_SYNTAX;
 648 }
 649
 650 static HRESULT WINAPI DEVENUM_IMediaCatMoniker_IsSystemMoniker(
 651     LPMONIKER iface,
 652     DWORD* pdwMksys)
 653 {
 654     TRACE("(%p)->(%p)\n", iface, pdwMksys);
 655
 656     return S_FALSE;
 657 }
 658
 659 static const IMonikerVtbl IMoniker_Vtbl =
 660 {
 661     DEVENUM_IMediaCatMoniker_QueryInterface,
 662     DEVENUM_IMediaCatMoniker_AddRef,
 663     DEVENUM_IMediaCatMoniker_Release,
 664     DEVENUM_IMediaCatMoniker_GetClassID,
 665     DEVENUM_IMediaCatMoniker_IsDirty,
 666     DEVENUM_IMediaCatMoniker_Load,
 667     DEVENUM_IMediaCatMoniker_Save,
 668     DEVENUM_IMediaCatMoniker_GetSizeMax,
 669     DEVENUM_IMediaCatMoniker_BindToObject,
 670     DEVENUM_IMediaCatMoniker_BindToStorage,
 671     DEVENUM_IMediaCatMoniker_Reduce,
 672     DEVENUM_IMediaCatMoniker_ComposeWith,
 673     DEVENUM_IMediaCatMoniker_Enum,
 674     DEVENUM_IMediaCatMoniker_IsEqual,
 675     DEVENUM_IMediaCatMoniker_Hash,
 676     DEVENUM_IMediaCatMoniker_IsRunning,
 677     DEVENUM_IMediaCatMoniker_GetTimeOfLastChange,
 678     DEVENUM_IMediaCatMoniker_Inverse,
 679     DEVENUM_IMediaCatMoniker_CommonPrefixWith,
 680     DEVENUM_IMediaCatMoniker_RelativePathTo,
 681     DEVENUM_IMediaCatMoniker_GetDisplayName,
 682     DEVENUM_IMediaCatMoniker_ParseDisplayName,
 683     DEVENUM_IMediaCatMoniker_IsSystemMoniker
 684 };
 685
 686 MediaCatMoniker * DEVENUM_IMediaCatMoniker_Construct(void)
 687 {
 688     MediaCatMoniker * pMoniker = NULL;
 689     pMoniker = CoTaskMemAlloc(sizeof(MediaCatMoniker));
 690     if (!pMoniker)
 691         return NULL;
 692
 693     pMoniker->lpVtbl = &IMoniker_Vtbl;
 694     pMoniker->ref = 0;
 695     pMoniker->hkey = NULL;
 696
 697     DEVENUM_IMediaCatMoniker_AddRef((LPMONIKER)pMoniker);
 698
 699     DEVENUM_LockModule();
 700
 701     return pMoniker;
 702 }
 703
 704 /**********************************************************************
 705  * DEVENUM_IEnumMoniker_QueryInterface (also IUnknown)
 706  */
 707 static HRESULT WINAPI DEVENUM_IEnumMoniker_QueryInterface(
 708     LPENUMMONIKER iface,
 709     REFIID riid,
 710     LPVOID *ppvObj)
 711 {
 712     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
 713
 714     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), ppvObj);
 715
 716     if (This == NULL || ppvObj == NULL) return E_POINTER;
 717
 718     if (IsEqualGUID(riid, &IID_IUnknown) ||
 719         IsEqualGUID(riid, &IID_IEnumMoniker))
 720     {
 721         *ppvObj = iface;
 722         DEVENUM_IEnumMoniker_AddRef(iface);
 723         return S_OK;
 724     }
 725
 726     FIXME("- no interface\n\tIID:\t%s\n", debugstr_guid(riid));
 727     return E_NOINTERFACE;
 728 }
 729
 730 /**********************************************************************
 731  * DEVENUM_IEnumMoniker_AddRef (also IUnknown)
 732  */
 733 static ULONG WINAPI DEVENUM_IEnumMoniker_AddRef(LPENUMMONIKER iface)
 734 {
 735     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
 736     ULONG ref = InterlockedIncrement(&This->ref);
 737
 738     TRACE("(%p)->() AddRef from %d\n", iface, ref - 1);
 739
 740     return ref;
 741 }
 742
 743 /**********************************************************************
 744  * DEVENUM_IEnumMoniker_Release (also IUnknown)
 745  */
 746 static ULONG WINAPI DEVENUM_IEnumMoniker_Release(LPENUMMONIKER iface)
 747 {
 748     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
 749     ULONG ref = InterlockedDecrement(&This->ref);
 750
 751     TRACE("(%p)->() Release from %d\n", iface, ref + 1);
 752
 753     if (!ref)
 754     {
 755         RegCloseKey(This->hkey);
 756         CoTaskMemFree(This);
 757         DEVENUM_UnlockModule();
 758         return 0;
 759     }
 760     return ref;
 761 }
 762
 763 static HRESULT WINAPI DEVENUM_IEnumMoniker_Next(LPENUMMONIKER iface, ULONG celt, IMoniker ** rgelt, ULONG * pceltFetched)
 764 {
 765     WCHAR buffer[MAX_PATH + 1];
 766     LONG res;
 767     ULONG fetched = 0;
 768     MediaCatMoniker * pMoniker;
 769     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
 770
 771     TRACE("(%p)->(%d, %p, %p)\n", iface, celt, rgelt, pceltFetched);
 772
 773     while (fetched < celt)
 774     {
 775         res = RegEnumKeyW(This->hkey, This->index, buffer, sizeof(buffer) / sizeof(WCHAR));
 776         if (res != ERROR_SUCCESS)
 777         {
 778             break;
 779         }
 780         pMoniker = DEVENUM_IMediaCatMoniker_Construct();
 781         if (!pMoniker)
 782             return E_OUTOFMEMORY;
 783
 784         if (RegOpenKeyW(This->hkey, buffer, &pMoniker->hkey) != ERROR_SUCCESS)
 785         {
 786             DEVENUM_IMediaCatMoniker_Release((LPMONIKER)pMoniker);
 787             break;
 788         }
 789         rgelt[fetched] = (LPMONIKER)pMoniker;
 790         fetched++;
 791     }
 792
 793     This->index += fetched;
 794
 795     TRACE("-- fetched %d\n", fetched);
 796
 797     if (pceltFetched)
 798         *pceltFetched = fetched;
 799
 800     if (fetched != celt)
 801         return S_FALSE;
 802     else
 803         return S_OK;
 804 }
 805
 806 static HRESULT WINAPI DEVENUM_IEnumMoniker_Skip(LPENUMMONIKER iface, ULONG celt)
 807 {
 808     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
 809     DWORD subKeys;
 810
 811     TRACE("(%p)->(%d)\n", iface, celt);
 812
 813     /* Before incrementing, check if there are any more values to run thru.
 814        Some programs use the Skip() function to get the amount of devices */
 815     if(RegQueryInfoKeyW(This->hkey, NULL, NULL, NULL, &subKeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
 816     {
 817         return S_FALSE;
 818     }
 819     if((This->index + celt) >= subKeys)
 820     {
 821         return S_FALSE;
 822     }
 823
 824     This->index += celt;
 825
 826     return S_OK;
 827 }
 828
 829 static HRESULT WINAPI DEVENUM_IEnumMoniker_Reset(LPENUMMONIKER iface)
 830 {
 831     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
 832
 833     TRACE("(%p)->()\n", iface);
 834
 835     This->index = 0;
 836
 837     return S_OK;
 838 }
 839
 840 static HRESULT WINAPI DEVENUM_IEnumMoniker_Clone(LPENUMMONIKER iface, IEnumMoniker ** ppenum)
 841 {
 842     FIXME("(%p)->(%p): stub\n", iface, ppenum);
 843
 844     return E_NOTIMPL;
 845 }
 846
 847 /**********************************************************************
 848  * IEnumMoniker_Vtbl
 849  */
 850 static const IEnumMonikerVtbl IEnumMoniker_Vtbl =
 851 {
 852     DEVENUM_IEnumMoniker_QueryInterface,
 853     DEVENUM_IEnumMoniker_AddRef,
 854     DEVENUM_IEnumMoniker_Release,
 855     DEVENUM_IEnumMoniker_Next,
 856     DEVENUM_IEnumMoniker_Skip,
 857     DEVENUM_IEnumMoniker_Reset,
 858     DEVENUM_IEnumMoniker_Clone
 859 };
 860
 861 HRESULT DEVENUM_IEnumMoniker_Construct(HKEY hkey, IEnumMoniker ** ppEnumMoniker)
 862 {
 863     EnumMonikerImpl * pEnumMoniker = CoTaskMemAlloc(sizeof(EnumMonikerImpl));
 864     if (!pEnumMoniker)
 865         return E_OUTOFMEMORY;
 866
 867     pEnumMoniker->lpVtbl = &IEnumMoniker_Vtbl;
 868     pEnumMoniker->ref = 1;
 869     pEnumMoniker->index = 0;
 870     pEnumMoniker->hkey = hkey;
 871
 872     *ppEnumMoniker = (IEnumMoniker *)pEnumMoniker;
 873
 874     DEVENUM_LockModule();
 875
 876     return S_OK;
 877 }