ExResource.c File Reference

#include <kmt_test.h>
#include <debug.h>

Include dependency graph for ExResource.c:

Go to the source code of this file.

Classes
struct	THREAD_DATA

Macros
#define	CheckResourceFields(Res, Reinit)
#define	CheckResourceStatus(Res, Exclusive, Shared, ExclusiveWaiters, SharedWaiters)

Typedefs
typedef BOOLEAN(NTAPI *	PACQUIRE_FUNCTION) (PERESOURCE, BOOLEAN)
typedef struct THREAD_DATA *	PTHREAD_DATA

Functions
static	_IRQL_requires_max_ (APC_LEVEL)
static VOID	TestResourceSharedAccess (IN PERESOURCE Res)
static VOID	TestResourceExclusiveAccess (IN PERESOURCE Res)
static VOID	TestResourceUndocumentedShortcuts (IN PERESOURCE Res, IN BOOLEAN AreApcsDisabled)
static VOID NTAPI	AcquireResourceThread (PVOID Context)
static VOID	InitThreadDataEx (PTHREAD_DATA ThreadData, PERESOURCE Res, PACQUIRE_FUNCTION AcquireFunction, PKSTART_ROUTINE StartRoutine)
static VOID	InitThreadData (PTHREAD_DATA ThreadData, PERESOURCE Res, PACQUIRE_FUNCTION AcquireFunction)
static NTSTATUS	StartThread (PTHREAD_DATA ThreadData, PLARGE_INTEGER Timeout, BOOLEAN Wait, BOOLEAN RetExpected)
static VOID	FinishThread (PTHREAD_DATA ThreadData)
static VOID	TestResourceWithThreads (IN PERESOURCE Res)
static VOID NTAPI	TestOwnerRes (PVOID Context)
static VOID	TestResourceWithOwner (IN PERESOURCE Res)
	START_TEST (ExResource)

Variables
	ERESOURCE_2K3
*	PERESOURCE_2K3

Macro Definition Documentation

CheckResourceFields

#define CheckResourceFields	(		Res,
			Reinit
	)

Value:

do                                                                     \
{                                                                                                               \
    ok_eq_pointer((Res)->SystemResourcesList.Flink->Blink, &(Res)->SystemResourcesList);                        \
    ok_eq_pointer((Res)->SystemResourcesList.Blink->Flink, &(Res)->SystemResourcesList);                        \
    if (!Reinit) ok_eq_pointer((Res)->OwnerTable, NULL);                                                        \
    ok_eq_int((Res)->ActiveCount, 0);                                                                           \
    ok_eq_uint((Res)->Flag, 0);                                                                                 \
    if (!Reinit) ok_eq_pointer((Res)->SharedWaiters, NULL);                                                     \
    if (!Reinit) ok_eq_pointer((Res)->ExclusiveWaiters, NULL);                                                  \
    ok_eq_ulongptr((Res)->OwnerThreads[0].OwnerThread, 0);                                                      \
    ok_eq_ulong((Res)->OwnerThreads[0].TableSize, 0LU);                                                         \
    ok_eq_ulongptr((Res)->OwnerThreads[1].OwnerThread, 0);                                                      \
    ok_eq_ulong((Res)->OwnerThreads[1].TableSize, 0LU);                                                         \
    ok_eq_ulong((Res)->ContentionCount, 0LU);                                                                   \
    ok_eq_uint((Res)->NumberOfSharedWaiters, 0);                                                                \
    ok_eq_uint((Res)->NumberOfExclusiveWaiters, 0);                                                             \
    ok_eq_pointer((Res)->Address, NULL);                                                                        \
    ok_eq_ulongptr((Res)->SpinLock, 0);                                                                         \
} while (0)

Definition at line 78 of file ExResource.c.

CheckResourceStatus

#define CheckResourceStatus	(		Res,
			Exclusive,
			Shared,
			ExclusiveWaiters,
			SharedWaiters
	)

Value:

do                         \
{                                                                                                               \
    if (Exclusive)                                                                                              \
        ok_bool_true(ExIsResourceAcquiredExclusiveLite(Res), "ExIsResourceAcquiredExclusiveLite returned");     \
    else                                                                                                        \
        ok_bool_false(ExIsResourceAcquiredExclusiveLite(Res), "ExIsResourceAcquiredExclusiveLite returned");    \
    ok_eq_ulong(ExIsResourceAcquiredSharedLite(Res), Shared);                                                   \
    ok_eq_ulong(ExGetExclusiveWaiterCount(Res), ExclusiveWaiters);                                              \
    ok_eq_ulong(ExGetSharedWaiterCount(Res), SharedWaiters);                                                    \
} while (0)

Definition at line 98 of file ExResource.c.

Typedef Documentation

PACQUIRE_FUNCTION

typedef BOOLEAN(NTAPI * PACQUIRE_FUNCTION) (PERESOURCE, BOOLEAN)

Definition at line 263 of file ExResource.c.

PTHREAD_DATA

typedef struct THREAD_DATA * PTHREAD_DATA

Function Documentation

_IRQL_requires_max_()

static _IRQL_requires_max_ ( APC_LEVEL  )

static

Definition at line 14 of file ExResource.c.

                              {
  LIST_ENTRY SystemResourcesList;
  POWNER_ENTRY OwnerTable;
  SHORT ActiveCount;
  USHORT Flag;
  volatile PKSEMAPHORE SharedWaiters;
  volatile PKEVENT ExclusiveWaiters;
  OWNER_ENTRY OwnerThreads[2];
  ULONG ContentionCount;
  USHORT NumberOfSharedWaiters;
  USHORT NumberOfExclusiveWaiters;
  _ANONYMOUS_UNION union {
    PVOID Address;
    ULONG_PTR CreatorBackTraceIndex;
  } DUMMYUNIONNAME;
  KSPIN_LOCK SpinLock;
} ERESOURCE_2K3, *PERESOURCE_2K3;

AcquireResourceThread()

static VOID NTAPI AcquireResourceThread ( PVOID  Context )

static

Definition at line 281 of file ExResource.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    PTHREAD_DATA ThreadData = Context;
    BOOLEAN Ret;
 
    KeEnterCriticalRegion();
    Ret = ThreadData->AcquireResource(ThreadData->Res, ThreadData->Wait);
    if (ThreadData->RetExpected)
        ok_bool_true(Ret, "AcquireResource returned");
    else
        ok_bool_false(Ret, "AcquireResource returned");
 
    ok_bool_false(KeSetEvent(&ThreadData->OutEvent, 0, TRUE), "KeSetEvent returned");
    Status = KeWaitForSingleObject(&ThreadData->InEvent, Executive, KernelMode, FALSE, NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    if (Ret)
        ExReleaseResource(ThreadData->Res);
    KeLeaveCriticalRegion();
}

Referenced by InitThreadData().

FinishThread()

static VOID FinishThread ( PTHREAD_DATA  ThreadData )

static

Definition at line 353 of file ExResource.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
 
    KeSetEvent(&ThreadData->InEvent, 0, TRUE);
    Status = KeWaitForSingleObject(ThreadData->Thread, Executive, KernelMode, FALSE, NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    ObDereferenceObject(ThreadData->Thread);
    Status = ZwClose(ThreadData->Handle);
    ok_eq_hex(Status, STATUS_SUCCESS);
    KeClearEvent(&ThreadData->InEvent);
    KeClearEvent(&ThreadData->OutEvent);
}

Referenced by TestResourceWithOwner(), and TestResourceWithThreads().

InitThreadData()

static VOID InitThreadData ( PTHREAD_DATA  ThreadData, PERESOURCE  Res, PACQUIRE_FUNCTION  AcquireFunction  )

static

Definition at line 321 of file ExResource.c.

{
    InitThreadDataEx(ThreadData, Res, AcquireFunction, AcquireResourceThread);
}

Referenced by TestResourceWithThreads().

InitThreadDataEx()

static VOID InitThreadDataEx ( PTHREAD_DATA  ThreadData, PERESOURCE  Res, PACQUIRE_FUNCTION  AcquireFunction, PKSTART_ROUTINE  StartRoutine  )

static

Definition at line 306 of file ExResource.c.

{
    ThreadData->Res = Res;
    KeInitializeEvent(&ThreadData->InEvent, NotificationEvent, FALSE);
    KeInitializeEvent(&ThreadData->OutEvent, NotificationEvent, FALSE);
    ThreadData->AcquireResource = AcquireFunction;
    ThreadData->StartRoutine = StartRoutine;
}

Referenced by InitThreadData(), and TestResourceWithOwner().

START_TEST()

START_TEST

(

ExResource

)

Definition at line 521 of file ExResource.c.

{
    NTSTATUS Status;
    ERESOURCE Res;
    KIRQL Irql;
 
    pExEnterCriticalRegionAndAcquireResourceShared = KmtGetSystemRoutineAddress(L"ExEnterCriticalRegionAndAcquireResourceShared");
    pExEnterCriticalRegionAndAcquireSharedWaitForExclusive = KmtGetSystemRoutineAddress(L"ExEnterCriticalRegionAndAcquireSharedWaitForExclusive");
    pExEnterCriticalRegionAndAcquireResourceExclusive = KmtGetSystemRoutineAddress(L"ExEnterCriticalRegionAndAcquireResourceExclusive");
    pExReleaseResourceAndLeaveCriticalRegion = KmtGetSystemRoutineAddress(L"ExReleaseResourceAndLeaveCriticalRegion");
    pKeAreAllApcsDisabled = KmtGetSystemRoutineAddress(L"KeAreAllApcsDisabled");
 
    if (skip(pKeAreAllApcsDisabled != NULL, "KeAreAllApcsDisabled unavailable\n"))
    {
        /* We can live without this function here */
    }
 
    /* this must be true even with the different structure versions */
    ASSERT(sizeof(ERESOURCE) == sizeof(ERESOURCE_2K3));
 
    /* functional tests & internals */
    Irql = KeRaiseIrqlToDpcLevel();
      Status = ExInitializeResourceLite(&Res);
      ok_eq_hex(Status, STATUS_SUCCESS);
    KeLowerIrql(APC_LEVEL);
 
      Status = ExDeleteResourceLite(&Res);
      ok_eq_hex(Status, STATUS_SUCCESS);
    KeLowerIrql(Irql);
 
    memset(&Res, 0x55, sizeof Res);
    Status = ExInitializeResourceLite(&Res);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceFields((PERESOURCE_2K3)&Res, FALSE);
 
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
 
    TestResourceSharedAccess(&Res);
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
 
    TestResourceExclusiveAccess(&Res);
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
 
    TestResourceUndocumentedShortcuts(&Res, FALSE);
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
    KeRaiseIrql(APC_LEVEL, &Irql);
      TestResourceUndocumentedShortcuts(&Res, TRUE);
    KeLowerIrql(Irql);
    ok_bool_false(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
 
    TestResourceWithThreads(&Res);
 
    /* ExReinitializeResourceLite cleans up after us */
    Status = ExReinitializeResourceLite(&Res);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceFields((PERESOURCE_2K3)&Res, TRUE);
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
 
    TestResourceWithOwner(&Res);
    CheckResourceStatus(&Res, FALSE, 0LU, 0LU, 0LU);
 
    Status = ExDeleteResourceLite(&Res);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    /* parameter checks */
    Status = STATUS_SUCCESS;
    _SEH2_TRY {
        ExInitializeResourceLite(NULL);
    } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
        Status = _SEH2_GetExceptionCode();
    } _SEH2_END;
    ok_eq_hex(Status, STATUS_ACCESS_VIOLATION);
 
    /* these bugcheck
    ExDeleteResourceLite(NULL);
    Status = ExDeleteResourceLite(&Res);*/
}

StartThread()

static NTSTATUS StartThread ( PTHREAD_DATA  ThreadData, PLARGE_INTEGER  Timeout, BOOLEAN  Wait, BOOLEAN  RetExpected  )

static

Definition at line 331 of file ExResource.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    OBJECT_ATTRIBUTES Attributes;
 
    ThreadData->Wait = Wait;
    ThreadData->RetExpected = RetExpected;
    InitializeObjectAttributes(&Attributes, NULL, OBJ_KERNEL_HANDLE, NULL, NULL);
    Status = PsCreateSystemThread(&ThreadData->Handle, GENERIC_ALL, &Attributes, NULL, NULL, ThreadData->StartRoutine, ThreadData);
    ok_eq_hex(Status, STATUS_SUCCESS);
    Status = ObReferenceObjectByHandle(ThreadData->Handle, SYNCHRONIZE, *PsThreadType, KernelMode, (PVOID *)&ThreadData->Thread, NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    return KeWaitForSingleObject(&ThreadData->OutEvent, Executive, KernelMode, FALSE, Timeout);
}

Referenced by TestResourceWithOwner(), and TestResourceWithThreads().

TestOwnerRes()

static VOID NTAPI TestOwnerRes ( PVOID  Context )

static

Definition at line 474 of file ExResource.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    PTHREAD_DATA ThreadData = Context;
    BOOLEAN Ret;
 
    KeEnterCriticalRegion();
    Ret = ThreadData->AcquireResource(ThreadData->Res, ThreadData->Wait);
    if (ThreadData->RetExpected)
        ok_bool_true(Ret, "AcquireResource returned");
    else
        ok_bool_false(Ret, "AcquireResource returned");
    KeLeaveCriticalRegion();
 
    ExReleaseResourceForThreadLite(ThreadData->Res, (ULONG_PTR)ThreadData->Res | 3);
 
    ok_bool_false(KeSetEvent(&ThreadData->OutEvent, 0, TRUE), "KeSetEvent returned");
    Status = KeWaitForSingleObject(&ThreadData->InEvent, Executive, KernelMode, FALSE, NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
}

Referenced by TestResourceWithOwner().

TestResourceExclusiveAccess()

static VOID TestResourceExclusiveAccess ( IN PERESOURCE  Res )

static

Definition at line 144 of file ExResource.c.

{
    LONG Count = 0;
 
    KeEnterCriticalRegion();
    ok_bool_true(ExAcquireResourceExclusiveLite(Res, FALSE), "ExAcquireResourceExclusiveLite returned"); ++Count;
 
    CheckResourceStatus(Res, TRUE, Count, 0LU, 0LU);
 
    ok_bool_true(ExAcquireResourceExclusiveLite(Res, TRUE), "ExAcquireResourceExclusiveLite returned"); ++Count;
    CheckResourceStatus(Res, TRUE, Count, 0LU, 0LU);
 
    ok_bool_true(ExAcquireResourceSharedLite(Res, FALSE), "ExAcquireResourceSharedLite returned"); ++Count;
    ok_bool_true(ExAcquireResourceSharedLite(Res, TRUE), "ExAcquireResourceSharedLite returned"); ++Count;
    ok_bool_true(ExAcquireSharedStarveExclusive(Res, FALSE), "ExAcquireSharedStarveExclusive returned"); ++Count;
    ok_bool_true(ExAcquireSharedStarveExclusive(Res, TRUE), "ExAcquireSharedStarveExclusive returned"); ++Count;
    ok_bool_true(ExAcquireSharedWaitForExclusive(Res, FALSE), "ExAcquireSharedWaitForExclusive returned"); ++Count;
    ok_bool_true(ExAcquireSharedWaitForExclusive(Res, TRUE), "ExAcquireSharedWaitForExclusive returned"); ++Count;
    CheckResourceStatus(Res, TRUE, Count, 0LU, 0LU);
 
    ExConvertExclusiveToSharedLite(Res);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    while (Count--)
        ExReleaseResourceLite(Res);
    KeLeaveCriticalRegion();
}

Referenced by START_TEST().

TestResourceSharedAccess()

static VOID TestResourceSharedAccess ( IN PERESOURCE  Res )

static

Definition at line 111 of file ExResource.c.

{
    LONG Count = 0;
 
    KeEnterCriticalRegion();
    ok_bool_true(ExAcquireResourceSharedLite(Res, FALSE), "ExAcquireResourceSharedLite returned"); ++Count;
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    ok_bool_true(ExAcquireResourceSharedLite(Res, FALSE), "ExAcquireResourceSharedLite returned"); ++Count;
    ok_bool_true(ExAcquireResourceSharedLite(Res, TRUE), "ExAcquireResourceSharedLite returned"); ++Count;
    ok_bool_true(ExAcquireSharedStarveExclusive(Res, FALSE), "ExAcquireSharedStarveExclusive returned"); ++Count;
    ok_bool_true(ExAcquireSharedStarveExclusive(Res, TRUE), "ExAcquireSharedStarveExclusive returned"); ++Count;
    ok_bool_true(ExAcquireSharedWaitForExclusive(Res, FALSE), "ExAcquireSharedWaitForExclusive returned"); ++Count;
    ok_bool_true(ExAcquireSharedWaitForExclusive(Res, TRUE), "ExAcquireSharedWaitForExclusive returned"); ++Count;
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    /* this one fails, TRUE would deadlock */
    ok_bool_false(ExAcquireResourceExclusiveLite(Res, FALSE), "ExAcquireResourceExclusiveLite returned");
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    /* this asserts */
    if (!KmtIsCheckedBuild)
        ExConvertExclusiveToSharedLite(Res);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    while (Count--)
        ExReleaseResourceLite(Res);
    KeLeaveCriticalRegion();
}

Referenced by START_TEST().

TestResourceUndocumentedShortcuts()

static VOID TestResourceUndocumentedShortcuts ( IN PERESOURCE  Res, IN BOOLEAN  AreApcsDisabled  )

static

Definition at line 175 of file ExResource.c.

{
    PVOID Ret;
    LONG Count = 0;
 
    ok_bool_false(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_uint(pKeAreAllApcsDisabled(), AreApcsDisabled);
 
    if (skip(pExEnterCriticalRegionAndAcquireResourceShared &&
             pExEnterCriticalRegionAndAcquireSharedWaitForExclusive &&
             pExEnterCriticalRegionAndAcquireResourceExclusive &&
             pExReleaseResourceAndLeaveCriticalRegion, "No shortcuts\n"))
    {
        return;
    }
    /* ExEnterCriticalRegionAndAcquireResourceShared, ExEnterCriticalRegionAndAcquireSharedWaitForExclusive */
    Count = 0;
    Ret = pExEnterCriticalRegionAndAcquireResourceShared(Res); ++Count;
    ok_eq_pointer(Ret, KeGetCurrentThread()->Win32Thread);
    ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    Ret = pExEnterCriticalRegionAndAcquireResourceShared(Res); ++Count;
    ok_eq_pointer(Ret, KeGetCurrentThread()->Win32Thread);
    ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    pExEnterCriticalRegionAndAcquireSharedWaitForExclusive(Res); ++Count;
    ok_eq_pointer(Ret, KeGetCurrentThread()->Win32Thread);
    ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    while (Count-- > 1)
    {
        pExReleaseResourceAndLeaveCriticalRegion(Res);
        ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
        if (pKeAreAllApcsDisabled)
            ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
        CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
    }
 
    pExReleaseResourceAndLeaveCriticalRegion(Res);
    ok_bool_false(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
 
    /* ExEnterCriticalRegionAndAcquireResourceExclusive */
    Count = 0;
    ok_bool_false(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    Ret = pExEnterCriticalRegionAndAcquireResourceExclusive(Res); ++Count;
    ok_eq_pointer(Ret, KeGetCurrentThread()->Win32Thread);
    ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, TRUE, Count, 0LU, 0LU);
 
    Ret = pExEnterCriticalRegionAndAcquireResourceExclusive(Res); ++Count;
    ok_eq_pointer(Ret, KeGetCurrentThread()->Win32Thread);
    ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, TRUE, Count, 0LU, 0LU);
 
    pExReleaseResourceAndLeaveCriticalRegion(Res); --Count;
    ok_bool_true(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_bool(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, TRUE, Count, 0LU, 0LU);
 
    pExReleaseResourceAndLeaveCriticalRegion(Res); --Count;
    ok_bool_false(KeAreApcsDisabled(), "KeAreApcsDisabled returned");
    if (pKeAreAllApcsDisabled)
        ok_eq_uint(pKeAreAllApcsDisabled(), AreApcsDisabled);
    CheckResourceStatus(Res, FALSE, Count, 0LU, 0LU);
}

Referenced by START_TEST().

TestResourceWithOwner()

static VOID TestResourceWithOwner ( IN PERESOURCE  Res )

static

Definition at line 498 of file ExResource.c.

{
    NTSTATUS Status;
    THREAD_DATA ThreadDataOwner;
 
    InitThreadDataEx(&ThreadDataOwner, Res, ExAcquireResourceExclusiveLite, TestOwnerRes);
 
    KeEnterCriticalRegion();
    ok_bool_true(ExAcquireResourceExclusiveLite(Res, FALSE), "ExAcquireResourceExclusiveLite returned");
    ExSetResourceOwnerPointer(Res, (PVOID)(ULONG_PTR)3);
    ExReleaseResourceForThreadLite(Res, 3);
 
    ok_bool_true(ExAcquireResourceExclusiveLite(Res, FALSE), "ExAcquireResourceExclusiveLite returned");
    ExSetResourceOwnerPointer(Res, (PVOID)((ULONG_PTR)Res | 3));
    KeLeaveCriticalRegion();
 
    Status = StartThread(&ThreadDataOwner, NULL, FALSE, FALSE);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    FinishThread(&ThreadDataOwner);
}

Referenced by START_TEST().

TestResourceWithThreads()

static VOID TestResourceWithThreads ( IN PERESOURCE  Res )

static

Definition at line 371 of file ExResource.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    THREAD_DATA ThreadDataShared;
    THREAD_DATA ThreadDataShared2;
    THREAD_DATA ThreadDataExclusive;
    THREAD_DATA ThreadDataSharedStarve;
    THREAD_DATA ThreadDataSharedWait;
    LARGE_INTEGER Timeout;
    Timeout.QuadPart = -10 * 1000 * 10; /* 10 ms */
 
    InitThreadData(&ThreadDataShared, Res, ExAcquireResourceSharedLite);
    InitThreadData(&ThreadDataShared2, Res, ExAcquireResourceSharedLite);
    InitThreadData(&ThreadDataExclusive, Res, ExAcquireResourceExclusiveLite);
    InitThreadData(&ThreadDataSharedStarve, Res, ExAcquireSharedStarveExclusive);
    InitThreadData(&ThreadDataSharedWait, Res, ExAcquireSharedWaitForExclusive);
 
    /* have a thread acquire the resource shared */
    Status = StartThread(&ThreadDataShared, NULL, FALSE, TRUE);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
 
    /* a second thread should be able to acquire the resource shared */
    Status = StartThread(&ThreadDataShared2, NULL, FALSE, TRUE);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 2);
    FinishThread(&ThreadDataShared2);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
 
    /* now have a thread that tries to acquire the resource exclusive -- it should fail */
    Status = StartThread(&ThreadDataExclusive, NULL, FALSE, FALSE);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
    FinishThread(&ThreadDataExclusive);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
 
    /* as above, but this time it should block */
    Status = StartThread(&ThreadDataExclusive, &Timeout, TRUE, TRUE);
    ok_eq_hex(Status, STATUS_TIMEOUT);
    CheckResourceStatus(Res, FALSE, 0LU, 1LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
 
    /* now try another shared one -- it should fail */
    Status = StartThread(&ThreadDataShared2, NULL, FALSE, FALSE);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 1LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
    FinishThread(&ThreadDataShared2);
 
    /* same for ExAcquireSharedWaitForExclusive */
    Status = StartThread(&ThreadDataSharedWait, NULL, FALSE, FALSE);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 1LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
    FinishThread(&ThreadDataSharedWait);
 
    /* ExAcquireSharedStarveExclusive must get access though! */
    Status = StartThread(&ThreadDataSharedStarve, NULL, TRUE, TRUE);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 1LU, 0LU);
    ok_eq_int(Res->ActiveCount, 2);
    FinishThread(&ThreadDataSharedStarve);
    CheckResourceStatus(Res, FALSE, 0LU, 1LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
 
    /* block another shared one */
    Status = StartThread(&ThreadDataShared2, &Timeout, TRUE, TRUE);
    ok_eq_hex(Status, STATUS_TIMEOUT);
    CheckResourceStatus(Res, FALSE, 0LU, 1LU, 1LU);
    ok_eq_int(Res->ActiveCount, 1);
 
    /* finish the very first one */
    FinishThread(&ThreadDataShared);
 
    /* now the blocked exclusive one should get the resource */
    Status = KeWaitForSingleObject(&ThreadDataExclusive.OutEvent, Executive, KernelMode, FALSE, NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 1LU);
    ok_eq_int(Res->ActiveCount, 1);
    ok_eq_uint((Res->Flag & ResourceOwnedExclusive) != 0, 1);
 
    FinishThread(&ThreadDataExclusive);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
 
    /* now the blocked shared one should resume */
    Status = KeWaitForSingleObject(&ThreadDataShared2.OutEvent, Executive, KernelMode, FALSE, NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 1);
    FinishThread(&ThreadDataShared2);
    CheckResourceStatus(Res, FALSE, 0LU, 0LU, 0LU);
    ok_eq_int(Res->ActiveCount, 0);
}

Referenced by START_TEST().

Variable Documentation

ERESOURCE_2K3

ERESOURCE_2K3

Definition at line 76 of file ExResource.c.

Referenced by START_TEST().

PERESOURCE_2K3

* PERESOURCE_2K3

Definition at line 76 of file ExResource.c.

Referenced by START_TEST().