97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include "VBoxUsbMon.h"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include "../cmn/VBoxUsbTool.h"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <VBox/cdefs.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <VBox/types.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <iprt/process.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <iprt/assert.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <VBox/err.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <iprt/assert.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <stdio.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#pragma warning(disable : 4200)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include "usbdi.h"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#pragma warning(default : 4200)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include "usbdlib.h"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include "VBoxUSBFilterMgr.h"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <VBox/usblib.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#include <devguid.h>

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define szBusQueryDeviceId L"USB\\Vid_80EE&Pid_CAFE"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define szBusQueryHardwareIDs L"USB\\Vid_80EE&Pid_CAFE&Rev_0100\0USB\\Vid_80EE&Pid_CAFE\0\0"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define szBusQueryCompatibleIDs L"USB\\Class_ff&SubClass_00&Prot_00\0USB\\Class_ff&SubClass_00\0USB\\Class_ff\0\0"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define szDeviceTextDescription L"VirtualBox USB"

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic LPWSTR lpszStandardControllerName[1] =

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync L"\\Driver\\usbhub",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync};

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef enum

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_UNKNOWN = 0,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_REMOVED,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_REPLUGGING,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_ADDED,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_UNCAPTURED,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_CAPTURING,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_CAPTURED,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_USED_BY_GUEST,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE_32BIT_HACK = 0x7fffffff

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLT_DEVSTATE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLT_DEVICE

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY GlobalLe;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* auxiliary list to be used for gathering devices to be re-plugged

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY RepluggingLe;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* Owning session. Each matched device has an owning session. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync struct VBOXUSBFLTCTX *pOwner;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* filter id - if NULL AND device has an owner - the filter is destroyed */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uintptr_t uFltId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* true iff device is filtered with a one-shot filter */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fIsFilterOneShot;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* The device state. If the non-owner session is requesting the state while the device is grabbed,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE enmState;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync volatile uint32_t cRefs;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PDEVICE_OBJECT Pdo;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint16_t idVendor;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint16_t idProduct;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint16_t bcdDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint8_t bClass;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint8_t bSubClass;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint8_t bProtocol;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync char szSerial[MAX_USB_SERIAL_STRING];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync char szMfgName[MAX_USB_SERIAL_STRING];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync char szProduct[MAX_USB_SERIAL_STRING];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#if 0

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync char szDrvKeyName[512];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync BOOLEAN fHighSpeed;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLT_DEVICE, *PVBOXUSBFLT_DEVICE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define PVBOXUSBFLT_DEVICE_FROM_LE(_pLe) ( (PVBOXUSBFLT_DEVICE)( ((uint8_t*)(_pLe)) - RT_OFFSETOF(VBOXUSBFLT_DEVICE, GlobalLe) ) )

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define PVBOXUSBFLT_DEVICE_FROM_REPLUGGINGLE(_pLe) ( (PVBOXUSBFLT_DEVICE)( ((uint8_t*)(_pLe)) - RT_OFFSETOF(VBOXUSBFLT_DEVICE, RepluggingLe) ) )

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define PVBOXUSBFLTCTX_FROM_LE(_pLe) ( (PVBOXUSBFLTCTX)( ((uint8_t*)(_pLe)) - RT_OFFSETOF(VBOXUSBFLTCTX, ListEntry) ) )

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLT_LOCK

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KSPIN_LOCK Lock;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL OldIrql;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLT_LOCK, *PVBOXUSBFLT_LOCK;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define VBOXUSBFLT_LOCK_INIT() \

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeInitializeSpinLock(&g_VBoxUsbFltGlobals.Lock.Lock)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define VBOXUSBFLT_LOCK_TERM() do { } while (0)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define VBOXUSBFLT_LOCK_ACQUIRE() \

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeAcquireSpinLock(&g_VBoxUsbFltGlobals.Lock.Lock, &g_VBoxUsbFltGlobals.Lock.OldIrql);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define VBOXUSBFLT_LOCK_RELEASE() \

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeReleaseSpinLock(&g_VBoxUsbFltGlobals.Lock.Lock, g_VBoxUsbFltGlobals.Lock.OldIrql);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLTGLOBALS

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ContextList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK Lock;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLTGLOBALS, *PVBOXUSBFLTGLOBALS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic VBOXUSBFLTGLOBALS g_VBoxUsbFltGlobals;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncDECLINLINE(void) vboxUsbFltDevRetain(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->cRefs);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASMAtomicIncU32(&pDevice->cRefs);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltDevDestroy(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->cRefs);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REMOVED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncDECLINLINE(void) vboxUsbFltDevRelease(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint32_t cRefs = ASMAtomicDecU32(&pDevice->cRefs);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(cRefs < UINT32_MAX/2);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!cRefs)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevDestroy(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltDevOwnerSetLocked(PVBOXUSBFLT_DEVICE pDevice, PVBOXUSBFLTCTX pContext, uintptr_t uFltId, bool fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->pOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ++pContext->cActiveFilters;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->pOwner = pContext;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->uFltId = uFltId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->fIsFilterOneShot = fIsOneShot;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltDevOwnerClearLocked(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->pOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync --pDevice->pOwner->cActiveFilters;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->pOwner->cActiveFilters < UINT32_MAX/2);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->pOwner = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->uFltId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltDevOwnerUpdateLocked(PVBOXUSBFLT_DEVICE pDevice, PVBOXUSBFLTCTX pContext, uintptr_t uFltId, bool fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->pOwner != pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->pOwner)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerClearLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerSetLocked(pDevice, pContext, uFltId, fIsOneShot);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else if (pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->uFltId = uFltId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->fIsFilterOneShot = fIsOneShot;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic PVBOXUSBFLT_DEVICE vboxUsbFltDevGetLocked(PDEVICE_OBJECT pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef DEBUG_mista

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry2 = pEntry->Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry2 != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry2 = pEntry2->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice2 = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry2);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert( pDevice->idVendor != pDevice2->idVendor

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->idProduct != pDevice2->idProduct

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->bcdDevice != pDevice2->bcdDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert( pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_UNCAPTURED

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_CAPTURING

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_CAPTURED

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_USED_BY_GUEST);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->Pdo == pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncPVBOXUSBFLT_DEVICE vboxUsbFltDevGet(PDEVICE_OBJECT pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = vboxUsbFltDevGetLocked(pPdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->enmState > VBOXUSBFLT_DEVSTATE_ADDED)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRetain(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic NTSTATUS vboxUsbFltPdoReplug(PDEVICE_OBJECT pDo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status = VBoxUsbToolIoInternalCtlSendSync(pDo, IOCTL_INTERNAL_USB_CYCLE_PORT, NULL, NULL);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(Status == STATUS_SUCCESS);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic PVBOXUSBFLTCTX vboxUsbFltDevMatchLocked(PVBOXUSBFLT_DEVICE pDevice, uintptr_t *puId, bool fRemoveFltIfOneShot, bool *pfFilter, bool *pfIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFILTER DevFlt;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterInit(&DevFlt, USBFILTERTYPE_CAPTURE);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetNumExact(&DevFlt, USBFILTERIDX_VENDOR_ID, pDevice->idVendor, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetNumExact(&DevFlt, USBFILTERIDX_PRODUCT_ID, pDevice->idProduct, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetNumExact(&DevFlt, USBFILTERIDX_DEVICE_REV, pDevice->bcdDevice, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetNumExact(&DevFlt, USBFILTERIDX_DEVICE_CLASS, pDevice->bClass, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetNumExact(&DevFlt, USBFILTERIDX_DEVICE_SUB_CLASS, pDevice->bSubClass, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetNumExact(&DevFlt, USBFILTERIDX_DEVICE_PROTOCOL, pDevice->bProtocol, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetStringExact(&DevFlt, USBFILTERIDX_MANUFACTURER_STR, pDevice->szMfgName, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetStringExact(&DevFlt, USBFILTERIDX_PRODUCT_STR, pDevice->szProduct, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetStringExact(&DevFlt, USBFILTERIDX_SERIAL_NUMBER_STR, pDevice->szSerial, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* Run filters on the thing. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *puId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pfFilter = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pfIsOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pOwner = VBoxUSBFilterMatchEx(&DevFlt, puId, fRemoveFltIfOneShot, pfFilter, pfIsOneShot);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterDelete(&DevFlt);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pOwner;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltDevStateMarkReplugLocked(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerUpdateLocked(pDevice, NULL, 0, false);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_REPLUGGING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic bool vboxUsbFltDevStateIsNotFiltered(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pDevice->enmState == VBOXUSBFLT_DEVSTATE_UNCAPTURED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic bool vboxUsbFltDevStateIsFiltered(PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pDevice->enmState >= VBOXUSBFLT_DEVSTATE_CAPTURING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic NTSTATUS vboxUsbFltDevPopulate(PVBOXUSBFLT_DEVICE pDevice, PDEVICE_OBJECT pDo /*, BOOLEAN bPopulateNonFilterProps*/)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PUSB_DEVICE_DESCRIPTOR pDevDr = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->Pdo = pDo;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevDr = (PUSB_DEVICE_DESCRIPTOR)VBoxUsbMonMemAllocZ(sizeof(*pDevDr));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr == NULL)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed(("Failed to alloc mem for urb\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_INSUFFICIENT_RESOURCES;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync do

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbToolGetDescriptor(pDo, pDevDr, sizeof(*pDevDr), USB_DEVICE_DESCRIPTOR_TYPE, 0, 0);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LogRel(("VBoxUSBGetDeviceDescription: getting device descriptor failed\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("Device pid=%x vid=%x rev=%x\n", pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->idVendor = pDevDr->idVendor;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->idProduct = pDevDr->idProduct;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->bcdDevice = pDevDr->bcdDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->bClass = pDevDr->bDeviceClass;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->bSubClass = pDevDr->bDeviceSubClass;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->bProtocol = pDevDr->bDeviceProtocol;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->szSerial[0] = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->szMfgName[0] = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->szProduct[0] = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* If there are no strings, don't even try to get any string descriptors. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iSerialNumber || pDevDr->iManufacturer || pDevDr->iProduct)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync int langId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbToolGetLangID(pDo, &langId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": reading language ID failed\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iSerialNumber)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbToolGetStringDescriptorA(pDo, pDevice->szSerial, sizeof (pDevice->szSerial), pDevDr->iSerialNumber, langId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": reading serial number failed\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iManufacturer)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbToolGetStringDescriptorA(pDo, pDevice->szMfgName, sizeof (pDevice->szMfgName), pDevDr->iManufacturer, langId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": reading manufacturer name failed\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iProduct)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbToolGetStringDescriptorA(pDo, pDevice->szProduct, sizeof (pDevice->szProduct), pDevDr->iProduct, langId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": reading product name failed\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#if 0

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bPopulateNonFilterProps)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WCHAR RegKeyBuf[512];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ULONG cbRegKeyBuf = sizeof (RegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = IoGetDeviceProperty(pDo,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync DevicePropertyDriverKeyName,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync cbRegKeyBuf,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RegKeyBuf,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &cbRegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": IoGetDeviceProperty failed Status (0x%x)\n", Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ANSI_STRING Ansi;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UNICODE_STRING Unicode;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Ansi.Buffer = pDevice->szDrvKeyName;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Ansi.Length = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Ansi.MaximumLength = sizeof(pDevice->szDrvKeyName);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RtlInitUnicodeString(&Unicode, RegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = RtlUnicodeStringToAnsiString(&Ansi, &Unicode, FALSE /* do not allocate */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": RtlUnicodeStringToAnsiString failed Status (0x%x)\n", Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->fHighSpend = FALSE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbToolGetDeviceSpeed(pDo, &pDevice->fHighSpend);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertMsgFailed((__FUNCTION__": VBoxUsbToolGetDeviceSpeed failed Status (0x%x)\n", Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log((__FUNCTION__": strings: '%s':'%s':'%s' (lang ID %x)\n",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->szMfgName, pDevice->szProduct, pDevice->szSerial, langId));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync } while (0);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pDevDr);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltSignalChangeLocked()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.ContextList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.ContextList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pCtx = PVBOXUSBFLTCTX_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the removed context can not be in a list */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pCtx->bRemoved);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pCtx->pChangeEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeSetEvent(pCtx->pChangeEvent,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync 0, /* increment*/

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync FALSE /* wait */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsyncstatic bool vboxUsbFltDevCheckReplugLocked(PVBOXUSBFLT_DEVICE pDevice, PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Assert(pContext);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* check if device is already replugging */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->enmState <= VBOXUSBFLT_DEVSTATE_ADDED)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* it is, do nothing */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pDevice->pOwner && pContext != pDevice->pOwner)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* this device is owned by another context, we're not allowed to do anything */

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync return false;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uintptr_t uId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool bNeedReplug = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fFilter = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fIsOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pNewOwner = vboxUsbFltDevMatchLocked(pDevice, &uId,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync false, /* do not remove a one-shot filter */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &fFilter, &fIsOneShot);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pDevice->pOwner && pNewOwner && pDevice->pOwner != pNewOwner)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* the device is owned by another owner, we can not change the owner here */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device should NOT be filtered, check the current state */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevStateIsNotFiltered(pDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no changes */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Assert(pNewOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* remove a one-shot filter and keep the original filter data */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync int tmpRc = VBoxUSBFilterRemove(pNewOwner, uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertRC(tmpRc);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (!pDevice->pOwner)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* update owner for one-shot if the owner is changed (i.e. assigned) */

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync vboxUsbFltDevOwnerUpdateLocked(pDevice, pNewOwner, uId, true);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pNewOwner)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync vboxUsbFltDevOwnerUpdateLocked(pDevice, pNewOwner, uId, false);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

a2104ee1ba1f25c0efb4ac5d2cd058a7201ec553vboxsync /* the device is currently filtered, we should release it only if

a2104ee1ba1f25c0efb4ac5d2cd058a7201ec553vboxsync if (!pDevice->pOwner || fIsOneShot || !pDevice->fIsFilterOneShot)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync bNeedReplug = true;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* the device should be filtered, check the current state */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pNewOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevStateIsFiltered(pDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device is filtered */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pNewOwner == pDevice->pOwner)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no changes */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* remove a one-shot filter and keep the original filter data */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync int tmpRc = VBoxUSBFilterRemove(pNewOwner, uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertRC(tmpRc);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerUpdateLocked(pDevice, pDevice->pOwner, uId, false);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Assert(!pDevice->pOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device needs to be filtered, but the owner changes, replug needed */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bNeedReplug = true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

a2104ee1ba1f25c0efb4ac5d2cd058a7201ec553vboxsync /* the device is currently NOT filtered,

a2104ee1ba1f25c0efb4ac5d2cd058a7201ec553vboxsync if (!pDevice->pOwner || fIsOneShot || !pDevice->fIsFilterOneShot)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync bNeedReplug = true;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bNeedReplug)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevStateMarkReplugLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return bNeedReplug;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltReplugList(PLIST_ENTRY pList)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PLIST_ENTRY pNext;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = pList->Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != pList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pNext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pNext = pEntry->Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_REPLUGGINGLE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_REMOVED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltPdoReplug(pDevice->Pdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevice->Pdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRelease(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsyncNTSTATUS VBoxUsbFltFilterCheck(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UNICODE_STRING szStandardControllerName[RT_ELEMENTS(lpszStandardControllerName)];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL Irql = KeGetCurrentIrql();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(Irql == PASSIVE_LEVEL);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("==" __FUNCTION__"\n"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (int i=0;i<RT_ELEMENTS(lpszStandardControllerName);i++)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync szStandardControllerName[i].Length = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync szStandardControllerName[i].MaximumLength = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync szStandardControllerName[i].Buffer = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RtlInitUnicodeString(&szStandardControllerName[i], lpszStandardControllerName[i]);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (int i = 0; i < 16; i++)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync char szHubName[32];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WCHAR szwHubName[32];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UNICODE_STRING UnicodeName;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ANSI_STRING AnsiName;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PDEVICE_OBJECT pHubDevObj;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PFILE_OBJECT pHubFileObj;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync sprintf(szHubName, "\\Device\\USBPDO-%d", i);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UnicodeName.Length = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UnicodeName.MaximumLength = sizeof (szwHubName);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UnicodeName.Buffer = szwHubName;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RtlInitAnsiString(&AnsiName, szHubName);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RtlAnsiStringToUnicodeString(&UnicodeName, &AnsiName, FALSE);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = IoGetDeviceObjectPointer(&UnicodeName, FILE_READ_DATA, &pHubFileObj, &pHubDevObj);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (Status == STATUS_SUCCESS)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("IoGetDeviceObjectPointer for %s returned %p %p\n", szHubName, pHubDevObj, pHubFileObj));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pHubDevObj->DriverObject

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync && pHubDevObj->DriverObject->DriverName.Buffer

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync && pHubDevObj->DriverObject->DriverName.Length

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync )

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (int j = 0; j < RT_ELEMENTS(lpszStandardControllerName); ++j)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!RtlCompareUnicodeString(&szStandardControllerName[j], &pHubDevObj->DriverObject->DriverName, TRUE /* case insensitive */))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PDEVICE_RELATIONS pDevRelations = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef DEBUG

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("Associated driver "));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbDbgPrintUnicodeString(&pHubDevObj->DriverObject->DriverName);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log((" -> related dev obj=0x%p\n", IoGetRelatedDeviceObject(pHubFileObj)));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = VBoxUsbMonQueryBusRelations(pHubDevObj, pHubFileObj, &pDevRelations);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (Status == STATUS_SUCCESS && pDevRelations)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ULONG cReplugPdos = pDevRelations->Count;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ReplugDevList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeListHead(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (ULONG k = 0; k < pDevRelations->Count; ++k)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("Found existing USB PDO %p\n", pDevRelations->Objects[k]));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = vboxUsbFltDevGetLocked(pDevRelations->Objects[k]);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync bool bReplug = vboxUsbFltDevCheckReplugLocked(pDevice, pContext);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bReplug)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InsertHeadList(&ReplugDevList, &pDevice->RepluggingLe);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRetain(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* do not dereference object since we will use it later */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevRelations->Objects[k]);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevRelations->Objects[k] = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync --cReplugPdos;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert((uint32_t)cReplugPdos < UINT32_MAX/2);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVICE Device;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = vboxUsbFltDevPopulate(&Device, pDevRelations->Objects[k] /*, FALSE /* only need filter properties */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uintptr_t uId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fFilter = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fIsOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pCtx = vboxUsbFltDevMatchLocked(&Device, &uId,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync false, /* do not remove a one-shot filter */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &fFilter, &fIsOneShot);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this device should not be filtered, and it's not */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevRelations->Objects[k]);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevRelations->Objects[k] = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync --cReplugPdos;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert((uint32_t)cReplugPdos < UINT32_MAX/2);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this device needs to be filtered, but it's not,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (cReplugPdos)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (ULONG k = 0; k < pDevRelations->Count; ++k)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!pDevRelations->Objects[k])

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = vboxUsbFltPdoReplug(pDevRelations->Objects[k]);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(Status == STATUS_SUCCESS);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevRelations->Objects[k]);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!--cReplugPdos)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!cReplugPdos);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltReplugList(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ExFreePool(pDevRelations);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pHubFileObj);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltClose(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ReplugDevList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeListHead(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pContext);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL Irql = KeGetCurrentIrql();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(Irql == PASSIVE_LEVEL);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint32_t cActiveFilters = pContext->cActiveFilters;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pContext->bRemoved = TRUE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pContext->pChangeEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeSetEvent(pContext->pChangeEvent,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync 0, /* increment*/

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync FALSE /* wait */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pContext->pChangeEvent);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pContext->pChangeEvent = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RemoveEntryList(&pContext->ListEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* now re-arrange the filters */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* 1. remove filters */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUSBFilterRemoveOwner(pContext);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* 2. check if there are devices owned */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->pOwner != pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState != VBOXUSBFLT_DEVSTATE_ADDED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState != VBOXUSBFLT_DEVSTATE_REMOVED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerClearLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (vboxUsbFltDevCheckReplugLocked(pDevice, pContext))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InsertHeadList(&ReplugDevList, &pDevice->RepluggingLe);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* retain to ensure the device is not removed before we issue a replug */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRetain(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* keep the PDO alive */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObReferenceObject(pDevice->Pdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this should replug all devices that were either skipped or grabbed due to the context's */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltReplugList(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltCreate(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync memset(pContext, 0, sizeof (*pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pContext->Process = RTProcSelf();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InsertHeadList(&g_VBoxUsbFltGlobals.ContextList, &pContext->ListEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncint VBoxUsbFltAdd(PVBOXUSBFLTCTX pContext, PUSBFILTER pFilter, uintptr_t *pId)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* Log the filter details. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log((__FUNCTION__": %s %s %s\n",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetString(pFilter, USBFILTERIDX_MANUFACTURER_STR) ? USBFilterGetString(pFilter, USBFILTERIDX_MANUFACTURER_STR) : "<null>",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetString(pFilter, USBFILTERIDX_PRODUCT_STR) ? USBFilterGetString(pFilter, USBFILTERIDX_PRODUCT_STR) : "<null>",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetString(pFilter, USBFILTERIDX_SERIAL_NUMBER_STR) ? USBFilterGetString(pFilter, USBFILTERIDX_SERIAL_NUMBER_STR) : "<null>"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef DEBUG

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("VBoxUSBClient::addFilter: idVendor=%#x idProduct=%#x bcdDevice=%#x bDeviceClass=%#x bDeviceSubClass=%#x bDeviceProtocol=%#x bBus=%#x bPort=%#x\n",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_VENDOR_ID),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_PRODUCT_ID),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_DEVICE_REV),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_DEVICE_CLASS),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_DEVICE_SUB_CLASS),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_DEVICE_PROTOCOL),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_BUS),

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterGetNum(pFilter, USBFILTERIDX_PORT)));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* We can't get the bus/port numbers. Ignore them while matching. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetMustBePresent(pFilter, USBFILTERIDX_BUS, false);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBFilterSetMustBePresent(pFilter, USBFILTERIDX_PORT, false);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uintptr_t uId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* Add the filter. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync int rc = VBoxUSBFilterAdd(pFilter, pContext, &uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertRC(rc);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (RT_SUCCESS(rc))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef VBOX_USBMON_WITH_FILTER_AUTOAPPLY

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbFltFilterCheck();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pId = uId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return rc;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncint VBoxUsbFltRemove(PVBOXUSBFLTCTX pContext, uintptr_t uId)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync int rc = VBoxUSBFilterRemove(pContext, uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!RT_SUCCESS(rc))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return rc;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->fIsFilterOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->uFltId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->uFltId != uId)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Assert(pDevice->pOwner == pContext);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pDevice->pOwner != pContext)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync continue;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->fIsFilterOneShot);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->uFltId = 0;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* clear the fIsFilterOneShot flag to ensure the device is replugged on the next VBoxUsbFltFilterCheck call */

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync pDevice->fIsFilterOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (RT_SUCCESS(rc))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef VBOX_USBMON_WITH_FILTER_AUTOAPPLY

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbFltFilterCheck();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return rc;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltSetNotifyEvent(PVBOXUSBFLTCTX pContext, HANDLE hEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PKEVENT pEvent = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PKEVENT pOldEvent = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (hEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = ObReferenceObjectByHandle(hEvent,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync EVENT_MODIFY_STATE,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *ExEventObjectType, UserMode,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync (PVOID*)&pEvent,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NULL);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(Status == STATUS_SUCCESS);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pOldEvent = pContext->pChangeEvent;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pContext->pChangeEvent = pEvent;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pOldEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pOldEvent);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic USBDEVICESTATE vboxUsbDevGetUserState(PVBOXUSBFLTCTX pContext, PVBOXUSBFLT_DEVICE pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevStateIsNotFiltered(pDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_HOST_CAPTURABLE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device is filtered, or replugging */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->pOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->uFltId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no user state for this, we should not return it tu the user */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_HOST;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device is filtered, if owner differs from the context, return as USED_BY_HOST */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->pOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the id can be null if a filter is removed */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->pOwner != pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_HOST;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync switch (pDevice->enmState)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync case VBOXUSBFLT_DEVSTATE_UNCAPTURED:

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync case VBOXUSBFLT_DEVSTATE_CAPTURING:

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_HOST_CAPTURABLE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync case VBOXUSBFLT_DEVSTATE_CAPTURED:

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_HELD_BY_PROXY;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync case VBOXUSBFLT_DEVSTATE_USED_BY_GUEST:

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_GUEST;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync default:

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_UNSUPPORTED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbDevToUserInfo(PVBOXUSBFLTCTX pContext, PVBOXUSBFLT_DEVICE pDevice, PUSBSUP_DEVINFO pDevInfo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#if 0

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevInfo->usVendorId = pDevice->idVendor;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevInfo->usProductId = pDevice->idProduct;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevInfo->usRevision = pDevice->bcdDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevInfo->enmState = vboxUsbDevGetUserState(pContext, pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync strcpy(pDevInfo->szDrvKeyName, pDevice->szDrvKeyName);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevInfo->enmState == USBDEVICESTATE_HELD_BY_PROXY

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevInfo->enmState == USBDEVICESTATE_USED_BY_GUEST)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this is the only case where we return the obj name to the client */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync strcpy(pDevInfo->szObjName, pDevice->szObjName);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevInfo->fHighSpeed = pDevice->fHighSpeed;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltGetDevice(PVBOXUSBFLTCTX pContext, HVBOXUSBDEVUSR hDevice, PUSBSUP_GETDEV_MON pInfo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(hDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync memset (pInfo, 0, sizeof (*pInfo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState != VBOXUSBFLT_DEVSTATE_REMOVED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState != VBOXUSBFLT_DEVSTATE_ADDED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice != hDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USBDEVICESTATE enmUsrState = vboxUsbDevGetUserState(pContext, pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pInfo->enmState = enmUsrState;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this should not occur */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_INVALID_PARAMETER;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltPdoAdd(PDEVICE_OBJECT pPdo, BOOLEAN *pbFiltered)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pbFiltered = FALSE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* first check if device is in the a already */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = vboxUsbFltDevGetLocked(pPdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState != VBOXUSBFLT_DEVSTATE_ADDED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState != VBOXUSBFLT_DEVSTATE_REMOVED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pbFiltered = pDevice->enmState >= VBOXUSBFLT_DEVSTATE_CAPTURING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = (PVBOXUSBFLT_DEVICE)VBoxUsbMonMemAllocZ(sizeof (*pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_NO_MEMORY;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_ADDED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->cRefs = 1;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status = vboxUsbFltDevPopulate(pDevice, pPdo /* , TRUE /* need all props */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uintptr_t uId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fFilter = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fIsOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pCtx;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pTmpDev;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* (paranoia) re-check the device is still not here */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pTmpDev = vboxUsbFltDevGetLocked(pPdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pTmpDev)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pTmpDev->enmState != VBOXUSBFLT_DEVSTATE_ADDED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pTmpDev->enmState != VBOXUSBFLT_DEVSTATE_REMOVED);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pbFiltered = pTmpDev->enmState >= VBOXUSBFLT_DEVSTATE_CAPTURING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pCtx = vboxUsbFltDevMatchLocked(pDevice, &uId,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync true, /* remove a one-shot filter */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &fFilter, &fIsOneShot);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pCtx);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_CAPTURING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!uId == !pCtx); /* either both zero or both not */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_UNCAPTURED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pCtx)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerSetLocked(pDevice, pCtx, fIsOneShot ? 0 : uId, fIsOneShot);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InsertHeadList(&g_VBoxUsbFltGlobals.DeviceList, &pDevice->GlobalLe);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* do not need to signal anything here -

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pbFiltered = fFilter;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltPdoAddCompleted(PDEVICE_OBJECT pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltSignalChangeLocked();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncBOOLEAN VBoxUsbFltPdoIsFiltered(PDEVICE_OBJECT pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE enmState = VBOXUSBFLT_DEVSTATE_REMOVED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = vboxUsbFltDevGetLocked(pPdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync enmState = pDevice->enmState;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return enmState >= VBOXUSBFLT_DEVSTATE_CAPTURING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltPdoRemove(PDEVICE_OBJECT pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVSTATE enmOldState;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = vboxUsbFltDevGetLocked(pPdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RemoveEntryList(&pDevice->GlobalLe);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync enmOldState = pDevice->enmState;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_REMOVED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (enmOldState != VBOXUSBFLT_DEVSTATE_REPLUGGING)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltSignalChangeLocked();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device *should* reappear, do signlling on re-appear only

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRelease(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncHVBOXUSBFLTDEV VBoxUsbFltProxyStarted(PDEVICE_OBJECT pPdo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = vboxUsbFltDevGetLocked(pPdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->enmState = VBOXUSBFLT_DEVSTATE_CAPTURING)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_CAPTURED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRetain(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltSignalChangeLocked();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncvoid VBoxUsbFltProxyStopped(HVBOXUSBFLTDEV hDev)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = (PVBOXUSBFLT_DEVICE)hDev;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->enmState == VBOXUSBFLT_DEVSTATE_CAPTURED

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_USED_BY_GUEST)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this is due to devie was physically removed */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Log(("The proxy notified progy stop for the captured device 0x%x\n", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_CAPTURING;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltSignalChangeLocked();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRelease(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltInit()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync int rc = VBoxUSBFilterInit();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertRC(rc);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (RT_FAILURE(rc))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_UNSUCCESSFUL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync memset(&g_VBoxUsbFltGlobals, 0, sizeof (g_VBoxUsbFltGlobals));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeListHead(&g_VBoxUsbFltGlobals.DeviceList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeListHead(&g_VBoxUsbFltGlobals.ContextList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_INIT();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltTerm()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool bBusy = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync do

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!IsListEmpty(&g_VBoxUsbFltGlobals.ContextList))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bBusy = true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PLIST_ENTRY pNext = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pNext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pNext = pEntry->Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->uFltId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pDevice->pOwner);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice->cRefs != 1)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bBusy = true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync } while (0);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bBusy)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_DEVICE_BUSY;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = g_VBoxUsbFltGlobals.DeviceList.Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RemoveEntryList(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = PVBOXUSBFLT_DEVICE_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->enmState = VBOXUSBFLT_DEVSTATE_REMOVED;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRelease(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_TERM();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUSBFilterTerm();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync