static NTSTATUS vboxUsbPnPMnRemoveDevice(PVBOXUSBDEV_EXT pDevExt, PIRP pIrp) { ENMVBOXUSB_PNPSTATE enmState = vboxUsbPnPStateGet(pDevExt); NTSTATUS Status = STATUS_SUCCESS; if (enmState != ENMVBOXUSB_PNPSTATE_SURPRISE_REMOVED) { Status = vboxUsbPnPRmDev(pDevExt); Assert(Status == STATUS_SUCCESS); } vboxUsbPnPStateSet(pDevExt, ENMVBOXUSB_PNPSTATE_REMOVED); vboxUsbDdiStateRelease(pDevExt); vboxUsbDdiStateReleaseAndWaitRemoved(pDevExt); vboxUsbRtClear(pDevExt); pIrp->IoStatus.Status = STATUS_SUCCESS; pIrp->IoStatus.Information = 0; IoSkipCurrentIrpStackLocation(pIrp); Status = IoCallDriver(pDevExt->pLowerDO, pIrp); IoDetachDevice(pDevExt->pLowerDO); IoDeleteDevice(pDevExt->pFDO); return Status; }
DECLHIDDEN(NTSTATUS) vboxUsbDispatchPnP(IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp) { PVBOXUSBDEV_EXT pDevExt = (PVBOXUSBDEV_EXT)pDeviceObject->DeviceExtension; ENMVBOXUSB_PNPSTATE enmState = vboxUsbPnPStateGet(pDevExt); if (!vboxUsbDdiStateRetainIfNotRemoved(pDevExt)) { return VBoxDrvToolIoComplete(pIrp, STATUS_DELETE_PENDING, 0); } PIO_STACK_LOCATION pSl = IoGetCurrentIrpStackLocation(pIrp); switch (pSl->MinorFunction) { case IRP_MN_START_DEVICE: { return vboxUsbPnPMnStartDevice(pDevExt, pIrp); } case IRP_MN_QUERY_STOP_DEVICE: { return vboxUsbPnPMnQueryStopDevice(pDevExt, pIrp); } case IRP_MN_STOP_DEVICE: { return vboxUsbPnPMnStopDevice(pDevExt, pIrp); } case IRP_MN_CANCEL_STOP_DEVICE: { return vboxUsbPnPMnCancelStopDevice(pDevExt, pIrp); } case IRP_MN_QUERY_REMOVE_DEVICE: { return vboxUsbPnPMnQueryRemoveDevice(pDevExt, pIrp); } case IRP_MN_REMOVE_DEVICE: { return vboxUsbPnPMnRemoveDevice(pDevExt, pIrp); } case IRP_MN_CANCEL_REMOVE_DEVICE: { return vboxUsbPnPMnCancelRemoveDevice(pDevExt, pIrp); } case IRP_MN_SURPRISE_REMOVAL: { return vboxUsbPnPMnSurpriseRemoval(pDevExt, pIrp); } case IRP_MN_QUERY_CAPABILITIES: { return vboxUsbPnPMnQueryCapabilities(pDevExt, pIrp); } default: { return vboxUsbPnPMnDefault(pDevExt, pIrp); } } }
static NTSTATUS vboxUsbPnPMnCancelStopDevice(PVBOXUSBDEV_EXT pDevExt, PIRP pIrp) { ENMVBOXUSB_PNPSTATE enmState = vboxUsbPnPStateGet(pDevExt); NTSTATUS Status = STATUS_SUCCESS; IoCopyCurrentIrpStackLocationToNext(pIrp); Status = VBoxDrvToolIoPostSync(pDevExt->pLowerDO, pIrp); if (NT_SUCCESS(Status) && enmState == ENMVBOXUSB_PNPSTATE_STOP_PENDING) { vboxUsbPnPStateRestore(pDevExt); } Status = STATUS_SUCCESS; VBoxDrvToolIoComplete(pIrp, Status, 0); vboxUsbDdiStateRelease(pDevExt); return Status; }
static NTSTATUS vboxUsbDispatchCreate(IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp) { PVBOXUSBDEV_EXT pDevExt = (PVBOXUSBDEV_EXT)pDeviceObject->DeviceExtension; NTSTATUS Status = STATUS_INVALID_HANDLE; do { if (vboxUsbPnPStateGet(pDevExt) != ENMVBOXUSB_PNPSTATE_STARTED) { Status = STATUS_INVALID_DEVICE_STATE; break; } PIO_STACK_LOCATION pSl = IoGetCurrentIrpStackLocation(pIrp); PFILE_OBJECT pFObj = pSl->FileObject; if (!pFObj) { Status = STATUS_INVALID_PARAMETER; break; } pFObj->FsContext = NULL; if (pFObj->FileName.Length) { Status = STATUS_INVALID_PARAMETER; break; } Status = vboxUsbRtCreate(pDevExt, pIrp); if (!NT_SUCCESS(Status)) { AssertFailed(); break; } ASMAtomicIncU32(&pDevExt->cHandles); Status = STATUS_SUCCESS; break; } while (0); Status = VBoxDrvToolIoComplete(pIrp, Status, 0); return Status; }
static NTSTATUS vboxUsbPnPMnCancelRemoveDevice(PVBOXUSBDEV_EXT pDevExt, PIRP pIrp) { ENMVBOXUSB_PNPSTATE enmState = vboxUsbPnPStateGet(pDevExt); NTSTATUS Status = STATUS_SUCCESS; if (enmState == ENMVBOXUSB_PNPSTATE_REMOVE_PENDING) { IoCopyCurrentIrpStackLocationToNext(pIrp); Status = VBoxDrvToolIoPostSync(pDevExt->pLowerDO, pIrp); if (NT_SUCCESS(Status)) { vboxUsbPnPStateRestore(pDevExt); } } else { Assert(0); Assert(enmState == ENMVBOXUSB_PNPSTATE_STARTED); } VBoxDrvToolIoComplete(pIrp, Status, 0); vboxUsbDdiStateRelease(pDevExt); return Status; }