mirror of https://github.com/tongzx/nt5src
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.
484 lines
16 KiB
484 lines
16 KiB
/*
|
|
*************************************************************************
|
|
* File: PNP.C
|
|
*
|
|
* Module: USBCCGP.SYS
|
|
* USB Common Class Generic Parent driver.
|
|
*
|
|
* Copyright (c) 1998 Microsoft Corporation
|
|
*
|
|
*
|
|
* Author: ervinp
|
|
*
|
|
*************************************************************************
|
|
*/
|
|
|
|
|
|
#include <wdm.h>
|
|
#include <stdio.h>
|
|
#include <usbdi.h>
|
|
#include <usbdlib.h>
|
|
#include <usbioctl.h>
|
|
#include <wdmguid.h>
|
|
|
|
#include "usbccgp.h"
|
|
#include "debug.h"
|
|
|
|
|
|
#ifdef ALLOC_PRAGMA
|
|
#pragma alloc_text(PAGE, USBC_PnP)
|
|
#pragma alloc_text(PAGE, GetDeviceText)
|
|
|
|
#endif
|
|
|
|
|
|
NTSTATUS USBC_PnP(PDEVEXT devExt, PIRP irp)
|
|
{
|
|
PIO_STACK_LOCATION irpSp;
|
|
NTSTATUS status;
|
|
UCHAR minorFunction;
|
|
BOOLEAN isParentFdo;
|
|
enum deviceState oldState;
|
|
|
|
PAGED_CODE();
|
|
|
|
irpSp = IoGetCurrentIrpStackLocation(irp);
|
|
|
|
/*
|
|
* Keep these privately so we still have it after the IRP completes
|
|
* or after the device extension is freed on a REMOVE_DEVICE
|
|
*/
|
|
minorFunction = irpSp->MinorFunction;
|
|
isParentFdo = devExt->isParentFdo;
|
|
|
|
DBG_LOG_PNP_IRP(irp, minorFunction, isParentFdo, FALSE, 0);
|
|
|
|
if (isParentFdo){
|
|
PPARENT_FDO_EXT parentFdoExt = &devExt->parentFdoExt;
|
|
BOOLEAN irpAlreadyCompleted = FALSE;
|
|
status = NO_STATUS;
|
|
|
|
if (parentFdoExt->state == STATE_SUSPENDED ||
|
|
parentFdoExt->pendingIdleIrp) {
|
|
|
|
ParentSetD0(parentFdoExt);
|
|
}
|
|
|
|
switch (minorFunction){
|
|
|
|
case IRP_MN_START_DEVICE:
|
|
status = StartParentFdo(parentFdoExt, irp);
|
|
break;
|
|
|
|
case IRP_MN_QUERY_STOP_DEVICE:
|
|
if (parentFdoExt->state == STATE_SUSPENDED){
|
|
status = STATUS_DEVICE_POWER_FAILURE;
|
|
}
|
|
else {
|
|
/*
|
|
* We will pass this IRP down the driver stack.
|
|
* However, we need to change the default status
|
|
* from STATUS_NOT_SUPPORTED to STATUS_SUCCESS.
|
|
*/
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
}
|
|
break;
|
|
|
|
case IRP_MN_STOP_DEVICE:
|
|
if (parentFdoExt->state == STATE_SUSPENDED){
|
|
DBGERR(("Got STOP while device suspended"));
|
|
status = STATUS_DEVICE_POWER_FAILURE;
|
|
}
|
|
else {
|
|
/*
|
|
* Only set state to STOPPED if the device was
|
|
* previously started successfully.
|
|
*/
|
|
if (parentFdoExt->state == STATE_STARTED){
|
|
parentFdoExt->state = STATE_STOPPING;
|
|
ParentCloseConfiguration(parentFdoExt);
|
|
}
|
|
else {
|
|
DBGWARN(("Got STOP while in state is %xh.", parentFdoExt->state));
|
|
}
|
|
}
|
|
break;
|
|
|
|
case IRP_MN_QUERY_REMOVE_DEVICE:
|
|
/*
|
|
* We will pass this IRP down the driver stack.
|
|
* However, we need to change the default status
|
|
* from STATUS_NOT_SUPPORTED to STATUS_SUCCESS.
|
|
*/
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_REMOVE_DEVICE:
|
|
|
|
/*
|
|
* Cancel downward IO
|
|
* Set default status to SUCCESS
|
|
* Send the IRP down
|
|
* Detach
|
|
* Cleanup
|
|
*/
|
|
PrepareParentFDOForRemove(parentFdoExt);
|
|
|
|
irp->IoStatus.Status = STATUS_SUCCESS;
|
|
IoSkipCurrentIrpStackLocation(irp);
|
|
status = IoCallDriver(parentFdoExt->topDevObj, irp);
|
|
irpAlreadyCompleted = TRUE;
|
|
|
|
IoDetachDevice(parentFdoExt->topDevObj);
|
|
|
|
FreeParentFDOResources(parentFdoExt);
|
|
|
|
break;
|
|
|
|
case IRP_MN_QUERY_DEVICE_RELATIONS:
|
|
status = QueryParentDeviceRelations(parentFdoExt, irp);
|
|
break;
|
|
|
|
case IRP_MN_QUERY_CAPABILITIES:
|
|
/*
|
|
* Return the USB PDO's capabilities, but add the SurpriseRemovalOK bit.
|
|
*/
|
|
ASSERT(irpSp->Parameters.DeviceCapabilities.Capabilities);
|
|
IoCopyCurrentIrpStackLocationToNext(irp);
|
|
status = CallDriverSync(parentFdoExt->topDevObj, irp);
|
|
if (NT_SUCCESS(status)){
|
|
irpSp->Parameters.DeviceCapabilities.Capabilities->SurpriseRemovalOK = TRUE;
|
|
}
|
|
break;
|
|
|
|
}
|
|
|
|
if (status == NO_STATUS){
|
|
IoCopyCurrentIrpStackLocationToNext(irp);
|
|
IoSetCompletionRoutine(irp, USBC_PnpComplete, (PVOID)devExt, TRUE, TRUE, TRUE);
|
|
status = IoCallDriver(parentFdoExt->topDevObj, irp);
|
|
}
|
|
else if (irpAlreadyCompleted){
|
|
/*
|
|
* Don't touch the irp.
|
|
*/
|
|
}
|
|
else if (status != STATUS_PENDING){
|
|
/*
|
|
* Complete the IRP here
|
|
*/
|
|
irp->IoStatus.Status = status;
|
|
IoCompleteRequest(irp, IO_NO_INCREMENT);
|
|
}
|
|
}
|
|
else {
|
|
/*
|
|
* This is a child function PDO.
|
|
*/
|
|
PFUNCTION_PDO_EXT functionPdoExt = &devExt->functionPdoExt;
|
|
|
|
switch (minorFunction){
|
|
|
|
case IRP_MN_START_DEVICE:
|
|
functionPdoExt->state = STATE_STARTED;
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_QUERY_STOP_DEVICE:
|
|
case IRP_MN_CANCEL_STOP_DEVICE:
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_STOP_DEVICE:
|
|
/*
|
|
* Only set state to STOPPED if the device was
|
|
* previously started successfully.
|
|
*/
|
|
if (functionPdoExt->state == STATE_STARTED){
|
|
functionPdoExt->state = STATE_STOPPED;
|
|
}
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_QUERY_REMOVE_DEVICE:
|
|
case IRP_MN_CANCEL_REMOVE_DEVICE:
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_REMOVE_DEVICE:
|
|
/*
|
|
* Since we are the bus driver for the function-PDOs, we cannot
|
|
* delete the function pdo on a remove-device. We must wait
|
|
* for the parent to get the remove before deleting the function pdo.
|
|
*/
|
|
oldState = functionPdoExt->state;
|
|
functionPdoExt->state = STATE_REMOVED;
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_QUERY_ID:
|
|
status = QueryFunctionPdoID(functionPdoExt, irp);
|
|
break;
|
|
|
|
case IRP_MN_QUERY_DEVICE_RELATIONS:
|
|
status = QueryFunctionDeviceRelations(functionPdoExt, irp);
|
|
break;
|
|
|
|
case IRP_MN_QUERY_CAPABILITIES:
|
|
status = QueryFunctionCapabilities(functionPdoExt, irp);
|
|
break;
|
|
|
|
case IRP_MN_QUERY_PNP_DEVICE_STATE:
|
|
irp->IoStatus.Information = 0;
|
|
switch (functionPdoExt->state){
|
|
case STATE_START_FAILED:
|
|
irp->IoStatus.Information |= PNP_DEVICE_FAILED;
|
|
break;
|
|
case STATE_STOPPED:
|
|
irp->IoStatus.Information |= PNP_DEVICE_DISABLED;
|
|
break;
|
|
case STATE_REMOVING:
|
|
case STATE_REMOVED:
|
|
irp->IoStatus.Information |= PNP_DEVICE_REMOVED;
|
|
break;
|
|
}
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_QUERY_RESOURCE_REQUIREMENTS:
|
|
// Adrian says that once PnP sends this IRP, the PDO is valid for
|
|
// PnP functions like IoGetDeviceProperty, etc.
|
|
//
|
|
// And since we know that the PDO is valid and the DevNode now exists,
|
|
// this would also be a good time to handle the MS ExtPropDesc.
|
|
//
|
|
InstallExtPropDesc(functionPdoExt);
|
|
|
|
status = STATUS_SUCCESS;
|
|
break;
|
|
|
|
case IRP_MN_QUERY_BUS_INFORMATION:
|
|
{
|
|
PPNP_BUS_INFORMATION busInfo = ALLOCPOOL(PagedPool, sizeof(PNP_BUS_INFORMATION));
|
|
if (busInfo) {
|
|
busInfo->BusTypeGuid = GUID_BUS_TYPE_USB;
|
|
busInfo->LegacyBusType = PNPBus;
|
|
busInfo->BusNumber = 0;
|
|
irp->IoStatus.Information = (ULONG_PTR)busInfo;
|
|
status = STATUS_SUCCESS;
|
|
}
|
|
else {
|
|
status = STATUS_INSUFFICIENT_RESOURCES;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case IRP_MN_QUERY_DEVICE_TEXT:
|
|
status = GetDeviceText(functionPdoExt, irp);
|
|
break;
|
|
|
|
case IRP_MN_QUERY_INTERFACE:
|
|
/*
|
|
* Send this down to the parent.
|
|
*/
|
|
IoCopyCurrentIrpStackLocationToNext(irp);
|
|
status = CallDriverSync(functionPdoExt->parentFdoExt->fdo, irp);
|
|
break;
|
|
|
|
default:
|
|
/*
|
|
* Complete the IRP with the default status.
|
|
*/
|
|
status = irp->IoStatus.Status;
|
|
break;
|
|
}
|
|
|
|
/*
|
|
* Complete the IRP here
|
|
*/
|
|
irp->IoStatus.Status = status;
|
|
IoCompleteRequest(irp, IO_NO_INCREMENT);
|
|
}
|
|
|
|
DBG_LOG_PNP_IRP(irp, minorFunction, isParentFdo, TRUE, status);
|
|
|
|
return status;
|
|
}
|
|
|
|
|
|
NTSTATUS USBC_PnpComplete(IN PDEVICE_OBJECT devObj, IN PIRP irp, IN PVOID context)
|
|
{
|
|
PDEVEXT devExt = (PDEVEXT)context;
|
|
PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(irp);
|
|
NTSTATUS status = irp->IoStatus.Status;
|
|
|
|
ASSERT(devExt->signature == USBCCGP_TAG);
|
|
|
|
if (devExt->isParentFdo){
|
|
PPARENT_FDO_EXT parentFdoExt = &devExt->parentFdoExt;
|
|
|
|
switch (irpSp->MinorFunction){
|
|
|
|
case IRP_MN_START_DEVICE:
|
|
ASSERT(0);
|
|
break;
|
|
|
|
case IRP_MN_STOP_DEVICE:
|
|
/*
|
|
* Only set the state to STOPPED if the device
|
|
* was previously successfully started;
|
|
* otherwise, leave the state alone.
|
|
*/
|
|
if (parentFdoExt->state == STATE_STOPPING){
|
|
|
|
/*
|
|
* Free the interface list's .Interface pointers,
|
|
* which we have to re-allocate on a start-after-stop.
|
|
*/
|
|
FreeInterfaceList(parentFdoExt, FALSE);
|
|
|
|
parentFdoExt->state = STATE_STOPPED;
|
|
}
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
/*
|
|
* Must propagate the pending bit if a lower driver returned pending.
|
|
*/
|
|
if (irp->PendingReturned){
|
|
IoMarkIrpPending(irp);
|
|
}
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
********************************************************************************
|
|
* GetDeviceText
|
|
********************************************************************************
|
|
*
|
|
* If the interface descriptor for this function has a non-zero iInterface
|
|
* string, return that string. Otherwise, pass this request off to the
|
|
* parent, which will return the iProduct string from the device descriptor.
|
|
*
|
|
*/
|
|
NTSTATUS GetDeviceText(PFUNCTION_PDO_EXT functionPdoExt, PIRP irp)
|
|
{
|
|
NTSTATUS status;
|
|
UCHAR ifaceStringIndex;
|
|
ULONG ulBytes = 0;
|
|
|
|
PAGED_CODE();
|
|
|
|
ifaceStringIndex = functionPdoExt->functionInterfaceList[0].InterfaceDescriptor->iInterface;
|
|
if (ifaceStringIndex){
|
|
PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(irp);
|
|
PUSB_STRING_DESCRIPTOR ifaceStringDesc;
|
|
PWCHAR deviceText;
|
|
|
|
switch (irpSp->Parameters.QueryDeviceText.DeviceTextType){
|
|
|
|
case DeviceTextDescription:
|
|
ifaceStringDesc = ALLOCPOOL(NonPagedPool, MAXIMUM_USB_STRING_LENGTH);
|
|
if (ifaceStringDesc){
|
|
LANGID langId = (LANGID)(irpSp->Parameters.QueryDeviceText.LocaleId >> 16);
|
|
|
|
if (langId == 0){
|
|
/*
|
|
* Force to English.
|
|
*/
|
|
langId = 0x0409;
|
|
}
|
|
|
|
QDT_Retry:
|
|
status = GetStringDescriptor( functionPdoExt->parentFdoExt,
|
|
ifaceStringIndex,
|
|
langId,
|
|
ifaceStringDesc,
|
|
MAXIMUM_USB_STRING_LENGTH);
|
|
|
|
if (NT_SUCCESS(status) &&
|
|
ifaceStringDesc->bLength == 0) {
|
|
|
|
status = STATUS_UNSUCCESSFUL;
|
|
}
|
|
|
|
if (NT_SUCCESS(status)){
|
|
ULONG numWchars = (ifaceStringDesc->bLength - 2*sizeof(UCHAR))/sizeof(WCHAR);
|
|
|
|
deviceText = ALLOCPOOL(PagedPool, (numWchars+1)*sizeof(WCHAR));
|
|
if (deviceText){
|
|
RtlZeroMemory(deviceText, (numWchars+1)*sizeof(WCHAR));
|
|
RtlCopyMemory(deviceText, ifaceStringDesc->bString, numWchars*sizeof(WCHAR));
|
|
irp->IoStatus.Information = (ULONG_PTR)deviceText;
|
|
}
|
|
else {
|
|
status = STATUS_INSUFFICIENT_RESOURCES;
|
|
}
|
|
} else if (langId != 0x409) {
|
|
// We are running a non-English flavor of the OS, but the
|
|
// attached USB device does not contain device text in
|
|
// the requested language. Let's try again for English.
|
|
|
|
langId = 0x0409;
|
|
goto QDT_Retry;
|
|
}
|
|
|
|
FREEPOOL(ifaceStringDesc);
|
|
}
|
|
else {
|
|
status = STATUS_INSUFFICIENT_RESOURCES;
|
|
}
|
|
|
|
if (!NT_SUCCESS(status) && GenericCompositeUSBDeviceString) {
|
|
// Return generic English string if one could not be
|
|
// obtained from the device.
|
|
|
|
STRLEN(ulBytes, GenericCompositeUSBDeviceString);
|
|
|
|
ulBytes += sizeof(UNICODE_NULL);
|
|
|
|
deviceText = ALLOCPOOL(PagedPool, ulBytes);
|
|
if (deviceText) {
|
|
RtlZeroMemory(deviceText, ulBytes);
|
|
RtlCopyMemory(deviceText,
|
|
GenericCompositeUSBDeviceString,
|
|
ulBytes);
|
|
irp->IoStatus.Information = (ULONG_PTR) deviceText;
|
|
status = STATUS_SUCCESS;
|
|
} else {
|
|
status = STATUS_INSUFFICIENT_RESOURCES;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case DeviceTextLocationInformation:
|
|
/*
|
|
* BUGBUG
|
|
* Supporting this call to return phys port # is optional.
|
|
*
|
|
* We may be able to service it with
|
|
* a call to IOCTL_INTERNAL_USB_GET_PARENT_HUB_INFO
|
|
* (Pass a PULONG in Argument2, this will be filled in with the port #.).
|
|
*/
|
|
status = irp->IoStatus.Status;
|
|
break;
|
|
|
|
default:
|
|
DBGWARN(("GetDeviceText: unhandled DeviceTextType %xh.", (ULONG)irpSp->Parameters.QueryDeviceText.DeviceTextType));
|
|
status = irp->IoStatus.Status;
|
|
break;
|
|
}
|
|
}
|
|
else {
|
|
IoCopyCurrentIrpStackLocationToNext(irp);
|
|
status = CallNextDriverSync(functionPdoExt->parentFdoExt, irp);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|