Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-10-01 06:57:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[KMTESTS:KE]

Browse source 
- Add a small stress test for scheduling behavior after event set/wake
CORE-10212

svn path=/trunk/; revision=69283
This commit is contained in:
Thomas Faber 2015-09-19 12:57:51 +00:00
parent 8b7a8bb461
commit 7a7073bedc
1 changed files with 126 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

126
rostests/kmtests/ntos_ke/KeEvent.c
View file

@ -222,8 +222,131 @@ TestEventConcurrent(
}
}
#define NUM_SCHED_TESTS 1000
typedef struct
{
KEVENT Event;
KEVENT WaitEvent;
ULONG Counter;
KPRIORITY PriorityIncrement;
ULONG CounterValues[NUM_SCHED_TESTS];
} COUNT_THREAD_DATA, *PCOUNT_THREAD_DATA;
static
VOID
NTAPI
CountThread(
IN OUT PVOID Context)
{
PCOUNT_THREAD_DATA ThreadData = Context;
PKEVENT Event = &ThreadData->Event;
volatile ULONG *Counter = &ThreadData->Counter;
ULONG *CounterValue = ThreadData->CounterValues;
KPRIORITY Priority;
Priority = KeQueryPriorityThread(KeGetCurrentThread());
ok_eq_long(Priority, 8L);
while (CounterValue < &ThreadData->CounterValues[NUM_SCHED_TESTS])
{
KeSetEvent(&ThreadData->WaitEvent, IO_NO_INCREMENT, TRUE);
KeWaitForSingleObject(Event, Executive, KernelMode, FALSE, NULL);
*CounterValue++ = *Counter;
}
Priority = KeQueryPriorityThread(KeGetCurrentThread());
ok_eq_long(Priority, 8L + min(ThreadData->PriorityIncrement, 7));
}
static
VOID
NTAPI
TestEventScheduling(
_In_ PVOID Context)
{
PCOUNT_THREAD_DATA ThreadData;
PKTHREAD Thread;
NTSTATUS Status;
LONG PreviousState;
ULONG i;
volatile ULONG *Counter;
KPRIORITY PriorityIncrement;
KPRIORITY Priority;
UNREFERENCED_PARAMETER(Context);
ThreadData = ExAllocatePoolWithTag(PagedPool, sizeof(*ThreadData), 'CEmK');
if (skip(ThreadData != NULL, "Out of memory\n"))
{
return;
}
KeInitializeEvent(&ThreadData->Event, SynchronizationEvent, FALSE);
KeInitializeEvent(&ThreadData->WaitEvent, SynchronizationEvent, FALSE);
Counter = &ThreadData->Counter;
for (PriorityIncrement = 0; PriorityIncrement <= 8; PriorityIncrement++)
{
ThreadData->PriorityIncrement = PriorityIncrement;
ThreadData->Counter = 0;
RtlFillMemory(ThreadData->CounterValues,
sizeof(ThreadData->CounterValues),
0xFE);
Thread = KmtStartThread(CountThread, ThreadData);
Priority = KeQueryPriorityThread(KeGetCurrentThread());
ok(Priority == 8, "[%lu] Priority = %lu\n", PriorityIncrement, Priority);
for (i = 1; i <= NUM_SCHED_TESTS; i++)
{
Status = KeWaitForSingleObject(&ThreadData->WaitEvent, Executive, KernelMode, FALSE, NULL);
ok_eq_hex(Status, STATUS_SUCCESS);
PreviousState = KeSetEvent(&ThreadData->Event, PriorityIncrement, FALSE);
*Counter = i;
ok_eq_long(PreviousState, 0L);
}
Priority = KeQueryPriorityThread(KeGetCurrentThread());
ok(Priority == 8, "[%lu] Priority = %lu\n", PriorityIncrement, Priority);
KmtFinishThread(Thread, NULL);
if (PriorityIncrement == 0)
{
/* Both threads have the same priority, so either can win the race */
ok(ThreadData->CounterValues[0] == 0 || ThreadData->CounterValues[0] == 1,
"[%lu] Counter 0 = %lu\n",
PriorityIncrement, ThreadData->CounterValues[0]);
}
else
{
/* CountThread has the higher priority, it will always win */
ok(ThreadData->CounterValues[0] == 0,
"[%lu] Counter 0 = %lu\n",
PriorityIncrement, ThreadData->CounterValues[0]);
}
for (i = 1; i < NUM_SCHED_TESTS; i++)
{
if (PriorityIncrement == 0)
{
ok(ThreadData->CounterValues[i] == i ||
ThreadData->CounterValues[i] == i + 1,
"[%lu] Counter %lu = %lu, expected %lu or %lu\n",
PriorityIncrement, i,
ThreadData->CounterValues[i], i, i + 1);
}
else
{
ok(ThreadData->CounterValues[i] == ThreadData->CounterValues[i - 1] + 1,
"[%lu] Counter %lu = %lu, expected %lu\n",
PriorityIncrement, i,
ThreadData->CounterValues[i], ThreadData->CounterValues[i - 1] + 1);
}
}
}
ExFreePoolWithTag(ThreadData, 'CEmK');
}
START_TEST(KeEvent)
{
PKTHREAD Thread;
KEVENT Event;
KIRQL Irql;
KIRQL Irqls[] = { PASSIVE_LEVEL, APC_LEVEL, DISPATCH_LEVEL };
@ -264,4 +387,7 @@ START_TEST(KeEvent)
ok_irql(PASSIVE_LEVEL);
KmtSetIrql(PASSIVE_LEVEL);
Thread = KmtStartThread(TestEventScheduling, NULL);
KmtFinishThread(Thread, NULL);
}