huwcampbell.com
Reactos
1/*
2 * Unit test suite for drive functions.
3 *
4 * Copyright 2002 Dmitry Timoshkov
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 <stdarg.h>
22
23#include "wine/test.h"
24#include "windef.h"
25#include "winbase.h"
26#include "winerror.h"
27
28static DWORD (WINAPI *pGetDiskFreeSpaceExA)(LPCSTR, PULARGE_INTEGER, PULARGE_INTEGER, PULARGE_INTEGER);
29
30static void test_GetDriveTypeA(void)
31{
32 char drive[] = "?:\\";
33 char existing_drive_letter = 0;
34 DWORD logical_drives;
35 UINT type;
36
37 logical_drives = GetLogicalDrives();
38 ok(logical_drives != 0, "GetLogicalDrives error %ld\n", GetLastError());
39
40 for (drive[0] = 'A'; drive[0] <= 'Z'; drive[0]++)
41 {
42 type = GetDriveTypeA(drive);
43 ok(type > DRIVE_UNKNOWN && type <= DRIVE_RAMDISK,
44 "not a valid drive %c: type %u\n", drive[0], type);
45
46 if (!(logical_drives & 1))
47 ok(type == DRIVE_NO_ROOT_DIR,
48 "GetDriveTypeA should return DRIVE_NO_ROOT_DIR for inexistent drive %c: but not %u\n",
49 drive[0], type);
50 else if (type != DRIVE_NO_ROOT_DIR)
51 existing_drive_letter = drive[0];
52
53 logical_drives >>= 1;
54 }
55
56 if (!existing_drive_letter) {
57 skip("No drives found, skipping drive spec format tests.\n");
58 return;
59 }
60
61 drive[0] = existing_drive_letter;
62 drive[2] = 0; /* C: */
63 type = GetDriveTypeA(drive);
64 ok(type > DRIVE_NO_ROOT_DIR && type <= DRIVE_RAMDISK, "got %u for drive spec '%s'\n", type, drive);
65
66 drive[1] = '?'; /* C? */
67 type = GetDriveTypeA(drive);
68 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, drive);
69
70 drive[1] = 0; /* C */
71 type = GetDriveTypeA(drive);
72 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, drive);
73
74 drive[0] = '?'; /* the string "?" */
75 type = GetDriveTypeA(drive);
76 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, drive);
77
78 drive[0] = 0; /* the empty string */
79 type = GetDriveTypeA(drive);
80 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, drive);
81}
82
83static void test_GetDriveTypeW(void)
84{
85 WCHAR drive[] = {'?',':','\\',0};
86 WCHAR existing_drive_letter = 0;
87 DWORD logical_drives;
88 UINT type;
89
90 logical_drives = GetLogicalDrives();
91 ok(logical_drives != 0, "GetLogicalDrives error %ld\n", GetLastError());
92
93 for (drive[0] = 'A'; drive[0] <= 'Z'; drive[0]++)
94 {
95 type = GetDriveTypeW(drive);
96 ok(type > DRIVE_UNKNOWN && type <= DRIVE_RAMDISK,
97 "not a valid drive %c: type %u\n", drive[0], type);
98
99 if (!(logical_drives & 1))
100 ok(type == DRIVE_NO_ROOT_DIR,
101 "GetDriveTypeW should return DRIVE_NO_ROOT_DIR for inexistent drive %c: but not %u\n",
102 drive[0], type);
103 else if (type != DRIVE_NO_ROOT_DIR)
104 existing_drive_letter = drive[0];
105
106 logical_drives >>= 1;
107 }
108
109 if (!existing_drive_letter) {
110 skip("No drives found, skipping drive spec format tests.\n");
111 return;
112 }
113
114 drive[0] = existing_drive_letter;
115 drive[2] = 0; /* C: */
116 type = GetDriveTypeW(drive);
117 ok(type > DRIVE_NO_ROOT_DIR && type <= DRIVE_RAMDISK, "got %u for drive spec '%s'\n",
118 type, wine_dbgstr_w(drive));
119
120 drive[1] = '?'; /* C? */
121 type = GetDriveTypeW(drive);
122 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, wine_dbgstr_w(drive));
123
124 drive[1] = 0; /* C */
125 type = GetDriveTypeW(drive);
126 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, wine_dbgstr_w(drive));
127
128 drive[0] = '?'; /* the string "?" */
129 type = GetDriveTypeW(drive);
130 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, wine_dbgstr_w(drive));
131
132 drive[0] = 0; /* the empty string */
133 type = GetDriveTypeW(drive);
134 ok(type == DRIVE_NO_ROOT_DIR, "got %u for drive spec '%s'\n", type, wine_dbgstr_w(drive));
135}
136
137static void test_GetDiskFreeSpaceA(void)
138{
139 BOOL ret;
140 DWORD sectors_per_cluster, bytes_per_sector, free_clusters, total_clusters;
141 char volume_guid_path[50];
142 char drive[] = "?:\\";
143 DWORD logical_drives;
144
145 ret = GetDiskFreeSpaceA(NULL, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
146 ok(ret, "GetDiskFreeSpaceA error %ld\n", GetLastError());
147
148 ret = GetDiskFreeSpaceA("", §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
149 ok(!ret && (GetLastError() == ERROR_PATH_NOT_FOUND || GetLastError() == ERROR_INVALID_NAME),
150 "GetDiskFreeSpaceA(\"\"): ret=%d GetLastError=%ld\n",
151 ret, GetLastError());
152
153 ret = GetDiskFreeSpaceA("\\", §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
154 ok(ret, "GetDiskFreeSpaceA error %ld\n", GetLastError());
155
156 ret = GetDiskFreeSpaceA("/", §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
157 ok(ret, "GetDiskFreeSpaceA error %ld\n", GetLastError());
158
159 ret = GetDiskFreeSpaceA("C:\\", §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
160 ok(ret, "GetDiskFreeSpaceA error %ld\n", GetLastError());
161
162#ifdef __REACTOS__
163 if (GetNTVersion() >= _WIN32_WINNT_VISTA) {
164#endif
165 ret = GetVolumeNameForVolumeMountPointA("C:\\", volume_guid_path, ARRAY_SIZE(volume_guid_path));
166 ok(ret, "GetVolumeNameForVolumeMountPointA error %ld\n", GetLastError());
167
168 ret = GetDiskFreeSpaceA(volume_guid_path, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
169 ok(ret, "GetDiskFreeSpaceA error %ld\n", GetLastError());
170#ifdef __REACTOS__
171 }
172#endif
173
174 logical_drives = GetLogicalDrives();
175 ok(logical_drives != 0, "GetLogicalDrives error %ld\n", GetLastError());
176
177 for (drive[0] = 'A'; drive[0] <= 'Z'; drive[0]++)
178 {
179 UINT drivetype = GetDriveTypeA(drive);
180 /* Skip floppy drives because NT pops up a MessageBox if no
181 * floppy is present
182 */
183 if (drivetype != DRIVE_REMOVABLE && drivetype != DRIVE_NO_ROOT_DIR)
184 {
185 ret = GetDiskFreeSpaceA(drive, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
186 if (!(logical_drives & 1))
187 ok(!ret && (GetLastError() == ERROR_PATH_NOT_FOUND || GetLastError() == ERROR_INVALID_DRIVE),
188 "GetDiskFreeSpaceA(%s): ret=%d GetLastError=%ld\n",
189 drive, ret, GetLastError());
190 else
191 {
192
193 if (!ret)
194 /* GetDiskFreeSpaceA() should succeed, but it can fail with too many
195 different GetLastError() results to be usable for an ok() */
196 trace("GetDiskFreeSpaceA(%s) failed with %ld\n", drive, GetLastError());
197
198 if( GetVersion() & 0x80000000)
199 /* win3.0 through winME */
200 ok( total_clusters <= 65535,
201 "total clusters is %ld > 65535\n", total_clusters);
202 else if (pGetDiskFreeSpaceExA) {
203 /* NT, 2k, XP : GetDiskFreeSpace should be accurate */
204 ULARGE_INTEGER totEx, tot, d;
205
206 tot.QuadPart = sectors_per_cluster;
207 tot.QuadPart = (tot.QuadPart * bytes_per_sector) * total_clusters;
208 ret = pGetDiskFreeSpaceExA( drive, &d, &totEx, NULL);
209
210 if (!ret)
211 /* GetDiskFreeSpaceExA() should succeed, but it can fail with too many
212 different GetLastError() results to be usable for an ok() */
213 trace("GetDiskFreeSpaceExA(%s) failed with %ld\n", drive, GetLastError());
214
215 ok( bytes_per_sector == 0 || /* empty cd rom drive */
216 totEx.QuadPart <= tot.QuadPart,
217 "GetDiskFreeSpaceA should report at least as much bytes on disk %s as GetDiskFreeSpaceExA\n", drive);
218 }
219 }
220 }
221 logical_drives >>= 1;
222 }
223}
224
225static void test_GetDiskFreeSpaceW(void)
226{
227 BOOL ret;
228 DWORD sectors_per_cluster, bytes_per_sector, free_clusters, total_clusters;
229 WCHAR volume_guid_path[50];
230 WCHAR drive[] = {'?',':','\\',0};
231 DWORD logical_drives;
232 static const WCHAR empty_pathW[] = { 0 };
233 static const WCHAR root_pathW[] = { '\\', 0 };
234 static const WCHAR unix_style_root_pathW[] = { '/', 0 };
235 static const WCHAR c_drive_pathW[] = { 'C', ':', '\\', 0 };
236
237 ret = GetDiskFreeSpaceW(NULL, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
238 if (ret == 0 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
239 {
240 win_skip("GetDiskFreeSpaceW is not available\n");
241 return;
242 }
243 ok(ret, "GetDiskFreeSpaceW error %ld\n", GetLastError());
244
245 ret = GetDiskFreeSpaceW(empty_pathW, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
246 ok(!ret && GetLastError() == ERROR_PATH_NOT_FOUND,
247 "GetDiskFreeSpaceW(\"\"): ret=%d GetLastError=%ld\n",
248 ret, GetLastError());
249
250 ret = GetDiskFreeSpaceW(root_pathW, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
251 ok(ret, "GetDiskFreeSpaceW(\"\") error %ld\n", GetLastError());
252
253 ret = GetDiskFreeSpaceW(unix_style_root_pathW, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
254 ok(ret, "GetDiskFreeSpaceW error %ld\n", GetLastError());
255
256 ret = GetDiskFreeSpaceW(c_drive_pathW, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
257 ok(ret, "GetDiskFreeSpaceW error %ld\n", GetLastError());
258
259 ret = GetVolumeNameForVolumeMountPointW(c_drive_pathW, volume_guid_path, ARRAY_SIZE(volume_guid_path));
260 ok(ret, "GetVolumeNameForVolumeMountPointW error %ld\n", GetLastError());
261
262 ret = GetDiskFreeSpaceW(volume_guid_path, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
263 ok(ret, "GetDiskFreeSpaceW error %ld\n", GetLastError());
264
265 logical_drives = GetLogicalDrives();
266 ok(logical_drives != 0, "GetLogicalDrives error %ld\n", GetLastError());
267
268 for (drive[0] = 'A'; drive[0] <= 'Z'; drive[0]++)
269 {
270 UINT drivetype = GetDriveTypeW(drive);
271 /* Skip floppy drives because NT4 pops up a MessageBox if no floppy is present */
272 if (drivetype != DRIVE_REMOVABLE && drivetype != DRIVE_NO_ROOT_DIR)
273 {
274 ret = GetDiskFreeSpaceW(drive, §ors_per_cluster, &bytes_per_sector, &free_clusters, &total_clusters);
275 if (!(logical_drives & 1))
276 ok(!ret && GetLastError() == ERROR_PATH_NOT_FOUND,
277 "GetDiskFreeSpaceW(%c): ret=%d GetLastError=%ld\n",
278 drive[0], ret, GetLastError());
279 else if (!ret)
280 /* GetDiskFreeSpaceW() should succeed, but it can fail with too many
281 different GetLastError() results to be usable for an ok() */
282 trace("GetDiskFreeSpaceW(%c) failed with %ld\n", drive[0], GetLastError());
283 }
284 logical_drives >>= 1;
285 }
286}
287
288START_TEST(drive)
289{
290 HANDLE hkernel32 = GetModuleHandleA("kernel32");
291 pGetDiskFreeSpaceExA = (void *) GetProcAddress(hkernel32, "GetDiskFreeSpaceExA");
292
293 test_GetDriveTypeA();
294 test_GetDriveTypeW();
295
296 test_GetDiskFreeSpaceA();
297 test_GetDiskFreeSpaceW();
298}