dlls/quartz/filtermapper.c

   1 /*
   2  * IFilterMapper & IFilterMapper2 Implementations
   3  *
   4  * Copyright 2003 Robert Shearman
   5  * Copyright 2004 Christian Costa
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #define NONAMELESSUNION
  23 #define NONAMELESSSTRUCT
  24 #include <stdarg.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "winreg.h"
  30 #include "winerror.h"
  31
  32 #include "quartz_private.h"
  33
  34 #include "ole2.h"
  35 #include "olectl.h"
  36 #include "strmif.h"
  37 #include "wine/unicode.h"
  38 #include "uuids.h"
  39
  40 #include "wine/debug.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
  43
  44 #define ARRAYSIZE(array) (sizeof(array)/sizeof((array)[0]))
  45
  46 /* Unexposed IAMFilterData interface */
  47 typedef struct IAMFilterData IAMFilterData;
  48
  49 typedef struct IAMFilterDataVtbl
  50 {
  51     BEGIN_INTERFACE
  52
  53     /*** IUnknown methods ***/
  54     HRESULT (STDMETHODCALLTYPE *QueryInterface)(
  55         IAMFilterData *This,
  56         REFIID riid,
  57         void **ppvObject);
  58
  59     ULONG (STDMETHODCALLTYPE *AddRef)(
  60         IAMFilterData *This);
  61
  62     ULONG (STDMETHODCALLTYPE *Release)(
  63         IAMFilterData *This);
  64
  65     /*** IAMFilterData methods ***/
  66     HRESULT (STDMETHODCALLTYPE *ParseFilterData)(
  67         IAMFilterData *This,
  68         BYTE *pData,
  69         ULONG cb,
  70         BYTE **ppRegFilter2);
  71
  72     HRESULT (STDMETHODCALLTYPE *CreateFilterData)(
  73         IAMFilterData* This,
  74         REGFILTER2 *prf2,
  75         BYTE **pRegFilterData,
  76         ULONG *pcb);
  77
  78     END_INTERFACE
  79 } IAMFilterDataVtbl;
  80 struct IAMFilterData
  81 {
  82     const IAMFilterDataVtbl *lpVtbl;
  83 };
  84 const GUID IID_IAMFilterData = {
  85  0x97f7c4d4, 0x547b, 0x4a5f, { 0x83,0x32, 0x53,0x64,0x30,0xad,0x2e,0x4d }
  86 };
  87
  88
  89 typedef struct FilterMapper2Impl
  90 {
  91     const IFilterMapper2Vtbl *lpVtbl;
  92     const IFilterMapperVtbl  *lpVtblFilterMapper;
  93     const IAMFilterDataVtbl  *lpVtblAMFilterData;
  94     LONG refCount;
  95 } FilterMapper2Impl;
  96
  97 static const IFilterMapper2Vtbl fm2vtbl;
  98 static const IFilterMapperVtbl fmvtbl;
  99 static const IAMFilterDataVtbl AMFilterDataVtbl;
 100
 101 static inline FilterMapper2Impl *impl_from_IFilterMapper( IFilterMapper *iface )
 102 {
 103     return (FilterMapper2Impl *)((char*)iface - FIELD_OFFSET(FilterMapper2Impl, lpVtblFilterMapper));
 104 }
 105
 106 static inline FilterMapper2Impl *impl_from_IAMFilterData( IAMFilterData *iface )
 107 {
 108     return (FilterMapper2Impl *)((char*)iface - FIELD_OFFSET(FilterMapper2Impl, lpVtblAMFilterData));
 109 }
 110
 111 static const WCHAR wszClsidSlash[] = {'C','L','S','I','D','\\',0};
 112 static const WCHAR wszSlashInstance[] = {'\\','I','n','s','t','a','n','c','e','\\',0};
 113 static const WCHAR wszSlash[] = {'\\',0};
 114
 115 /* CLSID property in media category Moniker */
 116 static const WCHAR wszClsidName[] = {'C','L','S','I','D',0};
 117 /* FriendlyName property in media category Moniker */
 118 static const WCHAR wszFriendlyName[] = {'F','r','i','e','n','d','l','y','N','a','m','e',0};
 119 /* Merit property in media category Moniker (CLSID_ActiveMovieCategories only) */
 120 static const WCHAR wszMeritName[] = {'M','e','r','i','t',0};
 121 /* FilterData property in media category Moniker (not CLSID_ActiveMovieCategories) */
 122 static const WCHAR wszFilterDataName[] = {'F','i','l','t','e','r','D','a','t','a',0};
 123 /* For filters registered with IFilterMapper */
 124 static const WCHAR wszFilterSlash[] = {'F','i','l','t','e','r','\\',0};
 125 static const WCHAR wszFilter[] = {'F','i','l','t','e','r',0};
 126 /* For pins registered with IFilterMapper */
 127 static const WCHAR wszPins[] = {'P','i','n','s',0};
 128 static const WCHAR wszAllowedMany[] = {'A','l','l','o','w','e','d','M','a','n','y',0};
 129 static const WCHAR wszAllowedZero[] = {'A','l','l','o','w','e','d','Z','e','r','o',0};
 130 static const WCHAR wszDirection[] = {'D','i','r','e','c','t','i','o','n',0};
 131 static const WCHAR wszIsRendered[] = {'I','s','R','e','n','d','e','r','e','d',0};
 132 /* For types registered with IFilterMapper */
 133 static const WCHAR wszTypes[] = {'T','y','p','e','s',0};
 134
 135
 136 /* registry format for REGFILTER2 */
 137 struct REG_RF
 138 {
 139     DWORD dwVersion;
 140     DWORD dwMerit;
 141     DWORD dwPins;
 142     DWORD dwUnused;
 143 };
 144
 145 struct REG_RFP
 146 {
 147     BYTE signature[4]; /* e.g. "0pi3" */
 148     DWORD dwFlags;
 149     DWORD dwInstances;
 150     DWORD dwMediaTypes;
 151     DWORD dwMediums;
 152     DWORD bCategory; /* is there a category clsid? */
 153     /* optional: dwOffsetCategoryClsid */
 154 };
 155
 156 struct REG_TYPE
 157 {
 158     BYTE signature[4]; /* e.g. "0ty3" */
 159     DWORD dwUnused;
 160     DWORD dwOffsetMajor;
 161     DWORD dwOffsetMinor;
 162 };
 163
 164 struct MONIKER_MERIT
 165 {
 166     IMoniker * pMoniker;
 167     DWORD dwMerit;
 168 };
 169
 170 struct Vector
 171 {
 172     LPBYTE pData;
 173     int capacity; /* in bytes */
 174     int current; /* pointer to next free byte */
 175 };
 176
 177 /* returns the position it was added at */
 178 static int add_data(struct Vector * v, const BYTE * pData, int size)
 179 {
 180     int index = v->current;
 181     if (v->current + size > v->capacity)
 182     {
 183         LPBYTE pOldData = v->pData;
 184         v->capacity = (v->capacity + size) * 2;
 185         v->pData = CoTaskMemAlloc(v->capacity);
 186         memcpy(v->pData, pOldData, v->current);
 187         CoTaskMemFree(pOldData);
 188     }
 189     memcpy(v->pData + v->current, pData, size);
 190     v->current += size;
 191     return index;
 192 }
 193
 194 static int find_data(const struct Vector * v, const BYTE * pData, int size)
 195 {
 196     int index;
 197     for (index = 0; index < v->current; index++)
 198         if (!memcmp(v->pData + index, pData, size))
 199             return index;
 200     /* not found */
 201     return -1;
 202 }
 203
 204 static void delete_vector(struct Vector * v)
 205 {
 206     CoTaskMemFree(v->pData);
 207     v->current = 0;
 208     v->capacity = 0;
 209 }
 210
 211 HRESULT FilterMapper2_create(IUnknown *pUnkOuter, LPVOID *ppObj)
 212 {
 213     FilterMapper2Impl * pFM2impl;
 214
 215     TRACE("(%p, %p)\n", pUnkOuter, ppObj);
 216
 217     if (pUnkOuter)
 218         return CLASS_E_NOAGGREGATION;
 219
 220     pFM2impl = CoTaskMemAlloc(sizeof(*pFM2impl));
 221     if (!pFM2impl)
 222         return E_OUTOFMEMORY;
 223
 224     pFM2impl->lpVtbl = &fm2vtbl;
 225     pFM2impl->lpVtblFilterMapper = &fmvtbl;
 226     pFM2impl->lpVtblAMFilterData = &AMFilterDataVtbl;
 227     pFM2impl->refCount = 1;
 228
 229     *ppObj = pFM2impl;
 230
 231     TRACE("-- created at %p\n", pFM2impl);
 232
 233     return S_OK;
 234 }
 235
 236 HRESULT FilterMapper_create(IUnknown *pUnkOuter, LPVOID *ppObj)
 237 {
 238     FilterMapper2Impl *pFM2impl;
 239     HRESULT hr;
 240
 241     TRACE("(%p, %p)\n", pUnkOuter, ppObj);
 242
 243     hr = FilterMapper2_create(pUnkOuter, (LPVOID*)&pFM2impl);
 244     if (FAILED(hr))
 245         return hr;
 246
 247     *ppObj = &pFM2impl->lpVtblFilterMapper;
 248
 249     return hr;
 250 }
 251
 252 /*** IUnknown methods ***/
 253
 254 static HRESULT WINAPI FilterMapper2_QueryInterface(IFilterMapper2 * iface, REFIID riid, LPVOID *ppv)
 255 {
 256     FilterMapper2Impl *This = (FilterMapper2Impl *)iface;
 257
 258     TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), ppv);
 259
 260     *ppv = NULL;
 261
 262     if (IsEqualIID(riid, &IID_IUnknown))
 263         *ppv = iface;
 264     else if (IsEqualIID(riid, &IID_IFilterMapper2))
 265         *ppv = iface;
 266     else if (IsEqualIID(riid, &IID_IFilterMapper))
 267         *ppv = &This->lpVtblFilterMapper;
 268     else if (IsEqualIID(riid, &IID_IAMFilterData))
 269         *ppv = &This->lpVtblAMFilterData;
 270
 271     if (*ppv != NULL)
 272     {
 273         IUnknown_AddRef((IUnknown *)*ppv);
 274         return S_OK;
 275     }
 276
 277     FIXME("No interface for %s\n", debugstr_guid(riid));
 278     return E_NOINTERFACE;
 279 }
 280
 281 static ULONG WINAPI FilterMapper2_AddRef(IFilterMapper2 * iface)
 282 {
 283     FilterMapper2Impl *This = (FilterMapper2Impl *)iface;
 284     ULONG refCount = InterlockedIncrement(&This->refCount);
 285
 286     TRACE("(%p)->() AddRef from %d\n", This, refCount - 1);
 287
 288     return refCount;
 289 }
 290
 291 static ULONG WINAPI FilterMapper2_Release(IFilterMapper2 * iface)
 292 {
 293     FilterMapper2Impl *This = (FilterMapper2Impl *)iface;
 294     ULONG refCount = InterlockedDecrement(&This->refCount);
 295
 296     TRACE("(%p)->() Release from %d\n", This, refCount + 1);
 297
 298     if (refCount == 0)
 299     {
 300         CoTaskMemFree(This);
 301         return 0;
 302     }
 303     return refCount;
 304 }
 305
 306 /*** IFilterMapper2 methods ***/
 307
 308 static HRESULT WINAPI FilterMapper2_CreateCategory(
 309     IFilterMapper2 * iface,
 310     REFCLSID clsidCategory,
 311     DWORD dwCategoryMerit,
 312     LPCWSTR szDescription)
 313 {
 314     LPWSTR wClsidAMCat = NULL;
 315     LPWSTR wClsidCategory = NULL;
 316     WCHAR wszKeyName[ARRAYSIZE(wszClsidSlash)-1 + ARRAYSIZE(wszSlashInstance)-1 + (CHARS_IN_GUID-1) * 2 + 1];
 317     HKEY hKey = NULL;
 318     LONG lRet;
 319     HRESULT hr;
 320
 321     TRACE("(%s, %x, %s)\n", debugstr_guid(clsidCategory), dwCategoryMerit, debugstr_w(szDescription));
 322
 323     hr = StringFromCLSID(&CLSID_ActiveMovieCategories, &wClsidAMCat);
 324
 325     if (SUCCEEDED(hr))
 326     {
 327         hr = StringFromCLSID(clsidCategory, &wClsidCategory);
 328     }
 329
 330     if (SUCCEEDED(hr))
 331     {
 332         strcpyW(wszKeyName, wszClsidSlash);
 333         strcatW(wszKeyName, wClsidAMCat);
 334         strcatW(wszKeyName, wszSlashInstance);
 335         strcatW(wszKeyName, wClsidCategory);
 336
 337         lRet = RegCreateKeyExW(HKEY_CLASSES_ROOT, wszKeyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hKey, NULL);
 338         hr = HRESULT_FROM_WIN32(lRet);
 339     }
 340
 341     if (SUCCEEDED(hr))
 342     {
 343         lRet = RegSetValueExW(hKey, wszFriendlyName, 0, REG_SZ, (const BYTE*)szDescription, (strlenW(szDescription) + 1) * sizeof(WCHAR));
 344         hr = HRESULT_FROM_WIN32(lRet);
 345     }
 346
 347     if (SUCCEEDED(hr))
 348     {
 349         lRet = RegSetValueExW(hKey, wszClsidName, 0, REG_SZ, (LPBYTE)wClsidCategory, (strlenW(wClsidCategory) + 1) * sizeof(WCHAR));
 350         hr = HRESULT_FROM_WIN32(lRet);
 351     }
 352
 353     if (SUCCEEDED(hr))
 354     {
 355         lRet = RegSetValueExW(hKey, wszMeritName, 0, REG_DWORD, (LPBYTE)&dwCategoryMerit, sizeof(dwCategoryMerit));
 356         hr = HRESULT_FROM_WIN32(lRet);
 357     }
 358
 359     CloseHandle(hKey);
 360     CoTaskMemFree(wClsidCategory);
 361     CoTaskMemFree(wClsidAMCat);
 362
 363     return hr;
 364 }
 365
 366 static HRESULT WINAPI FilterMapper2_UnregisterFilter(
 367     IFilterMapper2 * iface,
 368     const CLSID *pclsidCategory,
 369     const OLECHAR *szInstance,
 370     REFCLSID Filter)
 371 {
 372     WCHAR wszKeyName[MAX_PATH];
 373     LPWSTR wClsidCategory = NULL;
 374     LPWSTR wFilter = NULL;
 375     HRESULT hr;
 376
 377     TRACE("(%p, %s, %s)\n", pclsidCategory, debugstr_w(szInstance), debugstr_guid(Filter));
 378
 379     if (!pclsidCategory)
 380         pclsidCategory = &CLSID_LegacyAmFilterCategory;
 381
 382     hr = StringFromCLSID(pclsidCategory, &wClsidCategory);
 383
 384     if (SUCCEEDED(hr))
 385     {
 386         strcpyW(wszKeyName, wszClsidSlash);
 387         strcatW(wszKeyName, wClsidCategory);
 388         strcatW(wszKeyName, wszSlashInstance);
 389         if (szInstance)
 390             strcatW(wszKeyName, szInstance);
 391         else
 392         {
 393             hr = StringFromCLSID(Filter, &wFilter);
 394             if (SUCCEEDED(hr))
 395                 strcatW(wszKeyName, wFilter);
 396         }
 397     }
 398
 399     if (SUCCEEDED(hr))
 400     {
 401         LONG lRet = RegDeleteKeyW(HKEY_CLASSES_ROOT, wszKeyName);
 402         hr = HRESULT_FROM_WIN32(lRet);
 403     }
 404
 405     CoTaskMemFree(wClsidCategory);
 406     CoTaskMemFree(wFilter);
 407
 408     return hr;
 409 }
 410
 411 static HRESULT FM2_WriteFriendlyName(IPropertyBag * pPropBag, LPCWSTR szName)
 412 {
 413     VARIANT var;
 414
 415     V_VT(&var) = VT_BSTR;
 416     V_UNION(&var, bstrVal) = (BSTR)szName;
 417
 418     return IPropertyBag_Write(pPropBag, wszFriendlyName, &var);
 419 }
 420
 421 static HRESULT FM2_WriteClsid(IPropertyBag * pPropBag, REFCLSID clsid)
 422 {
 423     LPWSTR wszClsid = NULL;
 424     VARIANT var;
 425     HRESULT hr;
 426
 427     hr = StringFromCLSID(clsid, &wszClsid);
 428
 429     if (SUCCEEDED(hr))
 430     {
 431         V_VT(&var) = VT_BSTR;
 432         V_UNION(&var, bstrVal) = wszClsid;
 433         hr = IPropertyBag_Write(pPropBag, wszClsidName, &var);
 434     }
 435     CoTaskMemFree(wszClsid);
 436     return hr;
 437 }
 438
 439 static HRESULT FM2_WriteFilterData(const REGFILTER2 * prf2, BYTE **ppData, ULONG *pcbData)
 440 {
 441     int size = sizeof(struct REG_RF);
 442     unsigned int i;
 443     struct Vector mainStore = {NULL, 0, 0};
 444     struct Vector clsidStore = {NULL, 0, 0};
 445     struct REG_RF rrf;
 446     HRESULT hr = S_OK;
 447
 448     rrf.dwVersion = prf2->dwVersion;
 449     rrf.dwMerit = prf2->dwMerit;
 450     rrf.dwPins = prf2->u.s1.cPins2;
 451     rrf.dwUnused = 0;
 452
 453     add_data(&mainStore, (LPBYTE)&rrf, sizeof(rrf));
 454
 455     for (i = 0; i < prf2->u.s1.cPins2; i++)
 456     {
 457         size += sizeof(struct REG_RFP);
 458         if (prf2->u.s1.rgPins2[i].clsPinCategory)
 459             size += sizeof(DWORD);
 460         size += prf2->u.s1.rgPins2[i].nMediaTypes * sizeof(struct REG_TYPE);
 461         size += prf2->u.s1.rgPins2[i].nMediums * sizeof(DWORD);
 462     }
 463
 464     for (i = 0; i < prf2->u.s1.cPins2; i++)
 465     {
 466         struct REG_RFP rrfp;
 467         REGFILTERPINS2 rgPin2 = prf2->u.s1.rgPins2[i];
 468         unsigned int j;
 469
 470         rrfp.signature[0] = '0';
 471         rrfp.signature[1] = 'p';
 472         rrfp.signature[2] = 'i';
 473         rrfp.signature[3] = '3';
 474         rrfp.signature[0] += i;
 475         rrfp.dwFlags = rgPin2.dwFlags;
 476         rrfp.dwInstances = rgPin2.cInstances;
 477         rrfp.dwMediaTypes = rgPin2.nMediaTypes;
 478         rrfp.dwMediums = rgPin2.nMediums;
 479         rrfp.bCategory = rgPin2.clsPinCategory ? 1 : 0;
 480
 481         add_data(&mainStore, (LPBYTE)&rrfp, sizeof(rrfp));
 482         if (rrfp.bCategory)
 483         {
 484             DWORD index = find_data(&clsidStore, (const BYTE*)rgPin2.clsPinCategory, sizeof(CLSID));
 485             if (index == -1)
 486                 index = add_data(&clsidStore, (const BYTE*)rgPin2.clsPinCategory, sizeof(CLSID));
 487             index += size;
 488
 489             add_data(&mainStore, (LPBYTE)&index, sizeof(index));
 490         }
 491
 492         for (j = 0; j < rgPin2.nMediaTypes; j++)
 493         {
 494             struct REG_TYPE rt;
 495             rt.signature[0] = '0';
 496             rt.signature[1] = 't';
 497             rt.signature[2] = 'y';
 498             rt.signature[3] = '3';
 499             rt.signature[0] += j;
 500
 501             rt.dwUnused = 0;
 502             rt.dwOffsetMajor = find_data(&clsidStore, (const BYTE*)rgPin2.lpMediaType[j].clsMajorType, sizeof(CLSID));
 503             if (rt.dwOffsetMajor == -1)
 504                 rt.dwOffsetMajor = add_data(&clsidStore, (const BYTE*)rgPin2.lpMediaType[j].clsMajorType, sizeof(CLSID));
 505             rt.dwOffsetMajor += size;
 506             rt.dwOffsetMinor = find_data(&clsidStore, (const BYTE*)rgPin2.lpMediaType[j].clsMinorType, sizeof(CLSID));
 507             if (rt.dwOffsetMinor == -1)
 508                 rt.dwOffsetMinor = add_data(&clsidStore, (const BYTE*)rgPin2.lpMediaType[j].clsMinorType, sizeof(CLSID));
 509             rt.dwOffsetMinor += size;
 510
 511             add_data(&mainStore, (LPBYTE)&rt, sizeof(rt));
 512         }
 513
 514         for (j = 0; j < rgPin2.nMediums; j++)
 515         {
 516             DWORD index = find_data(&clsidStore, (const BYTE*)(rgPin2.lpMedium + j), sizeof(REGPINMEDIUM));
 517             if (index == -1)
 518                 index = add_data(&clsidStore, (const BYTE*)(rgPin2.lpMedium + j), sizeof(REGPINMEDIUM));
 519             index += size;
 520
 521             add_data(&mainStore, (LPBYTE)&index, sizeof(index));
 522         }
 523     }
 524
 525     if (SUCCEEDED(hr))
 526     {
 527         *pcbData = mainStore.current + clsidStore.current;
 528         *ppData = CoTaskMemAlloc(*pcbData);
 529         if (!*ppData)
 530             hr = E_OUTOFMEMORY;
 531     }
 532
 533     if (SUCCEEDED(hr))
 534     {
 535         memcpy(*ppData, mainStore.pData, mainStore.current);
 536         memcpy((*ppData) + mainStore.current, clsidStore.pData, clsidStore.current);
 537     }
 538
 539     delete_vector(&mainStore);
 540     delete_vector(&clsidStore);
 541     return hr;
 542 }
 543
 544 static HRESULT FM2_ReadFilterData(BYTE *pData, REGFILTER2 * prf2)
 545 {
 546     HRESULT hr = S_OK;
 547     struct REG_RF * prrf;
 548     LPBYTE pCurrent;
 549     DWORD i;
 550     REGFILTERPINS2 * rgPins2;
 551
 552     prrf = (struct REG_RF *)pData;
 553     pCurrent = pData;
 554
 555     if (prrf->dwVersion != 2)
 556     {
 557         FIXME("Filter registry version %d not supported\n", prrf->dwVersion);
 558         ZeroMemory(prf2, sizeof(*prf2));
 559         hr = E_FAIL;
 560     }
 561
 562     if (SUCCEEDED(hr))
 563     {
 564         TRACE("version = %d, merit = %x, #pins = %d, unused = %x\n",
 565             prrf->dwVersion, prrf->dwMerit, prrf->dwPins, prrf->dwUnused);
 566
 567         prf2->dwVersion = prrf->dwVersion;
 568         prf2->dwMerit = prrf->dwMerit;
 569         prf2->u.s1.cPins2 = prrf->dwPins;
 570         rgPins2 = CoTaskMemAlloc(prrf->dwPins * sizeof(*rgPins2));
 571         prf2->u.s1.rgPins2 = rgPins2;
 572         pCurrent += sizeof(struct REG_RF);
 573
 574         for (i = 0; i < prrf->dwPins; i++)
 575         {
 576             struct REG_RFP * prrfp = (struct REG_RFP *)pCurrent;
 577             REGPINTYPES * lpMediaType;
 578             REGPINMEDIUM * lpMedium;
 579             UINT j;
 580
 581             /* FIXME: check signature */
 582
 583             TRACE("\tsignature = %s\n", debugstr_an((const char*)prrfp->signature, 4));
 584
 585             TRACE("\tpin[%d]: flags = %x, instances = %d, media types = %d, mediums = %d\n",
 586                 i, prrfp->dwFlags, prrfp->dwInstances, prrfp->dwMediaTypes, prrfp->dwMediums);
 587
 588             rgPins2[i].dwFlags = prrfp->dwFlags;
 589             rgPins2[i].cInstances = prrfp->dwInstances;
 590             rgPins2[i].nMediaTypes = prrfp->dwMediaTypes;
 591             rgPins2[i].nMediums = prrfp->dwMediums;
 592             pCurrent += sizeof(struct REG_RFP);
 593             if (prrfp->bCategory)
 594             {
 595                 CLSID * clsCat = CoTaskMemAlloc(sizeof(CLSID));
 596                 memcpy(clsCat, pData + *(DWORD*)(pCurrent), sizeof(CLSID));
 597                 pCurrent += sizeof(DWORD);
 598                 rgPins2[i].clsPinCategory = clsCat;
 599             }
 600             else
 601                 rgPins2[i].clsPinCategory = NULL;
 602
 603             if (rgPins2[i].nMediaTypes > 0)
 604                 lpMediaType = CoTaskMemAlloc(rgPins2[i].nMediaTypes * sizeof(*lpMediaType));
 605             else
 606                 lpMediaType = NULL;
 607
 608             rgPins2[i].lpMediaType = lpMediaType;
 609
 610             for (j = 0; j < rgPins2[i].nMediaTypes; j++)
 611             {
 612                 struct REG_TYPE * prt = (struct REG_TYPE *)pCurrent;
 613                 CLSID * clsMajor = CoTaskMemAlloc(sizeof(CLSID));
 614                 CLSID * clsMinor = CoTaskMemAlloc(sizeof(CLSID));
 615
 616                 /* FIXME: check signature */
 617                 TRACE("\t\tsignature = %s\n", debugstr_an((const char*)prt->signature, 4));
 618
 619                 memcpy(clsMajor, pData + prt->dwOffsetMajor, sizeof(CLSID));
 620                 memcpy(clsMinor, pData + prt->dwOffsetMinor, sizeof(CLSID));
 621
 622                 lpMediaType[j].clsMajorType = clsMajor;
 623                 lpMediaType[j].clsMinorType = clsMinor;
 624
 625                 pCurrent += sizeof(*prt);
 626             }
 627
 628             if (rgPins2[i].nMediums > 0)
 629                 lpMedium = CoTaskMemAlloc(rgPins2[i].nMediums * sizeof(*lpMedium));
 630             else
 631                 lpMedium = NULL;
 632
 633             rgPins2[i].lpMedium = lpMedium;
 634
 635             for (j = 0; j < rgPins2[i].nMediums; j++)
 636             {
 637                 DWORD dwOffset = *(DWORD *)pCurrent;
 638
 639                 memcpy(lpMedium + j, pData + dwOffset, sizeof(REGPINMEDIUM));
 640
 641                 pCurrent += sizeof(dwOffset);
 642             }
 643         }
 644
 645     }
 646
 647     return hr;
 648 }
 649
 650 static void FM2_DeleteRegFilter(REGFILTER2 * prf2)
 651 {
 652     UINT i;
 653     for (i = 0; i < prf2->u.s1.cPins2; i++)
 654     {
 655         UINT j;
 656         if (prf2->u.s1.rgPins2[i].clsPinCategory)
 657             CoTaskMemFree((LPVOID)prf2->u.s1.rgPins2[i].clsPinCategory);
 658
 659         for (j = 0; j < prf2->u.s1.rgPins2[i].nMediaTypes; j++)
 660         {
 661             CoTaskMemFree((LPVOID)prf2->u.s1.rgPins2[i].lpMediaType[j].clsMajorType);
 662             CoTaskMemFree((LPVOID)prf2->u.s1.rgPins2[i].lpMediaType[j].clsMinorType);
 663         }
 664         CoTaskMemFree((LPVOID)prf2->u.s1.rgPins2[i].lpMedium);
 665     }
 666 }
 667
 668 static HRESULT WINAPI FilterMapper2_RegisterFilter(
 669     IFilterMapper2 * iface,
 670     REFCLSID clsidFilter,
 671     LPCWSTR szName,
 672     IMoniker **ppMoniker,
 673     const CLSID *pclsidCategory,
 674     const OLECHAR *szInstance,
 675     const REGFILTER2 *prf2)
 676 {
 677     IParseDisplayName * pParser = NULL;
 678     IBindCtx * pBindCtx = NULL;
 679     IMoniker * pMoniker = NULL;
 680     IPropertyBag * pPropBag = NULL;
 681     HRESULT hr;
 682     LPWSTR pwszParseName = NULL;
 683     LPWSTR pCurrent;
 684     static const WCHAR wszDevice[] = {'@','d','e','v','i','c','e',':','s','w',':',0};
 685     int nameLen;
 686     ULONG ulEaten;
 687     LPWSTR szClsidTemp = NULL;
 688     REGFILTER2 regfilter2;
 689     REGFILTERPINS2* pregfp2 = NULL;
 690
 691     TRACE("(%s, %s, %p, %s, %s, %p)\n",
 692         debugstr_guid(clsidFilter),
 693         debugstr_w(szName),
 694         ppMoniker,
 695         debugstr_guid(pclsidCategory),
 696         debugstr_w(szInstance),
 697         prf2);
 698
 699     if (prf2->dwVersion == 2)
 700     {
 701         regfilter2 = *prf2;
 702     }
 703     else if (prf2->dwVersion == 1)
 704     {
 705         ULONG i;
 706         DWORD flags;
 707         /* REGFILTER2 structure is converted from version 1 to 2. Tested on Win2k. */
 708         regfilter2.dwVersion = 2;
 709         regfilter2.dwMerit = prf2->dwMerit;
 710         regfilter2.u.s1.cPins2 = prf2->u.s.cPins;
 711         pregfp2 = CoTaskMemAlloc(prf2->u.s.cPins * sizeof(REGFILTERPINS2));
 712         regfilter2.u.s1.rgPins2 = pregfp2;
 713         for (i = 0; i < prf2->u.s.cPins; i++)
 714         {
 715             flags = 0;
 716             if (prf2->u.s.rgPins[i].bRendered)
 717                 flags |= REG_PINFLAG_B_RENDERER;
 718             if (prf2->u.s.rgPins[i].bOutput)
 719                 flags |= REG_PINFLAG_B_OUTPUT;
 720             if (prf2->u.s.rgPins[i].bZero)
 721                 flags |= REG_PINFLAG_B_ZERO;
 722             if (prf2->u.s.rgPins[i].bMany)
 723                 flags |= REG_PINFLAG_B_MANY;
 724             pregfp2[i].dwFlags = flags;
 725             pregfp2[i].cInstances = 1;
 726             pregfp2[i].nMediaTypes = prf2->u.s.rgPins[i].nMediaTypes;
 727             pregfp2[i].lpMediaType = prf2->u.s.rgPins[i].lpMediaType;
 728             pregfp2[i].nMediums = 0;
 729             pregfp2[i].lpMedium = NULL;
 730             pregfp2[i].clsPinCategory = NULL;
 731         }
 732     }
 733     else
 734     {
 735         FIXME("dwVersion other that 1 or 2 not supported at the moment\n");
 736         return E_NOTIMPL;
 737     }
 738
 739     if (ppMoniker)
 740         *ppMoniker = NULL;
 741
 742     if (!pclsidCategory)
 743         /* MSDN mentions the inexistent CLSID_ActiveMovieFilters GUID.
 744          * In fact this is the CLSID_LegacyAmFilterCategory one */
 745         pclsidCategory = &CLSID_LegacyAmFilterCategory;
 746
 747     /* sizeof... will include the null terminator and
 748      * the + 1 is for the separator ('\\'). The -1 is
 749      * because CHARS_IN_GUID includes the null terminator
 750      */
 751     nameLen = sizeof(wszDevice)/sizeof(wszDevice[0]) + CHARS_IN_GUID - 1 + 1;
 752
 753     if (szInstance)
 754         nameLen += strlenW(szInstance);
 755     else
 756         nameLen += CHARS_IN_GUID - 1; /* CHARS_IN_GUID includes null terminator */
 757
 758     pCurrent = pwszParseName = CoTaskMemAlloc(nameLen*sizeof(WCHAR));
 759     if (!pwszParseName)
 760         return E_OUTOFMEMORY;
 761
 762     strcpyW(pwszParseName, wszDevice);
 763     pCurrent += strlenW(wszDevice);
 764
 765     hr = StringFromCLSID(pclsidCategory, &szClsidTemp);
 766
 767     if (SUCCEEDED(hr))
 768     {
 769         memcpy(pCurrent, szClsidTemp, CHARS_IN_GUID * sizeof(WCHAR));
 770         pCurrent += CHARS_IN_GUID - 1;
 771         pCurrent[0] = '\\';
 772
 773         if (szInstance)
 774             strcpyW(pCurrent+1, szInstance);
 775         else
 776         {
 777             CoTaskMemFree(szClsidTemp);
 778             szClsidTemp = NULL;
 779
 780             hr = StringFromCLSID(clsidFilter, &szClsidTemp);
 781             if (SUCCEEDED(hr))
 782                 strcpyW(pCurrent+1, szClsidTemp);
 783         }
 784     }
 785
 786     if (SUCCEEDED(hr))
 787         hr = CoCreateInstance(&CLSID_CDeviceMoniker, NULL, CLSCTX_INPROC, &IID_IParseDisplayName, (LPVOID *)&pParser);
 788
 789     if (SUCCEEDED(hr))
 790         hr = CreateBindCtx(0, &pBindCtx);
 791
 792     if (SUCCEEDED(hr))
 793         hr = IParseDisplayName_ParseDisplayName(pParser, pBindCtx, pwszParseName, &ulEaten, &pMoniker);
 794
 795     if (pBindCtx)
 796         IBindCtx_Release(pBindCtx);
 797     if (pParser)
 798         IParseDisplayName_Release(pParser);
 799
 800     if (SUCCEEDED(hr))
 801         hr = IMoniker_BindToStorage(pMoniker, NULL, NULL, &IID_IPropertyBag, (LPVOID)&pPropBag);
 802
 803     if (SUCCEEDED(hr))
 804         hr = FM2_WriteFriendlyName(pPropBag, szName);
 805
 806     if (SUCCEEDED(hr))
 807         hr = FM2_WriteClsid(pPropBag, clsidFilter);
 808
 809     if (SUCCEEDED(hr))
 810     {
 811         BYTE *pData;
 812         ULONG cbData;
 813
 814         hr = FM2_WriteFilterData(&regfilter2, &pData, &cbData);
 815         if (SUCCEEDED(hr))
 816         {
 817             VARIANT var;
 818             SAFEARRAY *psa;
 819             SAFEARRAYBOUND saBound;
 820
 821             saBound.lLbound = 0;
 822             saBound.cElements = cbData;
 823             psa = SafeArrayCreate(VT_UI1, 1, &saBound);
 824             if (!psa)
 825             {
 826                 ERR("Couldn't create SAFEARRAY\n");
 827                 hr = E_FAIL;
 828             }
 829
 830             if (SUCCEEDED(hr))
 831             {
 832                 LPBYTE pbSAData;
 833                 hr = SafeArrayAccessData(psa, (LPVOID *)&pbSAData);
 834                 if (SUCCEEDED(hr))
 835                 {
 836                     memcpy(pbSAData, pData, cbData);
 837                     hr = SafeArrayUnaccessData(psa);
 838                 }
 839             }
 840
 841             V_VT(&var) = VT_ARRAY | VT_UI1;
 842             V_UNION(&var, parray) = psa;
 843
 844             if (SUCCEEDED(hr))
 845                 hr = IPropertyBag_Write(pPropBag, wszFilterDataName, &var);
 846
 847             if (psa)
 848                 SafeArrayDestroy(psa);
 849             CoTaskMemFree(pData);
 850         }
 851     }
 852
 853     if (pPropBag)
 854         IPropertyBag_Release(pPropBag);
 855     CoTaskMemFree(szClsidTemp);
 856
 857     if (SUCCEEDED(hr) && ppMoniker)
 858         *ppMoniker = pMoniker;
 859     else if (pMoniker)
 860         IMoniker_Release(pMoniker);
 861
 862     CoTaskMemFree(pregfp2);
 863
 864     TRACE("-- returning %x\n", hr);
 865
 866     return hr;
 867 }
 868
 869 /* internal helper function */
 870 static BOOL MatchTypes(
 871     BOOL bExactMatch,
 872     DWORD nPinTypes,
 873     const REGPINTYPES * pPinTypes,
 874     DWORD nMatchTypes,
 875     const GUID * pMatchTypes)
 876 {
 877     BOOL bMatch = FALSE;
 878     DWORD j;
 879
 880     if ((nMatchTypes == 0) && (nPinTypes > 0))
 881         bMatch = TRUE;
 882
 883     for (j = 0; j < nPinTypes; j++)
 884     {
 885         DWORD i;
 886         for (i = 0; i < nMatchTypes*2; i+=2)
 887         {
 888             if (((!bExactMatch && IsEqualGUID(pPinTypes[j].clsMajorType, &GUID_NULL)) || IsEqualGUID(&pMatchTypes[i], &GUID_NULL) || IsEqualGUID(pPinTypes[j].clsMajorType, &pMatchTypes[i])) &&
 889                 ((!bExactMatch && IsEqualGUID(pPinTypes[j].clsMinorType, &GUID_NULL)) || IsEqualGUID(&pMatchTypes[i+1], &GUID_NULL) || IsEqualGUID(pPinTypes[j].clsMinorType, &pMatchTypes[i+1])))
 890             {
 891                 bMatch = TRUE;
 892                 break;
 893             }
 894         }
 895     }
 896     return bMatch;
 897 }
 898
 899 /* internal helper function for qsort of MONIKER_MERIT array */
 900 static int mm_compare(const void * left, const void * right)
 901 {
 902     const struct MONIKER_MERIT * mmLeft = (const struct MONIKER_MERIT *)left;
 903     const struct MONIKER_MERIT * mmRight = (const struct MONIKER_MERIT *)right;
 904
 905     if (mmLeft->dwMerit == mmRight->dwMerit)
 906         return 0;
 907     if (mmLeft->dwMerit > mmRight->dwMerit)
 908         return -1;
 909     return 1;
 910 }
 911
 912 /* NOTES:
 913  *   Exact match means whether or not to treat GUID_NULL's in filter data as wild cards
 914  *    (GUID_NULL's in input to function automatically treated as wild cards)
 915  *   Input/Output needed means match only on criteria if TRUE (with zero input types
 916  *    meaning match any input/output pin as long as one exists), otherwise match any
 917  *    filter that meets the rest of the requirements.
 918  */
 919 static HRESULT WINAPI FilterMapper2_EnumMatchingFilters(
 920     IFilterMapper2 * iface,
 921     IEnumMoniker **ppEnum,
 922     DWORD dwFlags,
 923     BOOL bExactMatch,
 924     DWORD dwMerit,
 925     BOOL bInputNeeded,
 926     DWORD cInputTypes,
 927     const GUID *pInputTypes,
 928     const REGPINMEDIUM *pMedIn,
 929     const CLSID *pPinCategoryIn,
 930     BOOL bRender,
 931     BOOL bOutputNeeded,
 932     DWORD cOutputTypes,
 933     const GUID *pOutputTypes,
 934     const REGPINMEDIUM *pMedOut,
 935     const CLSID *pPinCategoryOut)
 936 {
 937     ICreateDevEnum * pCreateDevEnum;
 938     IMoniker * pMonikerCat;
 939     IEnumMoniker * pEnumCat;
 940     HRESULT hr;
 941     struct Vector monikers = {NULL, 0, 0};
 942
 943     TRACE("(%p, %x, %s, %x, %s, %d, %p, %p, %p, %s, %s, %p, %p, %p)\n",
 944         ppEnum,
 945         dwFlags,
 946         bExactMatch ? "true" : "false",
 947         dwMerit,
 948         bInputNeeded ? "true" : "false",
 949         cInputTypes,
 950         pInputTypes,
 951         pMedIn,
 952         pPinCategoryIn,
 953         bRender ? "true" : "false",
 954         bOutputNeeded ? "true" : "false",
 955         pOutputTypes,
 956         pMedOut,
 957         pPinCategoryOut);
 958
 959     if (dwFlags != 0)
 960     {
 961         FIXME("dwFlags = %x not implemented\n", dwFlags);
 962     }
 963
 964     *ppEnum = NULL;
 965
 966     hr = CoCreateInstance(&CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC, &IID_ICreateDevEnum, (LPVOID*)&pCreateDevEnum);
 967     if (FAILED(hr))
 968         return hr;
 969
 970     hr = ICreateDevEnum_CreateClassEnumerator(pCreateDevEnum, &CLSID_ActiveMovieCategories, &pEnumCat, 0);
 971     if (FAILED(hr)) {
 972         ICreateDevEnum_Release(pCreateDevEnum);
 973         return hr;
 974     }
 975
 976     while (IEnumMoniker_Next(pEnumCat, 1, &pMonikerCat, NULL) == S_OK)
 977     {
 978         IPropertyBag * pPropBagCat = NULL;
 979         VARIANT var;
 980         HRESULT hrSub; /* this is so that one buggy filter
 981                           doesn't make the whole lot fail */
 982
 983         VariantInit(&var);
 984
 985         hrSub = IMoniker_BindToStorage(pMonikerCat, NULL, NULL, &IID_IPropertyBag, (LPVOID*)&pPropBagCat);
 986
 987         if (SUCCEEDED(hrSub))
 988             hrSub = IPropertyBag_Read(pPropBagCat, wszMeritName, &var, NULL);
 989
 990         if (SUCCEEDED(hrSub) && (V_UNION(&var, ulVal) >= dwMerit))
 991         {
 992             CLSID clsidCat;
 993             IEnumMoniker * pEnum;
 994             IMoniker * pMoniker;
 995
 996             VariantClear(&var);
 997
 998             if (TRACE_ON(quartz))
 999             {
1000                 VARIANT temp;
1001                 V_VT(&temp) = VT_EMPTY;
1002                 IPropertyBag_Read(pPropBagCat, wszFriendlyName, &temp, NULL);
1003                 TRACE("Considering category %s\n", debugstr_w(V_UNION(&temp, bstrVal)));
1004                 VariantClear(&temp);
1005             }
1006
1007             hrSub = IPropertyBag_Read(pPropBagCat, wszClsidName, &var, NULL);
1008
1009             if (SUCCEEDED(hrSub))
1010                 hrSub = CLSIDFromString(V_UNION(&var, bstrVal), &clsidCat);
1011
1012             if (SUCCEEDED(hrSub))
1013                 hrSub = ICreateDevEnum_CreateClassEnumerator(pCreateDevEnum, &clsidCat, &pEnum, 0);
1014
1015             if (hrSub == S_OK)
1016             {
1017                 while (IEnumMoniker_Next(pEnum, 1, &pMoniker, NULL) == S_OK)
1018                 {
1019                     IPropertyBag * pPropBag = NULL;
1020                     VARIANT var;
1021                     BYTE *pData = NULL;
1022                     REGFILTER2 rf2;
1023                     DWORD i;
1024                     BOOL bInputMatch = !bInputNeeded;
1025                     BOOL bOutputMatch = !bOutputNeeded;
1026
1027                     ZeroMemory(&rf2, sizeof(rf2));
1028                     VariantInit(&var);
1029
1030                     hrSub = IMoniker_BindToStorage(pMoniker, NULL, NULL, &IID_IPropertyBag, (LPVOID*)&pPropBag);
1031
1032                     if (TRACE_ON(quartz))
1033                     {
1034                         VARIANT temp;
1035                         V_VT(&temp) = VT_EMPTY;
1036                         IPropertyBag_Read(pPropBag, wszFriendlyName, &temp, NULL);
1037                         TRACE("Considering filter %s\n", debugstr_w(V_UNION(&temp, bstrVal)));
1038                         VariantClear(&temp);
1039                     }
1040
1041                     if (SUCCEEDED(hrSub))
1042                     {
1043                         hrSub = IPropertyBag_Read(pPropBag, wszFilterDataName, &var, NULL);
1044                     }
1045
1046                     if (SUCCEEDED(hrSub))
1047                         hrSub = SafeArrayAccessData(V_UNION(&var, parray), (LPVOID*)&pData);
1048
1049                     if (SUCCEEDED(hrSub))
1050                         hrSub = FM2_ReadFilterData(pData, &rf2);
1051
1052                     if (pData)
1053                         SafeArrayUnaccessData(V_UNION(&var, parray));
1054
1055                     VariantClear(&var);
1056
1057                     /* Logic used for bInputMatch expression:
1058                      * There exists some pin such that bInputNeeded implies (pin is an input and
1059                      * (bRender implies pin has render flag) and major/minor types members of
1060                      * pInputTypes )
1061                      * bOutputMatch is similar, but without the "bRender implies ..." part
1062                      * and substituting variables names containing input for output
1063                      */
1064
1065                     /* determine whether filter meets requirements */
1066                     if (SUCCEEDED(hrSub) && (rf2.dwMerit >= dwMerit))
1067                     {
1068                         for (i = 0; (i < rf2.u.s1.cPins2) && (!bInputMatch || !bOutputMatch); i++)
1069                         {
1070                             const REGFILTERPINS2 * rfp2 = rf2.u.s1.rgPins2 + i;
1071
1072                             bInputMatch = bInputMatch || (!(rfp2->dwFlags & REG_PINFLAG_B_OUTPUT) &&
1073                                 (!bRender || (rfp2->dwFlags & REG_PINFLAG_B_RENDERER)) &&
1074                                 MatchTypes(bExactMatch, rfp2->nMediaTypes, rfp2->lpMediaType, cInputTypes, pInputTypes));
1075                             bOutputMatch = bOutputMatch || ((rfp2->dwFlags & REG_PINFLAG_B_OUTPUT) &&
1076                                 MatchTypes(bExactMatch, rfp2->nMediaTypes, rfp2->lpMediaType, cOutputTypes, pOutputTypes));
1077                         }
1078
1079                         if (bInputMatch && bOutputMatch)
1080                         {
1081                             struct MONIKER_MERIT mm = {pMoniker, rf2.dwMerit};
1082                             IMoniker_AddRef(pMoniker);
1083                             add_data(&monikers, (LPBYTE)&mm, sizeof(mm));
1084                         }
1085                     }
1086
1087                     FM2_DeleteRegFilter(&rf2);
1088                     if (pPropBag)
1089                         IPropertyBag_Release(pPropBag);
1090                     IMoniker_Release(pMoniker);
1091                 }
1092                 IEnumMoniker_Release(pEnum);
1093             }
1094         }
1095
1096         VariantClear(&var);
1097         if (pPropBagCat)
1098             IPropertyBag_Release(pPropBagCat);
1099         IMoniker_Release(pMonikerCat);
1100     }
1101
1102     if (SUCCEEDED(hr))
1103     {
1104         IMoniker ** ppMoniker;
1105         unsigned int i;
1106         ULONG nMonikerCount = monikers.current / sizeof(struct MONIKER_MERIT);
1107
1108         /* sort the monikers in descending merit order */
1109         qsort(monikers.pData, nMonikerCount,
1110               sizeof(struct MONIKER_MERIT),
1111               mm_compare);
1112
1113         /* construct an IEnumMoniker interface */
1114         ppMoniker = CoTaskMemAlloc(nMonikerCount * sizeof(IMoniker *));
1115         for (i = 0; i < nMonikerCount; i++)
1116         {
1117             /* no need to AddRef here as already AddRef'd above */
1118             ppMoniker[i] = ((struct MONIKER_MERIT *)monikers.pData)[i].pMoniker;
1119         }
1120         hr = EnumMonikerImpl_Create(ppMoniker, nMonikerCount, ppEnum);
1121         CoTaskMemFree(ppMoniker);
1122     }
1123
1124     delete_vector(&monikers);
1125     IEnumMoniker_Release(pEnumCat);
1126     ICreateDevEnum_Release(pCreateDevEnum);
1127
1128     return hr;
1129 }
1130
1131 static const IFilterMapper2Vtbl fm2vtbl =
1132 {
1133
1134     FilterMapper2_QueryInterface,
1135     FilterMapper2_AddRef,
1136     FilterMapper2_Release,
1137
1138     FilterMapper2_CreateCategory,
1139     FilterMapper2_UnregisterFilter,
1140     FilterMapper2_RegisterFilter,
1141     FilterMapper2_EnumMatchingFilters
1142 };
1143
1144 /*** IUnknown methods ***/
1145
1146 static HRESULT WINAPI FilterMapper_QueryInterface(IFilterMapper * iface, REFIID riid, LPVOID *ppv)
1147 {
1148     FilterMapper2Impl *This = impl_from_IFilterMapper(iface);
1149
1150     TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), ppv);
1151
1152     return FilterMapper2_QueryInterface((IFilterMapper2*)&This->lpVtbl, riid, ppv);
1153 }
1154
1155 static ULONG WINAPI FilterMapper_AddRef(IFilterMapper * iface)
1156 {
1157     FilterMapper2Impl *This = impl_from_IFilterMapper(iface);
1158
1159     return FilterMapper2_AddRef((IFilterMapper2*)This);
1160 }
1161
1162 static ULONG WINAPI FilterMapper_Release(IFilterMapper * iface)
1163 {
1164     FilterMapper2Impl *This = impl_from_IFilterMapper(iface);
1165
1166     return FilterMapper2_Release((IFilterMapper2*)This);
1167 }
1168
1169 /*** IFilterMapper methods ***/
1170
1171 static HRESULT WINAPI FilterMapper_EnumMatchingFilters(
1172     IFilterMapper * iface,
1173     IEnumRegFilters **ppEnum,
1174     DWORD dwMerit,
1175     BOOL bInputNeeded,
1176     CLSID clsInMaj,
1177     CLSID clsInSub,
1178     BOOL bRender,
1179     BOOL bOutputNeeded,
1180     CLSID clsOutMaj,
1181     CLSID clsOutSub)
1182 {
1183     FilterMapper2Impl *This = impl_from_IFilterMapper(iface);
1184     GUID InputType[2];
1185     GUID OutputType[2];
1186     IEnumMoniker* ppEnumMoniker;
1187     IMoniker* IMon;
1188     ULONG nb;
1189     ULONG idx = 0, nb_mon = 0;
1190     REGFILTER* regfilters;
1191     HRESULT hr;
1192
1193     TRACE("(%p/%p)->(%p, %x, %s, %s, %s, %s, %s, %s, %s) stub!\n",
1194         iface,This,
1195         ppEnum,
1196         dwMerit,
1197         bInputNeeded ? "true" : "false",
1198         debugstr_guid(&clsInMaj),
1199         debugstr_guid(&clsInSub),
1200         bRender ? "true" : "false",
1201         bOutputNeeded ? "true" : "false",
1202         debugstr_guid(&clsOutMaj),
1203         debugstr_guid(&clsOutSub));
1204
1205     InputType[0] = clsInMaj;
1206     InputType[1] = clsInSub;
1207     OutputType[0] = clsOutMaj;
1208     OutputType[1] = clsOutSub;
1209
1210     hr = IFilterMapper2_EnumMatchingFilters((IFilterMapper2*)This,
1211                                        &ppEnumMoniker,
1212                                        0,
1213                                        TRUE,
1214                                        dwMerit,
1215                                        bInputNeeded,
1216                                        1,
1217                                        InputType,
1218                                        NULL,
1219                                        &GUID_NULL,
1220                                        bRender,
1221                                        bOutputNeeded,
1222                                        1,
1223                                        OutputType,
1224                                        NULL,
1225                                        &GUID_NULL);
1226
1227     if (!SUCCEEDED(hr))
1228         return hr;
1229
1230     while(IEnumMoniker_Next(ppEnumMoniker, 1, &IMon, &nb) == S_OK)
1231     {
1232         IMoniker_Release(IMon);
1233         nb_mon++;
1234     }
1235
1236     *ppEnum = NULL;
1237     if (!nb_mon)
1238     {
1239         IEnumMoniker_Release(ppEnumMoniker);
1240         return IEnumRegFiltersImpl_Construct(NULL, 0, ppEnum);
1241     }
1242
1243     regfilters = CoTaskMemAlloc(nb_mon * sizeof(REGFILTER));
1244     if (!regfilters)
1245     {
1246         IEnumMoniker_Release(ppEnumMoniker);
1247         return E_OUTOFMEMORY;
1248     }
1249
1250     IEnumMoniker_Reset(ppEnumMoniker);
1251     while(IEnumMoniker_Next(ppEnumMoniker, 1, &IMon, &nb) == S_OK)
1252     {
1253         IPropertyBag * pPropBagCat = NULL;
1254         VARIANT var;
1255         HRESULT hrSub;
1256         GUID clsid;
1257         int len;
1258
1259         VariantInit(&var);
1260         V_VT(&var) = VT_BSTR;
1261
1262         hrSub = IMoniker_BindToStorage(IMon, NULL, NULL, &IID_IPropertyBag, (LPVOID*)&pPropBagCat);
1263
1264         if (SUCCEEDED(hrSub))
1265             hrSub = IPropertyBag_Read(pPropBagCat, wszClsidName, &var, NULL);
1266
1267         if (SUCCEEDED(hrSub))
1268             hrSub = CLSIDFromString(V_UNION(&var, bstrVal), &clsid);
1269
1270         if (SUCCEEDED(hrSub))
1271             hrSub = IPropertyBag_Read(pPropBagCat, wszFriendlyName, &var, NULL);
1272
1273         if (SUCCEEDED(hrSub))
1274         {
1275             len = (strlenW((WCHAR*)V_UNION(&var, bstrVal))+1) * sizeof(WCHAR);
1276             if (!(regfilters[idx].Name = CoTaskMemAlloc(len*2)))
1277                 hr = E_OUTOFMEMORY;
1278         }
1279
1280         if (SUCCEEDED(hrSub))
1281         {
1282             memcpy(regfilters[idx].Name, V_UNION(&var, bstrVal), len);
1283             regfilters[idx].Clsid = clsid;
1284             idx++;
1285         }
1286
1287         if (pPropBagCat)
1288             IPropertyBag_Release(pPropBagCat);
1289         IMoniker_Release(IMon);
1290     }
1291
1292     /* In case of release all resources */
1293     if (!SUCCEEDED(hr))
1294     {
1295         for (idx = 0; idx < nb_mon; idx++)
1296             CoTaskMemFree(regfilters[idx].Name);
1297         CoTaskMemFree(regfilters);
1298         IEnumMoniker_Release(ppEnumMoniker);
1299         return hr;
1300     }
1301
1302     hr = IEnumRegFiltersImpl_Construct(regfilters, nb_mon, ppEnum);
1303     CoTaskMemFree(regfilters);
1304     IEnumMoniker_Release(ppEnumMoniker);
1305
1306     return hr;
1307 }
1308
1309
1310 static HRESULT WINAPI FilterMapper_RegisterFilter(IFilterMapper * iface, CLSID clsid, LPCWSTR szName, DWORD dwMerit)
1311 {
1312     HRESULT hr;
1313     LPWSTR wszClsid = NULL;
1314     HKEY hKey;
1315     LONG lRet;
1316     WCHAR wszKeyName[strlenW(wszFilterSlash) + (CHARS_IN_GUID-1) + 1];
1317
1318     TRACE("(%p)->(%s, %s, %x)\n", iface, debugstr_guid(&clsid), debugstr_w(szName), dwMerit);
1319
1320     hr = StringFromCLSID(&clsid, &wszClsid);
1321
1322     if (SUCCEEDED(hr))
1323     {
1324         strcpyW(wszKeyName, wszFilterSlash);
1325         strcatW(wszKeyName, wszClsid);
1326
1327         lRet = RegCreateKeyExW(HKEY_CLASSES_ROOT, wszKeyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hKey, NULL);
1328         hr = HRESULT_FROM_WIN32(lRet);
1329     }
1330
1331     if (SUCCEEDED(hr))
1332     {
1333         lRet = RegSetValueExW(hKey, NULL, 0, REG_SZ, (const BYTE*)szName, (strlenW(szName) + 1) * sizeof(WCHAR));
1334         hr = HRESULT_FROM_WIN32(lRet);
1335         CloseHandle(hKey);
1336     }
1337
1338     if (SUCCEEDED(hr))
1339     {
1340         strcpyW(wszKeyName, wszClsidSlash);
1341         strcatW(wszKeyName, wszClsid);
1342
1343         lRet = RegCreateKeyExW(HKEY_CLASSES_ROOT, wszKeyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hKey, NULL);
1344         hr = HRESULT_FROM_WIN32(lRet);
1345     }
1346
1347     if (SUCCEEDED(hr))
1348     {
1349         lRet = RegSetValueExW(hKey, wszMeritName, 0, REG_DWORD, (LPBYTE)&dwMerit, sizeof(dwMerit));
1350         hr = HRESULT_FROM_WIN32(lRet);
1351         CloseHandle(hKey);
1352     }
1353
1354     return hr;
1355 }
1356
1357 static HRESULT WINAPI FilterMapper_RegisterFilterInstance(IFilterMapper * iface, CLSID clsid, LPCWSTR szName, CLSID *MRId)
1358 {
1359     TRACE("(%p)->(%s, %s, %p)\n", iface, debugstr_guid(&clsid), debugstr_w(szName), MRId);
1360
1361     /* Not implemented in Windows (tested on Win2k) */
1362
1363     return E_NOTIMPL;
1364 }
1365
1366 static HRESULT WINAPI FilterMapper_RegisterPin(
1367     IFilterMapper * iface,
1368     CLSID Filter,
1369     LPCWSTR szName,
1370     BOOL bRendered,
1371     BOOL bOutput,
1372     BOOL bZero,
1373     BOOL bMany,
1374     CLSID ConnectsToFilter,
1375     LPCWSTR ConnectsToPin)
1376 {
1377     HRESULT hr;
1378     LONG lRet;
1379     LPWSTR wszClsid = NULL;
1380     HKEY hKey = NULL;
1381     HKEY hPinsKey = NULL;
1382     WCHAR * wszPinsKeyName;
1383     WCHAR wszKeyName[strlenW(wszClsidSlash) + (CHARS_IN_GUID-1) + 1];
1384
1385     TRACE("(%p)->(%s, %s, %d, %d, %d, %d, %s, %s)\n", iface, debugstr_guid(&Filter), debugstr_w(szName), bRendered,
1386                 bOutput, bZero, bMany, debugstr_guid(&ConnectsToFilter), debugstr_w(ConnectsToPin));
1387
1388     hr = StringFromCLSID(&Filter, &wszClsid);
1389
1390     if (SUCCEEDED(hr))
1391     {
1392         strcpyW(wszKeyName, wszClsidSlash);
1393         strcatW(wszKeyName, wszClsid);
1394
1395         lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszKeyName, 0, KEY_WRITE, &hKey);
1396         hr = HRESULT_FROM_WIN32(lRet);
1397     }
1398
1399     if (SUCCEEDED(hr))
1400     {
1401         wszPinsKeyName = CoTaskMemAlloc((strlenW(wszPins) + 1 + strlenW(szName) + 1) * 2);
1402         if (!wszPinsKeyName)
1403              hr = E_OUTOFMEMORY;
1404     }
1405
1406     if (SUCCEEDED(hr))
1407     {
1408         strcpyW(wszPinsKeyName, wszPins);
1409         strcatW(wszPinsKeyName, wszSlash);
1410         strcatW(wszPinsKeyName, szName);
1411
1412         lRet = RegCreateKeyExW(hKey, wszPinsKeyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hPinsKey, NULL);
1413         hr = HRESULT_FROM_WIN32(lRet);
1414         CoTaskMemFree(wszPinsKeyName);
1415     }
1416
1417     if (SUCCEEDED(hr))
1418     {
1419         lRet = RegSetValueExW(hPinsKey, wszAllowedMany, 0, REG_DWORD, (LPBYTE)&bMany, sizeof(bMany));
1420         hr = HRESULT_FROM_WIN32(lRet);
1421     }
1422
1423     if (SUCCEEDED(hr))
1424     {
1425         lRet = RegSetValueExW(hPinsKey, wszAllowedZero, 0, REG_DWORD, (LPBYTE)&bZero, sizeof(bZero));
1426         hr = HRESULT_FROM_WIN32(lRet);
1427     }
1428
1429     if (SUCCEEDED(hr))
1430     {
1431         lRet = RegSetValueExW(hPinsKey, wszDirection, 0, REG_DWORD, (LPBYTE)&bOutput, sizeof(bOutput));
1432         hr = HRESULT_FROM_WIN32(lRet);
1433     }
1434
1435     if (SUCCEEDED(hr))
1436     {
1437         lRet = RegSetValueExW(hPinsKey, wszIsRendered, 0, REG_DWORD, (LPBYTE)&bRendered, sizeof(bRendered));
1438         hr = HRESULT_FROM_WIN32(lRet);
1439     }
1440
1441     if (SUCCEEDED(hr))
1442     {
1443         HKEY hkeyDummy = NULL;
1444
1445         lRet = RegCreateKeyExW(hPinsKey, wszTypes, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkeyDummy, NULL);
1446         hr = HRESULT_FROM_WIN32(lRet);
1447
1448         if (hkeyDummy) RegCloseKey(hkeyDummy);
1449     }
1450
1451     CoTaskMemFree(wszClsid);
1452     if (hKey)
1453         CloseHandle(hKey);
1454     if (hPinsKey)
1455         CloseHandle(hPinsKey);
1456
1457     return hr;
1458 }
1459
1460
1461 static HRESULT WINAPI FilterMapper_RegisterPinType(
1462     IFilterMapper * iface,
1463     CLSID clsFilter,
1464     LPCWSTR szName,
1465     CLSID clsMajorType,
1466     CLSID clsSubType)
1467 {
1468     HRESULT hr;
1469     LONG lRet;
1470     LPWSTR wszClsid = NULL;
1471     LPWSTR wszClsidMajorType = NULL;
1472     LPWSTR wszClsidSubType = NULL;
1473     HKEY hKey = NULL;
1474     WCHAR * wszTypesKey;
1475     WCHAR wszKeyName[strlenW(wszClsidSlash) + (CHARS_IN_GUID-1) + 1];
1476
1477     TRACE("(%p)->(%s, %s, %s, %s)\n", iface, debugstr_guid(&clsFilter), debugstr_w(szName),
1478                     debugstr_guid(&clsMajorType), debugstr_guid(&clsSubType));
1479
1480     hr = StringFromCLSID(&clsFilter, &wszClsid);
1481
1482     if (SUCCEEDED(hr))
1483     {
1484         hr = StringFromCLSID(&clsMajorType, &wszClsidMajorType);
1485     }
1486
1487     if (SUCCEEDED(hr))
1488     {
1489         hr = StringFromCLSID(&clsSubType, &wszClsidSubType);
1490     }
1491
1492     if (SUCCEEDED(hr))
1493     {
1494         wszTypesKey = CoTaskMemAlloc((strlenW(wszClsidSlash) + strlenW(wszClsid) + strlenW(wszPins) +
1495                         strlenW(szName) + strlenW(wszTypes) + 3 + 1) * 2);
1496         if (!wszTypesKey)
1497             hr = E_OUTOFMEMORY;
1498     }
1499
1500     if (SUCCEEDED(hr))
1501     {
1502         strcpyW(wszTypesKey, wszClsidSlash);
1503         strcatW(wszTypesKey, wszClsid);
1504         strcatW(wszTypesKey, wszSlash);
1505         strcatW(wszTypesKey, wszPins);
1506         strcatW(wszTypesKey, wszSlash);
1507         strcatW(wszTypesKey, szName);
1508         strcatW(wszTypesKey, wszSlash);
1509         strcatW(wszTypesKey, wszTypes);
1510
1511         lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszTypesKey, 0, KEY_WRITE, &hKey);
1512         hr = HRESULT_FROM_WIN32(lRet);
1513         CoTaskMemFree(wszTypesKey);
1514     }
1515
1516     if (SUCCEEDED(hr))
1517     {
1518         HKEY hkeyDummy = NULL;
1519
1520         strcpyW(wszKeyName, wszClsidMajorType);
1521         strcatW(wszKeyName, wszSlash);
1522         strcatW(wszKeyName, wszClsidSubType);
1523
1524         lRet = RegCreateKeyExW(hKey, wszKeyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkeyDummy, NULL);
1525         hr = HRESULT_FROM_WIN32(lRet);
1526         CloseHandle(hKey);
1527
1528         if (hkeyDummy) RegCloseKey(hkeyDummy);
1529     }
1530
1531     CoTaskMemFree(wszClsid);
1532     CoTaskMemFree(wszClsidMajorType);
1533     CoTaskMemFree(wszClsidSubType);
1534
1535     return hr;
1536 }
1537
1538 static HRESULT WINAPI FilterMapper_UnregisterFilter(IFilterMapper * iface, CLSID Filter)
1539 {
1540     HRESULT hr;
1541     LONG lRet;
1542     LPWSTR wszClsid = NULL;
1543     HKEY hKey;
1544     WCHAR wszKeyName[strlenW(wszClsidSlash) + (CHARS_IN_GUID-1) + 1];
1545
1546     TRACE("(%p)->(%s)\n", iface, debugstr_guid(&Filter));
1547
1548     hr = StringFromCLSID(&Filter, &wszClsid);
1549
1550     if (SUCCEEDED(hr))
1551     {
1552         lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszFilter, 0, KEY_WRITE, &hKey);
1553         hr = HRESULT_FROM_WIN32(lRet);
1554     }
1555
1556     if (SUCCEEDED(hr))
1557     {
1558         lRet = RegDeleteKeyW(hKey, wszClsid);
1559         hr = HRESULT_FROM_WIN32(lRet);
1560         CloseHandle(hKey);
1561     }
1562
1563     if (SUCCEEDED(hr))
1564     {
1565         strcpyW(wszKeyName, wszClsidSlash);
1566         strcatW(wszKeyName, wszClsid);
1567
1568         lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszKeyName, 0, KEY_WRITE, &hKey);
1569         hr = HRESULT_FROM_WIN32(lRet);
1570     }
1571
1572     if (SUCCEEDED(hr))
1573     {
1574         lRet = RegDeleteKeyW(hKey, wszMeritName);
1575         hr = HRESULT_FROM_WIN32(lRet);
1576         CloseHandle(hKey);
1577     }
1578
1579     CoTaskMemFree(wszClsid);
1580
1581     return hr;
1582 }
1583
1584 static HRESULT WINAPI FilterMapper_UnregisterFilterInstance(IFilterMapper * iface, CLSID MRId)
1585 {
1586     TRACE("(%p)->(%s)\n", iface, debugstr_guid(&MRId));
1587
1588     /* Not implemented in Windows (tested on Win2k) */
1589
1590     return E_NOTIMPL;
1591 }
1592
1593 static HRESULT WINAPI FilterMapper_UnregisterPin(IFilterMapper * iface, CLSID Filter, LPCWSTR Name)
1594 {
1595     HRESULT hr;
1596     LONG lRet;
1597     LPWSTR wszClsid = NULL;
1598     HKEY hKey = NULL;
1599     WCHAR * wszPinNameKey;
1600     WCHAR wszKeyName[strlenW(wszClsidSlash) + (CHARS_IN_GUID-1) + 1];
1601
1602     TRACE("(%p)->(%s, %s)\n", iface, debugstr_guid(&Filter), debugstr_w(Name));
1603
1604     if (!Name)
1605         return E_INVALIDARG;
1606
1607     hr = StringFromCLSID(&Filter, &wszClsid);
1608
1609     if (SUCCEEDED(hr))
1610     {
1611         strcpyW(wszKeyName, wszClsidSlash);
1612         strcatW(wszKeyName, wszClsid);
1613
1614         lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszKeyName, 0, KEY_WRITE, &hKey);
1615         hr = HRESULT_FROM_WIN32(lRet);
1616     }
1617
1618     if (SUCCEEDED(hr))
1619     {
1620         wszPinNameKey = CoTaskMemAlloc((strlenW(wszPins) + 1 + strlenW(Name) + 1) * 2);
1621         if (!wszPinNameKey)
1622             hr = E_OUTOFMEMORY;
1623     }
1624
1625     if (SUCCEEDED(hr))
1626     {
1627         strcpyW(wszPinNameKey, wszPins);
1628         strcatW(wszPinNameKey, wszSlash);
1629         strcatW(wszPinNameKey, Name);
1630
1631         lRet = RegDeleteKeyW(hKey, wszPinNameKey);
1632         hr = HRESULT_FROM_WIN32(lRet);
1633         CoTaskMemFree(wszPinNameKey);
1634     }
1635
1636     CoTaskMemFree(wszClsid);
1637     if (hKey)
1638         CloseHandle(hKey);
1639
1640     return hr;
1641 }
1642
1643 static const IFilterMapperVtbl fmvtbl =
1644 {
1645
1646     FilterMapper_QueryInterface,
1647     FilterMapper_AddRef,
1648     FilterMapper_Release,
1649
1650     FilterMapper_RegisterFilter,
1651     FilterMapper_RegisterFilterInstance,
1652     FilterMapper_RegisterPin,
1653     FilterMapper_RegisterPinType,
1654     FilterMapper_UnregisterFilter,
1655     FilterMapper_UnregisterFilterInstance,
1656     FilterMapper_UnregisterPin,
1657     FilterMapper_EnumMatchingFilters
1658 };
1659
1660
1661 /*** IUnknown methods ***/
1662 static HRESULT WINAPI AMFilterData_QueryInterface(IAMFilterData * iface, REFIID riid, LPVOID *ppv)
1663 {
1664     FilterMapper2Impl *This = impl_from_IAMFilterData(iface);
1665
1666     return FilterMapper2_QueryInterface((IFilterMapper2*)This, riid, ppv);
1667 }
1668
1669 static ULONG WINAPI AMFilterData_AddRef(IAMFilterData * iface)
1670 {
1671     FilterMapper2Impl *This = impl_from_IAMFilterData(iface);
1672
1673     return FilterMapper2_AddRef((IFilterMapper2*)This);
1674 }
1675
1676 static ULONG WINAPI AMFilterData_Release(IAMFilterData * iface)
1677 {
1678     FilterMapper2Impl *This = impl_from_IAMFilterData(iface);
1679
1680     return FilterMapper2_Release((IFilterMapper2*)This);
1681 }
1682
1683 /*** IAMFilterData methods ***/
1684 static HRESULT WINAPI AMFilterData_ParseFilterData(IAMFilterData* iface,
1685                                                    BYTE *pData, ULONG cb,
1686                                                    BYTE **ppRegFilter2)
1687 {
1688     FilterMapper2Impl *This = impl_from_IAMFilterData(iface);
1689     HRESULT hr = S_OK;
1690     REGFILTER2 *prf2;
1691
1692     TRACE("(%p/%p)->(%p, %d, %p)\n", This, iface, pData, cb, ppRegFilter2);
1693
1694     prf2 = CoTaskMemAlloc(sizeof(*prf2));
1695     if (!prf2)
1696         return E_OUTOFMEMORY;
1697     *ppRegFilter2 = (BYTE *)&prf2;
1698
1699     hr = FM2_ReadFilterData(pData, prf2);
1700     if (FAILED(hr))
1701     {
1702         CoTaskMemFree(prf2);
1703         *ppRegFilter2 = NULL;
1704     }
1705
1706     return hr;
1707 }
1708
1709 static HRESULT WINAPI AMFilterData_CreateFilterData(IAMFilterData* iface,
1710                                                     REGFILTER2 *prf2,
1711                                                     BYTE **pRegFilterData,
1712                                                     ULONG *pcb)
1713 {
1714     FilterMapper2Impl *This = impl_from_IAMFilterData(iface);
1715
1716     TRACE("(%p/%p)->(%p, %p, %p)\n", This, iface, prf2, pRegFilterData, pcb);
1717
1718     return FM2_WriteFilterData(prf2, pRegFilterData, pcb);
1719 }
1720
1721 static const IAMFilterDataVtbl AMFilterDataVtbl = {
1722     AMFilterData_QueryInterface,
1723     AMFilterData_AddRef,
1724     AMFilterData_Release,
1725     AMFilterData_ParseFilterData,
1726     AMFilterData_CreateFilterData
1727 };