[tfaber] 68166: [NTDLL_APITEST] - Also test FileFsAttributeInformation in NtQueryVolumeInformationFile

Author: tfaber
Date: Wed Jun 17 06:54:17 2015
New Revision: 68166

URL: http://svn.reactos.org/svn/reactos?rev=68166&view=rev
Log:
[NTDLL_APITEST]
- Also test FileFsAttributeInformation in NtQueryVolumeInformationFile

Modified:
    trunk/rostests/apitests/ntdll/NtQueryVolumeInformationFile.c

Modified: trunk/rostests/apitests/ntdll/NtQueryVolumeInformationFile.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/NtQueryVol…
==============================================================================
--- trunk/rostests/apitests/ntdll/NtQueryVolumeInformationFile.c	[iso-8859-1] (original)
+++ trunk/rostests/apitests/ntdll/NtQueryVolumeInformationFile.c	[iso-8859-1] Wed Jun 17
06:54:17 2015
@@ -133,16 +133,122 @@
     status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, VolumeInfo,
sizeof(Buffer), FileFsVolumeInformation);
     ok(status == STATUS_SUCCESS, "Got status 0x%lx\n", status);
     ok(IoStatusBlock.Status == STATUS_SUCCESS, "IoStatusBlock.Status =
0x%lx\n", IoStatusBlock.Status);
-    ok(IoStatusBlock.Information >= FIELD_OFFSET(FILE_FS_VOLUME_INFORMATION,
VolumeLabel),
-           "IoStatusBlock.Information = %Iu, expected >=%lu\n",
IoStatusBlock.Information, (ULONG)sizeof(FILE_FS_VOLUME_INFORMATION));
+    ok(IoStatusBlock.Information == FIELD_OFFSET(FILE_FS_VOLUME_INFORMATION, VolumeLabel)
+ VolumeInfo->VolumeLabelLength,
+       "IoStatusBlock.Information = %Iu, expected %lu+%lu\n",
IoStatusBlock.Information, (ULONG)FIELD_OFFSET(FILE_FS_VOLUME_INFORMATION, VolumeLabel),
VolumeInfo->VolumeLabelLength);
     ok(VolumeInfo->VolumeCreationTime.QuadPart != 0x5555555555555555,
"VolumeInfo->VolumeCreationTime = %I64d\n",
VolumeInfo->VolumeCreationTime.QuadPart);
     ok(VolumeInfo->VolumeSerialNumber != 0x55555555,
"VolumeInfo->VolumeSerialNumber = %lu\n",
VolumeInfo->VolumeSerialNumber);
     ok(VolumeInfo->SupportsObjects == FALSE || VolumeInfo->SupportsObjects == TRUE,
"VolumeInfo->SupportsObjects = %u\n", VolumeInfo->SupportsObjects);
     ok(VolumeInfo->VolumeLabelLength % sizeof(WCHAR) == 0,
"VolumeInfo->VolumeLabelLength = %Iu\n", VolumeInfo->VolumeLabelLength);
     if (VolumeInfo->VolumeLabelLength >= sizeof(WCHAR))
         ok(VolumeInfo->VolumeLabel[VolumeInfo->VolumeLabelLength / sizeof(WCHAR) -
1] != 0x5555, "Incorrect VolumeLabel or Length\n");
+    trace("VolumeLabel = %.*ls\n", (int)VolumeInfo->VolumeLabelLength /
sizeof(WCHAR), VolumeInfo->VolumeLabel);
     ok(VolumeInfo->VolumeLabel[VolumeInfo->VolumeLabelLength / sizeof(WCHAR)] ==
0x5555,
        "Got %x\n", VolumeInfo->VolumeLabel[VolumeInfo->VolumeLabelLength
/ sizeof(WCHAR)]);
+}
+
+static
+VOID
+TestFileFsAttributeInformation(HANDLE handle)
+{
+    IO_STATUS_BLOCK IoStatusBlock;
+    ULONG Buffer[(sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + 32 * sizeof(WCHAR)) /
sizeof(ULONG)];
+    PFILE_FS_ATTRIBUTE_INFORMATION AttributeInfo =
(PFILE_FS_ATTRIBUTE_INFORMATION)Buffer;
+    NTSTATUS status;
+
+    status = NtQueryVolumeInformationFile(handle, NULL, NULL, 0,
FileFsAttributeInformation);
+    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Got status 0x%lx\n", status);
+
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, NULL, 0,
FileFsAttributeInformation);
+    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == 0x55555555, "IoStatusBlock.Status = 0x%lx\n",
IoStatusBlock.Status);
+    ok(IoStatusBlock.Information == (ULONG_PTR)0x5555555555555555,
"IoStatusBlock.Information = %Iu\n", IoStatusBlock.Information);
+
+    status = NtQueryVolumeInformationFile(handle, NULL, AttributeInfo, 0,
FileFsAttributeInformation);
+    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Got status 0x%lx\n", status);
+
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, (PUCHAR)Buffer + 2,
sizeof(FILE_FS_ATTRIBUTE_INFORMATION), FileFsAttributeInformation);
+    ok(status == STATUS_DATATYPE_MISALIGNMENT, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == 0x55555555, "IoStatusBlock.Status = 0x%lx\n",
IoStatusBlock.Status);
+    ok(IoStatusBlock.Information == (ULONG_PTR)0x5555555555555555,
"IoStatusBlock.Information = %Iu\n", IoStatusBlock.Information);
+
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo, 0,
FileFsAttributeInformation);
+    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == 0x55555555, "IoStatusBlock.Status = 0x%lx\n",
IoStatusBlock.Status);
+    ok(IoStatusBlock.Information == (ULONG_PTR)0x5555555555555555,
"IoStatusBlock.Information = %Iu\n", IoStatusBlock.Information);
+
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo,
FIELD_OFFSET(FILE_FS_ATTRIBUTE_INFORMATION, FileSystemName), FileFsAttributeInformation);
+    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == 0x55555555, "IoStatusBlock.Status = 0x%lx\n",
IoStatusBlock.Status);
+    ok(IoStatusBlock.Information == (ULONG_PTR)0x5555555555555555,
"IoStatusBlock.Information = %Iu\n", IoStatusBlock.Information);
+
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo,
sizeof(FILE_FS_ATTRIBUTE_INFORMATION) - 1, FileFsAttributeInformation);
+    ok(status == STATUS_INFO_LENGTH_MISMATCH, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == 0x55555555, "IoStatusBlock.Status = 0x%lx\n",
IoStatusBlock.Status);
+    ok(IoStatusBlock.Information == (ULONG_PTR)0x5555555555555555,
"IoStatusBlock.Information = %Iu\n", IoStatusBlock.Information);
+
+    RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo,
sizeof(FILE_FS_ATTRIBUTE_INFORMATION), FileFsAttributeInformation);
+    ok(status == STATUS_BUFFER_OVERFLOW, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == STATUS_BUFFER_OVERFLOW, "IoStatusBlock.Status =
0x%lx\n", IoStatusBlock.Status);
+    ok(AttributeInfo->FileSystemNameLength == sizeof(FILE_FS_ATTRIBUTE_INFORMATION) -
FIELD_OFFSET(FILE_FS_ATTRIBUTE_INFORMATION, FileSystemName),
+       "AttributeInfo->FileSystemNameLength = %Iu\n",
AttributeInfo->FileSystemNameLength);
+    ok(IoStatusBlock.Information == sizeof(FILE_FS_ATTRIBUTE_INFORMATION),
+       "IoStatusBlock.Information = %Iu, expected %lu\n",
IoStatusBlock.Information, (ULONG)sizeof(FILE_FS_ATTRIBUTE_INFORMATION));
+    ok(AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)] == 0x5555,
+       "Got %x\n",
AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)]);
+
+    RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo,
sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + 1, FileFsAttributeInformation);
+    ok(status == STATUS_BUFFER_OVERFLOW, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == STATUS_BUFFER_OVERFLOW, "IoStatusBlock.Status =
0x%lx\n", IoStatusBlock.Status);
+    ok(AttributeInfo->FileSystemNameLength == sizeof(FILE_FS_ATTRIBUTE_INFORMATION) +
1 - FIELD_OFFSET(FILE_FS_ATTRIBUTE_INFORMATION, FileSystemName),
+       "AttributeInfo->FileSystemNameLength = %Iu\n",
AttributeInfo->FileSystemNameLength);
+    ok(IoStatusBlock.Information == sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + 1,
+       "IoStatusBlock.Information = %Iu, expected %lu\n",
IoStatusBlock.Information, (ULONG)sizeof(FILE_FS_ATTRIBUTE_INFORMATION));
+    ok((AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)] & 0xff00) == 0x5500,
+       "Got %x\n",
AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)]);
+    ok(AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR) + 1] == 0x5555,
+       "Got %x\n",
AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength / sizeof(WCHAR) +
1]);
+
+    RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo,
sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + sizeof(WCHAR), FileFsAttributeInformation);
+    if (status != STATUS_SUCCESS)
+    {
+        ok(status == STATUS_BUFFER_OVERFLOW, "Got status 0x%lx\n", status);
+        ok(IoStatusBlock.Status == STATUS_BUFFER_OVERFLOW, "IoStatusBlock.Status =
0x%lx\n", IoStatusBlock.Status);
+        ok(AttributeInfo->FileSystemNameLength ==
sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + sizeof(WCHAR) -
FIELD_OFFSET(FILE_FS_ATTRIBUTE_INFORMATION, FileSystemName),
+           "AttributeInfo->FileSystemNameLength = %Iu\n",
AttributeInfo->FileSystemNameLength);
+        ok(IoStatusBlock.Information == sizeof(FILE_FS_ATTRIBUTE_INFORMATION) +
sizeof(WCHAR),
+           "IoStatusBlock.Information = %Iu, expected %lu\n",
IoStatusBlock.Information, (ULONG)sizeof(FILE_FS_ATTRIBUTE_INFORMATION));
+        ok(AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)] == 0x5555,
+           "Got %x\n",
AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)]);
+    }
+
+    RtlFillMemory(Buffer, sizeof(Buffer), 0x55);
+    RtlFillMemory(&IoStatusBlock, sizeof(IoStatusBlock), 0x55);
+    status = NtQueryVolumeInformationFile(handle, &IoStatusBlock, AttributeInfo,
sizeof(Buffer), FileFsAttributeInformation);
+    ok(status == STATUS_SUCCESS, "Got status 0x%lx\n", status);
+    ok(IoStatusBlock.Status == STATUS_SUCCESS, "IoStatusBlock.Status =
0x%lx\n", IoStatusBlock.Status);
+    ok(IoStatusBlock.Information == FIELD_OFFSET(FILE_FS_ATTRIBUTE_INFORMATION,
FileSystemName) + AttributeInfo->FileSystemNameLength,
+       "IoStatusBlock.Information = %Iu, expected %lu+%lu\n",
IoStatusBlock.Information, (ULONG)FIELD_OFFSET(FILE_FS_ATTRIBUTE_INFORMATION,
FileSystemName), AttributeInfo->FileSystemNameLength);
+    ok(AttributeInfo->FileSystemAttributes != 0x55555555,
"AttributeInfo->FileSystemAttributes = 0x%lx\n",
AttributeInfo->FileSystemAttributes);
+    ok(AttributeInfo->MaximumComponentNameLength != 0x55555555,
"AttributeInfo->MaximumComponentNameLength = 0x%lx\n",
AttributeInfo->MaximumComponentNameLength);
+    ok(AttributeInfo->FileSystemNameLength % sizeof(WCHAR) == 0,
"AttributeInfo->FileSystemNameLength = %Iu\n",
AttributeInfo->FileSystemNameLength);
+    ok(!wcsncmp(AttributeInfo->FileSystemName, L"NTFS", 4) ||
+       !wcsncmp(AttributeInfo->FileSystemName, L"FAT", 3) ||
+       !wcsncmp(AttributeInfo->FileSystemName, L"FAT32", 5),
+       "FileSystemName = %.*ls\n", (int)AttributeInfo->FileSystemNameLength
/ sizeof(WCHAR), AttributeInfo->FileSystemName);
+    trace("FileSystemName = %.*ls\n",
(int)AttributeInfo->FileSystemNameLength / sizeof(WCHAR),
AttributeInfo->FileSystemName);
+    ok(AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)] == 0x5555,
+       "Got %x\n",
AttributeInfo->FileSystemName[AttributeInfo->FileSystemNameLength /
sizeof(WCHAR)]);
 }
 
 START_TEST(NtQueryVolumeInformationFile)
@@ -160,7 +266,7 @@
 
     InitializeObjectAttributes(&attr, &pathW, OBJ_CASE_INSENSITIVE, NULL, NULL);
     status = NtOpenFile(&handle, SYNCHRONIZE|FILE_LIST_DIRECTORY, &attr,
&IoStatusBlock, FILE_SHARE_READ|FILE_SHARE_WRITE,
FILE_DIRECTORY_FILE|FILE_SYNCHRONOUS_IO_NONALERT);
-
+    ok(status == STATUS_SUCCESS, "NtOpenFile failed: 0x%lx\n", status);
     RtlFreeUnicodeString(&pathW);
 
     if (!NT_SUCCESS(status))
@@ -177,6 +283,7 @@
 
     TestFileFsDeviceInformation(handle);
     TestFileFsVolumeInformation(handle);
+    TestFileFsAttributeInformation(handle);
 
     NtClose(handle);
 }