https://git.reactos.org/?p=reactos.git;a=commitdiff;h=fcb156388d88eb6c0949890fde6d0efe19f00deb
commit fcb156388d88eb6c0949890fde6d0efe19f00deb Author: Bișoc George <[email protected]> AuthorDate: Thu Mar 26 12:11:29 2020 +0100 Commit: Thomas Faber <[email protected]> CommitDate: Sat Mar 28 13:10:44 2020 +0100 [NTDLL_APITEST] Add testcase for process priority class. CORE-16757 Currently the testcase function only addresses the behaviour of getting an unaligned buffer (either of correct length size or not) for now. --- .../apitests/ntdll/NtQueryInformationProcess.c | 69 ++++++++++++++++++++++ 1 file changed, 69 insertions(+) diff --git a/modules/rostests/apitests/ntdll/NtQueryInformationProcess.c b/modules/rostests/apitests/ntdll/NtQueryInformationProcess.c index bdbfc3f726a..e392ead8760 100644 --- a/modules/rostests/apitests/ntdll/NtQueryInformationProcess.c +++ b/modules/rostests/apitests/ntdll/NtQueryInformationProcess.c @@ -185,6 +185,74 @@ Test_ProcessTimes(void) #undef SPIN_TIME } +static +void +Test_ProcessPriorityClassAlignment(void) +{ + NTSTATUS Status; + PPROCESS_PRIORITY_CLASS ProcPriority; + + /* Allocate some memory for the priority class structure */ + ProcPriority = malloc(sizeof(PROCESS_PRIORITY_CLASS)); + if (ProcPriority == NULL) + { + skip("Failed to allocate memory for PROCESS_PRIORITY_CLASS!\n"); + return; + } + + /* + * Initialize the PriorityClass member to ensure the test won't randomly succeed (if such data is uninitialized). + * Filling 85 to the data member makes sure that if the test fails continously then NtQueryInformationProcess() + * didn't initialize the structure with data. + */ + RtlFillMemory(&ProcPriority->PriorityClass, sizeof(ProcPriority->PriorityClass), 0x55); + + /* Unaligned buffer -- wrong size */ + Status = NtQueryInformationProcess(NtCurrentProcess(), + ProcessPriorityClass, + (PVOID)1, + 0, + NULL); + ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH); + + /* Unaligned buffer -- correct size */ + Status = NtQueryInformationProcess(NtCurrentProcess(), + ProcessPriorityClass, + (PVOID)1, + sizeof(PROCESS_PRIORITY_CLASS), + NULL); + ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT); + + /* Unaligned buffer -- wrong size (but this time do with an alignment of 2) */ + Status = NtQueryInformationProcess(NtCurrentProcess(), + ProcessPriorityClass, + (PVOID)2, + 0, + NULL); + ok_hex(Status, STATUS_INFO_LENGTH_MISMATCH); + + /* Unaligned buffer -- correct size (but this time do with an alignment of 2) */ + Status = NtQueryInformationProcess(NtCurrentProcess(), + ProcessPriorityClass, + (PVOID)2, + sizeof(PROCESS_PRIORITY_CLASS), + NULL); + ok_hex(Status, STATUS_DATATYPE_MISALIGNMENT); + + /* Do not care for the length but expect to return the priority class */ + Status = NtQueryInformationProcess(NtCurrentProcess(), + ProcessPriorityClass, + ProcPriority, + sizeof(PROCESS_PRIORITY_CLASS), + NULL); + ok_hex(Status, STATUS_SUCCESS); + + /* Make sure the returned priority class is a valid number (non negative) but also it should be within the PROCESS_PRIORITY_CLASS range */ + ok(ProcPriority->PriorityClass > PROCESS_PRIORITY_CLASS_INVALID && ProcPriority->PriorityClass <= PROCESS_PRIORITY_CLASS_ABOVE_NORMAL, + "Expected a valid number from priority class range but got %d\n", ProcPriority->PriorityClass); + free(ProcPriority); +} + START_TEST(NtQueryInformationProcess) { NTSTATUS Status; @@ -193,4 +261,5 @@ START_TEST(NtQueryInformationProcess) ok_hex(Status, STATUS_SUCCESS); Test_ProcessTimes(); + Test_ProcessPriorityClassAlignment(); }
