dlls/windowscodecs/tests/metadata.c

   1 /*
   2  * Copyright 2011 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 <stdio.h>
  21 #include <stdarg.h>
  22 #include <math.h>
  23
  24 #define COBJMACROS
  25
  26 #include "windef.h"
  27 #include "objbase.h"
  28 #include "wincodec.h"
  29 #include "wincodecsdk.h"
  30 #include "wine/test.h"
  31
  32 #define expect_blob(propvar, data, length) do { \
  33     ok((propvar).vt == VT_BLOB, "unexpected vt: %i\n", (propvar).vt); \
  34     if ((propvar).vt == VT_BLOB) { \
  35         ok(U(propvar).blob.cbSize == (length), "expected size %u, got %u\n", (ULONG)(length), U(propvar).blob.cbSize); \
  36         if (U(propvar).blob.cbSize == (length)) { \
  37             ok(!memcmp(U(propvar).blob.pBlobData, (data), (length)), "unexpected data\n"); \
  38         } \
  39     } \
  40 } while (0)
  41
  42 #define IFD_SHORT 3
  43 #define IFD_LONG 4
  44 #define IFD_RATIONAL 5
  45
  46 #include "pshpack2.h"
  47 struct IFD_entry
  48 {
  49     SHORT id;
  50     SHORT type;
  51     ULONG length;
  52     LONG  value;
  53 };
  54
  55 struct IFD_rational
  56 {
  57     LONG numerator;
  58     LONG denominator;
  59 };
  60
  61 static const struct
  62 {
  63     USHORT number_of_entries;
  64     struct IFD_entry entry[6];
  65     ULONG next_IFD;
  66     struct IFD_rational xres;
  67 } IFD_data =
  68 {
  69     6,
  70     {
  71         { 0xfe,  IFD_SHORT, 1, 1 }, /* NEWSUBFILETYPE */
  72         { 0x100, IFD_LONG, 1, 222 }, /* IMAGEWIDTH */
  73         { 0x101, IFD_LONG, 1, 333 }, /* IMAGELENGTH */
  74         { 0x102, IFD_SHORT, 1, 24 }, /* BITSPERSAMPLE */
  75         { 0x103, IFD_LONG, 1, 32773 }, /* COMPRESSION: packbits */
  76         { 0x11a, IFD_RATIONAL, 1, /* XRESOLUTION */
  77           sizeof(USHORT) + sizeof(struct IFD_entry) * 6 + sizeof(ULONG) }
  78     },
  79     0,
  80     { 900, 3 }
  81 };
  82 #include "poppack.h"
  83
  84 static const char metadata_unknown[] = "lalala";
  85
  86 static const char metadata_tEXt[] = {
  87     0,0,0,14, /* chunk length */
  88     't','E','X','t', /* chunk type */
  89     'w','i','n','e','t','e','s','t',0, /* keyword */
  90     'v','a','l','u','e', /* text */
  91     0x3f,0x64,0x19,0xf3 /* chunk CRC */
  92 };
  93
  94 static const char *debugstr_guid(REFIID riid)
  95 {
  96     static char buf[50];
  97
  98     if(!riid)
  99         return "(null)";
 100
 101     sprintf(buf, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
 102             riid->Data1, riid->Data2, riid->Data3, riid->Data4[0],
 103             riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4],
 104             riid->Data4[5], riid->Data4[6], riid->Data4[7]);
 105
 106     return buf;
 107 }
 108
 109 static IStream *create_stream(const char *data, int data_size)
 110 {
 111     HRESULT hr;
 112     IStream *stream;
 113     HGLOBAL hdata;
 114     void *locked_data;
 115
 116     hdata = GlobalAlloc(GMEM_MOVEABLE, data_size);
 117     ok(hdata != 0, "GlobalAlloc failed\n");
 118     if (!hdata) return NULL;
 119
 120     locked_data = GlobalLock(hdata);
 121     memcpy(locked_data, data, data_size);
 122     GlobalUnlock(hdata);
 123
 124     hr = CreateStreamOnHGlobal(hdata, TRUE, &stream);
 125     ok(hr == S_OK, "CreateStreamOnHGlobal failed, hr=%x\n", hr);
 126
 127     return stream;
 128 }
 129
 130 static void load_stream(IUnknown *reader, const char *data, int data_size)
 131 {
 132     HRESULT hr;
 133     IWICPersistStream *persist;
 134     IStream *stream;
 135
 136     stream = create_stream(data, data_size);
 137     if (!stream)
 138         return;
 139
 140     hr = IUnknown_QueryInterface(reader, &IID_IWICPersistStream, (void**)&persist);
 141     ok(hr == S_OK, "QueryInterface failed, hr=%x\n", hr);
 142
 143     if (SUCCEEDED(hr))
 144     {
 145         hr = IWICPersistStream_LoadEx(persist, stream, NULL, WICPersistOptionsDefault);
 146         ok(hr == S_OK, "LoadEx failed, hr=%x\n", hr);
 147
 148         IWICPersistStream_Release(persist);
 149     }
 150
 151     IStream_Release(stream);
 152 }
 153
 154 static void test_metadata_unknown(void)
 155 {
 156     HRESULT hr;
 157     IWICMetadataReader *reader;
 158     IWICEnumMetadataItem *enumerator;
 159     PROPVARIANT schema, id, value;
 160     ULONG items_returned;
 161
 162     hr = CoCreateInstance(&CLSID_WICUnknownMetadataReader, NULL, CLSCTX_INPROC_SERVER,
 163         &IID_IWICMetadataReader, (void**)&reader);
 164     ok(hr == S_OK, "CoCreateInstance failed, hr=%x\n", hr);
 165     if (FAILED(hr)) return;
 166
 167     load_stream((IUnknown*)reader, metadata_unknown, sizeof(metadata_unknown));
 168
 169     hr = IWICMetadataReader_GetEnumerator(reader, &enumerator);
 170     ok(hr == S_OK, "GetEnumerator failed, hr=%x\n", hr);
 171
 172     if (SUCCEEDED(hr))
 173     {
 174         PropVariantInit(&schema);
 175         PropVariantInit(&id);
 176         PropVariantInit(&value);
 177
 178         hr = IWICEnumMetadataItem_Next(enumerator, 1, &schema, &id, &value, &items_returned);
 179         ok(hr == S_OK, "Next failed, hr=%x\n", hr);
 180         ok(items_returned == 1, "unexpected item count %i\n", items_returned);
 181
 182         if (hr == S_OK && items_returned == 1)
 183         {
 184             ok(schema.vt == VT_EMPTY, "unexpected vt: %i\n", schema.vt);
 185             ok(id.vt == VT_EMPTY, "unexpected vt: %i\n", id.vt);
 186             expect_blob(value, metadata_unknown, sizeof(metadata_unknown));
 187
 188             PropVariantClear(&schema);
 189             PropVariantClear(&id);
 190             PropVariantClear(&value);
 191         }
 192
 193         hr = IWICEnumMetadataItem_Next(enumerator, 1, &schema, &id, &value, &items_returned);
 194         ok(hr == S_FALSE, "Next failed, hr=%x\n", hr);
 195         ok(items_returned == 0, "unexpected item count %i\n", items_returned);
 196
 197         IWICEnumMetadataItem_Release(enumerator);
 198     }
 199
 200     IWICMetadataReader_Release(reader);
 201 }
 202
 203 static void test_metadata_tEXt(void)
 204 {
 205     HRESULT hr;
 206     IWICMetadataReader *reader;
 207     IWICEnumMetadataItem *enumerator;
 208     PROPVARIANT schema, id, value;
 209     ULONG items_returned, count;
 210     GUID format;
 211
 212     PropVariantInit(&schema);
 213     PropVariantInit(&id);
 214     PropVariantInit(&value);
 215
 216     hr = CoCreateInstance(&CLSID_WICPngTextMetadataReader, NULL, CLSCTX_INPROC_SERVER,
 217         &IID_IWICMetadataReader, (void**)&reader);
 218     todo_wine ok(hr == S_OK, "CoCreateInstance failed, hr=%x\n", hr);
 219     if (FAILED(hr)) return;
 220
 221     hr = IWICMetadataReader_GetCount(reader, NULL);
 222     ok(hr == E_INVALIDARG, "GetCount failed, hr=%x\n", hr);
 223
 224     hr = IWICMetadataReader_GetCount(reader, &count);
 225     ok(hr == S_OK, "GetCount failed, hr=%x\n", hr);
 226     ok(count == 0, "unexpected count %i\n", count);
 227
 228     load_stream((IUnknown*)reader, metadata_tEXt, sizeof(metadata_tEXt));
 229
 230     hr = IWICMetadataReader_GetCount(reader, &count);
 231     ok(hr == S_OK, "GetCount failed, hr=%x\n", hr);
 232     ok(count == 1, "unexpected count %i\n", count);
 233
 234     hr = IWICMetadataReader_GetEnumerator(reader, NULL);
 235     ok(hr == E_INVALIDARG, "GetEnumerator failed, hr=%x\n", hr);
 236
 237     hr = IWICMetadataReader_GetEnumerator(reader, &enumerator);
 238     ok(hr == S_OK, "GetEnumerator failed, hr=%x\n", hr);
 239
 240     if (SUCCEEDED(hr))
 241     {
 242         hr = IWICEnumMetadataItem_Next(enumerator, 1, &schema, &id, &value, &items_returned);
 243         ok(hr == S_OK, "Next failed, hr=%x\n", hr);
 244         ok(items_returned == 1, "unexpected item count %i\n", items_returned);
 245
 246         if (hr == S_OK && items_returned == 1)
 247         {
 248             ok(schema.vt == VT_EMPTY, "unexpected vt: %i\n", schema.vt);
 249             ok(id.vt == VT_LPSTR, "unexpected vt: %i\n", id.vt);
 250             ok(!strcmp(U(id).pszVal, "winetest"), "unexpected id: %s\n", U(id).pszVal);
 251             ok(value.vt == VT_LPSTR, "unexpected vt: %i\n", value.vt);
 252             ok(!strcmp(U(value).pszVal, "value"), "unexpected value: %s\n", U(value).pszVal);
 253
 254             PropVariantClear(&schema);
 255             PropVariantClear(&id);
 256             PropVariantClear(&value);
 257         }
 258
 259         hr = IWICEnumMetadataItem_Next(enumerator, 1, &schema, &id, &value, &items_returned);
 260         ok(hr == S_FALSE, "Next failed, hr=%x\n", hr);
 261         ok(items_returned == 0, "unexpected item count %i\n", items_returned);
 262
 263         IWICEnumMetadataItem_Release(enumerator);
 264     }
 265
 266     hr = IWICMetadataReader_GetMetadataFormat(reader, &format);
 267     ok(hr == S_OK, "GetMetadataFormat failed, hr=%x\n", hr);
 268     ok(IsEqualGUID(&format, &GUID_MetadataFormatChunktEXt), "unexpected format %s\n", debugstr_guid(&format));
 269
 270     hr = IWICMetadataReader_GetMetadataFormat(reader, NULL);
 271     ok(hr == E_INVALIDARG, "GetMetadataFormat failed, hr=%x\n", hr);
 272
 273     id.vt = VT_LPSTR;
 274     U(id).pszVal = CoTaskMemAlloc(strlen("winetest") + 1);
 275     strcpy(U(id).pszVal, "winetest");
 276
 277     hr = IWICMetadataReader_GetValue(reader, NULL, &id, NULL);
 278     ok(hr == S_OK, "GetValue failed, hr=%x\n", hr);
 279
 280     hr = IWICMetadataReader_GetValue(reader, &schema, NULL, &value);
 281     ok(hr == E_INVALIDARG, "GetValue failed, hr=%x\n", hr);
 282
 283     hr = IWICMetadataReader_GetValue(reader, &schema, &id, &value);
 284     ok(hr == S_OK, "GetValue failed, hr=%x\n", hr);
 285     ok(value.vt == VT_LPSTR, "unexpected vt: %i\n", id.vt);
 286     ok(!strcmp(U(value).pszVal, "value"), "unexpected value: %s\n", U(value).pszVal);
 287     PropVariantClear(&value);
 288
 289     strcpy(U(id).pszVal, "test");
 290
 291     hr = IWICMetadataReader_GetValue(reader, &schema, &id, &value);
 292     ok(hr == WINCODEC_ERR_PROPERTYNOTFOUND, "GetValue failed, hr=%x\n", hr);
 293
 294     PropVariantClear(&id);
 295
 296     hr = IWICMetadataReader_GetValueByIndex(reader, 0, NULL, NULL, NULL);
 297     ok(hr == S_OK, "GetValueByIndex failed, hr=%x\n", hr);
 298
 299     hr = IWICMetadataReader_GetValueByIndex(reader, 0, &schema, NULL, NULL);
 300     ok(hr == S_OK, "GetValueByIndex failed, hr=%x\n", hr);
 301     ok(schema.vt == VT_EMPTY, "unexpected vt: %i\n", schema.vt);
 302
 303     hr = IWICMetadataReader_GetValueByIndex(reader, 0, NULL, &id, NULL);
 304     ok(hr == S_OK, "GetValueByIndex failed, hr=%x\n", hr);
 305     ok(id.vt == VT_LPSTR, "unexpected vt: %i\n", id.vt);
 306     ok(!strcmp(U(id).pszVal, "winetest"), "unexpected id: %s\n", U(id).pszVal);
 307     PropVariantClear(&id);
 308
 309     hr = IWICMetadataReader_GetValueByIndex(reader, 0, NULL, NULL, &value);
 310     ok(hr == S_OK, "GetValueByIndex failed, hr=%x\n", hr);
 311     ok(value.vt == VT_LPSTR, "unexpected vt: %i\n", value.vt);
 312     ok(!strcmp(U(value).pszVal, "value"), "unexpected value: %s\n", U(value).pszVal);
 313     PropVariantClear(&value);
 314
 315     hr = IWICMetadataReader_GetValueByIndex(reader, 1, NULL, NULL, NULL);
 316     ok(hr == E_INVALIDARG, "GetValueByIndex failed, hr=%x\n", hr);
 317
 318     IWICMetadataReader_Release(reader);
 319 }
 320
 321 static void test_metadata_IFD(void)
 322 {
 323     static const struct test_data
 324     {
 325         ULONG type, id;
 326         LONGLONG value;
 327     } td[6] =
 328     {
 329         { VT_UI2, 0xfe, 1 },
 330         { VT_UI4, 0x100, 222 },
 331         { VT_UI4, 0x101, 333 },
 332         { VT_UI2, 0x102, 24 },
 333         { VT_UI4, 0x103, 32773 },
 334         { VT_UI8, 0x11a,  ((LONGLONG)3 << 32) | 900 }
 335     };
 336     HRESULT hr;
 337     IWICMetadataReader *reader;
 338     IWICEnumMetadataItem *enumerator;
 339     PROPVARIANT schema, id, value;
 340     ULONG items_returned, count, i;
 341     GUID format;
 342
 343     PropVariantInit(&schema);
 344     PropVariantInit(&id);
 345     PropVariantInit(&value);
 346
 347     hr = CoCreateInstance(&CLSID_WICIfdMetadataReader, NULL, CLSCTX_INPROC_SERVER,
 348         &IID_IWICMetadataReader, (void**)&reader);
 349     todo_wine ok(hr == S_OK, "CoCreateInstance error %#x\n", hr);
 350     if (FAILED(hr)) return;
 351
 352     hr = IWICMetadataReader_GetCount(reader, NULL);
 353     ok(hr == E_INVALIDARG, "GetCount error %#x\n", hr);
 354
 355     hr = IWICMetadataReader_GetCount(reader, &count);
 356     ok(hr == S_OK, "GetCount error %#x\n", hr);
 357     ok(count == 0, "unexpected count %u\n", count);
 358
 359     load_stream((IUnknown*)reader, (const char *)&IFD_data, sizeof(IFD_data));
 360
 361     hr = IWICMetadataReader_GetCount(reader, &count);
 362     ok(hr == S_OK, "GetCount error %#x\n", hr);
 363     ok(count == 6, "unexpected count %u\n", count);
 364
 365     hr = IWICMetadataReader_GetEnumerator(reader, NULL);
 366     ok(hr == E_INVALIDARG, "GetEnumerator error %#x\n", hr);
 367
 368     hr = IWICMetadataReader_GetEnumerator(reader, &enumerator);
 369     ok(hr == S_OK, "GetEnumerator error %#x\n", hr);
 370
 371     for (i = 0; i < count; i++)
 372     {
 373         hr = IWICEnumMetadataItem_Next(enumerator, 1, &schema, &id, &value, &items_returned);
 374         ok(hr == S_OK, "Next error %#x\n", hr);
 375         ok(items_returned == 1, "unexpected item count %u\n", items_returned);
 376
 377         ok(schema.vt == VT_EMPTY, "%u: unexpected vt: %u\n", i, schema.vt);
 378         ok(id.vt == VT_UI2, "%u: unexpected vt: %u\n", i, id.vt);
 379         ok(U(id).uiVal == td[i].id, "%u: unexpected id: %#x\n", i, U(id).uiVal);
 380         ok(value.vt == td[i].type, "%u: unexpected vt: %u\n", i, value.vt);
 381         ok(U(value).uhVal.QuadPart == td[i].value, "%u: unexpected id: %d/%d\n", i, U(value).uhVal.LowPart, U(value).uhVal.HighPart);
 382
 383         PropVariantClear(&schema);
 384         PropVariantClear(&id);
 385         PropVariantClear(&value);
 386     }
 387
 388     hr = IWICEnumMetadataItem_Next(enumerator, 1, &schema, &id, &value, &items_returned);
 389     ok(hr == S_FALSE, "Next should fail\n");
 390     ok(items_returned == 0, "unexpected item count %u\n", items_returned);
 391
 392     IWICEnumMetadataItem_Release(enumerator);
 393
 394     hr = IWICMetadataReader_GetMetadataFormat(reader, &format);
 395     ok(hr == S_OK, "GetMetadataFormat error %#x\n", hr);
 396     ok(IsEqualGUID(&format, &GUID_MetadataFormatIfd), "unexpected format %s\n", debugstr_guid(&format));
 397
 398     hr = IWICMetadataReader_GetMetadataFormat(reader, NULL);
 399     ok(hr == E_INVALIDARG, "GetMetadataFormat should fail\n");
 400
 401     hr = IWICMetadataReader_GetValueByIndex(reader, 0, NULL, NULL, NULL);
 402     ok(hr == S_OK, "GetValueByIndex error %#x\n", hr);
 403
 404     hr = IWICMetadataReader_GetValueByIndex(reader, count - 1, NULL, NULL, NULL);
 405     ok(hr == S_OK, "GetValueByIndex error %#x\n", hr);
 406
 407     hr = IWICMetadataReader_GetValueByIndex(reader, 0, &schema, NULL, NULL);
 408     ok(hr == S_OK, "GetValueByIndex error %#x\n", hr);
 409     ok(schema.vt == VT_EMPTY, "unexpected vt: %u\n", schema.vt);
 410
 411     hr = IWICMetadataReader_GetValueByIndex(reader, count - 1, &schema, NULL, NULL);
 412     ok(hr == S_OK, "GetValueByIndex error %#x\n", hr);
 413     ok(schema.vt == VT_EMPTY, "unexpected vt: %u\n", schema.vt);
 414
 415     hr = IWICMetadataReader_GetValueByIndex(reader, 0, NULL, &id, NULL);
 416     ok(hr == S_OK, "GetValueByIndex error %#x\n", hr);
 417     ok(id.vt == VT_UI2, "unexpected vt: %u\n", id.vt);
 418     ok(U(id).uiVal == 0xfe, "unexpected id: %#x\n", U(id).uiVal);
 419     PropVariantClear(&id);
 420
 421     hr = IWICMetadataReader_GetValueByIndex(reader, 0, NULL, NULL, &value);
 422     ok(hr == S_OK, "GetValueByIndex error %#x\n", hr);
 423     ok(value.vt == VT_UI2, "unexpected vt: %u\n", value.vt);
 424     ok(U(value).ulVal == 1, "unexpected id: %u\n", U(value).ulVal);
 425     PropVariantClear(&value);
 426
 427     hr = IWICMetadataReader_GetValueByIndex(reader, count, &schema, NULL, NULL);
 428     ok(hr == E_INVALIDARG, "GetValueByIndex should fail\n");
 429
 430     IWICMetadataReader_Release(reader);
 431 }
 432
 433 static void test_create_reader(void)
 434 {
 435     HRESULT hr;
 436     IWICComponentFactory *factory;
 437     IStream *stream;
 438     IWICMetadataReader *reader;
 439     UINT count=0;
 440     GUID format;
 441
 442     hr = CoCreateInstance(&CLSID_WICImagingFactory, NULL, CLSCTX_INPROC_SERVER,
 443         &IID_IWICComponentFactory, (void**)&factory);
 444     ok(hr == S_OK, "CoCreateInstance failed, hr=%x\n", hr);
 445
 446     stream = create_stream(metadata_tEXt, sizeof(metadata_tEXt));
 447
 448     hr = IWICComponentFactory_CreateMetadataReaderFromContainer(factory,
 449         &GUID_ContainerFormatPng, NULL, WICPersistOptionsDefault,
 450         stream, &reader);
 451 todo_wine
 452     ok(hr == S_OK, "CreateMetadataReaderFromContainer failed, hr=%x\n", hr);
 453     if (FAILED(hr)) return;
 454
 455     if (SUCCEEDED(hr))
 456     {
 457         hr = IWICMetadataReader_GetCount(reader, &count);
 458         ok(hr == S_OK, "GetCount failed, hr=%x\n", hr);
 459         ok(count == 1, "unexpected count %i\n", count);
 460
 461         hr = IWICMetadataReader_GetMetadataFormat(reader, &format);
 462         ok(hr == S_OK, "GetMetadataFormat failed, hr=%x\n", hr);
 463         ok(IsEqualGUID(&format, &GUID_MetadataFormatChunktEXt), "unexpected format %s\n", debugstr_guid(&format));
 464
 465         IWICMetadataReader_Release(reader);
 466     }
 467
 468     hr = IWICComponentFactory_CreateMetadataReaderFromContainer(factory,
 469         &GUID_ContainerFormatWmp, NULL, WICPersistOptionsDefault,
 470         stream, &reader);
 471     ok(hr == S_OK, "CreateMetadataReaderFromContainer failed, hr=%x\n", hr);
 472
 473     if (SUCCEEDED(hr))
 474     {
 475         hr = IWICMetadataReader_GetCount(reader, &count);
 476         ok(hr == S_OK, "GetCount failed, hr=%x\n", hr);
 477         ok(count == 1, "unexpected count %i\n", count);
 478
 479         hr = IWICMetadataReader_GetMetadataFormat(reader, &format);
 480         ok(hr == S_OK, "GetMetadataFormat failed, hr=%x\n", hr);
 481         ok(IsEqualGUID(&format, &GUID_MetadataFormatUnknown), "unexpected format %s\n", debugstr_guid(&format));
 482
 483         IWICMetadataReader_Release(reader);
 484     }
 485
 486     IStream_Release(stream);
 487
 488     IWICComponentFactory_Release(factory);
 489 }
 490
 491 START_TEST(metadata)
 492 {
 493     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
 494
 495     test_metadata_unknown();
 496     test_metadata_tEXt();
 497     test_metadata_IFD();
 498     test_create_reader();
 499
 500     CoUninitialize();
 501 }