如何通過系統(tǒng)API函數(shù)讀取USB設(shè)備消息，獲得設(shè)備加載及移除的消息
有時應(yīng)用需要檢查USB設(shè)備的加載與卸除消息，如U盤的插上與拔出。一種辦法是以輪詢的方式調(diào)用設(shè)備接口嘗試訪問設(shè)備，但這樣的做法效率很低，并且實時性不高。本文將介紹更為通用的方法，通過系統(tǒng)API函數(shù)RequestDeviceNotificaTIons，讀取設(shè)備消息，獲得設(shè)備加載及移除的消息。
查找設(shè)備IClass值每個USB設(shè)備都有一個IClass值，相當(dāng)于在系統(tǒng)里的編號，IClass值可以通過查詢板子注冊表獲得。以U盤為例：
如圖，[HKEY_LOCAL_MACHINEDriversUSBClientDrivers]下存有所有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)聽設(shè)備消息
監(jiān)聽流程如下：
創(chuàng)建消息隊列->綁定設(shè)備和消息隊列->讀取消息隊列
以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 ", 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盤在驅(qū)動中的名稱，如“DSK1”，并非U盤盤符。
消息隊列里讀到的消息包括已加載設(shè)備的加載消息。
根據(jù)設(shè)備加載消息獲得準(zhǔn)確的設(shè)備信息
以U盤為例，它的設(shè)備類型為“DSK”，而SD卡，NANDFLASH的設(shè)備類型也是DSK，所以從設(shè)備消息里讀出設(shè)備名如“DSK1”、“DSK2”后并不能直觀確定哪個是U盤的設(shè)備名。
這時可以調(diào)用存儲器相關(guān)API函數(shù)OpenStore根據(jù)設(shè)備名，查詢U盤盤符等信息。
必要的頭文件及l(fā)ib庫
#include "Storemgr.h"
#pragma comment(lib,"Storeapi.lib")
封裝OpenStore函數(shù)
增加超時是有必要的，因為設(shè)備加載消息來得更快，可能存儲設(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;
}
獲得詳細(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 ", detail.d.szName, StoreInfo.szStoreName, detail.d.fAttached ? L"appeared" : L"was removed", flags);
用鏈表存儲設(shè)備信息在U盤加載時，可以通過訪問存儲驅(qū)動API來查詢設(shè)備信息。但是當(dāng)設(shè)備移除后，設(shè)備信息就不存在于驅(qū)動中了，所以用OpenStore和GetStoreInfo就無法查到已移除的設(shè)備信息了。
想要知道移除的設(shè)備詳細(xì)信息，就只有在加載的時候?qū)⑿畔⒋嬖跀?shù)據(jù)結(jié)構(gòu)中。這樣在移除時，通過查詢數(shù)據(jù)結(jié)構(gòu)中的數(shù)據(jù)，就可以獲得設(shè)備詳細(xì)信息了。
下載文檔
