1 /* Unit test suite for Ntdll directory functions
3 * Copyright 2007 Jeff Latimer
4 * Copyright 2007 Andrey Turkin
5 * Copyright 2008 Jeff Zaroyko
6 * Copyright 2009 Dan Kegel
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 * We use function pointers here as there is no import library for NTDLL on
31 /* Define WIN32_NO_STATUS so MSVC does not give us duplicate macro
32 * definition errors when we get to winnt.h
34 #define WIN32_NO_STATUS
36 #include "wine/test.h"
39 static NTSTATUS (WINAPI
*pNtClose
)( PHANDLE
);
40 static NTSTATUS (WINAPI
*pNtOpenFile
) ( PHANDLE
, ACCESS_MASK
, POBJECT_ATTRIBUTES
, PIO_STATUS_BLOCK
, ULONG
, ULONG
);
41 static NTSTATUS (WINAPI
*pNtQueryDirectoryFile
)(HANDLE
,HANDLE
,PIO_APC_ROUTINE
,PVOID
,PIO_STATUS_BLOCK
,
42 PVOID
,ULONG
,FILE_INFORMATION_CLASS
,BOOLEAN
,PUNICODE_STRING
,BOOLEAN
);
43 static BOOLEAN (WINAPI
*pRtlCreateUnicodeStringFromAsciiz
)(PUNICODE_STRING
,LPCSTR
);
44 static BOOL (WINAPI
*pRtlDosPathNameToNtPathName_U
)( LPCWSTR
, PUNICODE_STRING
, PWSTR
*, CURDIR
* );
45 static VOID (WINAPI
*pRtlInitUnicodeString
)( PUNICODE_STRING
, LPCWSTR
);
46 static NTSTATUS (WINAPI
*pRtlMultiByteToUnicodeN
)( LPWSTR dst
, DWORD dstlen
, LPDWORD reslen
,
47 LPCSTR src
, DWORD srclen
);
49 /* The attribute sets to test */
51 int todo
; /* set if it doesn't work on wine yet */
52 const DWORD attr
; /* desired attribute */
53 const char *name
; /* filename to use */
54 const char *target
; /* what to point to (only for reparse pts) */
55 const char *description
; /* for error messages */
56 int nfound
; /* How many were found (expect 1) */
57 WCHAR nameW
[20]; /* unicode version of name (filled in later) */
59 { 0, FILE_ATTRIBUTE_NORMAL
, "n.tmp", NULL
, "normal" },
60 { 1, FILE_ATTRIBUTE_HIDDEN
, "h.tmp", NULL
, "hidden" },
61 { 1, FILE_ATTRIBUTE_SYSTEM
, "s.tmp", NULL
, "system" },
62 { 0, FILE_ATTRIBUTE_DIRECTORY
, "d.tmp", NULL
, "directory" },
65 static const int max_test_dir_size
= 20; /* size of above plus some for .. etc */
67 /* Create a test directory full of attribute test files, clear counts */
68 static void set_up_attribute_test(const char *testdirA
)
72 ok(CreateDirectoryA(testdirA
, NULL
),
73 "couldn't create dir '%s', error %d\n", testdirA
, GetLastError());
75 for (i
=0; testfiles
[i
].name
; i
++) {
77 pRtlMultiByteToUnicodeN(testfiles
[i
].nameW
, sizeof(testfiles
[i
].nameW
), NULL
, testfiles
[i
].name
, strlen(testfiles
[i
].name
)+1);
79 sprintf(buf
, "%s\\%s", testdirA
, testfiles
[i
].name
);
80 testfiles
[i
].nfound
= 0;
81 if (testfiles
[i
].attr
& FILE_ATTRIBUTE_DIRECTORY
) {
82 ok(CreateDirectoryA(buf
, NULL
),
83 "couldn't create dir '%s', error %d\n", buf
, GetLastError());
85 HANDLE h
= CreateFileA(buf
,
86 GENERIC_READ
|GENERIC_WRITE
,
87 0, NULL
, CREATE_ALWAYS
,
88 testfiles
[i
].attr
, 0);
89 ok( h
!= INVALID_HANDLE_VALUE
, "failed to create temp file '%s'\n", buf
);
95 /* Remove the given test directory and the attribute test files, if any */
96 static void tear_down_attribute_test(const char *testdirA
)
100 for (i
=0; testfiles
[i
].name
; i
++) {
103 sprintf(buf
, "%s\\%s", testdirA
, testfiles
[i
].name
);
104 if (testfiles
[i
].attr
& FILE_ATTRIBUTE_DIRECTORY
) {
105 ret
= RemoveDirectory(buf
);
106 ok(ret
|| (GetLastError() == ERROR_PATH_NOT_FOUND
),
107 "Failed to rmdir %s, error %d\n", buf
, GetLastError());
109 ret
= DeleteFile(buf
);
110 ok(ret
|| (GetLastError() == ERROR_PATH_NOT_FOUND
),
111 "Failed to rm %s, error %d\n", buf
, GetLastError());
114 RemoveDirectoryA(testdirA
);
117 /* Match one found file against testfiles[], increment count if found */
118 static void tally_test_file(FILE_BOTH_DIRECTORY_INFORMATION
*dir_info
)
122 (FILE_ATTRIBUTE_SYSTEM
|FILE_ATTRIBUTE_HIDDEN
|FILE_ATTRIBUTE_DIRECTORY
|FILE_ATTRIBUTE_REPARSE_POINT
);
123 DWORD attrib
= dir_info
->FileAttributes
& attribmask
;
124 WCHAR
*nameW
= dir_info
->FileName
;
125 int namelen
= dir_info
->FileNameLength
/ sizeof(WCHAR
);
130 for (i
=0; testfiles
[i
].name
; i
++) {
131 int len
= strlen(testfiles
[i
].name
);
132 if (namelen
!= len
|| memcmp(nameW
, testfiles
[i
].nameW
, len
*sizeof(WCHAR
)))
134 if (testfiles
[i
].todo
) {
136 ok (attrib
== (testfiles
[i
].attr
& attribmask
), "file %s: expected %s (%x), got %x (is your linux new enough?)\n", testfiles
[i
].name
, testfiles
[i
].description
, testfiles
[i
].attr
, attrib
);
138 ok (attrib
== (testfiles
[i
].attr
& attribmask
), "file %s: expected %s (%x), got %x (is your linux new enough?)\n", testfiles
[i
].name
, testfiles
[i
].description
, testfiles
[i
].attr
, attrib
);
140 testfiles
[i
].nfound
++;
143 ok(testfiles
[i
].name
!= NULL
, "unexpected file found\n");
146 static void test_NtQueryDirectoryFile(void)
148 OBJECT_ATTRIBUTES attr
;
149 UNICODE_STRING ntdirname
;
150 char testdirA
[MAX_PATH
];
151 WCHAR testdirW
[MAX_PATH
];
155 UINT data_len
; /* length of dir data */
156 BYTE data
[8192]; /* directory data */
157 FILE_BOTH_DIRECTORY_INFORMATION
*dir_info
;
162 /* Clean up from prior aborted run, if any, then set up test files */
163 ok(GetTempPathA(MAX_PATH
, testdirA
), "couldn't get temp dir\n");
164 strcat(testdirA
, "NtQueryDirectoryFile.tmp");
165 tear_down_attribute_test(testdirA
);
166 set_up_attribute_test(testdirA
);
168 /* Read the directory and note which files are found */
169 pRtlMultiByteToUnicodeN(testdirW
, sizeof(testdirW
), NULL
, testdirA
, strlen(testdirA
)+1);
170 if (!pRtlDosPathNameToNtPathName_U(testdirW
, &ntdirname
, NULL
, NULL
))
172 ok(0,"RtlDosPathNametoNtPathName_U failed\n");
175 InitializeObjectAttributes(&attr
, &ntdirname
, 0, 0, NULL
);
176 status
= pNtOpenFile( &dirh
, SYNCHRONIZE
| FILE_LIST_DIRECTORY
, &attr
, &io
,
178 FILE_SYNCHRONOUS_IO_NONALERT
|FILE_OPEN_FOR_BACKUP_INTENT
|FILE_DIRECTORY_FILE
);
179 ok (status
== STATUS_SUCCESS
, "failed to open dir '%s', ret 0x%x, error %d\n", testdirA
, status
, GetLastError());
180 if (status
!= STATUS_SUCCESS
) {
181 skip("can't test if we can't open the directory\n");
185 pNtQueryDirectoryFile( dirh
, NULL
, NULL
, NULL
, &io
, data
, sizeof(data
),
186 FileBothDirectoryInformation
, FALSE
, NULL
, TRUE
);
187 ok (U(io
).Status
== STATUS_SUCCESS
, "filed to query directory; status %x\n", U(io
).Status
);
188 data_len
= io
.Information
;
189 ok (data_len
>= sizeof(FILE_BOTH_DIRECTORY_INFORMATION
), "not enough data in directory\n");
193 while ((data_pos
< data_len
) && (numfiles
< max_test_dir_size
)) {
194 dir_info
= (FILE_BOTH_DIRECTORY_INFORMATION
*)(data
+ data_pos
);
196 tally_test_file(dir_info
);
198 if (dir_info
->NextEntryOffset
== 0) {
199 pNtQueryDirectoryFile( dirh
, 0, NULL
, NULL
, &io
, data
, sizeof(data
),
200 FileBothDirectoryInformation
, FALSE
, NULL
, FALSE
);
201 if (U(io
).Status
== STATUS_NO_MORE_FILES
)
203 ok (U(io
).Status
== STATUS_SUCCESS
, "filed to query directory; status %x\n", U(io
).Status
);
204 data_len
= io
.Information
;
205 if (data_len
< sizeof(FILE_BOTH_DIRECTORY_INFORMATION
))
209 data_pos
+= dir_info
->NextEntryOffset
;
213 ok(numfiles
< max_test_dir_size
, "too many loops\n");
215 for (i
=0; testfiles
[i
].name
; i
++)
216 ok(testfiles
[i
].nfound
== 1, "Wrong number %d of %s files found\n",
217 testfiles
[i
].nfound
, testfiles
[i
].description
);
221 tear_down_attribute_test(testdirA
);
224 START_TEST(directory
)
226 HMODULE hntdll
= GetModuleHandleA("ntdll.dll");
229 skip("not running on NT, skipping test\n");
233 pNtClose
= (void *)GetProcAddress(hntdll
, "NtClose");
234 pNtOpenFile
= (void *)GetProcAddress(hntdll
, "NtOpenFile");
235 pNtQueryDirectoryFile
= (void *)GetProcAddress(hntdll
, "NtQueryDirectoryFile");
236 pRtlCreateUnicodeStringFromAsciiz
= (void *)GetProcAddress(hntdll
, "RtlCreateUnicodeStringFromAsciiz");
237 pRtlDosPathNameToNtPathName_U
= (void *)GetProcAddress(hntdll
, "RtlDosPathNameToNtPathName_U");
238 pRtlInitUnicodeString
= (void *)GetProcAddress(hntdll
, "RtlInitUnicodeString");
239 pRtlMultiByteToUnicodeN
= (void *)GetProcAddress(hntdll
,"RtlMultiByteToUnicodeN");
241 test_NtQueryDirectoryFile();