Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

4906 lines
146 KiB

/*++
Copyright (c) 1996-2001 Microsoft Corporation
Module Name:
USBMASS.C
Abstract:
This source file contains the DriverEntry() and AddDevice() entry points
for the USBSTOR driver and the dispatch routines which handle:
IRP_MJ_POWER
IRP_MJ_SYSTEM_CONTROL
IRP_MJ_PNP
Environment:
kernel mode
Revision History:
06-01-98 : started rewrite
--*/
//*****************************************************************************
// I N C L U D E S
//*****************************************************************************
#include <ntddk.h>
#include <usbdi.h>
#include <usbdlib.h>
#include <initguid.h>
#include <usbbusif.h>
#include <stdio.h>
#include "usbmass.h"
//*****************************************************************************
// L O C A L F U N C T I O N P R O T O T Y P E S
//*****************************************************************************
NTSTATUS
USBSTOR_GetBusInterface (
IN PDEVICE_OBJECT DeviceObject,
IN PUSB_BUS_INTERFACE_USBDI_V1 BusInterface
);
#ifdef ALLOC_PRAGMA
#pragma alloc_text(PAGE, DriverEntry)
#pragma alloc_text(PAGE, USBSTOR_Unload)
#pragma alloc_text(PAGE, USBSTOR_AddDevice)
#pragma alloc_text(PAGE, USBSTOR_QueryFdoParams)
#pragma alloc_text(PAGE, USBSTOR_Power)
#pragma alloc_text(PAGE, USBSTOR_FdoSetPower)
#pragma alloc_text(PAGE, USBSTOR_SystemControl)
#pragma alloc_text(PAGE, USBSTOR_Pnp)
#pragma alloc_text(PAGE, USBSTOR_FdoStartDevice)
#pragma alloc_text(PAGE, USBSTOR_GetDescriptors)
#pragma alloc_text(PAGE, USBSTOR_GetStringDescriptors)
#pragma alloc_text(PAGE, USBSTOR_AdjustConfigurationDescriptor)
#pragma alloc_text(PAGE, USBSTOR_GetPipes)
#pragma alloc_text(PAGE, USBSTOR_CreateChildPDO)
#pragma alloc_text(PAGE, USBSTOR_FdoStopDevice)
#pragma alloc_text(PAGE, USBSTOR_FdoRemoveDevice)
#pragma alloc_text(PAGE, USBSTOR_FdoQueryStopRemoveDevice)
#pragma alloc_text(PAGE, USBSTOR_FdoCancelStopRemoveDevice)
#pragma alloc_text(PAGE, USBSTOR_FdoQueryDeviceRelations)
#pragma alloc_text(PAGE, USBSTOR_FdoQueryCapabilities)
#pragma alloc_text(PAGE, USBSTOR_PdoStartDevice)
#pragma alloc_text(PAGE, USBSTOR_PdoRemoveDevice)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryID)
#pragma alloc_text(PAGE, USBSTOR_PdoDeviceTypeString)
#pragma alloc_text(PAGE, USBSTOR_PdoGenericTypeString)
#pragma alloc_text(PAGE, CopyField)
#pragma alloc_text(PAGE, USBSTOR_StringArrayToMultiSz)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryDeviceId)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryHardwareIds)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryCompatibleIds)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryDeviceText)
#pragma alloc_text(PAGE, USBSTOR_PdoBusQueryInstanceId)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryDeviceRelations)
#pragma alloc_text(PAGE, USBSTOR_PdoQueryCapabilities)
#pragma alloc_text(PAGE, USBSTOR_SyncPassDownIrp)
#pragma alloc_text(PAGE, USBSTOR_SyncSendUsbRequest)
#pragma alloc_text(PAGE, USBSTOR_GetDescriptor)
#pragma alloc_text(PAGE, USBSTOR_GetMaxLun)
#pragma alloc_text(PAGE, USBSTOR_SelectConfiguration)
#pragma alloc_text(PAGE, USBSTOR_UnConfigure)
#pragma alloc_text(PAGE, USBSTOR_ResetPipe)
#pragma alloc_text(PAGE, USBSTOR_AbortPipe)
#pragma alloc_text(PAGE, USBSTOR_GetBusInterface)
#endif
//******************************************************************************
//
// DriverEntry()
//
//******************************************************************************
NTSTATUS
DriverEntry (
IN PDRIVER_OBJECT DriverObject,
IN PUNICODE_STRING RegistryPath
)
{
PAGED_CODE();
#if DBG
// Query the registry for global parameters
//
USBSTOR_QueryGlobalParams();
#endif
DBGPRINT(2, ("enter: DriverEntry\n"));
DBGFBRK(DBGF_BRK_DRIVERENTRY);
LOGINIT();
//
// Initialize the Driver Object with the driver's entry points
//
//
// USBMASS.C
//
DriverObject->DriverUnload = USBSTOR_Unload;
DriverObject->DriverExtension->AddDevice = USBSTOR_AddDevice;
//
// OCRW.C
//
DriverObject->MajorFunction[IRP_MJ_CREATE] = USBSTOR_Create;
DriverObject->MajorFunction[IRP_MJ_CLOSE] = USBSTOR_Close;
DriverObject->MajorFunction[IRP_MJ_READ] = USBSTOR_ReadWrite;
DriverObject->MajorFunction[IRP_MJ_WRITE] = USBSTOR_ReadWrite;
//
// SCSI.C
//
DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = USBSTOR_DeviceControl;
DriverObject->MajorFunction[IRP_MJ_SCSI] = USBSTOR_Scsi;
DriverObject->DriverStartIo = USBSTOR_StartIo;
//
// USBMASS.C
//
DriverObject->MajorFunction[IRP_MJ_POWER] = USBSTOR_Power;
DriverObject->MajorFunction[IRP_MJ_SYSTEM_CONTROL] = USBSTOR_SystemControl;
DriverObject->MajorFunction[IRP_MJ_PNP] = USBSTOR_Pnp;
DBGPRINT(2, ("exit: DriverEntry\n"));
return STATUS_SUCCESS;
}
//******************************************************************************
//
// USBSTOR_Unload()
//
//******************************************************************************
VOID
USBSTOR_Unload (
IN PDRIVER_OBJECT DriverObject
)
{
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_Unload\n"));
LOGENTRY('UNLD', DriverObject, 0, 0);
DBGFBRK(DBGF_BRK_UNLOAD);
LOGUNINIT();
DBGPRINT(2, ("exit: USBSTOR_Unload\n"));
}
//******************************************************************************
//
// USBSTOR_AddDevice()
//
//******************************************************************************
NTSTATUS
USBSTOR_AddDevice (
IN PDRIVER_OBJECT DriverObject,
IN PDEVICE_OBJECT PhysicalDeviceObject
)
{
PDEVICE_OBJECT deviceObject;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_AddDevice\n"));
LOGENTRY('ADDD', DriverObject, PhysicalDeviceObject, 0);
DBGFBRK(DBGF_BRK_ADDDEVICE);
// Create the FDO
//
ntStatus = IoCreateDevice(DriverObject,
sizeof(FDO_DEVICE_EXTENSION),
NULL,
FILE_DEVICE_BUS_EXTENDER,
FILE_AUTOGENERATED_DEVICE_NAME,
FALSE,
&deviceObject);
if (!NT_SUCCESS(ntStatus))
{
return ntStatus;
}
// StartIo should not be called recursively and should be deferred until
// the previous StartIo call returns to the IO manager. This will prevent
// a recursive stack overflow death if a device error occurs when there
// are many requests queued on the device queue.
//
IoSetStartIoAttributes(deviceObject,
TRUE, // DeferredStartIo
FALSE // NonCancelable
);
// Initialize the FDO DeviceExtension
//
fdoDeviceExtension = deviceObject->DeviceExtension;
// Set all DeviceExtension pointers to NULL and all variable to zero
//
RtlZeroMemory(fdoDeviceExtension, sizeof(FDO_DEVICE_EXTENSION));
// Tag this as the FDO on top of the USB PDO
//
fdoDeviceExtension->Type = USBSTOR_DO_TYPE_FDO;
// Store a back point to the DeviceObject to which the DeviceExtension
// is attached.
//
fdoDeviceExtension->FdoDeviceObject = deviceObject;
// Remember our PDO
//
fdoDeviceExtension->PhysicalDeviceObject = PhysicalDeviceObject;
// Attach the FDO we created to the top of the PDO stack
//
fdoDeviceExtension->StackDeviceObject = IoAttachDeviceToDeviceStack(
deviceObject,
PhysicalDeviceObject);
// Initialize the list of child PDOs
//
InitializeListHead(&fdoDeviceExtension->ChildPDOs);
// Initialize to one in AddDevice, decrement by one in REMOVE_DEVICE
//
fdoDeviceExtension->PendingIoCount = 1;
// Initialize the event which is set when OpenCount is decremented to zero.
//
KeInitializeEvent(&fdoDeviceExtension->RemoveEvent,
SynchronizationEvent,
FALSE);
// Set the initial system and device power states
//
fdoDeviceExtension->SystemPowerState = PowerSystemWorking;
fdoDeviceExtension->DevicePowerState = PowerDeviceD0;
KeInitializeEvent(&fdoDeviceExtension->PowerDownEvent,
SynchronizationEvent,
FALSE);
// Initialize the spinlock which protects the PDO DeviceFlags
//
KeInitializeSpinLock(&fdoDeviceExtension->ExtensionDataSpinLock);
KeInitializeEvent(&fdoDeviceExtension->CancelEvent,
SynchronizationEvent,
FALSE);
// Initialize timeout timer
//
IoInitializeTimer(deviceObject, USBSTOR_TimerTick, NULL);
USBSTOR_QueryFdoParams(deviceObject);
fdoDeviceExtension->LastSenseWasReset = TRUE;
deviceObject->Flags |= DO_DIRECT_IO;
deviceObject->Flags |= DO_POWER_PAGABLE;
deviceObject->Flags &= ~DO_DEVICE_INITIALIZING;
DBGPRINT(2, ("exit: USBSTOR_AddDevice\n"));
LOGENTRY('addd', deviceObject, fdoDeviceExtension,
fdoDeviceExtension->StackDeviceObject);
return STATUS_SUCCESS;
}
//******************************************************************************
//
// USBSTOR_QueryFdoParams()
//
// This is called at AddDevice() time when the FDO is being created to query
// device parameters from the registry.
//
//******************************************************************************
VOID
USBSTOR_QueryFdoParams (
IN PDEVICE_OBJECT DeviceObject
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
RTL_QUERY_REGISTRY_TABLE paramTable[3];
ULONG driverFlags;
ULONG nonRemovable;
HANDLE handle;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_QueryFdoParams\n"));
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Set the default value in case the registry key does not exist.
// Currently the flags are only used to specify the device protocol:
// {Bulk-Only, Control/Bulk/Interrupt, Control/Bulk}
//
// If the driver is loaded during textmode setup then the registry key
// will not yet exist. That should be the only case in which the registry
// key does not exist. In this case DeviceProtocolUnspecified will be
// treated as DeviceProtocolCB. If that causes the first request to fail
// then we will switch to DeviceProtocolBulkOnly.
//
driverFlags = DeviceProtocolUnspecified;
nonRemovable = 0;
ntStatus = IoOpenDeviceRegistryKey(
fdoDeviceExtension->PhysicalDeviceObject,
PLUGPLAY_REGKEY_DRIVER,
STANDARD_RIGHTS_ALL,
&handle);
if (NT_SUCCESS(ntStatus))
{
RtlZeroMemory (&paramTable[0], sizeof(paramTable));
paramTable[0].Flags = RTL_QUERY_REGISTRY_DIRECT;
paramTable[0].Name = L"DriverFlags";
paramTable[0].EntryContext = &driverFlags;
paramTable[0].DefaultType = REG_BINARY;
paramTable[0].DefaultData = &driverFlags;
paramTable[0].DefaultLength = sizeof(ULONG);
paramTable[1].Flags = RTL_QUERY_REGISTRY_DIRECT;
paramTable[1].Name = L"NonRemovable";
paramTable[1].EntryContext = &nonRemovable;
paramTable[1].DefaultType = REG_BINARY;
paramTable[1].DefaultData = &nonRemovable;
paramTable[1].DefaultLength = sizeof(ULONG);
RtlQueryRegistryValues(RTL_REGISTRY_HANDLE,
(PCWSTR)handle,
&paramTable[0],
NULL, // Context
NULL); // Environment
ZwClose(handle);
}
if (driverFlags >= DeviceProtocolLast)
{
driverFlags = DeviceProtocolUnspecified;
}
fdoDeviceExtension->DriverFlags = driverFlags;
fdoDeviceExtension->NonRemovable = nonRemovable;
DBGPRINT(2, ("deviceFlags %08X\n", driverFlags));
DBGPRINT(2, ("nonRemovable %08X\n", nonRemovable));
DBGPRINT(2, ("exit: USBSTOR_QueryFdoParams\n"));
}
//******************************************************************************
//
// USBSTOR_Power()
//
// Dispatch routine which handles IRP_MJ_POWER
//
//******************************************************************************
NTSTATUS
USBSTOR_Power (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PDEVICE_EXTENSION deviceExtension;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
NTSTATUS ntStatus;
PAGED_CODE();
deviceExtension = DeviceObject->DeviceExtension;
irpStack = IoGetCurrentIrpStackLocation(Irp);
DBGPRINT(2, ("enter: USBSTOR_Power %s %08X %08X %s\n",
(deviceExtension->Type == USBSTOR_DO_TYPE_FDO) ?
"FDO" : "PDO",
DeviceObject,
Irp,
PowerMinorFunctionString(irpStack->MinorFunction)));
LOGENTRY('POWR', DeviceObject, Irp, irpStack->MinorFunction);
if (irpStack->MinorFunction == IRP_MN_SET_POWER)
{
DBGPRINT(2, ("%s IRP_MN_SET_POWER %s\n",
(deviceExtension->Type == USBSTOR_DO_TYPE_FDO) ?
"FDO" : "PDO",
(irpStack->Parameters.Power.Type == SystemPowerState) ?
PowerSystemStateString(irpStack->Parameters.Power.State.SystemState) :
PowerDeviceStateString(irpStack->Parameters.Power.State.DeviceState)));
}
if (deviceExtension->Type == USBSTOR_DO_TYPE_FDO)
{
// This is an FDO attached to the USB PDO.
//
fdoDeviceExtension = (PFDO_DEVICE_EXTENSION)deviceExtension;
if (irpStack->MinorFunction == IRP_MN_SET_POWER)
{
// Handle powering the FDO down and up...
//
ntStatus = USBSTOR_FdoSetPower(DeviceObject,
Irp);
}
else
{
// No special processing for IRP_MN_QUERY_POWER, IRP_MN_WAIT_WAKE,
// or IRP_MN_POWER_SEQUENCE at this time. Just pass the request
// down to the next lower driver now.
//
PoStartNextPowerIrp(Irp);
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = PoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
}
}
else
{
// This is a PDO enumerated by our FDO.
if (irpStack->MinorFunction == IRP_MN_SET_POWER)
{
// Handle powering the PDO down and up...
//
ntStatus = USBSTOR_PdoSetPower(DeviceObject,
Irp);
}
else
{
if (irpStack->MinorFunction == IRP_MN_QUERY_POWER)
{
// Always return SUCCESS for IRP_MN_QUERY_POWER for the PDO.
//
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
}
else
{
// No special processing for IRP_MN_WAIT_WAKE or
// IRP_MN_POWER_SEQUENCE. Just complete the request
// now without changing the status.
//
ntStatus = Irp->IoStatus.Status;
}
PoStartNextPowerIrp(Irp);
IoCompleteRequest(Irp, IO_NO_INCREMENT);
}
}
DBGPRINT(2, ("exit: USBSTOR_Power %08X\n", ntStatus));
LOGENTRY('powr', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoSetPower()
//
// Dispatch routine which handles IRP_MJ_POWER, IRP_MN_SET_POWER for the FDO
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoSetPower (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
POWER_STATE_TYPE powerType;
POWER_STATE powerState;
POWER_STATE newState;
BOOLEAN passRequest;
NTSTATUS ntStatus;
PAGED_CODE();
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Get our Irp parameters
//
irpStack = IoGetCurrentIrpStackLocation(Irp);
powerType = irpStack->Parameters.Power.Type;
powerState = irpStack->Parameters.Power.State;
DBGPRINT(2, ("enter: USBSTOR_FdoSetPower %08X %s\n",
DeviceObject,
(powerType == SystemPowerState) ?
PowerSystemStateString(powerState.SystemState) :
PowerDeviceStateString(powerState.DeviceState)));
LOGENTRY('FDSP', DeviceObject, Irp, irpStack->MinorFunction);
// Pass the request down here, unless we request a device state power
// Irp, in which case we pass the request down in our completion routine.
//
passRequest = TRUE;
if (powerType == SystemPowerState)
{
// Remember the current system state.
//
fdoDeviceExtension->SystemPowerState = powerState.SystemState;
// Map the new system state to a new device state
//
if (powerState.SystemState != PowerSystemWorking)
{
newState.DeviceState = PowerDeviceD3;
}
else
{
newState.DeviceState = PowerDeviceD0;
}
// If the new device state is different than the current device
// state, request a device state power Irp.
//
if (fdoDeviceExtension->DevicePowerState != newState.DeviceState)
{
DBGPRINT(2, ("Requesting power Irp %08X %08X from %s to %s\n",
DeviceObject, Irp,
PowerDeviceStateString(fdoDeviceExtension->DevicePowerState),
PowerDeviceStateString(newState.DeviceState)));
ASSERT(fdoDeviceExtension->CurrentPowerIrp == NULL);
fdoDeviceExtension->CurrentPowerIrp = Irp;
ntStatus = PoRequestPowerIrp(fdoDeviceExtension->PhysicalDeviceObject,
IRP_MN_SET_POWER,
newState,
USBSTOR_FdoSetPowerCompletion,
DeviceObject,
NULL);
passRequest = FALSE;
}
}
else if (powerType == DevicePowerState)
{
POWER_STATE oldState;
DBGPRINT(2, ("Received power Irp %08X %08X from %s to %s\n",
DeviceObject, Irp,
PowerDeviceStateString(fdoDeviceExtension->DevicePowerState),
PowerDeviceStateString(powerState.DeviceState)));
// Update the current device state.
//
oldState.DeviceState = fdoDeviceExtension->DevicePowerState;
fdoDeviceExtension->DevicePowerState = powerState.DeviceState;
if (oldState.DeviceState == PowerDeviceD0 &&
powerState.DeviceState > PowerDeviceD0)
{
// Powering down. Stick this Irp in the device queue and
// then wait. When USBSTOR_StartIo() pulls this Irp out
// of the device queue, we'll know that no transfer requests
// are active at that time and then this power Irp can be
// passed down the stack.
ULONG zero;
DBGPRINT(2, ("FDO Powering Down\n"));
LOGENTRY('PWRD', DeviceObject, Irp, 0);
zero = 0; // Front of the queue please
IoStartPacket(DeviceObject,
Irp,
&zero,
NULL);
KeWaitForSingleObject(&fdoDeviceExtension->PowerDownEvent,
Executive,
KernelMode,
FALSE,
NULL);
}
else if (oldState.DeviceState > PowerDeviceD0 &&
powerState.DeviceState == PowerDeviceD0)
{
DBGPRINT(2, ("PDO Powering Up\n"));
LOGENTRY('PWRU', DeviceObject, Irp, 0);
IoCopyCurrentIrpStackLocationToNext(Irp);
IoSetCompletionRoutine(Irp,
USBSTOR_FdoSetPowerD0Completion,
NULL,
TRUE,
TRUE,
TRUE);
ntStatus = PoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
passRequest = FALSE;
}
}
if (passRequest)
{
//
// Pass the request down to the next lower driver
//
PoStartNextPowerIrp(Irp);
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = PoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
}
DBGPRINT(2, ("exit: USBSTOR_FdoSetPower %08X\n", ntStatus));
LOGENTRY('fdsp', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoSetPowerCompletion()
//
// Completion routine for PoRequestPowerIrp() in USBSTOR_FdoSetPower.
//
// The purpose of this routine is to block passing down the SystemPowerState
// Irp until the requested DevicePowerState Irp completes.
//
//******************************************************************************
VOID
USBSTOR_FdoSetPowerCompletion(
IN PDEVICE_OBJECT PdoDeviceObject,
IN UCHAR MinorFunction,
IN POWER_STATE PowerState,
IN PVOID Context,
IN PIO_STATUS_BLOCK IoStatus
)
{
PDEVICE_OBJECT fdoDeviceObject;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIRP irp;
NTSTATUS ntStatus;
fdoDeviceObject = (PDEVICE_OBJECT)Context;
fdoDeviceExtension = fdoDeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
ASSERT(fdoDeviceExtension->CurrentPowerIrp != NULL);
irp = fdoDeviceExtension->CurrentPowerIrp;
fdoDeviceExtension->CurrentPowerIrp = NULL;
#if DBG
{
PIO_STACK_LOCATION irpStack;
SYSTEM_POWER_STATE systemState;
irpStack = IoGetCurrentIrpStackLocation(irp);
systemState = irpStack->Parameters.Power.State.SystemState;
ntStatus = IoStatus->Status;
DBGPRINT(2, ("USBSTOR_FdoSetPowerCompletion %08X %08X %s %08X\n",
fdoDeviceObject, irp,
PowerSystemStateString(systemState),
ntStatus));
LOGENTRY('fspc', fdoDeviceObject, systemState, ntStatus);
}
#endif
// The requested DevicePowerState Irp has completed.
// Now pass down the SystemPowerState Irp which requested the
// DevicePowerState Irp.
PoStartNextPowerIrp(irp);
IoCopyCurrentIrpStackLocationToNext(irp);
// Mark the Irp pending since USBSTOR_FdoSetPower() would have
// originally returned STATUS_PENDING after calling PoRequestPowerIrp().
//
IoMarkIrpPending(irp);
ntStatus = PoCallDriver(fdoDeviceExtension->StackDeviceObject,
irp);
}
//******************************************************************************
//
// USBSTOR_FdoSetPowerD0Completion()
//
// Completion routine used by USBSTOR_FdoSetPower when passing down a
// IRP_MN_SET_POWER DevicePowerState PowerDeviceD0 Irp for the FDO.
//
// The purpose of this routine is to delay unblocking the device queue
// until after the DevicePowerState PowerDeviceD0 Irp completes.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoSetPowerD0Completion (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp,
IN PVOID NotUsed
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
DEVICE_POWER_STATE deviceState;
KIRQL irql;
NTSTATUS ntStatus;
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
irpStack = IoGetCurrentIrpStackLocation(Irp);
deviceState = irpStack->Parameters.Power.State.DeviceState;
ASSERT(deviceState == PowerDeviceD0);
ntStatus = Irp->IoStatus.Status;
DBGPRINT(2, ("USBSTOR_FdoSetPowerD0Completion %08X %08X %s %08X\n",
DeviceObject, Irp,
PowerDeviceStateString(deviceState),
ntStatus));
LOGENTRY('fs0c', DeviceObject, deviceState, ntStatus);
// Powering up. Unblock the device queue which was left blocked
// after USBSTOR_StartIo() passed down the power down Irp.
KeRaiseIrql(DISPATCH_LEVEL, &irql);
{
IoStartNextPacket(DeviceObject, TRUE);
}
KeLowerIrql(irql);
PoStartNextPowerIrp(Irp);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoSetPower()
//
// Dispatch routine which handles IRP_MJ_POWER, IRP_MN_SET_POWER for the PDO
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoSetPower (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
POWER_STATE_TYPE powerType;
POWER_STATE powerState;
BOOLEAN completeRequest;
NTSTATUS ntStatus;
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
// Get our Irp parameters
//
irpStack = IoGetCurrentIrpStackLocation(Irp);
powerType = irpStack->Parameters.Power.Type;
powerState = irpStack->Parameters.Power.State;
DBGPRINT(2, ("enter: USBSTOR_PdoSetPower %08X %s\n",
DeviceObject,
(powerType == SystemPowerState) ?
PowerSystemStateString(powerState.SystemState) :
PowerDeviceStateString(powerState.DeviceState)));
LOGENTRY('PDSP', DeviceObject, Irp, irpStack->MinorFunction);
// Complete the request here, unless we are powering up or down and need
// to wait before completing the request later.
//
completeRequest = TRUE;
if (powerType == SystemPowerState)
{
POWER_STATE newState;
// Update the current system state.
//
pdoDeviceExtension->SystemPowerState = powerState.SystemState;
// Map the new system state to a new device state
//
if (powerState.SystemState != PowerSystemWorking)
{
newState.DeviceState = PowerDeviceD3;
}
else
{
newState.DeviceState = PowerDeviceD0;
}
// If the new device state is different than the current device
// state, request a device state power Irp.
//
if (pdoDeviceExtension->DevicePowerState != newState.DeviceState)
{
DBGPRINT(2, ("Requesting power Irp %08X %08X from %s to %s\n",
DeviceObject, Irp,
PowerDeviceStateString(pdoDeviceExtension->DevicePowerState),
PowerDeviceStateString(newState.DeviceState)));
ASSERT(pdoDeviceExtension->CurrentPowerIrp == NULL);
pdoDeviceExtension->CurrentPowerIrp = Irp;
ntStatus = PoRequestPowerIrp(DeviceObject,
IRP_MN_SET_POWER,
newState,
USBSTOR_PdoSetPowerCompletion,
NULL,
NULL);
ASSERT(ntStatus == STATUS_PENDING);
completeRequest = FALSE;
}
}
else if (powerType == DevicePowerState)
{
POWER_STATE oldState;
DBGPRINT(2, ("Received power Irp %08X %08X from %s to %s\n",
DeviceObject, Irp,
PowerDeviceStateString(pdoDeviceExtension->DevicePowerState),
PowerDeviceStateString(powerState.DeviceState)));
// Update the current device state.
//
oldState.DeviceState = pdoDeviceExtension->DevicePowerState;
pdoDeviceExtension->DevicePowerState = powerState.DeviceState;
if (oldState.DeviceState == PowerDeviceD0 &&
powerState.DeviceState > PowerDeviceD0)
{
// Powering down.
DBGPRINT(2, ("PDO Powering Down\n"));
LOGENTRY('pwrd', DeviceObject, Irp, 0);
}
else if (oldState.DeviceState > PowerDeviceD0 &&
powerState.DeviceState == PowerDeviceD0)
{
// Powering up.
DBGPRINT(2, ("PDO Powering Up\n"));
LOGENTRY('pwru', DeviceObject, Irp, 0);
}
}
if (completeRequest)
{
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
PoStartNextPowerIrp(Irp);
IoCompleteRequest(Irp, IO_NO_INCREMENT);
}
DBGPRINT(2, ("exit: USBSTOR_PdoSetPower %08X\n", ntStatus));
LOGENTRY('pdsp', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoSetPowerCompletion()
//
// Completion routine for PoRequestPowerIrp() in USBSTOR_PdoSetPower.
//
// The purpose of this routine is to block completing the SystemPowerState
// Irp until the requested DevicePowerState Irp completes.
//
//******************************************************************************
VOID
USBSTOR_PdoSetPowerCompletion(
IN PDEVICE_OBJECT DeviceObject,
IN UCHAR MinorFunction,
IN POWER_STATE PowerState,
IN PVOID Context,
IN PIO_STATUS_BLOCK IoStatus
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PIRP irp;
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
ASSERT(pdoDeviceExtension->CurrentPowerIrp != NULL);
irp = pdoDeviceExtension->CurrentPowerIrp;
pdoDeviceExtension->CurrentPowerIrp = NULL;
#if DBG
{
PIO_STACK_LOCATION irpStack;
SYSTEM_POWER_STATE systemState;
NTSTATUS ntStatus;
irpStack = IoGetCurrentIrpStackLocation(irp);
systemState = irpStack->Parameters.Power.State.SystemState;
ntStatus = IoStatus->Status;
DBGPRINT(2, ("USBSTOR_PdoSetPowerCompletion %08X %08X %s %08X\n",
DeviceObject, irp,
PowerSystemStateString(systemState),
ntStatus));
LOGENTRY('pspc', DeviceObject, systemState, ntStatus);
}
#endif
// The requested DevicePowerState Irp has completed.
// Now complete the SystemPowerState Irp which requested the
// DevicePowerState Irp.
// Mark the Irp pending since USBSTOR_PdoSetPower() would have
// originally returned STATUS_PENDING after calling PoRequestPowerIrp().
//
IoMarkIrpPending(irp);
irp->IoStatus.Status = STATUS_SUCCESS;
PoStartNextPowerIrp(irp);
IoCompleteRequest(irp, IO_NO_INCREMENT);
}
//******************************************************************************
//
// USBSTOR_SystemControl()
//
// Dispatch routine which handles IRP_MJ_SYSTEM_CONTROL
//
//******************************************************************************
NTSTATUS
USBSTOR_SystemControl (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PDEVICE_EXTENSION deviceExtension;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
NTSTATUS ntStatus;
PAGED_CODE();
deviceExtension = DeviceObject->DeviceExtension;
irpStack = IoGetCurrentIrpStackLocation(Irp);
DBGPRINT(2, ("enter: USBSTOR_SystemControl %2X\n", irpStack->MinorFunction));
LOGENTRY('SYSC', DeviceObject, Irp, irpStack->MinorFunction);
if (deviceExtension->Type == USBSTOR_DO_TYPE_FDO)
{
// This is an FDO attached to the USB PDO.
//
fdoDeviceExtension = DeviceObject->DeviceExtension;
switch (irpStack->MinorFunction)
{
//
// XXXXX Need to handle any of these?
//
default:
//
// Pass the request down to the next lower driver
//
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
break;
}
}
else
{
// This is a PDO enumerated by our FDO.
ntStatus = Irp->IoStatus.Status;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
}
DBGPRINT(2, ("exit: USBSTOR_SystemControl %08X\n", ntStatus));
LOGENTRY('sysc', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_Pnp()
//
// Dispatch routine which handles IRP_MJ_PNP
//
//******************************************************************************
NTSTATUS
USBSTOR_Pnp (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PDEVICE_EXTENSION deviceExtension;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
NTSTATUS ntStatus;
PAGED_CODE();
deviceExtension = DeviceObject->DeviceExtension;
irpStack = IoGetCurrentIrpStackLocation(Irp);
DBGPRINT(2, ("enter: USBSTOR_Pnp %s\n",
PnPMinorFunctionString(irpStack->MinorFunction)));
LOGENTRY('PNP ', DeviceObject, Irp, irpStack->MinorFunction);
if (deviceExtension->Type == USBSTOR_DO_TYPE_FDO)
{
// This is an FDO attached to the USB PDO.
// We have some real work to do.
//
fdoDeviceExtension = DeviceObject->DeviceExtension;
switch (irpStack->MinorFunction)
{
case IRP_MN_START_DEVICE:
ntStatus = USBSTOR_FdoStartDevice(DeviceObject, Irp);
break;
case IRP_MN_STOP_DEVICE:
ntStatus = USBSTOR_FdoStopDevice(DeviceObject, Irp);
break;
case IRP_MN_REMOVE_DEVICE:
ntStatus = USBSTOR_FdoRemoveDevice(DeviceObject, Irp);
break;
case IRP_MN_QUERY_STOP_DEVICE:
case IRP_MN_QUERY_REMOVE_DEVICE:
ntStatus = USBSTOR_FdoQueryStopRemoveDevice(DeviceObject, Irp);
break;
case IRP_MN_CANCEL_STOP_DEVICE:
case IRP_MN_CANCEL_REMOVE_DEVICE:
ntStatus = USBSTOR_FdoCancelStopRemoveDevice(DeviceObject, Irp);
break;
case IRP_MN_QUERY_DEVICE_RELATIONS:
ntStatus = USBSTOR_FdoQueryDeviceRelations(DeviceObject, Irp);
break;
case IRP_MN_QUERY_CAPABILITIES:
ntStatus = USBSTOR_FdoQueryCapabilities(DeviceObject, Irp);
break;
case IRP_MN_SURPRISE_REMOVAL:
//
// The documentation says to set the status before passing the
// Irp down the stack
//
Irp->IoStatus.Status = STATUS_SUCCESS;
// nothing else special yet, just fall through to default
default:
//
// Pass the request down to the next lower driver
//
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
break;
}
}
else
{
// This is a PDO enumerated by our FDO.
// We don't have too much to do.
//
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
switch (irpStack->MinorFunction)
{
case IRP_MN_START_DEVICE:
ntStatus = USBSTOR_PdoStartDevice(DeviceObject, Irp);
break;
case IRP_MN_QUERY_ID:
ntStatus = USBSTOR_PdoQueryID(DeviceObject, Irp);
break;
case IRP_MN_QUERY_DEVICE_TEXT:
ntStatus = USBSTOR_PdoQueryDeviceText(DeviceObject, Irp);
break;
case IRP_MN_QUERY_DEVICE_RELATIONS:
ntStatus = USBSTOR_PdoQueryDeviceRelations(DeviceObject, Irp);
break;
case IRP_MN_QUERY_CAPABILITIES:
ntStatus = USBSTOR_PdoQueryCapabilities(DeviceObject, Irp);
break;
case IRP_MN_REMOVE_DEVICE:
ntStatus = USBSTOR_PdoRemoveDevice(DeviceObject, Irp);
break;
case IRP_MN_SURPRISE_REMOVAL:
case IRP_MN_STOP_DEVICE:
case IRP_MN_QUERY_STOP_DEVICE:
case IRP_MN_QUERY_REMOVE_DEVICE:
case IRP_MN_CANCEL_STOP_DEVICE:
case IRP_MN_CANCEL_REMOVE_DEVICE:
case IRP_MN_QUERY_PNP_DEVICE_STATE:
// We have no value add for IRP_MN_QUERY_PNP_DEVICE_STATE
// at the moment. At some point we might have reason to
// return PNP_DEVICE_REMOVED or PNP_DEVICE_FAILED.
DBGPRINT(2, ("Succeeding PnP for Child PDO %s\n",
PnPMinorFunctionString(irpStack->MinorFunction)));
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
break;
default:
DBGPRINT(2, ("Unhandled PnP Irp for Child PDO %s\n",
PnPMinorFunctionString(irpStack->MinorFunction)));
ntStatus = Irp->IoStatus.Status;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
break;
}
}
DBGPRINT(2, ("exit: USBSTOR_Pnp %08X\n", ntStatus));
LOGENTRY('pnp ', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoStartDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_START_DEVICE for the FDO
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
// This IRP must be handled first by the underlying bus driver for a device
// and then by each higher driver in the device stack.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoStartDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
USB_BUS_INTERFACE_USBDI_V1 busInterface;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_FdoStartDevice\n"));
DBGFBRK(DBGF_BRK_STARTDEVICE);
LOGENTRY('STRT', DeviceObject, Irp, 0);
fdoDeviceExtension = DeviceObject->DeviceExtension;
// Pass IRP_MN_START_DEVICE Irp down the stack first before we do anything.
//
ntStatus = USBSTOR_SyncPassDownIrp(DeviceObject,
Irp);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Lower driver failed IRP_MN_START_DEVICE\n"));
goto USBSTOR_FdoStartDeviceDone;
}
// Allocate Reset Pipe / Reset Port IoWorkItem
//
if (fdoDeviceExtension->IoWorkItem == NULL)
{
fdoDeviceExtension->IoWorkItem = IoAllocateWorkItem(DeviceObject);
if (fdoDeviceExtension->IoWorkItem == NULL)
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
goto USBSTOR_FdoStartDeviceDone;
}
}
// If this is the first time the device as been started, retrieve the
// Device and Configuration Descriptors from the device.
//
if (fdoDeviceExtension->DeviceDescriptor == NULL)
{
ntStatus = USBSTOR_GetDescriptors(DeviceObject);
if (!NT_SUCCESS(ntStatus))
{
goto USBSTOR_FdoStartDeviceDone;
}
}
// Now configure the device
//
ntStatus = USBSTOR_SelectConfiguration(DeviceObject);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Configure device failed\n"));
goto USBSTOR_FdoStartDeviceDone;
}
// If the driver is loaded during textmode setup then the registry
// value won't exist yet to indicate what type of device this is. If
// the Interface Descriptor indicates that the device is a Bulk-Only
// device then believe it.
//
if ((fdoDeviceExtension->InterfaceDescriptor->bInterfaceClass ==
USB_DEVICE_CLASS_STORAGE) &&
(fdoDeviceExtension->InterfaceDescriptor->bInterfaceProtocol ==
USBSTOR_PROTOCOL_BULK_ONLY) &&
(fdoDeviceExtension->DriverFlags == DeviceProtocolUnspecified))
{
fdoDeviceExtension->DriverFlags = DeviceProtocolBulkOnly;
}
// Find the bulk and interrupt pipes we'll use in this configuration.
//
ntStatus = USBSTOR_GetPipes(DeviceObject);
if (!NT_SUCCESS(ntStatus))
{
goto USBSTOR_FdoStartDeviceDone;
}
// Enable hacks for certain revs of the Y-E Data USB Floppy
//
if (fdoDeviceExtension->DeviceDescriptor->idVendor == 0x057B &&
fdoDeviceExtension->DeviceDescriptor->idProduct == 0x0000 &&
fdoDeviceExtension->DeviceDescriptor->bcdDevice < 0x0128)
{
SET_FLAG(fdoDeviceExtension->DeviceHackFlags, DHF_FORCE_REQUEST_SENSE);
#if 0
SET_FLAG(fdoDeviceExtension->DeviceHackFlags, DHF_TUR_START_UNIT);
SET_FLAG(fdoDeviceExtension->DeviceHackFlags, DHF_MEDIUM_CHANGE_RESET);
#endif
}
// Start timeout timer
//
IoStartTimer(DeviceObject);
// Everything looks good so far, go ahead and create the list of
// child PDOs if this is the first time we have been started and
// the list is empty.
//
if (IsListEmpty(&fdoDeviceExtension->ChildPDOs))
{
UCHAR maxLun;
UCHAR lun;
maxLun = 0;
// Only check devices which claim to be USB Mass Storage Class
// Bulk-Only spec compliant for Multiple LUN support.
//
if ((fdoDeviceExtension->InterfaceDescriptor->bInterfaceClass ==
USB_DEVICE_CLASS_STORAGE) &&
(fdoDeviceExtension->InterfaceDescriptor->bInterfaceProtocol ==
USBSTOR_PROTOCOL_BULK_ONLY))
{
// See if the device supports Multiple LUNs
//
ntStatus = USBSTOR_GetMaxLun(DeviceObject,
&maxLun);
if (NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("GetMaxLun returned %02x\n", maxLun));
// We need to provide a unique InstanceID for each logical unit.
// We use the device USB SerialNumber string as part of the
// unique InstanceID. Without a device USB SerialNumber string
// we can't support multiple logical units on the device.
//
// The Bulk-Only USB Mass Storage class specification requires
// a SerialNumber string so if the device does not have one it
// is not really spec compliant anyway.
//
if (fdoDeviceExtension->SerialNumber == NULL)
{
DBGPRINT(1, ("Multiple Lun but no SerialNumber!\n"));
maxLun = 0;
}
}
}
for (lun = 0; lun <= maxLun; lun++)
{
ntStatus = USBSTOR_CreateChildPDO(DeviceObject, lun);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Create Child PDO %d failed\n", lun));
goto USBSTOR_FdoStartDeviceDone;
}
}
}
if (NT_SUCCESS(USBSTOR_GetBusInterface(DeviceObject, &busInterface)))
{
fdoDeviceExtension->DeviceIsHighSpeed =
busInterface.IsDeviceHighSpeed(busInterface.BusContext);
DBGPRINT(1, ("DeviceIsHighSpeed: %s\n",
fdoDeviceExtension->DeviceIsHighSpeed ? "TRUE" : "FALSE"));
}
else
{
fdoDeviceExtension->DeviceIsHighSpeed = FALSE;
}
USBSTOR_FdoStartDeviceDone:
// Must complete request since completion routine returned
// STATUS_MORE_PROCESSING_REQUIRED
//
Irp->IoStatus.Status = ntStatus;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_FdoStartDevice %08X\n", ntStatus));
LOGENTRY('strt', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_GetDescriptors()
//
// This routine is called at START_DEVICE time for the FDO to retrieve the
// Device and Configurations descriptors from the device and store them in
// the device extension.
//
//******************************************************************************
NTSTATUS
USBSTOR_GetDescriptors (
IN PDEVICE_OBJECT DeviceObject
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PUCHAR descriptor;
ULONG descriptorLength;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_GetDescriptors\n"));
LOGENTRY('GDSC', DeviceObject, 0, 0);
fdoDeviceExtension = DeviceObject->DeviceExtension;
//
// Get Device Descriptor
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_DEVICE_DESCRIPTOR_TYPE,
0, // Index
0, // LanguageId
2, // RetryCount
sizeof(USB_DEVICE_DESCRIPTOR),
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Device Descriptor failed\n"));
goto USBSTOR_GetDescriptorsDone;
}
ASSERT(fdoDeviceExtension->DeviceDescriptor == NULL);
fdoDeviceExtension->DeviceDescriptor = (PUSB_DEVICE_DESCRIPTOR)descriptor;
//
// Get Configuration Descriptor (just the Configuration Descriptor)
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_CONFIGURATION_DESCRIPTOR_TYPE,
0, // Index
0, // LanguageId
2, // RetryCount
sizeof(USB_CONFIGURATION_DESCRIPTOR),
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Configuration Descriptor failed (1)\n"));
goto USBSTOR_GetDescriptorsDone;
}
descriptorLength = ((PUSB_CONFIGURATION_DESCRIPTOR)descriptor)->wTotalLength;
ExFreePool(descriptor);
if (descriptorLength < sizeof(USB_CONFIGURATION_DESCRIPTOR))
{
ntStatus = STATUS_DEVICE_DATA_ERROR;
DBGPRINT(1, ("Get Configuration Descriptor failed (2)\n"));
goto USBSTOR_GetDescriptorsDone;
}
//
// Get Configuration Descriptor (and Interface and Endpoint Descriptors)
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_CONFIGURATION_DESCRIPTOR_TYPE,
0, // Index
0, // LanguageId
2, // RetryCount
descriptorLength,
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Configuration Descriptor failed (3)\n"));
goto USBSTOR_GetDescriptorsDone;
}
ASSERT(fdoDeviceExtension->ConfigurationDescriptor == NULL);
fdoDeviceExtension->ConfigurationDescriptor = (PUSB_CONFIGURATION_DESCRIPTOR)descriptor;
//
// Get the Serial Number String Descriptor, if there is one
//
if (fdoDeviceExtension->DeviceDescriptor->iSerialNumber)
{
USBSTOR_GetStringDescriptors(DeviceObject);
}
#if DBG
DumpDeviceDesc(fdoDeviceExtension->DeviceDescriptor);
DumpConfigDesc(fdoDeviceExtension->ConfigurationDescriptor);
#endif
USBSTOR_GetDescriptorsDone:
DBGPRINT(2, ("exit: USBSTOR_GetDescriptors %08X\n", ntStatus));
LOGENTRY('gdsc', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_GetStringDescriptors()
//
// This routine is called at START_DEVICE time for the FDO to retrieve the
// Serial Number string descriptor from the device and store it in
// the device extension.
//
//******************************************************************************
USBSTOR_GetStringDescriptors (
IN PDEVICE_OBJECT DeviceObject
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PUCHAR descriptor;
ULONG descriptorLength;
USHORT languageId;
ULONG i, numIds;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_GetStringDescriptors\n"));
LOGENTRY('GSDC', DeviceObject, 0, 0);
fdoDeviceExtension = DeviceObject->DeviceExtension;
//
// Get the list of Language IDs (descriptor header only)
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_STRING_DESCRIPTOR_TYPE,
0, // Index
0, // LanguageId
2, // RetryCount
sizeof(USB_COMMON_DESCRIPTOR),
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Language IDs failed (1) %08X\n", ntStatus));
goto USBSTOR_GetStringDescriptorsDone;
}
descriptorLength = ((PUSB_COMMON_DESCRIPTOR)descriptor)->bLength;
ExFreePool(descriptor);
if ((descriptorLength < sizeof(USB_COMMON_DESCRIPTOR) + sizeof(USHORT)) ||
(descriptorLength & 1))
{
ntStatus = STATUS_DEVICE_DATA_ERROR;
DBGPRINT(1, ("Get Language IDs failed (2) %d\n", descriptorLength));
goto USBSTOR_GetStringDescriptorsDone;
}
//
// Get the list of Language IDs (complete descriptor)
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_STRING_DESCRIPTOR_TYPE,
0, // Index
0, // LanguageId
2, // RetryCount
descriptorLength,
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Language IDs failed (3) %08X\n", ntStatus));
goto USBSTOR_GetStringDescriptorsDone;
}
// Search the list of LanguageIDs for US-English (0x0409). If we find
// it in the list, that's the LanguageID we'll use. Else just default
// to the first LanguageID in the list.
numIds = (descriptorLength - sizeof(USB_COMMON_DESCRIPTOR)) / sizeof(USHORT);
languageId = ((PUSHORT)descriptor)[1];
for (i = 2; i <= numIds; i++)
{
if (((PUSHORT)descriptor)[i] == 0x0409)
{
languageId = 0x0409;
break;
}
}
ExFreePool(descriptor);
//
// Get the Serial Number (descriptor header only)
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_STRING_DESCRIPTOR_TYPE,
fdoDeviceExtension->DeviceDescriptor->iSerialNumber,
languageId,
2, // RetryCount
sizeof(USB_COMMON_DESCRIPTOR),
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Serial Number failed (1) %08X\n", ntStatus));
goto USBSTOR_GetStringDescriptorsDone;
}
descriptorLength = ((PUSB_COMMON_DESCRIPTOR)descriptor)->bLength;
ExFreePool(descriptor);
if ((descriptorLength < sizeof(USB_COMMON_DESCRIPTOR) + sizeof(USHORT)) ||
(descriptorLength & 1))
{
ntStatus = STATUS_DEVICE_DATA_ERROR;
DBGPRINT(1, ("Get Serial Number failed (2) %d\n", descriptorLength));
goto USBSTOR_GetStringDescriptorsDone;
}
//
// Get the Serial Number (complete descriptor)
//
ntStatus = USBSTOR_GetDescriptor(DeviceObject,
USB_RECIPIENT_DEVICE,
USB_STRING_DESCRIPTOR_TYPE,
fdoDeviceExtension->DeviceDescriptor->iSerialNumber,
languageId,
2, // RetryCount
descriptorLength,
&descriptor);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Get Serial Number failed (3) %08X\n", ntStatus));
goto USBSTOR_GetStringDescriptorsDone;
}
ASSERT(fdoDeviceExtension->SerialNumber == NULL);
fdoDeviceExtension->SerialNumber = (PUSB_STRING_DESCRIPTOR)descriptor;
USBSTOR_GetStringDescriptorsDone:
DBGPRINT(2, ("exit: USBSTOR_GetStringDescriptors %08X %08X\n",
ntStatus, fdoDeviceExtension->SerialNumber));
LOGENTRY('gdsc', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_AdjustConfigurationDescriptor()
//
// This routine is called at START_DEVICE time for the FDO to adjust the
// Configuration Descriptor, if necessary.
//
// Removes Endpoint Descriptors we won't use. The Configuration Descriptor
// is modified in place.
//
//******************************************************************************
VOID
USBSTOR_AdjustConfigurationDescriptor (
IN PDEVICE_OBJECT DeviceObject,
IN PUSB_CONFIGURATION_DESCRIPTOR ConfigDesc,
OUT PUSB_INTERFACE_DESCRIPTOR *InterfaceDesc,
OUT PLONG BulkInIndex,
OUT PLONG BulkOutIndex,
OUT PLONG InterruptInIndex
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PUCHAR descEnd;
PUSB_COMMON_DESCRIPTOR commonDesc;
PUSB_INTERFACE_DESCRIPTOR interfaceDesc;
PUSB_ENDPOINT_DESCRIPTOR endpointDesc;
LONG endpointIndex;
BOOLEAN removeEndpoint;
PAGED_CODE();
fdoDeviceExtension = DeviceObject->DeviceExtension;
descEnd = (PUCHAR)ConfigDesc + ConfigDesc->wTotalLength;
commonDesc = (PUSB_COMMON_DESCRIPTOR)ConfigDesc;
interfaceDesc = NULL;
*BulkInIndex = -1;
*BulkOutIndex = -1;
*InterruptInIndex = -1;
endpointIndex = 0;
while ((PUCHAR)commonDesc + sizeof(USB_COMMON_DESCRIPTOR) < descEnd &&
(PUCHAR)commonDesc + commonDesc->bLength <= descEnd)
{
// Is this an Interface Descriptor?
//
if ((commonDesc->bDescriptorType == USB_INTERFACE_DESCRIPTOR_TYPE) &&
(commonDesc->bLength == sizeof(USB_INTERFACE_DESCRIPTOR)))
{
// Only bother looking at the first Interface Descriptor
//
if (interfaceDesc != NULL)
{
break;
}
// Remember the first Interface Descriptor we have seen
//
interfaceDesc = (PUSB_INTERFACE_DESCRIPTOR)commonDesc;
}
// Is this an Endpoint Descriptor?
//
if ((commonDesc->bDescriptorType == USB_ENDPOINT_DESCRIPTOR_TYPE) &&
(commonDesc->bLength == sizeof(USB_ENDPOINT_DESCRIPTOR)) &&
(interfaceDesc != NULL))
{
endpointDesc = (PUSB_ENDPOINT_DESCRIPTOR)commonDesc;
// There is currently a bug in the composite parent driver
// that doesn't handle the case where the number of
// endpoints in an Interface Descriptor differs from the
// Interface Descriptor originally returned by the deivce.
// Until that bug is fixed avoid the bug by not stripping
// out endpoints that won't be used.
//
removeEndpoint = FALSE;
if (((endpointDesc->bmAttributes & USB_ENDPOINT_TYPE_MASK) ==
USB_ENDPOINT_TYPE_BULK) &&
(USB_ENDPOINT_DIRECTION_IN(endpointDesc->bEndpointAddress)))
{
if (*BulkInIndex == -1)
{
*BulkInIndex = endpointIndex;
removeEndpoint = FALSE;
}
}
else if (((endpointDesc->bmAttributes & USB_ENDPOINT_TYPE_MASK) ==
USB_ENDPOINT_TYPE_BULK) &&
(USB_ENDPOINT_DIRECTION_OUT(endpointDesc->bEndpointAddress)))
{
if (*BulkOutIndex == -1)
{
*BulkOutIndex = endpointIndex;
removeEndpoint = FALSE;
}
}
else if (((endpointDesc->bmAttributes & USB_ENDPOINT_TYPE_MASK) ==
USB_ENDPOINT_TYPE_INTERRUPT) &&
(USB_ENDPOINT_DIRECTION_IN(endpointDesc->bEndpointAddress)))
{
// Only keep the Interrupt endpoint if we know for sure
// that the device is a CBI device. Don't trust the
// bInterfaceProtocol value of the device. Devices can lie.
//
if ((*InterruptInIndex == -1) &&
(fdoDeviceExtension->DriverFlags == DeviceProtocolCBI))
{
*InterruptInIndex = endpointIndex;
removeEndpoint = FALSE;
}
}
if (removeEndpoint)
{
// Remove this endpoint, we won't use it.
//
DBGPRINT(1, ("Removing Endpoint addr %02X, attr %02X\n",
endpointDesc->bEndpointAddress,
endpointDesc->bmAttributes));
RtlMoveMemory(endpointDesc,
endpointDesc + 1,
descEnd - (PUCHAR)(endpointDesc + 1));
ConfigDesc->wTotalLength -= sizeof(USB_ENDPOINT_DESCRIPTOR);
interfaceDesc->bNumEndpoints -= 1;
descEnd -= sizeof(USB_ENDPOINT_DESCRIPTOR);
continue;
}
else
{
DBGPRINT(1, ("Keeping Endpoint addr %02X, attr %02X\n",
endpointDesc->bEndpointAddress,
endpointDesc->bmAttributes));
endpointIndex++;
}
}
// Advance past this descriptor
//
(PUCHAR)commonDesc += commonDesc->bLength;
}
ASSERT(*BulkInIndex != -1);
ASSERT(*BulkOutIndex != -1);
ASSERT((*InterruptInIndex != -1) ==
(fdoDeviceExtension->DriverFlags == DeviceProtocolCBI));
*InterfaceDesc = interfaceDesc;
}
//******************************************************************************
//
// USBSTOR_GetPipes()
//
// This routine is called at START_DEVICE time find the Bulk IN, Bulk OUT,
// and Interrupt IN endpoints for the device.
//
//******************************************************************************
NTSTATUS
USBSTOR_GetPipes (
IN PDEVICE_OBJECT DeviceObject
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PUSBD_PIPE_INFORMATION pipe;
ULONG i;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_GetPipes\n"));
LOGENTRY('GPIP', DeviceObject, 0, 0);
fdoDeviceExtension = DeviceObject->DeviceExtension;
fdoDeviceExtension->BulkInPipe = NULL;
fdoDeviceExtension->BulkOutPipe = NULL;
fdoDeviceExtension->InterruptInPipe = NULL;
// Find the Bulk IN, Bulk OUT, and Interrupt IN endpoints.
//
for (i=0; i<fdoDeviceExtension->InterfaceInfo->NumberOfPipes; i++)
{
pipe = &fdoDeviceExtension->InterfaceInfo->Pipes[i];
if (pipe->PipeType == UsbdPipeTypeBulk)
{
if (USBD_PIPE_DIRECTION_IN(pipe) &&
fdoDeviceExtension->BulkInPipe == NULL)
{
fdoDeviceExtension->BulkInPipe = pipe;
}
else if (!USBD_PIPE_DIRECTION_IN(pipe) &&
fdoDeviceExtension->BulkOutPipe == NULL)
{
fdoDeviceExtension->BulkOutPipe = pipe;
}
}
else if (pipe->PipeType == UsbdPipeTypeInterrupt)
{
if (USBD_PIPE_DIRECTION_IN(pipe) &&
fdoDeviceExtension->InterruptInPipe == NULL &&
fdoDeviceExtension->DriverFlags == DeviceProtocolCBI)
{
fdoDeviceExtension->InterruptInPipe = pipe;
}
}
}
ntStatus = STATUS_SUCCESS;
if (fdoDeviceExtension->BulkInPipe == NULL)
{
DBGPRINT(1, ("Missing Bulk IN pipe\n"));
ntStatus = STATUS_DEVICE_CONFIGURATION_ERROR;
}
if (fdoDeviceExtension->BulkOutPipe == NULL)
{
DBGPRINT(1, ("Missing Bulk OUT pipe\n"));
ntStatus = STATUS_DEVICE_CONFIGURATION_ERROR;
}
DBGPRINT(2, ("exit: USBSTOR_GetPipes %08X\n", ntStatus));
LOGENTRY('gpip', ntStatus, fdoDeviceExtension->BulkInPipe,
fdoDeviceExtension->BulkOutPipe);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_CreateChildPDO()
//
// This routine is called during START_DEVICE of the FDO to create the
// child PDO. This is only called the first time the FDO is started,
// after the device has its USB configuration selected.
//
//******************************************************************************
NTSTATUS
USBSTOR_CreateChildPDO (
IN PDEVICE_OBJECT FdoDeviceObject,
IN UCHAR Lun
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PDEVICE_OBJECT pdoDeviceObject;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_CreateChildPDO %d\n", Lun));
LOGENTRY('CCPD', FdoDeviceObject, Lun, 0);
fdoDeviceExtension = FdoDeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Create the PDO
//
ntStatus = IoCreateDevice(FdoDeviceObject->DriverObject,
sizeof(PDO_DEVICE_EXTENSION),
NULL,
FILE_DEVICE_MASS_STORAGE,
FILE_AUTOGENERATED_DEVICE_NAME,
FALSE,
&pdoDeviceObject);
if (!NT_SUCCESS(ntStatus))
{
return ntStatus;
}
// The PDO and the FDO are effectively at the same stack level.
// Irps directed at the PDO will sometimes be passed down with
// IoCallDriver() to the FDO->StackDeviceObject.
//
pdoDeviceObject->StackSize = FdoDeviceObject->StackSize;
// Initialize the PDO DeviceExtension
//
pdoDeviceExtension = pdoDeviceObject->DeviceExtension;
// Set all DeviceExtension pointers to NULL and all variable to zero
//
RtlZeroMemory(pdoDeviceExtension, sizeof(PDO_DEVICE_EXTENSION));
// Tag this as a PDO which is the child of an FDO
//
pdoDeviceExtension->Type = USBSTOR_DO_TYPE_PDO;
// Point back to our own DeviceObject
//
pdoDeviceExtension->PdoDeviceObject = pdoDeviceObject;
// Remember the PDO's parent FDO
//
pdoDeviceExtension->ParentFDO = FdoDeviceObject;
// Set the initial system and device power states
//
pdoDeviceExtension->SystemPowerState = PowerSystemWorking;
pdoDeviceExtension->DevicePowerState = PowerDeviceD0;
// Initialize the PDO's PnP device state
//
pdoDeviceExtension->DeviceState = DeviceStateCreated;
// Add the child PDO we just created to the parent's list of child PDOs
//
InsertTailList(&fdoDeviceExtension->ChildPDOs,
&pdoDeviceExtension->ListEntry);
pdoDeviceObject->Flags |= DO_DIRECT_IO;
pdoDeviceObject->Flags |= DO_POWER_PAGABLE;
pdoDeviceObject->Flags &= ~DO_DEVICE_INITIALIZING;
pdoDeviceExtension->LUN = Lun;
// Get the Inquiry Data from the device
//
ntStatus = USBSTOR_GetInquiryData(pdoDeviceObject);
// If the device is a DIRECT_ACCESS_DEVICE, see if it is a floppy
//
if (NT_SUCCESS(ntStatus))
{
PINQUIRYDATA inquiryData;
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
if (inquiryData->DeviceType == DIRECT_ACCESS_DEVICE)
{
pdoDeviceExtension->IsFloppy = USBSTOR_IsFloppyDevice(pdoDeviceObject);
}
}
DBGPRINT(2, ("exit: USBSTOR_CreateChildPDO %08X\n", ntStatus));
LOGENTRY('ccpd', FdoDeviceObject, pdoDeviceObject, ntStatus);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoStopDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_STOP_DEVICE for the FDO
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
// The PnP Manager only sends this IRP if a prior IRP_MN_QUERY_STOP_DEVICE
// completed successfully.
//
// This IRP is handled first by the driver at the top of the device stack and
// then by each lower driver in the attachment chain.
//
// A driver must set Irp->IoStatus.Status to STATUS_SUCCESS. A driver must
// not fail this IRP. If a driver cannot release the device's hardware
// resources, it can fail a query-stop IRP, but once it succeeds the query-stop
// request it must succeed the stop request.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoStopDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_FdoStopDevice\n"));
LOGENTRY('STOP', DeviceObject, Irp, 0);
DBGFBRK(DBGF_BRK_STOPDEVICE);
fdoDeviceExtension = DeviceObject->DeviceExtension;
// Release the device resources allocated during IRP_MN_START_DEVICE
//
// Stop the timeout timer
//
IoStopTimer(DeviceObject);
// Unconfigure the device
//
ntStatus = USBSTOR_UnConfigure(DeviceObject);
// The documentation says to set the status before passing the
// Irp down the stack
//
Irp->IoStatus.Status = STATUS_SUCCESS;
// Pass the IRP_MN_STOP_DEVICE Irp down the stack.
//
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
DBGPRINT(2, ("exit: USBSTOR_FdoStopDevice %08X\n", ntStatus));
LOGENTRY('stop', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoRemoveDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_REMOVE_DEVICE for the FDO
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
// This IRP is handled first by the driver at the top of the device stack and
// then by each lower driver in the attachment chain.
//
// A driver must set Irp->IoStatus.Status to STATUS_SUCCESS. Drivers must not
// fail this IRP.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoRemoveDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_FdoRemoveDevice\n"));
LOGENTRY('REMV', DeviceObject, Irp, 0);
DBGFBRK(DBGF_BRK_REMOVEDEVICE);
fdoDeviceExtension = DeviceObject->DeviceExtension;
// Decrement by one to match the initial one in AddDevice
//
DECREMENT_PENDING_IO_COUNT(fdoDeviceExtension);
LOGENTRY('rem1', DeviceObject, 0, 0);
// Wait for all pending requests to complete
//
KeWaitForSingleObject(&fdoDeviceExtension->RemoveEvent,
Executive,
KernelMode,
FALSE,
NULL);
LOGENTRY('rem2', DeviceObject, 0, 0);
// The child PDOs should have received REMOVE_DEVICE before the FDO.
// Go ahead and delete them now.
//
while (!IsListEmpty(&fdoDeviceExtension->ChildPDOs))
{
PLIST_ENTRY listEntry;
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
listEntry = RemoveTailList(&fdoDeviceExtension->ChildPDOs);
pdoDeviceExtension = CONTAINING_RECORD(listEntry,
PDO_DEVICE_EXTENSION,
ListEntry);
ASSERT(pdoDeviceExtension->DeviceState == DeviceStateCreated ||
pdoDeviceExtension->DeviceState == DeviceStateRemoved);
LOGENTRY('remc', DeviceObject, pdoDeviceExtension->PdoDeviceObject, 0);
IoDeleteDevice(pdoDeviceExtension->PdoDeviceObject);
}
// Free everything that was allocated during IRP_MN_START_DEVICE
//
if (fdoDeviceExtension->IoWorkItem != NULL)
{
IoFreeWorkItem(fdoDeviceExtension->IoWorkItem);
}
if (fdoDeviceExtension->DeviceDescriptor != NULL)
{
ExFreePool(fdoDeviceExtension->DeviceDescriptor);
}
if (fdoDeviceExtension->ConfigurationDescriptor != NULL)
{
ExFreePool(fdoDeviceExtension->ConfigurationDescriptor);
}
if (fdoDeviceExtension->SerialNumber != NULL)
{
ExFreePool(fdoDeviceExtension->SerialNumber);
}
if (fdoDeviceExtension->InterfaceInfo != NULL)
{
ExFreePool(fdoDeviceExtension->InterfaceInfo);
}
// The documentation says to set the status before passing the Irp down
//
Irp->IoStatus.Status = STATUS_SUCCESS;
// Pass the IRP_MN_REMOVE_DEVICE Irp down the stack.
//
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
LOGENTRY('rem3', DeviceObject, 0, 0);
// Free everything that was allocated during AddDevice
//
IoDetachDevice(fdoDeviceExtension->StackDeviceObject);
IoDeleteDevice(DeviceObject);
DBGPRINT(2, ("exit: USBSTOR_FdoRemoveDevice %08X\n", ntStatus));
LOGENTRY('remv', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoQueryStopRemoveDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_STOP_DEVICE and
// IRP_MN_QUERY_REMOVE_DEVICE for the FDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
// This IRP is handled first by the driver at the top of the device stack and
// then by each lower driver in the attachment chain.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoQueryStopRemoveDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_FdoQueryStopRemoveDevice\n"));
LOGENTRY('QSRD', DeviceObject, Irp, 0);
DBGFBRK(DBGF_BRK_QUERYSTOPDEVICE);
fdoDeviceExtension = DeviceObject->DeviceExtension;
// The documentation says to set the status before passing the Irp down
//
Irp->IoStatus.Status = STATUS_SUCCESS;
// Pass the IRP_MN_QUERY_STOP/REMOVE_DEVICE Irp down the stack.
//
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
DBGPRINT(2, ("exit: USBSTOR_FdoQueryStopRemoveDevice %08X\n", ntStatus));
LOGENTRY('qsrd', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoCancelStopRemoveDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_CANCEL_STOP_DEVICE and
// IRP_MN_CANCEL_REMOVE_DEVICE for the FDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
// This IRP must be handled first by the underlying bus driver for a device
// and then by each higher driver in the device stack.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoCancelStopRemoveDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_FdoCancelStopRemoveDevice\n"));
LOGENTRY('CSRD', DeviceObject, Irp, 0);
DBGFBRK(DBGF_BRK_CANCELSTOPDEVICE);
fdoDeviceExtension = DeviceObject->DeviceExtension;
// The documentation says to set the status before passing the Irp down
//
Irp->IoStatus.Status = STATUS_SUCCESS;
// Pass the IRP_MN_CANCEL_STOP/REMOVE_DEVICE Irp down the stack.
//
ntStatus = USBSTOR_SyncPassDownIrp(DeviceObject,
Irp);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Lower driver failed IRP_MN_CANCEL_STOP/REMOVE_DEVICE\n"));
goto USBSTOR_CancelStopRemoveDeviceDone;
}
USBSTOR_CancelStopRemoveDeviceDone:
// Must complete request since completion routine returned
// STATUS_MORE_PROCESSING_REQUIRED
//
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_FdoCancelStopRemoveDevice %08X\n", ntStatus));
LOGENTRY('csrd', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoQueryDeviceRelations()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_DEVICE_RELATIONS for the FDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// an arbitrary thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoQueryDeviceRelations (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
DEVICE_RELATION_TYPE relationType;
NTSTATUS ntStatus;
PAGED_CODE();
fdoDeviceExtension = DeviceObject->DeviceExtension;
irpStack = IoGetCurrentIrpStackLocation(Irp);
relationType = irpStack->Parameters.QueryDeviceRelations.Type;
DBGPRINT(2, ("enter: USBSTOR_FdoQueryDeviceRelations %d\n",
relationType));
LOGENTRY('FQDR', DeviceObject, Irp, relationType);
switch (relationType)
{
case BusRelations:
if (!IsListEmpty(&fdoDeviceExtension->ChildPDOs))
{
// If we have children to return, add them to the existing
// relation list, if there is one, else create and add them
// to a new relation list.
//
// Then in either case, pass the request down the driver stack.
//
PDEVICE_RELATIONS oldRelations;
PDEVICE_RELATIONS newRelations;
PLIST_ENTRY listHead;
PLIST_ENTRY listEntry;
ULONG oldCount;
ULONG childCount;
ULONG index;
listHead = &fdoDeviceExtension->ChildPDOs;
// How many children?
//
for (listEntry = listHead->Flink, childCount = 0;
listEntry != listHead;
listEntry = listEntry->Flink, childCount++)
;
oldRelations = (PDEVICE_RELATIONS)Irp->IoStatus.Information;
if (oldRelations)
{
// Add our children to the existing relation list.
oldCount = oldRelations->Count;
// A DEVICE_RELATIONS structure has room for one
// PDEVICE_OBJECT to start with, so subtract that
// out of the size we allocate.
//
newRelations = ExAllocatePoolWithTag(
PagedPool,
sizeof(DEVICE_RELATIONS) +
sizeof(PDEVICE_OBJECT) *
(oldCount + childCount - 1),
POOL_TAG);
if (newRelations)
{
// Copy the existing relation list
//
for (index = 0; index < oldCount; index++)
{
newRelations->Objects[index] =
oldRelations->Objects[index];
}
}
// Now we're done the the existing relation list, free it
//
ExFreePool(oldRelations);
}
else
{
// Create a new relation list for our children
newRelations = ExAllocatePoolWithTag(
PagedPool,
sizeof(DEVICE_RELATIONS) +
sizeof(PDEVICE_OBJECT) *
(childCount - 1),
POOL_TAG);
oldCount = 0;
index = 0;
}
if (newRelations)
{
newRelations->Count = oldCount + childCount;
// Add our child relations at the end of the list
//
for (listEntry = listHead->Flink;
listEntry != listHead;
listEntry = listEntry->Flink)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
pdoDeviceExtension = CONTAINING_RECORD(
listEntry,
PDO_DEVICE_EXTENSION,
ListEntry);
newRelations->Objects[index++] =
pdoDeviceExtension->PdoDeviceObject;
ObReferenceObject(pdoDeviceExtension->PdoDeviceObject);
DBGPRINT(2, ("returning ChildPDO %08X\n",
pdoDeviceExtension->PdoDeviceObject));
}
ASSERT(index == oldCount + childCount);
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
Irp->IoStatus.Information = (ULONG_PTR)newRelations;
}
else
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
Irp->IoStatus.Status = ntStatus;
Irp->IoStatus.Information = 0;
}
}
else
{
// If we don't have a child to return, just pass the request
// down without doing anything.
//
ntStatus = STATUS_SUCCESS;
}
break;
case EjectionRelations:
case PowerRelations:
case RemovalRelations:
case TargetDeviceRelation:
default:
//
// Pass the request down the driver stack without doing anything.
//
ntStatus = STATUS_SUCCESS;
break;
}
if (NT_SUCCESS(ntStatus))
{
// Pass the Irp down the driver stack if successful so far.
//
IoSkipCurrentIrpStackLocation(Irp);
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
}
else
{
// Unsuccessful, just complete the request now.
//
IoCompleteRequest(Irp, IO_NO_INCREMENT);
}
DBGPRINT(2, ("exit: USBSTOR_FdoQueryDeviceRelations %08X\n", ntStatus));
LOGENTRY('fqdr', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_FdoQueryCapabilities()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_CAPABILITIES for the FDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// an arbitrary thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_FdoQueryCapabilities (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
PDEVICE_CAPABILITIES deviceCapabilities;
NTSTATUS ntStatus;
PAGED_CODE();
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
DBGPRINT(2, ("enter: USBSTOR_FdoQueryCapabilities\n"));
LOGENTRY('FQCP', DeviceObject, Irp, 0);
irpStack = IoGetCurrentIrpStackLocation(Irp);
deviceCapabilities = irpStack->Parameters.DeviceCapabilities.Capabilities;
// Pass IRP_MN_QUERY_CAPABILITIES Irp down the stack first before we do
// anything.
//
ntStatus = USBSTOR_SyncPassDownIrp(DeviceObject,
Irp);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Lower driver failed IRP_MN_QUERY_CAPABILITIES\n"));
}
else
{
if (fdoDeviceExtension->NonRemovable)
{
deviceCapabilities->Removable = FALSE;
}
}
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_FdoQueryCapabilities %08X\n", ntStatus));
LOGENTRY('fqcp', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoStartDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_START_DEVICE for the PDO
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoStartDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoStartDevice\n"));
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_PdoStartDevice %08X\n", ntStatus));
LOGENTRY('pstr', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoRemoveDevice()
//
// This routine handles IRP_MJ_PNP, IRP_MN_REMOVE_DEVICE for the PDO
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// system thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoRemoveDevice (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoRemoveDevice\n"));
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
pdoDeviceExtension->Claimed = FALSE;
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_PdoRemoveDevice %08X\n", ntStatus));
LOGENTRY('prmd', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoQueryID()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_ID for the PDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// an arbitrary thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoQueryID (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PIO_STACK_LOCATION irpStack;
UNICODE_STRING unicodeStr;
BOOLEAN multiStrings;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoQueryID\n"));
LOGENTRY('PQID', DeviceObject, Irp, 0);
irpStack = IoGetCurrentIrpStackLocation(Irp);
// Initialize return value to NULL
//
RtlInitUnicodeString(&unicodeStr, NULL);
switch (irpStack->Parameters.QueryId.IdType)
{
case BusQueryDeviceID:
ntStatus = USBSTOR_PdoQueryDeviceId(
DeviceObject,
&unicodeStr);
multiStrings = FALSE;
break;
case BusQueryHardwareIDs:
ntStatus = USBSTOR_PdoQueryHardwareIds(
DeviceObject,
&unicodeStr);
multiStrings = TRUE;
break;
case BusQueryCompatibleIDs:
ntStatus = USBSTOR_PdoQueryCompatibleIds(
DeviceObject,
&unicodeStr);
multiStrings = TRUE;
break;
case BusQueryInstanceID:
ntStatus = USBSTOR_PdoBusQueryInstanceId(
DeviceObject,
&unicodeStr);
multiStrings = FALSE;
break;
default:
ntStatus = STATUS_NOT_SUPPORTED;
break;
}
if (NT_SUCCESS(ntStatus) && unicodeStr.Buffer)
{
PWCHAR idString;
//
// fix up all invalid characters
//
idString = unicodeStr.Buffer;
while (*idString)
{
if ((*idString <= L' ') ||
(*idString > (WCHAR)0x7F) ||
(*idString == L','))
{
*idString = L'_';
}
idString++;
if ((*idString == L'\0') && multiStrings)
{
idString++;
}
}
Irp->IoStatus.Information = (ULONG_PTR)unicodeStr.Buffer;
}
else
{
Irp->IoStatus.Information = (ULONG_PTR)NULL;
}
Irp->IoStatus.Status = ntStatus;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_PdoQueryID %08X\n", ntStatus));
LOGENTRY('pqid', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoDeviceTypeString()
//
// This routine returns a device type string for the PDO.
//
//******************************************************************************
PCHAR
USBSTOR_PdoDeviceTypeString (
IN PDEVICE_OBJECT DeviceObject
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PINQUIRYDATA inquiryData;
PAGED_CODE();
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
switch (inquiryData->DeviceType)
{
case DIRECT_ACCESS_DEVICE:
return pdoDeviceExtension->IsFloppy ? "SFloppy" : "Disk";
case WRITE_ONCE_READ_MULTIPLE_DEVICE:
return "Worm";
case READ_ONLY_DIRECT_ACCESS_DEVICE:
return "CdRom";
case OPTICAL_DEVICE:
return "Optical";
case MEDIUM_CHANGER:
return "Changer";
case SEQUENTIAL_ACCESS_DEVICE:
return "Sequential";
default:
return "Other";
}
}
//******************************************************************************
//
// USBSTOR_PdoGenericTypeString()
//
// This routine returns a device type string for the PDO.
//
//******************************************************************************
PCHAR
USBSTOR_PdoGenericTypeString (
IN PDEVICE_OBJECT DeviceObject
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PINQUIRYDATA inquiryData;
PAGED_CODE();
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
switch (inquiryData->DeviceType)
{
case DIRECT_ACCESS_DEVICE:
return pdoDeviceExtension->IsFloppy ? "GenSFloppy" : "GenDisk";
case WRITE_ONCE_READ_MULTIPLE_DEVICE:
return "GenWorm";
case READ_ONLY_DIRECT_ACCESS_DEVICE:
return "GenCdRom";
case OPTICAL_DEVICE:
return "GenOptical";
case MEDIUM_CHANGER:
return "GenChanger";
case SEQUENTIAL_ACCESS_DEVICE:
return "GenSequential";
default:
return "UsbstorOther";
}
}
//******************************************************************************
//
// CopyField()
//
// This routine will copy Count string bytes from Source to Destination.
// If it finds a nul byte in the Source it will translate that and any
// subsequent bytes into Change. It will also replace spaces with the
// specified Change character.
//
//******************************************************************************
VOID
CopyField (
IN PUCHAR Destination,
IN PUCHAR Source,
IN ULONG Count,
IN UCHAR Change
)
{
ULONG i;
BOOLEAN pastEnd;
PAGED_CODE();
pastEnd = FALSE;
for (i = 0; i < Count; i++)
{
if (!pastEnd)
{
if (Source[i] == 0)
{
pastEnd = TRUE;
Destination[i] = Change;
} else if (Source[i] == ' ')
{
Destination[i] = Change;
} else
{
Destination[i] = Source[i];
}
}
else
{
Destination[i] = Change;
}
}
return;
}
//******************************************************************************
//
// USBSTOR_StringArrayToMultiSz()
//
// This routine will take a null terminated array of ascii strings and merge
// them together into a unicode multi-string block.
//
// This routine allocates memory for the string buffer - it is the caller's
// responsibility to free it.
//
//******************************************************************************
NTSTATUS
USBSTOR_StringArrayToMultiSz(
PUNICODE_STRING MultiString,
PCSTR StringArray[]
)
{
ANSI_STRING ansiEntry;
UNICODE_STRING unicodeEntry;
UCHAR i;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_StringArrayToMultiSz %08X %08X\n",
MultiString, StringArray));
// Make sure we aren't going to leak any memory
//
ASSERT(MultiString->Buffer == NULL);
RtlInitUnicodeString(MultiString, NULL);
// First add up the sizes of the converted ascii strings to determine
// how big the multisz will be.
//
for (i = 0; StringArray[i] != NULL; i++)
{
RtlInitAnsiString(&ansiEntry, StringArray[i]);
MultiString->Length += (USHORT)RtlAnsiStringToUnicodeSize(&ansiEntry);
}
ASSERT(MultiString->Length != 0);
// Add room for the double NULL terminator
//
MultiString->MaximumLength = MultiString->Length + sizeof(UNICODE_NULL);
// Now allocate a buffer for the multisz
//
MultiString->Buffer = ExAllocatePoolWithTag(PagedPool,
MultiString->MaximumLength,
POOL_TAG);
if (MultiString->Buffer == NULL)
{
return STATUS_INSUFFICIENT_RESOURCES;
}
RtlZeroMemory(MultiString->Buffer, MultiString->MaximumLength);
unicodeEntry = *MultiString;
// Now convert each ascii string in the array into a unicode string
// in the multisz
//
for (i = 0; StringArray[i] != NULL; i++)
{
RtlInitAnsiString(&ansiEntry, StringArray[i]);
ntStatus = RtlAnsiStringToUnicodeString(&unicodeEntry,
&ansiEntry,
FALSE);
// Since we're not allocating any memory the only failure possible
// is if this function is bad
ASSERT(NT_SUCCESS(ntStatus));
// Push the buffer location up and reduce the maximum count
//
((PSTR) unicodeEntry.Buffer) += unicodeEntry.Length + sizeof(WCHAR);
unicodeEntry.MaximumLength -= unicodeEntry.Length + sizeof(WCHAR);
};
DBGPRINT(2, ("exit: USBSTOR_StringArrayToMultiSz\n"));
return STATUS_SUCCESS;
}
//******************************************************************************
//
// USBSTOR_PdoQueryDeviceId()
//
// This routine handles IRP_MN_QUERY_ID BusQueryDeviceID for the PDO.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoQueryDeviceId (
IN PDEVICE_OBJECT DeviceObject,
OUT PUNICODE_STRING UnicodeString
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PINQUIRYDATA inquiryData;
UCHAR buffer[128];
PUCHAR rawIdString;
ANSI_STRING ansiIdString;
ULONG whichString;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoQueryDeviceId\n"));
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
RtlZeroMemory(buffer, sizeof(buffer));
rawIdString = USBSTOR_PdoDeviceTypeString(DeviceObject);
sprintf(buffer, "USBSTOR\\%s", rawIdString);
rawIdString = buffer + strlen(buffer);
for (whichString = 0; whichString < 3; whichString++)
{
PUCHAR headerString;
PUCHAR sourceString;
ULONG sourceStringLength;
ULONG i;
switch (whichString)
{
//
// Vendor Id
//
case 0:
sourceString = inquiryData->VendorId;
sourceStringLength = sizeof(inquiryData->VendorId);
headerString = "Ven";
break;
//
// Product Id
//
case 1:
sourceString = inquiryData->ProductId;
sourceStringLength = sizeof(inquiryData->ProductId);
headerString = "Prod";
break;
//
// Product Revision Level
//
case 2:
sourceString = inquiryData->ProductRevisionLevel;
sourceStringLength = sizeof(inquiryData->ProductRevisionLevel);
headerString = "Rev";
break;
}
//
// Start at the end of the source string and back up until we find a
// non-space, non-null character.
//
for (; sourceStringLength > 0; sourceStringLength--)
{
if((sourceString[sourceStringLength - 1] != ' ') &&
(sourceString[sourceStringLength - 1] != '\0'))
{
break;
}
}
//
// Throw the header string into the block
//
sprintf(rawIdString, "&%s_", headerString);
rawIdString += strlen(headerString) + 2;
//
// Spew the string into the device id
//
for(i = 0; i < sourceStringLength; i++)
{
*rawIdString = (sourceString[i] != ' ') ? (sourceString[i]) :
('_');
rawIdString++;
}
}
RtlInitAnsiString(&ansiIdString, buffer);
ntStatus = RtlAnsiStringToUnicodeString(UnicodeString, &ansiIdString, TRUE);
DBGPRINT(2, ("exit: USBSTOR_PdoQueryDeviceId %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoQueryHardwareIds()
//
// This routine handles IRP_MN_QUERY_ID BusQueryHardwareIDs for the PDO.
//
//******************************************************************************
#define NUMBER_HARDWARE_STRINGS 7
NTSTATUS
USBSTOR_PdoQueryHardwareIds (
IN PDEVICE_OBJECT DeviceObject,
OUT PUNICODE_STRING UnicodeString
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PINQUIRYDATA inquiryData;
PUCHAR devTypeString;
PUCHAR genTypeString;
ULONG i;
PSTR strings[NUMBER_HARDWARE_STRINGS + 1];
UCHAR scratch[128];
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoQueryHardwareIds\n"));
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
devTypeString = USBSTOR_PdoDeviceTypeString(DeviceObject);
genTypeString = USBSTOR_PdoGenericTypeString(DeviceObject);
ntStatus = STATUS_SUCCESS;
RtlZeroMemory(strings, sizeof(strings));
for (i = 0; i < NUMBER_HARDWARE_STRINGS; i++)
{
RtlZeroMemory(scratch, sizeof(scratch));
// First build each string in the scratch buffer
//
switch (i)
{
//
// Bus + Dev Type + Vendor + Product + Revision
//
case 0:
sprintf(scratch, "USBSTOR\\%s", devTypeString);
CopyField(scratch + strlen(scratch),
inquiryData->VendorId,
8,
'_');
CopyField(scratch + strlen(scratch),
inquiryData->ProductId,
16,
'_');
CopyField(scratch + strlen(scratch),
inquiryData->ProductRevisionLevel,
4,
'_');
break;
//
// Bus + Dev Type + Vendor + Product
//
case 1:
sprintf(scratch, "USBSTOR\\%s", devTypeString);
CopyField(scratch + strlen(scratch),
inquiryData->VendorId,
8,
'_');
CopyField(scratch + strlen(scratch),
inquiryData->ProductId,
16,
'_');
break;
//
// Bus + Dev Type + Vendor
//
case 2:
sprintf(scratch, "USBSTOR\\%s", devTypeString);
CopyField(scratch + strlen(scratch),
inquiryData->VendorId,
8,
'_');
break;
//
// Bus + Vendor + Product + Revision[0]
//
case 3:
sprintf(scratch, "USBSTOR\\");
//
// Fall through to the next set.
//
//
// Vendor + Product + Revision[0] (win9x)
//
case 4:
CopyField(scratch + strlen(scratch),
inquiryData->VendorId,
8,
'_');
CopyField(scratch + strlen(scratch),
inquiryData->ProductId,
16,
'_');
CopyField(scratch + strlen(scratch),
inquiryData->ProductRevisionLevel,
1,
'_');
break;
//
// Bus + Generic Type
//
case 5:
sprintf(scratch, "USBSTOR\\%s", genTypeString);
break;
//
// Generic Type
//
case 6:
sprintf(scratch, "%s", genTypeString);
break;
default:
ASSERT(FALSE);
break;
}
// Now allocate a tmp buffer for this string and copy the scratch
// buffer to the tmp buffer
//
if (strlen(scratch) != 0)
{
strings[i] = ExAllocatePoolWithTag(
PagedPool,
strlen(scratch) + sizeof(UCHAR),
POOL_TAG);
if (strings[i] == NULL)
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
break;
}
else
{
strcpy(strings[i], scratch);
}
}
}
if (NT_SUCCESS(ntStatus))
{
// Now convert the array of stings to one Unicode MultiSz
//
ntStatus = USBSTOR_StringArrayToMultiSz(UnicodeString, strings);
}
// Now free up the tmp buffers for each string
//
for (i = 0; i < NUMBER_HARDWARE_STRINGS; i++)
{
if (strings[i])
{
ExFreePool(strings[i]);
}
}
DBGPRINT(2, ("exit: USBSTOR_PdoQueryHardwareIds %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoQueryCompatibleIds()
//
// This routine handles IRP_MN_QUERY_ID BusQueryCompatibleIDs for the PDO.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoQueryCompatibleIds (
IN PDEVICE_OBJECT DeviceObject,
OUT PUNICODE_STRING UnicodeString
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PINQUIRYDATA inquiryData;
PUCHAR devTypeString;
UCHAR s[sizeof("USBSTOR\\DEVICE_TYPE_GOES_HERE")];
PSTR strings[] = {s, "USBSTOR\\RAW", NULL};
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoQueryCompatibleIds\n"));
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
devTypeString = USBSTOR_PdoDeviceTypeString(DeviceObject);
sprintf(s, "USBSTOR\\%s", devTypeString);
ntStatus = USBSTOR_StringArrayToMultiSz(UnicodeString, strings);
DBGPRINT(2, ("exit: USBSTOR_PdoQueryCompatibleIds %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoQueryDeviceText()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_DEVICE_TEXT for the PDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// an arbitrary thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoQueryDeviceText (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PINQUIRYDATA inquiryData;
PIO_STACK_LOCATION irpStack;
DEVICE_TEXT_TYPE textType;
UCHAR ansiBuffer[256];
ANSI_STRING ansiText;
UNICODE_STRING unicodeText;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoQueryDeviceText\n"));
LOGENTRY('PQDT', DeviceObject, Irp, 0);
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
inquiryData = (PINQUIRYDATA)pdoDeviceExtension->InquiryDataBuffer;
irpStack = IoGetCurrentIrpStackLocation(Irp);
textType = irpStack->Parameters.QueryDeviceText.DeviceTextType;
if (textType == DeviceTextDescription)
{
PUCHAR c;
LONG i;
RtlZeroMemory(ansiBuffer, sizeof(ansiBuffer));
RtlCopyMemory(ansiBuffer,
inquiryData->VendorId,
sizeof(inquiryData->VendorId));
c = ansiBuffer;
for (i = sizeof(inquiryData->VendorId)-1; i >= 0; i--)
{
if((c[i] != '\0') &&
(c[i] != ' '))
{
i++;
break;
}
}
c += i;
*c++ = ' ';
RtlCopyMemory(c,
inquiryData->ProductId,
sizeof(inquiryData->ProductId));
for (i = sizeof(inquiryData->ProductId)-1; i >= 0; i--)
{
if((c[i] != '\0') &&
(c[i] != ' '))
{
i++;
break;
}
}
c += i;
*c++ = ' ';
sprintf(c, "USB Device");
RtlInitAnsiString(&ansiText, ansiBuffer);
ntStatus = RtlAnsiStringToUnicodeString(&unicodeText,
&ansiText,
TRUE);
if (NT_SUCCESS(ntStatus))
{
Irp->IoStatus.Information = (ULONG_PTR)unicodeText.Buffer;
}
else
{
Irp->IoStatus.Information = (ULONG_PTR)NULL;
}
}
else
{
// If a device does not provide description or location information,
// the device's underlying bus driver completes the IRP without
// modifying Irp->IoStatus.Status or Ipr->IoStatus.Information.
//
ntStatus = Irp->IoStatus.Status;
}
Irp->IoStatus.Status = ntStatus;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_PdoQueryDeviceText %08X\n", ntStatus));
LOGENTRY('pqdt', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoBusQueryInstanceId()
//
// This routine handles IRP_MN_QUERY_ID BusQueryInstanceID for the PDO.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoBusQueryInstanceId (
IN PDEVICE_OBJECT DeviceObject,
OUT PUNICODE_STRING UnicodeString
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
USHORT length;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_PdoBusQueryInstanceId\n"));
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
fdoDeviceExtension = pdoDeviceExtension->ParentFDO->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
if (fdoDeviceExtension->SerialNumber == NULL)
{
// If we set DEVICE_CAPABILITIES.UniqueID = 0 in response to a
// IRP_MN_QUERY_CAPABILITIES, we can return a NULL ID in response
// to a BusQueryInstanceID.
//
ntStatus = STATUS_SUCCESS;
}
else
{
// Return an NULL-terminated InstanceId string with the format:
// <USB Device SerialNumberString> + '&' + <LUN value in hex>
//
length = fdoDeviceExtension->SerialNumber->bLength -
sizeof(USB_COMMON_DESCRIPTOR) +
3 * sizeof(WCHAR);
UnicodeString->Buffer = ExAllocatePoolWithTag(
PagedPool,
length,
POOL_TAG);
if (UnicodeString->Buffer == NULL)
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
else
{
UnicodeString->Length = length - sizeof(WCHAR);
UnicodeString->MaximumLength = length;
// Copy the USB Device SerialNumberString
//
RtlCopyMemory(UnicodeString->Buffer,
&fdoDeviceExtension->SerialNumber->bString[0],
length - 3 * sizeof(WCHAR));
// Append a '&'
//
UnicodeString->Buffer[length/sizeof(WCHAR) - 3] = (WCHAR)'&';
// Append the LUN value in hex
//
if (pdoDeviceExtension->LUN <= 9)
{
UnicodeString->Buffer[length/sizeof(WCHAR) - 2] =
(WCHAR)('0' + pdoDeviceExtension->LUN);
}
else
{
UnicodeString->Buffer[length/sizeof(WCHAR) - 2] =
(WCHAR)('A' + pdoDeviceExtension->LUN - 0xA);
}
UnicodeString->Buffer[length/sizeof(WCHAR) - 1] =
UNICODE_NULL;
ntStatus = STATUS_SUCCESS;
}
}
DBGPRINT(2, ("exit: USBSTOR_PdoBusQueryInstanceId %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoQueryDeviceRelations()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_DEVICE_RELATIONS for the PDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// an arbitrary thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoQueryDeviceRelations (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PIO_STACK_LOCATION irpStack;
DEVICE_RELATION_TYPE relationType;
PDEVICE_RELATIONS newRelations;
NTSTATUS ntStatus;
PAGED_CODE();
irpStack = IoGetCurrentIrpStackLocation(Irp);
relationType = irpStack->Parameters.QueryDeviceRelations.Type;
DBGPRINT(2, ("enter: USBSTOR_PdoQueryDeviceRelations %d\n",
relationType));
LOGENTRY('PQDR', DeviceObject, Irp, relationType);
switch (relationType)
{
case TargetDeviceRelation:
//
// Return a relation list containing ourself.
//
newRelations = ExAllocatePoolWithTag(
PagedPool,
sizeof(DEVICE_RELATIONS),
POOL_TAG);
if (newRelations)
{
newRelations->Count = 1;
newRelations->Objects[0] = DeviceObject;
ObReferenceObject(DeviceObject);
ntStatus = STATUS_SUCCESS;
Irp->IoStatus.Status = ntStatus;
Irp->IoStatus.Information = (ULONG_PTR)newRelations;
}
else
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
Irp->IoStatus.Status = ntStatus;
Irp->IoStatus.Information = 0;
}
break;
case BusRelations:
case EjectionRelations:
case PowerRelations:
case RemovalRelations:
default:
//
// Just complete the request with it's current status
//
ntStatus = Irp->IoStatus.Status;
break;
}
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_PdoQueryDeviceRelations %08X\n", ntStatus));
LOGENTRY('pqdr', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_PdoQueryCapabilities()
//
// This routine handles IRP_MJ_PNP, IRP_MN_QUERY_CAPABILITIES for the PDO.
//
// The PnP Manager sends this IRP at IRQL PASSIVE_LEVEL in the context of a
// an arbitrary thread.
//
//******************************************************************************
NTSTATUS
USBSTOR_PdoQueryCapabilities (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PPDO_DEVICE_EXTENSION pdoDeviceExtension;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIO_STACK_LOCATION irpStack;
PDEVICE_CAPABILITIES deviceCapabilities;
NTSTATUS ntStatus;
PAGED_CODE();
pdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(pdoDeviceExtension->Type == USBSTOR_DO_TYPE_PDO);
fdoDeviceExtension = pdoDeviceExtension->ParentFDO->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
DBGPRINT(2, ("enter: USBSTOR_PdoQueryCapabilities\n"));
LOGENTRY('PQCP', DeviceObject, Irp, 0);
irpStack = IoGetCurrentIrpStackLocation(Irp);
deviceCapabilities = irpStack->Parameters.DeviceCapabilities.Capabilities;
// Pass IRP_MN_QUERY_CAPABILITIES Irp down the stack first before we do
// anything.
//
ntStatus = USBSTOR_SyncPassDownIrp(pdoDeviceExtension->ParentFDO,
Irp);
if (!NT_SUCCESS(ntStatus))
{
DBGPRINT(1, ("Lower driver failed IRP_MN_QUERY_CAPABILITIES\n"));
}
else
{
if (fdoDeviceExtension->SerialNumber == NULL)
{
deviceCapabilities->UniqueID = FALSE;
}
deviceCapabilities->Removable = FALSE;
}
IoCompleteRequest(Irp, IO_NO_INCREMENT);
DBGPRINT(2, ("exit: USBSTOR_PdoQueryCapabilities %08X\n", ntStatus));
LOGENTRY('pqcp', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_SyncPassDownIrp()
//
//******************************************************************************
NTSTATUS
USBSTOR_SyncPassDownIrp (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
KEVENT localevent;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_SyncPassDownIrp\n"));
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Initialize the event we'll wait on
//
KeInitializeEvent(&localevent,
SynchronizationEvent,
FALSE);
// Copy down Irp params for the next driver
//
IoCopyCurrentIrpStackLocationToNext(Irp);
// Set the completion routine, which will signal the event
//
IoSetCompletionRoutine(Irp,
USBSTOR_SyncCompletionRoutine,
&localevent,
TRUE, // InvokeOnSuccess
TRUE, // InvokeOnError
TRUE); // InvokeOnCancel
// Pass the Irp down the stack
//
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
Irp);
// If the request is pending, block until it completes
//
if (ntStatus == STATUS_PENDING)
{
KeWaitForSingleObject(&localevent,
Executive,
KernelMode,
FALSE,
NULL);
ntStatus = Irp->IoStatus.Status;
}
DBGPRINT(2, ("exit: USBSTOR_SyncPassDownIrp %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_SyncCompletionRoutine()
//
// Completion routine used by USBSTOR_SyncPassDownIrp and
// USBSTOR_SyncSendUsbRequest
//
// If the Irp is one we allocated ourself, DeviceObject is NULL.
//
//******************************************************************************
NTSTATUS
USBSTOR_SyncCompletionRoutine (
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp,
IN PVOID Context
)
{
PKEVENT kevent;
LOGENTRY('SCR ', DeviceObject, Irp, Irp->IoStatus.Status);
kevent = (PKEVENT)Context;
KeSetEvent(kevent,
IO_NO_INCREMENT,
FALSE);
return STATUS_MORE_PROCESSING_REQUIRED;
}
//******************************************************************************
//
// USBSTOR_SyncSendUsbRequest()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_SyncSendUsbRequest (
IN PDEVICE_OBJECT DeviceObject,
IN PURB Urb
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
KEVENT localevent;
PIRP irp;
PIO_STACK_LOCATION nextStack;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(3, ("enter: USBSTOR_SyncSendUsbRequest\n"));
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Initialize the event we'll wait on
//
KeInitializeEvent(&localevent,
SynchronizationEvent,
FALSE);
// Allocate the Irp
//
irp = IoAllocateIrp(fdoDeviceExtension->StackDeviceObject->StackSize, FALSE);
LOGENTRY('SSUR', DeviceObject, irp, Urb);
if (irp == NULL)
{
return STATUS_INSUFFICIENT_RESOURCES;
}
// Set the Irp parameters
//
nextStack = IoGetNextIrpStackLocation(irp);
nextStack->MajorFunction = IRP_MJ_INTERNAL_DEVICE_CONTROL;
nextStack->Parameters.DeviceIoControl.IoControlCode =
IOCTL_INTERNAL_USB_SUBMIT_URB;
nextStack->Parameters.Others.Argument1 = Urb;
// Set the completion routine, which will signal the event
//
IoSetCompletionRoutine(irp,
USBSTOR_SyncCompletionRoutine,
&localevent,
TRUE, // InvokeOnSuccess
TRUE, // InvokeOnError
TRUE); // InvokeOnCancel
// Pass the Irp & Urb down the stack
//
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
irp);
// If the request is pending, block until it completes
//
if (ntStatus == STATUS_PENDING)
{
LARGE_INTEGER timeout;
// Specify a timeout of 5 seconds to wait for this call to complete.
//
timeout.QuadPart = -10000 * 5000;
ntStatus = KeWaitForSingleObject(&localevent,
Executive,
KernelMode,
FALSE,
&timeout);
if (ntStatus == STATUS_TIMEOUT)
{
ntStatus = STATUS_IO_TIMEOUT;
// Cancel the Irp we just sent.
//
IoCancelIrp(irp);
// And wait until the cancel completes
//
KeWaitForSingleObject(&localevent,
Executive,
KernelMode,
FALSE,
NULL);
}
else
{
ntStatus = irp->IoStatus.Status;
}
}
// Done with the Irp, now free it.
//
IoFreeIrp(irp);
LOGENTRY('ssur', ntStatus, Urb, Urb->UrbHeader.Status);
DBGPRINT(3, ("exit: USBSTOR_SyncSendUsbRequest %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_GetDescriptor()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_GetDescriptor (
IN PDEVICE_OBJECT DeviceObject,
IN UCHAR Recipient,
IN UCHAR DescriptorType,
IN UCHAR Index,
IN USHORT LanguageId,
IN ULONG RetryCount,
IN ULONG DescriptorLength,
OUT PUCHAR *Descriptor
)
{
USHORT function;
PURB urb;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_GetDescriptor\n"));
*Descriptor = NULL;
// Set the URB function based on Recipient {Device, Interface, Endpoint}
//
switch (Recipient)
{
case USB_RECIPIENT_DEVICE:
function = URB_FUNCTION_GET_DESCRIPTOR_FROM_DEVICE;
break;
case USB_RECIPIENT_INTERFACE:
function = URB_FUNCTION_GET_DESCRIPTOR_FROM_INTERFACE;
break;
case USB_RECIPIENT_ENDPOINT:
function = URB_FUNCTION_GET_DESCRIPTOR_FROM_ENDPOINT;
break;
default:
return STATUS_INVALID_PARAMETER;
}
// Allocate a descriptor buffer
//
*Descriptor = ExAllocatePoolWithTag(NonPagedPool,
DescriptorLength,
POOL_TAG);
if (*Descriptor != NULL)
{
// Allocate a URB for the Get Descriptor request
//
urb = ExAllocatePoolWithTag(NonPagedPool,
sizeof(struct _URB_CONTROL_DESCRIPTOR_REQUEST),
POOL_TAG);
if (urb != NULL)
{
do
{
// Initialize the URB
//
urb->UrbHeader.Function = function;
urb->UrbHeader.Length = sizeof(struct _URB_CONTROL_DESCRIPTOR_REQUEST);
urb->UrbControlDescriptorRequest.TransferBufferLength = DescriptorLength;
urb->UrbControlDescriptorRequest.TransferBuffer = *Descriptor;
urb->UrbControlDescriptorRequest.TransferBufferMDL = NULL;
urb->UrbControlDescriptorRequest.UrbLink = NULL;
urb->UrbControlDescriptorRequest.DescriptorType = DescriptorType;
urb->UrbControlDescriptorRequest.Index = Index;
urb->UrbControlDescriptorRequest.LanguageId = LanguageId;
// Send the URB down the stack
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject,
urb);
if (NT_SUCCESS(ntStatus))
{
// No error, make sure the length and type are correct
//
if ((DescriptorLength ==
urb->UrbControlDescriptorRequest.TransferBufferLength) &&
(DescriptorType ==
((PUSB_COMMON_DESCRIPTOR)*Descriptor)->bDescriptorType))
{
// The length and type are correct, all done
//
break;
}
else
{
// No error, but the length or type is incorrect
//
ntStatus = STATUS_DEVICE_DATA_ERROR;
}
}
} while (RetryCount-- > 0);
ExFreePool(urb);
}
else
{
// Failed to allocate the URB
//
ExFreePool(*Descriptor);
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
}
else
{
// Failed to allocate the descriptor buffer
//
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
if (!NT_SUCCESS(ntStatus))
{
if (*Descriptor != NULL)
{
ExFreePool(*Descriptor);
*Descriptor = NULL;
}
}
DBGPRINT(2, ("exit: USBSTOR_GetDescriptor %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_GetMaxLun()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_GetMaxLun (
IN PDEVICE_OBJECT DeviceObject,
OUT PUCHAR MaxLun
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PUCHAR maxLunBuf;
ULONG retryCount;
PURB urb;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_GetMaxLun\n"));
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Return zero unless we successfully return a non-zero value
//
*MaxLun = 0;
// Allocate a URB for the Get Max LUN request, plus an extra byte at
// the end for the transfer buffer.
//
urb = ExAllocatePoolWithTag(NonPagedPool,
sizeof(URB) + 1,
POOL_TAG);
if (urb != NULL)
{
// Get a pointer to the transfer buffer, which is the byte immediately
// after the end of the URB.
//
maxLunBuf = (PUCHAR)(urb + 1);
retryCount = 2;
do
{
// Initialize the Control Transfer URB, all fields default to zero
//
RtlZeroMemory(urb, sizeof(URB) + 1);
CLASS_URB(urb).Hdr.Length = sizeof(CLASS_URB(urb));
CLASS_URB(urb).Hdr.Function = URB_FUNCTION_CLASS_INTERFACE;
CLASS_URB(urb).TransferFlags = USBD_TRANSFER_DIRECTION_IN;
CLASS_URB(urb).TransferBufferLength = 1;
CLASS_URB(urb).TransferBuffer = maxLunBuf;
// CLASS_URB(urb).TransferBufferMDL is already zero
// CLASS_URB(urb).RequestTypeReservedBits is already zero
CLASS_URB(urb).Request = BULK_ONLY_GET_MAX_LUN;
// CLASS_URB(urb).Value is already zero
// Target the request at the proper interface on the device
//
CLASS_URB(urb).Index = fdoDeviceExtension->InterfaceInfo->InterfaceNumber;
// Send the URB down the stack
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject,
urb);
if (NT_SUCCESS(ntStatus))
{
// No error, make sure the length is correct
//
if (CLASS_URB(urb).TransferBufferLength == 1)
{
// The length is correct, return the value if it looks ok
//
if (*maxLunBuf <= BULK_ONLY_MAXIMUM_LUN)
{
*MaxLun = *maxLunBuf;
}
else
{
ntStatus = STATUS_DEVICE_DATA_ERROR;
}
break;
}
else
{
// No error, but the length or type is incorrect
//
ntStatus = STATUS_DEVICE_DATA_ERROR;
}
}
else if (USBD_STATUS(CLASS_URB(urb).Hdr.Status) ==
USBD_STATUS(USBD_STATUS_STALL_PID))
{
// Some devices which do not support the Get Max LUN request
// get confused and will STALL a CBW on the Bulk endpoint
// it if immediately follows the Get Max LUN request.
//
// It should never be necessary to send a Clear_Feature
// Endpoint_Stall for Control EP0, but doing so appears to
// be one way to unconfuse devices which are confused by the
// Get Max LUN request.
// Initialize the Control Transfer URB, all fields default to zero
//
RtlZeroMemory(urb, sizeof(URB));
FEATURE_URB(urb).Hdr.Length = sizeof(FEATURE_URB(urb));
FEATURE_URB(urb).Hdr.Function = URB_FUNCTION_CLEAR_FEATURE_TO_ENDPOINT;
FEATURE_URB(urb).FeatureSelector = USB_FEATURE_ENDPOINT_STALL;
// FEATURE_URB(urb).Index is already zero
// Send the URB down the stack
//
USBSTOR_SyncSendUsbRequest(DeviceObject,
urb);
}
} while (retryCount-- > 0);
ExFreePool(urb);
}
else
{
// Failed to allocate the URB
//
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
DBGPRINT(2, ("exit: USBSTOR_GetMaxLun %08X\n", ntStatus));
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_SelectConfiguration()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_SelectConfiguration (
IN PDEVICE_OBJECT DeviceObject
)
{
PURB urb;
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PUSB_CONFIGURATION_DESCRIPTOR configurationDescriptor;
PUSBD_INTERFACE_LIST_ENTRY interfaceList;
PUSB_INTERFACE_DESCRIPTOR interfaceDescriptor;
PUSBD_INTERFACE_INFORMATION interfaceInfo;
LONG i;
LONG bulkInIndex;
LONG bulkOutIndex;
LONG interruptInIndex;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_SelectConfiguration\n"));
LOGENTRY('SCON', DeviceObject, 0, 0);
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
configurationDescriptor = fdoDeviceExtension->ConfigurationDescriptor;
// Allocate storage for an Inteface List to use as an input/output
// parameter to USBD_CreateConfigurationRequestEx().
//
interfaceList = ExAllocatePoolWithTag(
PagedPool,
sizeof(USBD_INTERFACE_LIST_ENTRY) * 2,
POOL_TAG);
if (interfaceList)
{
// Mutate configuration descriptor to suit our wishes
//
USBSTOR_AdjustConfigurationDescriptor(
DeviceObject,
fdoDeviceExtension->ConfigurationDescriptor,
&interfaceDescriptor,
&bulkInIndex,
&bulkOutIndex,
&interruptInIndex);
// Save the Interface Descriptor pointer so we don't have
// to parse the Configuration Descriptor again in case we
// want to look at it.
//
fdoDeviceExtension->InterfaceDescriptor = interfaceDescriptor;
if (interfaceDescriptor)
{
// Add the single Interface Descriptor we care about to the
// interface list, then terminate the list.
//
interfaceList[0].InterfaceDescriptor = interfaceDescriptor;
interfaceList[1].InterfaceDescriptor = NULL;
// USBD will fail a SELECT_CONFIGURATION request if the Config
// Descriptor bNumInterfaces does not match the number of interfaces
// in the SELECT_CONFIGURATION request. Since we are ignoring
// any interfaces other than the first interface, set the Config
// Descriptor bNumInterfaces to 1.
//
// This is only necessary in case this driver is loaded for an
// entire multiple interface device and not as a single interface
// child of the composite parent driver.
//
configurationDescriptor->bNumInterfaces = 1;
// Create a SELECT_CONFIGURATION URB, turning the Interface
// Descriptors in the interfaceList into USBD_INTERFACE_INFORMATION
// structures in the URB.
//
urb = USBD_CreateConfigurationRequestEx(
configurationDescriptor,
interfaceList
);
if (urb)
{
// Now issue the USB request to set the Configuration
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject,
urb);
if (NT_SUCCESS(ntStatus))
{
// Save the configuration handle for this device in
// the Device Extension.
//
fdoDeviceExtension->ConfigurationHandle =
urb->UrbSelectConfiguration.ConfigurationHandle;
interfaceInfo = &urb->UrbSelectConfiguration.Interface;
// Save a copy of the interface information returned
// by the SELECT_CONFIGURATION request in the Device
// Extension. This gives us a list of PIPE_INFORMATION
// structures for each pipe opened in this configuration.
//
ASSERT(fdoDeviceExtension->InterfaceInfo == NULL);
fdoDeviceExtension->InterfaceInfo =
ExAllocatePoolWithTag(NonPagedPool,
interfaceInfo->Length,
POOL_TAG);
if (fdoDeviceExtension->InterfaceInfo)
{
RtlCopyMemory(fdoDeviceExtension->InterfaceInfo,
interfaceInfo,
interfaceInfo->Length);
}
else
{
// Could not allocate a copy of interface information
//
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
}
if (NT_SUCCESS(ntStatus))
{
// Reuse the SELECT_CONFIGURATION request URB as a
// SELECT_INTERFACE request URB and send down a request to
// select the default alternate interface setting that is
// currently in effect. The point of this seemingly
// useless request is to make sure the endpoint
// MaximumTransferSize values are in effect.
//
// When USBHUB is loaded as a composite parent for a
// multiple interface device it ignores SELECT_CONFIGURATION
// requests from child device drivers. In particular the
// MaximumTransferSize values of child driver SELECT_CONFIGURATION
// requests are ignored and the default 4KB value remains
// in effect. The composite parent driver will respect the
// MaximumTransferSize values of child driver SELECT_INTERFACE
// requests.
//
ASSERT(GET_SELECT_INTERFACE_REQUEST_SIZE(fdoDeviceExtension->InterfaceInfo->NumberOfPipes) <
GET_SELECT_CONFIGURATION_REQUEST_SIZE(1, fdoDeviceExtension->InterfaceInfo->NumberOfPipes));
RtlZeroMemory(urb, GET_SELECT_INTERFACE_REQUEST_SIZE(fdoDeviceExtension->InterfaceInfo->NumberOfPipes));
urb->UrbSelectInterface.Hdr.Length =
(USHORT)GET_SELECT_INTERFACE_REQUEST_SIZE(fdoDeviceExtension->InterfaceInfo->NumberOfPipes);
urb->UrbSelectInterface.Hdr.Function =
URB_FUNCTION_SELECT_INTERFACE;
urb->UrbSelectInterface.ConfigurationHandle =
fdoDeviceExtension->ConfigurationHandle;
interfaceInfo = &urb->UrbSelectInterface.Interface;
RtlCopyMemory(interfaceInfo,
fdoDeviceExtension->InterfaceInfo,
fdoDeviceExtension->InterfaceInfo->Length);
// Override the USBD_DEFAULT_MAXIMUM_TRANSFER_SIZE
// for all pipes.
//
for (i=0; i<(LONG)interfaceInfo->NumberOfPipes; i++)
{
if (i == bulkInIndex || i == bulkOutIndex)
{
interfaceInfo->Pipes[i].MaximumTransferSize =
USBSTOR_MAX_TRANSFER_SIZE;
DBGPRINT(1, ("Set pipe %d MaximumTransferSize to %X\n",
i,
interfaceInfo->Pipes[i].MaximumTransferSize));
}
else if (i == interruptInIndex)
{
interfaceInfo->Pipes[i].MaximumTransferSize =
sizeof(USHORT);
DBGPRINT(1, ("Set pipe %d MaximumTransferSize to %X\n",
i,
interfaceInfo->Pipes[i].MaximumTransferSize));
}
}
// Now issue the USB request to set the Interface
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject,
urb);
if (NT_SUCCESS(ntStatus))
{
ASSERT(interfaceInfo->Length ==
fdoDeviceExtension->InterfaceInfo->Length);
RtlCopyMemory(fdoDeviceExtension->InterfaceInfo,
interfaceInfo,
fdoDeviceExtension->InterfaceInfo->Length);
}
}
// Done with the URB
//
ExFreePool(urb);
}
else
{
// Could not allocate urb
//
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
}
else
{
// Did not parse an Interface Descriptor out of the Configuration
// Descriptor, the Configuration Descriptor must be bad.
//
ntStatus = STATUS_UNSUCCESSFUL;
}
// Done with the interface list
//
ExFreePool(interfaceList);
}
else
{
// Could not allocate Interface List
//
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
DBGPRINT(2, ("exit: USBSTOR_SelectConfiguration %08X\n", ntStatus));
LOGENTRY('scon', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_UnConfigure()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_UnConfigure (
IN PDEVICE_OBJECT DeviceObject
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
NTSTATUS ntStatus;
PURB urb;
ULONG ulSize;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_UnConfigure\n"));
LOGENTRY('UCON', DeviceObject, 0, 0);
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
// Allocate a URB for the SELECT_CONFIGURATION request. As we are
// unconfiguring the device, the request needs no pipe and interface
// information structures.
//
ulSize = sizeof(struct _URB_SELECT_CONFIGURATION) -
sizeof(USBD_INTERFACE_INFORMATION);
urb = ExAllocatePoolWithTag(NonPagedPool, ulSize, POOL_TAG);
if (urb)
{
// Initialize the URB. A NULL Configuration Descriptor indicates
// that the device should be unconfigured.
//
UsbBuildSelectConfigurationRequest(urb,
(USHORT)ulSize,
NULL);
// Now issue the USB request to set the Configuration
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject,
urb);
// Done with the URB now.
//
ExFreePool(urb);
fdoDeviceExtension->ConfigurationHandle = 0;
// Free the copy of the interface information that was allocated in
// USBSTOR_SelectConfiguration().
//
if (fdoDeviceExtension->InterfaceInfo != NULL)
{
ExFreePool(fdoDeviceExtension->InterfaceInfo);
fdoDeviceExtension->InterfaceInfo = NULL;
}
}
else
{
// Could not allocate the URB.
//
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
DBGPRINT(2, ("exit: USBSTOR_UnConfigure %08X\n", ntStatus));
LOGENTRY('ucon', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_ResetPipe()
//
// This will reset the host pipe to Data0 and should also reset the device
// endpoint to Data0 for Bulk and Interrupt pipes by issuing a Clear_Feature
// Endpoint_Stall to the device endpoint.
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_ResetPipe (
IN PDEVICE_OBJECT DeviceObject,
IN USBD_PIPE_HANDLE Pipe
)
{
PURB urb;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_ResetPipe\n"));
LOGENTRY('RESP', DeviceObject, Pipe, 0);
// Allocate URB for RESET_PIPE request
//
urb = ExAllocatePoolWithTag(NonPagedPool,
sizeof(struct _URB_PIPE_REQUEST),
POOL_TAG);
if (urb != NULL)
{
// Initialize RESET_PIPE request URB
//
urb->UrbHeader.Length = sizeof (struct _URB_PIPE_REQUEST);
urb->UrbHeader.Function = URB_FUNCTION_RESET_PIPE;
urb->UrbPipeRequest.PipeHandle = Pipe;
// Submit RESET_PIPE request URB
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject, urb);
// Done with URB for RESET_PIPE request, free it
//
ExFreePool(urb);
}
else
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
DBGPRINT(2, ("exit: USBSTOR_ResetPipe %08X\n", ntStatus));
LOGENTRY('resp', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_AbortPipe()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_AbortPipe (
IN PDEVICE_OBJECT DeviceObject,
IN USBD_PIPE_HANDLE Pipe
)
{
PURB urb;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(2, ("enter: USBSTOR_AbortPipe\n"));
LOGENTRY('ABRT', DeviceObject, Pipe, 0);
// Allocate URB for ABORT_PIPE request
//
urb = ExAllocatePoolWithTag(NonPagedPool,
sizeof(struct _URB_PIPE_REQUEST),
POOL_TAG);
if (urb != NULL)
{
// Initialize ABORT_PIPE request URB
//
urb->UrbHeader.Length = sizeof (struct _URB_PIPE_REQUEST);
urb->UrbHeader.Function = URB_FUNCTION_ABORT_PIPE;
urb->UrbPipeRequest.PipeHandle = Pipe;
// Submit ABORT_PIPE request URB
//
ntStatus = USBSTOR_SyncSendUsbRequest(DeviceObject, urb);
// Done with URB for ABORT_PIPE request, free it
//
ExFreePool(urb);
}
else
{
ntStatus = STATUS_INSUFFICIENT_RESOURCES;
}
DBGPRINT(2, ("exit: USBSTOR_AbortPipe %08X\n", ntStatus));
LOGENTRY('abrt', ntStatus, 0, 0);
return ntStatus;
}
//******************************************************************************
//
// USBSTOR_GetBusInterface()
//
// Must be called at IRQL PASSIVE_LEVEL
//
//******************************************************************************
NTSTATUS
USBSTOR_GetBusInterface (
IN PDEVICE_OBJECT DeviceObject,
IN PUSB_BUS_INTERFACE_USBDI_V1 BusInterface
)
{
PFDO_DEVICE_EXTENSION fdoDeviceExtension;
PIRP irp;
KEVENT localevent;
PIO_STACK_LOCATION nextStack;
NTSTATUS ntStatus;
PAGED_CODE();
DBGPRINT(1, ("enter: USBSTOR_GetBusInterface\n"));
fdoDeviceExtension = DeviceObject->DeviceExtension;
ASSERT(fdoDeviceExtension->Type == USBSTOR_DO_TYPE_FDO);
RtlZeroMemory(BusInterface, sizeof(*BusInterface));
// Allocate the Irp
//
irp = IoAllocateIrp((CCHAR)(fdoDeviceExtension->StackDeviceObject->StackSize),
FALSE);
if (irp == NULL)
{
return STATUS_INSUFFICIENT_RESOURCES;
}
// Initialize the event we'll wait on.
//
KeInitializeEvent(&localevent,
SynchronizationEvent,
FALSE);
// Set the Irp parameters
//
nextStack = IoGetNextIrpStackLocation(irp);
nextStack->MajorFunction = IRP_MJ_PNP;
nextStack->MinorFunction = IRP_MN_QUERY_INTERFACE;
nextStack->Parameters.DeviceIoControl.IoControlCode =
IOCTL_INTERNAL_USB_GET_PORT_STATUS;
nextStack->Parameters.QueryInterface.Interface =
(PINTERFACE)BusInterface;
nextStack->Parameters.QueryInterface.InterfaceSpecificData =
NULL;
nextStack->Parameters.QueryInterface.InterfaceType =
&USB_BUS_INTERFACE_USBDI_GUID;
nextStack->Parameters.QueryInterface.Size =
sizeof(*BusInterface);
nextStack->Parameters.QueryInterface.Version =
USB_BUSIF_USBDI_VERSION_1;
// Set the completion routine, which will signal the event
//
IoSetCompletionRoutineEx(DeviceObject,
irp,
USBSTOR_SyncCompletionRoutine,
&localevent,
TRUE, // InvokeOnSuccess
TRUE, // InvokeOnError
TRUE); // InvokeOnCancel
// All PnP IRP's need the Status field initialized to STATUS_NOT_SUPPORTED
//
irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
// Pass the Irp down the stack
//
ntStatus = IoCallDriver(fdoDeviceExtension->StackDeviceObject,
irp);
// If the request is pending, block until it completes
//
if (ntStatus == STATUS_PENDING)
{
KeWaitForSingleObject(&localevent,
Executive,
KernelMode,
FALSE,
NULL);
ntStatus = irp->IoStatus.Status;
}
IoFreeIrp(irp);
if (NT_SUCCESS(ntStatus))
{
ASSERT(BusInterface->Version == USB_BUSIF_USBDI_VERSION_1);
ASSERT(BusInterface->Size == sizeof(*BusInterface));
}
DBGPRINT(1, ("exit: USBSTOR_GetBusInterface %08X\n", ntStatus));
return ntStatus;
}