query.c File Reference

#include "precomp.h"
#include <ntddstor.h>
#include <ntstrsafe.h>
#include <ndk/iofuncs.h>
#include <ndk/obfuncs.h>

Include dependency graph for query.c:

Go to the source code of this file.

Macros
#define	NTOS_MODE_USER

Functions
BOOLEAN NTAPI	QueryAvailableFileSystemFormat (IN DWORD Index, IN OUT PWCHAR FileSystem, OUT UCHAR *Major, OUT UCHAR *Minor, OUT BOOLEAN *LatestVersion)
BOOL NTAPI	QueryDeviceInformation (_In_ PCWSTR DriveRoot, _Out_ PVOID DeviceInformation, _In_ ULONG BufferSize)
	Retrieves disk device information.

Macro Definition Documentation

NTOS_MODE_USER

#define NTOS_MODE_USER

Definition at line 14 of file query.c.

Function Documentation

QueryAvailableFileSystemFormat()

BOOLEAN NTAPI QueryAvailableFileSystemFormat	(	IN DWORD	Index,
		IN OUT PWCHAR	FileSystem,
		OUT UCHAR *	Major,
		OUT UCHAR *	Minor,
		OUT BOOLEAN *	LatestVersion
	)

Definition at line 20 of file query.c.

{
    PLIST_ENTRY ListEntry;
    PIFS_PROVIDER Provider;
 
     if (!FileSystem || !Major ||!Minor ||!LatestVersion)
        return FALSE;
 
    ListEntry = ProviderListHead.Flink;
    while (TRUE)
    {
        if (ListEntry == &ProviderListHead)
            return FALSE;
        if (Index == 0)
            break;
        ListEntry = ListEntry->Flink;
        Index--;
    }
 
    Provider = CONTAINING_RECORD(ListEntry, IFS_PROVIDER, ListEntry);
    wcscpy(FileSystem, Provider->Name);
    *Major = 0; /* FIXME */
    *Minor = 0; /* FIXME */
    *LatestVersion = TRUE; /* FIXME */
 
    return TRUE;
}

Referenced by InitializeFormatDriveDlg(), ShowInstalledFileSystems(), and Usage().

QueryDeviceInformation()

BOOL NTAPI QueryDeviceInformation	(	_In_ PCWSTR	DriveRoot,
		_Out_ PVOID	DeviceInformation,
		_In_ ULONG	BufferSize
	)

Retrieves disk device information.

Parameters

[in]	DriveRoot	String which contains a DOS device name,
[out]	DeviceInformation	Pointer to buffer which will receive DEVICE_INFORMATION data.
[in]	BufferSize	Size of DeviceInformation buffer, in bytes.

Returns

TRUE if the buffer was large enough (pre-Vista at least, Vista+ if possible) and was filled with the requested information, FALSE otherwise.

Remarks

The returned flags are mostly related to Sony Memory Stick devices. On Vista+, the returned information is disk sector size and volume length in sectors, regardless of the type of disk. ReactOS returns DEVICE_HOTPLUG flag if inspected device is a hotplug device, as well as sector size and volume length of disk device.

Definition at line 79 of file query.c.

{
    PDEVICE_INFORMATION DeviceInfo = DeviceInformation;
    IO_STATUS_BLOCK Iosb;
    DISK_GEOMETRY DiskGeometry;
    STORAGE_HOTPLUG_INFO HotplugInfo;
    GET_LENGTH_INFORMATION LengthInformation;
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING DeviceName;
    HANDLE FileHandle;
    NTSTATUS Status;
    WCHAR DiskDevice[MAX_PATH];
    WCHAR DriveName[MAX_PATH];
 
    /* Buffer should be able to at least hold DeviceFlags */
    if (BufferSize < RTL_SIZEOF_THROUGH_FIELD(DEVICE_INFORMATION, DeviceFlags) ||
        !NT_SUCCESS(RtlStringCchCopyW(DriveName, ARRAYSIZE(DriveName), DriveRoot)))
    {
        return FALSE;
    }
 
    if (DriveName[wcslen(DriveName) - 1] != L'\\')
    {
        /* Append the trailing backslash for GetVolumeNameForVolumeMountPointW */
        if (!NT_SUCCESS(RtlStringCchCatW(DriveName, ARRAYSIZE(DriveName), L"\\")))
            return FALSE;
    }
 
    if (!GetVolumeNameForVolumeMountPointW(DriveName, DiskDevice, ARRAYSIZE(DiskDevice)) ||
        !RtlDosPathNameToNtPathName_U(DiskDevice, &DeviceName, NULL, NULL))
    {
        /* Disk has no volume GUID, fallback to QueryDosDevice */
        DriveName[wcslen(DriveName) - 1] = UNICODE_NULL;
        if (!QueryDosDeviceW(DriveName, DiskDevice, ARRAYSIZE(DiskDevice)))
            return FALSE;
        RtlInitUnicodeString(&DeviceName, DiskDevice);
    }
    else
    {
        /* Trim the trailing backslash since we will work with a device object */
        DeviceName.Length -= sizeof(WCHAR);
    }
 
    InitializeObjectAttributes(&ObjectAttributes,
                               &DeviceName,
                               OBJ_CASE_INSENSITIVE,
                               NULL,
                               NULL);
 
    Status = NtOpenFile(&FileHandle,
                        FILE_READ_DATA | FILE_READ_ATTRIBUTES | SYNCHRONIZE,
                        &ObjectAttributes,
                        &Iosb,
                        FILE_SHARE_READ | FILE_SHARE_WRITE,
                        FILE_SYNCHRONOUS_IO_NONALERT);
    if (!NT_SUCCESS(Status))
        return FALSE;
 
    Status = NtDeviceIoControlFile(FileHandle,
                                   NULL,
                                   NULL,
                                   NULL,
                                   &Iosb,
                                   IOCTL_STORAGE_GET_HOTPLUG_INFO,
                                   NULL,
                                   0,
                                   &HotplugInfo,
                                   sizeof(HotplugInfo));
    if (!NT_SUCCESS(Status))
        goto Quit;
 
    DeviceInfo->DeviceFlags = 0;
    if (HotplugInfo.MediaHotplug || HotplugInfo.DeviceHotplug)
    {
        /* This is a hotplug device */
        DeviceInfo->DeviceFlags |= DEVICE_HOTPLUG;
    }
 
    /* UNIMPLEMENTED: Other flags that would be set here are related to Sony "Memory Stick"
     * type of devices, which we do not have any special support for */
 
    if (BufferSize >= RTL_SIZEOF_THROUGH_FIELD(DEVICE_INFORMATION, SectorCount))
    {
        /* This is the Vista+ version of the structure.
         * We need to also provide disk sector size and volume length in sectors. */
        Status = NtDeviceIoControlFile(FileHandle,
                                       NULL,
                                       NULL,
                                       NULL,
                                       &Iosb,
                                       IOCTL_DISK_GET_DRIVE_GEOMETRY,
                                       NULL,
                                       0,
                                       &DiskGeometry,
                                       sizeof(DiskGeometry));
        if (!NT_SUCCESS(Status))
            goto Quit;
 
        Status = NtDeviceIoControlFile(FileHandle,
                                       NULL,
                                       NULL,
                                       NULL,
                                       &Iosb,
                                       IOCTL_DISK_GET_LENGTH_INFO,
                                       NULL,
                                       0,
                                       &LengthInformation,
                                       sizeof(LengthInformation));
        if (!NT_SUCCESS(Status))
            goto Quit;
 
        LengthInformation.Length.QuadPart /= DiskGeometry.BytesPerSector;
        DeviceInfo->SectorSize = DiskGeometry.BytesPerSector;
        DeviceInfo->SectorCount = LengthInformation.Length;
    }
 
    Status = STATUS_SUCCESS;
 
Quit:
    NtClose(FileHandle);
    return NT_SUCCESS(Status);
}

Referenced by MountMgrCreatePointWorker(), MountMgrNextDriveLetterWorker(), MountMgrVolumeMountPointChanged(), MountMgrVolumeMountPointCreated(), QueryPointsFromMemory(), QueryPointsFromSymbolicLinkName(), USBD_GetDeviceInformationEx(), USBH_GetDeviceType(), and wmain().