[KERNEL32_APITEST] Add a PCH.
[reactos.git] / modules / rostests / apitests / kernel32 / dosdev.c
1 /*
2 * Unit test suite for virtual substituted drive functions.
3 *
4 * Copyright 2011 Sam Arun Raj
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include "precomp.h"
22
23 #define SUBST_DRIVE_LETTER 'M'
24 #define SUBST_DRIVE "M:"
25 #define SUBST_DRIVE_NON_EXIST_DIR "M:\\deadbeef"
26 #define SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR "M:\\"
27 #define SUBST_DRIVE_SEARCH "M:\\*"
28 #define SUBST_DRIVE_LOWERCASE "m:"
29 #define SUBST_DRIVE_LOWERCASE_SEARCH "m:\\*"
30 #define SUBST_DRIVE2_LETTER 'R'
31 #define SUBST_DRIVE2 "R:"
32 #define SUBST_DRIVE2_WITH_TRAILING_PATH_SEPERATOR "R:\\"
33 #define SUBST_DRIVE2_SEARCH "R:\\*"
34
35 static void test_DefineDosDeviceA(void)
36 {
37 CHAR Buffer[MAX_PATH], Target[MAX_PATH];
38 BOOL Result;
39 UINT CharCount;
40 HANDLE hnd;
41 WIN32_FIND_DATAA Data;
42 UINT SystemDriveType, DriveType1, DriveType2;
43 DWORD dwMaskPrev, dwMaskCur;
44 CHAR c;
45
46 /* Choose the symbolic link target */
47 CharCount = GetSystemWindowsDirectoryA(Target, MAX_PATH);
48 ok(CharCount > 0, "Failed to get windows directory\n");
49 c = Target[3];
50 Target[3] = '\0';
51 SystemDriveType = GetDriveTypeA(Target);
52 Target[3] = c;
53
54 /* Test with a subst drive pointing to another substed drive */
55 dwMaskPrev = GetLogicalDrives();
56 Result = DefineDosDeviceA(0, SUBST_DRIVE, Target);
57 ok(Result, "Failed to subst drive\n");
58 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
59 Result = DefineDosDeviceA(0, SUBST_DRIVE2, SUBST_DRIVE);
60 ok(Result, "Failed to subst drive\n");
61 DriveType2 = GetDriveTypeA(SUBST_DRIVE2_WITH_TRAILING_PATH_SEPERATOR);
62 dwMaskCur = GetLogicalDrives();
63 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
64 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
65 ok((dwMaskCur & (1 << (SUBST_DRIVE2_LETTER - 'A'))), "Drive bit is not set\n");
66 hnd = FindFirstFileA(SUBST_DRIVE2_SEARCH, &Data);
67 ok(hnd != INVALID_HANDLE_VALUE, "Failed to open subst drive\n");
68 if (hnd) FindClose(hnd);
69 ok(DriveType1 == DriveType2, "subst drive types don't match\n");
70 ok(DriveType1 == SystemDriveType, "subst drive types don't match\n");
71 ok(DriveType2 == SystemDriveType, "subst drive types don't match\n");
72 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, Target);
73 ok(Result, "Failed to remove subst drive using NULL Target name\n");
74 hnd = FindFirstFileA(SUBST_DRIVE2_SEARCH, &Data);
75 ok(hnd == INVALID_HANDLE_VALUE, "Opened subst drive when it should fail, we removed the target\n");
76 if (hnd) FindClose(hnd);
77 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE2, SUBST_DRIVE);
78 ok(Result, "Failed to remove subst drive using NULL Target name\n");
79 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
80 ok(!Result, "Subst drive is present even after remove attempt\n");
81 Result = QueryDosDeviceA(SUBST_DRIVE2, Buffer, MAX_PATH);
82 ok(!Result, "Subst drive is present even after remove attempt\n");
83 dwMaskCur = GetLogicalDrives();
84 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
85
86 /* Test using lowercase drive letter */
87 dwMaskPrev = GetLogicalDrives();
88 Result = DefineDosDeviceA(0, SUBST_DRIVE_LOWERCASE, Target);
89 ok(Result, "Failed to subst drive using lowercase drive letter\n");
90 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
91 ok(DriveType1 == SystemDriveType, "subst drive types don't match\n");
92 dwMaskCur = GetLogicalDrives();
93 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
94 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
95 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
96 ok(hnd != INVALID_HANDLE_VALUE, "Failed to open subst drive\n");
97 if (hnd) FindClose(hnd);
98 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE_LOWERCASE, Target);
99 ok(Result, "Failed to remove subst drive using lowercase drive letter\n");
100 Result = QueryDosDeviceA(SUBST_DRIVE_LOWERCASE, Buffer, MAX_PATH);
101 ok(!Result, "Subst drive is present even after remove attempt\n");
102 dwMaskCur = GetLogicalDrives();
103 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
104
105 /* Test remove without using DDD_EXACT_MATCH_ON_REMOVE */
106 dwMaskPrev = GetLogicalDrives();
107 Result = DefineDosDeviceA(0, SUBST_DRIVE, Target);
108 ok(Result, "Failed to subst drive\n");
109 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
110 ok(DriveType1 == SystemDriveType, "subst drive types don't match\n");
111 dwMaskCur = GetLogicalDrives();
112 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
113 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
114 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
115 ok(hnd != INVALID_HANDLE_VALUE, "Failed to open subst drive\n");
116 if (hnd) FindClose(hnd);
117 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION, SUBST_DRIVE, NULL);
118 ok(Result, "Failed to remove subst drive using NULL Target name\n");
119 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
120 ok(!Result, "Subst drive is present even after remove attempt\n");
121 dwMaskCur = GetLogicalDrives();
122 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
123
124 /* Test multiple adds and multiple removes in add order */
125 dwMaskPrev = GetLogicalDrives();
126 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp1");
127 ok(Result, "Failed to subst drive\n");
128 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp2");
129 ok(Result, "Failed to subst drive\n");
130 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp3");
131 ok(Result, "Failed to subst drive\n");
132 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
133 ok(DriveType1 != SystemDriveType, "subst drive types match when it shouldn't\n");
134 ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_FILE_NOT_FOUND), GetLastError());
135 dwMaskCur = GetLogicalDrives();
136 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
137 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
138 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp1");
139 ok(Result, "Failed to remove subst drive\n");
140 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
141 ok(Result, "Failed to query subst drive\n");
142 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp3") == 0), "Subst drive is not pointing to correct target\n");
143 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp2");
144 ok(Result, "Failed to remove subst drive\n");
145 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
146 ok(Result, "Failed to query subst drive\n");
147 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp3") == 0), "Subst drive is not pointing to correct target\n");
148 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp3");
149 ok(Result, "Failed to remove subst drive\n");
150 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
151 ok(!Result, "Subst drive is present even after remove attempt\n");
152 dwMaskCur = GetLogicalDrives();
153 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
154
155 /* Test multiple adds and multiple removes in reverse order */
156 dwMaskPrev = GetLogicalDrives();
157 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp1");
158 ok(Result, "Failed to subst drive\n");
159 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp2");
160 ok(Result, "Failed to subst drive\n");
161 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp3");
162 ok(Result, "Failed to subst drive\n");
163 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
164 ok(DriveType1 != SystemDriveType, "subst drive types match when it shouldn't\n");
165 ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_FILE_NOT_FOUND), GetLastError());
166 dwMaskCur = GetLogicalDrives();
167 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
168 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
169 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp3");
170 ok(Result, "Failed to remove subst drive\n");
171 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
172 ok(Result, "Failed to query subst drive\n");
173 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp2") == 0), "Subst drive is not pointing to correct target\n");
174 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp2");
175 ok(Result, "Failed to remove subst drive\n");
176 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
177 ok(Result, "Failed to query subst drive\n");
178 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp1") == 0), "Subst drive is not pointing to correct target\n");
179 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp1");
180 ok(Result, "Failed to remove subst drive\n");
181 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
182 ok(!Result, "Subst drive is present even after remove attempt\n");
183 dwMaskCur = GetLogicalDrives();
184 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
185
186 /* Test multiple adds and multiple removes out of order */
187 dwMaskPrev = GetLogicalDrives();
188 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp1");
189 ok(Result, "Failed to subst drive\n");
190 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp2");
191 ok(Result, "Failed to subst drive\n");
192 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp3");
193 ok(Result, "Failed to subst drive\n");
194 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp4");
195 ok(Result, "Failed to subst drive\n");
196 Result = DefineDosDeviceA(0, SUBST_DRIVE, "C:\\temp5");
197 ok(Result, "Failed to subst drive\n");
198 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
199 ok(DriveType1 != SystemDriveType, "subst drive types match when it shouldn't\n");
200 ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_FILE_NOT_FOUND), GetLastError());
201 dwMaskCur = GetLogicalDrives();
202 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
203 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
204 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp2");
205 ok(Result, "Failed to remove subst drive\n");
206 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
207 ok(Result, "Failed to query subst drive\n");
208 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp5") == 0), "Subst drive is not pointing to correct target\n");
209 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp5");
210 ok(Result, "Failed to remove subst drive\n");
211 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
212 ok(Result, "Failed to query subst drive\n");
213 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp4") == 0), "Subst drive is not pointing to correct target\n");
214 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp1");
215 ok(Result, "Failed to remove subst drive\n");
216 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
217 ok(Result, "Failed to query subst drive\n");
218 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp4") == 0), "Subst drive is not pointing to correct target\n");
219 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp3");
220 ok(Result, "Failed to remove subst drive\n");
221 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
222 ok(Result, "Failed to query subst drive\n");
223 if (Result) ok((_stricmp(Buffer, "\\??\\C:\\temp4") == 0), "Subst drive is not pointing to correct target");
224 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, "C:\\temp4");
225 ok(Result, "Failed to remove subst drive\n");
226 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
227 ok(!Result, "Subst drive is present even after remove attempt\n");
228 dwMaskCur = GetLogicalDrives();
229 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
230
231 /* Test with trailing '\' appended to TargetPath */
232 dwMaskPrev = GetLogicalDrives();
233 snprintf(Buffer, sizeof(Buffer), "%s\\\\\\", Target);
234 Result = DefineDosDeviceA(0, SUBST_DRIVE, Buffer);
235 ok(Result, "Failed to subst drive\n");
236 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
237 ok(DriveType1 == SystemDriveType, "subst drive types don't match\n");
238 dwMaskCur = GetLogicalDrives();
239 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
240 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
241 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
242 ok(hnd != INVALID_HANDLE_VALUE, "Failed to open subst drive\n");
243 if (hnd) FindClose(hnd);
244 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, Buffer);
245 ok(Result, "Failed to remove subst drive using NULL Target name\n");
246 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
247 ok(!Result, "Subst drive is present even after remove attempt\n");
248 dwMaskCur = GetLogicalDrives();
249 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
250
251 /* Test with trailing '\' appended to TargetPath and DDD_RAW_TARGET_PATH flag */
252 dwMaskPrev = GetLogicalDrives();
253 snprintf(Buffer, sizeof(Buffer), "\\??\\%s\\\\\\", Target);
254 Result = DefineDosDeviceA(DDD_RAW_TARGET_PATH, SUBST_DRIVE, Buffer);
255 ok(Result, "Failed to subst drive\n");
256 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
257 ok(DriveType1 != SystemDriveType, "subst drive types match when they shouldn't\n");
258 dwMaskCur = GetLogicalDrives();
259 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
260 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
261 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
262 ok(hnd == INVALID_HANDLE_VALUE, "Opened subst drive when it should fail\n");
263 ok(GetLastError() == ERROR_INVALID_NAME, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_INVALID_NAME), GetLastError());
264 if (hnd) FindClose(hnd);
265 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION, SUBST_DRIVE, NULL);
266 ok(Result, "Failed to remove subst drive using NULL Target name\n");
267 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
268 ok(!Result, "Subst drive is present even after remove attempt\n");
269 dwMaskCur = GetLogicalDrives();
270 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
271
272 /* Test using trailing \ against drive letter */
273 dwMaskPrev = GetLogicalDrives();
274 Result = DefineDosDeviceA(0, SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR, Target);
275 ok(!Result, "Subst drive using trailing path seperator, this should not happen\n");
276 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
277 ok(DriveType1 != SystemDriveType, "subst drive types match when it shouldn't\n");
278 dwMaskCur = GetLogicalDrives();
279 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
280 ok(!(dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is set when it shouldn't\n");
281 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
282 ok(hnd == INVALID_HANDLE_VALUE, "Opened subst drive when it should fail\n");
283 if (hnd) FindClose(hnd);
284 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR, Target);
285 ok(!Result, "Removing Subst drive using trailing path seperator passed when it should fail\n");
286 Result = QueryDosDeviceA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR, Buffer, MAX_PATH);
287 ok(!Result, "Subst drive is present when it should not be created in the first place\n");
288 dwMaskCur = GetLogicalDrives();
289 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
290
291 /* Test using arbitary string, not necessarily a DOS drive letter */
292 dwMaskPrev = GetLogicalDrives();
293 Result = DefineDosDeviceA(0, "!QHello:", Target);
294 ok(Result, "Failed to subst drive using non-DOS drive name\n");
295 dwMaskCur = GetLogicalDrives();
296 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
297 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, "!QHello:", Target);
298 ok(Result, "Failed to subst drive using non-DOS drive name\n");
299 Result = QueryDosDeviceA("!QHello:", Buffer, MAX_PATH);
300 ok(!Result, "Subst drive is present even after remove attempt\n");
301 dwMaskCur = GetLogicalDrives();
302 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
303
304 /* Test by subst a drive to itself */
305 dwMaskPrev = GetLogicalDrives();
306 Result = DefineDosDeviceA(0, SUBST_DRIVE, SUBST_DRIVE);
307 ok(Result, "Failed to subst drive\n");
308 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
309 ok(DriveType1 != SystemDriveType, "subst drive types match when it shouldn't\n");
310 ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_FILE_NOT_FOUND), GetLastError());
311 dwMaskCur = GetLogicalDrives();
312 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
313 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
314 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
315 ok(hnd == INVALID_HANDLE_VALUE, "Opened subst drive when it should fail\n");
316 if (hnd) FindClose(hnd);
317 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, SUBST_DRIVE);
318 ok(Result, "Failed to remove subst drive using lowercase drive letter\n");
319 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
320 ok(!Result, "Subst drive is present even after remove attempt\n");
321 dwMaskCur = GetLogicalDrives();
322 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
323
324 /* Test by subst a drive to an non-existent folder under itself */
325 dwMaskPrev = GetLogicalDrives();
326 Result = DefineDosDeviceA(0, SUBST_DRIVE, SUBST_DRIVE_NON_EXIST_DIR);
327 ok(Result, "Failed to subst drive\n");
328 DriveType1 = GetDriveTypeA(SUBST_DRIVE_WITH_TRAILING_PATH_SEPERATOR);
329 ok(DriveType1 != SystemDriveType, "subst drive types match when it shouldn't\n");
330 ok(GetLastError() == ERROR_FILE_NOT_FOUND, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_FILE_NOT_FOUND), GetLastError());
331 dwMaskCur = GetLogicalDrives();
332 ok(dwMaskCur != dwMaskPrev, "Drive masks match when it shouldn't\n");
333 ok((dwMaskCur & (1 << (SUBST_DRIVE_LETTER - 'A'))), "Drive bit is not set\n");
334 hnd = FindFirstFileA(SUBST_DRIVE_SEARCH, &Data);
335 ok(hnd == INVALID_HANDLE_VALUE, "Opened subst drive when it should fail\n");
336 if (hnd) FindClose(hnd);
337 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, SUBST_DRIVE_NON_EXIST_DIR);
338 ok(Result, "Failed to remove subst drive using lowercase drive letter\n");
339 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
340 ok(!Result, "Subst drive is present even after remove attempt\n");
341 dwMaskCur = GetLogicalDrives();
342 ok(dwMaskCur == dwMaskPrev, "Drive masks don't match\n");
343 }
344
345 static void test_QueryDosDeviceA(void)
346 {
347 CHAR Buffer[MAX_PATH], Target[MAX_PATH];
348 BOOL Result;
349 UINT CharCount;
350
351 /* Choose the symbolic link target */
352 CharCount = GetSystemWindowsDirectoryA(Target, MAX_PATH);
353 ok(CharCount > 0, "Failed to get windows directory\n");
354
355 Result = DefineDosDeviceA(0, SUBST_DRIVE, Target);
356 ok(Result, "Failed to subst drive\n");
357 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, 0);
358 ok(!Result, "Should fail as the buffer passed is supposed to be small\n");
359 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_INSUFFICIENT_BUFFER), GetLastError());
360 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
361 ok(Result, "failed to get target path\n");
362 ok(_strnicmp(Buffer, "\\??\\", 4) == 0, "The target returned does have correct prefix set\n");
363 ok(stricmp(&Buffer[4], Target) == 0, "The target returned does not match the one set\n");
364 Result = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_EXACT_MATCH_ON_REMOVE, SUBST_DRIVE, Target);
365 ok(Result, "Failed to remove subst drive using lowercase drive letter\n");
366 Result = QueryDosDeviceA(SUBST_DRIVE, Buffer, MAX_PATH);
367
368 /* This will try to retrieve all existing MS-DOS device names */
369 Result = QueryDosDeviceA(NULL, Buffer, 0);
370 ok(!Result, "Should fail as the buffer passed is supposed to be small\n");
371 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Wrong last error. Expected %lu, got %lu\n", (DWORD)(ERROR_INSUFFICIENT_BUFFER), GetLastError());
372 }
373
374 START_TEST(dosdev)
375 {
376 test_DefineDosDeviceA();
377 test_QueryDosDeviceA();
378 }