mirror of
https://github.com/reactos/reactos.git
synced 2024-09-08 03:40:17 +00:00
475 lines
12 KiB
C
475 lines
12 KiB
C
![]() |
/*
|
||
|
vfdpnp.c
|
||
|
|
||
|
Virtual Floppy Drive for Windows NT platform
|
||
|
Kernel mode driver: Plug & Play functions
|
||
|
|
||
|
Copyright (C) 2003-2005 Ken Kato
|
||
|
*/
|
||
|
|
||
|
#ifndef VFD_PNP
|
||
|
/*
|
||
|
Not in working order for the time being
|
||
|
so DO NOT define VFD_PNP macro
|
||
|
unless you know exactly what you are doing...
|
||
|
*/
|
||
|
// suppress empty compile unit warning
|
||
|
#ifdef _MSC_VER
|
||
|
#pragma warning (disable: 4206)
|
||
|
#pragma message ("Plug and play support feature is disabled.")
|
||
|
#endif
|
||
|
|
||
|
#else // VFD_PNP
|
||
|
|
||
|
#include "imports.h"
|
||
|
#include "vfddrv.h"
|
||
|
#include "vfddbg.h"
|
||
|
|
||
|
static NTSTATUS
|
||
|
VfdReportDevice(
|
||
|
PDEVICE_EXTENSION device_extension);
|
||
|
|
||
|
#ifdef ALLOC_PRAGMA
|
||
|
#pragma alloc_text(PAGE, VfdPlugAndPlay)
|
||
|
#pragma alloc_text(PAGE, VfdPowerControl)
|
||
|
#pragma alloc_text(PAGE, VfdSystemControl)
|
||
|
#pragma alloc_text(PAGE, VfdAddDevice)
|
||
|
#pragma alloc_text(PAGE, VfdReportDevice)
|
||
|
#endif // ALLOC_PRAGMA
|
||
|
|
||
|
#define REMLOCK_TAG 'LdfV' // "VfdL"
|
||
|
#define REMLOCK_MAXIMUM 1 // Max minutes system allows lock to be held
|
||
|
#define REMLOCK_HIGHWATER 10 // Max number of irps holding lock at one time
|
||
|
|
||
|
#if DBG
|
||
|
static PCSTR StateTable[] ={
|
||
|
{ "STOPPED" },
|
||
|
{ "WORKING" },
|
||
|
{ "PENDINGSTOP" },
|
||
|
{ "PENDINGREMOVE" },
|
||
|
{ "SURPRISEREMOVED" },
|
||
|
{ "REMOVED" },
|
||
|
{ "UNKNOWN" }
|
||
|
};
|
||
|
#endif // DBG
|
||
|
|
||
|
//
|
||
|
// PnP I/O request dispatch
|
||
|
//
|
||
|
NTSTATUS
|
||
|
VfdPlugAndPlay(
|
||
|
IN PDEVICE_OBJECT DeviceObject,
|
||
|
IN PIRP Irp)
|
||
|
{
|
||
|
PIO_STACK_LOCATION io_stack;
|
||
|
PDEVICE_EXTENSION device_extension;
|
||
|
NTSTATUS status = STATUS_SUCCESS;
|
||
|
BOOLEAN lockHeld = TRUE;
|
||
|
|
||
|
//
|
||
|
// setup necessary pointers
|
||
|
//
|
||
|
io_stack = IoGetCurrentIrpStackLocation( Irp );
|
||
|
device_extension = (PDEVICE_EXTENSION) DeviceObject->DeviceExtension;
|
||
|
|
||
|
ASSERT(device_extension->DeviceState < VFD_MAX_STATE);
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdPlugAndPlay - IN. %ws %s Device State=%s\n",
|
||
|
device_extension->device_name.Buffer,
|
||
|
GetPnpIrpName(io_stack->MinorFunction),
|
||
|
StateTable[device_extension->DeviceState]));
|
||
|
|
||
|
//
|
||
|
// Acquire remove lock
|
||
|
//
|
||
|
status = IoAcquireRemoveLock(&device_extension->RemoveLock, Irp);
|
||
|
|
||
|
if (!NT_SUCCESS(status)) {
|
||
|
VFDTRACE(0, ("Acquire RemoveLock failed - %s\n", NtStatusToStr(status)));
|
||
|
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Process the PnP I/O request
|
||
|
//
|
||
|
switch (io_stack->MinorFunction) {
|
||
|
case IRP_MN_START_DEVICE: // 0x00
|
||
|
//
|
||
|
// Start the device
|
||
|
//
|
||
|
device_extension->DeviceState = VFD_WORKING;
|
||
|
status = STATUS_SUCCESS;
|
||
|
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_QUERY_REMOVE_DEVICE: // 0x01
|
||
|
//
|
||
|
// Prepare device removal
|
||
|
//
|
||
|
device_extension->DeviceState = VFD_PENDINGREMOVE;
|
||
|
status = STATUS_SUCCESS;
|
||
|
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_REMOVE_DEVICE: // 0x02
|
||
|
//
|
||
|
// Remove the device
|
||
|
//
|
||
|
status = STATUS_SUCCESS;
|
||
|
|
||
|
// complete the current request
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
|
||
|
//
|
||
|
// Set the device status to REMOVED and wait for other drivers
|
||
|
// to release the lock, then delete the device object
|
||
|
//
|
||
|
device_extension->DeviceState = VFD_REMOVED;
|
||
|
IoReleaseRemoveLockAndWait(&device_extension->RemoveLock, Irp);
|
||
|
lockHeld = FALSE;
|
||
|
|
||
|
VfdRemoveDevice(DeviceObject);
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_CANCEL_REMOVE_DEVICE: // 0x03
|
||
|
//
|
||
|
// Before sending the IRP down make sure we have received
|
||
|
// a IRP_MN_QUERY_REMOVE_DEVICE. We may get Cancel Remove
|
||
|
// without receiving a Query Remove earlier, if the
|
||
|
// driver on top fails a Query Remove and passes down the
|
||
|
// Cancel Remove.
|
||
|
//
|
||
|
|
||
|
if (device_extension->DeviceState == VFD_PENDINGREMOVE) {
|
||
|
device_extension->DeviceState = VFD_WORKING;
|
||
|
}
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_STOP_DEVICE: // 0x04
|
||
|
device_extension->DeviceState = VFD_STOPPED;
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_QUERY_STOP_DEVICE: // 0x05
|
||
|
device_extension->DeviceState = VFD_PENDINGSTOP;
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_CANCEL_STOP_DEVICE: // 0x06
|
||
|
//
|
||
|
// Before sending the IRP down make sure we have received
|
||
|
// a IRP_MN_QUERY_STOP_DEVICE. We may get Cancel Stop
|
||
|
// without receiving a Query Stop earlier, if the
|
||
|
// driver on top fails a Query Stop and passes down the
|
||
|
// Cancel Stop.
|
||
|
//
|
||
|
|
||
|
if (device_extension->DeviceState == VFD_PENDINGSTOP ) {
|
||
|
device_extension->DeviceState = VFD_WORKING;
|
||
|
}
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
|
||
|
case IRP_MN_QUERY_DEVICE_RELATIONS: // 0x07
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
|
||
|
switch (io_stack->Parameters.QueryDeviceRelations.Type) {
|
||
|
case BusRelations:
|
||
|
VFDTRACE(VFDINFO, ("------- BusRelations Query\n"));
|
||
|
break;
|
||
|
|
||
|
case EjectionRelations:
|
||
|
VFDTRACE(VFDINFO, ("------- EjectionRelations Query\n"));
|
||
|
break;
|
||
|
|
||
|
case PowerRelations:
|
||
|
VFDTRACE(VFDINFO, ("------- PowerRelations Query\n"));
|
||
|
break;
|
||
|
|
||
|
case RemovalRelations:
|
||
|
VFDTRACE(VFDINFO, ("------- RemovalRelations Query\n"));
|
||
|
break;
|
||
|
|
||
|
case TargetDeviceRelation:
|
||
|
VFDTRACE(VFDINFO, ("------- TargetDeviceRelation Query\n"));
|
||
|
|
||
|
Irp->IoStatus.Information = (LONG)ExAllocatePoolWithTag(
|
||
|
PagedPool, sizeof(DEVICE_RELATIONS), VFD_POOL_TAG);
|
||
|
|
||
|
if (Irp->IoStatus.Information) {
|
||
|
PDEVICE_RELATIONS rel = (PDEVICE_RELATIONS)Irp->IoStatus.Information;
|
||
|
|
||
|
rel->Count = 1;
|
||
|
rel->Objects[0] = device_extension->device_object;
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
}
|
||
|
else {
|
||
|
status = STATUS_INSUFFICIENT_RESOURCES;
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
VFDTRACE(VFDINFO, ("------- Unknown Query\n"));
|
||
|
break;
|
||
|
}
|
||
|
Irp->IoStatus.Status = status;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
|
||
|
break;
|
||
|
|
||
|
// case IRP_MN_QUERY_INTERFACE: // 0x08
|
||
|
// case IRP_MN_QUERY_CAPABILITIES: // 0x09
|
||
|
// case IRP_MN_QUERY_RESOURCES: // 0x0A
|
||
|
// case IRP_MN_QUERY_RESOURCE_REQUIREMENTS: // 0x0B
|
||
|
// case IRP_MN_QUERY_DEVICE_TEXT: // 0x0C
|
||
|
// case IRP_MN_FILTER_RESOURCE_REQUIREMENTS: // 0x0D
|
||
|
// case IRP_MN_READ_CONFIG: // 0x0F
|
||
|
// case IRP_MN_WRITE_CONFIG: // 0x10
|
||
|
// case IRP_MN_EJECT: // 0x11
|
||
|
// case IRP_MN_SET_LOCK: // 0x12
|
||
|
// case IRP_MN_QUERY_ID: // 0x13
|
||
|
// case IRP_MN_QUERY_PNP_DEVICE_STATE: // 0x14
|
||
|
// case IRP_MN_QUERY_BUS_INFORMATION: // 0x15
|
||
|
// case IRP_MN_DEVICE_USAGE_NOTIFICATION: // 0x16
|
||
|
|
||
|
case IRP_MN_SURPRISE_REMOVAL: // 0x17
|
||
|
device_extension->DeviceState = VFD_SURPRISEREMOVED;
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
|
||
|
// case IRP_MN_QUERY_LEGACY_BUS_INFORMATION: // 0x18
|
||
|
|
||
|
default:
|
||
|
//
|
||
|
// unknown request -- simply pass it to the lower device
|
||
|
//
|
||
|
status = STATUS_INVALID_DEVICE_REQUEST;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Device Extenion is gone if the current IRP is IRP_MN_REMOVE_DEVICE
|
||
|
//
|
||
|
if (lockHeld == TRUE) {
|
||
|
IoReleaseRemoveLock(&device_extension->RemoveLock, Irp);
|
||
|
}
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdPlugAndPlay - %s\n", NtStatusToStr(status)));
|
||
|
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Power management I/O request dispatch
|
||
|
//
|
||
|
NTSTATUS
|
||
|
VfdPowerControl(
|
||
|
IN PDEVICE_OBJECT DeviceObject,
|
||
|
IN PIRP Irp)
|
||
|
{
|
||
|
PIO_STACK_LOCATION io_stack;
|
||
|
PDEVICE_EXTENSION device_extension;
|
||
|
NTSTATUS status;
|
||
|
|
||
|
io_stack = IoGetCurrentIrpStackLocation( Irp );
|
||
|
device_extension = (PDEVICE_EXTENSION)DeviceObject->DeviceExtension;
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdPowerControl - IN. %ws %s Device State=%s\n",
|
||
|
device_extension->device_name.Buffer,
|
||
|
GetPnpIrpName(io_stack->MinorFunction),
|
||
|
StateTable[device_extension->DeviceState]));
|
||
|
|
||
|
PoStartNextPowerIrp(Irp);
|
||
|
|
||
|
//
|
||
|
// If the device has been removed, the driver should not pass
|
||
|
// the IRP down to the next lower driver.
|
||
|
//
|
||
|
|
||
|
if (device_extension->DeviceState == VFD_REMOVED) {
|
||
|
status = STATUS_DELETE_PENDING;
|
||
|
}
|
||
|
else {
|
||
|
status = STATUS_SUCCESS;
|
||
|
}
|
||
|
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdPowerControl - %s\n", NtStatusToStr(status)));
|
||
|
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// WMI I/O request dispatch
|
||
|
//
|
||
|
NTSTATUS
|
||
|
VfdSystemControl(
|
||
|
IN PDEVICE_OBJECT DeviceObject,
|
||
|
IN PIRP Irp)
|
||
|
{
|
||
|
PIO_STACK_LOCATION io_stack;
|
||
|
PDEVICE_EXTENSION device_extension;
|
||
|
NTSTATUS status;
|
||
|
|
||
|
io_stack = IoGetCurrentIrpStackLocation(Irp);
|
||
|
device_extension = (PDEVICE_EXTENSION)DeviceObject->DeviceExtension;
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdSystemControl - IN. %ws %s Device State=%s\n",
|
||
|
device_extension->device_name.Buffer,
|
||
|
GetPnpIrpName(io_stack->MinorFunction),
|
||
|
StateTable[device_extension->DeviceState]));
|
||
|
|
||
|
status = STATUS_SUCCESS;
|
||
|
Irp->IoStatus.Status = status;
|
||
|
Irp->IoStatus.Information = 0;
|
||
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdSystemControl - %s\n", NtStatusToStr(status)));
|
||
|
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// PnP AddDevice function
|
||
|
//
|
||
|
NTSTATUS
|
||
|
VfdAddDevice(
|
||
|
IN PDRIVER_OBJECT DriverObject,
|
||
|
IN OUT PDEVICE_OBJECT PhysicalDevice)
|
||
|
{
|
||
|
PDEVICE_OBJECT device_object;
|
||
|
PDEVICE_EXTENSION device_extension;
|
||
|
NTSTATUS status;
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] VfdAddDevice - IN\n"));
|
||
|
|
||
|
status = VfdCreateDevice(DriverObject, &device_object);
|
||
|
|
||
|
if (NT_SUCCESS(status)) {
|
||
|
|
||
|
device_object->Flags |= DO_POWER_PAGABLE;
|
||
|
|
||
|
device_extension =
|
||
|
(PDEVICE_EXTENSION)device_object->DeviceExtension;
|
||
|
|
||
|
// Device starts in Stopped state
|
||
|
device_extension->DeviceState = VFD_STOPPED;
|
||
|
|
||
|
VFDTRACE(VFDINFO, ("[VFD] Initializing the remove lock\n"));
|
||
|
|
||
|
IoInitializeRemoveLock(
|
||
|
&device_extension->RemoveLock,
|
||
|
REMLOCK_TAG,
|
||
|
REMLOCK_MAXIMUM,
|
||
|
REMLOCK_HIGHWATER);
|
||
|
|
||
|
if (PhysicalDevice) {
|
||
|
device_extension->PhysicalDevice = PhysicalDevice;
|
||
|
}
|
||
|
else {
|
||
|
VfdReportDevice(device_extension);
|
||
|
}
|
||
|
VfdRegisterInterface(device_extension);
|
||
|
VfdMountMgrNotifyVolume(device_extension);
|
||
|
}
|
||
|
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Report a VFD device to the PnP manager
|
||
|
//
|
||
|
NTSTATUS
|
||
|
VfdReportDevice(
|
||
|
PDEVICE_EXTENSION device_extension)
|
||
|
{
|
||
|
NTSTATUS status = STATUS_SUCCESS;
|
||
|
CM_RESOURCE_LIST list = {0};
|
||
|
PCM_FULL_RESOURCE_DESCRIPTOR full = &(list.List[0]);
|
||
|
PCM_PARTIAL_RESOURCE_LIST part = &(full->PartialResourceList);
|
||
|
PCM_PARTIAL_RESOURCE_DESCRIPTOR desc = &(part->PartialDescriptors[0]);;
|
||
|
|
||
|
list.Count = 1;
|
||
|
|
||
|
full->InterfaceType = Internal;
|
||
|
full->BusNumber = device_extension->device_number;
|
||
|
|
||
|
part->Version = 1;
|
||
|
part->Revision = 1;
|
||
|
part->Count = 1;
|
||
|
|
||
|
desc->Type = CmResourceTypeDeviceSpecific;
|
||
|
desc->ShareDisposition = CmResourceShareShared;
|
||
|
desc->Flags = 0;
|
||
|
|
||
|
VFDTRACE(VFDINFO,("[VFD] Reporting device %lu to the PnP manager\n",
|
||
|
device_extension->device_number));
|
||
|
|
||
|
status = IoReportDetectedDevice(
|
||
|
device_extension->device_object->DriverObject, // IN PDRIVER_OBJECT DriverObject,
|
||
|
Internal, // IN INTERFACE_TYPE LegacyBusType,
|
||
|
(ULONG)-1, // IN ULONG BusNumber,
|
||
|
(ULONG)-1, // IN ULONG SlotNumber,
|
||
|
&list, // IN PCM_RESOURCE_LIST ResourceList,
|
||
|
NULL, // IN PIO_RESOURCE_REQUIREMENTS_LIST OPTIONAL,
|
||
|
TRUE, // IN BOOLEAN ResourceAssigned,
|
||
|
&(device_extension->PhysicalDevice) // IN OUT PDEVICE_OBJECT *DeviceObject
|
||
|
);
|
||
|
|
||
|
if (!NT_SUCCESS(status)) {
|
||
|
VFDTRACE(0,
|
||
|
("[VFD] IoReportDetectedDevice - %s\n",
|
||
|
NtStatusToStr(status)));
|
||
|
}
|
||
|
|
||
|
device_extension->TargetDevice = IoAttachDeviceToDeviceStack(
|
||
|
device_extension->device_object,
|
||
|
device_extension->PhysicalDevice);
|
||
|
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
#endif // VFD_PNP
|