有時(shí)應(yīng)用需要檢查USB設(shè)備的加載與卸除消息,如U盤的插上與拔出。一種辦法是以輪詢的方式調(diào)用設(shè)備接口嘗試訪問(wèn)設(shè)備,但這樣的做法效率很低,并且實(shí)時(shí)性不高。本文將介紹更為通用的方法,通過(guò)系統(tǒng)API函數(shù)RequestDeviceNotifications,讀取設(shè)備消息,獲得設(shè)備加載及移除的消息。
查找設(shè)備IClass值
每個(gè)USB設(shè)備都有一個(gè)IClass值,相當(dāng)于在系統(tǒng)里的編號(hào),IClass值可以通過(guò)查詢板子注冊(cè)表獲得。以U盤為例:
如圖,[HKEY_LOCAL_MACHINE\Drivers\USB\ClientDrivers]下存有所有USB設(shè)備類型的信息。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)聽(tīng)設(shè)備消息
監(jiān)聽(tīng)流程如下:
創(chuàng)建消息隊(duì)列->綁定設(shè)備和消息隊(duì)列->讀取消息隊(duì)列
以U盤為例,代碼如下
MSGQUEUEOPTIONS msgopts;
//{A4E7EDDA-E575-4252-9D6B-4195D48BB865} ? ? ? ?這個(gè)是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) ? ? ? ?//參數(shù)“1”表示1ms超時(shí)
{
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ū)動(dòng)中的名稱,如“DSK1”,并非U盤盤符。
消息隊(duì)列里讀到的消息包括已加載設(shè)備的加載消息。
根據(jù)設(shè)備加載消息獲得準(zhǔn)確的設(shè)備信息
以U盤為例,它的設(shè)備類型為“DSK”,而SD卡,NANDFLASH的設(shè)備類型也是DSK,所以從設(shè)備消息里讀出設(shè)備名如“DSK1”、“DSK2”后并不能直觀確定哪個(gè)是U盤的設(shè)備名。
這時(shí)可以調(diào)用存儲(chǔ)器相關(guān)API函數(shù)OpenStore根據(jù)設(shè)備名,查詢U盤盤符等信息。
必要的頭文件及l(fā)ib庫(kù)
#include "Storemgr.h"
#pragma comment(lib,"Storeapi.lib")
封裝OpenStore函數(shù)
增加超時(shí)是有必要的,因?yàn)樵O(shè)備加載消息來(lái)得更快,可能存儲(chǔ)設(shè)備還沒(méi)完全加載好。最長(zhǎng)可能慢1ms,所以稍加延時(shí)即可。
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;
}
獲得詳細(xì)信息
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);
用鏈表存儲(chǔ)設(shè)備信息
在U盤加載時(shí),可以通過(guò)訪問(wèn)存儲(chǔ)驅(qū)動(dòng)API來(lái)查詢?cè)O(shè)備信息。但是當(dāng)設(shè)備移除后,設(shè)備信息就不存在于驅(qū)動(dòng)中了,所以用OpenStore和GetStoreInfo就無(wú)法查到已移除的設(shè)備信息了。
想要知道移除的設(shè)備詳細(xì)信息,就只有在加載的時(shí)候?qū)⑿畔⒋嬖跀?shù)據(jù)結(jié)構(gòu)中。這樣在移除時(shí),通過(guò)查詢數(shù)據(jù)結(jié)構(gòu)中的數(shù)據(jù),就可以獲得設(shè)備詳細(xì)信息了。
評(píng)論