rostests/winetests/ole32/propvariant.c

   1 /*
   2  * PropVariant Tests
   3  *
   4  * Copyright 2004 Robert Shearman
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 //#include "windows.h"
  22
  23 #define WIN32_NO_STATUS
  24 #define _INC_WINDOWS
  25 #define COM_NO_WINDOWS_H
  26
  27 #include <stdarg.h>
  28
  29 #include <windef.h>
  30 #include <winbase.h>
  31 #include <winnls.h>
  32 #include <ddeml.h>
  33 #include <ole2.h>
  34
  35 #include <wine/test.h>
  36
  37 /* invalid in all versions */
  38 #define PROP_INV 0x7f
  39 /* valid in v0 and above (NT4+) */
  40 #define PROP_V0  0
  41 /* valid in v1 and above (Win2k+) */
  42 #define PROP_V1  1
  43 /* valid in v1a and above (WinXP+) */
  44 #define PROP_V1A 2
  45 #define PROP_TODO 0x80
  46
  47 static const struct valid_mapping
  48 {
  49     BYTE simple;
  50     BYTE with_array;
  51     BYTE with_vector;
  52     BYTE byref;
  53 } valid_types[] =
  54 {
  55     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_EMPTY */
  56     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_NULL */
  57     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_I2 */
  58     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_I4 */
  59     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_R4 */
  60     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_R8 */
  61     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_CY */
  62     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_DATE */
  63     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_BSTR */
  64     { PROP_V1 , PROP_V1 | PROP_TODO , PROP_INV, PROP_V1 | PROP_TODO  }, /* VT_DISPATCH */
  65     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_ERROR */
  66     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_BOOL */
  67     { PROP_V1 | PROP_TODO , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_VARIANT */
  68     { PROP_V1 , PROP_V1 | PROP_TODO , PROP_INV, PROP_V1 | PROP_TODO  }, /* VT_UNKNOWN */
  69     { PROP_V1 , PROP_V1 | PROP_TODO , PROP_INV, PROP_V1 | PROP_TODO  }, /* VT_DECIMAL */
  70     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 15 */
  71     { PROP_V1 , PROP_V1 | PROP_TODO , PROP_V1 , PROP_V1 | PROP_TODO  }, /* VT_I1 */
  72     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_UI1 */
  73     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_UI2 */
  74     { PROP_V0 , PROP_V1 | PROP_TODO , PROP_V0 , PROP_V1 | PROP_TODO  }, /* VT_UI4 */
  75     { PROP_V0 , PROP_V1A | PROP_TODO, PROP_V0 , PROP_V1A | PROP_TODO }, /* VT_I8 */
  76     { PROP_V0 , PROP_V1A | PROP_TODO, PROP_V0 , PROP_V1A | PROP_TODO }, /* VT_UI8 */
  77     { PROP_V1 , PROP_V1 | PROP_TODO , PROP_INV, PROP_V1 | PROP_TODO  }, /* VT_INT */
  78     { PROP_V1 , PROP_V1 | PROP_TODO , PROP_INV, PROP_V1 | PROP_TODO  }, /* VT_UINT */
  79     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_VOID */
  80     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_HRESULT */
  81     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_PTR */
  82     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_SAFEARRAY */
  83     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_CARRAY */
  84     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_USERDEFINED */
  85     { PROP_V0 , PROP_INV, PROP_V0 , PROP_INV }, /* VT_LPSTR */
  86     { PROP_V0 , PROP_INV, PROP_V0 , PROP_INV }, /* VT_LPWSTR */
  87     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 32 */
  88     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 33 */
  89     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 34 */
  90     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 35 */
  91     { PROP_V1 | PROP_TODO , PROP_V1 | PROP_TODO , PROP_INV, PROP_V1 | PROP_TODO  }, /* VT_RECORD */
  92     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_INT_PTR */
  93     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* VT_UINT_PTR */
  94     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 39 */
  95     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 40 */
  96     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 41 */
  97     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 42 */
  98     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 43 */
  99     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 44 */
 100     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 45 */
 101     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 46 */
 102     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 47 */
 103     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 48 */
 104     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 49 */
 105     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 50 */
 106     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 51 */
 107     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 52 */
 108     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 53 */
 109     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 54 */
 110     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 55 */
 111     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 56 */
 112     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 57 */
 113     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 58 */
 114     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 59 */
 115     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 60 */
 116     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 61 */
 117     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 62 */
 118     { PROP_INV, PROP_INV, PROP_INV, PROP_INV }, /* 63 */
 119     { PROP_V0 , PROP_INV, PROP_V0 , PROP_INV }, /* VT_FILETIME */
 120     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_BLOB */
 121     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_STREAM */
 122     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_STORAGE */
 123     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_STREAMED_OBJECT */
 124     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_STORED_OBJECT */
 125     { PROP_V0 , PROP_INV, PROP_INV, PROP_INV }, /* VT_BLOB_OBJECT */
 126     { PROP_V0 , PROP_INV, PROP_V0 , PROP_INV }  /* VT_CF */
 127 };
 128
 129 static const char* wine_vtypes[VT_CLSID+1] =
 130 {
 131   "VT_EMPTY","VT_NULL","VT_I2","VT_I4","VT_R4","VT_R8","VT_CY","VT_DATE",
 132   "VT_BSTR","VT_DISPATCH","VT_ERROR","VT_BOOL","VT_VARIANT","VT_UNKNOWN",
 133   "VT_DECIMAL","15","VT_I1","VT_UI1","VT_UI2","VT_UI4","VT_I8","VT_UI8",
 134   "VT_INT","VT_UINT","VT_VOID","VT_HRESULT","VT_PTR","VT_SAFEARRAY",
 135   "VT_CARRAY","VT_USERDEFINED","VT_LPSTR","VT_LPWSTR","32","33","34","35",
 136   "VT_RECORD","VT_INT_PTR","VT_UINT_PTR","39","40","41","42","43","44","45",
 137   "46","47","48","49","50","51","52","53","54","55","56","57","58","59","60",
 138   "61","62","63","VT_FILETIME","VT_BLOB","VT_STREAM","VT_STORAGE",
 139   "VT_STREAMED_OBJECT","VT_STORED_OBJECT","VT_BLOB_OBJECT","VT_CF","VT_CLSID"
 140 };
 141
 142
 143 static void expect(HRESULT hr, VARTYPE vt, BOOL copy, int line)
 144 {
 145     int idx = vt & VT_TYPEMASK;
 146     BYTE flags;
 147     const char *modifier;
 148
 149     if(vt & VT_BYREF)
 150     {
 151         flags = valid_types[idx].byref;
 152         modifier = "byref";
 153     }
 154     else if(vt & VT_ARRAY)
 155     {
 156         flags = valid_types[idx].with_array;
 157         modifier = "array";
 158     }
 159     else if(vt & VT_VECTOR)
 160     {
 161         flags = valid_types[idx].with_vector;
 162         modifier = "vector";
 163     }
 164     else
 165     {
 166         flags = valid_types[idx].simple;
 167         modifier = "simple";
 168     }
 169
 170     if(flags == PROP_INV)
 171     {
 172         if (copy && (vt & VT_VECTOR))
 173             ok(hr == DISP_E_BADVARTYPE || hr == STG_E_INVALIDPARAMETER, "%s (%s): got %08x (line %d)\n", wine_vtypes[idx], modifier, hr, line);
 174         else
 175             ok(hr == (copy ? DISP_E_BADVARTYPE : STG_E_INVALIDPARAMETER), "%s (%s): got %08x (line %d)\n", wine_vtypes[idx], modifier, hr, line);
 176     }
 177     else if(flags == PROP_V0)
 178         ok(hr == S_OK, "%s (%s): got %08x\n", wine_vtypes[idx], modifier, hr);
 179     else if(flags & PROP_TODO)
 180     {
 181         todo_wine
 182         {
 183         if(hr != S_OK)
 184             win_skip("%s (%s): unsupported\n", wine_vtypes[idx], modifier);
 185         else ok(hr == S_OK, "%s (%s): got %08x\n", wine_vtypes[idx], modifier, hr);
 186         }
 187     }
 188     else
 189     {
 190         if(hr != S_OK)
 191             win_skip("%s (%s): unsupported\n", wine_vtypes[idx], modifier);
 192         else ok(hr == S_OK, "%s (%s): got %08x\n", wine_vtypes[idx], modifier, hr);
 193     }
 194 }
 195
 196 static void test_validtypes(void)
 197 {
 198     PROPVARIANT propvar, copy, uninit;
 199     HRESULT hr;
 200     unsigned int i, ret;
 201
 202     memset(&uninit, 0x77, sizeof(uninit));
 203
 204     memset(&propvar, 0x55, sizeof(propvar));
 205     hr = PropVariantClear(&propvar);
 206     ok(hr == STG_E_INVALIDPARAMETER, "expected STG_E_INVALIDPARAMETER, got %08x\n", hr);
 207     ok(propvar.vt == 0, "expected 0, got %d\n", propvar.vt);
 208     ok(U(propvar).uhVal.QuadPart == 0, "expected 0, got %#x/%#x\n",
 209        U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart);
 210
 211     for (i = 0; i < sizeof(valid_types)/sizeof(valid_types[0]); i++)
 212     {
 213         VARTYPE vt;
 214
 215         memset(&propvar, 0x55, sizeof(propvar));
 216         if (i == VT_RECORD)
 217             memset(&propvar, 0, sizeof(propvar));
 218         else if (i == VT_BLOB || i == VT_BLOB_OBJECT)
 219         {
 220             U(propvar).blob.cbSize = 0;
 221             U(propvar).blob.pBlobData = NULL;
 222         }
 223         else
 224             U(propvar).pszVal = NULL;
 225         vt = propvar.vt = i;
 226         memset(&copy, 0x77, sizeof(copy));
 227         hr = PropVariantCopy(&copy, &propvar);
 228         expect(hr, vt, TRUE, __LINE__);
 229         if (hr == S_OK)
 230         {
 231             ok(copy.vt == propvar.vt, "expected %d, got %d\n", propvar.vt, copy.vt);
 232             ok(U(copy).uhVal.QuadPart == U(propvar).uhVal.QuadPart, "%u: expected %#x/%#x, got %#x/%#x\n",
 233                i, U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart,
 234                U(copy).uhVal.u.LowPart, U(copy).uhVal.u.HighPart);
 235         }
 236         else
 237         {
 238             ret = memcmp(&copy, &uninit, sizeof(copy));
 239             ok(!ret || broken(ret) /* win2000 */, "%d: copy should stay unchanged\n", i);
 240         }
 241         hr = PropVariantClear(&propvar);
 242         expect(hr, vt, FALSE, __LINE__);
 243         ok(propvar.vt == 0, "expected 0, got %d\n", propvar.vt);
 244         ok(U(propvar).uhVal.QuadPart == 0, "%u: expected 0, got %#x/%#x\n",
 245            i, U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart);
 246
 247         memset(&propvar, 0x55, sizeof(propvar));
 248         U(propvar).pszVal = NULL;
 249         vt = propvar.vt = i | VT_ARRAY;
 250         memset(&copy, 0x77, sizeof(copy));
 251         hr = PropVariantCopy(&copy, &propvar);
 252         expect(hr, vt, TRUE, __LINE__);
 253         if (hr == S_OK)
 254         {
 255             ok(copy.vt == propvar.vt, "expected %d, got %d\n", propvar.vt, copy.vt);
 256             ok(U(copy).uhVal.QuadPart == 0, "%u: expected 0, got %#x/%#x\n",
 257                i, U(copy).uhVal.u.LowPart, U(copy).uhVal.u.HighPart);
 258         }
 259         else
 260         {
 261             ret = memcmp(&copy, &uninit, sizeof(copy));
 262             ok(!ret || broken(ret) /* win2000 */, "%d: copy should stay unchanged\n", i);
 263         }
 264         hr = PropVariantClear(&propvar);
 265         expect(hr, vt, FALSE, __LINE__);
 266         ok(propvar.vt == 0, "expected 0, got %d\n", propvar.vt);
 267         ok(U(propvar).uhVal.QuadPart == 0, "%u: expected 0, got %#x/%#x\n",
 268            i, U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart);
 269
 270         memset(&propvar, 0x55, sizeof(propvar));
 271         U(propvar).caub.cElems = 0;
 272         U(propvar).caub.pElems = NULL;
 273         vt = propvar.vt = i | VT_VECTOR;
 274         memset(&copy, 0x77, sizeof(copy));
 275         hr = PropVariantCopy(&copy, &propvar);
 276         expect(hr, vt, TRUE, __LINE__);
 277         if (hr == S_OK)
 278         {
 279             ok(copy.vt == propvar.vt, "expected %d, got %d\n", propvar.vt, copy.vt);
 280             ok(!U(copy).caub.cElems, "%u: expected 0, got %d\n", i, U(copy).caub.cElems);
 281             ok(!U(copy).caub.pElems, "%u: expected NULL, got %p\n", i, U(copy).caub.pElems);
 282         }
 283         else
 284         {
 285             ret = memcmp(&copy, &uninit, sizeof(copy));
 286             ok(!ret || broken(ret) /* win2000 */, "%d: copy should stay unchanged\n", i);
 287         }
 288         hr = PropVariantClear(&propvar);
 289         expect(hr, vt, FALSE, __LINE__);
 290         ok(propvar.vt == 0, "expected 0, got %d\n", propvar.vt);
 291         ok(U(propvar).uhVal.QuadPart == 0, "%u: expected 0, got %#x/%#x\n",
 292            i, U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart);
 293
 294         memset(&propvar, 0x55, sizeof(propvar));
 295         U(propvar).pszVal = NULL;
 296         vt = propvar.vt = i | VT_BYREF;
 297         memset(&copy, 0x77, sizeof(copy));
 298         hr = PropVariantCopy(&copy, &propvar);
 299         expect(hr, vt, TRUE, __LINE__);
 300         if (hr == S_OK)
 301         {
 302             ok(copy.vt == propvar.vt, "expected %d, got %d\n", propvar.vt, copy.vt);
 303             ok(U(copy).uhVal.QuadPart == U(propvar).uhVal.QuadPart, "%u: expected %#x/%#x, got %#x/%#x\n",
 304                i, U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart,
 305                U(copy).uhVal.u.LowPart, U(copy).uhVal.u.HighPart);
 306         }
 307         else
 308         {
 309             ret = memcmp(&copy, &uninit, sizeof(copy));
 310             ok(!ret || broken(ret) /* win2000 */, "%d: copy should stay unchanged\n", i);
 311         }
 312         hr = PropVariantClear(&propvar);
 313         expect(hr, vt, FALSE, __LINE__);
 314         ok(propvar.vt == 0, "expected 0, got %d\n", propvar.vt);
 315         ok(U(propvar).uhVal.QuadPart == 0, "%u: expected 0, got %#x/%#x\n",
 316            i, U(propvar).uhVal.u.LowPart, U(propvar).uhVal.u.HighPart);
 317     }
 318 }
 319
 320 static void test_copy(void)
 321 {
 322     static char szTestString[] = "Test String";
 323     static WCHAR wszTestString[] = {'T','e','s','t',' ','S','t','r','i','n','g',0};
 324     PROPVARIANT propvarSrc;
 325     PROPVARIANT propvarDst;
 326     HRESULT hr;
 327
 328     propvarSrc.vt = VT_BSTR;
 329     U(propvarSrc).bstrVal = SysAllocString(wszTestString);
 330
 331     hr = PropVariantCopy(&propvarDst, &propvarSrc);
 332     ok(hr == S_OK, "PropVariantCopy(...VT_BSTR...) failed\n");
 333     ok(!lstrcmpW(U(propvarSrc).bstrVal, U(propvarDst).bstrVal), "BSTR not copied properly\n");
 334     hr = PropVariantClear(&propvarSrc);
 335     ok(hr == S_OK, "PropVariantClear(...VT_BSTR...) failed\n");
 336     hr = PropVariantClear(&propvarDst);
 337     ok(hr == S_OK, "PropVariantClear(...VT_BSTR...) failed\n");
 338
 339     propvarSrc.vt = VT_LPWSTR;
 340     U(propvarSrc).pwszVal = wszTestString;
 341     hr = PropVariantCopy(&propvarDst, &propvarSrc);
 342     ok(hr == S_OK, "PropVariantCopy(...VT_LPWSTR...) failed\n");
 343     ok(!lstrcmpW(U(propvarSrc).pwszVal, U(propvarDst).pwszVal), "Wide string not copied properly\n");
 344     hr = PropVariantClear(&propvarDst);
 345     ok(hr == S_OK, "PropVariantClear(...VT_LPWSTR...) failed\n");
 346     memset(&propvarSrc, 0, sizeof(propvarSrc));
 347
 348     propvarSrc.vt = VT_LPSTR;
 349     U(propvarSrc).pszVal = szTestString;
 350     hr = PropVariantCopy(&propvarDst, &propvarSrc);
 351     ok(hr == S_OK, "PropVariantCopy(...VT_LPSTR...) failed\n");
 352     ok(!strcmp(U(propvarSrc).pszVal, U(propvarDst).pszVal), "String not copied properly\n");
 353     hr = PropVariantClear(&propvarDst);
 354     ok(hr == S_OK, "PropVariantClear(...VT_LPSTR...) failed\n");
 355     memset(&propvarSrc, 0, sizeof(propvarSrc));
 356 }
 357
 358 struct _PMemoryAllocator_vtable {
 359     void *Allocate; /* virtual void* Allocate(ULONG cbSize); */
 360     void *Free; /* virtual void Free(void *pv); */
 361 };
 362
 363 typedef struct _PMemoryAllocator {
 364     struct _PMemoryAllocator_vtable *vt;
 365 } PMemoryAllocator;
 366
 367 static void * WINAPI PMemoryAllocator_Allocate(PMemoryAllocator *_this, ULONG cbSize)
 368 {
 369     return CoTaskMemAlloc(cbSize);
 370 }
 371
 372 static void WINAPI PMemoryAllocator_Free(PMemoryAllocator *_this, void *pv)
 373 {
 374     CoTaskMemFree(pv);
 375 }
 376
 377 #ifdef __i386__
 378
 379 #include "pshpack1.h"
 380 typedef struct
 381 {
 382     BYTE pop_eax;  /* popl  %eax  */
 383     BYTE push_ecx; /* pushl %ecx  */
 384     BYTE push_eax; /* pushl %eax  */
 385     BYTE jmp_func; /* jmp   $func */
 386     DWORD func;
 387 } THISCALL_TO_STDCALL_THUNK;
 388 #include "poppack.h"
 389
 390 static THISCALL_TO_STDCALL_THUNK *wrapperCodeMem = NULL;
 391
 392 static void fill_thunk(THISCALL_TO_STDCALL_THUNK *thunk, void *fn)
 393 {
 394     thunk->pop_eax = 0x58;
 395     thunk->push_ecx = 0x51;
 396     thunk->push_eax = 0x50;
 397     thunk->jmp_func = 0xe9;
 398     thunk->func = (char*)fn - (char*)(&thunk->func + 1);
 399 }
 400
 401 static void setup_vtable(struct _PMemoryAllocator_vtable *vtable)
 402 {
 403     wrapperCodeMem = VirtualAlloc(NULL, 2 * sizeof(*wrapperCodeMem),
 404                                   MEM_COMMIT, PAGE_EXECUTE_READWRITE);
 405
 406     fill_thunk(&wrapperCodeMem[0], PMemoryAllocator_Allocate);
 407     fill_thunk(&wrapperCodeMem[1], PMemoryAllocator_Free);
 408
 409     vtable->Allocate = &wrapperCodeMem[0];
 410     vtable->Free = &wrapperCodeMem[1];
 411 }
 412
 413 #else
 414
 415 static void setup_vtable(struct _PMemoryAllocator_vtable *vtable)
 416 {
 417     vtable->Allocate = PMemoryAllocator_Allocate;
 418     vtable->Free = PMemoryAllocator_Free;
 419 }
 420
 421 #endif
 422
 423 static const char serialized_empty[] = {
 424     0,0, /* VT_EMPTY */
 425     0,0, /* padding */
 426 };
 427
 428 static const char serialized_null[] = {
 429     1,0, /* VT_NULL */
 430     0,0, /* padding */
 431 };
 432
 433 static const char serialized_i4[] = {
 434     3,0, /* VT_I4 */
 435     0,0, /* padding */
 436     0xef,0xcd,0xab,0xfe
 437 };
 438
 439 static const char serialized_bstr_wc[] = {
 440     8,0, /* VT_BSTR */
 441     0,0, /* padding */
 442     10,0,0,0, /* size */
 443     't',0,'e',0,
 444     's',0,'t',0,
 445     0,0,0,0
 446 };
 447
 448 static const char serialized_bstr_mb[] = {
 449     8,0, /* VT_BSTR */
 450     0,0, /* padding */
 451     5,0,0,0, /* size */
 452     't','e','s','t',
 453     0,0,0,0
 454 };
 455
 456 static void test_propertytovariant(void)
 457 {
 458     HANDLE hole32;
 459     BOOLEAN (__stdcall *pStgConvertPropertyToVariant)(const SERIALIZEDPROPERTYVALUE*,USHORT,PROPVARIANT*,PMemoryAllocator*);
 460     PROPVARIANT propvar;
 461     PMemoryAllocator allocator;
 462     struct _PMemoryAllocator_vtable vtable;
 463     BOOLEAN ret;
 464     static const WCHAR test_string[] = {'t','e','s','t',0};
 465
 466     hole32 = GetModuleHandleA("ole32");
 467
 468     pStgConvertPropertyToVariant = (void*)GetProcAddress(hole32, "StgConvertPropertyToVariant");
 469
 470     if (!pStgConvertPropertyToVariant)
 471     {
 472         win_skip("StgConvertPropertyToVariant not available\n");
 473         return;
 474     }
 475
 476     setup_vtable(&vtable);
 477     allocator.vt = &vtable;
 478
 479     ret = pStgConvertPropertyToVariant((SERIALIZEDPROPERTYVALUE*)serialized_empty,
 480         CP_WINUNICODE, &propvar, &allocator);
 481
 482     ok(ret == 0, "StgConvertPropertyToVariant returned %i\n", ret);
 483     ok(propvar.vt == VT_EMPTY, "unexpected vt %x\n", propvar.vt);
 484
 485     ret = pStgConvertPropertyToVariant((SERIALIZEDPROPERTYVALUE*)serialized_null,
 486         CP_WINUNICODE, &propvar, &allocator);
 487
 488     ok(ret == 0, "StgConvertPropertyToVariant returned %i\n", ret);
 489     ok(propvar.vt == VT_NULL, "unexpected vt %x\n", propvar.vt);
 490
 491     ret = pStgConvertPropertyToVariant((SERIALIZEDPROPERTYVALUE*)serialized_i4,
 492         CP_WINUNICODE, &propvar, &allocator);
 493
 494     ok(ret == 0, "StgConvertPropertyToVariant returned %i\n", ret);
 495     ok(propvar.vt == VT_I4, "unexpected vt %x\n", propvar.vt);
 496     ok(U(propvar).lVal == 0xfeabcdef, "unexpected lVal %x\n", U(propvar).lVal);
 497
 498     ret = pStgConvertPropertyToVariant((SERIALIZEDPROPERTYVALUE*)serialized_bstr_wc,
 499         CP_WINUNICODE, &propvar, &allocator);
 500
 501     ok(ret == 0, "StgConvertPropertyToVariant returned %i\n", ret);
 502     ok(propvar.vt == VT_BSTR, "unexpected vt %x\n", propvar.vt);
 503     ok(!lstrcmpW(U(propvar).bstrVal, test_string), "unexpected string value\n");
 504     PropVariantClear(&propvar);
 505
 506     ret = pStgConvertPropertyToVariant((SERIALIZEDPROPERTYVALUE*)serialized_bstr_mb,
 507         CP_UTF8, &propvar, &allocator);
 508
 509     ok(ret == 0, "StgConvertPropertyToVariant returned %i\n", ret);
 510     ok(propvar.vt == VT_BSTR, "unexpected vt %x\n", propvar.vt);
 511     ok(!lstrcmpW(U(propvar).bstrVal, test_string), "unexpected string value\n");
 512     PropVariantClear(&propvar);
 513 }
 514
 515 static void test_varianttoproperty(void)
 516 {
 517     HANDLE hole32;
 518     PROPVARIANT propvar;
 519     SERIALIZEDPROPERTYVALUE *propvalue, *own_propvalue;
 520     SERIALIZEDPROPERTYVALUE* (__stdcall *pStgConvertVariantToProperty)(
 521         const PROPVARIANT*,USHORT,SERIALIZEDPROPERTYVALUE*,ULONG*,PROPID,BOOLEAN,ULONG*);
 522     ULONG len;
 523     static const WCHAR test_string[] = {'t','e','s','t',0};
 524     BSTR test_string_bstr;
 525
 526     hole32 = GetModuleHandleA("ole32");
 527
 528     pStgConvertVariantToProperty = (void*)GetProcAddress(hole32, "StgConvertVariantToProperty");
 529
 530     if (!pStgConvertVariantToProperty)
 531     {
 532         win_skip("StgConvertVariantToProperty not available\n");
 533         return;
 534     }
 535
 536     own_propvalue = HeapAlloc(GetProcessHeap(), 0, sizeof(SERIALIZEDPROPERTYVALUE) + 20);
 537
 538     PropVariantInit(&propvar);
 539
 540     propvar.vt = VT_I4;
 541     U(propvar).lVal = 0xfeabcdef;
 542
 543     len = 0xdeadbeef;
 544     propvalue = pStgConvertVariantToProperty(&propvar, CP_WINUNICODE, NULL, &len,
 545         0, FALSE, 0);
 546
 547     ok(propvalue == NULL, "got nonnull propvalue\n");
 548     todo_wine ok(len == 8, "unexpected length %d\n", len);
 549
 550     if (len == 0xdeadbeef)
 551     {
 552         HeapFree(GetProcessHeap(), 0, own_propvalue);
 553         return;
 554     }
 555
 556     len = 20;
 557     propvalue = pStgConvertVariantToProperty(&propvar, CP_WINUNICODE, own_propvalue, &len,
 558         0, FALSE, 0);
 559
 560     ok(propvalue == own_propvalue, "unexpected propvalue %p\n", propvalue);
 561     ok(len == 8, "unexpected length %d\n", len);
 562     ok(!memcmp(propvalue, serialized_i4, 8), "got wrong data\n");
 563
 564     propvar.vt = VT_EMPTY;
 565     len = 20;
 566     own_propvalue->dwType = 0xdeadbeef;
 567     propvalue = pStgConvertVariantToProperty(&propvar, CP_WINUNICODE, own_propvalue, &len,
 568         0, FALSE, 0);
 569
 570     ok(propvalue == own_propvalue, "unexpected propvalue %p\n", propvalue);
 571     ok(len == 4 || broken(len == 0) /* before Vista */, "unexpected length %d\n", len);
 572     if (len) ok(!memcmp(propvalue, serialized_empty, 4), "got wrong data\n");
 573     else ok(propvalue->dwType == 0xdeadbeef, "unexpected type %d\n", propvalue->dwType);
 574
 575     propvar.vt = VT_NULL;
 576     len = 20;
 577     propvalue = pStgConvertVariantToProperty(&propvar, CP_WINUNICODE, own_propvalue, &len,
 578         0, FALSE, 0);
 579
 580     ok(propvalue == own_propvalue, "unexpected propvalue %p\n", propvalue);
 581     ok(len == 4, "unexpected length %d\n", len);
 582     ok(!memcmp(propvalue, serialized_null, 4), "got wrong data\n");
 583
 584     test_string_bstr = SysAllocString(test_string);
 585
 586     propvar.vt = VT_BSTR;
 587     U(propvar).bstrVal = test_string_bstr;
 588     len = 20;
 589     propvalue = pStgConvertVariantToProperty(&propvar, CP_WINUNICODE, own_propvalue, &len,
 590         0, FALSE, 0);
 591
 592     ok(propvalue == own_propvalue, "unexpected propvalue %p\n", propvalue);
 593     ok(len == 20, "unexpected length %d\n", len);
 594     ok(!memcmp(propvalue, serialized_bstr_wc, 20), "got wrong data\n");
 595
 596     len = 20;
 597     propvalue = pStgConvertVariantToProperty(&propvar, CP_UTF8, own_propvalue, &len,
 598         0, FALSE, 0);
 599
 600     ok(propvalue == own_propvalue, "unexpected propvalue %p\n", propvalue);
 601     ok(len == 16, "unexpected length %d\n", len);
 602     ok(!memcmp(propvalue, serialized_bstr_mb, 16), "got wrong data\n");
 603
 604     SysFreeString(test_string_bstr);
 605
 606     HeapFree(GetProcessHeap(), 0, own_propvalue);
 607 }
 608
 609 START_TEST(propvariant)
 610 {
 611     test_validtypes();
 612     test_copy();
 613     test_propertytovariant();
 614     test_varianttoproperty();
 615 }