ntdll/tests: Test NtQueryDirectoryFile with different ReturnSingleEntry and RestartScan flags.

This commit is contained in:
Grazvydas Ignotas 2011-08-04 23:24:48 +03:00 committed by Alexandre Julliard
parent d38909dade
commit fbf537fb60
1 changed files with 65 additions and 41 deletions

View File

@ -51,7 +51,8 @@ static NTSTATUS (WINAPI *pRtlWow64EnableFsRedirectionEx)( ULONG disable, ULONG *
/* The attribute sets to test */ /* The attribute sets to test */
static struct testfile_s { static struct testfile_s {
int todo; /* set if it doesn't work on wine yet */ BOOL todo; /* set if it doesn't work on wine yet */
BOOL attr_done; /* set if attributes were tested for this file already */
const DWORD attr; /* desired attribute */ const DWORD attr; /* desired attribute */
const char *name; /* filename to use */ const char *name; /* filename to use */
const char *target; /* what to point to (only for reparse pts) */ const char *target; /* what to point to (only for reparse pts) */
@ -59,13 +60,13 @@ static struct testfile_s {
int nfound; /* How many were found (expect 1) */ int nfound; /* How many were found (expect 1) */
WCHAR nameW[20]; /* unicode version of name (filled in later) */ WCHAR nameW[20]; /* unicode version of name (filled in later) */
} testfiles[] = { } testfiles[] = {
{ 0, FILE_ATTRIBUTE_NORMAL, "n.tmp", NULL, "normal" }, { 0, 0, FILE_ATTRIBUTE_NORMAL, "n.tmp", NULL, "normal" },
{ 1, FILE_ATTRIBUTE_HIDDEN, "h.tmp", NULL, "hidden" }, { 1, 0, FILE_ATTRIBUTE_HIDDEN, "h.tmp", NULL, "hidden" },
{ 1, FILE_ATTRIBUTE_SYSTEM, "s.tmp", NULL, "system" }, { 1, 0, FILE_ATTRIBUTE_SYSTEM, "s.tmp", NULL, "system" },
{ 0, FILE_ATTRIBUTE_DIRECTORY, "d.tmp", NULL, "directory" }, { 0, 0, FILE_ATTRIBUTE_DIRECTORY, "d.tmp", NULL, "directory" },
{ 0, FILE_ATTRIBUTE_DIRECTORY, ".", NULL, ". directory" }, { 0, 0, FILE_ATTRIBUTE_DIRECTORY, ".", NULL, ". directory" },
{ 0, FILE_ATTRIBUTE_DIRECTORY, "..", NULL, ".. directory" }, { 0, 0, FILE_ATTRIBUTE_DIRECTORY, "..", NULL, ".. directory" },
{ 0, 0, NULL } { 0, 0, 0, NULL }
}; };
static const int max_test_dir_size = 20; /* size of above plus some for .. etc */ static const int max_test_dir_size = 20; /* size of above plus some for .. etc */
@ -82,7 +83,6 @@ static void set_up_attribute_test(const char *testdirA)
char buf[MAX_PATH]; char buf[MAX_PATH];
pRtlMultiByteToUnicodeN(testfiles[i].nameW, sizeof(testfiles[i].nameW), NULL, testfiles[i].name, strlen(testfiles[i].name)+1); pRtlMultiByteToUnicodeN(testfiles[i].nameW, sizeof(testfiles[i].nameW), NULL, testfiles[i].name, strlen(testfiles[i].name)+1);
testfiles[i].nfound = 0;
if (strcmp(testfiles[i].name, ".") == 0 || strcmp(testfiles[i].name, "..") == 0) if (strcmp(testfiles[i].name, ".") == 0 || strcmp(testfiles[i].name, "..") == 0)
continue; continue;
sprintf(buf, "%s\\%s", testdirA, testfiles[i].name); sprintf(buf, "%s\\%s", testdirA, testfiles[i].name);
@ -100,6 +100,14 @@ static void set_up_attribute_test(const char *testdirA)
} }
} }
static void reset_found_files(void)
{
int i;
for (i = 0; testfiles[i].name; i++)
testfiles[i].nfound = 0;
}
/* Remove the given test directory and the attribute test files, if any */ /* Remove the given test directory and the attribute test files, if any */
static void tear_down_attribute_test(const char *testdirA) static void tear_down_attribute_test(const char *testdirA)
{ {
@ -138,11 +146,14 @@ static void tally_test_file(FILE_BOTH_DIRECTORY_INFORMATION *dir_info)
int len = strlen(testfiles[i].name); int len = strlen(testfiles[i].name);
if (namelen != len || memcmp(nameW, testfiles[i].nameW, len*sizeof(WCHAR))) if (namelen != len || memcmp(nameW, testfiles[i].nameW, len*sizeof(WCHAR)))
continue; continue;
if (testfiles[i].todo) { if (!testfiles[i].attr_done) {
todo_wine if (testfiles[i].todo) {
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); todo_wine
} else { 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);
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); } else {
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);
}
testfiles[i].attr_done = TRUE;
} }
testfiles[i].nfound++; testfiles[i].nfound++;
break; break;
@ -150,12 +161,9 @@ static void tally_test_file(FILE_BOTH_DIRECTORY_INFORMATION *dir_info)
ok(testfiles[i].name != NULL, "unexpected file found\n"); ok(testfiles[i].name != NULL, "unexpected file found\n");
} }
static void test_NtQueryDirectoryFile(void) static void test_flags_NtQueryDirectoryFile(OBJECT_ATTRIBUTES *attr, const char *testdirA,
BOOLEAN single_entry, BOOLEAN restart_flag)
{ {
OBJECT_ATTRIBUTES attr;
UNICODE_STRING ntdirname;
char testdirA[MAX_PATH];
WCHAR testdirW[MAX_PATH];
HANDLE dirh; HANDLE dirh;
IO_STATUS_BLOCK io; IO_STATUS_BLOCK io;
UINT data_pos; UINT data_pos;
@ -166,31 +174,19 @@ static void test_NtQueryDirectoryFile(void)
int numfiles; int numfiles;
int i; int i;
/* Clean up from prior aborted run, if any, then set up test files */ reset_found_files();
ok(GetTempPathA(MAX_PATH, testdirA), "couldn't get temp dir\n");
strcat(testdirA, "NtQueryDirectoryFile.tmp");
tear_down_attribute_test(testdirA);
set_up_attribute_test(testdirA);
/* Read the directory and note which files are found */ /* Read the directory and note which files are found */
pRtlMultiByteToUnicodeN(testdirW, sizeof(testdirW), NULL, testdirA, strlen(testdirA)+1); status = pNtOpenFile( &dirh, SYNCHRONIZE | FILE_LIST_DIRECTORY, attr, &io, FILE_OPEN,
if (!pRtlDosPathNameToNtPathName_U(testdirW, &ntdirname, NULL, NULL))
{
ok(0,"RtlDosPathNametoNtPathName_U failed\n");
goto done;
}
InitializeObjectAttributes(&attr, &ntdirname, OBJ_CASE_INSENSITIVE, 0, NULL);
status = pNtOpenFile( &dirh, SYNCHRONIZE | FILE_LIST_DIRECTORY, &attr, &io,
FILE_OPEN,
FILE_SYNCHRONOUS_IO_NONALERT|FILE_OPEN_FOR_BACKUP_INTENT|FILE_DIRECTORY_FILE); FILE_SYNCHRONOUS_IO_NONALERT|FILE_OPEN_FOR_BACKUP_INTENT|FILE_DIRECTORY_FILE);
ok (status == STATUS_SUCCESS, "failed to open dir '%s', ret 0x%x, error %d\n", testdirA, status, GetLastError()); ok (status == STATUS_SUCCESS, "failed to open dir '%s', ret 0x%x, error %d\n", testdirA, status, GetLastError());
if (status != STATUS_SUCCESS) { if (status != STATUS_SUCCESS) {
skip("can't test if we can't open the directory\n"); skip("can't test if we can't open the directory\n");
goto done; return;
} }
pNtQueryDirectoryFile( dirh, NULL, NULL, NULL, &io, data, sizeof(data), pNtQueryDirectoryFile( dirh, NULL, NULL, NULL, &io, data, sizeof(data),
FileBothDirectoryInformation, FALSE, NULL, FALSE ); FileBothDirectoryInformation, single_entry, NULL, restart_flag );
ok (U(io).Status == STATUS_SUCCESS, "filed to query directory; status %x\n", U(io).Status); ok (U(io).Status == STATUS_SUCCESS, "filed to query directory; status %x\n", U(io).Status);
data_len = io.Information; data_len = io.Information;
ok (data_len >= sizeof(FILE_BOTH_DIRECTORY_INFORMATION), "not enough data in directory\n"); ok (data_len >= sizeof(FILE_BOTH_DIRECTORY_INFORMATION), "not enough data in directory\n");
@ -204,7 +200,7 @@ static void test_NtQueryDirectoryFile(void)
if (dir_info->NextEntryOffset == 0) { if (dir_info->NextEntryOffset == 0) {
pNtQueryDirectoryFile( dirh, 0, NULL, NULL, &io, data, sizeof(data), pNtQueryDirectoryFile( dirh, 0, NULL, NULL, &io, data, sizeof(data),
FileBothDirectoryInformation, FALSE, NULL, FALSE ); FileBothDirectoryInformation, single_entry, NULL, FALSE );
if (U(io).Status == STATUS_NO_MORE_FILES) if (U(io).Status == STATUS_NO_MORE_FILES)
break; break;
ok (U(io).Status == STATUS_SUCCESS, "filed to query directory; status %x\n", U(io).Status); ok (U(io).Status == STATUS_SUCCESS, "filed to query directory; status %x\n", U(io).Status);
@ -220,17 +216,45 @@ static void test_NtQueryDirectoryFile(void)
ok(numfiles < max_test_dir_size, "too many loops\n"); ok(numfiles < max_test_dir_size, "too many loops\n");
for (i=0; testfiles[i].name; i++) { for (i=0; testfiles[i].name; i++) {
if (strcmp(testfiles[i].name, ".") == 0 || strcmp(testfiles[i].name, "..") == 0) { if ((strcmp(testfiles[i].name, ".") == 0 || strcmp(testfiles[i].name, "..") == 0) && (single_entry || !restart_flag)) {
todo_wine todo_wine
ok(testfiles[i].nfound == 1, "Wrong number %d of %s files found\n", ok(testfiles[i].nfound == 1, "Wrong number %d of %s files found (ReturnSingleEntry=%d,RestartScan=%d)\n",
testfiles[i].nfound, testfiles[i].description); testfiles[i].nfound, testfiles[i].description, single_entry, restart_flag);
} else { } else {
ok(testfiles[i].nfound == 1, "Wrong number %d of %s files found\n", ok(testfiles[i].nfound == 1, "Wrong number %d of %s files found (ReturnSingleEntry=%d,RestartScan=%d)\n",
testfiles[i].nfound, testfiles[i].description); testfiles[i].nfound, testfiles[i].description, single_entry, restart_flag);
} }
} }
pNtClose(dirh); pNtClose(dirh);
}
static void test_NtQueryDirectoryFile(void)
{
OBJECT_ATTRIBUTES attr;
UNICODE_STRING ntdirname;
char testdirA[MAX_PATH];
WCHAR testdirW[MAX_PATH];
/* Clean up from prior aborted run, if any, then set up test files */
ok(GetTempPathA(MAX_PATH, testdirA), "couldn't get temp dir\n");
strcat(testdirA, "NtQueryDirectoryFile.tmp");
tear_down_attribute_test(testdirA);
set_up_attribute_test(testdirA);
pRtlMultiByteToUnicodeN(testdirW, sizeof(testdirW), NULL, testdirA, strlen(testdirA)+1);
if (!pRtlDosPathNameToNtPathName_U(testdirW, &ntdirname, NULL, NULL))
{
ok(0, "RtlDosPathNametoNtPathName_U failed\n");
goto done;
}
InitializeObjectAttributes(&attr, &ntdirname, OBJ_CASE_INSENSITIVE, 0, NULL);
test_flags_NtQueryDirectoryFile(&attr, testdirA, FALSE, TRUE);
test_flags_NtQueryDirectoryFile(&attr, testdirA, FALSE, FALSE);
test_flags_NtQueryDirectoryFile(&attr, testdirA, TRUE, TRUE);
test_flags_NtQueryDirectoryFile(&attr, testdirA, TRUE, FALSE);
done: done:
tear_down_attribute_test(testdirA); tear_down_attribute_test(testdirA);
pRtlFreeUnicodeString(&ntdirname); pRtlFreeUnicodeString(&ntdirname);