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_ PWCHAR 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_ PWCHAR	DriveRoot,
		_Out_ PVOID	DeviceInformation,
		_In_ ULONG	BufferSize
	)

Retrieves disk device information.

Parameters

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

Returns

TRUE if the buffer was large enough and was filled with the requested information, FALSE otherwise.

Remarks

The returned information is 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 implementation 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 < sizeof(ULONG) ||
        !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;
    }
 
    /* 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 >= sizeof(DEVICE_INFORMATION))
    {
        /* 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().