硬件设备的监控处理之三——（USB设备的禁用和启用）

两种方法，

方法一 查找注册表

 

 

思路：首先枚举注册表路径"HKEY_LOCAL_MACHINE//Drivers//Active"下的子键，从子键中查找键项"Key"的值是否包含"Drivers//USB//ClientDrivers//Mass_Storage_Class"字段，如果包含再取该键的键项"Name"的值，获取名字devName后调用如下代码，实现禁用

 

HANDLE hnd = OpenStore(devName);//(L"DSK1:")

if(hnd == NULL)

{

::MessageBox(NULL, L"USB打开失败", L"notice", MB_OK );

return ;

}

BOOL ret = DismountStore(hnd);

CloseHandle(hnd);

 

流程：

    启动对话框的时候先调用EnumActiveDir枚举注册表项"HKEY_LOCAL_MACHINE//Drivers//Active"下的子键，然后调用 SearchAndDismountUsbStro(PREG_ITEM_NAME change_itemStruct)函数对USB存储设备进行dismount

    响应窗口回调函数WindowProc，当接收到WM_DEVICECHANGE消息时，判断是否接入或者移除了USB存储设备，弹出提示对话框，根据用户的选择来dismount或者不处理。

    启用的话就是重新插入设备

相关函数：

SearchAndDismountUsbStro

#include "storemgr.h" void SearchAndDismountUsbStro(PREG_ITEM_NAME change_itemStruct)//查找并delete. { PREG_ITEM_NAME tempchange_itemStruct = change_itemStruct; while(tempchange_itemStruct) { char szKeyName[10] = {0}; strcpy(szKeyName, tempchange_itemStruct->szItemName); ZeroMemory(&szKeyName[strlen(szKeyName)+1],10-strlen(szKeyName)-1); TCHAR szUsbKey[MAX_PATH] = TEXT("Drivers//Active//"); wcscat(szUsbKey,short2wide(szKeyName)); HKEY hUsbKey; if(ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE,szUsbKey,0,0,&hUsbKey)) { DWORD keyType; DWORD dwLength = 100; TCHAR buf[MAX_PATH] = {0}; if(ERROR_SUCCESS == RegQueryValueEx(hUsbKey,TEXT("Key"),0,&keyType,(BYTE *)buf,&dwLength)) if(strstr(wide2short(buf), "Drivers//USB//ClientDrivers//Mass_Storage_Class"))//判断是否是USB存储设备 { TCHAR devName[MAX_PATH] = {0}; if(ERROR_SUCCESS == RegQueryValueEx(hUsbKey,TEXT("Name"),0,&keyType,(BYTE *)devName,&dwLength)) { HANDLE hnd = OpenStore(devName);//(L"DSK1:") if(hnd == NULL) { ::MessageBox(NULL, L"USB打开失败", L"notice", MB_OK ); return ; } BOOL ret = DismountStore(hnd); CloseHandle(hnd); } } } RegCloseKey(hUsbKey); tempchange_itemStruct = tempchange_itemStruct->pNext; } } 

EnumActiveDir

typedef struct __REG_ITEM_NAME { int num;//编号 char szItemName[MAX_PATH];//键值内容 __REG_ITEM_NAME *pNext; } REG_ITEM_NAME, *PREG_ITEM_NAME; #define REGITEMLEN sizeof(struct __REG_ITEM_NAME) PREG_ITEM_NAME EnumActiveDir() { PREG_ITEM_NAME p,last,head; HKEY hKey; LONG lResult ; TCHAR szKeyName[64] ; if(ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE,TEXT("Drivers//Active"),0,0,&hKey)) { int count=0; head=last=NULL; p = (PREG_ITEM_NAME)malloc(REGITEMLEN); do { DWORD dwKeynameLength = MAX_PATH; lResult = RegEnumKeyEx(hKey,count,szKeyName,&dwKeynameLength,0,NULL,NULL,NULL); if(lResult != ERROR_SUCCESS) { break; } p->num = count; strcpy(p->szItemName, wide2short(szKeyName)); p->pNext=NULL; ++count; if(count == 1) head=p; else last->pNext=p; last=p; p = (PREG_ITEM_NAME)malloc(REGITEMLEN); } while (lResult != ERROR_NO_MORE_ITEMS); RegCloseKey(hKey); free(p); return head; } return NULL; } 

WindowProc

LRESULT CDlgUsbCtrlDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) { if(message == WM_DEVICECHANGE) { if ( DBT_DEVICEARRIVAL == wParam || DBT_DEVICEREMOVECOMPLETE == wParam ) { PDEV_BROADCAST_HDR pHdr = (PDEV_BROADCAST_HDR)lParam; switch( pHdr->dbch_devicetype ) { case DBT_DEVTYP_PORT: if(DBT_DEVICEARRIVAL == wParam) { change_itemStruct = EnumActiveDir(); if(::MessageBox(NULL, L"有新增的USB设备，是否禁用", L"Notice", MB_OKCANCEL)==IDOK) SearchAndDismountUsbStro(change_itemStruct); else MessageBox(L"用户选择了运行该USB存储设备加入"); } else if(DBT_DEVICEREMOVECOMPLETE == wParam) { //count_afterchange = EnumActiveDir(); //MessageBox(L"移除USB存储设备"); } break; default: break; } } } return CDialog::WindowProc(message, wParam, lParam); }  

 

方法二 存储设备处理

 

 

思路：

禁用：

    USB存储设备，如果存储设备被打开其它人就没办法使用即可实现禁用效果，并且将"System//StorageManager//Profiles//USBHDProfile";中写入键项  MountHidden= 1，dismount掉SD卡

启用：

    将"System//StorageManager//Profiles//SDMemory";中写入键项  MountHidden= 0，mountSD卡

     重新插入SD卡即可启用

 

 

禁用USB存储设备

void DisableUsbStorage() { STOREINFO StoreInfo; memset(&StoreInfo, 0, sizeof(STOREINFO)); //切记定义后要初始化，否则后面将查找不到任何存储器 StoreInfo.cbSize = sizeof(STOREINFO); HANDLE hDevice = FindFirstStore( &StoreInfo ); if( INVALID_HANDLE_VALUE != hDevice ) { do { //判断是否为USB存储设备 if( 0 == wcscmp( StoreInfo.szStoreName, TEXT("USB Hard Disk Drive") ) ) { //如果查找到指定存储设备，则打印信息到调试串口 wchar_t * devName = StoreInfo.szDeviceName; HANDLE hnd = OpenStore(devName);//(L"DSK1:") if(hnd == NULL) { //::MessageBox(NULL, L"USB打开失败", L"notice", MB_OK ); const char * notice = "USB打开失败！"; log(notice, strlen(notice)); return ; } BOOL ret = DismountStore(hnd); CloseHandle(hnd); break ; } }while( FindNextStore( hDevice, &StoreInfo ) ); } CloseHandle(hDevice); DWORD dwdata = 1; wchar_t *skeyName = L"System//StorageManager//Profiles//USBHDProfile"; if(WriteReg(dwdata, skeyName)) { const char * notice = "DisableUsbStorage写入MountHidden= 1成功！"; //log(notice, strlen(notice)); const char * notice1 = "成功禁用usb存储设备！"; log(notice1, strlen(notice1)); } else { const char * notice = "DisableUsbStorage写入MountHidden= 1失败！"; //log(notice, strlen(notice)); const char * notice1 = "禁用usb存储设备失败！"; log(notice1, strlen(notice1)); } } 

启用USB存储设备

void EnableUsbStorage() { DWORD dwdata = 0; wchar_t *skeyName = L"System//StorageManager//Profiles//USBHDProfile"; if(WriteReg(dwdata, skeyName)) { const char * notice = "EnableUsbStorage写入MountHidden= 0成功！"; //log(notice, strlen(notice)); const char * notice1 = "成功启用usb存储设备！"; log(notice1, strlen(notice1)); } else { const char * notice = "EnableUsbStorage写入MountHidden= 0失败！"; //log(notice, strlen(notice)); const char * notice1 = "启用usb存储设备失败！"; log(notice1, strlen(notice1)); } }