RtlComputePrivatizedDllName_U.c File Reference

#include "precomp.h"

Include dependency graph for RtlComputePrivatizedDllName_U.c:

Go to the source code of this file.

Macros
#define	ok_strings(Real, Local, Dll)   ok_strings_(Real, Local, Dll, __LINE__)

Functions
static BOOL	InitTestData ()
static void	ok_strings_ (PCUNICODE_STRING RealName, PCUNICODE_STRING LocalName, LPCWSTR DllName, int line)
static void	cleanup (PUNICODE_STRING String, WCHAR *Buffer, USHORT BufferSize)
static void	test_dllnames (void)
static void	test_allocations (void)
	START_TEST (RtlComputePrivatizedDllName_U)

Variables
static WCHAR	ProcessDir [MAX_PATH]
static WCHAR	LocalDir [MAX_PATH+10]

Macro Definition Documentation

ok_strings

#define ok_strings	(		Real,
			Local,
			Dll
	)		ok_strings_(Real, Local, Dll, __LINE__)

Definition at line 64 of file RtlComputePrivatizedDllName_U.c.

Function Documentation

cleanup()

static void cleanup ( PUNICODE_STRING  String, WCHAR *  Buffer, USHORT  BufferSize  )

static

Definition at line 67 of file RtlComputePrivatizedDllName_U.c.

{
    if (String->Buffer != Buffer)
    {
        RtlFreeUnicodeString(String);
        RtlInitEmptyUnicodeString(String, Buffer, BufferSize);
    }
}

InitTestData()

static BOOL InitTestData ( )

static

Definition at line 14 of file RtlComputePrivatizedDllName_U.c.

{
    static const UNICODE_STRING PathDividerFind = RTL_CONSTANT_STRING(L"\\/");
    UNICODE_STRING StrU;
    USHORT PathDivider;
 
    GetModuleFileNameW(NULL, ProcessDir, RTL_NUMBER_OF(ProcessDir));
    GetModuleFileNameW(NULL, LocalDir, RTL_NUMBER_OF(LocalDir));
 
    RtlInitUnicodeString(&StrU, ProcessDir);
 
    if (!NT_SUCCESS(RtlFindCharInUnicodeString(RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END,
                                               &StrU, &PathDividerFind, &PathDivider)))
    {
        skip("Failed to find path divider\n");
        return FALSE;
    }
    ProcessDir[PathDivider / sizeof(WCHAR) + 1] = UNICODE_NULL;
 
    RtlInitUnicodeString(&StrU, LocalDir);
    StrU.MaximumLength = sizeof(LocalDir);
 
    if (!NT_SUCCESS(RtlAppendUnicodeToString(&StrU, L".Local\\")))
    {
        skip("Failed to append .Local\\\n");
        return FALSE;
    }
    return TRUE;
}

Referenced by START_TEST().

ok_strings_()

static void ok_strings_ ( PCUNICODE_STRING  RealName, PCUNICODE_STRING  LocalName, LPCWSTR  DllName, int  line  )

static

Definition at line 44 of file RtlComputePrivatizedDllName_U.c.

{
    WCHAR ExpectReal[MAX_PATH*2];
    WCHAR ExpectLocal[MAX_PATH*2];
    int RealLen;
    int ExpectLen;
 
    RealLen = swprintf(ExpectReal, L"%s%s", ProcessDir, DllName) * sizeof(WCHAR);
    ExpectLen = swprintf(ExpectLocal, L"%s%s", LocalDir, DllName) * sizeof(WCHAR);
 
    ok_(__FILE__, line)(RealLen == RealName->Length, "Expected Real %u, got %u\n",
                        (UINT)RealLen, (UINT)RealName->Length);
    ok_(__FILE__, line)(ExpectLen == LocalName->Length, "Expected Local %u, got %u\n",
                        (UINT)ExpectLen, (UINT)LocalName->Length);
 
    ok_(__FILE__, line)(!wcscmp(RealName->Buffer, ExpectReal), "Expected Real %s, got %s\n",
                        wine_dbgstr_w(ExpectReal), wine_dbgstr_w(RealName->Buffer));
    ok_(__FILE__, line)(!wcscmp(LocalName->Buffer, ExpectLocal), "Expected Local %s, got %s\n",
                        wine_dbgstr_w(ExpectLocal), wine_dbgstr_w(LocalName->Buffer));
}

START_TEST()

START_TEST

(

RtlComputePrivatizedDllName_U

)

Definition at line 229 of file RtlComputePrivatizedDllName_U.c.

{
    if (!InitTestData())
        return;
 
    test_dllnames();
    test_allocations();
}

test_allocations()

static void test_allocations ( void  )

static

Definition at line 206 of file RtlComputePrivatizedDllName_U.c.

{
    WCHAR Buf1[10];
    WCHAR Buf2[10];
 
    UNICODE_STRING Str1, Str2;
    UNICODE_STRING DllName;
    NTSTATUS Status;
 
    RtlInitEmptyUnicodeString(&Str1, Buf1, sizeof(Buf1));
    RtlInitEmptyUnicodeString(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"kernel32.dll");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.dll");
    ok(Str1.Buffer != Buf1, "Expected a changed buffer\n");
    ok(Str2.Buffer != Buf2, "Expected a changed buffer\n");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
}

Referenced by START_TEST().

test_dllnames()

static void test_dllnames ( void  )

static

Definition at line 77 of file RtlComputePrivatizedDllName_U.c.

{
    WCHAR Buf1[MAX_PATH];
    WCHAR Buf2[MAX_PATH];
 
    UNICODE_STRING Str1, Str2;
    UNICODE_STRING DllName;
    NTSTATUS Status;
 
    RtlInitEmptyUnicodeString(&Str1, Buf1, sizeof(Buf1));
    RtlInitEmptyUnicodeString(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"kernel32.dll");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.dll");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
 
    RtlInitUnicodeString(&DllName, L"kernel32");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.DLL");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"kernel32.dll.dll");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.dll.dll");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"kernel32.dll.exe");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.dll.exe");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"kernel32.");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L".kernel32");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L".kernel32.DLL");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"..kernel32");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"..kernel32");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L".kernel32.");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L".kernel32.");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
 
    RtlInitUnicodeString(&DllName, L"test\\kernel32.dll");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.dll");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"test/kernel32.dll");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.dll");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"test.dll/kernel32");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.DLL");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"test.dll\\kernel32");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L"kernel32.DLL");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"//");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L".DLL");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    // Bug :)
    RtlInitUnicodeString(&DllName, L"\\");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    if (wcsstr(Str1.Buffer, L"\\\\"))
    {
        trace("Allowing bug found in windows' implementation\n");
        ok_strings(&Str1, &Str2, L"\\.DLL");
    }
    else
    {
        ok_strings(&Str1, &Str2, L".DLL");
    }
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
 
    RtlInitUnicodeString(&DllName, L"");
    Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
    ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
    ok_strings(&Str1, &Str2, L".DLL");
    cleanup(&Str1, Buf1, sizeof(Buf1));
    cleanup(&Str2, Buf2, sizeof(Buf2));
}

Referenced by START_TEST().

Variable Documentation

LocalDir

WCHAR LocalDir[MAX_PATH+10]

static

Definition at line 12 of file RtlComputePrivatizedDllName_U.c.

Referenced by InitTestData(), and ok_strings_().

ProcessDir

WCHAR ProcessDir[MAX_PATH]

static

Definition at line 11 of file RtlComputePrivatizedDllName_U.c.

Referenced by InitTestData(), and ok_strings_().