如何通過系統(tǒng)API函數(shù)讀取USB設備消息，獲得設備加載及移除的消息

有時應用需要檢查USB設備的加載與卸除消息，如U盤的插上與拔出。一種辦法是以輪詢的方式調(diào)用設備接口嘗試訪問設備，但這樣的做法效率很低，并且實時性不高。本文將介紹更為通用的方法，通過系統(tǒng)API函數(shù)RequestDeviceNotifications，讀取設備消息，獲得設備加載及移除的消息。

查找設備IClass值

每個USB設備都有一個IClass值，相當于在系統(tǒng)里的編號，IClass值可以通過查詢板子注冊表獲得。以U盤為例：

如圖，[HKEY_LOCAL_MACHINE\Drivers\USB\ClientDrivers]下存有所有USB設備類型的信息。Mass_Storage_Class即U盤的IClass值為{A4E7EDDA-E575-4252-9D6B-4195D48BB865}

用代碼表示即

GUID guid = { 0xA4E7EDDA, 0xE575, 0x4252, { 0x9D, 0x6B, ?0x41, 0x95, 0xD4, 0x8B, 0xB8, 0x65 } };

調(diào)用API監(jiān)聽設備消息

監(jiān)聽流程如下：

創(chuàng)建消息隊列->綁定設備和消息隊列->讀取消息隊列

以U盤為例，代碼如下

MSGQUEUEOPTIONS msgopts;

//{A4E7EDDA-E575-4252-9D6B-4195D48BB865} ? ? ? ?這個是USB storage的

GUID guid = { 0xA4E7EDDA, 0xE575, 0x4252, { 0x9D, 0x6B, 0x41, 0x95, 0xD4, 0x8B, 0xB8, 0x65 } };

msgopts.dwSize ? ? ? ?= sizeof(MSGQUEUEOPTIONS);

msgopts.dwFlags ? ? ? = 0;

msgopts.dwMaxMessages = 10; //?

msgopts.cbMaxMessage = sizeof(MYDEV);

msgopts.bReadAccess ? = TRUE;

HANDLE m_hReadMsgQ = CreateMsgQueue(NULL, &msgopts);

HANDLE hNotify;

DWORD ret;

hNotify = RequestDeviceNotifications(&guid, m_hReadMsgQ, TRUE);

flags = 0;

size = 0;

//while(true){

ret = WaitForSingleObject(m_hReadMsgQ, INFINITE);

if(ret == WAIT_OBJECT_0)

{ ?

while (ReadMsgQueue(m_hReadMsgQ, &detail, sizeof(detail), &size, 1, &flags) == TRUE) ? ? ? ?//參數(shù)“1”表示1ms超時

{

if (detail.d.fAttached)

{

wprintf(L"%s %s,f=0x%x\r\n", detail.d.szName, detail.d.fAttached ? L"appeared" : L"was removed", flags);

}

Else

{

wprintf(L"%s %s,f=0x%x\r\n", detail.d.szName, detail.d.fAttached ? L"appeared" : L"was removed", flags);

}// if (detail.d.fAttached)

}//while

}//if(ret == WAIT_OBJECT_0)

//}while(true)

CloseMsgQueue(m_hReadMsgQ);

注意：如果是U盤，這里讀到的detail.d.szName是U盤在驅(qū)動中的名稱，如“DSK1”，并非U盤盤符。

消息隊列里讀到的消息包括已加載設備的加載消息。

根據(jù)設備加載消息獲得準確的設備信息

以U盤為例，它的設備類型為“DSK”，而SD卡，NANDFLASH的設備類型也是DSK，所以從設備消息里讀出設備名如“DSK1”、“DSK2”后并不能直觀確定哪個是U盤的設備名。

這時可以調(diào)用存儲器相關API函數(shù)OpenStore根據(jù)設備名，查詢U盤盤符等信息。

必要的頭文件及l(fā)ib庫

#include "Storemgr.h"

#pragma comment(lib,"Storeapi.lib")

封裝OpenStore函數(shù)

增加超時是有必要的，因為設備加載消息來得更快，可能存儲設備還沒完全加載好。最長可能慢1ms，所以稍加延時即可。

HANDLE WINAPI OpenStoreEx(LPCTSTR szDeviceName, DWORD timeout = 3)

{

HANDLE ?hStore;

int i;

for(i=0;i

{

hStore = OpenStore(szDeviceName);

if (hStore != INVALID_HANDLE_VALUE)

{

break;;

}

Sleep(1);

}

return hStore;

}

獲得詳細信息

GetStoreInfo(hStore, &StoreInfo);

//if(wcscmp(StoreInfo.szStoreName, L"SD Memory Card") == 0) ? ? ? ? //判斷是SD卡

//else if(wcscmp(StoreInfo.szStoreName, L"NANDFS") == 0) ? ? ? ? ? ?//判斷是NandFlash

//else if (wcscmp(StoreInfo.szStoreName, L"USB Hard Disk Drive") == 0) ?//判斷是U盤

wprintf(L"%s %s %s,f=0x%x\r\n", detail.d.szName, StoreInfo.szStoreName, detail.d.fAttached ? L"appeared" : L"was removed", flags);

用鏈表存儲設備信息

在U盤加載時，可以通過訪問存儲驅(qū)動API來查詢設備信息。但是當設備移除后，設備信息就不存在于驅(qū)動中了，所以用OpenStore和GetStoreInfo就無法查到已移除的設備信息了。

想要知道移除的設備詳細信息，就只有在加載的時候?qū)⑿畔⒋嬖跀?shù)據(jù)結(jié)構中。這樣在移除時，通過查詢數(shù)據(jù)結(jié)構中的數(shù)據(jù)，就可以獲得設備詳細信息了。