mirror of
https://github.com/reactos/reactos.git
synced 2026-06-22 01:12:52 +08:00
99 lines
3.7 KiB
C
99 lines
3.7 KiB
C
/*
|
|
* PROJECT: ReactOS API tests
|
|
* LICENSE: MIT (https://spdx.org/licenses/MIT)
|
|
* PURPOSE: Tests for RtlQueryEnvironmentVariable
|
|
* COPYRIGHT: Copyright 2026 Timo Kreuzer <timo.kreuzer@reactos.org>
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
typedef
|
|
NTSTATUS
|
|
NTAPI
|
|
FN_RtlQueryEnvironmentVariable(
|
|
_In_opt_ PVOID Environment,
|
|
_In_reads_(NameLength) PWSTR Name,
|
|
_In_ SIZE_T NameLength,
|
|
_Out_writes_(ValueLength) PWSTR Value,
|
|
_In_ SIZE_T ValueLength,
|
|
_Out_ PSIZE_T ReturnLength);
|
|
|
|
static FN_RtlQueryEnvironmentVariable* pRtlQueryEnvironmentVariable;
|
|
|
|
START_TEST(RtlQueryEnvironmentVariable)
|
|
{
|
|
static WCHAR TestEnv[] = L"TestVar=TestVal\0Foo=4\0EmptyVar\0Bar=8\0\0";
|
|
WCHAR Buffer[32];
|
|
SIZE_T ReturnLength;
|
|
NTSTATUS Status;
|
|
|
|
HINSTANCE hinst = GetModuleHandleA("ntdll.dll");
|
|
pRtlQueryEnvironmentVariable = (FN_RtlQueryEnvironmentVariable*)
|
|
GetProcAddress(hinst, "RtlQueryEnvironmentVariable");
|
|
if (pRtlQueryEnvironmentVariable == NULL)
|
|
{
|
|
ok(GetNTVersion() < _WIN32_WINNT_VISTA, "RtlQueryEnvironmentVariable not available on NT6+\n");
|
|
skip("RtlQueryEnvironmentVariable is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* Query the TestVar environment variable length */
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"TestVar", 7, NULL, 0, &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_BUFFER_TOO_SMALL);
|
|
ok_eq_size(ReturnLength, 8);
|
|
|
|
/* Test a buffer size that is too small */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"TestVar", 7, Buffer, 2, &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_BUFFER_TOO_SMALL);
|
|
ok_eq_size(ReturnLength, 8);
|
|
ok_eq_wchar(Buffer[0], UNICODE_NULL);
|
|
ok_eq_wchar(Buffer[1], 0xABAB);
|
|
|
|
/* Test a buffer size that doesn't fit the terminating NULL char */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"TestVar", 7, Buffer, 7, &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_BUFFER_TOO_SMALL);
|
|
ok_eq_size(ReturnLength, 8);
|
|
ok_eq_wchar(Buffer[0], UNICODE_NULL);
|
|
ok_eq_wchar(Buffer[1], 0xABAB);
|
|
|
|
/* Test a buffer size that is just large enough */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"TestVar", 7, Buffer, 8, &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
|
ok_eq_size(ReturnLength, 7);
|
|
ok_eq_wstr(Buffer, L"TestVal");
|
|
ok_eq_wchar(Buffer[7], UNICODE_NULL);
|
|
ok_eq_wchar(Buffer[8], 0xABAB);
|
|
|
|
/* Test a buffer size that is larger than needed */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"TestVar", 7, Buffer, ARRAYSIZE(Buffer), &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
|
ok_eq_size(ReturnLength, 7);
|
|
ok_eq_wstr(Buffer, L"TestVal");
|
|
ok_eq_wchar(Buffer[7], UNICODE_NULL);
|
|
ok_eq_wchar(Buffer[8], 0xABAB);
|
|
|
|
/* Test a too big variable Length */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"TestVar", 8, Buffer, 8, &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_VARIABLE_NOT_FOUND);
|
|
ok_eq_wchar(Buffer[0], 0xABAB);
|
|
|
|
/* Query the EmptyVar environment variable */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"EmptyVar", 8, Buffer, ARRAYSIZE(Buffer), &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_VARIABLE_NOT_FOUND);
|
|
ok_eq_wchar(Buffer[0], 0xABAB);
|
|
|
|
/* Query the Bar environment variable */
|
|
memset(Buffer, 0xAB, sizeof(Buffer));
|
|
Status = pRtlQueryEnvironmentVariable(TestEnv, L"Bar", 3, Buffer, 8, &ReturnLength);
|
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
|
ok_eq_wchar(Buffer[0], L'8');
|
|
ok_eq_wchar(Buffer[1], UNICODE_NULL);
|
|
ok_eq_wchar(Buffer[2], 0xABAB);
|
|
}
|