mirror of
https://github.com/reactos/reactos.git
synced 2024-07-04 11:44:33 +00:00
123 lines
3.9 KiB
C
123 lines
3.9 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>
|
|
|
|
static
|
|
NTSTATUS
|
|
GetNextVolumeDevice(
|
|
_Inout_ PUNICODE_STRING VolumeDeviceName,
|
|
_Inout_ PULONG VolumeNumber,
|
|
_In_ NTSTATUS PreviousStatus)
|
|
{
|
|
NTSTATUS Status;
|
|
|
|
#ifndef __REACTOS__
|
|
*VolumeNumber++;
|
|
Status = RtlStringCbPrintfW(VolumeDeviceName->Buffer,
|
|
VolumeDeviceName->MaximumLength,
|
|
L"\\Device\\HarddiskVolume%lu",
|
|
*VolumeNumber);
|
|
#else
|
|
/* ROS's storage stack is old an broken, we don't have HarddiskVolumeN */
|
|
ULONG DiskNumber, PartitionNumber;
|
|
DiskNumber = *VolumeNumber >> 16;
|
|
PartitionNumber = *VolumeNumber & 0xffff;
|
|
if (!NT_SUCCESS(PreviousStatus))
|
|
{
|
|
if (PartitionNumber == 1)
|
|
{
|
|
/* Looks like this disk doesn't exist (or has no partitions),
|
|
* so we're done */
|
|
return STATUS_NO_MORE_ENTRIES;
|
|
}
|
|
DiskNumber++;
|
|
PartitionNumber = 0;
|
|
}
|
|
PartitionNumber++;
|
|
Status = RtlStringCbPrintfW(VolumeDeviceName->Buffer,
|
|
VolumeDeviceName->MaximumLength,
|
|
L"\\Device\\Harddisk%lu\\Partition%lu",
|
|
DiskNumber,
|
|
PartitionNumber);
|
|
*VolumeNumber = DiskNumber << 16 | PartitionNumber;
|
|
#endif
|
|
return Status;
|
|
}
|
|
|
|
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));
|
|
VolumeNumber = 0;
|
|
Status = STATUS_SUCCESS;
|
|
while (1)
|
|
{
|
|
Status = GetNextVolumeDevice(&VolumeDeviceName,
|
|
&VolumeNumber,
|
|
Status);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
trace("GetNextVolumeDevice(0x%lx) failed with %lx\n",
|
|
VolumeNumber, Status);
|
|
break;
|
|
}
|
|
|
|
RtlInitUnicodeString(&VolumeDeviceName, VolumeDeviceNameBuffer);
|
|
Status = IoGetDeviceObjectPointer(&VolumeDeviceName,
|
|
READ_CONTROL,
|
|
&FileObject,
|
|
&DeviceObject);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
trace("IoGetDeviceObjectPointer(%wZ) failed with %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);
|
|
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);
|
|
}
|
|
RtlFreeUnicodeString(&DosName);
|
|
}
|
|
ObDereferenceObject(FileObject);
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
ok(VolumeNumber > 1, "No volumes found\n");
|
|
}
|
|
|
|
START_TEST(IoVolume)
|
|
{
|
|
TestIoVolumeDeviceToDosName();
|
|
}
|