Rob Shearman : oleaut32:
Add some tests for user marshaling a VT_UNKNOWN variant.
Alexandre Julliard
julliard at wine.codeweavers.com
Tue Mar 20 08:11:01 CDT 2007
Module: wine
Branch: master
Commit: 35772a70d3f414c19f1ea92f9f145b7c35f5009a
URL: http://source.winehq.org/git/wine.git/?a=commit;h=35772a70d3f414c19f1ea92f9f145b7c35f5009a
Author: Rob Shearman <rob at codeweavers.com>
Date: Mon Mar 19 14:54:57 2007 +0000
oleaut32: Add some tests for user marshaling a VT_UNKNOWN variant.
---
dlls/oleaut32/tests/usrmarshal.c | 90 ++++++++++++++++++++++++++++++++++++++
1 files changed, 90 insertions(+), 0 deletions(-)
diff --git a/dlls/oleaut32/tests/usrmarshal.c b/dlls/oleaut32/tests/usrmarshal.c
index 9bc282d..447b30e 100644
--- a/dlls/oleaut32/tests/usrmarshal.c
+++ b/dlls/oleaut32/tests/usrmarshal.c
@@ -18,6 +18,8 @@
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
+#define COBJMACROS
+
#include <stdarg.h>
#include "windef.h"
@@ -404,6 +406,45 @@ static void test_marshal_BSTR(void)
SysFreeString(b);
}
+typedef struct
+{
+ const IUnknownVtbl *lpVtbl;
+ ULONG refs;
+} HeapUnknown;
+
+static HRESULT WINAPI HeapUnknown_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
+{
+ if (IsEqualIID(riid, &IID_IUnknown))
+ {
+ IUnknown_AddRef(iface);
+ *ppv = (LPVOID)iface;
+ return S_OK;
+ }
+ *ppv = NULL;
+ return E_NOINTERFACE;
+}
+
+static ULONG WINAPI HeapUnknown_AddRef(IUnknown *iface)
+{
+ HeapUnknown *This = (HeapUnknown *)iface;
+ return InterlockedIncrement((LONG*)&This->refs);
+}
+
+static ULONG WINAPI HeapUnknown_Release(IUnknown *iface)
+{
+ HeapUnknown *This = (HeapUnknown *)iface;
+ ULONG refs = InterlockedDecrement((LONG*)&This->refs);
+ if (!refs) HeapFree(GetProcessHeap(), 0, This);
+ return refs;
+}
+
+static const IUnknownVtbl HeapUnknown_Vtbl =
+{
+ HeapUnknown_QueryInterface,
+ HeapUnknown_AddRef,
+ HeapUnknown_Release
+};
+
static void check_variant_header(DWORD *wirev, VARIANT *v, unsigned long size)
{
WORD *wp;
@@ -445,6 +486,7 @@ static void test_marshal_VARIANT(void)
SAFEARRAYBOUND sab;
LPSAFEARRAY lpsa;
DECIMAL dec, dec2;
+ HeapUnknown *heap_unknown;
umcb.Flags = MAKELONG(MSHCTX_DIFFERENTMACHINE, NDR_LOCAL_DATA_REPRESENTATION);
umcb.pReserve = NULL;
@@ -1058,6 +1100,54 @@ static void test_marshal_VARIANT(void)
VARIANT_UserFree(&umcb.Flags, &v3);
}
HeapFree(GetProcessHeap(), 0, buffer);
+
+ /*** UNKNOWN ***/
+ heap_unknown = HeapAlloc(GetProcessHeap(), 0, sizeof(*heap_unknown));
+ heap_unknown->lpVtbl = &HeapUnknown_Vtbl;
+ heap_unknown->refs = 1;
+ VariantInit(&v);
+ VariantInit(&v2);
+ V_VT(&v) = VT_UNKNOWN;
+ V_UNKNOWN(&v) = (IUnknown *)heap_unknown;
+
+ size = VARIANT_UserSize(&umcb.Flags, 0, &v);
+ ok(size > 32, "size %ld\n", size);
+ buffer = HeapAlloc(GetProcessHeap(), 0, size);
+ memset(buffer, 0xcc, size);
+ next = VARIANT_UserMarshal(&umcb.Flags, buffer, &v);
+ wirev = (DWORD*)buffer;
+ check_variant_header(wirev, &v, next - buffer);
+ wirev += 5;
+
+ todo_wine
+ ok(*wirev == (DWORD_PTR)V_UNKNOWN(&v), "wv[5] %08x\n", *wirev);
+ wirev++;
+ todo_wine
+ ok(*wirev == next - buffer - 0x20, "wv[6] %08x\n", *wirev);
+ wirev++;
+ todo_wine
+ ok(*wirev == next - buffer - 0x20, "wv[7] %08x\n", *wirev);
+ wirev++;
+ todo_wine
+ ok(*wirev == 0x574f454d, "wv[8] %08x\n", *wirev);
+ if (VARIANT_UNMARSHAL_WORKS)
+ {
+ VARIANT v3;
+ VariantInit(&v3);
+ V_VT(&v3) = VT_UNKNOWN;
+ V_UNKNOWN(&v3) = (IUnknown *)heap_unknown;
+ IUnknown_AddRef(V_UNKNOWN(&v3));
+ next = VARIANT_UserUnmarshal(&umcb.Flags, buffer, &v3);
+ ok(V_VT(&v) == V_VT(&v3), "got vt %d expect %d\n", V_VT(&v), V_VT(&v3));
+ ok(V_VT(V_VARIANTREF(&v)) == V_VT(V_VARIANTREF(&v3)), "vts differ %x %x\n",
+ V_VT(V_VARIANTREF(&v)), V_VT(V_VARIANTREF(&v3)));
+ ok(V_R8(V_VARIANTREF(&v)) == V_R8(V_VARIANTREF(&v3)), "r8s differ\n");
+ VARIANT_UserFree(&umcb.Flags, &v3);
+ todo_wine
+ ok(heap_unknown->refs == 1, "%d refcounts of IUnknown leaked\n", heap_unknown->refs - 1);
+ IUnknown_Release((IUnknown *)heap_unknown);
+ }
+ HeapFree(GetProcessHeap(), 0, buffer);
}
More information about the wine-cvs
mailing list