5dee20057e4c051b2d15d9e87134938504e74ba3
[reactos.git] / rostests / apitests / ntdll / RtlDetermineDosPathNameType.c
1 /*
2 * PROJECT: ReactOS api tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: Test for RtlDetermineDosPathNameType_U/RtlDetermineDosPathNameType_Ustr
5 * PROGRAMMER: Thomas Faber <thfabba@gmx.de>
6 */
7
8 #define WIN32_NO_STATUS
9 #include <wine/test.h>
10 #include <pseh/pseh2.h>
11 #include <ndk/mmfuncs.h>
12 #include <ndk/rtlfuncs.h>
13
14 /*
15 ULONG
16 NTAPI
17 RtlDetermineDosPathNameType_U(
18 IN PCWSTR Path
19 );
20
21 ULONG
22 NTAPI
23 RtlDetermineDosPathNameType_Ustr(
24 IN PCUNICODE_STRING Path
25 );
26 */
27
28 static
29 ULONG
30 (NTAPI
31 *RtlDetermineDosPathNameType_Ustr)(
32 IN PCUNICODE_STRING Path
33 )
34 //= (PVOID)0x7c830669;
35 ;
36
37 static
38 PVOID
39 AllocateGuarded(
40 SIZE_T SizeRequested)
41 {
42 NTSTATUS Status;
43 SIZE_T Size = PAGE_ROUND_UP(SizeRequested + PAGE_SIZE);
44 PVOID VirtualMemory = NULL;
45 PCHAR StartOfBuffer;
46
47 Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_RESERVE, PAGE_NOACCESS);
48
49 if (!NT_SUCCESS(Status))
50 return NULL;
51
52 Size -= PAGE_SIZE;
53 if (Size)
54 {
55 Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_COMMIT, PAGE_READWRITE);
56 if (!NT_SUCCESS(Status))
57 {
58 Size = 0;
59 Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
60 ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
61 return NULL;
62 }
63 }
64
65 StartOfBuffer = VirtualMemory;
66 StartOfBuffer += Size - SizeRequested;
67
68 return StartOfBuffer;
69 }
70
71 static
72 VOID
73 FreeGuarded(
74 PVOID Pointer)
75 {
76 NTSTATUS Status;
77 PVOID VirtualMemory = (PVOID)PAGE_ROUND_DOWN((SIZE_T)Pointer);
78 SIZE_T Size = 0;
79
80 Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
81 ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
82 }
83
84 #define StartSeh() ExceptionStatus = STATUS_SUCCESS; _SEH2_TRY {
85 #define EndSeh(ExpectedStatus) } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { ExceptionStatus = _SEH2_GetExceptionCode(); } _SEH2_END; ok(ExceptionStatus == ExpectedStatus, "Exception %lx, expected %lx\n", ExceptionStatus, ExpectedStatus)
86
87 START_TEST(RtlDetermineDosPathNameType)
88 {
89 NTSTATUS ExceptionStatus;
90 RTL_PATH_TYPE PathType;
91 struct
92 {
93 PCWSTR FileName;
94 RTL_PATH_TYPE PathType;
95 } Tests[] =
96 {
97 { L"", RtlPathTypeRelative },
98 { L"xyz", RtlPathTypeRelative },
99 { L"CON", RtlPathTypeRelative },
100 { L"NUL", RtlPathTypeRelative },
101 { L":", RtlPathTypeRelative },
102 { L"::", RtlPathTypeDriveRelative },
103 { L":::", RtlPathTypeDriveRelative },
104 { L"::::", RtlPathTypeDriveRelative },
105 { L"\\", RtlPathTypeRooted },
106 { L"\\:", RtlPathTypeRooted },
107 { L"\\C:", RtlPathTypeRooted },
108 { L"\\C:\\", RtlPathTypeRooted },
109 { L"/", RtlPathTypeRooted },
110 { L"/:", RtlPathTypeRooted },
111 { L"/C:", RtlPathTypeRooted },
112 { L"/C:/", RtlPathTypeRooted },
113 { L"C", RtlPathTypeRelative },
114 { L"C:", RtlPathTypeDriveRelative },
115 { L"C:a", RtlPathTypeDriveRelative },
116 { L"C:a\\", RtlPathTypeDriveRelative },
117 { L"C:\\", RtlPathTypeDriveAbsolute },
118 { L"C:/", RtlPathTypeDriveAbsolute },
119 { L"C:\\a", RtlPathTypeDriveAbsolute },
120 { L"C:/a", RtlPathTypeDriveAbsolute },
121 { L"C:\\\\", RtlPathTypeDriveAbsolute },
122 { L"\\\\", RtlPathTypeUncAbsolute },
123 { L"\\\\\\", RtlPathTypeUncAbsolute },
124 { L"\\\\;", RtlPathTypeUncAbsolute },
125 { L"\\??\\", RtlPathTypeRooted },
126 { L"\\??\\UNC", RtlPathTypeRooted },
127 { L"\\??\\UNC\\", RtlPathTypeRooted },
128 { L"\\?", RtlPathTypeRooted },
129 { L"\\?\\", RtlPathTypeRooted },
130 { L"\\?\\UNC", RtlPathTypeRooted },
131 { L"\\?\\UNC\\", RtlPathTypeRooted },
132 { L"\\\\?\\UNC\\", RtlPathTypeLocalDevice },
133 { L"\\\\?", RtlPathTypeRootLocalDevice },
134 { L"\\\\??", RtlPathTypeUncAbsolute },
135 { L"\\\\??\\", RtlPathTypeUncAbsolute },
136 { L"\\\\??\\C:\\", RtlPathTypeUncAbsolute },
137 { L"\\\\.", RtlPathTypeRootLocalDevice },
138 { L"\\\\.\\", RtlPathTypeLocalDevice },
139 { L"\\\\.\\C:\\", RtlPathTypeLocalDevice },
140 { L"\\/", RtlPathTypeUncAbsolute },
141 { L"/\\", RtlPathTypeUncAbsolute },
142 { L"//", RtlPathTypeUncAbsolute },
143 { L"///", RtlPathTypeUncAbsolute },
144 { L"//;", RtlPathTypeUncAbsolute },
145 { L"//?", RtlPathTypeRootLocalDevice },
146 { L"/\\?", RtlPathTypeRootLocalDevice },
147 { L"\\/?", RtlPathTypeRootLocalDevice },
148 { L"//??", RtlPathTypeUncAbsolute },
149 { L"//??/", RtlPathTypeUncAbsolute },
150 { L"//??/C:/", RtlPathTypeUncAbsolute },
151 { L"//.", RtlPathTypeRootLocalDevice },
152 { L"\\/.", RtlPathTypeRootLocalDevice },
153 { L"/\\.", RtlPathTypeRootLocalDevice },
154 { L"//./", RtlPathTypeLocalDevice },
155 { L"//./C:/", RtlPathTypeLocalDevice },
156 };
157 ULONG i;
158 PWSTR FileName;
159 USHORT Length;
160
161 if (!RtlDetermineDosPathNameType_Ustr)
162 skip(0, "RtlDetermineDosPathNameType_Ustr unavailable\n");
163
164 StartSeh() RtlDetermineDosPathNameType_U(NULL); EndSeh(STATUS_ACCESS_VIOLATION);
165
166 if (RtlDetermineDosPathNameType_Ustr)
167 {
168 UNICODE_STRING PathString;
169 StartSeh() RtlDetermineDosPathNameType_Ustr(NULL); EndSeh(STATUS_ACCESS_VIOLATION);
170
171 RtlInitEmptyUnicodeString(&PathString, NULL, MAXUSHORT);
172 PathType = RtlDetermineDosPathNameType_Ustr(&PathString);
173 ok(PathType == RtlPathTypeRelative, "PathType = %d\n", PathType);
174 }
175
176 for (i = 0; i < sizeof(Tests) / sizeof(Tests[0]); i++)
177 {
178 Length = (USHORT)wcslen(Tests[i].FileName) * sizeof(WCHAR);
179 FileName = AllocateGuarded(Length + sizeof(UNICODE_NULL));
180 RtlCopyMemory(FileName, Tests[i].FileName, Length + sizeof(UNICODE_NULL));
181 StartSeh()
182 PathType = RtlDetermineDosPathNameType_U(FileName);
183 EndSeh(STATUS_SUCCESS);
184 ok(PathType == Tests[i].PathType, "PathType is %d, expected %d for '%S'\n", PathType, Tests[i].PathType, Tests[i].FileName);
185 FreeGuarded(FileName);
186
187 if (RtlDetermineDosPathNameType_Ustr)
188 {
189 UNICODE_STRING PathString;
190
191 FileName = AllocateGuarded(Length);
192 RtlCopyMemory(FileName, Tests[i].FileName, Length);
193 PathString.Buffer = FileName;
194 PathString.Length = Length;
195 PathString.MaximumLength = MAXUSHORT;
196 StartSeh()
197 PathType = RtlDetermineDosPathNameType_Ustr(&PathString);
198 EndSeh(STATUS_SUCCESS);
199 ok(PathType == Tests[i].PathType, "PathType is %d, expected %d for '%S'\n", PathType, Tests[i].PathType, Tests[i].FileName);
200 FreeGuarded(FileName);
201 }
202 }
203 }