Rob Shearman : ole32:
Add tests for default handler creation and handler marshaling.
Alexandre Julliard
julliard at wine.codeweavers.com
Mon Feb 26 07:59:26 CST 2007
Module: wine
Branch: master
Commit: 0101f89a8d08328c84425d08d1ed6513d7bd781e
URL: http://source.winehq.org/git/wine.git/?a=commit;h=0101f89a8d08328c84425d08d1ed6513d7bd781e
Author: Rob Shearman <rob at codeweavers.com>
Date: Fri Feb 23 19:38:11 2007 +0000
ole32: Add tests for default handler creation and handler marshaling.
---
dlls/ole32/tests/marshal.c | 142 ++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 142 insertions(+), 0 deletions(-)
diff --git a/dlls/ole32/tests/marshal.c b/dlls/ole32/tests/marshal.c
index 11f03df..f6397eb 100644
--- a/dlls/ole32/tests/marshal.c
+++ b/dlls/ole32/tests/marshal.c
@@ -48,6 +48,7 @@ static const IID IID_IWineTest =
}; /* 5201163f-8164-4fd0-a1a2-5d5a3654d3bd */
#define EXTENTID_WineTest IID_IWineTest
+#define CLSID_WineTest IID_IWineTest
static void test_cocreateinstance_proxy(void)
{
@@ -1963,6 +1964,145 @@ static void test_freethreadedmarshaler(void)
IMarshal_Release(pFTMarshal);
}
+static void test_inproc_handler(void)
+{
+ HRESULT hr;
+ IUnknown *pObject;
+ IUnknown *pObject2;
+ char buffer[256];
+ LPOLESTR pszClsid;
+ HKEY hkey;
+ DWORD dwDisposition;
+ DWORD error;
+
+ hr = StringFromCLSID(&CLSID_WineTest, &pszClsid);
+ ok_ole_success(hr, "StringFromCLSID");
+ strcpy(buffer, "CLSID\\");
+ WideCharToMultiByte(CP_ACP, 0, pszClsid, -1, buffer + strlen(buffer), sizeof(buffer) - strlen(buffer), NULL, NULL);
+ CoTaskMemFree(pszClsid);
+ strcat(buffer, "\\InprocHandler32");
+ error = RegCreateKeyEx(HKEY_CLASSES_ROOT, buffer, 0, NULL, 0, KEY_SET_VALUE, NULL, &hkey, &dwDisposition);
+ ok(error == ERROR_SUCCESS, "RegCreateKeyEx failed with error %d\n", error);
+ error = RegSetValueEx(hkey, NULL, 0, REG_SZ, (const unsigned char *)"ole32.dll", strlen("ole32.dll") + 1);
+ ok(error == ERROR_SUCCESS, "RegSetValueEx failed with error %d\n", error);
+ RegCloseKey(hkey);
+
+ hr = CoCreateInstance(&CLSID_WineTest, NULL, CLSCTX_INPROC_HANDLER, &IID_IUnknown, (void **)&pObject);
+ todo_wine
+ ok_ole_success(hr, "CoCreateInstance");
+
+ if (SUCCEEDED(hr))
+ {
+ hr = IUnknown_QueryInterface(pObject, &IID_IWineTest, (void **)&pObject2);
+ ok(hr == E_NOINTERFACE, "IUnknown_QueryInterface on handler for invalid interface returned 0x%08x instead of E_NOINTERFACE\n", hr);
+
+ /* it's a handler as it supports IOleObject */
+ hr = IUnknown_QueryInterface(pObject, &IID_IOleObject, (void **)&pObject2);
+ ok_ole_success(hr, "IUnknown_QueryInterface(&IID_IOleObject)");
+ IUnknown_Release(pObject2);
+
+ IUnknown_Release(pObject);
+ }
+
+ RegDeleteKey(HKEY_CLASSES_ROOT, buffer);
+ *strrchr(buffer, '\\') = '\0';
+ RegDeleteKey(HKEY_CLASSES_ROOT, buffer);
+}
+
+static HRESULT WINAPI Test_SMI_QueryInterface(
+ IStdMarshalInfo *iface,
+ REFIID riid,
+ LPVOID *ppvObj)
+{
+ if (ppvObj == NULL) return E_POINTER;
+
+ if (IsEqualGUID(riid, &IID_IUnknown) ||
+ IsEqualGUID(riid, &IID_IStdMarshalInfo))
+ {
+ *ppvObj = (LPVOID)iface;
+ IClassFactory_AddRef(iface);
+ return S_OK;
+ }
+
+ return E_NOINTERFACE;
+}
+
+static ULONG WINAPI Test_SMI_AddRef(IStdMarshalInfo *iface)
+{
+ LockModule();
+ return 2; /* non-heap-based object */
+}
+
+static ULONG WINAPI Test_SMI_Release(IStdMarshalInfo *iface)
+{
+ UnlockModule();
+ return 1; /* non-heap-based object */
+}
+
+static HRESULT WINAPI Test_SMI_GetClassForHandler(
+ IStdMarshalInfo *iface,
+ DWORD dwDestContext,
+ void *pvDestContext,
+ CLSID *pClsid)
+{
+ *pClsid = CLSID_WineTest;
+ return S_OK;
+}
+
+static const IStdMarshalInfoVtbl Test_SMI_Vtbl =
+{
+ Test_SMI_QueryInterface,
+ Test_SMI_AddRef,
+ Test_SMI_Release,
+ Test_SMI_GetClassForHandler
+};
+
+static IStdMarshalInfo Test_SMI = {&Test_SMI_Vtbl};
+
+static void test_handler_marshaling(void)
+{
+ HRESULT hr;
+ IStream *pStream = NULL;
+ IUnknown *pProxy = NULL;
+ IUnknown *pObject;
+ DWORD tid;
+ HANDLE thread;
+ static const LARGE_INTEGER ullZero;
+
+ cLocks = 0;
+
+ hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream);
+ ok_ole_success(hr, "CreateStreamOnHGlobal");
+ tid = start_host_object(pStream, &IID_IUnknown, (IUnknown*)&Test_SMI, MSHLFLAGS_NORMAL, &thread);
+
+ ok_more_than_one_lock();
+
+ IStream_Seek(pStream, ullZero, STREAM_SEEK_SET, NULL);
+ hr = CoUnmarshalInterface(pStream, &IID_IUnknown, (void **)&pProxy);
+ ok_ole_success(hr, "CoUnmarshalInterface");
+ IStream_Release(pStream);
+
+ ok_more_than_one_lock();
+
+ hr = IUnknown_QueryInterface(pProxy, &IID_IWineTest, (void **)&pObject);
+ ok(hr == E_NOINTERFACE, "IUnknown_QueryInterface with unknown IID should have returned E_NOINTERFACE instead of 0x%08x\n", hr);
+
+ /* it's a handler as it supports IOleObject */
+ hr = IUnknown_QueryInterface(pProxy, &IID_IOleObject, (void **)&pObject);
+ todo_wine
+ ok_ole_success(hr, "IUnknown_QueryInterface(&IID_IOleObject)");
+ if (SUCCEEDED(hr)) IUnknown_Release(pObject);
+
+ IUnknown_Release(pProxy);
+
+ ok_no_locks();
+
+ end_host_object(tid, thread);
+
+ /* FIXME: test IPersist interface has the same effect as IStdMarshalInfo */
+}
+
+
static HANDLE heventShutdown;
static void LockModuleOOP(void)
@@ -2441,6 +2581,8 @@ START_TEST(marshal)
test_call_from_message();
test_WM_QUIT_handling();
test_freethreadedmarshaler();
+ test_inproc_handler();
+ test_handler_marshaling();
/* doesn't pass with Win9x COM DLLs (even though Essential COM says it should) */
if (0) test_out_of_process_com();
More information about the wine-cvs
mailing list