mirror of
https://github.com/reactos/reactos.git
synced 2026-06-09 09:23:04 +08:00
bae799382a
- Limit the HarddiskVolume loop to 32 trials (instead of an unknown number of them). Ideally the list of volumes has to be queried from the volume manager, instead of hardcoding them (TODO). - Verify that the buffer returned by IoVolumeDeviceToDosName() is indeed a unicode multi-string (ends with two NULs), and, in the case it specifies a volume GUID, that it's a valid one.
99 lines
3.8 KiB
C
99 lines
3.8 KiB
C
/*
|
|
* PROJECT: ReactOS kernel-mode tests
|
|
* LICENSE: LGPL-2.1-or-later (https://spdx.org/licenses/LGPL-2.1-or-later)
|
|
* PURPOSE: Kernel-Mode Test Suite Volume Device test
|
|
* COPYRIGHT: Copyright 2019 Thomas Faber (thomas.faber@reactos.org)
|
|
*/
|
|
|
|
#include <kmt_test.h>
|
|
#include <mountmgr.h>
|
|
|
|
static
|
|
void
|
|
TestIoVolumeDeviceToDosName(void)
|
|
{
|
|
NTSTATUS Status;
|
|
ULONG VolumeNumber;
|
|
WCHAR VolumeDeviceNameBuffer[32];
|
|
UNICODE_STRING VolumeDeviceName;
|
|
PFILE_OBJECT FileObject;
|
|
PDEVICE_OBJECT DeviceObject;
|
|
UNICODE_STRING DosName;
|
|
UNICODE_STRING DosVolumePrefix = RTL_CONSTANT_STRING(L"\\\\?\\Volume");
|
|
|
|
RtlInitEmptyUnicodeString(&VolumeDeviceName,
|
|
VolumeDeviceNameBuffer,
|
|
sizeof(VolumeDeviceNameBuffer));
|
|
// TODO: Query the partition/volume manager for the list of volumes.
|
|
for (VolumeNumber = 0; VolumeNumber < 32; ++VolumeNumber)
|
|
{
|
|
Status = RtlStringCbPrintfW(VolumeDeviceName.Buffer,
|
|
VolumeDeviceName.MaximumLength,
|
|
L"\\Device\\HarddiskVolume%lu",
|
|
VolumeNumber);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
trace("RtlStringCbPrintfW(%lu) failed with 0x%lx\n",
|
|
VolumeNumber, Status);
|
|
break;
|
|
}
|
|
|
|
RtlInitUnicodeString(&VolumeDeviceName, VolumeDeviceNameBuffer);
|
|
Status = IoGetDeviceObjectPointer(&VolumeDeviceName,
|
|
READ_CONTROL,
|
|
&FileObject,
|
|
&DeviceObject);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
trace("IoGetDeviceObjectPointer(%wZ) failed with 0x%lx\n",
|
|
&VolumeDeviceName, Status);
|
|
continue;
|
|
}
|
|
|
|
Status = IoVolumeDeviceToDosName(DeviceObject, &DosName);
|
|
ok_eq_hex(Status, STATUS_SUCCESS);
|
|
if (!skip(NT_SUCCESS(Status), "No DOS name\n"))
|
|
{
|
|
trace("DOS name for %wZ is %wZ\n", &VolumeDeviceName, &DosName);
|
|
|
|
/* The DosName should contain one NUL-terminated string (always there?),
|
|
* plus one final NUL-terminator */
|
|
ok(DosName.MaximumLength == DosName.Length + sizeof(UNICODE_NULL),
|
|
"Unexpected DOS name maximum length %hu, expected %hu\n",
|
|
DosName.MaximumLength, DosName.Length + sizeof(UNICODE_NULL));
|
|
ok(DosName.Length >= sizeof(UNICODE_NULL),
|
|
"DOS name too short (length: %lu)\n",
|
|
DosName.Length / sizeof(WCHAR));
|
|
ok(DosName.Buffer[DosName.Length / sizeof(WCHAR)] == UNICODE_NULL,
|
|
"Missing NUL-terminator (1)\n");
|
|
ok(DosName.Buffer[DosName.MaximumLength / sizeof(WCHAR) - 1] == UNICODE_NULL,
|
|
"Missing NUL-terminator (2)\n");
|
|
|
|
/* The DOS name is either a drive letter, or a
|
|
* volume GUID name (if the volume is not mounted) */
|
|
if (DosName.Length == 2 * sizeof(WCHAR))
|
|
{
|
|
ok(DosName.Buffer[0] >= L'A' &&
|
|
DosName.Buffer[0] <= L'Z' &&
|
|
DosName.Buffer[1] == L':',
|
|
"Unexpected drive letter: %wZ\n", &DosName);
|
|
}
|
|
else
|
|
{
|
|
ok(RtlPrefixUnicodeString(&DosVolumePrefix, &DosName, FALSE),
|
|
"Unexpected volume path: %wZ\n", &DosName);
|
|
ok(MOUNTMGR_IS_DOS_VOLUME_NAME(&DosName),
|
|
"Invalid DOS volume path returned: %wZ\n", &DosName);
|
|
}
|
|
RtlFreeUnicodeString(&DosName);
|
|
}
|
|
ObDereferenceObject(FileObject);
|
|
}
|
|
ok(VolumeNumber > 1, "No volumes found\n");
|
|
}
|
|
|
|
START_TEST(IoVolume)
|
|
{
|
|
TestIoVolumeDeviceToDosName();
|
|
}
|