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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsynctypedef struct VBOXUSBFLT_BLDEV

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync{

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync LIST_ENTRY ListEntry;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync uint16_t idVendor;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync uint16_t idProduct;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync uint16_t bcdDevice;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync} VBOXUSBFLT_BLDEV, *PVBOXUSBFLT_BLDEV;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsynctypedef struct VBOXUSBFLTGLOBALS

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY DeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ContextList;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync /* devices known to misbehave */

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync LIST_ENTRY BlackDeviceList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK Lock;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync} VBOXUSBFLTGLOBALS, *PVBOXUSBFLTGLOBALS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic VBOXUSBFLTGLOBALS g_VBoxUsbFltGlobals;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsyncstatic bool vboxUsbFltBlDevMatchLocked(uint16_t idVendor, uint16_t idProduct, uint16_t bcdDevice)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync{

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.BlackDeviceList.Flink;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pEntry != &g_VBoxUsbFltGlobals.BlackDeviceList;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pEntry = pEntry->Flink)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync PVBOXUSBFLT_BLDEV pDev = PVBOXUSBFLT_BLDEV_FROM_LE(pEntry);

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (pDev->idVendor != idVendor)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync continue;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (pDev->idProduct != idProduct)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync continue;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (pDev->bcdDevice != bcdDevice)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync continue;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync return true;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync return false;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync}

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsyncstatic NTSTATUS vboxUsbFltBlDevAddLocked(uint16_t idVendor, uint16_t idProduct, uint16_t bcdDevice)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync{

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (vboxUsbFltBlDevMatchLocked(idVendor, idProduct, bcdDevice))

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync return STATUS_SUCCESS;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync PVBOXUSBFLT_BLDEV pDev = (PVBOXUSBFLT_BLDEV)VBoxUsbMonMemAllocZ(sizeof (*pDev));

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (!pDev)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync AssertFailed();

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync return STATUS_INSUFFICIENT_RESOURCES;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pDev->idVendor = idVendor;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pDev->idProduct = idProduct;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pDev->bcdDevice = bcdDevice;

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync return STATUS_SUCCESS;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync}

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsyncstatic void vboxUsbFltBlDevClearLocked()

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync{

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync PLIST_ENTRY pNext;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync for (PLIST_ENTRY pEntry = g_VBoxUsbFltGlobals.BlackDeviceList.Flink;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pEntry != &g_VBoxUsbFltGlobals.BlackDeviceList;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pEntry = pNext)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync pNext = pEntry->Flink;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync VBoxUsbMonMemFree(pEntry);

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync}

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsyncstatic void vboxUsbFltBlDevPopulateWithKnownLocked()

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync{

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync vboxUsbFltBlDevAddLocked(0x5ac, 0x921c, 0x115);

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync}

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

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{

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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{

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync ASSERT_WARN(pDevice->pOwner, ("no owner for device 0x%p", pDevice));

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

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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{

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync#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);

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN( pDevice->idVendor != pDevice2->idVendor

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

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_USED_BY_GUEST,

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ("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);

dab0c4ff7639347758c6506c37a16b11009c111cvboxsync /*

e260ace5dca451359028bb3281df2e843bb87ff5vboxsync if (pDevice == NULL)

e260ace5dca451359028bb3281df2e843bb87ff5vboxsync {

e260ace5dca451359028bb3281df2e843bb87ff5vboxsync WARN(("failed to get device for PDO(0x%p)", pPdo));

e260ace5dca451359028bb3281df2e843bb87ff5vboxsync }

e260ace5dca451359028bb3281df2e843bb87ff5vboxsync else if (pDevice->enmState > VBOXUSBFLT_DEVSTATE_ADDED)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevRetain(pDevice);

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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{

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Replugging PDO(0x%p)", pDo));

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

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsyncstatic bool vboxUsbFltDevCanBeCaptured(PVBOXUSBFLT_DEVICE pDevice)

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync{

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync if (pDevice->bClass == USB_DEVICE_CLASS_HUB)

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync {

980f7af98b185c77821b3fa52bddcde30c1fe1a1vboxsync LOG(("device (0x%p), pdo (0x%p) is a hub, can not be captured", pDevice, pDevice->Pdo));

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync return false;

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync }

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync return true;

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync}

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync *puId = 0;

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync *pfFilter = false;

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync *pfIsOneShot = false;

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync if (!vboxUsbFltDevCanBeCaptured(pDevice))

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync {

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync LOG(("vboxUsbFltDevCanBeCaptured returned false"));

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync return NULL;

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync }

58b3546cdc5359d0eb80ffa5c29200e8e1773625vboxsync

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync#define VBOXUSBMON_POPULATE_REQUEST_TIMEOUT_MS 10000

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Populating Device(0x%p) for PDO(0x%p)", pDevice, pDo));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr == NULL)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync WARN(("Failed to alloc mem for urb"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_INSUFFICIENT_RESOURCES;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync do

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync break;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (vboxUsbFltBlDevMatchLocked(pDevDr->idVendor, pDevDr->idProduct, pDevDr->bcdDevice))

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = STATUS_UNSUCCESSFUL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync 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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = VBoxUsbToolGetLangID(pDo, &langId, VBOXUSBMON_POPULATE_REQUEST_TIMEOUT_MS);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync WARN(("reading language ID failed"));

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (Status == STATUS_CANCELLED)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

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

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = STATUS_UNSUCCESSFUL;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iSerialNumber)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync WARN(("reading serial number failed"));

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (Status == STATUS_CANCELLED)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

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

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = STATUS_UNSUCCESSFUL;

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync break;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync LOG(("pretending success.."));

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iManufacturer)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (Status == STATUS_CANCELLED)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

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

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = STATUS_UNSUCCESSFUL;

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync break;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync LOG(("pretending success.."));

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pDevDr->iProduct)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync WARN(("reading product name failed"));

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync if (Status == STATUS_CANCELLED)

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync {

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

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

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = STATUS_UNSUCCESSFUL;

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync break;

e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync }

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync LOG(("pretending success.."));

3bba48576f7513ee7941bdbe2b5d677715578114vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#if 0

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bPopulateNonFilterProps)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync WCHAR RegKeyBuf[512];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ULONG cbRegKeyBuf = sizeof (RegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = IoGetDeviceProperty(pDo,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync DevicePropertyDriverKeyName,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync cbRegKeyBuf,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RegKeyBuf,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &cbRegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync ANSI_STRING Ansi;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync UNICODE_STRING Unicode;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Ansi.Buffer = pDevice->szDrvKeyName;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Ansi.Length = 0;

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RtlInitUnicodeString(&Unicode, RegKeyBuf);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync 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 {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync AssertMsgFailed((__FUNCTION__": VBoxUsbToolGetDeviceSpeed failed Status (0x%x)", Status));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync break;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Status = STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync } while (0);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbMonMemFree(pDevDr);

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return Status;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic void vboxUsbFltSignalChangeLocked()

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry != &g_VBoxUsbFltGlobals.ContextList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pEntry = pEntry->Flink)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVBOXUSBFLTCTX pCtx = PVBOXUSBFLTCTX_FROM_LE(pEntry);

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync Assert(!pCtx->bRemoved);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (pCtx->pChangeEvent)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeSetEvent(pCtx->pChangeEvent,

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync 0, /* increment*/

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync FALSE /* wait */);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsyncstatic bool vboxUsbFltDevCheckReplugLocked(PVBOXUSBFLT_DEVICE pDevice, PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(pContext, ("context is NULL!"));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Current context is (0x%p)", pContext));

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* check if device is already replugging */

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync /* it is, do nothing */

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING,

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

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

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync return false;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync uintptr_t uId = 0;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool bNeedReplug = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fFilter = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync bool fIsOneShot = false;

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync &fFilter, &fIsOneShot);

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (!fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Device (0x%p) is NOT filtered", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no changes */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(pNewOwner, ("no new owner"));

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(RT_SUCCESS(tmpRc), ("remove filter failed, rc (%d)", tmpRc));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (!pDevice->pOwner)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

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

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

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync else

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync {

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pNewOwner)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Device (0x%p) IS filtered", pDevice));

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

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Matching: Need replug"));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync bNeedReplug = true;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(uId, ("zero uid"));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(pNewOwner, ("zero pNewOwner"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (vboxUsbFltDevStateIsFiltered(pDevice))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Device (0x%p) IS filtered", pDevice));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* the device is filtered */

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Device owner match"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* no changes */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (fIsOneShot)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(RT_SUCCESS(tmpRc), ("remove filter failed, rc (%d)", tmpRc));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

a2104ee1ba1f25c0efb4ac5d2cd058a7201ec553vboxsync /* the device is currently NOT filtered,

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync {

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync bNeedReplug = true;

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Matching: Need replug"));

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync if (bNeedReplug)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Matching: Device needs replugging, marking as such"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevStateMarkReplugLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync else

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("Matching: Device does NOT need replugging"));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync }

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("replugging matched PDO(0x%p), pDevice(0x%p)", pDevice->Pdo, pDevice));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(pDevice->enmState == VBOXUSBFLT_DEVSTATE_REPLUGGING

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync || pDevice->enmState == VBOXUSBFLT_DEVSTATE_REMOVED,

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ("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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsynctypedef struct VBOXUSBFLTCHECKWALKER

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PVBOXUSBFLTCTX pContext;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync} VBOXUSBFLTCHECKWALKER, *PVBOXUSBFLTCHECKWALKER;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync{

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PVBOXUSBFLTCHECKWALKER pData = (PVBOXUSBFLTCHECKWALKER)pvContext;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PVBOXUSBFLTCTX pContext = pData->pContext;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL Irql = KeGetCurrentIrql();

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PDEVICE_RELATIONS pDevRelations = NULL;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync NTSTATUS Status = VBoxUsbMonQueryBusRelations(pTopDo, pFile, &pDevRelations);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (Status == STATUS_SUCCESS && pDevRelations)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ULONG cReplugPdos = pDevRelations->Count;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync LIST_ENTRY ReplugDevList;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync InitializeListHead(&ReplugDevList);

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PDEVICE_OBJECT pDevObj = pDevRelations->Objects[k];

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PVBOXUSBFLT_DEVICE pDevice = vboxUsbFltDevGetLocked(pDevObj);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (pDevice)

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync bool bReplug = vboxUsbFltDevCheckReplugLocked(pDevice, pContext);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (bReplug)

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync InsertHeadList(&ReplugDevList, &pDevice->RepluggingLe);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync vboxUsbFltDevRetain(pDevice);

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync else

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ObDereferenceObject(pDevObj);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync pDevRelations->Objects[k] = NULL;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync --cReplugPdos;

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync continue;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLT_DEVICE Device;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync Status = vboxUsbFltDevPopulate(&Device, pDevObj /*, FALSE /* only need filter properties */);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (NT_SUCCESS(Status))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync uintptr_t uId = 0;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync bool fFilter = false;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync bool fIsOneShot = false;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLT_LOCK_ACQUIRE();

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync PVBOXUSBFLTCTX pCtx = vboxUsbFltDevMatchLocked(&Device, &uId,

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync &fFilter, &fIsOneShot);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLT_LOCK_RELEASE();

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (fFilter)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

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

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync continue;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync else

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync {

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

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ObDereferenceObject(pDevObj);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync pDevRelations->Objects[k] = NULL;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync --cReplugPdos;

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (cReplugPdos)

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (!pDevRelations->Objects[k])

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync continue;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ASSERT_WARN(Status == STATUS_SUCCESS, ("vboxUsbFltPdoReplug ailed Status(0x%x)", Status));

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ObDereferenceObject(pDevRelations->Objects[k]);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync if (!--cReplugPdos)

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync break;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ASSERT_WARN(!cReplugPdos, ("cReplugPdosreached zero!"));

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync vboxUsbFltReplugList(&ReplugDevList);

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync ExFreePool(pDevRelations);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync else

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync {

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync pTopDo, Status, pDevRelations));

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync }

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync return TRUE;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync}

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsyncNTSTATUS VBoxUsbFltFilterCheck(PVBOXUSBFLTCTX pContext)

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync{

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync KIRQL Irql = KeGetCurrentIrql();

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

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

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync VBOXUSBFLTCHECKWALKER Data;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync Data.pContext = pContext;

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync vboxUsbMonHubDevWalk(vboxUsbFltFilterCheckWalker, &Data, VBOXUSBMONHUBWALK_F_FDO);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("DONE Running filters, Context (0x%p)", pContext));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltClose(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("Closing context(0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync LIST_ENTRY ReplugDevList;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeListHead(&ReplugDevList);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync ASSERT_WARN(pContext, ("null context"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KIRQL Irql = KeGetCurrentIrql();

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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 {

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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 }

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync else

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync {

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("no change event"));

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RemoveEntryList(&pContext->ListEntry);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("removing owner filters"));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync /* 1. remove filters */

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUSBFilterRemoveOwner(pContext);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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

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

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync pDevice, pDevice->Pdo, pDevice->enmState, pDevice->uFltId, (int)pDevice->fIsFilterOneShot));

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

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync vboxUsbFltDevOwnerClearLocked(pDevice);

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (vboxUsbFltDevCheckReplugLocked(pDevice, pContext))

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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 }

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync else

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync {

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("device does NOT need replug"));

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync }

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

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("SUCCESS done context(0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncNTSTATUS VBoxUsbFltCreate(PVBOXUSBFLTCTX pContext)

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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();

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("SUCCESS context(0x%p)", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return STATUS_SUCCESS;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync}

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync{

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("adding filter, Context (0x%p)..", pContext));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync *pId = 0;

67bfa93839cac50828c90f19a00256bf44cc0f33vboxsync /* LOG the filter details. */

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync#ifdef VBOX_USB_WITH_VERBOSE_LOGGING

ef357854a9842b56c249eea7ebf2f66e7c9da7c1vboxsync 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),

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync USBFilterGetNum(pFilter, USBFILTERIDX_PORT),

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("ADDED filer id 0x%p", uId));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(uId, ("uid is NULL"));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#ifdef VBOX_USBMON_WITH_FILTER_AUTOAPPLY

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBoxUsbFltFilterCheck();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync#endif

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync else

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync WARN(("VBoxUSBFilterAdd failed rc (%d)", rc));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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{

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync WARN(("VBoxUSBFilterRemove failed rc (%d)", rc));

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return rc;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync if (pDevice->pOwner != pContext)

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync continue;

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync

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

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync pDevice, pDevice->Pdo, pDevice->enmState, pDevice->uFltId, (int)pDevice->fIsFilterOneShot));

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDevice->uFltId = 0;

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

ea011c6d420ca798e246c09234845dfdedcda4acvboxsync pDevice->fIsFilterOneShot = false;

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync }

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync VBOXUSBFLT_LOCK_RELEASE();

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync LOG(("done enumerating devices"));

32b1e0e723743026551164b8b4238f2cb8ec4b3evboxsync

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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync ASSERT_WARN(!pDevice->pOwner, ("replugging device(0x%p) still has an owner(0x%p)", pDevice, pDevice->pOwner));

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 */

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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)

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync {

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

97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return USBDEVICESTATE_USED_BY_HOST;

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync }

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:

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

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

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

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync 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 {

85da04c2a4ab34e76410f9c1d83e2fa49800818cvboxsync LOG(("second try: found device (0x%p), state(%d) for PDO(0x%p)", pDevice, pDevice->enmState, pPdo));

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

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