dlls/windowscodecs/metadatahandler.c

   1 /*
   2  * Copyright 2012 Vincent Povirk for CodeWeavers
   3  * Copyright 2012 Dmitry Timoshkov
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include <stdarg.h>
  21 #include <stdio.h>
  22
  23 #define COBJMACROS
  24 #define NONAMELESSUNION
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winternl.h"
  29 #include "objbase.h"
  30 #include "propvarutil.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     DWORD 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=%lu\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=%lu\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_GetMetadataHandlerInfo(IWICMetadataWriter *iface,
 134     IWICMetadataHandlerInfo **ppIHandler)
 135 {
 136     HRESULT hr;
 137     IWICComponentInfo *component_info;
 138     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 139
 140     TRACE("%p,%p\n", iface, ppIHandler);
 141
 142     hr = CreateComponentInfo(This->vtable->clsid, &component_info);
 143     if (FAILED(hr)) return hr;
 144
 145     hr = IWICComponentInfo_QueryInterface(component_info, &IID_IWICMetadataHandlerInfo,
 146         (void **)ppIHandler);
 147
 148     IWICComponentInfo_Release(component_info);
 149     return hr;
 150 }
 151
 152 static HRESULT WINAPI MetadataHandler_GetMetadataFormat(IWICMetadataWriter *iface,
 153     GUID *pguidMetadataFormat)
 154 {
 155     HRESULT hr;
 156     IWICMetadataHandlerInfo *metadata_info;
 157
 158     TRACE("%p,%p\n", iface, pguidMetadataFormat);
 159
 160     if (!pguidMetadataFormat) return E_INVALIDARG;
 161
 162     hr = MetadataHandler_GetMetadataHandlerInfo(iface, &metadata_info);
 163     if (FAILED(hr)) return hr;
 164
 165     hr = IWICMetadataHandlerInfo_GetMetadataFormat(metadata_info, pguidMetadataFormat);
 166     IWICMetadataHandlerInfo_Release(metadata_info);
 167
 168     return hr;
 169 }
 170
 171 static HRESULT WINAPI MetadataHandler_GetCount(IWICMetadataWriter *iface,
 172     UINT *pcCount)
 173 {
 174     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 175
 176     TRACE("%p,%p\n", iface, pcCount);
 177
 178     if (!pcCount) return E_INVALIDARG;
 179
 180     *pcCount = This->item_count;
 181     return S_OK;
 182 }
 183
 184 static HRESULT WINAPI MetadataHandler_GetValueByIndex(IWICMetadataWriter *iface,
 185     UINT index, PROPVARIANT *schema, PROPVARIANT *id, PROPVARIANT *value)
 186 {
 187     HRESULT hr = S_OK;
 188     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 189
 190     TRACE("%p,%u,%p,%p,%p\n", iface, index, schema, id, value);
 191
 192     EnterCriticalSection(&This->lock);
 193
 194     if (index >= This->item_count)
 195     {
 196         LeaveCriticalSection(&This->lock);
 197         return E_INVALIDARG;
 198     }
 199
 200     if (schema)
 201         hr = PropVariantCopy(schema, &This->items[index].schema);
 202
 203     if (SUCCEEDED(hr) && id)
 204         hr = PropVariantCopy(id, &This->items[index].id);
 205
 206     if (SUCCEEDED(hr) && value)
 207         hr = PropVariantCopy(value, &This->items[index].value);
 208
 209     LeaveCriticalSection(&This->lock);
 210     return hr;
 211 }
 212
 213 static HRESULT WINAPI MetadataHandler_GetValue(IWICMetadataWriter *iface,
 214     const PROPVARIANT *schema, const PROPVARIANT *id, PROPVARIANT *value)
 215 {
 216     UINT i;
 217     HRESULT hr = WINCODEC_ERR_PROPERTYNOTFOUND;
 218     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 219
 220     TRACE("(%p,%s,%s,%p)\n", iface, wine_dbgstr_variant((const VARIANT *)schema), wine_dbgstr_variant((const VARIANT *)id), value);
 221
 222     if (!id) return E_INVALIDARG;
 223
 224     EnterCriticalSection(&This->lock);
 225
 226     for (i = 0; i < This->item_count; i++)
 227     {
 228         if (schema && This->items[i].schema.vt != VT_EMPTY)
 229         {
 230             if (PropVariantCompareEx(schema, &This->items[i].schema, 0, PVCF_USESTRCMPI) != 0) continue;
 231         }
 232
 233         if (PropVariantCompareEx(id, &This->items[i].id, 0, PVCF_USESTRCMPI) != 0) continue;
 234
 235         hr = value ? PropVariantCopy(value, &This->items[i].value) : S_OK;
 236         break;
 237     }
 238
 239     LeaveCriticalSection(&This->lock);
 240     return hr;
 241 }
 242
 243 static HRESULT WINAPI MetadataHandler_GetEnumerator(IWICMetadataWriter *iface,
 244     IWICEnumMetadataItem **ppIEnumMetadata)
 245 {
 246     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
 247     TRACE("(%p,%p)\n", iface, ppIEnumMetadata);
 248     return MetadataHandlerEnum_Create(This, 0, ppIEnumMetadata);
 249 }
 250
 251 static HRESULT WINAPI MetadataHandler_SetValue(IWICMetadataWriter *iface,
 252     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, const PROPVARIANT *pvarValue)
 253 {
 254     FIXME("(%p,%p,%p,%p): stub\n", iface, pvarSchema, pvarId, pvarValue);
 255     return E_NOTIMPL;
 256 }
 257
 258 static HRESULT WINAPI MetadataHandler_SetValueByIndex(IWICMetadataWriter *iface,
 259     UINT nIndex, const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, const PROPVARIANT *pvarValue)
 260 {
 261     FIXME("(%p,%u,%p,%p,%p): stub\n", iface, nIndex, pvarSchema, pvarId, pvarValue);
 262     return E_NOTIMPL;
 263 }
 264
 265 static HRESULT WINAPI MetadataHandler_RemoveValue(IWICMetadataWriter *iface,
 266     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId)
 267 {
 268     FIXME("(%p,%p,%p): stub\n", iface, pvarSchema, pvarId);
 269     return E_NOTIMPL;
 270 }
 271
 272 static HRESULT WINAPI MetadataHandler_RemoveValueByIndex(IWICMetadataWriter *iface,
 273     UINT nIndex)
 274 {
 275     FIXME("(%p,%u): stub\n", iface, nIndex);
 276     return E_NOTIMPL;
 277 }
 278
 279 static const IWICMetadataWriterVtbl MetadataHandler_Vtbl = {
 280     MetadataHandler_QueryInterface,
 281     MetadataHandler_AddRef,
 282     MetadataHandler_Release,
 283     MetadataHandler_GetMetadataFormat,
 284     MetadataHandler_GetMetadataHandlerInfo,
 285     MetadataHandler_GetCount,
 286     MetadataHandler_GetValueByIndex,
 287     MetadataHandler_GetValue,
 288     MetadataHandler_GetEnumerator,
 289     MetadataHandler_SetValue,
 290     MetadataHandler_SetValueByIndex,
 291     MetadataHandler_RemoveValue,
 292     MetadataHandler_RemoveValueByIndex
 293 };
 294
 295 static HRESULT WINAPI MetadataHandler_PersistStream_QueryInterface(IWICPersistStream *iface,
 296     REFIID iid, void **ppv)
 297 {
 298     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 299     return IWICMetadataWriter_QueryInterface(&This->IWICMetadataWriter_iface, iid, ppv);
 300 }
 301
 302 static ULONG WINAPI MetadataHandler_PersistStream_AddRef(IWICPersistStream *iface)
 303 {
 304     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 305     return IWICMetadataWriter_AddRef(&This->IWICMetadataWriter_iface);
 306 }
 307
 308 static ULONG WINAPI MetadataHandler_PersistStream_Release(IWICPersistStream *iface)
 309 {
 310     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 311     return IWICMetadataWriter_Release(&This->IWICMetadataWriter_iface);
 312 }
 313
 314 static HRESULT WINAPI MetadataHandler_GetClassID(IWICPersistStream *iface,
 315     CLSID *pClassID)
 316 {
 317     FIXME("(%p,%p): stub\n", iface, pClassID);
 318     return E_NOTIMPL;
 319 }
 320
 321 static HRESULT WINAPI MetadataHandler_IsDirty(IWICPersistStream *iface)
 322 {
 323     FIXME("(%p): stub\n", iface);
 324     return E_NOTIMPL;
 325 }
 326
 327 static HRESULT WINAPI MetadataHandler_Load(IWICPersistStream *iface,
 328     IStream *pStm)
 329 {
 330     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 331     TRACE("(%p,%p)\n", iface, pStm);
 332     return IWICPersistStream_LoadEx(&This->IWICPersistStream_iface, pStm, NULL, WICPersistOptionDefault);
 333 }
 334
 335 static HRESULT WINAPI MetadataHandler_Save(IWICPersistStream *iface,
 336     IStream *pStm, BOOL fClearDirty)
 337 {
 338     FIXME("(%p,%p,%i): stub\n", iface, pStm, fClearDirty);
 339     return E_NOTIMPL;
 340 }
 341
 342 static HRESULT WINAPI MetadataHandler_GetSizeMax(IWICPersistStream *iface,
 343     ULARGE_INTEGER *pcbSize)
 344 {
 345     FIXME("(%p,%p): stub\n", iface, pcbSize);
 346     return E_NOTIMPL;
 347 }
 348
 349 static HRESULT WINAPI MetadataHandler_LoadEx(IWICPersistStream *iface,
 350     IStream *pIStream, const GUID *pguidPreferredVendor, DWORD dwPersistOptions)
 351 {
 352     MetadataHandler *This = impl_from_IWICPersistStream(iface);
 353     HRESULT hr;
 354     MetadataItem *new_items=NULL;
 355     DWORD item_count=0;
 356
 357     TRACE("(%p,%p,%s,%lx)\n", iface, pIStream, debugstr_guid(pguidPreferredVendor), dwPersistOptions);
 358
 359     EnterCriticalSection(&This->lock);
 360
 361     hr = This->vtable->fnLoad(pIStream, pguidPreferredVendor, dwPersistOptions,
 362         &new_items, &item_count);
 363
 364     if (SUCCEEDED(hr))
 365     {
 366         MetadataHandler_FreeItems(This);
 367         This->items = new_items;
 368         This->item_count = item_count;
 369     }
 370
 371     LeaveCriticalSection(&This->lock);
 372
 373     return hr;
 374 }
 375
 376 static HRESULT WINAPI MetadataHandler_SaveEx(IWICPersistStream *iface,
 377     IStream *pIStream, DWORD dwPersistOptions, BOOL fClearDirty)
 378 {
 379     FIXME("(%p,%p,%lx,%i): stub\n", iface, pIStream, dwPersistOptions, fClearDirty);
 380     return E_NOTIMPL;
 381 }
 382
 383 static const IWICPersistStreamVtbl MetadataHandler_PersistStream_Vtbl = {
 384     MetadataHandler_PersistStream_QueryInterface,
 385     MetadataHandler_PersistStream_AddRef,
 386     MetadataHandler_PersistStream_Release,
 387     MetadataHandler_GetClassID,
 388     MetadataHandler_IsDirty,
 389     MetadataHandler_Load,
 390     MetadataHandler_Save,
 391     MetadataHandler_GetSizeMax,
 392     MetadataHandler_LoadEx,
 393     MetadataHandler_SaveEx
 394 };
 395
 396 HRESULT MetadataReader_Create(const MetadataHandlerVtbl *vtable, REFIID iid, void** ppv)
 397 {
 398     MetadataHandler *This;
 399     HRESULT hr;
 400
 401     TRACE("%s\n", debugstr_guid(vtable->clsid));
 402
 403     *ppv = NULL;
 404
 405     This = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataHandler));
 406     if (!This) return E_OUTOFMEMORY;
 407
 408     This->IWICMetadataWriter_iface.lpVtbl = &MetadataHandler_Vtbl;
 409     This->IWICPersistStream_iface.lpVtbl = &MetadataHandler_PersistStream_Vtbl;
 410     This->ref = 1;
 411     This->vtable = vtable;
 412     This->items = NULL;
 413     This->item_count = 0;
 414
 415     InitializeCriticalSection(&This->lock);
 416     This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": MetadataHandler.lock");
 417
 418     hr = IWICMetadataWriter_QueryInterface(&This->IWICMetadataWriter_iface, iid, ppv);
 419
 420     IWICMetadataWriter_Release(&This->IWICMetadataWriter_iface);
 421
 422     return hr;
 423 }
 424
 425 typedef struct MetadataHandlerEnum {
 426     IWICEnumMetadataItem IWICEnumMetadataItem_iface;
 427     LONG ref;
 428     MetadataHandler *parent;
 429     DWORD index;
 430 } MetadataHandlerEnum;
 431
 432 static inline MetadataHandlerEnum *impl_from_IWICEnumMetadataItem(IWICEnumMetadataItem *iface)
 433 {
 434     return CONTAINING_RECORD(iface, MetadataHandlerEnum, IWICEnumMetadataItem_iface);
 435 }
 436
 437 static HRESULT WINAPI MetadataHandlerEnum_QueryInterface(IWICEnumMetadataItem *iface, REFIID iid,
 438     void **ppv)
 439 {
 440     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 441     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
 442
 443     if (!ppv) return E_INVALIDARG;
 444
 445     if (IsEqualIID(&IID_IUnknown, iid) ||
 446         IsEqualIID(&IID_IWICEnumMetadataItem, iid))
 447     {
 448         *ppv = &This->IWICEnumMetadataItem_iface;
 449     }
 450     else
 451     {
 452         *ppv = NULL;
 453         return E_NOINTERFACE;
 454     }
 455
 456     IUnknown_AddRef((IUnknown*)*ppv);
 457     return S_OK;
 458 }
 459
 460 static ULONG WINAPI MetadataHandlerEnum_AddRef(IWICEnumMetadataItem *iface)
 461 {
 462     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 463     ULONG ref = InterlockedIncrement(&This->ref);
 464
 465     TRACE("(%p) refcount=%lu\n", iface, ref);
 466
 467     return ref;
 468 }
 469
 470 static ULONG WINAPI MetadataHandlerEnum_Release(IWICEnumMetadataItem *iface)
 471 {
 472     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 473     ULONG ref = InterlockedDecrement(&This->ref);
 474
 475     TRACE("(%p) refcount=%lu\n", iface, ref);
 476
 477     if (ref == 0)
 478     {
 479         IWICMetadataWriter_Release(&This->parent->IWICMetadataWriter_iface);
 480         HeapFree(GetProcessHeap(), 0, This);
 481     }
 482
 483     return ref;
 484 }
 485
 486 static HRESULT WINAPI MetadataHandlerEnum_Next(IWICEnumMetadataItem *iface,
 487     ULONG celt, PROPVARIANT *rgeltSchema, PROPVARIANT *rgeltId,
 488     PROPVARIANT *rgeltValue, ULONG *pceltFetched)
 489 {
 490     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 491     ULONG new_index;
 492     HRESULT hr=S_FALSE;
 493     ULONG i;
 494     ULONG fetched;
 495
 496     TRACE("(%p,%li)\n", iface, celt);
 497
 498     if (!pceltFetched)
 499         pceltFetched = &fetched;
 500
 501     EnterCriticalSection(&This->parent->lock);
 502
 503     if (This->index >= This->parent->item_count)
 504     {
 505         *pceltFetched = 0;
 506         LeaveCriticalSection(&This->parent->lock);
 507         return S_FALSE;
 508     }
 509
 510     new_index = min(This->parent->item_count, This->index + celt);
 511     *pceltFetched = new_index - This->index;
 512
 513     if (rgeltSchema)
 514     {
 515         for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
 516             hr = PropVariantCopy(&rgeltSchema[i], &This->parent->items[i+This->index].schema);
 517     }
 518
 519     for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
 520         hr = PropVariantCopy(&rgeltId[i], &This->parent->items[i+This->index].id);
 521
 522     if (rgeltValue)
 523     {
 524         for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
 525             hr = PropVariantCopy(&rgeltValue[i], &This->parent->items[i+This->index].value);
 526     }
 527
 528     if (SUCCEEDED(hr))
 529     {
 530         This->index = new_index;
 531     }
 532
 533     LeaveCriticalSection(&This->parent->lock);
 534
 535     return hr;
 536 }
 537
 538 static HRESULT WINAPI MetadataHandlerEnum_Skip(IWICEnumMetadataItem *iface,
 539     ULONG celt)
 540 {
 541     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 542
 543     EnterCriticalSection(&This->parent->lock);
 544
 545     This->index += celt;
 546
 547     LeaveCriticalSection(&This->parent->lock);
 548
 549     return S_OK;
 550 }
 551
 552 static HRESULT WINAPI MetadataHandlerEnum_Reset(IWICEnumMetadataItem *iface)
 553 {
 554     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 555
 556     EnterCriticalSection(&This->parent->lock);
 557
 558     This->index = 0;
 559
 560     LeaveCriticalSection(&This->parent->lock);
 561
 562     return S_OK;
 563 }
 564
 565 static HRESULT WINAPI MetadataHandlerEnum_Clone(IWICEnumMetadataItem *iface,
 566     IWICEnumMetadataItem **ppIEnumMetadataItem)
 567 {
 568     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
 569     HRESULT hr;
 570
 571     EnterCriticalSection(&This->parent->lock);
 572
 573     hr = MetadataHandlerEnum_Create(This->parent, This->index, ppIEnumMetadataItem);
 574
 575     LeaveCriticalSection(&This->parent->lock);
 576
 577     return hr;
 578 }
 579
 580 static const IWICEnumMetadataItemVtbl MetadataHandlerEnum_Vtbl = {
 581     MetadataHandlerEnum_QueryInterface,
 582     MetadataHandlerEnum_AddRef,
 583     MetadataHandlerEnum_Release,
 584     MetadataHandlerEnum_Next,
 585     MetadataHandlerEnum_Skip,
 586     MetadataHandlerEnum_Reset,
 587     MetadataHandlerEnum_Clone
 588 };
 589
 590 static HRESULT MetadataHandlerEnum_Create(MetadataHandler *parent, DWORD index,
 591     IWICEnumMetadataItem **ppIEnumMetadataItem)
 592 {
 593     MetadataHandlerEnum *This;
 594
 595     if (!ppIEnumMetadataItem) return E_INVALIDARG;
 596
 597     *ppIEnumMetadataItem = NULL;
 598
 599     This = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataHandlerEnum));
 600     if (!This) return E_OUTOFMEMORY;
 601
 602     IWICMetadataWriter_AddRef(&parent->IWICMetadataWriter_iface);
 603
 604     This->IWICEnumMetadataItem_iface.lpVtbl = &MetadataHandlerEnum_Vtbl;
 605     This->ref = 1;
 606     This->parent = parent;
 607     This->index = index;
 608
 609     *ppIEnumMetadataItem = &This->IWICEnumMetadataItem_iface;
 610
 611     return S_OK;
 612 }
 613
 614 static HRESULT LoadUnknownMetadata(IStream *input, const GUID *preferred_vendor,
 615     DWORD persist_options, MetadataItem **items, DWORD *item_count)
 616 {
 617     HRESULT hr;
 618     MetadataItem *result;
 619     STATSTG stat;
 620     BYTE *data;
 621     ULONG bytesread;
 622
 623     TRACE("\n");
 624
 625     hr = IStream_Stat(input, &stat, STATFLAG_NONAME);
 626     if (FAILED(hr))
 627         return hr;
 628
 629     data = HeapAlloc(GetProcessHeap(), 0, stat.cbSize.QuadPart);
 630     if (!data) return E_OUTOFMEMORY;
 631
 632     hr = IStream_Read(input, data, stat.cbSize.QuadPart, &bytesread);
 633     if (bytesread != stat.cbSize.QuadPart) hr = E_FAIL;
 634     if (hr != S_OK)
 635     {
 636         HeapFree(GetProcessHeap(), 0, data);
 637         return hr;
 638     }
 639
 640     result = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(MetadataItem));
 641     if (!result)
 642     {
 643         HeapFree(GetProcessHeap(), 0, data);
 644         return E_OUTOFMEMORY;
 645     }
 646
 647     PropVariantInit(&result[0].schema);
 648     PropVariantInit(&result[0].id);
 649     PropVariantInit(&result[0].value);
 650
 651     result[0].value.vt = VT_BLOB;
 652     result[0].value.blob.cbSize = bytesread;
 653     result[0].value.blob.pBlobData = data;
 654
 655     *items = result;
 656     *item_count = 1;
 657
 658     return S_OK;
 659 }
 660
 661 static const MetadataHandlerVtbl UnknownMetadataReader_Vtbl = {
 662     0,
 663     &CLSID_WICUnknownMetadataReader,
 664     LoadUnknownMetadata
 665 };
 666
 667 HRESULT UnknownMetadataReader_CreateInstance(REFIID iid, void** ppv)
 668 {
 669     return MetadataReader_Create(&UnknownMetadataReader_Vtbl, iid, ppv);
 670 }
 671
 672 #define SWAP_USHORT(x) do { if (!native_byte_order) (x) = RtlUshortByteSwap(x); } while(0)
 673 #define SWAP_ULONG(x) do { if (!native_byte_order) (x) = RtlUlongByteSwap(x); } while(0)
 674 #define SWAP_ULONGLONG(x) do { if (!native_byte_order) (x) = RtlUlonglongByteSwap(x); } while(0)
 675
 676 struct IFD_entry
 677 {
 678     SHORT id;
 679     SHORT type;
 680     ULONG count;
 681     LONG  value;
 682 };
 683
 684 #define IFD_BYTE 1
 685 #define IFD_ASCII 2
 686 #define IFD_SHORT 3
 687 #define IFD_LONG 4
 688 #define IFD_RATIONAL 5
 689 #define IFD_SBYTE 6
 690 #define IFD_UNDEFINED 7
 691 #define IFD_SSHORT 8
 692 #define IFD_SLONG 9
 693 #define IFD_SRATIONAL 10
 694 #define IFD_FLOAT 11
 695 #define IFD_DOUBLE 12
 696 #define IFD_IFD 13
 697
 698 static int tag_to_vt(SHORT tag)
 699 {
 700     static const int tag2vt[] =
 701     {
 702         VT_EMPTY, /* 0 */
 703         VT_UI1,   /* IFD_BYTE 1 */
 704         VT_LPSTR, /* IFD_ASCII 2 */
 705         VT_UI2,   /* IFD_SHORT 3 */
 706         VT_UI4,   /* IFD_LONG 4 */
 707         VT_UI8,   /* IFD_RATIONAL 5 */
 708         VT_I1,    /* IFD_SBYTE 6 */
 709         VT_BLOB,  /* IFD_UNDEFINED 7 */
 710         VT_I2,    /* IFD_SSHORT 8 */
 711         VT_I4,    /* IFD_SLONG 9 */
 712         VT_I8,    /* IFD_SRATIONAL 10 */
 713         VT_R4,    /* IFD_FLOAT 11 */
 714         VT_R8,    /* IFD_DOUBLE 12 */
 715         VT_BLOB,  /* IFD_IFD 13 */
 716     };
 717     return (tag > 0 && tag <= 13) ? tag2vt[tag] : VT_BLOB;
 718 }
 719
 720 static HRESULT load_IFD_entry(IStream *input, const struct IFD_entry *entry,
 721                               MetadataItem *item, BOOL native_byte_order)
 722 {
 723     ULONG count, value, i, bytesread;
 724     SHORT type;
 725     LARGE_INTEGER pos;
 726     HRESULT hr;
 727
 728     item->schema.vt = VT_EMPTY;
 729     item->id.vt = VT_UI2;
 730     item->id.uiVal = entry->id;
 731     SWAP_USHORT(item->id.uiVal);
 732
 733     count = entry->count;
 734     SWAP_ULONG(count);
 735     type = entry->type;
 736     SWAP_USHORT(type);
 737     item->value.vt = tag_to_vt(type);
 738     value = entry->value;
 739     SWAP_ULONG(value);
 740
 741     switch (type)
 742     {
 743      case IFD_BYTE:
 744      case IFD_SBYTE:
 745         if (!count) count = 1;
 746
 747         if (count <= 4)
 748         {
 749             const BYTE *data = (const BYTE *)&entry->value;
 750
 751             if (count == 1)
 752                 item->value.bVal = data[0];
 753             else
 754             {
 755                 item->value.vt |= VT_VECTOR;
 756                 item->value.caub.cElems = count;
 757                 item->value.caub.pElems = HeapAlloc(GetProcessHeap(), 0, count);
 758                 memcpy(item->value.caub.pElems, data, count);
 759             }
 760             break;
 761         }
 762
 763         item->value.vt |= VT_VECTOR;
 764         item->value.caub.cElems = count;
 765         item->value.caub.pElems = HeapAlloc(GetProcessHeap(), 0, count);
 766         if (!item->value.caub.pElems) return E_OUTOFMEMORY;
 767
 768         pos.QuadPart = value;
 769         hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 770         if (FAILED(hr))
 771         {
 772             HeapFree(GetProcessHeap(), 0, item->value.caub.pElems);
 773             return hr;
 774         }
 775         hr = IStream_Read(input, item->value.caub.pElems, count, &bytesread);
 776         if (bytesread != count) hr = E_FAIL;
 777         if (hr != S_OK)
 778         {
 779             HeapFree(GetProcessHeap(), 0, item->value.caub.pElems);
 780             return hr;
 781         }
 782         break;
 783     case IFD_SHORT:
 784     case IFD_SSHORT:
 785         if (!count) count = 1;
 786
 787         if (count <= 2)
 788         {
 789             const SHORT *data = (const SHORT *)&entry->value;
 790
 791             if (count == 1)
 792             {
 793                 item->value.uiVal = data[0];
 794                 SWAP_USHORT(item->value.uiVal);
 795             }
 796             else
 797             {
 798                 item->value.vt |= VT_VECTOR;
 799                 item->value.caui.cElems = count;
 800                 item->value.caui.pElems = HeapAlloc(GetProcessHeap(), 0, count * 2);
 801                 memcpy(item->value.caui.pElems, data, count * 2);
 802                 for (i = 0; i < count; i++)
 803                     SWAP_USHORT(item->value.caui.pElems[i]);
 804             }
 805             break;
 806         }
 807
 808         item->value.vt |= VT_VECTOR;
 809         item->value.caui.cElems = count;
 810         item->value.caui.pElems = HeapAlloc(GetProcessHeap(), 0, count * 2);
 811         if (!item->value.caui.pElems) return E_OUTOFMEMORY;
 812
 813         pos.QuadPart = value;
 814         hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 815         if (FAILED(hr))
 816         {
 817             HeapFree(GetProcessHeap(), 0, item->value.caui.pElems);
 818             return hr;
 819         }
 820         hr = IStream_Read(input, item->value.caui.pElems, count * 2, &bytesread);
 821         if (bytesread != count * 2) hr = E_FAIL;
 822         if (hr != S_OK)
 823         {
 824             HeapFree(GetProcessHeap(), 0, item->value.caui.pElems);
 825             return hr;
 826         }
 827         for (i = 0; i < count; i++)
 828             SWAP_USHORT(item->value.caui.pElems[i]);
 829         break;
 830     case IFD_LONG:
 831     case IFD_SLONG:
 832     case IFD_FLOAT:
 833         if (!count) count = 1;
 834
 835         if (count == 1)
 836         {
 837             item->value.ulVal = value;
 838             break;
 839         }
 840
 841         item->value.vt |= VT_VECTOR;
 842         item->value.caul.cElems = count;
 843         item->value.caul.pElems = HeapAlloc(GetProcessHeap(), 0, count * 4);
 844         if (!item->value.caul.pElems) return E_OUTOFMEMORY;
 845
 846         pos.QuadPart = value;
 847         hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 848         if (FAILED(hr))
 849         {
 850             HeapFree(GetProcessHeap(), 0, item->value.caul.pElems);
 851             return hr;
 852         }
 853         hr = IStream_Read(input, item->value.caul.pElems, count * 4, &bytesread);
 854         if (bytesread != count * 4) hr = E_FAIL;
 855         if (hr != S_OK)
 856         {
 857             HeapFree(GetProcessHeap(), 0, item->value.caul.pElems);
 858             return hr;
 859         }
 860         for (i = 0; i < count; i++)
 861             SWAP_ULONG(item->value.caul.pElems[i]);
 862         break;
 863     case IFD_RATIONAL:
 864     case IFD_SRATIONAL:
 865     case IFD_DOUBLE:
 866         if (!count)
 867         {
 868             FIXME("IFD field type %d, count 0\n", type);
 869             item->value.vt = VT_EMPTY;
 870             break;
 871         }
 872
 873         if (count == 1)
 874         {
 875             ULONGLONG ull;
 876
 877             pos.QuadPart = value;
 878             hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 879             if (FAILED(hr)) return hr;
 880
 881             hr = IStream_Read(input, &ull, sizeof(ull), &bytesread);
 882             if (bytesread != sizeof(ull)) hr = E_FAIL;
 883             if (hr != S_OK) return hr;
 884
 885             item->value.uhVal.QuadPart = ull;
 886
 887             if (type == IFD_DOUBLE)
 888                 SWAP_ULONGLONG(item->value.uhVal.QuadPart);
 889             else
 890             {
 891                 SWAP_ULONG(item->value.uhVal.LowPart);
 892                 SWAP_ULONG(item->value.uhVal.HighPart);
 893             }
 894             break;
 895         }
 896         else
 897         {
 898             item->value.vt |= VT_VECTOR;
 899             item->value.cauh.cElems = count;
 900             item->value.cauh.pElems = HeapAlloc(GetProcessHeap(), 0, count * 8);
 901             if (!item->value.cauh.pElems) return E_OUTOFMEMORY;
 902
 903             pos.QuadPart = value;
 904             hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 905             if (FAILED(hr))
 906             {
 907                 HeapFree(GetProcessHeap(), 0, item->value.cauh.pElems);
 908                 return hr;
 909             }
 910             hr = IStream_Read(input, item->value.cauh.pElems, count * 8, &bytesread);
 911             if (bytesread != count * 8) hr = E_FAIL;
 912             if (hr != S_OK)
 913             {
 914                 HeapFree(GetProcessHeap(), 0, item->value.cauh.pElems);
 915                 return hr;
 916             }
 917             for (i = 0; i < count; i++)
 918             {
 919                 if (type == IFD_DOUBLE)
 920                     SWAP_ULONGLONG(item->value.cauh.pElems[i].QuadPart);
 921                 else
 922                 {
 923                     SWAP_ULONG(item->value.cauh.pElems[i].LowPart);
 924                     SWAP_ULONG(item->value.cauh.pElems[i].HighPart);
 925                 }
 926             }
 927         }
 928         break;
 929     case IFD_ASCII:
 930         item->value.pszVal = HeapAlloc(GetProcessHeap(), 0, count + 1);
 931         if (!item->value.pszVal) return E_OUTOFMEMORY;
 932
 933         if (count <= 4)
 934         {
 935             const char *data = (const char *)&entry->value;
 936             memcpy(item->value.pszVal, data, count);
 937             item->value.pszVal[count] = 0;
 938             break;
 939         }
 940
 941         pos.QuadPart = value;
 942         hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 943         if (FAILED(hr))
 944         {
 945             HeapFree(GetProcessHeap(), 0, item->value.pszVal);
 946             return hr;
 947         }
 948         hr = IStream_Read(input, item->value.pszVal, count, &bytesread);
 949         if (bytesread != count) hr = E_FAIL;
 950         if (hr != S_OK)
 951         {
 952             HeapFree(GetProcessHeap(), 0, item->value.pszVal);
 953             return hr;
 954         }
 955         item->value.pszVal[count] = 0;
 956         break;
 957     case IFD_UNDEFINED:
 958         if (!count)
 959         {
 960             FIXME("IFD field type %d, count 0\n", type);
 961             item->value.vt = VT_EMPTY;
 962             break;
 963         }
 964
 965         item->value.blob.pBlobData = HeapAlloc(GetProcessHeap(), 0, count);
 966         if (!item->value.blob.pBlobData) return E_OUTOFMEMORY;
 967
 968         item->value.blob.cbSize = count;
 969
 970         if (count <= 4)
 971         {
 972             const char *data = (const char *)&entry->value;
 973             memcpy(item->value.blob.pBlobData, data, count);
 974             break;
 975         }
 976
 977         pos.QuadPart = value;
 978         hr = IStream_Seek(input, pos, SEEK_SET, NULL);
 979         if (FAILED(hr))
 980         {
 981             HeapFree(GetProcessHeap(), 0, item->value.blob.pBlobData);
 982             return hr;
 983         }
 984         hr = IStream_Read(input, item->value.blob.pBlobData, count, &bytesread);
 985         if (bytesread != count) hr = E_FAIL;
 986         if (hr != S_OK)
 987         {
 988             HeapFree(GetProcessHeap(), 0, item->value.blob.pBlobData);
 989             return hr;
 990         }
 991         break;
 992     default:
 993         FIXME("loading field of type %d, count %lu is not implemented\n", type, count);
 994         break;
 995     }
 996     return S_OK;
 997 }
 998
 999 static HRESULT LoadIfdMetadata(IStream *input, const GUID *preferred_vendor,
1000     DWORD persist_options, MetadataItem **items, DWORD *item_count)
1001 {
1002     HRESULT hr;
1003     MetadataItem *result;
1004     USHORT count, i;
1005     struct IFD_entry *entry;
1006     BOOL native_byte_order = TRUE;
1007     ULONG bytesread;
1008
1009     TRACE("\n");
1010
1011 #ifdef WORDS_BIGENDIAN
1012     if (persist_options & WICPersistOptionLittleEndian)
1013 #else
1014     if (persist_options & WICPersistOptionBigEndian)
1015 #endif
1016         native_byte_order = FALSE;
1017
1018     hr = IStream_Read(input, &count, sizeof(count), &bytesread);
1019     if (bytesread != sizeof(count)) hr = E_FAIL;
1020     if (hr != S_OK) return hr;
1021
1022     SWAP_USHORT(count);
1023
1024     entry = HeapAlloc(GetProcessHeap(), 0, count * sizeof(*entry));
1025     if (!entry) return E_OUTOFMEMORY;
1026
1027     hr = IStream_Read(input, entry, count * sizeof(*entry), &bytesread);
1028     if (bytesread != count * sizeof(*entry)) hr = E_FAIL;
1029     if (hr != S_OK)
1030     {
1031         HeapFree(GetProcessHeap(), 0, entry);
1032         return hr;
1033     }
1034
1035     /* limit number of IFDs to 4096 to avoid infinite loop */
1036     for (i = 0; i < 4096; i++)
1037     {
1038         ULONG next_ifd_offset;
1039         LARGE_INTEGER pos;
1040         USHORT next_ifd_count;
1041
1042         hr = IStream_Read(input, &next_ifd_offset, sizeof(next_ifd_offset), &bytesread);
1043         if (bytesread != sizeof(next_ifd_offset)) hr = E_FAIL;
1044         if (hr != S_OK) break;
1045
1046         SWAP_ULONG(next_ifd_offset);
1047         if (!next_ifd_offset) break;
1048
1049         pos.QuadPart = next_ifd_offset;
1050         hr = IStream_Seek(input, pos, SEEK_SET, NULL);
1051         if (FAILED(hr)) break;
1052
1053         hr = IStream_Read(input, &next_ifd_count, sizeof(next_ifd_count), &bytesread);
1054         if (bytesread != sizeof(next_ifd_count)) hr = E_FAIL;
1055         if (hr != S_OK) break;
1056
1057         SWAP_USHORT(next_ifd_count);
1058
1059         pos.QuadPart = next_ifd_count * sizeof(*entry);
1060         hr = IStream_Seek(input, pos, SEEK_CUR, NULL);
1061         if (FAILED(hr)) break;
1062     }
1063
1064     if (hr != S_OK || i == 4096)
1065     {
1066         HeapFree(GetProcessHeap(), 0, entry);
1067         return WINCODEC_ERR_BADMETADATAHEADER;
1068     }
1069
1070     result = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, count * sizeof(*result));
1071     if (!result)
1072     {
1073         HeapFree(GetProcessHeap(), 0, entry);
1074         return E_OUTOFMEMORY;
1075     }
1076
1077     for (i = 0; i < count; i++)
1078     {
1079         hr = load_IFD_entry(input, &entry[i], &result[i], native_byte_order);
1080         if (FAILED(hr))
1081         {
1082             HeapFree(GetProcessHeap(), 0, entry);
1083             HeapFree(GetProcessHeap(), 0, result);
1084             return hr;
1085         }
1086     }
1087
1088     HeapFree(GetProcessHeap(), 0, entry);
1089
1090     *items = result;
1091     *item_count = count;
1092
1093     return S_OK;
1094 }
1095
1096 static const MetadataHandlerVtbl IfdMetadataReader_Vtbl = {
1097     0,
1098     &CLSID_WICIfdMetadataReader,
1099     LoadIfdMetadata
1100 };
1101
1102 HRESULT IfdMetadataReader_CreateInstance(REFIID iid, void **ppv)
1103 {
1104     return MetadataReader_Create(&IfdMetadataReader_Vtbl, iid, ppv);
1105 }