现在的位置: 首页 > 综合 > 正文

一个可以禁用USB存储设备的程序(SetupAPI的使用方法)

2013年04月06日 ⁄ 综合 ⁄ 共 12780字 ⁄ 字号 小 中 大 ⁄ 评论关闭

 

  这个程序主要是通过SetupAPI和DDK中的CM_XX函数族对设备进行枚举、状态查询和控制。下面先是对使用的SetupAPI函数的说明，然后是程序代码，该程序其实可以禁用任何可以被禁用的设备，只不过我没有去做罢了，运行第一次会将USB Mass Storage设备都禁用，第二次会将它们都启用。请使用DDK编译。或者在VC中使用driverstdio将环境设为DDK环境编译

SetupDiGetClassDevs如果第二个参数设为NULL,第四个参数为DIGCF_ALLCLASSES，则返回所有设备信息。

 

SetupDiGetClassImageList建立一个包含所有已经安装的类位图的映像列表并返回含有这个列表的数据结构：SP_CLASSIMAGE_DATA

  1. typedef struct _SP_CLASSIMAGE_DATA {
  3.   DWORD  cbSize;
  5.   HIMAGELIST  ImageList;
  7.   DWORD  Reserved;
  9. } SP_CLASSIMAGE_DATA, *PSP_CLASSIMAGE_DATA;

ImageList映像列表的句柄

CbSize必须先初始化，sizeof(SP_CLASSIMAGE_DATA)

 

SetupDiEnumDeviceInfo为一个设备的设备信息集单元返回上下文结构：SP_DEVINFO_DATA

  1. typedef struct _SP_DEVINFO_DATA {
  3.  DWORD  cbSize;
  5.  GUID   ClassGuid;
  7.  DWORD  DevInst;
  9.  ULONG_PTR  Reserved;
  11. } SP_DEVINFO_DATA, *PSP_DEVINFO_DATA;


该结构定义了一个设备信息集成员的设备实例

ClassGuid设备的安装GUID

DevInst不透明的设备实例句柄(??我也不太清楚，反正DDK上英文这么说的)

 

CM_Get_DevNode_Status用来获得设备实例的状态

第三个输入参数就可以是上面结构的DevInst

 

SetupDiOpenClassRegKey打开设备类的主键或者指定类的子键

 

SetupDiGetDeviceRegistryProperty获得指定的即插即用的设备属性

 

SetupDiGetClassImageIndex获得指定设备在设备映像列表中的索引

 

SetupDiGetDeviceInstallParams获得指定设备的安装信息

 

SetupDiSetClassInstallParams为设备信息集或者是一个实际的设备信息单元设置或清除类安装参数,需要传入SP_PROPCHANGE_PARAMS结构指针，这个结构是负责协调DIF_PROPERTYCHANGE 安装请求

 

SetupDiCallClassInstaller根据DIF代码调用安装服务

  1. /*---------------------------------------------------------common.h----------------------------------------------------------------*/
  3. #include <stdio.h>
  4. #include <malloc.h>
  5. #include <tchar.h>
  6. #include <string.h>
  7. #include <windows.h>
  8. #include <setupapi.h>
  9. #include <cfgmgr32.h>
  10. #include <regstr.h>
  12. #define UnknownDevice TEXT("<Unknown Device>")
  13. #define USBDISK      TEXT("Mass")
  15. //设备信息结构链表
  16. typedef struct _DEVICE_INFO
  17. {
  18.  SP_DEVINFO_DATA  WY_DevInfoData; //设备信息包括实例句柄 和设备类的GUID
  19.  DWORD     WY_DevID;   //在此次全局的枚举中设备ID
  20.  TCHAR               *WY_lpszDevName; //设备的名称
  21.  ULONG     WY_ulDevNameLen; //设备名称的长度
  22.  BOOL     WY_isDisableable; //是否可以被禁用
  23.  BOOL     WY_isDisabled;  //是否已经被禁用
  24.  struct _DEVICE_INFO        *WY_pNextDevInfo;  //下一个结构块
  25. }WY_DEVICE_INFO,*WY_pDEVICE_INFO;
  27. TCHAR WY_tcDebugStr[1024];     //输出的调试信息
  28. //StateChange函数用来改变设备的状态，可以设为启用或者禁用
  29. BOOL StateChange(DWORD WY_dwNewState,DWORD WY_dwDevID,HDEVINFO WY_hDevInfo);
  30. //EnumDevice函数用来枚举计算机中所包含的所有设备
  31. BOOL EnumDevices(HWND WY_hDevInfo);
  32. //GetRegistryProperty函数用来获得设备注册表中的内容
  33. BOOL GetRegistryProperty(HDEVINFO WY_hDevInfo,PSP_DEVINFO_DATA WY_DevInfoData,ULONG WY_ulProperty,PVOID WY_pBuffer,PULONG WY_pulBufLen);
  34. //ConstructDeviceName函数用来构造设备名称
  35. BOOL  ConstructDeviceName( HDEVINFO  DeviceInfoSet,PSP_DEVINFO_DATA  DeviceInfoData,PVOID Buffer, PULONG  Length);
  36. //EnableDevice函数用来启用设备
  37. BOOL EnableDevice(DWORD WY_dwDevID,HDEVINFO WY_hDevInfo);
  38. //DisableDevice函数用来禁止设备
  39. BOOL DisableDevice(DWORD WY_dwDevID,HDEVINFO WY_hDevInfo);
  40. //查询该设备是否可以被禁用
  41. BOOL IsDisableable (DWORD WY_dwDevID, HDEVINFO WY_hDevInfo);
  42. //查询该设备是否已经禁用
  43. BOOL  IsDisabled (DWORD WY_dwDevID, HDEVINFO WY_hDevInfo);
  44. //GUID结构复制函数，
  45. //因为sdk的函数的头文件不知道为什么不能在DDK的环境下使用，只好自己写
  46. void  GuidCopy(GUID WY_guidSrc,GUID *WY_guidDest);
  47. //控制USB设备
  48. BOOL ControlDevice(DWORD WY_dwDevID,HDEVINFO WY_hDevInfo);
  50. /*----------------------------------------------devctrl.c----------------------------------------------------------------------------------*/
  52. #include "common.h"
  54. WY_DEVICE_INFO WY_DevInfo;  //设备链表头
  56. int APIENTRY WinMain(HINSTANCE WY_hInstance,HINSTANCE WY_hPreInstance,LPSTR WY_lpCmd,int WY_nCmdNum)
  57. {
  58.  HDEVINFO   WY_hDevInfo;
  59.  WY_pDEVICE_INFO WY_pDevInfo;
  60.  //返回设备信息
  61.  WY_hDevInfo = SetupDiGetClassDevs(NULL,NULL,NULL,DIGCF_PRESENT|DIGCF_ALLCLASSES);
  62.  _stprintf(WY_tcDebugStr,"WY_hDevInfo %x",WY_hDevInfo);
  63.  OutputDebugString(WY_tcDebugStr);
  64.  
  65.  WY_DevInfo.WY_DevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
  66.  WY_DevInfo.WY_ulDevNameLen = 256;
  67.  WY_DevInfo.WY_pNextDevInfo = NULL;
  68.  //枚举系统设备
  69.  EnumDevices(WY_hDevInfo);
  70.  WY_pDevInfo = WY_DevInfo.WY_pNextDevInfo;
  71.  while(WY_pDevInfo)
  72.  { //从系统设备中查找usb mass storage设备
  73.   _stprintf(WY_tcDebugStr,"DevID : %d GUID:%x-%x-%x-%x%x%x%x%x%x%x%x DevInst : %x,DevName:%s IsDisableable:%d, IsDisabled:%d"
  74.                     ,WY_pDevInfo->WY_DevID
  75.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data1
  76.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data2
  77.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data3
  78.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[0]
  79.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[1]
  80.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[2]
  81.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[3]
  82.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[4]
  83.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[5]
  84.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[6]
  85.                     ,WY_pDevInfo->WY_DevInfoData.ClassGuid.Data4[7]
  86.                     ,WY_pDevInfo->WY_DevInfoData.DevInst
  87.                     ,WY_pDevInfo->WY_lpszDevName
  88.                                ,WY_pDevInfo->WY_isDisableable
  89.                     ,WY_pDevInfo->WY_isDisabled);
  91.    OutputDebugString(WY_tcDebugStr);
  92.   if(strstr(WY_pDevInfo->WY_lpszDevName,USBDISK))
  93.   {
  94.    
  95.    //控制usb mass storage 设备的状态，如果启用则禁用，如果仅用则启用
  96.    if(ControlDevice(WY_pDevInfo->WY_DevID,WY_hDevInfo))
  97.    {
  98.     OutputDebugString("Successful");
  99.    }
  100.    else OutputDebugString("FAILED");
  101.     
  102.   }
  103.   WY_pDevInfo = WY_pDevInfo->WY_pNextDevInfo;
  104.  }
  105.  SetupDiDestroyDeviceInfoList(WY_hDevInfo);
  106.  return 0;
  107. }
  109. BOOL StateChange(DWORD WY_dwNewState, DWORD WY_dwDevID, HDEVINFO WY_hDevInfo)
  110. {
  111.  SP_PROPCHANGE_PARAMS  WY_PropChangeParams;
  112.  SP_DEVINFO_DATA   WY_DevInfoData = {sizeof(SP_DEVINFO_DATA)};
  113.  SP_DEVINSTALL_PARAMS  WY_devParams;
  115.  //查询设备信息
  116.  if(!SetupDiEnumDeviceInfo(WY_hDevInfo,WY_dwDevID,&WY_DevInfoData))
  117.  {
  118.   OutputDebugString("SetupDiEnumDeviceInfo FAILED");
  119.   return FALSE;
  120.  }
  122.  //设置设备属性变化参数
  123.  WY_PropChangeParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
  124.  WY_PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
  125.  WY_PropChangeParams.Scope = DICS_FLAG_GLOBAL;  //使修改的属性保存在所有的硬件属性文件
  126.  WY_PropChangeParams.StateChange = WY_dwNewState;
  127.  WY_PropChangeParams.HwProfile = 0;
  128.  //改变设备属性
  129.  if(!SetupDiSetClassInstallParams(WY_hDevInfo,&WY_DevInfoData,(SP_CLASSINSTALL_HEADER *)&WY_PropChangeParams,sizeof(WY_PropChangeParams)))
  130.  {
  131.   OutputDebugString("SetupDiSetClassInstallParams FAILED");
  132.   return FALSE;
  133.  }
  134.  
  135.  WY_PropChangeParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
  136.  WY_PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
  137.  WY_PropChangeParams.Scope = DICS_FLAG_CONFIGSPECIFIC;//使修改的属性保存在指定的属性文件
  138.  WY_PropChangeParams.StateChange = WY_dwNewState;
  139.  WY_PropChangeParams.HwProfile = 0;
  140.  //改变设备属性并调用安装服务
  141.  if(!SetupDiSetClassInstallParams(WY_hDevInfo,&WY_DevInfoData,(SP_CLASSINSTALL_HEADER *)&WY_PropChangeParams,sizeof(WY_PropChangeParams))
  142.   || !SetupDiCallClassInstaller(DIF_PROPERTYCHANGE,WY_hDevInfo,&WY_DevInfoData))
  143.  {
  144.   OutputDebugString("SetupDiSetClassInstallParams or SetupDiCallClassInstaller FAILED");
  145.   return TRUE;
  146.  }
  147.  else
  148.  { //判断是否需要重新启动
  149.   WY_devParams.cbSize = sizeof(WY_devParams);
  150.   if(!SetupDiGetDeviceInstallParams(WY_hDevInfo,&WY_DevInfoData,&WY_devParams))
  151.   {
  152.    OutputDebugString("SetupDiGetDeviceInstallParams FAILED");
  153.    return FALSE;
  154.   }
  155.   if(!WY_devParams.Flags & (DI_NEEDRESTART|DI_NEEDREBOOT))
  156.   {
  157.    OutputDebugString("Need Restart Computer");
  158.    return TRUE;
  159.   }
  160.   return TRUE;
  161.  }
  162. }
  164. BOOL EnumDevices(HWND WY_hDevInfo)
  165. {
  166.  DWORD     WY_dwDevID,WY_dwStatus,WY_dwProblem;
  167.  SP_DEVINFO_DATA  WY_DevInfoData = {sizeof(SP_DEVINFO_DATA)};
  168.  WY_pDEVICE_INFO  WY_pDevInfo,WY_ppreDevInfo;
  169.  TCHAR         *WY_DeviceName;
  170.  ULONG     WY_ulNameLen = 256;
  171.  
  172.  WY_ppreDevInfo = &WY_DevInfo;
  173.  //穷举每个设备
  174.  for(WY_dwDevID = 0;SetupDiEnumDeviceInfo(WY_hDevInfo,WY_dwDevID,&WY_DevInfoData);WY_dwDevID++)
  175.  {
  176.   //构造设备信息块
  177.   WY_pDevInfo = (WY_pDEVICE_INFO)malloc(sizeof(WY_DEVICE_INFO));
  178.   WY_pDevInfo->WY_pNextDevInfo = NULL;
  179.   WY_pDevInfo->WY_DevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
  180.   WY_pDevInfo->WY_ulDevNameLen;
  181.   WY_pDevInfo->WY_DevID = WY_dwDevID;
  182.   GuidCopy(WY_DevInfoData.ClassGuid,&WY_pDevInfo->WY_DevInfoData.ClassGuid);
  183.   WY_pDevInfo->WY_DevInfoData.DevInst = WY_DevInfoData.DevInst;
  184.   WY_pDevInfo->WY_DevInfoData.Reserved = WY_DevInfoData.Reserved;
  185.   //这里有个问题，如果用设备信息块结构中的字符串指针
  186.   //SetupDiGetDeviceRegistyProperty函数就会返回数据无效，不知道为什么。
  187.   //所以下面只能在重新复制一遍了。
  188.   ConstructDeviceName(WY_hDevInfo,&WY_DevInfoData,&WY_DeviceName,&WY_ulNameLen);
  189.   WY_pDevInfo->WY_ulDevNameLen = WY_ulNameLen;
  190.   WY_pDevInfo->WY_lpszDevName = (TCHAR *)malloc(WY_ulNameLen*sizeof(TCHAR));
  191.   memcpy(WY_pDevInfo->WY_lpszDevName,WY_DeviceName,WY_ulNameLen);
  193.   WY_pDevInfo->WY_isDisableable = IsDisableable(WY_dwDevID, WY_hDevInfo);
  194.   WY_pDevInfo->WY_isDisabled = IsDisabled(WY_dwDevID, WY_hDevInfo);
  196.   WY_ppreDevInfo->WY_pNextDevInfo = WY_pDevInfo;
  197.   WY_ppreDevInfo = WY_pDevInfo;
  198.  }
  199.  return FALSE;
  200. }
  202. BOOL GetRegistryProperty(HDEVINFO WY_hDevInfo, PSP_DEVINFO_DATA WY_DevInfoData, ULONG WY_ulProperty, PVOID WY_pBuffer, PULONG WY_pulBufLen)
  203. {
  204.  //读取设备的注册表信息
  205.  while (!SetupDiGetDeviceRegistryProperty(WY_hDevInfo,
  206.         WY_DevInfoData,
  207.         WY_ulProperty,  //信息字段
  208.         NULL,
  209.         (PVOID)*(TCHAR **)WY_pBuffer,
  210.         *WY_pulBufLen,
  211.         WY_pulBufLen
  212.         ))
  213.     {
  214.         if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
  215.         { //如果长度不够则重新分配缓冲区
  216.          
  217.             if (*(LPTSTR *)WY_pBuffer)
  218.                 LocalFree(*(LPTSTR *)WY_pBuffer);
  219.             *(LPTSTR *)WY_pBuffer = LocalAlloc(LPTR,*WY_pulBufLen);
  220.         }
  221.         else
  222.         {
  223.            
  224.             if (GetLastError() != ERROR_INVALID_DATA)
  225.                OutputDebugString(TEXT("GetDeviceRegistryProperty"));
  226.       return FALSE;
  227.         }           
  228.     }
  230.     return (*(LPTSTR *)WY_pBuffer)[0];
  231. }
  233. BOOL ConstructDeviceName(HDEVINFO WY_DevInfoSet, PSP_DEVINFO_DATA WY_DevInfoData, PVOID WY_pBuffer, PULONG Length)
  234. {
  235.  //尝试通过各种类型的设备描述获得其设备名称
  236.  if (!GetRegistryProperty(WY_DevInfoSet,
  237.         WY_DevInfoData,
  238.         SPDRP_FRIENDLYNAME ,//设备的现实名称
  239.         WY_pBuffer,
  240.         Length))
  241.     {
  242.         if (!GetRegistryProperty(WY_DevInfoSet,
  243.             WY_DevInfoData,
  244.             SPDRP_DEVICEDESC ,//设备的描述
  245.             WY_pBuffer,
  246.             Length))
  247.         {
  248.             if (!GetRegistryProperty(WY_DevInfoSet,
  249.                 WY_DevInfoData,
  250.                 SPDRP_CLASS ,//设备类名
  251.                 WY_pBuffer,
  252.                 Length))
  253.             {
  254.                 if (!GetRegistryProperty(WY_DevInfoSet,
  255.                     WY_DevInfoData,
  256.                     SPDRP_CLASSGUID ,//设备的GUID
  257.                     WY_pBuffer,
  258.                     Length))
  259.                 {//什么都没找到就返回UnknownDevice
  260.                      *Length = (_tcslen(UnknownDevice)+1)*sizeof(TCHAR);
  261.                      *(LPTSTR *)WY_pBuffer = LocalAlloc(LPTR,*Length);
  262.                      _tcscpy(*(LPTSTR *)WY_pBuffer,UnknownDevice);
  263.                 }
  264.             }
  265.         }
  266.     }
  267.  return TRUE;
  268. }
  270. BOOL EnableDevice(DWORD WY_dwDevID, HDEVINFO WY_hDevInfo)
  271. {
  272.  return StateChange(DICS_ENABLE,WY_dwDevID,WY_hDevInfo);
  273. }
  275. BOOL DisableDevice(DWORD WY_dwDevID, HDEVINFO WY_hDevInfo)
  276. {
  277.  return StateChange(DICS_DISABLE,WY_dwDevID,WY_hDevInfo);
  278. }
  280. BOOL IsDisableable(DWORD WY_dwDevID, HDEVINFO WY_hDevInfo)
  281. {
  282.  SP_DEVINFO_DATA WY_DevInfoData = {sizeof(SP_DEVINFO_DATA)};
  283.  DWORD    WY_dwDevStatus,WY_dwProblem;
  285.  if(!SetupDiEnumDeviceInfo(WY_hDevInfo,WY_dwDevID,&WY_DevInfoData))
  286.  {
  287.   OutputDebugString("SetupDiEnumDeviceInfo FAILED");
  288.   return FALSE;
  289.  }
  290.  //查询设备状态
  291.  if(CM_Get_DevNode_Status(&WY_dwDevStatus,&WY_dwProblem,WY_DevInfoData.DevInst,0)!=CR_SUCCESS)
  292.  {
  293.   OutputDebugString("CM_GET_DevNode_Status FAILED");
  294.   return FALSE;
  295.  }
  297.  return ((WY_dwDevStatus & DN_DISABLEABLE) && (WY_dwProblem != CM_PROB_HARDWARE_DISABLED));
  298. }
  300. BOOL IsDisabled(DWORD WY_dwDevID, HDEVINFO WY_hDevInfo)
  301. {
  302.  SP_DEVINFO_DATA WY_DevInfoData = {sizeof(SP_DEVINFO_DATA)};
  303.  DWORD    WY_dwDevStatus,WY_dwProblem;
  305.  if(!SetupDiEnumDeviceInfo(WY_hDevInfo,WY_dwDevID,&WY_DevInfoData))
  306.  {
  307.   OutputDebugString("SetupDiEnumDeviceInfo FAILED");
  308.   return FALSE;
  309.  }
  310.  //查询设备状态
  311.  if(CM_Get_DevNode_Status(&WY_dwDevStatus,&WY_dwProblem,WY_DevInfoData.DevInst,0)!=CR_SUCCESS)
  312.  {
  313.   OutputDebugString("CM_GET_DevNode_Status FAILED");
  314.   return FALSE;
  315.  }
  317.  return ((WY_dwDevStatus & DN_HAS_PROBLEM) && (WY_dwProblem == CM_PROB_DISABLED));
  318. }
  320. void GuidCopy(GUID WY_guidSrc, GUID *WY_guidDest)
  321. {
  322.  WY_guidDest->Data1 = WY_guidSrc.Data1;
  323.  WY_guidDest->Data2 = WY_guidSrc.Data2;
  324.  WY_guidDest->Data3 = WY_guidSrc.Data3;
  325.  memcpy(WY_guidDest->Data4,WY_guidSrc.Data4,8);
  326. }
  328. BOOL ControlDevice(DWORD WY_dwDevID, HDEVINFO WY_hDevInfo)
  329. {
  330.  BOOL WY_isCanDisable;
  332.  WY_isCanDisable = (IsDisableable(WY_dwDevID,WY_hDevInfo) && (!IsDisabled(WY_dwDevID,WY_hDevInfo)));
  333.  if(WY_isCanDisable)
  334.   return DisableDevice(WY_dwDevID,WY_hDevInfo);
  335.  else
  336.   return EnableDevice(WY_dwDevID,WY_hDevInfo);
  337. }
  339. /*----------------------------------------Source文件(MAKEFILE文件所有DDK中都一样)------------------------------------*/
  341. TARGETNAME=devctrl
  342. TARGETTYPE=PROGRAM
  344. UMTYPE=windows
  345. UMBASE=0x1000000
  346. UMENTRY=winmain
  348. TARGETPATH=obj
  350. TARGETLIBS=$(SDK_LIB_PATH)/setupapi.lib /
  351.            $(SDK_LIB_PATH)/comctl32.lib /
  352.      $(SDK_LIB_PATH)/cfgmgr32.lib/
  355. INCLUDES=
  356.  
  357. SOURCES=     /
  358.         devctrl.c

 

返回
【上篇】
【下篇】

作者:

抱歉!评论已关闭.

返回首页

Copyright © 2013-2018 学步园  保留所有权利.
软文销售 QQ客服：2265327166 点击这里给我发消息（其他合作也可洽谈）
必威体育
必威电竞