defaulthandler.c File Reference

#include <stdarg.h>
#include <stdio.h>
#include "windef.h"
#include "winbase.h"
#include "objbase.h"
#include "wine/test.h"

Include dependency graph for defaulthandler.c:

Go to the source code of this file.

Classes
struct	ole_stream_header_t

Macros
#define	COBJMACROS
#define	CONST_VTABLE
#define	DEFINE_EXPECT(func)    static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
#define	SET_EXPECT(func)    expect_ ## func = TRUE
#define	CHECK_EXPECT2(func)
#define	CHECK_EXPECT(func)
#define	CHECK_CALLED(func)
#define	CHECK_NOT_CALLED(func)

Functions
	DEFINE_EXPECT (CF_QueryInterface_ClassFactory)
	DEFINE_EXPECT (CF_CreateInstance)
	DEFINE_EXPECT (CF_QueryInterface_IMarshal)
static HRESULT	create_storage (IStorage **stg)
static void	test_olestream (void)
static HRESULT WINAPI	test_class_QueryInterface (IUnknown *iface, REFIID riid, void **ppv)
static ULONG WINAPI	test_class_AddRef (IUnknown *iface)
static ULONG WINAPI	test_class_Release (IUnknown *iface)
static HRESULT WINAPI	ClassFactory_QueryInterface (IClassFactory *iface, REFIID riid, void **ppv)
static ULONG WINAPI	ClassFactory_AddRef (IClassFactory *iface)
static ULONG WINAPI	ClassFactory_Release (IClassFactory *iface)
static HRESULT WINAPI	ClassFactory_CreateInstance (IClassFactory *iface, IUnknown *pUnkOuter, REFIID riid, void **ppv)
static HRESULT WINAPI	ClassFactory_LockServer (IClassFactory *iface, BOOL fLock)
static void	test_default_handler_run (void)
	START_TEST (defaulthandler)

Variables
static const IUnknownVtbl	test_class_vtbl
static IUnknown	test_class = { &test_class_vtbl }
static const IClassFactoryVtbl	ClassFactoryVtbl
static IClassFactory	ClassFactory = { &ClassFactoryVtbl }

Macro Definition Documentation

CHECK_CALLED

#define CHECK_CALLED

(

func

)

Value:

    do { \
        ok(called_ ## func, "expected " #func "\n"); \
        expect_ ## func = called_ ## func = FALSE; \
    }while(0)

Definition at line 51 of file defaulthandler.c.

CHECK_EXPECT

#define CHECK_EXPECT

(

func

)

Value:

    do { \
        CHECK_EXPECT2(func); \
        expect_ ## func = FALSE; \
    }while(0)

Definition at line 45 of file defaulthandler.c.

CHECK_EXPECT2

#define CHECK_EXPECT2

(

func

)

Value:

    do { \
        ok(expect_ ##func, "unexpected call " #func "\n"); \
        called_ ## func = TRUE; \
    }while(0)

Definition at line 39 of file defaulthandler.c.

CHECK_NOT_CALLED

#define CHECK_NOT_CALLED

(

func

)

Value:

    do { \
        ok(!called_ ## func, "unexpected " #func "\n"); \
        expect_ ## func = called_ ## func = FALSE; \
    }while(0)

Definition at line 57 of file defaulthandler.c.

COBJMACROS

#define COBJMACROS

Definition at line 21 of file defaulthandler.c.

CONST_VTABLE

#define CONST_VTABLE

Definition at line 22 of file defaulthandler.c.

DEFINE_EXPECT

#define DEFINE_EXPECT

(

func

)

static BOOL expect_ ## func = FALSE, called_ ## func = FALSE

Definition at line 33 of file defaulthandler.c.

SET_EXPECT

#define SET_EXPECT

(

func

)

expect_ ## func = TRUE

Definition at line 36 of file defaulthandler.c.

Function Documentation

ClassFactory_AddRef()

static ULONG WINAPI ClassFactory_AddRef ( IClassFactory *  iface )

static

Definition at line 190 of file defaulthandler.c.

{
    return 2;
}

ClassFactory_CreateInstance()

static HRESULT WINAPI ClassFactory_CreateInstance ( IClassFactory *  iface, IUnknown *  pUnkOuter, REFIID  riid, void **  ppv  )

static

Definition at line 200 of file defaulthandler.c.

{
    CHECK_EXPECT(CF_CreateInstance);
 
    ok(pUnkOuter == NULL, "pUnkOuter != NULL\n");
    todo_wine ok(IsEqualGUID(riid, &IID_IUnknown), "riid = %s\n", wine_dbgstr_guid(riid));
    if(IsEqualGUID(riid, &IID_IOleObject)) {
        *ppv = NULL;
        return E_NOINTERFACE;
    }
 
    *ppv = &test_class;
    return S_OK;
}

ClassFactory_LockServer()

static HRESULT WINAPI ClassFactory_LockServer ( IClassFactory *  iface, BOOL  fLock  )

static

Definition at line 216 of file defaulthandler.c.

{
    ok(0, "unexpected call\n");
    return E_NOTIMPL;
}

ClassFactory_QueryInterface()

static HRESULT WINAPI ClassFactory_QueryInterface ( IClassFactory *  iface, REFIID  riid, void **  ppv  )

static

Definition at line 170 of file defaulthandler.c.

{
    if(IsEqualGUID(riid, &IID_IUnknown)) {
        *ppv = iface;
        return S_OK;
    }else if(IsEqualGUID(riid, &IID_IMarshal)) {
        CHECK_EXPECT(CF_QueryInterface_IMarshal);
        *ppv = NULL;
        return E_NOINTERFACE;
    }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
        CHECK_EXPECT(CF_QueryInterface_ClassFactory);
        *ppv = iface;
        return S_OK;
    }
 
    ok(0, "unexpected interface: %s\n", wine_dbgstr_guid(riid));
    *ppv = NULL;
    return E_NOINTERFACE;
}

ClassFactory_Release()

static ULONG WINAPI ClassFactory_Release ( IClassFactory *  iface )

static

Definition at line 195 of file defaulthandler.c.

{
    return 1;
}

create_storage()

static HRESULT create_storage ( IStorage **  stg )

static

Definition at line 67 of file defaulthandler.c.

{
    HRESULT hr;
    ILockBytes *lock_bytes;
 
    hr = CreateILockBytesOnHGlobal(NULL, TRUE, &lock_bytes);
    if(SUCCEEDED(hr))
    {
        hr = StgCreateDocfileOnILockBytes(lock_bytes,
                  STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, stg);
        ILockBytes_Release(lock_bytes);
    }
    return hr;
}

DEFINE_EXPECT() [1/3]

DEFINE_EXPECT

(

CF_CreateInstance

)

DEFINE_EXPECT() [2/3]

DEFINE_EXPECT

(

CF_QueryInterface_ClassFactory

)

DEFINE_EXPECT() [3/3]

DEFINE_EXPECT

(

CF_QueryInterface_IMarshal

)

START_TEST()

START_TEST

(

defaulthandler

)

Definition at line 311 of file defaulthandler.c.

{
    OleInitialize(NULL);
 
    test_olestream();
    test_default_handler_run();
 
    OleUninitialize();
}

test_class_AddRef()

static ULONG WINAPI test_class_AddRef ( IUnknown *  iface )

static

Definition at line 152 of file defaulthandler.c.

{
    return 2;
}

test_class_QueryInterface()

static HRESULT WINAPI test_class_QueryInterface ( IUnknown *  iface, REFIID  riid, void **  ppv  )

static

Definition at line 137 of file defaulthandler.c.

{
    if(IsEqualGUID(riid, &IID_IUnknown)) {
        *ppv = iface;
        return S_OK;
    }else if(IsEqualGUID(riid, &IID_IOleObject)) {
        ok(0, "unexpected query for IOleObject interface\n");
        *ppv = NULL;
        return E_NOINTERFACE;
    }
 
    *ppv = NULL;
    return E_NOINTERFACE;
}

test_class_Release()

static ULONG WINAPI test_class_Release ( IUnknown *  iface )

static

Definition at line 157 of file defaulthandler.c.

{
    return 1;
}

test_default_handler_run()

static void test_default_handler_run ( void  )

static

Definition at line 232 of file defaulthandler.c.

{
    const CLSID test_server_clsid = {0x0f77e570,0x80c3,0x11e2,{0x9e,0x96,0x08,0x00,0x20,0x0c,0x9a,0x66}};
 
    IUnknown *unk;
    IRunnableObject *ro;
    IOleObject *oleobj;
    IPersistStorage *persist;
    DWORD class_reg;
    HRESULT hres;
 
    if(!GetProcAddress(GetModuleHandleA("ole32"), "CoRegisterSurrogateEx")) {
        win_skip("skipping OleCreateDefaultHandler tests\n");
        return;
    }
 
    hres = CoRegisterClassObject(&test_server_clsid, (IUnknown*)&ClassFactory,
            CLSCTX_INPROC_SERVER, 0, &class_reg);
    ok(hres == S_OK, "CoRegisterClassObject failed: %lx\n", hres);
 
    hres = OleCreateDefaultHandler(&test_server_clsid, NULL, &IID_IUnknown, (void**)&unk);
    ok(hres == S_OK, "OleCreateDefaultHandler failed: %lx\n", hres);
 
    hres = IUnknown_QueryInterface(unk, &IID_IRunnableObject, (void**)&ro);
    ok(hres == S_OK, "QueryInterface(IRunnableObject) failed: %lx\n", hres);
    IUnknown_Release(unk);
 
    hres = IRunnableObject_Run(ro, NULL);
    ok(hres == REGDB_E_CLASSNOTREG, "Run returned: %lx, expected REGDB_E_CLASSNOTREG\n", hres);
    IRunnableObject_Release(ro);
 
    SET_EXPECT(CF_QueryInterface_IMarshal);
    CoRevokeClassObject(class_reg);
    todo_wine CHECK_CALLED(CF_QueryInterface_IMarshal);
 
    hres = CoRegisterClassObject(&test_server_clsid, (IUnknown*)&ClassFactory,
            CLSCTX_LOCAL_SERVER, 0, &class_reg);
    ok(hres == S_OK, "CoRegisterClassObject failed: %lx\n", hres);
 
    hres = OleCreateDefaultHandler(&test_server_clsid, NULL, &IID_IUnknown, (void**)&unk);
    ok(hres == S_OK, "OleCreateDefaultHandler failed: %lx\n", hres);
 
    hres = IUnknown_QueryInterface(unk, &IID_IOleObject, (void**)&oleobj);
    ok(hres == S_OK, "QueryInterface(IID_IOleObject) failed: %lx\n", hres);
 
    hres = IOleObject_QueryInterface(oleobj, &IID_IPersistStorage, (void**)&persist);
    ok(hres == S_OK, "QueryInterface(IID_IPersistStorage) failed: %lx\n", hres);
    IPersistStorage_Release(persist);
    IOleObject_Release(oleobj);
 
    hres = IUnknown_QueryInterface(unk, &IID_IRunnableObject, (void**)&ro);
    ok(hres == S_OK, "QueryInterface(IRunnableObject) failed: %lx\n", hres);
    IUnknown_Release(unk);
 
    SET_EXPECT(CF_QueryInterface_ClassFactory);
    SET_EXPECT(CF_CreateInstance);
    hres = IRunnableObject_Run(ro, NULL);
    todo_wine
    ok(hres == S_OK, "Run failed: %lx\n", hres);
    CHECK_CALLED(CF_QueryInterface_ClassFactory);
    CHECK_CALLED(CF_CreateInstance);
    IRunnableObject_Release(ro);
 
    SET_EXPECT(CF_QueryInterface_ClassFactory);
    SET_EXPECT(CF_CreateInstance);
    hres = CoCreateInstance(&test_server_clsid, NULL, CLSCTX_LOCAL_SERVER,
                            &IID_IOleObject, (void**)&oleobj);
    todo_wine
    ok(hres == REGDB_E_CLASSNOTREG, "expected REGDB_E_CLASSNOTREG, got %lx\n", hres);
    todo_wine
    CHECK_NOT_CALLED(CF_QueryInterface_ClassFactory);
    todo_wine
    CHECK_NOT_CALLED(CF_CreateInstance);
 
    SET_EXPECT(CF_QueryInterface_IMarshal);
    CoRevokeClassObject(class_reg);
    todo_wine CHECK_CALLED(CF_QueryInterface_IMarshal);
}

Referenced by START_TEST().

test_olestream()

static void test_olestream ( void  )

static

Definition at line 91 of file defaulthandler.c.

{
    HRESULT hr;
    const CLSID non_existent_class = {0xa5f1772f, 0x3772, 0x490f, {0x9e, 0xc6, 0x77, 0x13, 0xe8, 0xb3, 0x4b, 0x5d}};
    IOleObject *ole_obj;
    IPersistStorage *persist;
    IStorage *stg;
    IStream *stm;
    static const WCHAR olestream[] = {1,'O','l','e',0};
    ULONG read;
    ole_stream_header_t header;
 
    hr = create_storage(&stg);
    ok(hr == S_OK, "got %08lx\n", hr);
 
    hr = IStorage_OpenStream(stg, olestream, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stm);
    ok(hr == STG_E_FILENOTFOUND, "got %08lx\n", hr);
 
    hr = OleCreateDefaultHandler(&non_existent_class, 0, &IID_IOleObject, (void**)&ole_obj);
    ok(hr == S_OK, "got %08lx\n", hr);
 
    hr = IOleObject_QueryInterface(ole_obj, &IID_IPersistStorage, (void**)&persist);
    ok(hr == S_OK, "got %08lx\n", hr);
 
    hr = IPersistStorage_InitNew(persist, stg);
    ok(hr == S_OK, "got %08lx\n", hr);
 
    hr = IStorage_OpenStream(stg, olestream, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stm);
    ok(hr == S_OK, "got %08lx\n", hr);
    hr = IStream_Read(stm, &header, sizeof(header), &read);
    ok(hr == S_OK, "got %08lx\n", hr);
    ok(read == sizeof(header), "read %ld\n", read);
    ok(header.version == 0x02000001, "got version %08lx\n", header.version);
    ok(header.flags == 0x0, "got flags %08lx\n", header.flags);
    ok(header.link_update_opt == 0x0, "got link update option %08lx\n", header.link_update_opt);
    ok(header.res == 0x0, "got reserved %08lx\n", header.res);
    ok(header.moniker_size == 0x0, "got moniker size %08lx\n", header.moniker_size);
 
    IStream_Release(stm);
 
    IPersistStorage_Release(persist);
    IOleObject_Release(ole_obj);
 
    IStorage_Release(stg);
}

Referenced by START_TEST().

Variable Documentation

ClassFactory

IClassFactory ClassFactory = { &ClassFactoryVtbl }

static

Definition at line 230 of file defaulthandler.c.

ClassFactoryVtbl

const IClassFactoryVtbl ClassFactoryVtbl

static

Initial value:

= {
    ClassFactory_QueryInterface,
    ClassFactory_AddRef,
    ClassFactory_Release,
    ClassFactory_CreateInstance,
    ClassFactory_LockServer
}

Definition at line 222 of file defaulthandler.c.

test_class

IUnknown test_class = { &test_class_vtbl }

static

Definition at line 168 of file defaulthandler.c.

Referenced by ClassFactory_CreateInstance().

test_class_vtbl

const IUnknownVtbl test_class_vtbl

static

Initial value:

= {
    test_class_QueryInterface,
    test_class_AddRef,
    test_class_Release,
}

Definition at line 162 of file defaulthandler.c.