ReactOS 0.4.16-dev-197-g92996da
RtlComputePrivatizedDllName_U.c
Go to the documentation of this file.
1/*
2 * PROJECT: ReactOS api tests
3 * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
4 * PURPOSE: Test for RtlComputePrivatizedDllName_U
5 * COPYRIGHT: Copyright 2019 Mark Jansen (mark.jansen@reactos.org)
6 */
7
8#include "precomp.h"
9
10
11static WCHAR ProcessDir[MAX_PATH];
12static WCHAR LocalDir[MAX_PATH + 10];
13
14static BOOL InitTestData()
15{
16 static const UNICODE_STRING PathDividerFind = RTL_CONSTANT_STRING(L"\\/");
17 UNICODE_STRING StrU;
18 USHORT PathDivider;
19
20 GetModuleFileNameW(NULL, ProcessDir, RTL_NUMBER_OF(ProcessDir));
21 GetModuleFileNameW(NULL, LocalDir, RTL_NUMBER_OF(LocalDir));
22
23 RtlInitUnicodeString(&StrU, ProcessDir);
24
25 if (!NT_SUCCESS(RtlFindCharInUnicodeString(RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END,
26 &StrU, &PathDividerFind, &PathDivider)))
27 {
28 skip("Failed to find path divider\n");
29 return FALSE;
30 }
31 ProcessDir[PathDivider / sizeof(WCHAR) + 1] = UNICODE_NULL;
32
33 RtlInitUnicodeString(&StrU, LocalDir);
34 StrU.MaximumLength = sizeof(LocalDir);
35
36 if (!NT_SUCCESS(RtlAppendUnicodeToString(&StrU, L".Local\\")))
37 {
38 skip("Failed to append .Local\\\n");
39 return FALSE;
40 }
41 return TRUE;
42}
43
44static void ok_strings_(PCUNICODE_STRING RealName, PCUNICODE_STRING LocalName, LPCWSTR DllName, int line)
45{
46 WCHAR ExpectReal[MAX_PATH*2];
47 WCHAR ExpectLocal[MAX_PATH*2];
48 int RealLen;
49 int ExpectLen;
50
51 RealLen = swprintf(ExpectReal, L"%s%s", ProcessDir, DllName) * sizeof(WCHAR);
52 ExpectLen = swprintf(ExpectLocal, L"%s%s", LocalDir, DllName) * sizeof(WCHAR);
53
54 ok_(__FILE__, line)(RealLen == RealName->Length, "Expected Real %u, got %u\n",
55 (UINT)RealLen, (UINT)RealName->Length);
56 ok_(__FILE__, line)(ExpectLen == LocalName->Length, "Expected Local %u, got %u\n",
57 (UINT)ExpectLen, (UINT)LocalName->Length);
58
59 ok_(__FILE__, line)(!wcscmp(RealName->Buffer, ExpectReal), "Expected Real %s, got %s\n",
60 wine_dbgstr_w(ExpectReal), wine_dbgstr_w(RealName->Buffer));
61 ok_(__FILE__, line)(!wcscmp(LocalName->Buffer, ExpectLocal), "Expected Local %s, got %s\n",
62 wine_dbgstr_w(ExpectLocal), wine_dbgstr_w(LocalName->Buffer));
63}
64#define ok_strings(Real, Local, Dll) ok_strings_(Real, Local, Dll, __LINE__)
65
66
67static void cleanup(PUNICODE_STRING String, WCHAR* Buffer, USHORT BufferSize)
68{
69 if (String->Buffer != Buffer)
70 {
71 RtlFreeUnicodeString(String);
72 RtlInitEmptyUnicodeString(String, Buffer, BufferSize);
73 }
74}
75
76
77static void test_dllnames(void)
78{
79 WCHAR Buf1[MAX_PATH];
80 WCHAR Buf2[MAX_PATH];
81
82 UNICODE_STRING Str1, Str2;
83 UNICODE_STRING DllName;
84 NTSTATUS Status;
85
86 RtlInitEmptyUnicodeString(&Str1, Buf1, sizeof(Buf1));
87 RtlInitEmptyUnicodeString(&Str2, Buf2, sizeof(Buf2));
88
89 RtlInitUnicodeString(&DllName, L"kernel32.dll");
90 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
91 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
92 ok_strings(&Str1, &Str2, L"kernel32.dll");
93 cleanup(&Str1, Buf1, sizeof(Buf1));
94 cleanup(&Str2, Buf2, sizeof(Buf2));
95
96
97 RtlInitUnicodeString(&DllName, L"kernel32");
98 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
99 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
100 ok_strings(&Str1, &Str2, L"kernel32.DLL");
101 cleanup(&Str1, Buf1, sizeof(Buf1));
102 cleanup(&Str2, Buf2, sizeof(Buf2));
103
104 RtlInitUnicodeString(&DllName, L"kernel32.dll.dll");
105 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
106 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
107 ok_strings(&Str1, &Str2, L"kernel32.dll.dll");
108 cleanup(&Str1, Buf1, sizeof(Buf1));
109 cleanup(&Str2, Buf2, sizeof(Buf2));
110
111 RtlInitUnicodeString(&DllName, L"kernel32.dll.exe");
112 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
113 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
114 ok_strings(&Str1, &Str2, L"kernel32.dll.exe");
115 cleanup(&Str1, Buf1, sizeof(Buf1));
116 cleanup(&Str2, Buf2, sizeof(Buf2));
117
118 RtlInitUnicodeString(&DllName, L"kernel32.");
119 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
120 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
121 ok_strings(&Str1, &Str2, L"kernel32.");
122 cleanup(&Str1, Buf1, sizeof(Buf1));
123 cleanup(&Str2, Buf2, sizeof(Buf2));
124
125 RtlInitUnicodeString(&DllName, L".kernel32");
126 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
127 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
128 ok_strings(&Str1, &Str2, L".kernel32.DLL");
129 cleanup(&Str1, Buf1, sizeof(Buf1));
130 cleanup(&Str2, Buf2, sizeof(Buf2));
131
132 RtlInitUnicodeString(&DllName, L"..kernel32");
133 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
134 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
135 ok_strings(&Str1, &Str2, L"..kernel32");
136 cleanup(&Str1, Buf1, sizeof(Buf1));
137 cleanup(&Str2, Buf2, sizeof(Buf2));
138
139 RtlInitUnicodeString(&DllName, L".kernel32.");
140 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
141 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
142 ok_strings(&Str1, &Str2, L".kernel32.");
143 cleanup(&Str1, Buf1, sizeof(Buf1));
144 cleanup(&Str2, Buf2, sizeof(Buf2));
145
146
147 RtlInitUnicodeString(&DllName, L"test\\kernel32.dll");
148 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
149 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
150 ok_strings(&Str1, &Str2, L"kernel32.dll");
151 cleanup(&Str1, Buf1, sizeof(Buf1));
152 cleanup(&Str2, Buf2, sizeof(Buf2));
153
154 RtlInitUnicodeString(&DllName, L"test/kernel32.dll");
155 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
156 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
157 ok_strings(&Str1, &Str2, L"kernel32.dll");
158 cleanup(&Str1, Buf1, sizeof(Buf1));
159 cleanup(&Str2, Buf2, sizeof(Buf2));
160
161 RtlInitUnicodeString(&DllName, L"test.dll/kernel32");
162 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
163 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
164 ok_strings(&Str1, &Str2, L"kernel32.DLL");
165 cleanup(&Str1, Buf1, sizeof(Buf1));
166 cleanup(&Str2, Buf2, sizeof(Buf2));
167
168 RtlInitUnicodeString(&DllName, L"test.dll\\kernel32");
169 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
170 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
171 ok_strings(&Str1, &Str2, L"kernel32.DLL");
172 cleanup(&Str1, Buf1, sizeof(Buf1));
173 cleanup(&Str2, Buf2, sizeof(Buf2));
174
175 RtlInitUnicodeString(&DllName, L"//");
176 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
177 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
178 ok_strings(&Str1, &Str2, L".DLL");
179 cleanup(&Str1, Buf1, sizeof(Buf1));
180 cleanup(&Str2, Buf2, sizeof(Buf2));
181
182 // Bug :)
183 RtlInitUnicodeString(&DllName, L"\\");
184 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
185 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
186 if (wcsstr(Str1.Buffer, L"\\\\"))
187 {
188 trace("Allowing bug found in windows' implementation\n");
189 ok_strings(&Str1, &Str2, L"\\.DLL");
190 }
191 else
192 {
193 ok_strings(&Str1, &Str2, L".DLL");
194 }
195 cleanup(&Str1, Buf1, sizeof(Buf1));
196 cleanup(&Str2, Buf2, sizeof(Buf2));
197
198 RtlInitUnicodeString(&DllName, L"");
199 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
200 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
201 ok_strings(&Str1, &Str2, L".DLL");
202 cleanup(&Str1, Buf1, sizeof(Buf1));
203 cleanup(&Str2, Buf2, sizeof(Buf2));
204}
205
206static void test_allocations(void)
207{
208 WCHAR Buf1[10];
209 WCHAR Buf2[10];
210
211 UNICODE_STRING Str1, Str2;
212 UNICODE_STRING DllName;
213 NTSTATUS Status;
214
215 RtlInitEmptyUnicodeString(&Str1, Buf1, sizeof(Buf1));
216 RtlInitEmptyUnicodeString(&Str2, Buf2, sizeof(Buf2));
217
218 RtlInitUnicodeString(&DllName, L"kernel32.dll");
219 Status = RtlComputePrivatizedDllName_U(&DllName, &Str1, &Str2);
220 ok(Status == STATUS_SUCCESS, "0x%lx\n", Status);
221 ok_strings(&Str1, &Str2, L"kernel32.dll");
222 ok(Str1.Buffer != Buf1, "Expected a changed buffer\n");
223 ok(Str2.Buffer != Buf2, "Expected a changed buffer\n");
224 cleanup(&Str1, Buf1, sizeof(Buf1));
225 cleanup(&Str2, Buf2, sizeof(Buf2));
226}
227
228
229START_TEST(RtlComputePrivatizedDllName_U)
230{
231 if (!InitTestData())
232 return;
233
234 test_dllnames();
235 test_allocations();
236}
LocalDir
static WCHAR LocalDir[MAX_PATH+10]
Definition: RtlComputePrivatizedDllName_U.c:12
ProcessDir
static WCHAR ProcessDir[MAX_PATH]
Definition: RtlComputePrivatizedDllName_U.c:11
ok_strings_
static void ok_strings_(PCUNICODE_STRING RealName, PCUNICODE_STRING LocalName, LPCWSTR DllName, int line)
Definition: RtlComputePrivatizedDllName_U.c:44
InitTestData
static BOOL InitTestData()
Definition: RtlComputePrivatizedDllName_U.c:14
test_dllnames
static void test_dllnames(void)
Definition: RtlComputePrivatizedDllName_U.c:77
test_allocations
static void test_allocations(void)
Definition: RtlComputePrivatizedDllName_U.c:206
ok_strings
#define ok_strings(Real, Local, Dll)
Definition: RtlComputePrivatizedDllName_U.c:64
RTL_NUMBER_OF
#define RTL_NUMBER_OF(x)
Definition: RtlRegistry.c:12
StrU
const UNICODE_STRING StrU
Definition: RtlUnicodeStringToCountedOemString.c:14
trace
#define trace
Definition: atltest.h:70
ok
#define ok(value,...)
Definition: atltest.h:57
skip
#define skip(...)
Definition: atltest.h:64
START_TEST
#define START_TEST(x)
Definition: atltest.h:75
ok_
#define ok_(x1, x2)
Definition: atltest.h:61
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
Buffer
Definition: bufpool.h:45
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
FALSE
#define FALSE
Definition: types.h:117
NT_SUCCESS
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:33
MAX_PATH
#define MAX_PATH
Definition: compat.h:34
cleanup
static void cleanup(void)
Definition: main.c:1335
GetModuleFileNameW
DWORD WINAPI GetModuleFileNameW(HINSTANCE hModule, LPWSTR lpFilename, DWORD nSize)
Definition: loader.c:600
swprintf
#define swprintf
Definition: precomp.h:40
RtlAppendUnicodeToString
NTSTATUS RtlAppendUnicodeToString(IN PUNICODE_STRING Str1, IN PWSTR Str2)
Definition: string_lib.cpp:62
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
Status
Status
Definition: gdiplustypes.h:25
wcsstr
_CONST_RETURN wchar_t *__cdecl wcsstr(_In_z_ const wchar_t *_Str, _In_z_ const wchar_t *_SubStr)
wine_dbgstr_w
#define wine_dbgstr_w
Definition: kernel32.h:34
UINT
unsigned int UINT
Definition: ndis.h:50
RtlComputePrivatizedDllName_U
NTSYSAPI NTSTATUS NTAPI RtlComputePrivatizedDllName_U(_In_ PUNICODE_STRING DllName, _Inout_ PUNICODE_STRING RealName, _Inout_ PUNICODE_STRING LocalName)
Definition: path.c:586
RtlInitUnicodeString
NTSYSAPI VOID NTAPI RtlInitUnicodeString(PUNICODE_STRING DestinationString, PCWSTR SourceString)
RtlFreeUnicodeString
NTSYSAPI VOID NTAPI RtlFreeUnicodeString(PUNICODE_STRING UnicodeString)
UNICODE_NULL
#define UNICODE_NULL
RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END
#define RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END
Definition: rtl.h:25
RtlFindCharInUnicodeString
NTSTATUS NTAPI RtlFindCharInUnicodeString(_In_ ULONG Flags, _In_ PCUNICODE_STRING SearchString, _In_ PCUNICODE_STRING MatchString, _Out_ PUSHORT Position)
L
#define L(x)
Definition: ntvdm.h:50
USHORT
unsigned short USHORT
Definition: pedump.c:61
wcscmp
_Check_return_ _CRTIMP int __cdecl wcscmp(_In_z_ const wchar_t *_Str1, _In_z_ const wchar_t *_Str2)
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
_UNICODE_STRING
Definition: env_spec_w32.h:368
_UNICODE_STRING::Length
USHORT Length
Definition: env_spec_w32.h:369
_UNICODE_STRING::MaximumLength
USHORT MaximumLength
Definition: env_spec_w32.h:370
_UNICODE_STRING::Buffer
PWSTR Buffer
Definition: env_spec_w32.h:371
line
Definition: parser.c:49
RTL_CONSTANT_STRING
#define RTL_CONSTANT_STRING(s)
Definition: tunneltest.c:14
String
_Must_inspect_result_ _In_ WDFDEVICE _In_ WDFSTRING String
Definition: wdfdevice.h:2433
BufferSize
_In_ WDFMEMORY _Out_opt_ size_t * BufferSize
Definition: wdfmemory.h:254
WCHAR
__wchar_t WCHAR
Definition: xmlstorage.h:180
LPCWSTR
const WCHAR * LPCWSTR
Definition: xmlstorage.h:185