如何通過系統(tǒng)API函數讀取USB設備消息，獲得設備加載及移除的消息
有時應用需要檢查USB設備的加載與卸除消息，如U盤的插上與拔出。一種辦法是以輪詢的方式調用設備接口嘗試訪問設備，但這樣的做法效率很低，并且實時性不高。本文將介紹更為通用的方法，通過系統(tǒng)API函數RequestDeviceNotificaTIons，讀取設備消息，獲得設備加載及移除的消息。
查找設備IClass值每個USB設備都有一個IClass值，相當于在系統(tǒng)里的編號，IClass值可以通過查詢板子注冊表獲得。以U盤為例：
如圖，[HKEY_LOCAL_MACHINEDriversUSBClientDrivers]下存有所有USB設備類型的信息。Mass_Storage_Class即U盤的IClass值為{A4E7EDDA-E575-4252-9D6B-4195D48BB865}
用代碼表示即
GUID guid = { 0xA4E7EDDA, 0xE575, 0x4252, { 0x9D, 0x6B, ?0x41, 0x95, 0xD4, 0x8B, 0xB8, 0x65 } };
調用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) ? ? ? ?//參數“1”表示1ms超時
{
if (detail.d.fAttached)
{
wprintf(L"%s %s,f=0x%x ", detail.d.szName, detail.d.fAttached ? L"appeared" : L"was removed", flags);
}
Else
{
wprintf(L"%s %s,f=0x%x ", 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盤在驅動中的名稱，如“DSK1”，并非U盤盤符。
消息隊列里讀到的消息包括已加載設備的加載消息。
根據設備加載消息獲得準確的設備信息
以U盤為例，它的設備類型為“DSK”，而SD卡，NANDFLASH的設備類型也是DSK，所以從設備消息里讀出設備名如“DSK1”、“DSK2”后并不能直觀確定哪個是U盤的設備名。
這時可以調用存儲器相關API函數OpenStore根據設備名，查詢U盤盤符等信息。
必要的頭文件及l(fā)ib庫
#include "Storemgr.h"
#pragma comment(lib,"Storeapi.lib")
封裝OpenStore函數
增加超時是有必要的，因為設備加載消息來得更快，可能存儲設備還沒完全加載好。最長可能慢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 ", detail.d.szName, StoreInfo.szStoreName, detail.d.fAttached ? L"appeared" : L"was removed", flags);
用鏈表存儲設備信息在U盤加載時，可以通過訪問存儲驅動API來查詢設備信息。但是當設備移除后，設備信息就不存在于驅動中了，所以用OpenStore和GetStoreInfo就無法查到已移除的設備信息了。
想要知道移除的設備詳細信息，就只有在加載的時候將信息存在數據結構中。這樣在移除時，通過查詢數據結構中的數據，就可以獲得設備詳細信息了。
下載文檔
