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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLT_BLDEV

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ListEntry;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint16_t idVendor;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint16_t idProduct;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint16_t bcdDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLT_BLDEV, *PVBOXUSBFLT_BLDEV;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define PVBOXUSBFLT_BLDEV_FROM_LE(_pLe) ( (PVBOXUSBFLT_BLDEV)( ((uint8_t*)(_pLe)) - RT_OFFSETOF(VBOXUSBFLT_BLDEV, ListEntry) ) )

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLTGLOBALS

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ContextList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* devices known to misbehave */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY BlackDeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK Lock;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLTGLOBALS, *PVBOXUSBFLTGLOBALS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic VBOXUSBFLTGLOBALS g_VBoxUsbFltGlobals;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic bool vboxUsbFltBlDevMatchLocked(uint16_t idVendor, uint16_t idProduct, uint16_t bcdDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.BlackDeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_BLDEV pDev = PVBOXUSBFLT_BLDEV_FROM_LE(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDev->idVendor != idVendor)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDev->idProduct != idProduct)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDev->bcdDevice != bcdDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic NTSTATUS vboxUsbFltBlDevAddLocked(uint16_t idVendor, uint16_t idProduct, uint16_t bcdDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltBlDevMatchLocked(idVendor, idProduct, bcdDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_BLDEV pDev = (PVBOXUSBFLT_BLDEV)VBoxUsbMonMemAllocZ(sizeof (*pDev));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!pDev)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync AssertFailed();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_INSUFFICIENT_RESOURCES;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDev->idVendor = idVendor;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDev->idProduct = idProduct;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDev->bcdDevice = bcdDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InsertHeadList(&g_VBoxUsbFltGlobals.BlackDeviceList, &pDev->ListEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltBlDevClearLocked()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PLIST_ENTRY pNext;

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.BlackDeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pNext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pNext = pEntry->Flink;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltBlDevPopulateWithKnownLocked()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* this one halts when trying to get string descriptors from it */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltBlDevAddLocked(0x5ac, 0x921c, 0x115);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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_WARN(!pDevice->pOwner, ("device 0x%p has an owner(0x%p)", pDevice, 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_WARN(pDevice->pOwner, ("no owner for device 0x%p", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->pOwner->cActiveFilters < UINT32_MAX/2, ("cActiveFilters (%d)", pDevice->pOwner->cActiveFilters));

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 VBOX_USB_WITH_VERBOSE_LOGGING

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_WARN( pDevice->idVendor != pDevice2->idVendor

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->bcdDevice != pDevice2->bcdDevice, ("duplicate devices in a list!!"));

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_WARN( 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 ("Invalid device state(%d) for device(0x%p) PDO(0x%p)", pDevice->enmState, pDevice, pDevice->Pdo));

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 LOG(("found device (0x%p), state(%d) for PDO(0x%p)", pDevice, pDevice->enmState, pPdo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("found replugging device (0x%p), state(%d) for PDO(0x%p)", pDevice, pDevice->enmState, pPdo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return pDevice;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic NTSTATUS vboxUsbFltPdoReplug(PDEVICE_OBJECT pDo)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Replugging PDO(0x%p)", pDo));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(Status == STATUS_SUCCESS, ("replugging PDO(0x%p) failed Status(0x%x)", pDo, Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Replugging PDO(0x%p) done with Status(0x%x)", pDo, Status));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#define VBOXUSBMON_POPULATE_REQUEST_TIMEOUT_MS 10000

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 LOG(("Populating Device(0x%p) for PDO(0x%p)", pDevice, pDo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr == NULL)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("Failed to alloc mem for urb"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_INSUFFICIENT_RESOURCES;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync do

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("getting device descriptor failed, Status (0x%x)", Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltBlDevMatchLocked(pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("found a known black list device, vid(0x%x), pid(0x%x), rev(0x%x)", pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_UNSUCCESSFUL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device pid=%x vid=%x rev=%x", 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, VBOXUSBMON_POPULATE_REQUEST_TIMEOUT_MS);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("reading language ID failed"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (Status == STATUS_CANCELLED)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("found a new black list device, vid(0x%x), pid(0x%x), rev(0x%x)", pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltBlDevAddLocked(pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_UNSUCCESSFUL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iSerialNumber)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("reading serial number failed"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->szSerial[0] == '\0', ("serial is not zero!!"));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (Status == STATUS_CANCELLED)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync WARN(("found a new black list device, vid(0x%x), pid(0x%x), rev(0x%x)", pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync vboxUsbFltBlDevAddLocked(pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Status = STATUS_UNSUCCESSFUL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync LOG(("pretending success.."));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pDevDr->iManufacturer)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (!NT_SUCCESS(Status))

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync WARN(("reading manufacturer name failed"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->szMfgName[0] == '\0', ("szMfgName is not zero!!"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (Status == STATUS_CANCELLED)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("found a new black list device, vid(0x%x), pid(0x%x), rev(0x%x)", pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltBlDevAddLocked(pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_UNSUCCESSFUL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("pretending success.."));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iProduct)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("reading product name failed"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->szProduct[0] == '\0', ("szProduct is not zero!!"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (Status == STATUS_CANCELLED)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("found a new black list device, vid(0x%x), pid(0x%x), rev(0x%x)", pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltBlDevAddLocked(pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_UNSUCCESSFUL;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync break;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync LOG(("pretending success.."));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Status = STATUS_SUCCESS;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#if 0

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (bPopulateNonFilterProps)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync WCHAR RegKeyBuf[512];

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync ULONG cbRegKeyBuf = sizeof (RegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = IoGetDeviceProperty(pDo,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync DevicePropertyDriverKeyName,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync cbRegKeyBuf,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RegKeyBuf,

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync &cbRegKeyBuf);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (!NT_SUCCESS(Status))

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync AssertMsgFailed((__FUNCTION__": IoGetDeviceProperty failed Status (0x%x)", Status));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ANSI_STRING Ansi;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UNICODE_STRING Unicode;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync 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)", 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)", Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG((": strings: '%s':'%s':'%s' (lang ID %x)",

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync pDevice->szMfgName, pDevice->szProduct, pDevice->szSerial, langId));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Populating Device(0x%p) for PDO(0x%p) Succeeded", pDevice, pDo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync } while (0);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync VBoxUsbMonMemFree(pDevDr);

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync LOG(("Populating Device(0x%p) for PDO(0x%p) Done, Status (0x%x)", pDevice, pDo, Status));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync return Status;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync}

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic bool vboxUsbFltDevCheckReplugLocked(PVBOXUSBFLT_DEVICE pDevice, PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pContext, ("context is NULL!"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Current context is (0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Current Device owner is (0x%p)", pDevice->pOwner));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* check if device is already replugging */

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device (0x%p) is already replugging, return..", pDevice));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* it is, do nothing */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ("Device (0x%p) state is NOT REPLUGGING (%d)", pDevice, pDevice->enmState));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device (0x%p) is owned by another context(0x%p), current is(0x%p)", pDevice, pDevice->pOwner, pContext));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching Info: Filter (0x%p), NewOwner(0x%p), fFilter(%d), fIsOneShot(%d)", uId, pNewOwner, (int)fFilter, (int)fIsOneShot));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Device (0x%p) is requested another owner(0x%p), current is(0x%p)", pDevice, pNewOwner, pDevice->pOwner));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Device (0x%p) should NOT be filtered", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevStateIsNotFiltered(pDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device (0x%p) is NOT filtered", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no changes */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pNewOwner, ("no new owner"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: This is a one-shot filter (0x%p), removing..", uId));

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(RT_SUCCESS(tmpRc), ("remove filter failed, rc (%d)", tmpRc));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!pDevice->pOwner)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: updating the one-shot owner to (0x%p), fltId(0x%p)", pNewOwner, uId));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerUpdateLocked(pDevice, pNewOwner, uId, true);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: device already has owner (0x%p) assigned", pDevice->pOwner));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: This is NOT a one-shot filter (0x%p), newOwner(0x%p)", uId, pNewOwner));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pNewOwner)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerUpdateLocked(pDevice, pNewOwner, uId, false);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device (0x%p) IS filtered", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!pDevice->pOwner || fIsOneShot || !pDevice->fIsFilterOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Need replug"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bNeedReplug = true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Device (0x%p) SHOULD be filtered", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(uId, ("zero uid"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pNewOwner, ("zero pNewOwner"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevStateIsFiltered(pDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device (0x%p) IS filtered", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device is filtered */

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device owner match"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no changes */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: This is a one-shot filter (0x%p), removing..", uId));

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(RT_SUCCESS(tmpRc), ("remove filter failed, rc (%d)", tmpRc));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: This is NOT a one-shot filter (0x%p), Owner(0x%p)", uId, pDevice->pOwner));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(!pDevice->pOwner, ("device should NOT have owner"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Need replug"));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bNeedReplug = true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device is currently NOT filtered,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!pDevice->pOwner || fIsOneShot || !pDevice->fIsFilterOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bNeedReplug = true;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Need replug"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bNeedReplug)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Device needs replugging, marking as such"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevStateMarkReplugLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: Device does NOT need replugging"));

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 LOG(("replugging matched PDO(0x%p), pDevice(0x%p)", pDevice->Pdo, pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_REMOVED,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ("invalid state(0x%x) for device(0x%p)", pDevice->enmState, pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltPdoReplug(pDevice->Pdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevice->Pdo);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRelease(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLTCHECKWALKER

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pContext;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLTCHECKWALKER, *PVBOXUSBFLTCHECKWALKER;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic DECLCALLBACK(BOOLEAN) vboxUsbFltFilterCheckWalker(PFILE_OBJECT pFile, PDEVICE_OBJECT pTopDo, PDEVICE_OBJECT pHubDo, PVOID pvContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCHECKWALKER pData = (PVBOXUSBFLTCHECKWALKER)pvContext;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pContext = pData->pContext;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Visiting pFile(0x%p), pTopDo(0x%p), pHubDo(0x%p), oContext(0x%p)", pFile, pTopDo, pHubDo, pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL Irql = KeGetCurrentIrql();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(Irql == PASSIVE_LEVEL, ("unexpected IRQL (%d)", Irql));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PDEVICE_RELATIONS pDevRelations = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status = VBoxUsbMonQueryBusRelations(pTopDo, pFile, &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 PDEVICE_OBJECT pDevObj = pDevRelations->Objects[k];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Found existing USB PDO 0x%p", pDevObj));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLT_DEVICE pDevice = vboxUsbFltDevGetLocked(pDevObj);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevice)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Found existing device info (0x%p) for PDO 0x%p", pDevice, pDevObj));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool bReplug = vboxUsbFltDevCheckReplugLocked(pDevice, pContext);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bReplug)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Replug needed for device (0x%p)", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRetain(pDevice);

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Replug NOT needed for device (0x%p)", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevObj);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync --cReplugPdos;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN((uint32_t)cReplugPdos < UINT32_MAX/2, ("cReplugPdos(%d) state broken", cReplugPdos));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("NO device info found for PDO 0x%p", pDevObj));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_DEVICE Device;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = vboxUsbFltDevPopulate(&Device, pDevObj /*, 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 LOG(("Matching Info: Filter (0x%p), pCtx(0x%p), fFilter(%d), fIsOneShot(%d)", uId, pCtx, (int)fFilter, (int)fIsOneShot));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: This device SHOULD be filtered"));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("vboxUsbFltDevPopulate for PDO 0x%p failed with Status 0x%x", pDevObj, Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Matching: This device should NOT be filtered"));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pDevObj);

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync --cReplugPdos;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN((uint32_t)cReplugPdos < UINT32_MAX/2, ("cReplugPdos is %d", cReplugPdos));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("(%d) non-matched PDOs to be replugged", cReplugPdos));

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_WARN(Status == STATUS_SUCCESS, ("vboxUsbFltPdoReplug ailed Status(0x%x)", Status));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!--cReplugPdos)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(!cReplugPdos, ("cReplugPdosreached zero!"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltReplugList(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ExFreePool(pDevRelations);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("VBoxUsbMonQueryBusRelations failed for DO(0x%p), Status(0x%x), pDevRelations(0x%p)",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pTopDo, Status, pDevRelations));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Done Visiting pFile(0x%p), pTopDo(0x%p), pHubDo(0x%p), oContext(0x%p)", pFile, pTopDo, pHubDo, pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return TRUE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltFilterCheck(PVBOXUSBFLTCTX pContext)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync{

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync KIRQL Irql = KeGetCurrentIrql();

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync ASSERT_WARN(Irql == PASSIVE_LEVEL, ("unexpected IRQL (%d)", Irql));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Running filters, Context (0x%p)..", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync VBOXUSBFLTCHECKWALKER Data;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync Data.pContext = pContext;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbMonHubDevWalk(vboxUsbFltFilterCheckWalker, &Data, VBOXUSBMONHUBWALK_F_FDO);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("DONE Running filters, Context (0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltClose(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Closing context(0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ReplugDevList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeListHead(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pContext, ("null context"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL Irql = KeGetCurrentIrql();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(Irql == PASSIVE_LEVEL, ("irql==(%d)", Irql));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uint32_t cActiveFilters = pContext->cActiveFilters;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pContext->bRemoved = TRUE;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pContext->pChangeEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("seting & closing change event (0x%p)", pContext->pChangeEvent));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeSetEvent(pContext->pChangeEvent,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync 0, /* increment*/

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync FALSE /* wait */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ObDereferenceObject(pContext->pChangeEvent);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pContext->pChangeEvent = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("no change event"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RemoveEntryList(&pContext->ListEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("removing owner filters"));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* 1. remove filters */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUSBFilterRemoveOwner(pContext);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("enumerating devices.."));

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 LOG(("found device(0x%p), pdo(0x%p), state(%d), filter id(0x%p), oneshot(%d)",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice, pDevice->Pdo, pDevice->enmState, pDevice->uFltId, (int)pDevice->fIsFilterOneShot));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState != VBOXUSBFLT_DEVSTATE_ADDED, ("VBOXUSBFLT_DEVSTATE_ADDED state for device(0x%p)", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState != VBOXUSBFLT_DEVSTATE_REMOVED, ("VBOXUSBFLT_DEVSTATE_REMOVED state for device(0x%p)", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerClearLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevCheckReplugLocked(pDevice, pContext))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("device needs replug"));

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 else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("device does NOT need replug"));

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 LOG(("SUCCESS done context(0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltCreate(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Creating context(0x%p)", pContext));

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 LOG(("SUCCESS context(0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("adding filter, Context (0x%p)..", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* LOG the filter details. */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG((__FUNCTION__": %s %s %s",

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 VBOX_USB_WITH_VERBOSE_LOGGING

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

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 USBFilterGetFilterType(pFilter)));

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 if (RT_SUCCESS(rc))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("ADDED filer id 0x%p", uId));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(uId, ("uid is NULL"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef VBOX_USBMON_WITH_FILTER_AUTOAPPLY

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbFltFilterCheck();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("VBoxUSBFilterAdd failed rc (%d)", rc));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(!uId, ("uid is not NULL"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pId = uId;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return rc;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncint VBoxUsbFltRemove(PVBOXUSBFLTCTX pContext, uintptr_t uId)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("removing filter id(0x%p), Context (0x%p)..", pContext, uId));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(uId);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!RT_SUCCESS(rc))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WARN(("VBoxUSBFilterRemove failed rc (%d)", rc));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return rc;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("enumerating devices.."));

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_WARN(!pDevice->uFltId, ("oneshot filter on device(0x%p): unexpected uFltId(%d)", pDevice, pDevice->uFltId));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->pOwner == pContext, ("Device(0x%p) owner(0x%p) not match to (0x%p)", pDevice, pDevice->pOwner, pContext));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("found device(0x%p), pdo(0x%p), state(%d), filter id(0x%p), oneshot(%d)",

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice, pDevice->Pdo, pDevice->enmState, pDevice->uFltId, (int)pDevice->fIsFilterOneShot));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(!pDevice->fIsFilterOneShot, ("device(0x%p) is filtered with a oneshot filter", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->uFltId = 0;

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->fIsFilterOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("done enumerating devices"));

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_WARN(!pDevice->pOwner, ("replugging device(0x%p) still has an owner(0x%p)", pDevice, pDevice->pOwner));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(!pDevice->uFltId, ("replugging device(0x%p) still has filter(0x%p)", pDevice, pDevice->uFltId));

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_WARN(pDevice->pOwner, ("device(0x%p) has noowner", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Device owner differs from the current context, returning used by host"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_HOST;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

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 WARN(("unexpected device state(%d) for device(0x%p)", pDevice->enmState, pDevice));

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 LOG(("found device (0x%p), state(%d) for PDO(0x%p)", pDevice, pDevice->enmState, pPdo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState != VBOXUSBFLT_DEVSTATE_ADDED, ("VBOXUSBFLT_DEVSTATE_ADDED state for device(0x%p)", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState != VBOXUSBFLT_DEVSTATE_REMOVED, ("VBOXUSBFLT_DEVSTATE_REMOVED state for device(0x%p)", pDevice));

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 WARN(("VBoxUsbMonMemAllocZ failed"));

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 WARN(("vboxUsbFltDevPopulate failed, Status 0x%x", Status));

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 LOG(("second try: found device (0x%p), state(%d) for PDO(0x%p)", pDevice, pDevice->enmState, pPdo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState != VBOXUSBFLT_DEVSTATE_ADDED, ("second try: VBOXUSBFLT_DEVSTATE_ADDED state for device(0x%p)", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ASSERT_WARN(pDevice->enmState != VBOXUSBFLT_DEVSTATE_REMOVED, ("second try: VBOXUSBFLT_DEVSTATE_REMOVED state for device(0x%p)", pDevice));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LOG(("Created Device 0x%p for PDO 0x%p", pDevice, pPdo));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pCtx = vboxUsbFltDevMatchLocked(pDevice, &uId,

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &fFilter, &fIsOneShot);

LOG(("Matching Info: Filter (0x%p), pCtx(0x%p), fFilter(%d), fIsOneShot(%d)", uId, pCtx, (int)fFilter, (int)fIsOneShot));

if (fFilter)

{

LOG(("Created Device 0x%p should be filtered", pDevice));

ASSERT_WARN(pCtx, ("zero ctx"));

ASSERT_WARN(uId, ("zero uId"));

pDevice->enmState = VBOXUSBFLT_DEVSTATE_CAPTURING;

}

else

{

LOG(("Created Device 0x%p should NOT be filtered", pDevice));

ASSERT_WARN(!uId == !pCtx, ("invalid uid(0x%p) - ctx(0x%p) pair", uId, pCtx)); /* either both zero or both not */

pDevice->enmState = VBOXUSBFLT_DEVSTATE_UNCAPTURED;

}

if (pCtx)

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

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

/* do not need to signal anything here -

VBOXUSBFLT_LOCK_RELEASE();

*pbFiltered = fFilter;

return STATUS_SUCCESS;

}

NTSTATUS VBoxUsbFltPdoAddCompleted(PDEVICE_OBJECT pPdo)

{

VBOXUSBFLT_LOCK_ACQUIRE();

vboxUsbFltSignalChangeLocked();

VBOXUSBFLT_LOCK_RELEASE();

return STATUS_SUCCESS;

}

BOOLEAN VBoxUsbFltPdoIsFiltered(PDEVICE_OBJECT pPdo)

{

VBOXUSBFLT_DEVSTATE enmState = VBOXUSBFLT_DEVSTATE_REMOVED;

VBOXUSBFLT_LOCK_ACQUIRE();

PVBOXUSBFLT_DEVICE pDevice = vboxUsbFltDevGetLocked(pPdo);

if (pDevice)

{

enmState = pDevice->enmState;

}

VBOXUSBFLT_LOCK_RELEASE();

return enmState >= VBOXUSBFLT_DEVSTATE_CAPTURING;

}

NTSTATUS VBoxUsbFltPdoRemove(PDEVICE_OBJECT pPdo)