RtlQueryEnvironmentVariable.c File Reference

#include "precomp.h"

Include dependency graph for RtlQueryEnvironmentVariable.c:

Go to the source code of this file.

Typedefs
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)

Functions
	START_TEST (RtlQueryEnvironmentVariable)

Variables
static FN_RtlQueryEnvironmentVariable *	pRtlQueryEnvironmentVariable

Typedef Documentation

FN_RtlQueryEnvironmentVariable

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)

Definition at line 13 of file RtlQueryEnvironmentVariable.c.

Function Documentation

START_TEST()

START_TEST

(

RtlQueryEnvironmentVariable

)

Definition at line 23 of file RtlQueryEnvironmentVariable.c.

{
    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);
}

Variable Documentation

pRtlQueryEnvironmentVariable

FN_RtlQueryEnvironmentVariable* pRtlQueryEnvironmentVariable

static

Definition at line 21 of file RtlQueryEnvironmentVariable.c.

Referenced by START_TEST(), and testQuery().