Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-06-10 20:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

[KMTESTS:KE]

Browse source 
- Add tests for acquiring mutants recursively
CORE-10218

svn path=/trunk/; revision=69300
This commit is contained in:
Thomas Faber 2015-09-19 23:59:02 +00:00
parent 97f2b6b128
commit e5d4c337e6
1 changed files with 193 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

209
rostests/kmtests/ntos_ke/KeMutex.c
View file

@ -21,7 +21,7 @@ C_ASSERT(sizeof(DISPATCHER_HEADER) == 8 + 2 * sizeof(PVOID));
C_ASSERT(sizeof(KMUTANT) == sizeof(DISPATCHER_HEADER) + 3 * sizeof(PVOID) + sizeof(PVOID)); C_ASSERT(sizeof(KMUTANT) == sizeof(DISPATCHER_HEADER) + 3 * sizeof(PVOID) + sizeof(PVOID));
C_ASSERT(sizeof(KMUTANT) == MUTANT_SIZE * sizeof(ULONG)); C_ASSERT(sizeof(KMUTANT) == MUTANT_SIZE * sizeof(ULONG));
#define CheckMutex(Mutex, Held, New, ExpectedApcDisable) do { \ #define CheckMutex(Mutex, State, New, ExpectedApcDisable) do { \
PKTHREAD Thread = KeGetCurrentThread(); \ PKTHREAD Thread = KeGetCurrentThread(); \
ok_eq_uint((Mutex)->Header.Type, MutantObject); \ ok_eq_uint((Mutex)->Header.Type, MutantObject); \
ok_eq_uint((Mutex)->Header.Abandoned, 0x55); \ ok_eq_uint((Mutex)->Header.Abandoned, 0x55); \
@ -31,9 +31,9 @@ C_ASSERT(sizeof(KMUTANT) == MUTANT_SIZE * sizeof(ULONG));
&(Mutex)->Header.WaitListHead); \ &(Mutex)->Header.WaitListHead); \
ok_eq_pointer((Mutex)->Header.WaitListHead.Blink, \ ok_eq_pointer((Mutex)->Header.WaitListHead.Blink, \
&(Mutex)->Header.WaitListHead); \ &(Mutex)->Header.WaitListHead); \
if (Held) \ if ((State) <= 0) \
{ \ { \
ok_eq_long((Mutex)->Header.SignalState, 0); \ ok_eq_long((Mutex)->Header.SignalState, State); \
ok_eq_pointer((Mutex)->MutantListEntry.Flink, &Thread->MutantListHead); \ ok_eq_pointer((Mutex)->MutantListEntry.Flink, &Thread->MutantListHead); \
ok_eq_pointer((Mutex)->MutantListEntry.Blink, &Thread->MutantListHead); \ ok_eq_pointer((Mutex)->MutantListEntry.Blink, &Thread->MutantListHead); \
ok_eq_pointer(Thread->MutantListHead.Flink, &(Mutex)->MutantListEntry); \ ok_eq_pointer(Thread->MutantListHead.Flink, &(Mutex)->MutantListEntry); \
@ -42,7 +42,7 @@ C_ASSERT(sizeof(KMUTANT) == MUTANT_SIZE * sizeof(ULONG));
} \ } \
else \ else \
{ \ { \
ok_eq_long((Mutex)->Header.SignalState, 1); \ ok_eq_long((Mutex)->Header.SignalState, State); \
if (New) \ if (New) \
{ \ { \
ok_eq_pointer((Mutex)->MutantListEntry.Flink, \ ok_eq_pointer((Mutex)->MutantListEntry.Flink, \
@ -75,21 +75,22 @@ TestMutant(VOID)
NTSTATUS Status; NTSTATUS Status;
KMUTANT Mutant; KMUTANT Mutant;
LONG State; LONG State;
LONG i;
PKTHREAD Thread = KeGetCurrentThread(); PKTHREAD Thread = KeGetCurrentThread();
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
RtlFillMemory(&Mutant, sizeof(Mutant), 0x55); RtlFillMemory(&Mutant, sizeof(Mutant), 0x55);
KeInitializeMutant(&Mutant, FALSE); KeInitializeMutant(&Mutant, FALSE);
CheckMutex(&Mutant, FALSE, TRUE, 0); CheckMutex(&Mutant, 1L, TRUE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
RtlFillMemory(&Mutant, sizeof(Mutant), 0x55); RtlFillMemory(&Mutant, sizeof(Mutant), 0x55);
KeInitializeMutant(&Mutant, TRUE); KeInitializeMutant(&Mutant, TRUE);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
CheckMutex(&Mutant, TRUE, TRUE, 0); CheckMutex(&Mutant, 0L, TRUE, 0);
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE); State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
ok_eq_long(State, 0); ok_eq_long(State, 0L);
CheckMutex(&Mutant, FALSE, FALSE, 0); CheckMutex(&Mutant, 1L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Acquire and release */ /* Acquire and release */
@ -99,12 +100,99 @@ TestMutant(VOID)
FALSE, FALSE,
NULL); NULL);
ok_eq_hex(Status, STATUS_SUCCESS); ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutant, TRUE, TRUE, 0); CheckMutex(&Mutant, 0L, TRUE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE); State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
ok_eq_long(State, 0); ok_eq_long(State, 0L);
CheckMutex(&Mutant, FALSE, FALSE, 0); CheckMutex(&Mutant, 1L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Acquire recursively */
for (i = 0; i < 8; i++)
{
KmtStartSeh()
Status = KeWaitForSingleObject(&Mutant,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutant, -i, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
}
for (i = 0; i < 7; i++)
{
KmtStartSeh()
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_long(State, -7L + i);
CheckMutex(&Mutant, -6L + i, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
}
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
ok_eq_long(State, 0L);
CheckMutex(&Mutant, 1L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Pretend to acquire it recursively -MINLONG times */
KmtStartSeh()
Status = KeWaitForSingleObject(&Mutant,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutant, 0L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
Mutant.Header.SignalState = MINLONG + 1;
KmtStartSeh()
Status = KeWaitForSingleObject(&Mutant,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutant, (LONG)MINLONG, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
KmtStartSeh()
KeWaitForSingleObject(&Mutant,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_MUTANT_LIMIT_EXCEEDED);
CheckMutex(&Mutant, (LONG)MINLONG, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
ok_eq_long(State, (LONG)MINLONG);
CheckMutex(&Mutant, (LONG)MINLONG + 1L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
Mutant.Header.SignalState = -1;
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
ok_eq_long(State, -1L);
CheckMutex(&Mutant, 0L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
State = KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
ok_eq_long(State, 0L);
CheckMutex(&Mutant, 1L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Now release it once too often */
KmtStartSeh()
KeReleaseMutant(&Mutant, 1, FALSE, FALSE);
KmtEndSeh(STATUS_MUTANT_NOT_OWNED);
CheckMutex(&Mutant, 1L, FALSE, 0);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
} }
@ -115,31 +203,120 @@ TestMutex(VOID)
NTSTATUS Status; NTSTATUS Status;
KMUTEX Mutex; KMUTEX Mutex;
LONG State; LONG State;
LONG i;
PKTHREAD Thread = KeGetCurrentThread(); PKTHREAD Thread = KeGetCurrentThread();
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
RtlFillMemory(&Mutex, sizeof(Mutex), 0x55); RtlFillMemory(&Mutex, sizeof(Mutex), 0x55);
KeInitializeMutex(&Mutex, 0); KeInitializeMutex(&Mutex, 0);
CheckMutex(&Mutex, FALSE, TRUE, 1); CheckMutex(&Mutex, 1L, TRUE, 1);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
RtlFillMemory(&Mutex, sizeof(Mutex), 0x55); RtlFillMemory(&Mutex, sizeof(Mutex), 0x55);
KeInitializeMutex(&Mutex, 123); KeInitializeMutex(&Mutex, 123);
CheckMutex(&Mutex, FALSE, TRUE, 1); CheckMutex(&Mutex, 1L, TRUE, 1);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Acquire and release */
Status = KeWaitForSingleObject(&Mutex, Status = KeWaitForSingleObject(&Mutex,
Executive, Executive,
KernelMode, KernelMode,
FALSE, FALSE,
NULL); NULL);
ok_eq_hex(Status, STATUS_SUCCESS); ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutex, TRUE, FALSE, 1); CheckMutex(&Mutex, 0L, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL); CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
State = KeReleaseMutex(&Mutex, FALSE); State = KeReleaseMutex(&Mutex, FALSE);
ok_eq_long(State, 0); ok_eq_long(State, 0L);
CheckMutex(&Mutex, FALSE, FALSE, 1); CheckMutex(&Mutex, 1L, FALSE, 1);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Acquire recursively */
for (i = 0; i < 8; i++)
{
KmtStartSeh()
Status = KeWaitForSingleObject(&Mutex,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutex, -i, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
}
for (i = 0; i < 7; i++)
{
KmtStartSeh()
State = KeReleaseMutex(&Mutex, FALSE);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_long(State, -7L + i);
CheckMutex(&Mutex, -6L + i, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
}
State = KeReleaseMutex(&Mutex, FALSE);
ok_eq_long(State, 0L);
CheckMutex(&Mutex, 1L, FALSE, 1);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Pretend to acquire it recursively -MINLONG times */
KmtStartSeh()
Status = KeWaitForSingleObject(&Mutex,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutex, 0L, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
Mutex.Header.SignalState = MINLONG + 1;
KmtStartSeh()
Status = KeWaitForSingleObject(&Mutex,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_SUCCESS);
ok_eq_hex(Status, STATUS_SUCCESS);
CheckMutex(&Mutex, (LONG)MINLONG, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
KmtStartSeh()
KeWaitForSingleObject(&Mutex,
Executive,
KernelMode,
FALSE,
NULL);
KmtEndSeh(STATUS_MUTANT_LIMIT_EXCEEDED);
CheckMutex(&Mutex, (LONG)MINLONG, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
State = KeReleaseMutex(&Mutex, FALSE);
ok_eq_long(State, (LONG)MINLONG);
CheckMutex(&Mutex, (LONG)MINLONG + 1L, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
Mutex.Header.SignalState = -1;
State = KeReleaseMutex(&Mutex, FALSE);
ok_eq_long(State, -1L);
CheckMutex(&Mutex, 0L, FALSE, 1);
CheckApcs(-1, 0, FALSE, PASSIVE_LEVEL);
State = KeReleaseMutex(&Mutex, FALSE);
ok_eq_long(State, 0L);
CheckMutex(&Mutex, 1L, FALSE, 1);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
/* Now release it once too often */
KmtStartSeh()
KeReleaseMutex(&Mutex, FALSE);
KmtEndSeh(STATUS_MUTANT_NOT_OWNED);
CheckMutex(&Mutex, 1L, FALSE, 1);
CheckApcs(0, 0, FALSE, PASSIVE_LEVEL); CheckApcs(0, 0, FALSE, PASSIVE_LEVEL);
} }