thread.c File Reference

#include <stdio.h>
#include <stdarg.h>
#include "windef.h"
#include "winbase.h"
#include "winerror.h"
#include "ole2.h"
#include "shlwapi.h"
#include "wine/test.h"

Include dependency graph for thread.c:

Go to the source code of this file.

Classes
struct	threadref

Macros
#define	COBJMACROS
#define	CONST_VTABLE

Functions
static	HRESULT (WINAPI *pSHCreateThreadRef)(LONG *
static threadref *	impl_from_IUnknown (IUnknown *iface)
static HRESULT WINAPI	threadref_QueryInterface (IUnknown *iface, REFIID riid, LPVOID *ppvObj)
static ULONG WINAPI	threadref_AddRef (IUnknown *iface)
static ULONG WINAPI	threadref_Release (IUnknown *iface)
static void	init_threadref (threadref *iface, LONG *refcount)
static void	test_SHCreateThreadRef (void)
static void	test_SHGetThreadRef (void)
static void	test_SHSetThreadRef (void)
	START_TEST (thread)

Variables
static DWORD	AddRef_called
static const IUnknownVtbl	threadref_vt

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 23 of file thread.c.

CONST_VTABLE

#define CONST_VTABLE

Definition at line 24 of file thread.c.

Function Documentation

HRESULT()

static HRESULT ( WINAPI *  pSHCreateThreadRef )

static

impl_from_IUnknown()

static threadref * impl_from_IUnknown ( IUnknown *  iface )

inlinestatic

Definition at line 46 of file thread.c.

{
  return CONTAINING_RECORD(iface, threadref, IUnknown_iface);
}

init_threadref()

static void init_threadref ( threadref *  iface, LONG *  refcount  )

static

Definition at line 84 of file thread.c.

{
  iface->IUnknown_iface.lpVtbl = &threadref_vt;
  iface->ref = refcount;
}

Referenced by test_SHSetThreadRef().

START_TEST()

START_TEST

(

thread

)

Definition at line 244 of file thread.c.

{
    HMODULE hshlwapi = GetModuleHandleA("shlwapi.dll");
 
    pSHCreateThreadRef = (void *) GetProcAddress(hshlwapi, "SHCreateThreadRef");
    pSHGetThreadRef = (void *) GetProcAddress(hshlwapi, "SHGetThreadRef");
    pSHSetThreadRef = (void *) GetProcAddress(hshlwapi, "SHSetThreadRef");
 
    test_SHCreateThreadRef();
    test_SHGetThreadRef();
    test_SHSetThreadRef();
 
}

test_SHCreateThreadRef()

static void test_SHCreateThreadRef ( void  )

static

Definition at line 92 of file thread.c.

{
    IUnknown *pobj;
    IUnknown *punk;
    LONG refcount;
    HRESULT hr;
 
    /* Not present before IE 6_XP_sp2 */
    if (!pSHCreateThreadRef) {
        win_skip("SHCreateThreadRef not found\n");
        return;
    }
 
    /* start with a clean state */
    hr = pSHSetThreadRef(NULL);
    ok(hr == S_OK, "got 0x%lx (expected S_OK)\n", hr);
 
    pobj = NULL;
    refcount = 0xdeadbeef;
    hr = pSHCreateThreadRef(&refcount, &pobj);
    ok((hr == S_OK) && pobj && (refcount == 1),
        "got 0x%lx and %p with %ld (expected S_OK and '!= NULL' with 1)\n",
        hr, pobj, refcount);
 
    /* the object is not automatic set as ThreadRef */
    punk = NULL;
    hr = pSHGetThreadRef(&punk);
    ok( (hr == E_NOINTERFACE) && (punk == NULL),
        "got 0x%lx and %p (expected E_NOINTERFACE and NULL)\n", hr, punk);
 
    /* set the object */
    hr = pSHSetThreadRef(pobj);
    ok(hr == S_OK, "got 0x%lx (expected S_OK)\n", hr);
 
    /* read back */
    punk = NULL;
    hr = pSHGetThreadRef(&punk);
    ok( (hr == S_OK) && (punk == pobj) && (refcount == 2),
        "got 0x%lx and %p with %ld (expected S_OK and %p with 2)\n",
        hr, punk, refcount, pobj);
 
    /* free the ref from SHGetThreadRef */
    if (SUCCEEDED(hr)) {
        hr = IUnknown_Release(pobj);
        ok((hr == 1) && (hr == refcount),
            "got %ld with %ld (expected 1 with 1)\n", hr, refcount);
    }
 
    /* free the object */
    if (pobj) {
        hr = IUnknown_Release(pobj);
        ok((hr == 0) && (hr == refcount),
            "got %ld with %ld (expected 0 with 0)\n", hr, refcount);
    }
 
    if (0) {
        /* the ThreadRef has still the pointer,
           but the object no longer exist after the *_Release */
        punk = NULL;
        hr = pSHGetThreadRef(&punk);
        trace("got 0x%lx and %p with %ld\n", hr, punk, refcount);
    }
 
    /* remove the dead object pointer */
    hr = pSHSetThreadRef(NULL);
    ok(hr == S_OK, "got 0x%lx (expected S_OK)\n", hr);
 
    /* parameter check */
    if (0) {
        /* vista: E_INVALIDARG, XP: crash */
        pobj = NULL;
        hr = pSHCreateThreadRef(NULL, &pobj);
        ok(hr == E_INVALIDARG, "got 0x%lx (expected E_INVALIDARG)\n", hr);
 
        refcount = 0xdeadbeef;
        /* vista: E_INVALIDARG, XP: crash */
        hr = pSHCreateThreadRef(&refcount, NULL);
        ok( (hr == E_INVALIDARG) && (refcount == 0xdeadbeef),
            "got 0x%lx with 0x%lx (expected E_INVALIDARG and oxdeadbeef)\n",
            hr, refcount);
    }
}

Referenced by START_TEST().

test_SHGetThreadRef()

static void test_SHGetThreadRef ( void  )

static

Definition at line 176 of file thread.c.

{
    IUnknown *punk;
    HRESULT hr;
 
    /* Not present before IE 5 */
    if (!pSHGetThreadRef) {
        win_skip("SHGetThreadRef not found\n");
        return;
    }
 
    punk = NULL;
    hr = pSHGetThreadRef(&punk);
    ok( (hr == E_NOINTERFACE) && (punk == NULL),
        "got 0x%lx and %p (expected E_NOINTERFACE and NULL)\n", hr, punk);
 
    if (0) {
        /* this crash on Windows */
        pSHGetThreadRef(NULL);
    }
}

Referenced by START_TEST().

test_SHSetThreadRef()

static void test_SHSetThreadRef ( void  )

static

Definition at line 198 of file thread.c.

{
    threadref ref;
    IUnknown *punk;
    HRESULT hr;
    LONG refcount;
 
    /* Not present before IE 5 */
    if (!pSHSetThreadRef) {
        win_skip("SHSetThreadRef not found\n");
        return;
    }
 
    /* start with a clean state */
    hr = pSHSetThreadRef(NULL);
    ok(hr == S_OK, "got 0x%lx (expected S_OK)\n", hr);
 
    /* build and set out object */
    init_threadref(&ref, &refcount);
    AddRef_called = 0;
    refcount = 1;
    hr = pSHSetThreadRef(&ref.IUnknown_iface);
    ok( (hr == S_OK) && (refcount == 1) && (!AddRef_called),
        "got 0x%lx with %ld, %ld (expected S_OK with 1, 0)\n",
        hr, refcount, AddRef_called);
 
    /* read back our object */
    AddRef_called = 0;
    refcount = 1;
    punk = NULL;
    hr = pSHGetThreadRef(&punk);
    ok( (hr == S_OK) && (punk == &ref.IUnknown_iface) && (refcount == 2) && (AddRef_called == 1),
        "got 0x%lx and %p with %ld, %ld (expected S_OK and %p with 2, 1)\n",
        hr, punk, refcount, AddRef_called, &ref);
 
    /* clear the object pointer */
    hr = pSHSetThreadRef(NULL);
    ok(hr == S_OK, "got 0x%lx (expected S_OK)\n", hr);
 
    /* verify, that our object is no longer known as ThreadRef */
    hr = pSHGetThreadRef(&punk);
    ok( (hr == E_NOINTERFACE) && (punk == NULL),
        "got 0x%lx and %p (expected E_NOINTERFACE and NULL)\n", hr, punk);
 
}

Referenced by START_TEST().

threadref_AddRef()

static ULONG WINAPI threadref_AddRef ( IUnknown *  iface )

static

Definition at line 60 of file thread.c.

{
    threadref * This = impl_from_IUnknown(iface);
 
    AddRef_called++;
    return InterlockedIncrement(This->ref);
}

threadref_QueryInterface()

static HRESULT WINAPI threadref_QueryInterface ( IUnknown *  iface, REFIID  riid, LPVOID *  ppvObj  )

static

Definition at line 51 of file thread.c.

{
    threadref * This = impl_from_IUnknown(iface);
 
    trace("unexpected QueryInterface(%p, %s, %p) called\n", This, wine_dbgstr_guid(riid), ppvObj);
    *ppvObj = NULL;
    return E_NOINTERFACE;
}

threadref_Release()

static ULONG WINAPI threadref_Release ( IUnknown *  iface )

static

Definition at line 68 of file thread.c.

{
    threadref * This = impl_from_IUnknown(iface);
 
    trace("unexpected Release(%p) called\n", This);
    return InterlockedDecrement(This->ref);
}

Variable Documentation

AddRef_called

DWORD AddRef_called

static

Definition at line 38 of file thread.c.

Referenced by test_SHSetThreadRef(), and threadref_AddRef().

threadref_vt

const IUnknownVtbl threadref_vt

static

Initial value:

=
{
  threadref_QueryInterface,
  threadref_AddRef,
  threadref_Release
}

Definition at line 77 of file thread.c.