dlls/windowscodecs/metadatahandler.c

   1 /*
   2  * Copyright 2012 Vincent Povirk for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24 #define NONAMELESSUNION
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "objbase.h"
  29 #include "wincodec.h"
  30 #include "wincodecsdk.h"
  31
  32 #include "wincodecs_private.h"
  33
  34 #include "wine/debug.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
  37
  38 typedef struct MetadataHandler {
  39     IWICMetadataWriter IWICMetadataWriter_iface;
  40     LONG ref;
  41     IWICPersistStream IWICPersistStream_iface;
  42     const MetadataHandlerVtbl *vtable;
  43     MetadataItem *items;
  44     DWORD item_count;
  45     CRITICAL_SECTION lock;
  46 } MetadataHandler;
  47
  48 static inline MetadataHandler *impl_from_IWICMetadataWriter(IWICMetadataWriter *iface)
  49 {
  50     return CONTAINING_RECORD(iface, MetadataHandler, IWICMetadataWriter_iface);
  51 }
  52
  53 static inline MetadataHandler *impl_from_IWICPersistStream(IWICPersistStream *iface)
  54 {
  55     return CONTAINING_RECORD(iface, MetadataHandler, IWICPersistStream_iface);
  56 }
  57
  58 static void MetadataHandler_FreeItems(MetadataHandler *This)
  59 {
  60     int i;
  61
  62     for (i=0; i<This->item_count; i++)
  63     {
  64         PropVariantClear(&This->items[i].schema);
  65         PropVariantClear(&This->items[i].id);
  66         PropVariantClear(&This->items[i].value);
  67     }
  68
  69     HeapFree(GetProcessHeap(), 0, This->items);
  70 }
  71
  72 static HRESULT MetadataHandlerEnum_Create(MetadataHandler *parent, DWORD index,
  73     IWICEnumMetadataItem **ppIEnumMetadataItem);
  74
  75 static HRESULT WINAPI MetadataHandler_QueryInterface(IWICMetadataWriter *iface, REFIID iid,
  76     void **ppv)
  77 {
  78     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
  79     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
  80
  81     if (!ppv) return E_INVALIDARG;
  82
  83     if (IsEqualIID(&IID_IUnknown, iid) ||
  84         IsEqualIID(&IID_IWICMetadataReader, iid) ||
  85         (IsEqualIID(&IID_IWICMetadataWriter, iid) && This->vtable->is_writer))
  86     {
  87         *ppv = &This->IWICMetadataWriter_iface;
  88     }
  89     else if (IsEqualIID(&IID_IPersist, iid) ||
  90              IsEqualIID(&IID_IPersistStream, iid) ||
  91              IsEqualIID(&IID_IWICPersistStream, iid))
  92     {
  93         *ppv = &This->IWICPersistStream_iface;
  94     }
  95     else
  96     {
  97         *ppv = NULL;
  98         return E_NOINTERFACE;
  99     }
 100
 101     IUnknown_AddRef((IUnknown*)*ppv);
 102     return S_OK;
 103 }
 104
 105 static ULONG WINAPI MetadataHandler_AddRef(IWICMetadataWriter *iface)
 106 {
 107     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 108     ULONG ref = InterlockedIncrement(&This->ref);
 109
 110     TRACE("(%p) refcount=%u\n", iface, ref);
 111
 112     return ref;
 113 }
 114
 115 static ULONG WINAPI MetadataHandler_Release(IWICMetadataWriter *iface)
 116 {
 117     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 118     ULONG ref = InterlockedDecrement(&This->ref);
 119
 120     TRACE("(%p) refcount=%u\n", iface, ref);
 121
 122     if (ref == 0)
 123     {
 124         MetadataHandler_FreeItems(This);
 125         This->lock.DebugInfo->Spare[0] = 0;
 126         DeleteCriticalSection(&This->lock);
 127         HeapFree(GetProcessHeap(), 0, This);
 128     }
 129
 130     return ref;
 131 }
 132
 133 static HRESULT WINAPI MetadataHandler_GetMetadataFormat(IWICMetadataWriter *iface,
 134     GUID *pguidMetadataFormat)
 135 {
 136     FIXME("(%p,%s): stub\n", iface, debugstr_guid(pguidMetadataFormat));
 137     return E_NOTIMPL;
 138 }
 139
 140 static HRESULT WINAPI MetadataHandler_GetMetadataHandlerInfo(IWICMetadataWriter *iface,
 141     IWICMetadataHandlerInfo **ppIHandler)
 142 {
 143     FIXME("(%p,%p): stub\n", iface, ppIHandler);
 144     return E_NOTIMPL;
 145 }
 146
 147 static HRESULT WINAPI MetadataHandler_GetCount(IWICMetadataWriter *iface,
 148     UINT *pcCount)
 149 {
 150     FIXME("(%p,%p): stub\n", iface, pcCount);
 151     return E_NOTIMPL;
 152 }
 153
 154 static HRESULT WINAPI MetadataHandler_GetValueByIndex(IWICMetadataWriter *iface,
 155     UINT nIndex, PROPVARIANT *pvarSchema, PROPVARIANT *pvarId, PROPVARIANT *pvarValue)
 156 {
 157     FIXME("(%p,%u,%p,%p,%p): stub\n", iface, nIndex, pvarSchema, pvarId, pvarValue);
 158     return E_NOTIMPL;
 159 }
 160
 161 static HRESULT WINAPI MetadataHandler_GetValue(IWICMetadataWriter *iface,
 162     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, PROPVARIANT *pvarValue)
 163 {
 164     FIXME("(%p,%p,%p,%p): stub\n", iface, pvarSchema, pvarId, pvarValue);
 165     return E_NOTIMPL;
 166 }
 167
 168 static HRESULT WINAPI MetadataHandler_GetEnumerator(IWICMetadataWriter *iface,
 169     IWICEnumMetadataItem **ppIEnumMetadata)
 170 {
 171     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 172     TRACE("(%p,%p)\n", iface, ppIEnumMetadata);
 173     return MetadataHandlerEnum_Create(This, 0, ppIEnumMetadata);
 174 }
 175
 176 static HRESULT WINAPI MetadataHandler_SetValue(IWICMetadataWriter *iface,
 177     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, const PROPVARIANT *pvarValue)
 178 {
 179     FIXME("(%p,%p,%p,%p): stub\n", iface, pvarSchema, pvarId, pvarValue);
 180     return E_NOTIMPL;
 181 }
 182
 183 static HRESULT WINAPI MetadataHandler_SetValueByIndex(IWICMetadataWriter *iface,
 184     UINT nIndex, const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, const PROPVARIANT *pvarValue)
 185 {
 186     FIXME("(%p,%u,%p,%p,%p): stub\n", iface, nIndex, pvarSchema, pvarId, pvarValue);
 187     return E_NOTIMPL;
 188 }
 189
 190 static HRESULT WINAPI MetadataHandler_RemoveValue(IWICMetadataWriter *iface,
 191     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId)
 192 {
 193     FIXME("(%p,%p,%p): stub\n", iface, pvarSchema, pvarId);
 194     return E_NOTIMPL;
 195 }
 196
 197 static HRESULT WINAPI MetadataHandler_RemoveValueByIndex(IWICMetadataWriter *iface,
 198     UINT nIndex)
 199 {
 200     FIXME("(%p,%u): stub\n", iface, nIndex);
 201     return E_NOTIMPL;
 202 }
 203
 204 static const IWICMetadataWriterVtbl MetadataHandler_Vtbl = {
 205     MetadataHandler_QueryInterface,
 206     MetadataHandler_AddRef,
 207     MetadataHandler_Release,
 208     MetadataHandler_GetMetadataFormat,
 209     MetadataHandler_GetMetadataHandlerInfo,
 210     MetadataHandler_GetCount,
 211     MetadataHandler_GetValueByIndex,
 212     MetadataHandler_GetValue,
 213     MetadataHandler_GetEnumerator,
 214     MetadataHandler_SetValue,
 215     MetadataHandler_SetValueByIndex,
 216     MetadataHandler_RemoveValue,
 217     MetadataHandler_RemoveValueByIndex
 218 };
 219
 220 static HRESULT WINAPI MetadataHandler_PersistStream_QueryInterface(IWICPersistStream *iface,
 221     REFIID iid, void **ppv)
 222 {
 223     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 224     return IWICMetadataWriter_QueryInterface(&This->IWICMetadataWriter_iface, iid, ppv);
 225 }
 226
 227 static ULONG WINAPI MetadataHandler_PersistStream_AddRef(IWICPersistStream *iface)
 228 {
 229     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 230     return IWICMetadataWriter_AddRef(&This->IWICMetadataWriter_iface);
 231 }
 232
 233 static ULONG WINAPI MetadataHandler_PersistStream_Release(IWICPersistStream *iface)
 234 {
 235     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 236     return IWICMetadataWriter_Release(&This->IWICMetadataWriter_iface);
 237 }
 238
 239 static HRESULT WINAPI MetadataHandler_GetClassID(IWICPersistStream *iface,
 240     CLSID *pClassID)
 241 {
 242     FIXME("(%p,%p): stub\n", iface, pClassID);
 243     return E_NOTIMPL;
 244 }
 245
 246 static HRESULT WINAPI MetadataHandler_IsDirty(IWICPersistStream *iface)
 247 {
 248     FIXME("(%p): stub\n", iface);
 249     return E_NOTIMPL;
 250 }
 251
 252 static HRESULT WINAPI MetadataHandler_Load(IWICPersistStream *iface,
 253     IStream *pStm)
 254 {
 255     FIXME("(%p,%p): stub\n", iface, pStm);
 256     return E_NOTIMPL;
 257 }
 258
 259 static HRESULT WINAPI MetadataHandler_Save(IWICPersistStream *iface,
 260     IStream *pStm, BOOL fClearDirty)
 261 {
 262     FIXME("(%p,%p,%i): stub\n", iface, pStm, fClearDirty);
 263     return E_NOTIMPL;
 264 }
 265
 266 static HRESULT WINAPI MetadataHandler_GetSizeMax(IWICPersistStream *iface,
 267     ULARGE_INTEGER *pcbSize)
 268 {
 269     FIXME("(%p,%p): stub\n", iface, pcbSize);
 270     return E_NOTIMPL;
 271 }
 272
 273 static HRESULT WINAPI MetadataHandler_LoadEx(IWICPersistStream *iface,
 274     IStream *pIStream, const GUID *pguidPreferredVendor, DWORD dwPersistOptions)
 275 {
 276     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 277     HRESULT hr;
 278     MetadataItem *new_items=NULL;
 279     DWORD item_count=0;
 280
 281     TRACE("(%p,%p,%s,%x)\n", iface, pIStream, debugstr_guid(pguidPreferredVendor), dwPersistOptions);
 282
 283     EnterCriticalSection(&This->lock);
 284
 285     hr = This->vtable->fnLoad(pIStream, pguidPreferredVendor, dwPersistOptions,
 286         &new_items, &item_count);
 287
 288     if (SUCCEEDED(hr))
 289     {
 290         MetadataHandler_FreeItems(This);
 291         This->items = new_items;
 292         This->item_count = item_count;
 293     }
 294
 295     LeaveCriticalSection(&This->lock);
 296
 297     return hr;
 298 }
 299
 300 static HRESULT WINAPI MetadataHandler_SaveEx(IWICPersistStream *iface,
 301     IStream *pIStream, DWORD dwPersistOptions, BOOL fClearDirty)
 302 {
 303     FIXME("(%p,%p,%x,%i): stub\n", iface, pIStream, dwPersistOptions, fClearDirty);
 304     return E_NOTIMPL;
 305 }
 306
 307 static const IWICPersistStreamVtbl MetadataHandler_PersistStream_Vtbl = {
 308     MetadataHandler_PersistStream_QueryInterface,
 309     MetadataHandler_PersistStream_AddRef,
 310     MetadataHandler_PersistStream_Release,
 311     MetadataHandler_GetClassID,
 312     MetadataHandler_IsDirty,
 313     MetadataHandler_Load,
 314     MetadataHandler_Save,
 315     MetadataHandler_GetSizeMax,
 316     MetadataHandler_LoadEx,
 317     MetadataHandler_SaveEx
 318 };
 319
 320 HRESULT MetadataReader_Create(const MetadataHandlerVtbl *vtable, IUnknown *pUnkOuter, REFIID iid, void** ppv)
 321 {
 322     MetadataHandler *This;
 323     HRESULT hr;
 324
 325     TRACE("%s\n", debugstr_guid(vtable->clsid));
 326
 327     *ppv = NULL;
 328
 329     if (pUnkOuter) return CLASS_E_NOAGGREGATION;
 330
 331     This = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataHandler));
 332     if (!This) return E_OUTOFMEMORY;
 333
 334     This->IWICMetadataWriter_iface.lpVtbl = &MetadataHandler_Vtbl;
 335     This->IWICPersistStream_iface.lpVtbl = &MetadataHandler_PersistStream_Vtbl;
 336     This->ref = 1;
 337     This->vtable = vtable;
 338     This->items = NULL;
 339     This->item_count = 0;
 340
 341     InitializeCriticalSection(&This->lock);
 342     This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": MetadataHandler.lock");
 343
 344     hr = IWICMetadataWriter_QueryInterface(&This->IWICMetadataWriter_iface, iid, ppv);
 345
 346     IWICMetadataWriter_Release(&This->IWICMetadataWriter_iface);
 347
 348     return hr;
 349 }
 350
 351 typedef struct MetadataHandlerEnum {
 352     IWICEnumMetadataItem IWICEnumMetadataItem_iface;
 353     LONG ref;
 354     MetadataHandler *parent;
 355     DWORD index;
 356 } MetadataHandlerEnum;
 357
 358 static inline MetadataHandlerEnum *impl_from_IWICEnumMetadataItem(IWICEnumMetadataItem *iface)
 359 {
 360     return CONTAINING_RECORD(iface, MetadataHandlerEnum, IWICEnumMetadataItem_iface);
 361 }
 362
 363 static HRESULT WINAPI MetadataHandlerEnum_QueryInterface(IWICEnumMetadataItem *iface, REFIID iid,
 364     void **ppv)
 365 {
 366     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 367     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
 368
 369     if (!ppv) return E_INVALIDARG;
 370
 371     if (IsEqualIID(&IID_IUnknown, iid) ||
 372         IsEqualIID(&IID_IWICEnumMetadataItem, iid))
 373     {
 374         *ppv = &This->IWICEnumMetadataItem_iface;
 375     }
 376     else
 377     {
 378         *ppv = NULL;
 379         return E_NOINTERFACE;
 380     }
 381
 382     IUnknown_AddRef((IUnknown*)*ppv);
 383     return S_OK;
 384 }
 385
 386 static ULONG WINAPI MetadataHandlerEnum_AddRef(IWICEnumMetadataItem *iface)
 387 {
 388     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 389     ULONG ref = InterlockedIncrement(&This->ref);
 390
 391     TRACE("(%p) refcount=%u\n", iface, ref);
 392
 393     return ref;
 394 }
 395
 396 static ULONG WINAPI MetadataHandlerEnum_Release(IWICEnumMetadataItem *iface)
 397 {
 398     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 399     ULONG ref = InterlockedDecrement(&This->ref);
 400
 401     TRACE("(%p) refcount=%u\n", iface, ref);
 402
 403     if (ref == 0)
 404     {
 405         IWICMetadataWriter_Release(&This->parent->IWICMetadataWriter_iface);
 406         HeapFree(GetProcessHeap(), 0, This);
 407     }
 408
 409     return ref;
 410 }
 411
 412 static HRESULT WINAPI MetadataHandlerEnum_Next(IWICEnumMetadataItem *iface,
 413     ULONG celt, PROPVARIANT *rgeltSchema, PROPVARIANT *rgeltId,
 414     PROPVARIANT *rgeltValue, ULONG *pceltFetched)
 415 {
 416     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 417     ULONG new_index;
 418     HRESULT hr=S_FALSE;
 419     int i;
 420
 421     TRACE("(%p,%i)\n", iface, celt);
 422
 423     EnterCriticalSection(&This->parent->lock);
 424
 425     if (This->index >= This->parent->item_count)
 426     {
 427         *pceltFetched = 0;
 428         LeaveCriticalSection(&This->parent->lock);
 429         return S_FALSE;
 430     }
 431
 432     new_index = min(This->parent->item_count, This->index + celt);
 433     *pceltFetched = new_index - This->index;
 434
 435     if (rgeltSchema)
 436     {
 437         for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
 438             hr = PropVariantCopy(&rgeltSchema[i], &This->parent->items[i+This->index].schema);
 439     }
 440
 441     for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
 442         hr = PropVariantCopy(&rgeltId[i], &This->parent->items[i+This->index].id);
 443
 444     if (rgeltValue)
 445     {
 446         for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
 447             hr = PropVariantCopy(&rgeltValue[i], &This->parent->items[i+This->index].value);
 448     }
 449
 450     if (SUCCEEDED(hr))
 451     {
 452         This->index = new_index;
 453     }
 454
 455     LeaveCriticalSection(&This->parent->lock);
 456
 457     return hr;
 458 }
 459
 460 static HRESULT WINAPI MetadataHandlerEnum_Skip(IWICEnumMetadataItem *iface,
 461     ULONG celt)
 462 {
 463     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 464
 465     EnterCriticalSection(&This->parent->lock);
 466
 467     This->index += celt;
 468
 469     LeaveCriticalSection(&This->parent->lock);
 470
 471     return S_OK;
 472 }
 473
 474 static HRESULT WINAPI MetadataHandlerEnum_Reset(IWICEnumMetadataItem *iface)
 475 {
 476     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 477
 478     EnterCriticalSection(&This->parent->lock);
 479
 480     This->index = 0;
 481
 482     LeaveCriticalSection(&This->parent->lock);
 483
 484     return S_OK;
 485 }
 486
 487 static HRESULT WINAPI MetadataHandlerEnum_Clone(IWICEnumMetadataItem *iface,
 488     IWICEnumMetadataItem **ppIEnumMetadataItem)
 489 {
 490     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 491     HRESULT hr;
 492
 493     EnterCriticalSection(&This->parent->lock);
 494
 495     hr = MetadataHandlerEnum_Create(This->parent, This->index, ppIEnumMetadataItem);
 496
 497     LeaveCriticalSection(&This->parent->lock);
 498
 499     return hr;
 500 }
 501
 502 static const IWICEnumMetadataItemVtbl MetadataHandlerEnum_Vtbl = {
 503     MetadataHandlerEnum_QueryInterface,
 504     MetadataHandlerEnum_AddRef,
 505     MetadataHandlerEnum_Release,
 506     MetadataHandlerEnum_Next,
 507     MetadataHandlerEnum_Skip,
 508     MetadataHandlerEnum_Reset,
 509     MetadataHandlerEnum_Clone
 510 };
 511
 512 static HRESULT MetadataHandlerEnum_Create(MetadataHandler *parent, DWORD index,
 513     IWICEnumMetadataItem **ppIEnumMetadataItem)
 514 {
 515     MetadataHandlerEnum *This;
 516
 517     *ppIEnumMetadataItem = NULL;
 518
 519     This = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataHandlerEnum));
 520     if (!This) return E_OUTOFMEMORY;
 521
 522     IWICMetadataWriter_AddRef(&parent->IWICMetadataWriter_iface);
 523
 524     This->IWICEnumMetadataItem_iface.lpVtbl = &MetadataHandlerEnum_Vtbl;
 525     This->ref = 1;
 526     This->parent = parent;
 527     This->index = index;
 528
 529     *ppIEnumMetadataItem = &This->IWICEnumMetadataItem_iface;
 530
 531     return S_OK;
 532 }
 533
 534 static HRESULT LoadUnknownMetadata(IStream *input, const GUID *preferred_vendor,
 535     DWORD persist_options, MetadataItem **items, DWORD *item_count)
 536 {
 537     HRESULT hr;
 538     MetadataItem *result;
 539     STATSTG stat;
 540     BYTE *data;
 541     ULONG bytesread;
 542
 543     TRACE("\n");
 544
 545     hr = IStream_Stat(input, &stat, STATFLAG_NONAME);
 546     if (FAILED(hr))
 547         return hr;
 548
 549     data = HeapAlloc(GetProcessHeap(), 0, stat.cbSize.QuadPart);
 550     if (!data) return E_OUTOFMEMORY;
 551
 552     hr = IStream_Read(input, data, stat.cbSize.QuadPart, &bytesread);
 553     if (FAILED(hr))
 554     {
 555         HeapFree(GetProcessHeap(), 0, data);
 556         return hr;
 557     }
 558
 559     result = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataItem));
 560     if (!result)
 561     {
 562         HeapFree(GetProcessHeap(), 0, data);
 563         return E_OUTOFMEMORY;
 564     }
 565
 566     PropVariantInit(&result[0].schema);
 567     PropVariantInit(&result[0].id);
 568     PropVariantInit(&result[0].value);
 569
 570     result[0].value.vt = VT_BLOB;
 571     result[0].value.u.blob.cbSize = bytesread;
 572     result[0].value.u.blob.pBlobData = data;
 573
 574     *items = result;
 575     *item_count = 1;
 576
 577     return S_OK;
 578 }
 579
 580 static const MetadataHandlerVtbl UnknownMetadataReader_Vtbl = {
 581     0,
 582     &CLSID_WICUnknownMetadataReader,
 583     LoadUnknownMetadata
 584 };
 585
 586 HRESULT UnknownMetadataReader_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
 587 {
 588     return MetadataReader_Create(&UnknownMetadataReader_Vtbl, pUnkOuter, iid, ppv);
 589 }