禁用设备

来源:岁月联盟 编辑:zhu 时间:2008-09-08

 参考

  Device Management Functions

// 必要的头文件和要链接的LIB文件
#include <setupapi.h>
#include <shlwapi.h>
#pragma comment(lib, "setupapi.lib")
#pragma comment(lib, "shlwapi.lib") // device information set(我把它译为设备信息集)
HDEVINFO hDevInfo = NULL;// 出错信息
void FormatMSG(DWORD dwError, LPTSTR * lpszMsg)
{  
  BOOL bOk = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
   NULL, dwError,
   MAKELANGID(LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED), (LPTSTR)lpszMsg, 0, NULL); 
  if (!bOk)
  {
    HMODULE hDll = LoadLibraryEx(_T("netmsg.dll"), NULL, DONT_RESOLVE_DLL_REFERENCES);
    if (NULL != hDll)
    {
      FormatMessage(FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_FROM_SYSTEM,
       hDll, dwError,
       MAKELANGID(LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED),(LPTSTR)lpszMsg,0, NULL);
      FreeLibrary(hDll);  
    } 
  }
}
<!--[if !supportEmptyParas]--><!--[endif]-->
BOOL ChangeStatus(DWORD NewStatus, DWORD SelectedItem, HDEVINFO hDevInfo)
{ 
  LPTSTR lpszMsg = NULL; HCURSOR hCursor = NULL;
  try 
  {  
    SP_PROPCHANGE_PARAMS PropChangeParams = {sizeof(SP_CLASSINSTALL_HEADER)};  
    SP_DEVINFO_DATA DeviceInfoData = {sizeof(SP_DEVINFO_DATA)};   
    hCursor = SetCursor(LoadCursor(NULL, IDC_WAIT));   //Get a handle to the Selected Item.  
    if (!SetupDiEnumDeviceInfo(hDevInfo, SelectedItem, &DeviceInfoData))  
    {   
      FormatMSG(GetLastError(), &lpszMsg);   
      throw lpszMsg;  
    }  //Set the PropChangeParams structure.  
    PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
    PropChangeParams.Scope = DICS_FLAG_GLOBAL;  
    PropChangeParams.StateChange = NewStatus;  
    if (!SetupDiSetClassInstallParams(hDevInfo, &DeviceInfoData,
       (SP_CLASSINSTALL_HEADER *)&PropChangeParams,sizeof(PropChangeParams)))  
    {   
      FormatMSG(GetLastError(), &lpszMsg);   
      throw lpszMsg;  
    }  //Call the ClassInstaller and perform the change.  
    if (!SetupDiCallClassInstaller(DIF_PROPERTYCHANGE,hDevInfo,&DeviceInfoData))  
    {   
      FormatMSG(GetLastError(), &lpszMsg);   
      throw lpszMsg;  
    }  
    SetCursor(hCursor);   
    return TRUE; 
  } 
  catch (TCHAR * pszError) 
  {  
    SetCursor(hCursor);  
    ::MessageBox(NULL,pszError,_T("提示"),MB_OK);  
    if (NULL != lpszMsg)  
    {   
      LocalFree((HLOCAL)lpszMsg);  
    }   
    return FALSE; 
  }
}
// 这些设备的启用和禁用主要有此函数实现
// 参数说明,nStatus 可取3个值,-1 :啥也不做// 0 禁用设备,1启用设备。
// nIndex用于控制是对光驱,软驱还是对USB启用或禁用
// 也是可取3个值,0 代表软驱,1 代表光驱 2 代表USB
// 譬如要对光驱实现禁用,可以这样调用此函数
// ControlDisk(0, 1);
bool ControlDisk(int nStatus, int nIndex)
{ 
  if (-1 == nStatus) 
  {   
    return FALSE; 
  }  
  LPTSTR lpszMsg = NULL; 
  try 
  {  
    TCHAR * GUIDString = NULL;  
    GUID guid;  
    ZeroMemory(&guid, sizeof(GUID));  
    switch(nIndex)  
    {   
    case 0: // 0 代表软驱    
      GUIDString = _T("4D36E980-E325-11CE-BFC1-08002BE10318");    
      UuidFromString((unsigned char *)GUIDString, &guid);    
      break;    
    case 1: // 1 代表光驱     
      GUIDString = _T("4D36E965-E325-11CE-BFC1-08002BE10318");    
      UuidFromString((unsigned char *)GUIDString, &guid);    
      break;   
    case 2: // 2 代表USB    
      GUIDString = _T("36FC9E60-C465-11CF-8056-444553540000");    
      UuidFromString((unsigned char *)GUIDString, &guid);    
      break;
    default:;  
    }  
    hDevInfo = SetupDiGetClassDevs(&guid,NULL,NULL,DIGCF_PRESENT);  
    if (INVALID_HANDLE_VALUE == hDevInfo)  
    {   
      FormatMSG(GetLastError(), &lpszMsg);   
      throw lpszMsg;  
    }  
    DWORD i;  
    SP_DEVINFO_DATA DeviceInfoData;  
    ZeroMemory(&DeviceInfoData, sizeof(SP_DEVINFO_DATA));  
    DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);  
    for (i = 0; SetupDiEnumDeviceInfo(hDevInfo, i, &DeviceInfoData); ++i)  
    {   
      if (1 == nStatus)   
      {    
        StateChange(DICS_ENABLE, i, hDevInfo);   
      }   
      else if (0 == nStatus)   
      {    
        StateChange(DICS_DISABLE, i, hDevInfo);   
      }   
    }  // 释放 device information set  
    return SetupDiDestroyDeviceInfoList(hDevInfo); 
  } 
  catch (TCHAR * pszError) 
  {  
    ::MessageBox(NULL,pszError,_T("提示"),MB_OK);  
    if (NULL != lpszMsg)  
    {   
      LocalFree((HLOCAL)lpszMsg);  
    }   
    return FALSE; 
  } 
  return FALSE;
}

  经测试这样对光驱,软驱和USB实现禁用没有问题,但是当禁用过之后如果要对USB实现启用必须两次调用ControlDisk(1, 2);这个函数才可,其余的两个启用没什么问题。还有要说的就是我不是用的枚举所有的设备,然后再过滤(网上的那个禁用网卡的就是这种方法)。

  据我观察,在注册表的此项下下:

  HKEY_LOCAL_MACHINESYSTEMCurrentControlSetControlClass

  有很多GUID,每个GUID代表一个设备,譬如这个是USB的

{36FC9E60-C465-11CF-8056-444553540000}

  这个是CDROM的

{4D36E965-E325-11CE-BFC1-08002BE10318}

上一篇:枚举系统设备
下一篇:被误解的C++