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

[KSECDD] Fix RtlEncryptMemory and improve test

Browse source
This commit is contained in:
Timo Kreuzer 2020-09-13 19:58:06 +02:00
parent ed02f41af5
commit ab43f86980
2 changed files with 123 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

106
modules/rostests/apitests/advapi32/RtlEncryptMemory.c
View file

@ -9,26 +9,112 @@
#include <ntsecapi.h>
START_TEST(RtlEncryptMemory)
static const CHAR TestData[32] = "This is some test Message!!!";
void TestEncrypt(ULONG OptionFlags)
{
static const CHAR TestData[32] = "This is some test Message!!!";
CHAR Buffer[32];
CHAR DECLSPEC_ALIGN(8) Buffer[32];
NTSTATUS Status;
/* Size must be aligned to 8 bytes */
Status = RtlEncryptMemory(Buffer, 7, RTL_ENCRYPT_OPTION_SAME_PROCESS);
/* Size must be aligned to 8 bytes (aka RTL_ENCRYPT_MEMORY_SIZE) */
RtlCopyMemory(Buffer, TestData, sizeof(Buffer));
Status = RtlEncryptMemory(Buffer, 7, OptionFlags);
ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
/* Buffer must not be aligned to 8 bytes */
Status = RtlEncryptMemory(&Buffer[1], 8, RTL_ENCRYPT_OPTION_SAME_PROCESS);
/* Buffer can be unaligned */
RtlCopyMemory(Buffer, TestData, sizeof(Buffer));
Status = RtlEncryptMemory(&Buffer[1], 8, OptionFlags);
ok_ntstatus(Status, STATUS_SUCCESS);
Status = RtlDecryptMemory(&Buffer[1], 8, OptionFlags);
ok_ntstatus(Status, STATUS_SUCCESS);
ok(RtlEqualMemory(Buffer, TestData, sizeof(Buffer)) == 1,
"OptionFlags=%lu, Buffer='%s', TestData='%s'\n",
OptionFlags, Buffer, TestData);
RtlCopyMemory(Buffer, TestData, sizeof(Buffer));
Status = RtlEncryptMemory(Buffer, sizeof(Buffer), RTL_ENCRYPT_OPTION_SAME_PROCESS);
Status = RtlEncryptMemory(Buffer, sizeof(Buffer), OptionFlags);
ok_ntstatus(Status, STATUS_SUCCESS);
ok_int(RtlEqualMemory(Buffer, TestData, sizeof(Buffer)), 0);
Status = RtlDecryptMemory(Buffer, sizeof(Buffer), RTL_ENCRYPT_OPTION_SAME_PROCESS);
Status = RtlDecryptMemory(Buffer, sizeof(Buffer), OptionFlags);
ok_ntstatus(Status, STATUS_SUCCESS);
ok_int(RtlEqualMemory(Buffer, TestData, sizeof(Buffer)), 1);
}
void TestCrossProcessDecrypt(PSTR Param)
{
UCHAR Buffer[32] = { 0 };
ULONG OptionFlags;
PSTR StrData;
ULONG i;
NTSTATUS Status;
OptionFlags = Param[3] == '1' ? RTL_ENCRYPT_OPTION_CROSS_PROCESS : RTL_ENCRYPT_OPTION_SAME_LOGON;
/* Convert the HEX string to binary ('-cp<1|2>:<data>') */
StrData = Param + 5;
for (i = 0; i < sizeof(Buffer); i++)
{
UINT x;
sscanf(&StrData[2 * i], "%02X", &x);
Buffer[i] = (UCHAR)x;
}
/* Decrypt the data */
Status = RtlDecryptMemory(Buffer, sizeof(Buffer), OptionFlags);
ok_ntstatus(Status, STATUS_SUCCESS);
ok_int(RtlEqualMemory(Buffer, TestData, sizeof(Buffer)), 1);
}
void TestCrossProcessEncrypt(ULONG OptionFlags)
{
UCHAR Buffer[32];
CHAR CmdLine[MAX_PATH];
PSTR StrBuffer;
NTSTATUS Status;
ULONG i;
INT result;
/* Encrypt the test string */
RtlCopyMemory(Buffer, TestData, sizeof(Buffer));
Status = RtlEncryptMemory(Buffer, sizeof(Buffer), OptionFlags);
ok_ntstatus(Status, STATUS_SUCCESS);
/* Start building a command line */
sprintf(CmdLine, "advapi32_apitest.exe RtlEncryptMemory -cp%lu:", OptionFlags);
StrBuffer = CmdLine + strlen(CmdLine);
/* Convert encrypted data into a hex string */
for (i = 0; i < sizeof(Buffer); i++)
{
sprintf(&StrBuffer[2 * i], "%02X", Buffer[i]);
}
result = system(CmdLine);
ok_int(result, 0);
}
START_TEST(RtlEncryptMemory)
{
CHAR Buffer[8] = { 0 };
PSTR CommandLine, Param;
NTSTATUS Status;
/* Check recursive call */
CommandLine = GetCommandLineA();
Param = strstr(CommandLine, "-cp");
if (Param != NULL)
{
TestCrossProcessDecrypt(Param);
return;
}
TestEncrypt(RTL_ENCRYPT_OPTION_SAME_PROCESS);
TestEncrypt(RTL_ENCRYPT_OPTION_CROSS_PROCESS);
TestEncrypt(RTL_ENCRYPT_OPTION_SAME_LOGON);
Status = RtlEncryptMemory(Buffer, sizeof(Buffer), 4);
ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
TestCrossProcessEncrypt(RTL_ENCRYPT_OPTION_CROSS_PROCESS);
TestCrossProcessEncrypt(RTL_ENCRYPT_OPTION_SAME_LOGON);
}