Обнаружение USB вставки / удаления событий в Windows с помощью C ++ - PullRequest
Новая
       56

Обнаружение USB вставки / удаления событий в Windows с помощью C ++

20 голосов
/ 02 ноября 2010

Я пишу расширение для существующего приложения, которое должно обрабатывать события USB вставки / удаления.Я знаю VID / PID интересующего устройства.Однако у меня нет доступа к дескриптору окна, поэтому я не знаю, будет ли RegisterDeviceNotification полезным, если только нет способа получить дескриптор через WINAPI.Как лучше всего обнаружить события вставки / удаления USB с помощью C ++?

Этот пример кода на веб-сайте Microsoft показывает, как получать уведомления о событиях через WMI:

Как его можно изменить, чтобы получать события вставки / удаления USB?Или есть другой способ, которым я должен идти об этом?Я использую Visual Studio 2008. Спасибо.

ДОПОЛНИТЕЛЬНАЯ ИНФОРМАЦИЯ

Это то, что я имею до сих пор (минус обработка ошибок): 

DEFINE_GUID(GUID_INTERFACE_CP210x, 0x993f7832, 0x6e2d, 0x4a0f, 0xb2, 0x72, 0xe2, 0xc7, 0x8e, 0x74, 0xf9, 0x3e);

MyClass::MyClass()
{
    // Generate message-only window
    _pWndClassEx = (WNDCLASSEX *)malloc( sizeof(WNDCLASSEX) );
    memset( _pWndClassEx, 0, sizeof(WNDCLASSEX) );
    _pWndClassEx->cbSize = sizeof(WNDCLASSEX);
    _pWndClassEx->lpfnWndProc = (WNDPROC)WndProc; // function which will handle messages
    _pWndClassEx->hInstance = GetCurrentModule();
    _pWndClassEx->lpszClassName = pClassName;
    atom = RegisterClassEx( _pWndClassEx );
    _hWnd = CreateWindowEx( 0, pClassName, pWindowName, 0, 0, 0, 0, 0, HWND_MESSAGE, NULL, NULL, NULL );

    // Register the USB device for notification
    _pDevIF = (DEV_BROADCAST_DEVICEINTERFACE *)malloc( sizeof(DEV_BROADCAST_DEVICEINTERFACE) );
    memset( _pDevIF, 0, sizeof(DEV_BROADCAST_DEVICEINTERFACE) );
    _pDevIF->dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    _pDevIF->dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    _pDevIF->dbcc_classguid = GUID_INTERFACE_CP210x;
    _hNotifyDevNode = RegisterDeviceNotification( _hWnd, _pDevIF, DEVICE_NOTIFY_WINDOW_HANDLE );
}

static bool OnDeviceChange(UINT nEventType, DWORD dwData)
{
    switch ( nEventType )
    {
    case DBT_DEVICEARRIVAL:
        // A device has been inserted adn is now available.
        break;

    case DBT_DEVICEREMOVECOMPLETE:
        // Device has been removed.
        break;

    default:
        break;
    }

    return true;
}

static LRESULT WndProc( HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam )
{
    switch ( message )
    {
    case WM_DEVICECHANGE:
        OnDeviceChange( wParam, lParam ); // Set breakpoint (never gets here)
        break;

    default:
        break;
    }

    return DefWindowProc(hwnd, message, wParam, lParam);
}

ПК попадает в WndProc, но не при извлечении / вставке USB-устройства.Кажется, что ПК никогда не входит в OnDeviceChange.Любые советы будут оценены.Мне нужно обрабатывать неожиданные вставки / удаления устройства USB.Если это имеет значение, устройство USB отображается как виртуальный COM-порт для Windows.Спасибо.

Дополнительная информация: Вызов CreateWindowEx с использованием класса atom, возвращаемого RegisterClassEx, завершается ошибкой с сообщением об ошибке "Не удается найти класс окна". 

_hWnd = CreateWindowEx( 0, (LPCTSTR)&atom, pWindowName, 0, 0, 0, 0, 0, HWND_MESSAGE, NULL, NULL, NULL );

НОВЫЙ ПОДХОД

Я также пробую этот новый подход.Я пытаюсь написать окно только для сообщений, чтобы получать уведомления об изменении устройства для устройства USB.Я использую MFC, C ++ и Visual Studio 2008. Все компилируется и работает без сбоев или блокировок, но обработчик событий никогда не срабатывает.Интересующее устройство устанавливается в Windows как виртуальный COM-порт.

Мое основное приложение создает экземпляр класса, описанного ниже, а затем ожидает ввода символов с клавиатуры, используя цикл while.Именно в течение этого времени ожидания я извлекаю и вставляю свое USB-устройство, ожидая, что событие будет запущено. 

class CMessageOnlyWindow : public CWnd
{
    DECLARE_DYNAMIC(CMessageOnlyWindow)
private:
    DEV_BROADCAST_DEVICEINTERFACE * _pDevIF; // The notification filter.
    HDEVNOTIFY _hNotifyDev;             // The device notification handle.
public:
    CMessageOnlyWindow();
    virtual ~CMessageOnlyWindow();
protected:
    afx_msg BOOL OnDeviceChange( UINT nEventType, DWORD dwData );
private:
    void RegisterNotification( void );
    void UnregisterNotification( void );
protected:
    DECLARE_MESSAGE_MAP()               // Must be last.
};

Для простоты я удалил всю очистку и обработку ошибок: 

DEFINE_GUID(GUID_INTERFACE_CP210x, 0x993f7832, 0x6e2d, 0x4a0f, \
    0xb2, 0x72, 0xe2, 0xc7, 0x8e, 0x74, 0xf9, 0x3e);

IMPLEMENT_DYNAMIC(CMessageOnlyWindow, CWnd)

CMessageOnlyWindow::CMessageOnlyWindow()
{
    CString cstrWndClassName = ::AfxRegisterWndClass( NULL );
    BOOL bCreated = this->CreateEx( 0, cstrWndClassName,
        L"CMessageOnlyWindow", 0, 0, 0, 0, 0, HWND_MESSAGE, 0 );
    this->RegisterNotification();
}

CMessageOnlyWindow::~CMessageOnlyWindow() {}

BEGIN_MESSAGE_MAP(CMessageOnlyWindow, CWnd)
    ON_WM_DEVICECHANGE()
END_MESSAGE_MAP()

afx_msg BOOL CMessageOnlyWindow::OnDeviceChange( UINT nEventType, DWORD dwData )
{
    switch ( nEventType ) // <-- Never gets here.
    {
    case DBT_DEVICEARRIVAL:
        break;

    case DBT_DEVICEREMOVECOMPLETE:
        break;

    default:
        break;
    }

    return TRUE;
}

void CMessageOnlyWindow::RegisterNotification(void)
{
    _pDevIF = (DEV_BROADCAST_DEVICEINTERFACE *)malloc( sizeof(DEV_BROADCAST_DEVICEINTERFACE) );
    memset( _pDevIF, 0, sizeof(DEV_BROADCAST_DEVICEINTERFACE) );
    _pDevIF->dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    _pDevIF->dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    _pDevIF->dbcc_classguid = GUID_INTERFACE_CP210x;
    _hNotifyDev = RegisterDeviceNotification( this->m_hWnd, _pDevIF, DEVICE_NOTIFY_WINDOW_HANDLE );
}

void CMessageOnlyWindow::UnregisterNotification(void)
{
    UnregisterDeviceNotification( _hNotifyDev );
}

Любые мысли или предложения будут высоко оценены.Если какие-либо детали отсутствуют, дайте мне знать, и я буду рад их добавить.Спасибо.

Нужно ли открывать окно только для сообщений в новом потоке, или создание нового окна автоматически приводит к появлению нового потока?

Ответы [ 3 ]

14 голосов
/ 02 ноября 2010

Создайте фиктивное окно, которое ничего не делает, кроме ожидания WM_DEVICECHANGE, и зарегистрируйте это окно, используя RegisterDeviceNotification.WMI - это излишество, ИМХО.

7 голосов
/ 02 ноября 2010

Существует образец MSDN специально для вашего случая, в нативном коде.

Регистрация для уведомления устройства

Лучше сделать это таким образом, чем через WMI.

6 голосов
/ 09 августа 2011

Я следовал вашему "новому подходу" и также обнаружил, что OnDeviceChange не вызывается. Проблема заключалась в том, что не было никакого цикла сообщений, потому что это было консольное приложение. Вызов следующей функции через равные промежутки времени исправил ее. 

void check_for_device_change()
{
    MSG msg; 

    const int val = PeekMessage( &msg, 0, 0, 0, PM_REMOVE );

    if( val > 0 )
    { 
        TranslateMessage( &msg );
        DispatchMessage( &msg );
    } 
}
...