reactos/drivers/usb/usbport/pnp.c

1808 lines
57 KiB
C
Raw Normal View History

/*
* PROJECT: ReactOS USB Port Driver
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* PURPOSE: USBPort plug and play functions
* COPYRIGHT: Copyright 2017 Vadim Galyant <vgal@rambler.ru>
*/
#include "usbport.h"
#define NDEBUG
#include <debug.h>
#define NDEBUG_USBPORT_CORE
#include "usbdebug.h"
IO_COMPLETION_ROUTINE USBPORT_FdoStartCompletion;
NTSTATUS
NTAPI
USBPORT_FdoStartCompletion(IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp,
IN PVOID Context)
{
KeSetEvent((PKEVENT)Context, EVENT_INCREMENT, FALSE);
return STATUS_MORE_PROCESSING_REQUIRED;
}
NTSTATUS
NTAPI
USBPORT_RegisterDeviceInterface(IN PDEVICE_OBJECT PdoDevice,
IN PDEVICE_OBJECT DeviceObject,
IN CONST GUID *InterfaceClassGuid,
IN BOOLEAN Enable)
{
PUSBPORT_RHDEVICE_EXTENSION DeviceExtension;
PUNICODE_STRING SymbolicLinkName;
NTSTATUS Status;
DPRINT("USBPORT_RegisterDeviceInterface: Enable - %x\n", Enable);
DeviceExtension = DeviceObject->DeviceExtension;
SymbolicLinkName = &DeviceExtension->CommonExtension.SymbolicLinkName;
if (Enable)
{
Status = IoRegisterDeviceInterface(PdoDevice,
InterfaceClassGuid,
NULL,
SymbolicLinkName);
if (NT_SUCCESS(Status))
{
DeviceExtension->CommonExtension.IsInterfaceEnabled = 1;
Status = USBPORT_SetRegistryKeyValue(PdoDevice,
FALSE,
REG_SZ,
L"SymbolicName",
SymbolicLinkName->Buffer,
SymbolicLinkName->Length);
if (NT_SUCCESS(Status))
{
DPRINT("USBPORT_RegisterDeviceInterface: LinkName - %wZ\n",
&DeviceExtension->CommonExtension.SymbolicLinkName);
Status = IoSetDeviceInterfaceState(SymbolicLinkName, TRUE);
}
}
}
else
{
/* Disable device interface */
Status = IoSetDeviceInterfaceState(SymbolicLinkName, FALSE);
if (NT_SUCCESS(Status))
{
RtlFreeUnicodeString(SymbolicLinkName);
DeviceExtension->CommonExtension.IsInterfaceEnabled = 0; // Disabled interface
}
}
return Status;
}
BOOLEAN
NTAPI
USBPORT_IsSelectiveSuspendEnabled(IN PDEVICE_OBJECT FdoDevice)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
ULONG Disabled = 0;
DPRINT("USBPORT_IsSelectiveSuspendEnabled: ... \n");
FdoExtension = FdoDevice->DeviceExtension;
USBPORT_GetRegistryKeyValueFullInfo(FdoDevice,
FdoExtension->CommonExtension.LowerPdoDevice,
TRUE,
L"HcDisableSelectiveSuspend",
sizeof(L"HcDisableSelectiveSuspend"),
&Disabled,
sizeof(Disabled));
return (Disabled == 0);
}
NTSTATUS
NTAPI
USBPORT_GetConfigValue(IN PWSTR ValueName,
IN ULONG ValueType,
IN PVOID ValueData,
IN ULONG ValueLength,
IN PVOID Context,
IN PVOID EntryContext)
{
NTSTATUS Status = STATUS_SUCCESS;
DPRINT("USBPORT_GetConfigValue \n");
if (ValueType == REG_DWORD)
{
*(PULONG)EntryContext = *(PULONG)ValueData;
}
else
{
Status = STATUS_INVALID_PARAMETER;
}
return Status;
}
NTSTATUS
NTAPI
USBPORT_GetDefaultBIOSx(IN PDEVICE_OBJECT FdoDevice,
IN PULONG UsbBIOSx,
IN PULONG DisableSelectiveSuspend,
IN PULONG DisableCcDetect,
IN PULONG IdleEpSupport,
IN PULONG IdleEpSupportEx,
IN PULONG SoftRetry)
{
RTL_QUERY_REGISTRY_TABLE QueryTable[7];
DPRINT("USBPORT_GetDefaultBIOS_X: ... \n");
RtlZeroMemory(QueryTable, sizeof(QueryTable));
*UsbBIOSx = 2;
QueryTable[0].QueryRoutine = USBPORT_GetConfigValue;
QueryTable[0].Flags = 0;
QueryTable[0].Name = L"UsbBIOSx";
QueryTable[0].EntryContext = UsbBIOSx;
QueryTable[0].DefaultType = REG_DWORD;
QueryTable[0].DefaultData = UsbBIOSx;
QueryTable[0].DefaultLength = sizeof(ULONG);
QueryTable[1].QueryRoutine = USBPORT_GetConfigValue;
QueryTable[1].Flags = 0;
QueryTable[1].Name = L"DisableSelectiveSuspend";
QueryTable[1].EntryContext = DisableSelectiveSuspend;
QueryTable[1].DefaultType = REG_DWORD;
QueryTable[1].DefaultData = DisableSelectiveSuspend;
QueryTable[1].DefaultLength = sizeof(ULONG);
QueryTable[2].QueryRoutine = USBPORT_GetConfigValue;
QueryTable[2].Flags = 0;
QueryTable[2].Name = L"DisableCcDetect";
QueryTable[2].EntryContext = DisableCcDetect;
QueryTable[2].DefaultType = REG_DWORD;
QueryTable[2].DefaultData = DisableCcDetect;
QueryTable[2].DefaultLength = sizeof(ULONG);
QueryTable[3].QueryRoutine = USBPORT_GetConfigValue;
QueryTable[3].Flags = 0;
QueryTable[3].Name = L"EnIdleEndpointSupport";
QueryTable[3].EntryContext = IdleEpSupport;
QueryTable[3].DefaultType = REG_DWORD;
QueryTable[3].DefaultData = IdleEpSupport;
QueryTable[3].DefaultLength = sizeof(ULONG);
QueryTable[4].QueryRoutine = USBPORT_GetConfigValue;
QueryTable[4].Flags = 0;
QueryTable[4].Name = L"EnIdleEndpointSupportEx";
QueryTable[4].EntryContext = IdleEpSupportEx;
QueryTable[4].DefaultType = REG_DWORD;
QueryTable[4].DefaultData = IdleEpSupportEx;
QueryTable[4].DefaultLength = sizeof(ULONG);
QueryTable[5].QueryRoutine = USBPORT_GetConfigValue;
QueryTable[5].Flags = 0;
QueryTable[5].Name = L"EnSoftRetry";
QueryTable[5].EntryContext = SoftRetry;
QueryTable[5].DefaultType = REG_DWORD;
QueryTable[5].DefaultData = SoftRetry;
QueryTable[5].DefaultLength = sizeof(ULONG);
return RtlQueryRegistryValues(RTL_REGISTRY_SERVICES,
L"usb",
QueryTable,
NULL,
NULL);
}
NTSTATUS
NTAPI
USBPORT_IsCompanionController(IN PDEVICE_OBJECT DeviceObject,
IN BOOLEAN *IsCompanion)
{
PDEVICE_OBJECT HighestDevice;
PIRP Irp;
KEVENT Event;
PIO_STACK_LOCATION IoStack;
PCI_DEVICE_PRESENT_INTERFACE PciInterface = {0};
PCI_DEVICE_PRESENCE_PARAMETERS Parameters = {0};
IO_STATUS_BLOCK IoStatusBlock;
NTSTATUS Status;
BOOLEAN IsPresent;
DPRINT("USBPORT_IsCompanionController: ... \n");
*IsCompanion = FALSE;
KeInitializeEvent(&Event, SynchronizationEvent, FALSE);
HighestDevice = IoGetAttachedDeviceReference(DeviceObject);
Irp = IoBuildSynchronousFsdRequest(IRP_MJ_PNP,
HighestDevice,
NULL,
0,
NULL,
&Event,
&IoStatusBlock);
if (!Irp)
{
Status = STATUS_INSUFFICIENT_RESOURCES;
ObDereferenceObject(HighestDevice);
return Status;
}
IoStack = IoGetNextIrpStackLocation(Irp);
Irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
Irp->IoStatus.Information = 0;
IoStack->MinorFunction = IRP_MN_QUERY_INTERFACE;
IoStack->Parameters.QueryInterface.InterfaceType = &GUID_PCI_DEVICE_PRESENT_INTERFACE;
IoStack->Parameters.QueryInterface.Size = sizeof(PCI_DEVICE_PRESENT_INTERFACE);
IoStack->Parameters.QueryInterface.Version = 1;
IoStack->Parameters.QueryInterface.Interface = (PINTERFACE)&PciInterface;
IoStack->Parameters.QueryInterface.InterfaceSpecificData = 0;
Status = IoCallDriver(HighestDevice, Irp);
if (Status == STATUS_PENDING)
{
KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, NULL);
Status = IoStatusBlock.Status;
}
if (!NT_SUCCESS(Status))
{
DPRINT1("USBPORT_IsCompanionController: query interface failed\\n");
ObDereferenceObject(HighestDevice);
return Status;
}
DPRINT("USBPORT_IsCompanionController: query interface succeeded\n");
if (PciInterface.Size < sizeof(PCI_DEVICE_PRESENT_INTERFACE))
{
DPRINT1("USBPORT_IsCompanionController: old version\n");
ObDereferenceObject(HighestDevice);
return Status;
}
Parameters.Size = sizeof(PCI_DEVICE_PRESENT_INTERFACE);
Parameters.BaseClass = PCI_CLASS_SERIAL_BUS_CTLR;
Parameters.SubClass = PCI_SUBCLASS_SB_USB;
Parameters.ProgIf = PCI_INTERFACE_USB_ID_EHCI;
Parameters.Flags = PCI_USE_LOCAL_BUS |
PCI_USE_LOCAL_DEVICE |
PCI_USE_CLASS_SUBCLASS |
PCI_USE_PROGIF;
IsPresent = (PciInterface.IsDevicePresentEx)(PciInterface.Context,
&Parameters);
if (IsPresent)
{
DPRINT("USBPORT_IsCompanionController: Present EHCI controller for FDO - %p\n",
DeviceObject);
}
else
{
DPRINT("USBPORT_IsCompanionController: No EHCI controller for FDO - %p\n",
DeviceObject);
}
*IsCompanion = IsPresent;
(PciInterface.InterfaceDereference)(PciInterface.Context);
ObDereferenceObject(HighestDevice);
return Status;
}
NTSTATUS
NTAPI
USBPORT_QueryPciBusInterface(IN PDEVICE_OBJECT FdoDevice)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
PBUS_INTERFACE_STANDARD BusInterface;
PIO_STACK_LOCATION IoStack;
IO_STATUS_BLOCK IoStatusBlock;
PDEVICE_OBJECT HighestDevice;
KEVENT Event;
PIRP Irp;
NTSTATUS Status;
DPRINT("USBPORT_QueryPciBusInterface: ... \n");
FdoExtension = FdoDevice->DeviceExtension;
BusInterface = &FdoExtension->BusInterface;
RtlZeroMemory(BusInterface, sizeof(BUS_INTERFACE_STANDARD));
KeInitializeEvent(&Event, SynchronizationEvent, FALSE);
HighestDevice = IoGetAttachedDeviceReference(FdoDevice);
Irp = IoBuildSynchronousFsdRequest(IRP_MJ_PNP,
HighestDevice,
NULL,
0,
NULL,
&Event,
&IoStatusBlock);
if (Irp)
{
IoStack = IoGetNextIrpStackLocation(Irp);
Irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
Irp->IoStatus.Information = 0;
IoStack->MinorFunction = IRP_MN_QUERY_INTERFACE;
IoStack->Parameters.QueryInterface.InterfaceType = &GUID_BUS_INTERFACE_STANDARD;
IoStack->Parameters.QueryInterface.Size = sizeof(BUS_INTERFACE_STANDARD);
IoStack->Parameters.QueryInterface.Version = 1;
IoStack->Parameters.QueryInterface.Interface = (PINTERFACE)BusInterface;
IoStack->Parameters.QueryInterface.InterfaceSpecificData = 0;
Status = IoCallDriver(HighestDevice, Irp);
if (Status == STATUS_PENDING)
{
KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, NULL);
Status = IoStatusBlock.Status;
}
}
else
{
Status = STATUS_INSUFFICIENT_RESOURCES;
}
ObDereferenceObject(HighestDevice);
DPRINT("USBPORT_QueryPciBusInterface: return Status - %x\n", Status);
return Status;
}
NTSTATUS
NTAPI
USBPORT_QueryCapabilities(IN PDEVICE_OBJECT FdoDevice,
IN PDEVICE_CAPABILITIES Capabilities)
{
PUSBPORT_DEVICE_EXTENSION FdoExtention;
PIRP Irp;
NTSTATUS Status;
PIO_STACK_LOCATION IoStack;
KEVENT Event;
DPRINT("USBPORT_QueryCapabilities: ... \n");
FdoExtention = FdoDevice->DeviceExtension;
RtlZeroMemory(Capabilities, sizeof(DEVICE_CAPABILITIES));
Capabilities->Size = sizeof(DEVICE_CAPABILITIES);
Capabilities->Version = 1;
Capabilities->Address = MAXULONG;
Capabilities->UINumber = MAXULONG;
Irp = IoAllocateIrp(FdoExtention->CommonExtension.LowerDevice->StackSize,
FALSE);
if (!Irp)
{
DPRINT1("USBPORT_QueryCapabilities: No resources - IoAllocateIrp!\n");
return STATUS_INSUFFICIENT_RESOURCES;
}
IoStack = IoGetCurrentIrpStackLocation(Irp);
Irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
IoStack = IoGetNextIrpStackLocation(Irp);
IoStack->MajorFunction = IRP_MJ_PNP;
IoStack->MinorFunction = IRP_MN_QUERY_CAPABILITIES;
KeInitializeEvent(&Event, NotificationEvent, FALSE);
IoSetCompletionRoutine(Irp,
USBPORT_FdoStartCompletion,
&Event,
TRUE,
TRUE,
TRUE);
IoStack->Parameters.DeviceCapabilities.Capabilities = Capabilities;
Status = IoCallDriver(FdoExtention->CommonExtension.LowerDevice, Irp);
if (Status == STATUS_PENDING)
{
KeWaitForSingleObject(&Event, Suspended, KernelMode, FALSE, NULL);
Status = Irp->IoStatus.Status;
}
if (NT_SUCCESS(Status) && Capabilities)
{
USBPORT_DumpingCapabilities(Capabilities);
}
IoFreeIrp(Irp);
return Status;
}
NTSTATUS
NTAPI
USBPORT_CreateLegacySymbolicLink(IN PDEVICE_OBJECT FdoDevice)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
WCHAR CharName[255] = {0};
WCHAR CharDosName[255] = {0};
UNICODE_STRING DeviceName;
NTSTATUS Status;
FdoExtension = FdoDevice->DeviceExtension;
RtlStringCbPrintfW(CharName,
sizeof(CharName),
L"\\Device\\USBFDO-%d",
FdoExtension->FdoNameNumber);
RtlInitUnicodeString(&DeviceName, CharName);
RtlStringCbPrintfW(CharDosName,
sizeof(CharDosName),
L"\\DosDevices\\HCD%d",
FdoExtension->FdoNameNumber);
RtlInitUnicodeString(&FdoExtension->DosDeviceSymbolicName, CharDosName);
DPRINT("USBPORT_CreateLegacySymbolicLink: DeviceName - %wZ, DosSymbolicName - %wZ\n",
&DeviceName,
&FdoExtension->DosDeviceSymbolicName);
Status = IoCreateSymbolicLink(&FdoExtension->DosDeviceSymbolicName,
&DeviceName);
if (NT_SUCCESS(Status))
{
FdoExtension->Flags |= USBPORT_FLAG_DOS_SYMBOLIC_NAME;
}
return Status;
}
NTSTATUS
NTAPI
USBPORT_StopDevice(IN PDEVICE_OBJECT FdoDevice)
{
DPRINT1("USBPORT_StopDevice: UNIMPLEMENTED. FIXME\n");
DbgBreakPoint();
return STATUS_SUCCESS;
}
NTSTATUS
NTAPI
USBPORT_StartDevice(IN PDEVICE_OBJECT FdoDevice,
IN PUSBPORT_RESOURCES UsbPortResources)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
PUSBPORT_REGISTRATION_PACKET Packet;
NTSTATUS Status;
PCI_COMMON_CONFIG PciConfig;
ULONG BytesRead;
DEVICE_DESCRIPTION DeviceDescription;
PDMA_ADAPTER DmaAdapter = NULL;
ULONG MiniPortStatus;
PUSBPORT_COMMON_BUFFER_HEADER HeaderBuffer;
ULONG ResultLength;
ULONG DisableSelectiveSuspend = 0;
ULONG DisableCcDetect = 0;
ULONG IdleEpSupport = 0;
ULONG IdleEpSupportEx = 0;
ULONG SoftRetry = 0;
ULONG Limit2GB = 0;
ULONG TotalBusBandwidth = 0;
BOOLEAN IsCompanion = FALSE;
ULONG LegacyBIOS;
ULONG MiniportFlags;
ULONG ix;
DPRINT("USBPORT_StartDevice: FdoDevice - %p, UsbPortResources - %p\n",
FdoDevice,
UsbPortResources);
FdoExtension = FdoDevice->DeviceExtension;
Packet = &FdoExtension->MiniPortInterface->Packet;
Status = USBPORT_QueryPciBusInterface(FdoDevice);
if (!NT_SUCCESS(Status))
goto ExitWithError;
BytesRead = (*FdoExtension->BusInterface.GetBusData)(FdoExtension->BusInterface.Context,
PCI_WHICHSPACE_CONFIG,
&PciConfig,
0,
PCI_COMMON_HDR_LENGTH);
if (BytesRead != PCI_COMMON_HDR_LENGTH)
{
DPRINT1("USBPORT_StartDevice: Failed to get pci config information!\n");
goto ExitWithError;
}
FdoExtension->VendorID = PciConfig.VendorID;
FdoExtension->DeviceID = PciConfig.DeviceID;
FdoExtension->RevisionID = PciConfig.RevisionID;
FdoExtension->ProgIf = PciConfig.ProgIf;
FdoExtension->SubClass = PciConfig.SubClass;
FdoExtension->BaseClass = PciConfig.BaseClass;
RtlZeroMemory(&DeviceDescription, sizeof(DeviceDescription));
DeviceDescription.Version = DEVICE_DESCRIPTION_VERSION;
DeviceDescription.Master = TRUE;
DeviceDescription.ScatterGather = TRUE;
DeviceDescription.Dma32BitAddresses = TRUE;
DeviceDescription.InterfaceType = PCIBus;
DeviceDescription.DmaWidth = Width32Bits;
DeviceDescription.DmaSpeed = Compatible;
DeviceDescription.MaximumLength = MAXULONG;
DmaAdapter = IoGetDmaAdapter(FdoExtension->CommonExtension.LowerPdoDevice,
&DeviceDescription,
&FdoExtension->NumberMapRegs);
FdoExtension->DmaAdapter = DmaAdapter;
if (!DmaAdapter)
{
DPRINT1("USBPORT_StartDevice: Failed to get DmaAdapter!\n");
Status = STATUS_INSUFFICIENT_RESOURCES;
goto ExitWithError;
}
Status = USBPORT_CreateWorkerThread(FdoDevice);
if (!NT_SUCCESS(Status))
goto ExitWithError;
Status = USBPORT_QueryCapabilities(FdoDevice, &FdoExtension->Capabilities);
if (!NT_SUCCESS(Status))
goto ExitWithError;
FdoExtension->PciDeviceNumber = FdoExtension->Capabilities.Address >> 16;
FdoExtension->PciFunctionNumber = FdoExtension->Capabilities.Address & 0xFFFF;
Status = IoGetDeviceProperty(FdoExtension->CommonExtension.LowerPdoDevice,
DevicePropertyBusNumber,
sizeof(ULONG),
&FdoExtension->BusNumber,
&ResultLength);
if (!NT_SUCCESS(Status))
goto ExitWithError;
KeInitializeSpinLock(&FdoExtension->EndpointListSpinLock);
KeInitializeSpinLock(&FdoExtension->EpStateChangeSpinLock);
KeInitializeSpinLock(&FdoExtension->EndpointClosedSpinLock);
KeInitializeSpinLock(&FdoExtension->DeviceHandleSpinLock);
KeInitializeSpinLock(&FdoExtension->IdleIoCsqSpinLock);
KeInitializeSpinLock(&FdoExtension->BadRequestIoCsqSpinLock);
KeInitializeSpinLock(&FdoExtension->MapTransferSpinLock);
KeInitializeSpinLock(&FdoExtension->FlushTransferSpinLock);
KeInitializeSpinLock(&FdoExtension->FlushPendingTransferSpinLock);
KeInitializeSpinLock(&FdoExtension->DoneTransferSpinLock);
KeInitializeSpinLock(&FdoExtension->WorkerThreadEventSpinLock);
KeInitializeSpinLock(&FdoExtension->MiniportSpinLock);
KeInitializeSpinLock(&FdoExtension->TimerFlagsSpinLock);
KeInitializeSpinLock(&FdoExtension->PowerWakeSpinLock);
KeInitializeSpinLock(&FdoExtension->SetPowerD0SpinLock);
KeInitializeSpinLock(&FdoExtension->RootHubCallbackSpinLock);
KeInitializeSpinLock(&FdoExtension->TtSpinLock);
KeInitializeDpc(&FdoExtension->IsrDpc, USBPORT_IsrDpc, FdoDevice);
KeInitializeDpc(&FdoExtension->TransferFlushDpc,
USBPORT_TransferFlushDpc,
FdoDevice);
KeInitializeDpc(&FdoExtension->WorkerRequestDpc,
USBPORT_WorkerRequestDpc,
FdoDevice);
KeInitializeDpc(&FdoExtension->HcWakeDpc,
USBPORT_HcWakeDpc,
FdoDevice);
IoCsqInitialize(&FdoExtension->IdleIoCsq,
USBPORT_InsertIdleIrp,
USBPORT_RemoveIdleIrp,
USBPORT_PeekNextIdleIrp,
USBPORT_AcquireIdleLock,
USBPORT_ReleaseIdleLock,
USBPORT_CompleteCanceledIdleIrp);
IoCsqInitialize(&FdoExtension->BadRequestIoCsq,
USBPORT_InsertBadRequest,
USBPORT_RemoveBadRequest,
USBPORT_PeekNextBadRequest,
USBPORT_AcquireBadRequestLock,
USBPORT_ReleaseBadRequestLock,
USBPORT_CompleteCanceledBadRequest);
FdoExtension->IsrDpcCounter = -1;
FdoExtension->IsrDpcHandlerCounter = -1;
FdoExtension->IdleLockCounter = -1;
FdoExtension->BadRequestLockCounter = -1;
FdoExtension->ChirpRootPortLock = -1;
FdoExtension->RHInitCallBackLock = 0;
FdoExtension->UsbAddressBitMap[0] = 1;
FdoExtension->UsbAddressBitMap[1] = 0;
FdoExtension->UsbAddressBitMap[2] = 0;
FdoExtension->UsbAddressBitMap[3] = 0;
USBPORT_GetDefaultBIOSx(FdoDevice,
&FdoExtension->UsbBIOSx,
&DisableSelectiveSuspend,
&DisableCcDetect,
&IdleEpSupport,
&IdleEpSupportEx,
&SoftRetry);
if (DisableSelectiveSuspend)
FdoExtension->Flags |= USBPORT_FLAG_BIOS_DISABLE_SS;
if (!DisableSelectiveSuspend &&
USBPORT_IsSelectiveSuspendEnabled(FdoDevice))
{
FdoExtension->Flags |= USBPORT_FLAG_SELECTIVE_SUSPEND;
}
MiniportFlags = Packet->MiniPortFlags;
if (MiniportFlags & USB_MINIPORT_FLAGS_POLLING)
FdoExtension->Flags |= USBPORT_FLAG_HC_POLLING;
if (MiniportFlags & USB_MINIPORT_FLAGS_WAKE_SUPPORT)
FdoExtension->Flags |= USBPORT_FLAG_HC_WAKE_SUPPORT;
if (MiniportFlags & USB_MINIPORT_FLAGS_DISABLE_SS)
FdoExtension->Flags = (FdoExtension->Flags & ~USBPORT_FLAG_SELECTIVE_SUSPEND) |
USBPORT_FLAG_BIOS_DISABLE_SS;
USBPORT_SetRegistryKeyValue(FdoExtension->CommonExtension.LowerPdoDevice,
TRUE,
REG_DWORD,
L"EnIdleEndpointSupport",
&IdleEpSupport,
sizeof(IdleEpSupport));
USBPORT_SetRegistryKeyValue(FdoExtension->CommonExtension.LowerPdoDevice,
TRUE,
REG_DWORD,
L"EnIdleEndpointSupportEx",
&IdleEpSupportEx,
sizeof(IdleEpSupportEx));
USBPORT_SetRegistryKeyValue(FdoExtension->CommonExtension.LowerPdoDevice,
TRUE,
REG_DWORD,
L"EnSoftRetry",
&SoftRetry,
sizeof(SoftRetry));
USBPORT_GetRegistryKeyValueFullInfo(FdoDevice,
FdoExtension->CommonExtension.LowerPdoDevice,
TRUE,
L"CommonBuffer2GBLimit",
sizeof(L"CommonBuffer2GBLimit"),
&Limit2GB,
sizeof(Limit2GB));
FdoExtension->CommonBufferLimit = (Limit2GB != 0);
if (FdoExtension->BaseClass == PCI_CLASS_SERIAL_BUS_CTLR &&
FdoExtension->SubClass == PCI_SUBCLASS_SB_USB &&
FdoExtension->ProgIf < PCI_INTERFACE_USB_ID_EHCI)
{
Status = USBPORT_IsCompanionController(FdoDevice, &IsCompanion);
if (!NT_SUCCESS(Status))
{
if (IsCompanion)
{
FdoExtension->Flags |= USBPORT_FLAG_COMPANION_HC;
}
else
{
FdoExtension->Flags &= ~USBPORT_FLAG_COMPANION_HC;
}
}
}
if (DisableCcDetect)
{
FdoExtension->Flags &= ~USBPORT_FLAG_COMPANION_HC;
}
TotalBusBandwidth = Packet->MiniPortBusBandwidth;
FdoExtension->TotalBusBandwidth = TotalBusBandwidth;
USBPORT_GetRegistryKeyValueFullInfo(FdoDevice,
FdoExtension->CommonExtension.LowerPdoDevice,
TRUE,
L"TotalBusBandwidth",
sizeof(L"TotalBusBandwidth"),
&TotalBusBandwidth,
sizeof(TotalBusBandwidth));
if (TotalBusBandwidth != FdoExtension->TotalBusBandwidth)
{
FdoExtension->TotalBusBandwidth = TotalBusBandwidth;
}
for (ix = 0; ix < USB2_FRAMES; ix++)
{
FdoExtension->Bandwidth[ix] = FdoExtension->TotalBusBandwidth -
FdoExtension->TotalBusBandwidth / 10;
}
FdoExtension->ActiveIrpTable = ExAllocatePoolWithTag(NonPagedPool,
sizeof(USBPORT_IRP_TABLE),
USB_PORT_TAG);
if (!FdoExtension->ActiveIrpTable)
{
DPRINT1("USBPORT_StartDevice: Allocate ActiveIrpTable failed!\n");
goto ExitWithError;
}
RtlZeroMemory(FdoExtension->ActiveIrpTable, sizeof(USBPORT_IRP_TABLE));
FdoExtension->PendingIrpTable = ExAllocatePoolWithTag(NonPagedPool,
sizeof(USBPORT_IRP_TABLE),
USB_PORT_TAG);
if (!FdoExtension->PendingIrpTable)
{
DPRINT1("USBPORT_StartDevice: Allocate PendingIrpTable failed!\n");
goto ExitWithError;
}
RtlZeroMemory(FdoExtension->PendingIrpTable, sizeof(USBPORT_IRP_TABLE));
Status = IoConnectInterrupt(&FdoExtension->InterruptObject,
USBPORT_InterruptService,
(PVOID)FdoDevice,
0,
UsbPortResources->InterruptVector,
UsbPortResources->InterruptLevel,
UsbPortResources->InterruptLevel,
UsbPortResources->InterruptMode,
UsbPortResources->ShareVector,
UsbPortResources->InterruptAffinity,
0);
if (!NT_SUCCESS(Status))
{
DPRINT1("USBPORT_StartDevice: IoConnectInterrupt failed!\n");
goto ExitWithError;
}
FdoExtension->Flags &= ~USBPORT_FLAG_INT_CONNECTED;
if (Packet->MiniPortExtensionSize)
{
RtlZeroMemory(FdoExtension->MiniPortExt, Packet->MiniPortExtensionSize);
}
if (Packet->MiniPortResourcesSize)
{
HeaderBuffer = USBPORT_AllocateCommonBuffer(FdoDevice,
Packet->MiniPortResourcesSize);
if (!HeaderBuffer)
{
DPRINT1("USBPORT_StartDevice: Failed to AllocateCommonBuffer!\n");
Status = STATUS_INSUFFICIENT_RESOURCES;
goto ExitWithError;
}
UsbPortResources->StartVA = HeaderBuffer->VirtualAddress;
UsbPortResources->StartPA = HeaderBuffer->PhysicalAddress;
FdoExtension->MiniPortCommonBuffer = HeaderBuffer;
}
else
{
FdoExtension->MiniPortCommonBuffer = NULL;
}
MiniPortStatus = Packet->StartController(FdoExtension->MiniPortExt,
UsbPortResources);
if (UsbPortResources->LegacySupport)
{
FdoExtension->Flags |= USBPORT_FLAG_LEGACY_SUPPORT;
LegacyBIOS = 1;
}
else
{
LegacyBIOS = 0;
}
USBPORT_SetRegistryKeyValue(FdoExtension->CommonExtension.LowerPdoDevice,
FALSE,
REG_DWORD,
L"DetectedLegacyBIOS",
&LegacyBIOS,
sizeof(LegacyBIOS));
if (MiniPortStatus)
{
DPRINT1("USBPORT_StartDevice: Failed to Start MiniPort. MiniPortStatus - %x\n",
MiniPortStatus);
if (FdoExtension->Flags & USBPORT_FLAG_INT_CONNECTED)
{
IoDisconnectInterrupt(FdoExtension->InterruptObject);
FdoExtension->Flags &= ~USBPORT_FLAG_INT_CONNECTED;
}
if (FdoExtension->MiniPortCommonBuffer)
{
USBPORT_FreeCommonBuffer(FdoDevice, FdoExtension->MiniPortCommonBuffer);
FdoExtension->MiniPortCommonBuffer = NULL;
}
goto ExitWithError;
}
else
{
FdoExtension->MiniPortFlags |= USBPORT_MPFLAG_INTERRUPTS_ENABLED;
USBPORT_MiniportInterrupts(FdoDevice, TRUE);
}
FdoExtension->TimerValue = 500;
USBPORT_StartTimer((PVOID)FdoDevice, 500);
Status = USBPORT_RegisterDeviceInterface(FdoExtension->CommonExtension.LowerPdoDevice,
FdoDevice,
&GUID_DEVINTERFACE_USB_HOST_CONTROLLER,
TRUE);
if (!NT_SUCCESS(Status))
{
DPRINT1("USBPORT_StartDevice: RegisterDeviceInterface failed!\n");
goto ExitWithError;
}
USBPORT_CreateLegacySymbolicLink(FdoDevice);
FdoExtension->Flags |= USBPORT_FLAG_HC_STARTED;
DPRINT("USBPORT_StartDevice: Exit Status - %p\n", Status);
return Status;
ExitWithError:
USBPORT_StopDevice(FdoDevice);
DPRINT1("USBPORT_StartDevice: ExitWithError Status - %lx\n", Status);
return Status;
}
NTSTATUS
NTAPI
USBPORT_ParseResources(IN PDEVICE_OBJECT FdoDevice,
IN PIRP Irp,
IN PUSBPORT_RESOURCES UsbPortResources)
{
PCM_RESOURCE_LIST AllocatedResourcesTranslated;
PCM_PARTIAL_RESOURCE_LIST ResourceList;
PCM_PARTIAL_RESOURCE_DESCRIPTOR PartialDescriptor;
PCM_PARTIAL_RESOURCE_DESCRIPTOR PortDescriptor = NULL;
PCM_PARTIAL_RESOURCE_DESCRIPTOR MemoryDescriptor = NULL;
PCM_PARTIAL_RESOURCE_DESCRIPTOR InterruptDescriptor = NULL;
PIO_STACK_LOCATION IoStack;
ULONG ix;
NTSTATUS Status = STATUS_SUCCESS;
DPRINT("USBPORT_ParseResources: ... \n");
IoStack = IoGetCurrentIrpStackLocation(Irp);
AllocatedResourcesTranslated = IoStack->Parameters.StartDevice.AllocatedResourcesTranslated;
if (AllocatedResourcesTranslated)
{
RtlZeroMemory(UsbPortResources, sizeof(USBPORT_RESOURCES));
ResourceList = &AllocatedResourcesTranslated->List[0].PartialResourceList;
PartialDescriptor = &ResourceList->PartialDescriptors[0];
for (ix = 0; ix < ResourceList->Count; ++ix)
{
if (PartialDescriptor->Type == CmResourceTypePort)
{
if (!PortDescriptor)
PortDescriptor = PartialDescriptor;
}
else if (PartialDescriptor->Type == CmResourceTypeInterrupt)
{
if (!InterruptDescriptor)
InterruptDescriptor = PartialDescriptor;
}
else if (PartialDescriptor->Type == CmResourceTypeMemory)
{
if (!MemoryDescriptor)
MemoryDescriptor = PartialDescriptor;
}
PartialDescriptor += 1;
}
if (PortDescriptor)
{
if (PortDescriptor->Flags & CM_RESOURCE_PORT_IO)
{
2018-02-27 20:56:06 +00:00
UsbPortResources->ResourceBase = (PVOID)(ULONG_PTR)PortDescriptor->u.Port.Start.QuadPart;
}
else
{
UsbPortResources->ResourceBase = MmMapIoSpace(PortDescriptor->u.Port.Start,
PortDescriptor->u.Port.Length,
0);
}
UsbPortResources->IoSpaceLength = PortDescriptor->u.Port.Length;
if (UsbPortResources->ResourceBase)
{
UsbPortResources->ResourcesTypes |= USBPORT_RESOURCES_PORT;
}
else
{
Status = STATUS_NONE_MAPPED;
}
}
if (MemoryDescriptor && NT_SUCCESS(Status))
{
UsbPortResources->IoSpaceLength = MemoryDescriptor->u.Memory.Length;
UsbPortResources->ResourceBase = MmMapIoSpace(MemoryDescriptor->u.Memory.Start,
MemoryDescriptor->u.Memory.Length,
0);
if (UsbPortResources->ResourceBase)
{
UsbPortResources->ResourcesTypes |= USBPORT_RESOURCES_MEMORY;
}
else
{
Status = STATUS_NONE_MAPPED;
}
}
if (InterruptDescriptor && NT_SUCCESS(Status))
{
UsbPortResources->ResourcesTypes |= USBPORT_RESOURCES_INTERRUPT;
UsbPortResources->InterruptVector = InterruptDescriptor->u.Interrupt.Vector;
UsbPortResources->InterruptLevel = InterruptDescriptor->u.Interrupt.Level;
UsbPortResources->InterruptAffinity = InterruptDescriptor->u.Interrupt.Affinity;
UsbPortResources->ShareVector = InterruptDescriptor->ShareDisposition ==
CmResourceShareShared;
UsbPortResources->InterruptMode = InterruptDescriptor->Flags ==
CM_RESOURCE_INTERRUPT_LATCHED;
}
}
else
{
Status = STATUS_NONE_MAPPED;
}
return Status;
}
NTSTATUS
NTAPI
USBPORT_CreatePdo(IN PDEVICE_OBJECT FdoDevice,
OUT PDEVICE_OBJECT *RootHubPdo)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
PUSBPORT_RHDEVICE_EXTENSION PdoExtension;
UNICODE_STRING DeviceName;
ULONG DeviceNumber = 0;
PDEVICE_OBJECT DeviceObject = NULL;
WCHAR CharDeviceName[64];
NTSTATUS Status = STATUS_SUCCESS;
DPRINT("USBPORT_CreatePdo: FdoDevice - %p, RootHubPdo - %p\n",
FdoDevice,
RootHubPdo);
FdoExtension = FdoDevice->DeviceExtension;
do
{
RtlStringCbPrintfW(CharDeviceName,
sizeof(CharDeviceName),
L"\\Device\\USBPDO-%d",
DeviceNumber);
RtlInitUnicodeString(&DeviceName, CharDeviceName);
DPRINT("USBPORT_CreatePdo: DeviceName - %wZ\n", &DeviceName);
Status = IoCreateDevice(FdoExtension->MiniPortInterface->DriverObject,
sizeof(USBPORT_RHDEVICE_EXTENSION),
&DeviceName,
FILE_DEVICE_BUS_EXTENDER,
0,
FALSE,
&DeviceObject);
++DeviceNumber;
}
while (Status == STATUS_OBJECT_NAME_COLLISION);
if (!NT_SUCCESS(Status))
{
*RootHubPdo = NULL;
DPRINT1("USBPORT_CreatePdo: Filed create HubPdo!\n");
return Status;
}
if (DeviceObject)
{
PdoExtension = DeviceObject->DeviceExtension;
RtlZeroMemory(PdoExtension, sizeof(USBPORT_RHDEVICE_EXTENSION));
PdoExtension->CommonExtension.SelfDevice = DeviceObject;
PdoExtension->CommonExtension.IsPDO = TRUE;
PdoExtension->FdoDevice = FdoDevice;
PdoExtension->PdoNameNumber = DeviceNumber;
USBPORT_AdjustDeviceCapabilities(FdoDevice, DeviceObject);
DeviceObject->StackSize = FdoDevice->StackSize;
DeviceObject->Flags |= DO_POWER_PAGABLE;
DeviceObject->Flags &= ~DO_DEVICE_INITIALIZING;
}
else
{
Status = STATUS_UNSUCCESSFUL;
}
if (!NT_SUCCESS(Status))
*RootHubPdo = NULL;
else
*RootHubPdo = DeviceObject;
DPRINT("USBPORT_CreatePdo: HubPdo - %p\n", DeviceObject);
return Status;
}
NTSTATUS
NTAPI
USBPORT_FdoPnP(IN PDEVICE_OBJECT FdoDevice,
IN PIRP Irp)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
PUSBPORT_COMMON_DEVICE_EXTENSION FdoCommonExtension;
PUSBPORT_REGISTRATION_PACKET Packet;
PUSBPORT_RESOURCES UsbPortResources;
PIO_STACK_LOCATION IoStack;
UCHAR Minor;
KEVENT Event;
NTSTATUS Status;
DEVICE_RELATION_TYPE RelationType;
PDEVICE_RELATIONS DeviceRelations;
PDEVICE_OBJECT RootHubPdo;
FdoExtension = FdoDevice->DeviceExtension;
FdoCommonExtension = &FdoExtension->CommonExtension;
UsbPortResources = &FdoExtension->UsbPortResources;
Packet = &FdoExtension->MiniPortInterface->Packet;
IoStack = IoGetCurrentIrpStackLocation(Irp);
Minor = IoStack->MinorFunction;
DPRINT("USBPORT_FdoPnP: FdoDevice - %p, Minor - %x\n", FdoDevice, Minor);
RelationType = IoStack->Parameters.QueryDeviceRelations.Type;
switch (Minor)
{
case IRP_MN_START_DEVICE:
DPRINT("IRP_MN_START_DEVICE\n");
KeInitializeEvent(&Event, NotificationEvent, FALSE);
IoCopyCurrentIrpStackLocationToNext(Irp);
IoSetCompletionRoutine(Irp,
USBPORT_FdoStartCompletion,
&Event,
TRUE,
TRUE,
TRUE);
Status = IoCallDriver(FdoCommonExtension->LowerDevice,
Irp);
if (Status == STATUS_PENDING)
{
KeWaitForSingleObject(&Event,
Suspended,
KernelMode,
FALSE,
NULL);
Status = Irp->IoStatus.Status;
}
if (!NT_SUCCESS(Status))
{
goto Exit;
}
Status = USBPORT_ParseResources(FdoDevice,
Irp,
UsbPortResources);
if (!NT_SUCCESS(Status))
{
FdoCommonExtension->PnpStateFlags |= USBPORT_PNP_STATE_STOPPED;
goto Exit;
}
Status = USBPORT_StartDevice(FdoDevice, UsbPortResources);
if (!NT_SUCCESS(Status))
{
FdoCommonExtension->PnpStateFlags |= USBPORT_PNP_STATE_STOPPED;
goto Exit;
}
FdoCommonExtension->PnpStateFlags &= ~USBPORT_PNP_STATE_NOT_INIT;
FdoCommonExtension->PnpStateFlags |= USBPORT_PNP_STATE_STARTED;
FdoCommonExtension->DevicePowerState = PowerDeviceD0;
if (Packet->MiniPortFlags & USB_MINIPORT_FLAGS_USB2)
{
USBPORT_AddUSB2Fdo(FdoDevice);
}
else
{
USBPORT_AddUSB1Fdo(FdoDevice);
}
Exit:
Irp->IoStatus.Status = Status;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return Status;
case IRP_MN_QUERY_REMOVE_DEVICE:
DPRINT("IRP_MN_QUERY_REMOVE_DEVICE\n");
if (Packet->MiniPortFlags & USB_MINIPORT_FLAGS_USB2)
{
DPRINT1("USBPORT_FdoPnP: Haction registry write FIXME\n");
}
Irp->IoStatus.Status = STATUS_SUCCESS;
goto ForwardIrp;
case IRP_MN_REMOVE_DEVICE:
DPRINT("USBPORT_FdoPnP: IRP_MN_REMOVE_DEVICE\n");
FdoCommonExtension->PnpStateFlags |= USBPORT_PNP_STATE_FAILED;
if (FdoCommonExtension->PnpStateFlags & USBPORT_PNP_STATE_STARTED &&
!(FdoCommonExtension->PnpStateFlags & USBPORT_PNP_STATE_NOT_INIT))
{
DPRINT1("USBPORT_FdoPnP: stop fdo FIXME\n");
FdoCommonExtension->PnpStateFlags |= USBPORT_PNP_STATE_NOT_INIT;
}
Irp->IoStatus.Status = STATUS_SUCCESS;
IoSkipCurrentIrpStackLocation(Irp);
Status = IoCallDriver(FdoCommonExtension->LowerDevice, Irp);
IoDetachDevice(FdoCommonExtension->LowerDevice);
RootHubPdo = FdoExtension->RootHubPdo;
IoDeleteDevice(FdoDevice);
if (RootHubPdo)
{
IoDeleteDevice(RootHubPdo);
}
return Status;
case IRP_MN_CANCEL_REMOVE_DEVICE:
DPRINT("IRP_MN_CANCEL_REMOVE_DEVICE\n");
Irp->IoStatus.Status = STATUS_SUCCESS;
goto ForwardIrp;
case IRP_MN_STOP_DEVICE:
DPRINT("IRP_MN_STOP_DEVICE\n");
if (FdoCommonExtension->PnpStateFlags & USBPORT_PNP_STATE_STARTED)
{
DPRINT1("USBPORT_FdoPnP: stop fdo FIXME\n");
FdoCommonExtension->PnpStateFlags &= ~USBPORT_PNP_STATE_STARTED;
FdoCommonExtension->PnpStateFlags |= USBPORT_PNP_STATE_NOT_INIT;
}
Irp->IoStatus.Status = STATUS_SUCCESS;
goto ForwardIrp;
case IRP_MN_QUERY_STOP_DEVICE:
DPRINT("IRP_MN_QUERY_STOP_DEVICE\n");
Irp->IoStatus.Status = STATUS_SUCCESS;
goto ForwardIrp;
case IRP_MN_CANCEL_STOP_DEVICE:
DPRINT("IRP_MN_CANCEL_STOP_DEVICE\n");
Irp->IoStatus.Status = STATUS_SUCCESS;
goto ForwardIrp;
case IRP_MN_QUERY_DEVICE_RELATIONS:
DPRINT("IRP_MN_QUERY_DEVICE_RELATIONS\n");
if (RelationType == BusRelations)
{
DeviceRelations = ExAllocatePoolWithTag(PagedPool,
sizeof(DEVICE_RELATIONS),
USB_PORT_TAG);
if (!DeviceRelations)
{
Status = STATUS_INSUFFICIENT_RESOURCES;
Irp->IoStatus.Status = Status;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return Status;
}
DeviceRelations->Count = 0;
DeviceRelations->Objects[0] = NULL;
if (!FdoExtension->RootHubPdo)
{
Status = USBPORT_CreatePdo(FdoDevice,
&FdoExtension->RootHubPdo);
if (!NT_SUCCESS(Status))
{
ExFreePoolWithTag(DeviceRelations, USB_PORT_TAG);
goto ForwardIrp;
}
}
else
{
Status = STATUS_SUCCESS;
}
DeviceRelations->Count = 1;
DeviceRelations->Objects[0] = FdoExtension->RootHubPdo;
ObReferenceObject(FdoExtension->RootHubPdo);
Irp->IoStatus.Information = (ULONG_PTR)DeviceRelations;
}
else
{
if (RelationType == RemovalRelations)
{
DPRINT1("USBPORT_FdoPnP: FIXME IRP_MN_QUERY_DEVICE_RELATIONS/RemovalRelations\n");
}
goto ForwardIrp;
}
Irp->IoStatus.Status = Status;
goto ForwardIrp;
case IRP_MN_QUERY_INTERFACE:
DPRINT("IRP_MN_QUERY_INTERFACE\n");
goto ForwardIrp;
case IRP_MN_QUERY_CAPABILITIES:
DPRINT("IRP_MN_QUERY_CAPABILITIES\n");
goto ForwardIrp;
case IRP_MN_QUERY_RESOURCES:
DPRINT("IRP_MN_QUERY_RESOURCES\n");
goto ForwardIrp;
case IRP_MN_QUERY_RESOURCE_REQUIREMENTS:
DPRINT("IRP_MN_QUERY_RESOURCE_REQUIREMENTS\n");
goto ForwardIrp;
case IRP_MN_QUERY_DEVICE_TEXT:
DPRINT("IRP_MN_QUERY_DEVICE_TEXT\n");
goto ForwardIrp;
case IRP_MN_FILTER_RESOURCE_REQUIREMENTS:
DPRINT("IRP_MN_FILTER_RESOURCE_REQUIREMENTS\n");
goto ForwardIrp;
case IRP_MN_READ_CONFIG:
DPRINT("IRP_MN_READ_CONFIG\n");
goto ForwardIrp;
case IRP_MN_WRITE_CONFIG:
DPRINT("IRP_MN_WRITE_CONFIG\n");
goto ForwardIrp;
case IRP_MN_EJECT:
DPRINT("IRP_MN_EJECT\n");
goto ForwardIrp;
case IRP_MN_SET_LOCK:
DPRINT("IRP_MN_SET_LOCK\n");
goto ForwardIrp;
case IRP_MN_QUERY_ID:
DPRINT("IRP_MN_QUERY_ID\n");
goto ForwardIrp;
case IRP_MN_QUERY_PNP_DEVICE_STATE:
DPRINT("IRP_MN_QUERY_PNP_DEVICE_STATE\n");
goto ForwardIrp;
case IRP_MN_QUERY_BUS_INFORMATION:
DPRINT("IRP_MN_QUERY_BUS_INFORMATION\n");
goto ForwardIrp;
case IRP_MN_DEVICE_USAGE_NOTIFICATION:
DPRINT("IRP_MN_DEVICE_USAGE_NOTIFICATION\n");
goto ForwardIrp;
case IRP_MN_SURPRISE_REMOVAL:
DPRINT1("IRP_MN_SURPRISE_REMOVAL\n");
if (!(FdoCommonExtension->PnpStateFlags & USBPORT_PNP_STATE_FAILED))
{
USBPORT_InvalidateControllerHandler(FdoDevice,
USBPORT_INVALIDATE_CONTROLLER_SURPRISE_REMOVE);
}
goto ForwardIrp;
default:
DPRINT("unknown IRP_MN_???\n");
ForwardIrp:
/* forward irp to next device object */
IoSkipCurrentIrpStackLocation(Irp);
break;
}
return IoCallDriver(FdoCommonExtension->LowerDevice, Irp);
}
PVOID
NTAPI
USBPORT_GetDeviceHwIds(IN PDEVICE_OBJECT FdoDevice,
IN USHORT VendorID,
IN USHORT DeviceID,
IN USHORT RevisionID)
{
PUSBPORT_DEVICE_EXTENSION FdoExtension;
PUSBPORT_REGISTRATION_PACKET Packet;
PVOID Id;
WCHAR Buffer[300] = {0};
2018-02-27 20:56:06 +00:00
SIZE_T Length = 0;
size_t Remaining = sizeof(Buffer);
PWCHAR EndBuffer;
FdoExtension = FdoDevice->DeviceExtension;
Packet = &FdoExtension->MiniPortInterface->Packet;
DPRINT("USBPORT_GetDeviceHwIds: FdoDevice - %p, Packet->MiniPortFlags - %p\n",
FdoDevice,
Packet->MiniPortFlags);
if (Packet->MiniPortFlags & USB_MINIPORT_FLAGS_USB2)
{
RtlStringCbPrintfExW(Buffer,
Remaining,
&EndBuffer,
&Remaining,
0,
L"USB\\ROOT_HUB20&VID%04x&PID%04x&REV%04x",
VendorID,
DeviceID,
RevisionID);
EndBuffer++;
Remaining -= sizeof(UNICODE_NULL);
RtlStringCbPrintfExW(EndBuffer,
Remaining,
&EndBuffer,
&Remaining,
0,
L"USB\\ROOT_HUB20&VID%04x&PID%04x",
VendorID,
DeviceID);
EndBuffer++;
Remaining -= sizeof(UNICODE_NULL);
RtlStringCbPrintfExW(EndBuffer,
Remaining,
NULL,
&Remaining,
0,
L"USB\\ROOT_HUB20");
}
else
{
RtlStringCbPrintfExW(Buffer,
Remaining,
&EndBuffer,
&Remaining,
0,
L"USB\\ROOT_HUB&VID%04x&PID%04x&REV%04x",
VendorID,
DeviceID,
RevisionID);
EndBuffer++;
Remaining -= sizeof(UNICODE_NULL);
RtlStringCbPrintfExW(EndBuffer,
Remaining,
&EndBuffer,
&Remaining,
0,
L"USB\\ROOT_HUB&VID%04x&PID%04x",
VendorID,
DeviceID);
EndBuffer++;
Remaining -= sizeof(UNICODE_NULL);
RtlStringCbPrintfExW(EndBuffer,
Remaining,
NULL,
&Remaining,
0,
L"USB\\ROOT_HUB");
}
Length = (sizeof(Buffer) - Remaining + 2 * sizeof(UNICODE_NULL));
/* for debug only */
if (FALSE)
{
DPRINT("Hardware IDs:\n");
USBPORT_DumpingIDs(Buffer);
}
Id = ExAllocatePoolWithTag(PagedPool, Length, USB_PORT_TAG);
if (!Id)
return NULL;
RtlMoveMemory(Id, Buffer, Length);
return Id;
}
NTSTATUS
NTAPI
USBPORT_PdoPnP(IN PDEVICE_OBJECT PdoDevice,
IN PIRP Irp)
{
PUSBPORT_RHDEVICE_EXTENSION PdoExtension;
PUSBPORT_COMMON_DEVICE_EXTENSION PdoCommonExtension;
PDEVICE_OBJECT FdoDevice;
PUSBPORT_DEVICE_EXTENSION FdoExtension;
PIO_STACK_LOCATION IoStack;
UCHAR Minor;
NTSTATUS Status;
PPNP_BUS_INFORMATION BusInformation;
PDEVICE_CAPABILITIES DeviceCapabilities;
PdoExtension = PdoDevice->DeviceExtension;
PdoCommonExtension = &PdoExtension->CommonExtension;
FdoDevice = PdoExtension->FdoDevice;
FdoExtension = FdoDevice->DeviceExtension;
IoStack = IoGetCurrentIrpStackLocation(Irp);
Minor = IoStack->MinorFunction;
Status = Irp->IoStatus.Status;
DPRINT("USBPORT_PdoPnP: PdoDevice - %p, Minor - %x\n", PdoDevice, Minor);
switch (Minor)
{
case IRP_MN_START_DEVICE:
DPRINT("IRP_MN_START_DEVICE\n");
Status = USBPORT_RootHubCreateDevice(FdoDevice, PdoDevice);
if (NT_SUCCESS(Status))
{
Status = USBPORT_RegisterDeviceInterface(PdoDevice,
PdoDevice,
&GUID_DEVINTERFACE_USB_HUB,
TRUE);
if (NT_SUCCESS(Status))
{
PdoCommonExtension->DevicePowerState = PowerDeviceD0;
PdoCommonExtension->PnpStateFlags = USBPORT_PNP_STATE_STARTED;
}
}
break;
case IRP_MN_QUERY_REMOVE_DEVICE:
DPRINT("USBPORT_PdoPnP: IRP_MN_QUERY_REMOVE_DEVICE\n");
Status = STATUS_SUCCESS;
break;
case IRP_MN_REMOVE_DEVICE:
DPRINT1("USBPORT_PdoPnP: IRP_MN_REMOVE_DEVICE UNIMPLEMENTED. FIXME. \n");
//USBPORT_StopRootHub();
Status = STATUS_SUCCESS;
break;
case IRP_MN_CANCEL_REMOVE_DEVICE:
DPRINT("IRP_MN_CANCEL_REMOVE_DEVICE\n");
Status = STATUS_SUCCESS;
break;
case IRP_MN_STOP_DEVICE:
DPRINT1("USBPORT_PdoPnP: IRP_MN_STOP_DEVICE UNIMPLEMENTED. FIXME. \n");
//USBPORT_StopRootHub();
Status = STATUS_SUCCESS;
break;
case IRP_MN_QUERY_STOP_DEVICE:
DPRINT("IRP_MN_QUERY_STOP_DEVICE\n");
Status = STATUS_SUCCESS;
break;
case IRP_MN_CANCEL_STOP_DEVICE:
DPRINT("IRP_MN_CANCEL_STOP_DEVICE\n");
Status = STATUS_SUCCESS;
break;
case IRP_MN_QUERY_DEVICE_RELATIONS:
{
PDEVICE_RELATIONS DeviceRelations;
DPRINT("IRP_MN_QUERY_DEVICE_RELATIONS\n");
if (IoStack->Parameters.QueryDeviceRelations.Type != TargetDeviceRelation)
{
break;
}
DeviceRelations = ExAllocatePoolWithTag(PagedPool,
sizeof(DEVICE_RELATIONS),
USB_PORT_TAG);
if (!DeviceRelations)
{
Status = STATUS_INSUFFICIENT_RESOURCES;
Irp->IoStatus.Information = 0;
break;
}
DeviceRelations->Count = 1;
DeviceRelations->Objects[0] = PdoDevice;
ObReferenceObject(PdoDevice);
Status = STATUS_SUCCESS;
Irp->IoStatus.Information = (ULONG_PTR)DeviceRelations;
break;
}
case IRP_MN_QUERY_INTERFACE:
DPRINT("IRP_MN_QUERY_INTERFACE\n");
Status = USBPORT_PdoQueryInterface(FdoDevice, PdoDevice, Irp);
break;
case IRP_MN_QUERY_CAPABILITIES:
DPRINT("IRP_MN_QUERY_CAPABILITIES\n");
DeviceCapabilities = IoStack->Parameters.DeviceCapabilities.Capabilities;
RtlCopyMemory(DeviceCapabilities,
&PdoExtension->Capabilities,
sizeof(DEVICE_CAPABILITIES));
Status = STATUS_SUCCESS;
break;
case IRP_MN_QUERY_RESOURCES:
DPRINT("USBPORT_PdoPnP: IRP_MN_QUERY_RESOURCES\n");
break;
case IRP_MN_QUERY_RESOURCE_REQUIREMENTS:
DPRINT("IRP_MN_QUERY_RESOURCE_REQUIREMENTS\n");
break;
case IRP_MN_QUERY_DEVICE_TEXT:
DPRINT("IRP_MN_QUERY_DEVICE_TEXT\n");
break;
case IRP_MN_FILTER_RESOURCE_REQUIREMENTS:
DPRINT("IRP_MN_FILTER_RESOURCE_REQUIREMENTS\n");
break;
case IRP_MN_READ_CONFIG:
DPRINT("IRP_MN_READ_CONFIG\n");
ASSERT(FALSE);
break;
case IRP_MN_WRITE_CONFIG:
DPRINT("IRP_MN_WRITE_CONFIG\n");
ASSERT(FALSE);
break;
case IRP_MN_EJECT:
DPRINT("IRP_MN_EJECT\n");
ASSERT(FALSE);
break;
case IRP_MN_SET_LOCK:
DPRINT("IRP_MN_SET_LOCK\n");
ASSERT(FALSE);
break;
case IRP_MN_QUERY_ID:
{
ULONG IdType;
LONG Length;
WCHAR Buffer[64] = {0};
PVOID Id;
Status = STATUS_SUCCESS;
IdType = IoStack->Parameters.QueryId.IdType;
DPRINT("IRP_MN_QUERY_ID/Type %x\n", IdType);
if (IdType == BusQueryDeviceID)
{
PUSBPORT_REGISTRATION_PACKET Packet;
Packet = &FdoExtension->MiniPortInterface->Packet;
if (Packet->MiniPortFlags & USB_MINIPORT_FLAGS_USB2)
{
RtlStringCbPrintfW(Buffer,
sizeof(Buffer),
L"USB\\ROOT_HUB20");
}
else
{
RtlStringCbPrintfW(Buffer,
sizeof(Buffer),
L"USB\\ROOT_HUB");
}
2018-08-16 12:10:35 +00:00
Length = (LONG)(wcslen(Buffer) + 1);
Id = ExAllocatePoolWithTag(PagedPool,
Length * sizeof(WCHAR),
USB_PORT_TAG);
if (Id)
{
RtlZeroMemory(Id, Length * sizeof(WCHAR));
RtlStringCbCopyW(Id, Length * sizeof(WCHAR), Buffer);
DPRINT("BusQueryDeviceID - %S, TotalLength - %hu\n",
Id,
Length);
}
Irp->IoStatus.Information = (ULONG_PTR)Id;
break;
}
if (IdType == BusQueryHardwareIDs)
{
Id = USBPORT_GetDeviceHwIds(FdoDevice,
FdoExtension->VendorID,
FdoExtension->DeviceID,
FdoExtension->RevisionID);
Irp->IoStatus.Information = (ULONG_PTR)Id;
break;
}
if (IdType == BusQueryCompatibleIDs ||
IdType == BusQueryInstanceID)
{
Irp->IoStatus.Information = 0;
break;
}
/* IdType == BusQueryDeviceSerialNumber */
Status = Irp->IoStatus.Status;
break;
}
case IRP_MN_QUERY_PNP_DEVICE_STATE:
DPRINT("IRP_MN_QUERY_PNP_DEVICE_STATE\n");
Status = STATUS_SUCCESS;
break;
case IRP_MN_QUERY_BUS_INFORMATION:
DPRINT("IRP_MN_QUERY_BUS_INFORMATION\n");
/* Allocate buffer for bus information */
BusInformation = ExAllocatePoolWithTag(PagedPool,
sizeof(PNP_BUS_INFORMATION),
USB_PORT_TAG);
if (!BusInformation)
{
/* No memory */
Status = STATUS_INSUFFICIENT_RESOURCES;
break;
}
RtlZeroMemory(BusInformation, sizeof(PNP_BUS_INFORMATION));
/* Copy BUS GUID */
RtlMoveMemory(&BusInformation->BusTypeGuid,
&GUID_BUS_TYPE_USB,
sizeof(GUID));
/* Set bus type */
BusInformation->LegacyBusType = PNPBus;
BusInformation->BusNumber = 0;
Status = STATUS_SUCCESS;
Irp->IoStatus.Information = (ULONG_PTR)BusInformation;
break;
case IRP_MN_DEVICE_USAGE_NOTIFICATION:
DPRINT("IRP_MN_DEVICE_USAGE_NOTIFICATION\n");
break;
case IRP_MN_SURPRISE_REMOVAL:
DPRINT("USBPORT_PdoPnP: IRP_MN_SURPRISE_REMOVAL\n");
Status = STATUS_SUCCESS;
break;
default:
DPRINT("unknown IRP_MN_???\n");
break;
}
Irp->IoStatus.Status = Status;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return Status;
}