有時應用需要檢查USB設備的加載與卸除消息,如U盤的插上與拔出。一種辦法是以輪詢的方式調用設備接口嘗試訪問設備,但這樣的做法效率很低,并且實時性不高。本文將介紹更為通用的方法,通過系統API函數RequestDeviceNotifications,讀取設備消息,獲得設備加載及移除的消息。
查找設備IClass值
每個USB設備都有一個IClass值,相當于在系統里的編號,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 } };
調用API監聽設備消息
監聽流程如下:
創建消息隊列->綁定設備和消息隊列->讀取消息隊列
以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);
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\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盤在驅動中的名稱,如“DSK1”,并非U盤盤符。
消息隊列里讀到的消息包括已加載設備的加載消息。
根據設備加載消息獲得準確的設備信息
以U盤為例,它的設備類型為“DSK”,而SD卡,NANDFLASH的設備類型也是DSK,所以從設備消息里讀出設備名如“DSK1”、“DSK2”后并不能直觀確定哪個是U盤的設備名。
這時可以調用存儲器相關API函數OpenStore根據設備名,查詢U盤盤符等信息。
必要的頭文件及lib庫
#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\r\n", detail.d.szName, StoreInfo.szStoreName, detail.d.fAttached ? L"appeared" : L"was removed", flags);
用鏈表存儲設備信息
在U盤加載時,可以通過訪問存儲驅動API來查詢設備信息。但是當設備移除后,設備信息就不存在于驅動中了,所以用OpenStore和GetStoreInfo就無法查到已移除的設備信息了。
想要知道移除的設備詳細信息,就只有在加載的時候將信息存在數據結構中。這樣在移除時,通過查詢數據結構中的數據,就可以獲得設備詳細信息了。
評論
查看更多