mirror of
https://github.com/reactos/reactos.git
synced 2025-08-02 00:03:22 +00:00
[PARTMGR_APITEST] Add a test suite for the partition manager (#7591)
CORE-13525 Add a test for IOCTL_STORAGE_GET_DEVICE_NUMBER.
This commit is contained in:
Hermès Bélusca-Maïto
parent
c3e14b54ae
commit
e0e45ffa1a
5 changed files with 354 additions and 0 deletions
|
|
@ -39,6 +39,7 @@ add_subdirectory(netshell)
|
||||||
add_subdirectory(ntdll)
|
add_subdirectory(ntdll)
|
||||||
add_subdirectory(ole32)
|
add_subdirectory(ole32)
|
||||||
add_subdirectory(opengl32)
|
add_subdirectory(opengl32)
|
||||||
|
add_subdirectory(partmgr)
|
||||||
add_subdirectory(pefile)
|
add_subdirectory(pefile)
|
||||||
add_subdirectory(powrprof)
|
add_subdirectory(powrprof)
|
||||||
add_subdirectory(rtl)
|
add_subdirectory(rtl)
|
||||||
|
|
|
||||||
17
modules/rostests/apitests/partmgr/CMakeLists.txt
Normal file
17
modules/rostests/apitests/partmgr/CMakeLists.txt
Normal file
|
|
@ -0,0 +1,17 @@
|
||||||
|
|
||||||
|
list(APPEND SOURCE
|
||||||
|
StorDeviceNumber.c)
|
||||||
|
|
||||||
|
list(APPEND PCH_SKIP_SOURCE
|
||||||
|
testlist.c)
|
||||||
|
|
||||||
|
add_executable(partmgr_apitest
|
||||||
|
${SOURCE}
|
||||||
|
${PCH_SKIP_SOURCE})
|
||||||
|
|
||||||
|
target_link_libraries(partmgr_apitest wine ${PSEH_LIB})
|
||||||
|
set_module_type(partmgr_apitest win32cui)
|
||||||
|
add_importlibs(partmgr_apitest msvcrt kernel32 ntdll)
|
||||||
|
# TODO: Enable this when we get more than one source file to justify its use
|
||||||
|
#add_pch(partmgr_apitest precomp.h "${PCH_SKIP_SOURCE}")
|
||||||
|
add_rostests_file(TARGET partmgr_apitest)
|
||||||
306
modules/rostests/apitests/partmgr/StorDeviceNumber.c
Normal file
306
modules/rostests/apitests/partmgr/StorDeviceNumber.c
Normal file
|
|
@ -0,0 +1,306 @@
|
||||||
|
/*
|
||||||
|
* PROJECT: ReactOS API Tests
|
||||||
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
||||||
|
* PURPOSE: Test for IOCTL_STORAGE_GET_DEVICE_NUMBER
|
||||||
|
* COPYRIGHT: Copyright 2024 Hermès Bélusca-Maïto <hermes.belusca-maito@reactos.org>
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "precomp.h"
|
||||||
|
#include <ntddstor.h>
|
||||||
|
|
||||||
|
static LPCSTR wine_dbgstr_us(const UNICODE_STRING *us)
|
||||||
|
{
|
||||||
|
if (!us) return "(null)";
|
||||||
|
return wine_dbgstr_wn(us->Buffer, us->Length / sizeof(WCHAR));
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Flags combination allowing all the read, write and delete share modes.
|
||||||
|
* Currently similar to FILE_SHARE_VALID_FLAGS. */
|
||||||
|
#define FILE_SHARE_ALL \
|
||||||
|
(FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE)
|
||||||
|
|
||||||
|
static BOOLEAN
|
||||||
|
Test_Device_StorDeviceNumber(
|
||||||
|
_In_ PCWSTR NtDeviceName)
|
||||||
|
{
|
||||||
|
BOOLEAN Success = FALSE; // Suppose failure.
|
||||||
|
NTSTATUS Status;
|
||||||
|
HANDLE DeviceHandle1 = NULL, DeviceHandle2 = NULL;
|
||||||
|
FILE_FS_DEVICE_INFORMATION DeviceInfo;
|
||||||
|
STORAGE_DEVICE_NUMBER DeviceNumber;
|
||||||
|
UNICODE_STRING DeviceName;
|
||||||
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
||||||
|
IO_STATUS_BLOCK IoStatusBlock;
|
||||||
|
WCHAR NtLegacyDeviceName[MAX_PATH];
|
||||||
|
|
||||||
|
ULONG BufferSize;
|
||||||
|
struct { OBJECT_NAME_INFORMATION; WCHAR Buffer[MAX_PATH]; } DeviceName1Buffer;
|
||||||
|
PUNICODE_STRING DeviceName1 = &DeviceName1Buffer.Name;
|
||||||
|
struct { OBJECT_NAME_INFORMATION; WCHAR Buffer[MAX_PATH]; } DeviceName2Buffer;
|
||||||
|
PUNICODE_STRING DeviceName2 = &DeviceName2Buffer.Name;
|
||||||
|
|
||||||
|
/* Open a handle to the device */
|
||||||
|
RtlInitUnicodeString(&DeviceName, NtDeviceName);
|
||||||
|
InitializeObjectAttributes(&ObjectAttributes,
|
||||||
|
&DeviceName,
|
||||||
|
OBJ_CASE_INSENSITIVE,
|
||||||
|
NULL,
|
||||||
|
NULL);
|
||||||
|
Status = NtOpenFile(&DeviceHandle1,
|
||||||
|
FILE_READ_ATTRIBUTES | SYNCHRONIZE,
|
||||||
|
&ObjectAttributes,
|
||||||
|
&IoStatusBlock,
|
||||||
|
FILE_SHARE_ALL,
|
||||||
|
/* FILE_NON_DIRECTORY_FILE | */ FILE_SYNCHRONOUS_IO_NONALERT);
|
||||||
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
||||||
|
if (!NT_SUCCESS(Status))
|
||||||
|
{
|
||||||
|
skip("Device '%s': Opening failed\n", wine_dbgstr_us(&DeviceName));
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Verify the device information before proceeding further */
|
||||||
|
Status = NtQueryVolumeInformationFile(DeviceHandle1,
|
||||||
|
&IoStatusBlock,
|
||||||
|
&DeviceInfo,
|
||||||
|
sizeof(DeviceInfo),
|
||||||
|
FileFsDeviceInformation);
|
||||||
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
||||||
|
if (!NT_SUCCESS(Status))
|
||||||
|
{
|
||||||
|
skip("FileFsDeviceInformation('%s') failed, Status 0x%08lx\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), Status);
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Ignore volumes that are NOT on usual disks */
|
||||||
|
switch (DeviceInfo.DeviceType)
|
||||||
|
{
|
||||||
|
/* Testable devices */
|
||||||
|
case FILE_DEVICE_CD_ROM:
|
||||||
|
// case FILE_DEVICE_CD_ROM_FILE_SYSTEM:
|
||||||
|
case FILE_DEVICE_DISK:
|
||||||
|
// case FILE_DEVICE_DISK_FILE_SYSTEM:
|
||||||
|
// case FILE_DEVICE_NETWORK:
|
||||||
|
// case FILE_DEVICE_NETWORK_FILE_SYSTEM:
|
||||||
|
case FILE_DEVICE_VIRTUAL_DISK:
|
||||||
|
break;
|
||||||
|
|
||||||
|
/* Untestable devices */
|
||||||
|
default:
|
||||||
|
skip("Device '%s': Cannot test, device type %lu\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), DeviceInfo.DeviceType);
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
#if 0
|
||||||
|
if (DeviceInfo.DeviceType != FILE_DEVICE_DISK &&
|
||||||
|
DeviceInfo.DeviceType != FILE_DEVICE_VIRTUAL_DISK &&
|
||||||
|
DeviceInfo.DeviceType != FILE_DEVICE_CD_ROM)
|
||||||
|
{
|
||||||
|
skip("Device '%s': Cannot test, device type %lu\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), DeviceInfo.DeviceType);
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Retrieve the storage device number.
|
||||||
|
* Note that this call is unsupported if this is a dynamic volume.
|
||||||
|
* NOTE: Usually fails for floppy disks.
|
||||||
|
*/
|
||||||
|
Status = NtDeviceIoControlFile(DeviceHandle1,
|
||||||
|
NULL, NULL, NULL,
|
||||||
|
&IoStatusBlock,
|
||||||
|
IOCTL_STORAGE_GET_DEVICE_NUMBER,
|
||||||
|
NULL, 0,
|
||||||
|
&DeviceNumber, sizeof(DeviceNumber));
|
||||||
|
if (!NT_SUCCESS(Status))
|
||||||
|
{
|
||||||
|
skip("Device '%s': Couldn't retrieve disk number\n", wine_dbgstr_us(&DeviceName));
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
ok(DeviceNumber.DeviceType == DeviceInfo.DeviceType,
|
||||||
|
"Device '%s': Device type mismatch\n", wine_dbgstr_us(&DeviceName));
|
||||||
|
|
||||||
|
/* NOTE: this value is set to 0xFFFFFFFF (-1) for the disks that
|
||||||
|
* represent the physical paths of a multipath I/O (MPIO) disk. */
|
||||||
|
ok(DeviceNumber.DeviceNumber != ULONG_MAX,
|
||||||
|
"Device '%s': Invalid disk number reported\n", wine_dbgstr_us(&DeviceName));
|
||||||
|
if (DeviceNumber.DeviceNumber == ULONG_MAX)
|
||||||
|
goto Quit;
|
||||||
|
|
||||||
|
switch (DeviceInfo.DeviceType)
|
||||||
|
{
|
||||||
|
/* Testable devices */
|
||||||
|
case FILE_DEVICE_CD_ROM:
|
||||||
|
// case FILE_DEVICE_CD_ROM_FILE_SYSTEM:
|
||||||
|
{
|
||||||
|
/* CD-ROMs don't have partitions, their partition number is -1 */
|
||||||
|
ok(DeviceNumber.PartitionNumber == ULONG_MAX,
|
||||||
|
"Device '%s': Invalid partition number (%lu) reported, expected ULONG_MAX (-1)\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), DeviceNumber.PartitionNumber);
|
||||||
|
|
||||||
|
/* Map to an NT device name */
|
||||||
|
RtlStringCchPrintfW(NtLegacyDeviceName, _countof(NtLegacyDeviceName),
|
||||||
|
L"\\Device\\CdRom%lu",
|
||||||
|
DeviceNumber.DeviceNumber);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case FILE_DEVICE_DISK:
|
||||||
|
// case FILE_DEVICE_DISK_FILE_SYSTEM:
|
||||||
|
case FILE_DEVICE_VIRTUAL_DISK:
|
||||||
|
{
|
||||||
|
/* Check whether this is a floppy or a partitionable device */
|
||||||
|
if (DeviceInfo.Characteristics & FILE_FLOPPY_DISKETTE)
|
||||||
|
{
|
||||||
|
/* Floppies don't have partitions, their partition number is -1 */
|
||||||
|
ok(DeviceNumber.PartitionNumber == ULONG_MAX,
|
||||||
|
"Device '%s': Invalid partition number (%lu) reported, expected ULONG_MAX (-1)\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), DeviceNumber.PartitionNumber);
|
||||||
|
|
||||||
|
/* Map to an NT device name */
|
||||||
|
RtlStringCchPrintfW(NtLegacyDeviceName, _countof(NtLegacyDeviceName),
|
||||||
|
L"\\Device\\Floppy%lu",
|
||||||
|
DeviceNumber.DeviceNumber);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
/* The device is partitionable, so it must have a valid partition number */
|
||||||
|
ok(DeviceNumber.PartitionNumber != ULONG_MAX,
|
||||||
|
"Device '%s': Invalid partition number (%lu) reported; unpartitionable device?\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), DeviceNumber.PartitionNumber);
|
||||||
|
if (DeviceNumber.PartitionNumber == ULONG_MAX)
|
||||||
|
goto Quit;
|
||||||
|
|
||||||
|
/* Map to an NT device name */
|
||||||
|
RtlStringCchPrintfW(NtLegacyDeviceName, _countof(NtLegacyDeviceName),
|
||||||
|
L"\\Device\\Harddisk%lu\\Partition%lu",
|
||||||
|
DeviceNumber.DeviceNumber, DeviceNumber.PartitionNumber);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Untestable devices */
|
||||||
|
default:
|
||||||
|
skip("Device '%s': Cannot test, device type %lu\n",
|
||||||
|
wine_dbgstr_us(&DeviceName), DeviceInfo.DeviceType);
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Open the device using the legacy path */
|
||||||
|
RtlInitUnicodeString(&DeviceName, NtLegacyDeviceName);
|
||||||
|
InitializeObjectAttributes(&ObjectAttributes,
|
||||||
|
&DeviceName,
|
||||||
|
OBJ_CASE_INSENSITIVE,
|
||||||
|
NULL,
|
||||||
|
NULL);
|
||||||
|
Status = NtOpenFile(&DeviceHandle2,
|
||||||
|
FILE_READ_ATTRIBUTES | SYNCHRONIZE,
|
||||||
|
&ObjectAttributes,
|
||||||
|
&IoStatusBlock,
|
||||||
|
FILE_SHARE_ALL,
|
||||||
|
/* FILE_NON_DIRECTORY_FILE | */ FILE_SYNCHRONOUS_IO_NONALERT);
|
||||||
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
||||||
|
if (!NT_SUCCESS(Status))
|
||||||
|
{
|
||||||
|
skip("Device '%s': Opening failed\n", wine_dbgstr_us(&DeviceName));
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Verify whether both retrieved handles refer to the same device.
|
||||||
|
* Since we're not running on Windows 10, we cannot use kernel32!CompareObjectHandles()
|
||||||
|
* or ntdll!NtCompareObjects(), therefore we have to rely on comparing
|
||||||
|
* whether the devices referred by both handles have the same canonical name.
|
||||||
|
*/
|
||||||
|
Status = NtQueryObject(DeviceHandle1,
|
||||||
|
ObjectNameInformation,
|
||||||
|
&DeviceName1Buffer,
|
||||||
|
sizeof(DeviceName1Buffer),
|
||||||
|
&BufferSize);
|
||||||
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
||||||
|
|
||||||
|
Status = NtQueryObject(DeviceHandle2,
|
||||||
|
ObjectNameInformation,
|
||||||
|
&DeviceName2Buffer,
|
||||||
|
sizeof(DeviceName2Buffer),
|
||||||
|
&BufferSize);
|
||||||
|
ok_ntstatus(Status, STATUS_SUCCESS);
|
||||||
|
|
||||||
|
Success = RtlEqualUnicodeString(DeviceName1, DeviceName2, FALSE);
|
||||||
|
ok(Success, "Devices '%s' and '%s' are not the same!\n",
|
||||||
|
wine_dbgstr_us(DeviceName1), wine_dbgstr_us(DeviceName2));
|
||||||
|
|
||||||
|
Quit:
|
||||||
|
/* Cleanup */
|
||||||
|
if (DeviceHandle2)
|
||||||
|
NtClose(DeviceHandle2);
|
||||||
|
if (DeviceHandle1)
|
||||||
|
NtClose(DeviceHandle1);
|
||||||
|
|
||||||
|
return Success;
|
||||||
|
}
|
||||||
|
|
||||||
|
static BOOLEAN
|
||||||
|
Test_Drive_StorDeviceNumber(
|
||||||
|
_In_ WCHAR Drive)
|
||||||
|
{
|
||||||
|
WCHAR NtDeviceName[] = L"\\DosDevices\\?:";
|
||||||
|
NtDeviceName[sizeof("\\DosDevices\\")-1] = Drive;
|
||||||
|
return Test_Device_StorDeviceNumber(NtDeviceName);
|
||||||
|
}
|
||||||
|
|
||||||
|
START_TEST(StorDeviceNumber)
|
||||||
|
{
|
||||||
|
DWORD Drives;
|
||||||
|
UCHAR i;
|
||||||
|
|
||||||
|
/* Enumerate existing testable drives */
|
||||||
|
Drives = GetLogicalDrives();
|
||||||
|
if (Drives == 0)
|
||||||
|
{
|
||||||
|
skip("Drives map unavailable, error 0x%lx\n", GetLastError());
|
||||||
|
goto otherTests;
|
||||||
|
}
|
||||||
|
for (i = 0; i <= 'Z'-'A'; ++i)
|
||||||
|
{
|
||||||
|
WCHAR DriveName[] = L"?:\\";
|
||||||
|
UINT DriveType;
|
||||||
|
|
||||||
|
/* Skip non-existing drives */
|
||||||
|
if (!(Drives & (1 << i)))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
/* Retrieve the drive type and see whether we can test it */
|
||||||
|
DriveName[0] = L'A' + i;
|
||||||
|
DriveType = GetDriveTypeW(DriveName);
|
||||||
|
|
||||||
|
switch (DriveType)
|
||||||
|
{
|
||||||
|
case DRIVE_REMOVABLE:
|
||||||
|
case DRIVE_FIXED:
|
||||||
|
case DRIVE_CDROM:
|
||||||
|
case DRIVE_RAMDISK:
|
||||||
|
{
|
||||||
|
Test_Drive_StorDeviceNumber(L'A' + i);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case DRIVE_UNKNOWN:
|
||||||
|
case DRIVE_NO_ROOT_DIR:
|
||||||
|
case DRIVE_REMOTE:
|
||||||
|
default:
|
||||||
|
/* Unhandled drive type, just skip it silently */
|
||||||
|
trace("Drive %c with unhandled type %u\n", 'A' + i, DriveType);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
otherTests:
|
||||||
|
/* Test the drive containing SystemRoot */
|
||||||
|
Test_Drive_StorDeviceNumber(SharedUserData->NtSystemRoot[0]);
|
||||||
|
|
||||||
|
/* Test \??\PhysicalDrive0, if it exists */
|
||||||
|
Test_Device_StorDeviceNumber(L"\\??\\PhysicalDrive0");
|
||||||
|
}
|
||||||
20
modules/rostests/apitests/partmgr/precomp.h
Normal file
20
modules/rostests/apitests/partmgr/precomp.h
Normal file
|
|
@ -0,0 +1,20 @@
|
||||||
|
/*
|
||||||
|
* PROJECT: ReactOS API Tests
|
||||||
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
||||||
|
* PURPOSE: Precompiled header
|
||||||
|
* COPYRIGHT: Copyright 2024 Hermès Bélusca-Maïto <hermes.belusca-maito@reactos.org>
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#define WIN32_NO_STATUS
|
||||||
|
#include <apitest.h>
|
||||||
|
|
||||||
|
#define NTOS_MODE_USER
|
||||||
|
#include <ndk/iofuncs.h>
|
||||||
|
#include <ndk/obfuncs.h>
|
||||||
|
#include <ndk/rtlfuncs.h>
|
||||||
|
|
||||||
|
#include <ntstrsafe.h>
|
||||||
|
|
||||||
|
/* EOF */
|
||||||
10
modules/rostests/apitests/partmgr/testlist.c
Normal file
10
modules/rostests/apitests/partmgr/testlist.c
Normal file
|
|
@ -0,0 +1,10 @@
|
||||||
|
#define STANDALONE
|
||||||
|
#include <apitest.h>
|
||||||
|
|
||||||
|
extern void func_StorDeviceNumber(void);
|
||||||
|
|
||||||
|
const struct test winetest_testlist[] =
|
||||||
|
{
|
||||||
|
{ "StorDeviceNumber", func_StorDeviceNumber },
|
||||||
|
{ 0, 0 }
|
||||||
|
};
|
||||||
Loading…
Add table
Add a link
Reference in a new issue