Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-07-02 18:54:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NTDLL_APITEST]

Browse source 
- Add a test for SystemFlagsInformation that demonstrates the parameter checks performed by NtQuerySystemInformation/NtSetSystemInformation
CORE-9849

svn path=/trunk/; revision=68225
This commit is contained in:
Thomas Faber 2015-06-21 08:50:10 +00:00
parent f633746280
commit e9138ce35a
1 changed files with 171 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

184
rostests/apitests/ntdll/SystemInfo.c
View file

@ -3,33 +3,182 @@
* LICENSE: GPL - See COPYING in the top level directory
* PURPOSE: Test for NtQuery/SetSystemInformation
* PROGRAMMERS: Timo Kreuzer
* Thomas Faber <thomas.faber@reactos.org>
*/
#include <apitest.h>
#define WIN32_NO_STATUS
#include <ndk/exfuncs.h>
#include <ndk/rtlfuncs.h>
#include <ndk/setypes.h>
#define ntv6(x) (LOBYTE(LOWORD(GetVersion())) >= 6 ? (x) : 0)
static
void
GetPrivilege()
Test_Flags(void)
{
HANDLE hToken;
TOKEN_PRIVILEGES tkp;
NTSTATUS Status;
ULONG ReturnLength;
ULONG Flags;
ULONG Buffer[2];
ULONG Buffer2[2];
PSYSTEM_FLAGS_INFORMATION FlagsInfo = (PVOID)Buffer;
BOOLEAN PrivilegeEnabled;
OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
&hToken);
/* Query */
ReturnLength = 0x55555555;
Status = NtQuerySystemInformation(SystemFlagsInformation, NULL, 0, &ReturnLength);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0 ||
ntv6(ReturnLength == sizeof(SYSTEM_FLAGS_INFORMATION)), "ReturnLength = %lu\n", ReturnLength);
LookupPrivilegeValue(NULL, SE_SYSTEMTIME_NAME, &tkp.Privileges[0].Luid);
ReturnLength = 0x55555555;
Status = NtQuerySystemInformation(SystemFlagsInformation, NULL, sizeof(SYSTEM_FLAGS_INFORMATION) - 1, &ReturnLength);
ok(Status == STATUS_ACCESS_VIOLATION, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0x55555555, "ReturnLength = %lu\n", ReturnLength);
tkp.PrivilegeCount = 1;
tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
ReturnLength = 0x55555555;
Status = NtQuerySystemInformation(SystemFlagsInformation, (PVOID)2, 0, &ReturnLength);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0 ||
ntv6(ReturnLength == sizeof(SYSTEM_FLAGS_INFORMATION)), "ReturnLength = %lu\n", ReturnLength);
AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, NULL, 0);
CloseHandle(hToken);
ReturnLength = 0x55555555;
Status = NtQuerySystemInformation(SystemFlagsInformation, (PVOID)4, 0, &ReturnLength);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0 ||
ntv6(ReturnLength == sizeof(SYSTEM_FLAGS_INFORMATION)), "ReturnLength = %lu\n", ReturnLength);
ReturnLength = 0x55555555;
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, (PUCHAR)FlagsInfo + 2, sizeof(SYSTEM_FLAGS_INFORMATION), &ReturnLength);
ok(Status == STATUS_DATATYPE_MISALIGNMENT, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0x55555555, "ReturnLength = %lu\n", ReturnLength);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
ReturnLength = 0x55555555;
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, (PUCHAR)FlagsInfo + 2, sizeof(SYSTEM_FLAGS_INFORMATION) - 1, &ReturnLength);
ok(Status == STATUS_DATATYPE_MISALIGNMENT, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0x55555555, "ReturnLength = %lu\n", ReturnLength);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
ReturnLength = 0x55555555;
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION) - 1, &ReturnLength);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0 ||
ntv6(ReturnLength == sizeof(SYSTEM_FLAGS_INFORMATION)), "ReturnLength = %lu\n", ReturnLength);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
ReturnLength = 0x55555555;
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION), &ReturnLength);
ok(Status == STATUS_SUCCESS, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == sizeof(SYSTEM_FLAGS_INFORMATION), "ReturnLength = %lu\n", ReturnLength);
ok(FlagsInfo->Flags != 0x55555555, "Flags = %lx\n", FlagsInfo->Flags);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
Flags = FlagsInfo->Flags;
ReturnLength = 0x55555555;
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION) + 1, &ReturnLength);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtQuerySystemInformation returned %lx\n", Status);
ok(ReturnLength == 0 ||
ntv6(ReturnLength == sizeof(SYSTEM_FLAGS_INFORMATION)), "ReturnLength = %lu\n", ReturnLength);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION), NULL);
ok(Status == STATUS_SUCCESS, "NtQuerySystemInformation returned %lx\n", Status);
ok(FlagsInfo->Flags == Flags, "Flags = %lx, expected %lx\n", FlagsInfo->Flags, Flags);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION), (PVOID)4);
ok(Status == STATUS_ACCESS_VIOLATION, "NtQuerySystemInformation returned %lx\n", Status);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION), (PVOID)2);
ok(Status == STATUS_ACCESS_VIOLATION, "NtQuerySystemInformation returned %lx\n", Status);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION), (PVOID)1);
ok(Status == STATUS_ACCESS_VIOLATION, "NtQuerySystemInformation returned %lx\n", Status);
ok(Buffer[0] == 0x55555555, "Buffer[0] = %lx\n", Buffer[0]);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
RtlFillMemory(Buffer2, sizeof(Buffer2), 0x55);
RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
Status = NtQuerySystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION), (PULONG)((PUCHAR)Buffer2 + 1));
ok(Status == STATUS_SUCCESS, "NtQuerySystemInformation returned %lx\n", Status);
ok(FlagsInfo->Flags == Flags, "Flags = %lx, expected %lx\n", FlagsInfo->Flags, Flags);
ok(Buffer[1] == 0x55555555, "Buffer[1] = %lx\n", Buffer[1]);
ok(Buffer2[0] == 0x00000455, "Buffer2[0] = %lx\n", Buffer2[0]);
ok(Buffer2[1] == 0x55555500, "Buffer2[1] = %lx\n", Buffer2[1]);
/* Set */
Status = NtSetSystemInformation(SystemFlagsInformation, NULL, 0);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, NULL, sizeof(SYSTEM_FLAGS_INFORMATION) - 1);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, NULL, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_ACCESS_DENIED, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, (PVOID)2, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_DATATYPE_MISALIGNMENT, "NtSetSystemInformation returned %lx\n", Status);
Status = RtlAdjustPrivilege(SE_DEBUG_PRIVILEGE, TRUE, FALSE, &PrivilegeEnabled);
if (!NT_SUCCESS(Status))
{
skip("Cannot acquire SeDebugPrivilege\n");
return;
}
Status = NtSetSystemInformation(SystemFlagsInformation, NULL, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_ACCESS_VIOLATION, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, (PVOID)2, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_DATATYPE_MISALIGNMENT, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, (PVOID)4, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_ACCESS_VIOLATION, "NtSetSystemInformation returned %lx\n", Status);
FlagsInfo->Flags = Flags;
Status = NtSetSystemInformation(SystemFlagsInformation, (PUCHAR)FlagsInfo + 2, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_DATATYPE_MISALIGNMENT, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, (PUCHAR)FlagsInfo + 2, sizeof(SYSTEM_FLAGS_INFORMATION) - 1);
ok(Status == STATUS_DATATYPE_MISALIGNMENT, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION) - 1);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION) + 1);
ok(Status == STATUS_INFO_LENGTH_MISMATCH, "NtSetSystemInformation returned %lx\n", Status);
Status = NtSetSystemInformation(SystemFlagsInformation, FlagsInfo, sizeof(SYSTEM_FLAGS_INFORMATION));
ok(Status == STATUS_SUCCESS, "NtSetSystemInformation returned %lx\n", Status);
ok(FlagsInfo->Flags == Flags, "Flags = %lx, expected %lu\n", FlagsInfo->Flags, Flags);
Status = RtlAdjustPrivilege(SE_DEBUG_PRIVILEGE, PrivilegeEnabled, FALSE, &PrivilegeEnabled);
ok(Status == STATUS_SUCCESS, "RtlAdjustPrivilege returned %lx\n", Status);
}
static
void
Test_TimeAdjustment(void)
{
@ -37,6 +186,7 @@ Test_TimeAdjustment(void)
SYSTEM_SET_TIME_ADJUST_INFORMATION SetTimeInfo;
NTSTATUS Status;
ULONG ReturnLength;
BOOLEAN PrivilegeEnabled;
SetTimeInfo.TimeAdjustment = 0;
SetTimeInfo.Enable = 0;
@ -54,7 +204,12 @@ Test_TimeAdjustment(void)
ok_ntstatus(Status, STATUS_PRIVILEGE_NOT_HELD);
/* Get the required privilege */
GetPrivilege();
Status = RtlAdjustPrivilege(SE_SYSTEMTIME_PRIVILEGE, TRUE, FALSE, &PrivilegeEnabled);
if (!NT_SUCCESS(Status))
{
skip("Cannot acquire SeSystemTimePrivilege\n");
return;
}
/* Test wrong length */
Status = NtSetSystemInformation(SystemTimeAdjustmentInformation,
@ -110,9 +265,12 @@ Test_TimeAdjustment(void)
sizeof(SetTimeInfo));
ok_ntstatus(Status, STATUS_SUCCESS);
Status = RtlAdjustPrivilege(SE_SYSTEMTIME_PRIVILEGE, PrivilegeEnabled, FALSE, &PrivilegeEnabled);
ok(Status == STATUS_SUCCESS, "RtlAdjustPrivilege returned %lx\n", Status);
}
START_TEST(NtSystemInformation)
{
Test_Flags();
Test_TimeAdjustment();
}