硬件设备的监控处理之三——（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;elselast->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));}}