rostests/winetests/ole32/defaulthandler.c

   1 /*
   2  * Default Handler Tests
   3  *
   4  * Copyright 2008 Huw Davies
   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 #define WIN32_NO_STATUS
  22 #define _INC_WINDOWS
  23 #define COM_NO_WINDOWS_H
  24
  25 #define COBJMACROS
  26 #define CONST_VTABLE
  27
  28 //#include <stdarg.h>
  29 #include <stdio.h>
  30
  31 #include <windef.h>
  32 #include <winbase.h>
  33 #include <ole2.h>
  34 //#include "objbase.h"
  35
  36 #include <wine/test.h>
  37
  38 #define DEFINE_EXPECT(func) \
  39     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  40
  41 #define SET_EXPECT(func) \
  42     expect_ ## func = TRUE
  43
  44 #define CHECK_EXPECT2(func) \
  45     do { \
  46         ok(expect_ ##func, "unexpected call " #func "\n"); \
  47         called_ ## func = TRUE; \
  48     }while(0)
  49
  50 #define CHECK_EXPECT(func) \
  51     do { \
  52         CHECK_EXPECT2(func); \
  53         expect_ ## func = FALSE; \
  54     }while(0)
  55
  56 #define CHECK_CALLED(func) \
  57     do { \
  58         ok(called_ ## func, "expected " #func "\n"); \
  59         expect_ ## func = called_ ## func = FALSE; \
  60     }while(0)
  61
  62 #define CHECK_NOT_CALLED(func) \
  63     do { \
  64         ok(!called_ ## func, "unexpected " #func "\n"); \
  65         expect_ ## func = called_ ## func = FALSE; \
  66     }while(0)
  67
  68 DEFINE_EXPECT(CF_QueryInterface_ClassFactory);
  69 DEFINE_EXPECT(CF_CreateInstance);
  70 DEFINE_EXPECT(CF_QueryInterface_IMarshal);
  71
  72 static HRESULT create_storage(IStorage **stg)
  73 {
  74     HRESULT hr;
  75     ILockBytes *lock_bytes;
  76
  77     hr = CreateILockBytesOnHGlobal(NULL, TRUE, &lock_bytes);
  78     if(SUCCEEDED(hr))
  79     {
  80         hr = StgCreateDocfileOnILockBytes(lock_bytes,
  81                   STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, stg);
  82         ILockBytes_Release(lock_bytes);
  83     }
  84     return hr;
  85 }
  86
  87 typedef struct
  88 {
  89     DWORD version;
  90     DWORD flags;
  91     DWORD link_update_opt;
  92     DWORD res;
  93     DWORD moniker_size;
  94 } ole_stream_header_t;
  95
  96 static void test_olestream(void)
  97 {
  98     HRESULT hr;
  99     const CLSID non_existent_class = {0xa5f1772f, 0x3772, 0x490f, {0x9e, 0xc6, 0x77, 0x13, 0xe8, 0xb3, 0x4b, 0x5d}};
 100     IOleObject *ole_obj;
 101     IPersistStorage *persist;
 102     IStorage *stg;
 103     IStream *stm;
 104     static const WCHAR olestream[] = {1,'O','l','e',0};
 105     ULONG read;
 106     ole_stream_header_t header;
 107
 108     hr = create_storage(&stg);
 109     ok(hr == S_OK, "got %08x\n", hr);
 110
 111     hr = IStorage_OpenStream(stg, olestream, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stm);
 112     ok(hr == STG_E_FILENOTFOUND, "got %08x\n", hr);
 113
 114     hr = OleCreateDefaultHandler(&non_existent_class, 0, &IID_IOleObject, (void**)&ole_obj);
 115     ok(hr == S_OK, "got %08x\n", hr);
 116
 117     hr = IOleObject_QueryInterface(ole_obj, &IID_IPersistStorage, (void**)&persist);
 118     ok(hr == S_OK, "got %08x\n", hr);
 119
 120     hr = IPersistStorage_InitNew(persist, stg);
 121     ok(hr == S_OK, "got %08x\n", hr);
 122
 123     hr = IStorage_OpenStream(stg, olestream, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stm);
 124     ok(hr == S_OK, "got %08x\n", hr);
 125     hr = IStream_Read(stm, &header, sizeof(header), &read);
 126     ok(hr == S_OK, "got %08x\n", hr);
 127     ok(read == sizeof(header), "read %d\n", read);
 128     ok(header.version == 0x02000001, "got version %08x\n", header.version);
 129     ok(header.flags == 0x0, "got flags %08x\n", header.flags);
 130     ok(header.link_update_opt == 0x0, "got link update option %08x\n", header.link_update_opt);
 131     ok(header.res == 0x0, "got reserved %08x\n", header.res);
 132     ok(header.moniker_size == 0x0, "got moniker size %08x\n", header.moniker_size);
 133
 134     IStream_Release(stm);
 135
 136     IPersistStorage_Release(persist);
 137     IOleObject_Release(ole_obj);
 138
 139     IStorage_Release(stg);
 140 }
 141
 142 static HRESULT WINAPI test_class_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
 143 {
 144     if(IsEqualGUID(riid, &IID_IUnknown)) {
 145         *ppv = iface;
 146         return S_OK;
 147     }else if(IsEqualGUID(riid, &IID_IOleObject)) {
 148         ok(0, "unexpected query for IOleObject interface\n");
 149         *ppv = NULL;
 150         return E_NOINTERFACE;
 151     }
 152
 153     *ppv = NULL;
 154     return E_NOINTERFACE;
 155 }
 156
 157 static ULONG WINAPI test_class_AddRef(IUnknown *iface)
 158 {
 159     return 2;
 160 }
 161
 162 static ULONG WINAPI test_class_Release(IUnknown *iface)
 163 {
 164     return 1;
 165 }
 166
 167 static const IUnknownVtbl test_class_vtbl = {
 168     test_class_QueryInterface,
 169     test_class_AddRef,
 170     test_class_Release,
 171 };
 172
 173 static IUnknown test_class = { &test_class_vtbl };
 174
 175 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
 176 {
 177     if(IsEqualGUID(riid, &IID_IUnknown)) {
 178         *ppv = iface;
 179         return S_OK;
 180     }else if(IsEqualGUID(riid, &IID_IMarshal)) {
 181         CHECK_EXPECT(CF_QueryInterface_IMarshal);
 182         *ppv = NULL;
 183         return E_NOINTERFACE;
 184     }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
 185         CHECK_EXPECT(CF_QueryInterface_ClassFactory);
 186         *ppv = iface;
 187         return S_OK;
 188     }
 189
 190     ok(0, "unexpected interface: %s\n", wine_dbgstr_guid(riid));
 191     *ppv = NULL;
 192     return E_NOINTERFACE;
 193 }
 194
 195 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
 196 {
 197     return 2;
 198 }
 199
 200 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
 201 {
 202     return 1;
 203 }
 204
 205 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface,
 206         IUnknown *pUnkOuter, REFIID riid, void **ppv)
 207 {
 208     CHECK_EXPECT(CF_CreateInstance);
 209
 210     ok(pUnkOuter == NULL, "pUnkOuter != NULL\n");
 211     todo_wine ok(IsEqualGUID(riid, &IID_IUnknown), "riid = %s\n", wine_dbgstr_guid(riid));
 212     if(IsEqualGUID(riid, &IID_IOleObject)) {
 213         *ppv = NULL;
 214         return E_NOINTERFACE;
 215     }
 216
 217     *ppv = &test_class;
 218     return S_OK;
 219 }
 220
 221 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
 222 {
 223     ok(0, "unexpected call\n");
 224     return E_NOTIMPL;
 225 }
 226
 227 static const IClassFactoryVtbl ClassFactoryVtbl = {
 228     ClassFactory_QueryInterface,
 229     ClassFactory_AddRef,
 230     ClassFactory_Release,
 231     ClassFactory_CreateInstance,
 232     ClassFactory_LockServer
 233 };
 234
 235 static IClassFactory ClassFactory = { &ClassFactoryVtbl };
 236
 237 static void test_default_handler_run(void)
 238 {
 239     const CLSID test_server_clsid = {0x0f77e570,0x80c3,0x11e2,{0x9e,0x96,0x08,0x00,0x20,0x0c,0x9a,0x66}};
 240
 241     IUnknown *unk;
 242     IRunnableObject *ro;
 243     IOleObject *oleobj;
 244     IPersistStorage *persist;
 245     DWORD class_reg;
 246     HRESULT hres;
 247
 248     if(!GetProcAddress(GetModuleHandleA("ole32"), "CoRegisterSurrogateEx")) {
 249         win_skip("skipping OleCreateDefaultHandler tests\n");
 250         return;
 251     }
 252
 253     hres = CoRegisterClassObject(&test_server_clsid, (IUnknown*)&ClassFactory,
 254             CLSCTX_INPROC_SERVER, 0, &class_reg);
 255     ok(hres == S_OK, "CoRegisterClassObject failed: %x\n", hres);
 256
 257     hres = OleCreateDefaultHandler(&test_server_clsid, NULL, &IID_IUnknown, (void**)&unk);
 258     ok(hres == S_OK, "OleCreateDefaultHandler failed: %x\n", hres);
 259
 260     hres = IUnknown_QueryInterface(unk, &IID_IRunnableObject, (void**)&ro);
 261     ok(hres == S_OK, "QueryInterface(IRunnableObject) failed: %x\n", hres);
 262     IUnknown_Release(unk);
 263
 264     hres = IRunnableObject_Run(ro, NULL);
 265     ok(hres == REGDB_E_CLASSNOTREG, "Run returned: %x, expected REGDB_E_CLASSNOTREG\n", hres);
 266     IRunnableObject_Release(ro);
 267
 268     SET_EXPECT(CF_QueryInterface_IMarshal);
 269     CoRevokeClassObject(class_reg);
 270     todo_wine CHECK_CALLED(CF_QueryInterface_IMarshal);
 271
 272     hres = CoRegisterClassObject(&test_server_clsid, (IUnknown*)&ClassFactory,
 273             CLSCTX_LOCAL_SERVER, 0, &class_reg);
 274     ok(hres == S_OK, "CoRegisterClassObject failed: %x\n", hres);
 275
 276     hres = OleCreateDefaultHandler(&test_server_clsid, NULL, &IID_IUnknown, (void**)&unk);
 277     ok(hres == S_OK, "OleCreateDefaultHandler failed: %x\n", hres);
 278
 279     hres = IUnknown_QueryInterface(unk, &IID_IOleObject, (void**)&oleobj);
 280     ok(hres == S_OK, "QueryInterface(IID_IOleObject) failed: %x\n", hres);
 281
 282     hres = IOleObject_QueryInterface(oleobj, &IID_IPersistStorage, (void**)&persist);
 283     ok(hres == S_OK, "QueryInterface(IID_IPersistStorage) failed: %x\n", hres);
 284     IPersistStorage_Release(persist);
 285     IOleObject_Release(oleobj);
 286
 287     hres = IUnknown_QueryInterface(unk, &IID_IRunnableObject, (void**)&ro);
 288     ok(hres == S_OK, "QueryInterface(IRunnableObject) failed: %x\n", hres);
 289     IUnknown_Release(unk);
 290
 291     SET_EXPECT(CF_QueryInterface_ClassFactory);
 292     SET_EXPECT(CF_CreateInstance);
 293     hres = IRunnableObject_Run(ro, NULL);
 294 todo_wine
 295     ok(hres == S_OK, "Run failed: %x\n", hres);
 296     CHECK_CALLED(CF_QueryInterface_ClassFactory);
 297     CHECK_CALLED(CF_CreateInstance);
 298     IRunnableObject_Release(ro);
 299
 300     SET_EXPECT(CF_QueryInterface_ClassFactory);
 301     SET_EXPECT(CF_CreateInstance);
 302     hres = CoCreateInstance(&test_server_clsid, NULL, CLSCTX_LOCAL_SERVER,
 303                             &IID_IOleObject, (void**)&oleobj);
 304 todo_wine
 305     ok(hres == REGDB_E_CLASSNOTREG, "expected REGDB_E_CLASSNOTREG, got %x\n", hres);
 306 todo_wine
 307     CHECK_NOT_CALLED(CF_QueryInterface_ClassFactory);
 308 todo_wine
 309     CHECK_NOT_CALLED(CF_CreateInstance);
 310
 311     SET_EXPECT(CF_QueryInterface_IMarshal);
 312     CoRevokeClassObject(class_reg);
 313     todo_wine CHECK_CALLED(CF_QueryInterface_IMarshal);
 314 }
 315
 316 START_TEST(defaulthandler)
 317 {
 318     OleInitialize(NULL);
 319
 320     test_olestream();
 321     test_default_handler_run();
 322
 323     OleUninitialize();
 324 }