mirror of
https://github.com/reactos/reactos.git
synced 2024-11-10 08:43:28 +00:00
130 lines
3.8 KiB
C
130 lines
3.8 KiB
C
/*
|
|
* PROJECT: ReactOS API Tests
|
|
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
|
* PURPOSE: Test for NtSaveKey
|
|
* PROGRAMMERS: Aleksandar Andrejevic <theflash AT sdf DOT lonestar DOT org>
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
static
|
|
NTSTATUS
|
|
OpenRegistryKeyHandle(PHANDLE KeyHandle,
|
|
ACCESS_MASK AccessMask,
|
|
PWCHAR RegistryPath)
|
|
{
|
|
UNICODE_STRING KeyName;
|
|
OBJECT_ATTRIBUTES Attributes;
|
|
|
|
RtlInitUnicodeString(&KeyName, RegistryPath);
|
|
InitializeObjectAttributes(&Attributes,
|
|
&KeyName,
|
|
OBJ_CASE_INSENSITIVE,
|
|
NULL,
|
|
NULL);
|
|
|
|
return NtOpenKey(KeyHandle, AccessMask, &Attributes);
|
|
}
|
|
|
|
START_TEST(NtSaveKey)
|
|
{
|
|
NTSTATUS Status;
|
|
HANDLE KeyHandle;
|
|
HANDLE FileHandle;
|
|
BOOLEAN PrivilegeEnabled = FALSE;
|
|
BOOLEAN OldPrivilegeStatus;
|
|
|
|
/* Make sure we don't have backup privileges initially, otherwise WHS testbot fails */
|
|
Status = RtlAdjustPrivilege(SE_BACKUP_PRIVILEGE, FALSE, FALSE, &PrivilegeEnabled);
|
|
ok(Status == STATUS_SUCCESS, "RtlAdjustPrivilege returned %lx\n", Status);
|
|
|
|
/* Open the file */
|
|
FileHandle = CreateFileW(L"saved_key.dat",
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_DELETE_ON_CLOSE,
|
|
NULL);
|
|
if (FileHandle == INVALID_HANDLE_VALUE)
|
|
{
|
|
skip("CreateFileW failed with error: %lu\n", GetLastError());
|
|
return;
|
|
}
|
|
|
|
/* Try saving HKEY_LOCAL_MACHINE\Hardware */
|
|
Status = OpenRegistryKeyHandle(&KeyHandle, KEY_READ, L"\\Registry\\Machine\\Hardware");
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
skip("NtOpenKey failed with status: 0x%08lX\n", Status);
|
|
NtClose(FileHandle);
|
|
return;
|
|
}
|
|
|
|
Status = NtSaveKey(KeyHandle, FileHandle);
|
|
ok_ntstatus(Status, STATUS_PRIVILEGE_NOT_HELD);
|
|
|
|
NtClose(KeyHandle);
|
|
|
|
/* Set the SeBackupPrivilege */
|
|
Status = RtlAdjustPrivilege(SE_BACKUP_PRIVILEGE,
|
|
TRUE,
|
|
FALSE,
|
|
&OldPrivilegeStatus);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
skip("RtlAdjustPrivilege failed with status: 0x%08lX\n", (ULONG)Status);
|
|
NtClose(FileHandle);
|
|
return;
|
|
}
|
|
|
|
/* Try saving HKEY_LOCAL_MACHINE\Hardware again */
|
|
Status = OpenRegistryKeyHandle(&KeyHandle, KEY_READ, L"\\Registry\\Machine\\Hardware");
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
skip("NtOpenKey failed with status: 0x%08lX\n", Status);
|
|
goto Cleanup;
|
|
}
|
|
|
|
Status = NtSaveKey(KeyHandle, FileHandle);
|
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
|
|
|
NtClose(KeyHandle);
|
|
|
|
/* Try saving HKEY_LOCAL_MACHINE */
|
|
Status = OpenRegistryKeyHandle(&KeyHandle, KEY_READ, L"\\Registry\\Machine");
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
skip("NtOpenKey failed with status: 0x%08lX\n", Status);
|
|
goto Cleanup;
|
|
}
|
|
|
|
Status = NtSaveKey(KeyHandle, FileHandle);
|
|
ok_ntstatus(Status, STATUS_ACCESS_DENIED);
|
|
|
|
NtClose(KeyHandle);
|
|
|
|
/* Try saving HKEY_USERS */
|
|
Status = OpenRegistryKeyHandle(&KeyHandle, KEY_READ, L"\\Registry\\User");
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
skip("NtOpenKey failed with status: 0x%08lX\n", Status);
|
|
goto Cleanup;
|
|
}
|
|
|
|
Status = NtSaveKey(KeyHandle, FileHandle);
|
|
ok_ntstatus(Status, STATUS_ACCESS_DENIED);
|
|
|
|
NtClose(KeyHandle);
|
|
|
|
Cleanup:
|
|
|
|
/* Restore the SeBackupPrivilege */
|
|
RtlAdjustPrivilege(SE_BACKUP_PRIVILEGE,
|
|
OldPrivilegeStatus,
|
|
FALSE,
|
|
&OldPrivilegeStatus);
|
|
|
|
/* Close the file handle */
|
|
NtClose(FileHandle);
|
|
}
|