ReactOS 0.4.16-dev-197-g92996da
ObSymbolicLink.c
Go to the documentation of this file.
1/*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
4 * PURPOSE: Kernel-Mode Test for Zw*SymbolicLinkObject
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8#include <kmt_test.h>
9
10#define NDEBUG
11#include <debug.h>
12
13static
14VOID
15TestQueryLink(
16 _In_ HANDLE LinkHandle,
17 _In_ PCUNICODE_STRING ExpectedTarget)
18{
19 NTSTATUS Status;
20 WCHAR QueriedTargetBuffer[32];
21 UNICODE_STRING QueriedTarget;
22 ULONG ResultLength;
23 PULONG pResultLength;
24 ULONG i;
25
26 for (i = 0; i < 2; i++)
27 {
28 if (i == 0)
29 {
30 pResultLength = &ResultLength;
31 }
32 else
33 {
34 pResultLength = NULL;
35 }
36
37 /* Query with NULL Buffer just gives us the length */
38 RtlInitEmptyUnicodeString(&QueriedTarget, NULL, 0);
39 if (pResultLength) ResultLength = 0x55555555;
40 Status = ZwQuerySymbolicLinkObject(LinkHandle,
41 &QueriedTarget,
42 pResultLength);
43 ok_eq_hex(Status, STATUS_BUFFER_TOO_SMALL);
44 ok_eq_uint(QueriedTarget.Length, 0);
45 ok_eq_uint(QueriedTarget.MaximumLength, 0);
46 ok_eq_pointer(QueriedTarget.Buffer, NULL);
47 if (pResultLength) ok_eq_ulong(ResultLength, ExpectedTarget->MaximumLength);
48
49 /* Query with Length-1 buffer */
50 RtlInitEmptyUnicodeString(&QueriedTarget,
51 QueriedTargetBuffer,
52 ExpectedTarget->Length - 1);
53 RtlFillMemory(&QueriedTargetBuffer, sizeof(QueriedTargetBuffer), 0x55);
54 if (pResultLength) ResultLength = 0x55555555;
55 Status = ZwQuerySymbolicLinkObject(LinkHandle,
56 &QueriedTarget,
57 pResultLength);
58 ok_eq_hex(Status, STATUS_BUFFER_TOO_SMALL);
59 ok_eq_uint(QueriedTarget.Length, 0);
60 ok_eq_uint(QueriedTarget.MaximumLength, ExpectedTarget->Length - 1);
61 ok_eq_pointer(QueriedTarget.Buffer, QueriedTargetBuffer);
62 ok_eq_uint(QueriedTarget.Buffer[0], 0x5555);
63 if (pResultLength) ok_eq_ulong(ResultLength, ExpectedTarget->MaximumLength);
64
65 /* Query with Length buffer */
66 RtlInitEmptyUnicodeString(&QueriedTarget,
67 QueriedTargetBuffer,
68 ExpectedTarget->Length);
69 RtlFillMemory(&QueriedTargetBuffer, sizeof(QueriedTargetBuffer), 0x55);
70 if (pResultLength) ResultLength = 0x55555555;
71 Status = ZwQuerySymbolicLinkObject(LinkHandle,
72 &QueriedTarget,
73 pResultLength);
74 ok_eq_uint(QueriedTarget.MaximumLength, ExpectedTarget->Length);
75 ok_eq_pointer(QueriedTarget.Buffer, QueriedTargetBuffer);
76 if (pResultLength &&
77 QueriedTarget.MaximumLength < ExpectedTarget->MaximumLength)
78 {
79 ok_eq_hex(Status, STATUS_BUFFER_TOO_SMALL);
80 ok_eq_uint(QueriedTarget.Length, 0);
81 ok_eq_uint(QueriedTarget.Buffer[0], 0x5555);
82 if (pResultLength) ok_eq_ulong(ResultLength, ExpectedTarget->MaximumLength);
83 }
84 else
85 {
86 ok_eq_hex(Status, STATUS_SUCCESS);
87 ok_eq_uint(QueriedTarget.Length, ExpectedTarget->Length);
88 ok(RtlEqualUnicodeString(&QueriedTarget, ExpectedTarget, FALSE), "%wZ != %wZ\n", &QueriedTarget, ExpectedTarget);
89 ok_eq_uint(QueriedTarget.Buffer[QueriedTarget.Length / sizeof(WCHAR)], 0x5555);
90 }
91
92 /* Query with Length+1 buffer */
93 RtlInitEmptyUnicodeString(&QueriedTarget,
94 QueriedTargetBuffer,
95 ExpectedTarget->Length + 1);
96 RtlFillMemory(&QueriedTargetBuffer, sizeof(QueriedTargetBuffer), 0x55);
97 if (pResultLength) ResultLength = 0x55555555;
98 Status = ZwQuerySymbolicLinkObject(LinkHandle,
99 &QueriedTarget,
100 pResultLength);
101 ok_eq_uint(QueriedTarget.MaximumLength, ExpectedTarget->Length + 1);
102 ok_eq_pointer(QueriedTarget.Buffer, QueriedTargetBuffer);
103 if (pResultLength &&
104 QueriedTarget.MaximumLength < ExpectedTarget->MaximumLength)
105 {
106 ok_eq_hex(Status, STATUS_BUFFER_TOO_SMALL);
107 ok_eq_uint(QueriedTarget.Length, 0);
108 ok_eq_uint(QueriedTarget.Buffer[0], 0x5555);
109 if (pResultLength) ok_eq_ulong(ResultLength, ExpectedTarget->MaximumLength);
110 }
111 else
112 {
113 ok_eq_hex(Status, STATUS_SUCCESS);
114 ok_eq_uint(QueriedTarget.Length, ExpectedTarget->Length);
115 ok(RtlEqualUnicodeString(&QueriedTarget, ExpectedTarget, FALSE), "%wZ != %wZ\n", &QueriedTarget, ExpectedTarget);
116 ok_eq_uint(QueriedTarget.Buffer[QueriedTarget.Length / sizeof(WCHAR)], 0x5555);
117 }
118
119 /* Query with Length+2 buffer */
120 RtlInitEmptyUnicodeString(&QueriedTarget,
121 QueriedTargetBuffer,
122 ExpectedTarget->Length + sizeof(WCHAR));
123 RtlFillMemory(&QueriedTargetBuffer, sizeof(QueriedTargetBuffer), 0x55);
124 if (pResultLength) ResultLength = 0x55555555;
125 Status = ZwQuerySymbolicLinkObject(LinkHandle,
126 &QueriedTarget,
127 pResultLength);
128 ok_eq_hex(Status, STATUS_SUCCESS);
129 ok_eq_uint(QueriedTarget.Length, ExpectedTarget->Length);
130 ok_eq_uint(QueriedTarget.MaximumLength, ExpectedTarget->Length + sizeof(WCHAR));
131 ok_eq_pointer(QueriedTarget.Buffer, QueriedTargetBuffer);
132 ok(RtlEqualUnicodeString(&QueriedTarget, ExpectedTarget, FALSE), "%wZ != %wZ\n", &QueriedTarget, ExpectedTarget);
133 if (pResultLength)
134 {
135 if (ExpectedTarget->MaximumLength >= ExpectedTarget->Length + sizeof(UNICODE_NULL))
136 ok_eq_uint(QueriedTarget.Buffer[QueriedTarget.Length / sizeof(WCHAR)], 0);
137 ok_eq_ulong(ResultLength, ExpectedTarget->MaximumLength);
138 }
139 else
140 {
141 ok_eq_uint(QueriedTarget.Buffer[QueriedTarget.Length / sizeof(WCHAR)], 0x5555);
142 }
143
144 /* Query with full-sized buffer */
145 RtlInitEmptyUnicodeString(&QueriedTarget,
146 QueriedTargetBuffer,
147 sizeof(QueriedTargetBuffer));
148 RtlFillMemory(&QueriedTargetBuffer, sizeof(QueriedTargetBuffer), 0x55);
149 if (pResultLength) ResultLength = 0x55555555;
150 Status = ZwQuerySymbolicLinkObject(LinkHandle,
151 &QueriedTarget,
152 pResultLength);
153 ok_eq_hex(Status, STATUS_SUCCESS);
154 ok_eq_uint(QueriedTarget.Length, ExpectedTarget->Length);
155 ok_eq_uint(QueriedTarget.MaximumLength, sizeof(QueriedTargetBuffer));
156 ok_eq_pointer(QueriedTarget.Buffer, QueriedTargetBuffer);
157 ok(RtlEqualUnicodeString(&QueriedTarget, ExpectedTarget, FALSE), "%wZ != %wZ\n", &QueriedTarget, ExpectedTarget);
158 if (pResultLength)
159 {
160 if (ExpectedTarget->MaximumLength >= ExpectedTarget->Length + sizeof(UNICODE_NULL))
161 ok_eq_uint(QueriedTarget.Buffer[QueriedTarget.Length / sizeof(WCHAR)], 0);
162 ok_eq_ulong(ResultLength, ExpectedTarget->MaximumLength);
163 }
164 else
165 {
166 ok_eq_uint(QueriedTarget.Buffer[QueriedTarget.Length / sizeof(WCHAR)], 0x5555);
167 }
168 }
169}
170
171START_TEST(ObSymbolicLink)
172{
173 NTSTATUS Status;
174 HANDLE LinkHandle;
175 HANDLE LinkHandle2;
176 UNICODE_STRING LinkName = RTL_CONSTANT_STRING(L"\\Device\\ObSymbolicLinkKmtestNamedPipe");
177 OBJECT_ATTRIBUTES ObjectAttributes;
178 UNICODE_STRING DefaultLinkTarget = RTL_CONSTANT_STRING(L"\\Device\\NamedPipe");
179 UNICODE_STRING LinkTarget = DefaultLinkTarget;
180
181 InitializeObjectAttributes(&ObjectAttributes,
182 &LinkName,
183 OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE,
184 NULL,
185 NULL);
186 LinkHandle = KmtInvalidPointer;
187 Status = ZwOpenSymbolicLinkObject(&LinkHandle,
188 SYMBOLIC_LINK_QUERY,
189 &ObjectAttributes);
190 ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
191 ok_eq_pointer(LinkHandle, NULL);
192 if (NT_SUCCESS(Status)) ObCloseHandle(LinkHandle, KernelMode);
193
194 /* Create it */
195 LinkHandle = KmtInvalidPointer;
196 Status = ZwCreateSymbolicLinkObject(&LinkHandle,
197 SYMBOLIC_LINK_QUERY,
198 &ObjectAttributes,
199 &LinkTarget);
200 ok_eq_hex(Status, STATUS_SUCCESS);
201 ok(LinkHandle != NULL && LinkHandle != KmtInvalidPointer, "LinkHandle = %p\n", LinkHandle);
202 if (skip(NT_SUCCESS(Status), "Failed to create link\n"))
203 {
204 return;
205 }
206
207 /* Now we should be able to open it */
208 LinkHandle2 = KmtInvalidPointer;
209 Status = ZwOpenSymbolicLinkObject(&LinkHandle2,
210 SYMBOLIC_LINK_QUERY,
211 &ObjectAttributes);
212 ok_eq_hex(Status, STATUS_SUCCESS);
213 ok(LinkHandle2 != NULL && LinkHandle2 != KmtInvalidPointer, "LinkHandle = %p\n", LinkHandle2);
214 if (!skip(NT_SUCCESS(Status), "Failed to open symlink\n"))
215 {
216 TestQueryLink(LinkHandle2, &LinkTarget);
217 ObCloseHandle(LinkHandle2, KernelMode);
218 }
219
220 /* Close it */
221 ObCloseHandle(LinkHandle, KernelMode);
222
223 /* Open fails again */
224 LinkHandle = KmtInvalidPointer;
225 Status = ZwOpenSymbolicLinkObject(&LinkHandle,
226 SYMBOLIC_LINK_QUERY,
227 &ObjectAttributes);
228 ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
229 ok_eq_pointer(LinkHandle, NULL);
230 if (NT_SUCCESS(Status)) ObCloseHandle(LinkHandle, KernelMode);
231
232 /* Create with broken LinkTarget */
233 LinkHandle = KmtInvalidPointer;
234 LinkTarget.Buffer = NULL;
235 Status = ZwCreateSymbolicLinkObject(&LinkHandle,
236 SYMBOLIC_LINK_QUERY,
237 &ObjectAttributes,
238 &LinkTarget);
239 ok_eq_hex(Status, STATUS_ACCESS_VIOLATION);
240 ok(LinkHandle == KmtInvalidPointer, "LinkHandle = %p\n", LinkHandle);
241 if (NT_SUCCESS(Status)) ObCloseHandle(LinkHandle, KernelMode);
242
243 /* Since it failed, open should also fail */
244 LinkHandle = KmtInvalidPointer;
245 Status = ZwOpenSymbolicLinkObject(&LinkHandle,
246 SYMBOLIC_LINK_QUERY,
247 &ObjectAttributes);
248 ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
249 ok_eq_pointer(LinkHandle, NULL);
250 if (NT_SUCCESS(Status)) ObCloseHandle(LinkHandle, KernelMode);
251
252 /* Create with valid, but not null-terminated LinkTarget */
253 LinkTarget = DefaultLinkTarget;
254 LinkTarget.MaximumLength = LinkTarget.Length;
255 LinkHandle = KmtInvalidPointer;
256 Status = ZwCreateSymbolicLinkObject(&LinkHandle,
257 SYMBOLIC_LINK_QUERY,
258 &ObjectAttributes,
259 &LinkTarget);
260 ok_eq_hex(Status, STATUS_SUCCESS);
261 ok(LinkHandle != NULL && LinkHandle != KmtInvalidPointer, "LinkHandle = %p\n", LinkHandle);
262 if (skip(NT_SUCCESS(Status), "Failed to create link\n"))
263 {
264 return;
265 }
266
267 /* Now we should be able to open it */
268 LinkHandle2 = KmtInvalidPointer;
269 Status = ZwOpenSymbolicLinkObject(&LinkHandle2,
270 SYMBOLIC_LINK_QUERY,
271 &ObjectAttributes);
272 ok_eq_hex(Status, STATUS_SUCCESS);
273 ok(LinkHandle2 != NULL && LinkHandle2 != KmtInvalidPointer, "LinkHandle = %p\n", LinkHandle2);
274 if (!skip(NT_SUCCESS(Status), "Failed to open symlink\n"))
275 {
276 TestQueryLink(LinkHandle2, &LinkTarget);
277 ObCloseHandle(LinkHandle2, KernelMode);
278 }
279
280 /* Close it */
281 ObCloseHandle(LinkHandle, KernelMode);
282}
ok_eq_pointer
#define ok_eq_pointer(value, expected)
Definition: apitest.h:59
ok_eq_hex
#define ok_eq_hex(value, expected)
Definition: apitest.h:77
ok_eq_ulong
#define ok_eq_ulong(value, expected)
Definition: apitest.h:63
ok_eq_uint
#define ok_eq_uint(value, expected)
Definition: apitest.h:61
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
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
ObjectAttributes
IN PUNICODE_STRING IN POBJECT_ATTRIBUTES ObjectAttributes
Definition: conport.c:36
NULL
#define NULL
Definition: types.h:112
FALSE
#define FALSE
Definition: types.h:117
NT_SUCCESS
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:33
Status
Status
Definition: gdiplustypes.h:25
i
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble const GLfloat const GLdouble const GLfloat GLint i
Definition: glfuncs.h:248
OBJ_KERNEL_HANDLE
#define OBJ_KERNEL_HANDLE
Definition: winternl.h:231
OBJ_CASE_INSENSITIVE
#define OBJ_CASE_INSENSITIVE
Definition: winternl.h:228
RtlFillMemory
#define RtlFillMemory(Dest, Length, Fill)
Definition: winternl.h:599
kmt_test.h
KmtInvalidPointer
#define KmtInvalidPointer
Definition: kmt_test.h:280
InitializeObjectAttributes
#define InitializeObjectAttributes(p, n, a, r, s)
Definition: reg.c:106
_In_
#define _In_
Definition: ms_sal.h:308
KernelMode
#define KernelMode
Definition: asm.h:34
ZwOpenSymbolicLinkObject
NTSYSAPI NTSTATUS NTAPI ZwOpenSymbolicLinkObject(_Out_ PHANDLE SymbolicLinkHandle, _In_ ACCESS_MASK DesiredAccess, _In_ POBJECT_ATTRIBUTES ObjectAttributes)
ZwCreateSymbolicLinkObject
NTSYSAPI NTSTATUS NTAPI ZwCreateSymbolicLinkObject(_Out_ PHANDLE SymbolicLinkHandle, _In_ ACCESS_MASK DesiredAccess, _In_ POBJECT_ATTRIBUTES ObjectAttributes, _In_ PUNICODE_STRING Name)
SYMBOLIC_LINK_QUERY
#define SYMBOLIC_LINK_QUERY
Definition: nt_native.h:1265
RtlEqualUnicodeString
NTSYSAPI BOOLEAN NTAPI RtlEqualUnicodeString(PUNICODE_STRING String1, PUNICODE_STRING String2, BOOLEAN CaseInSensitive)
UNICODE_NULL
#define UNICODE_NULL
STATUS_ACCESS_VIOLATION
#define STATUS_ACCESS_VIOLATION
Definition: ntstatus.h:242
L
#define L(x)
Definition: ntvdm.h:50
ObCloseHandle
NTSTATUS NTAPI ObCloseHandle(IN HANDLE Handle, IN KPROCESSOR_MODE AccessMode)
Definition: obhandle.c:3379
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
STATUS_BUFFER_TOO_SMALL
#define STATUS_BUFFER_TOO_SMALL
Definition: shellext.h:69
_OBJECT_ATTRIBUTES
Definition: umtypes.h:182
_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
RTL_CONSTANT_STRING
#define RTL_CONSTANT_STRING(s)
Definition: tunneltest.c:14
PULONG
uint32_t * PULONG
Definition: typedefs.h:59
ULONG
uint32_t ULONG
Definition: typedefs.h:59
STATUS_OBJECT_NAME_NOT_FOUND
#define STATUS_OBJECT_NAME_NOT_FOUND
Definition: udferr_usr.h:149
ResultLength
_Must_inspect_result_ _In_ WDFDEVICE _In_ DEVICE_REGISTRY_PROPERTY _In_ ULONG _Out_ PULONG ResultLength
Definition: wdfdevice.h:3776
WCHAR
__wchar_t WCHAR
Definition: xmlstorage.h:180
LinkTarget
_Inout_ PUNICODE_STRING LinkTarget
Definition: zwfuncs.h:292