[转载]光驱，软驱，USB的禁用和启用

天云上人 · 发表于 2005-3-16 11:27:43

  (dlutyuanhongl发表于2005-3-15 23:14:04)

这几天，由于工作的实际需要，需要对光驱，软驱，USB的启用和禁用实现控制。参考大家提供的对网卡禁用的代码,实现了光驱，软驱，USB的启用和禁用。主要实现代码如下:
// 必要的头文件和要链接的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);
  }
}
}

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 *)&

ropChangeParams,
           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;
  }

  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_MACHINE\SYSTEM\CurrentControlSet\Control\Class\
   有很多GUID，每个GUID代表一个设备，譬如这个是USB的
   {36FC9E60-C465-11CF-8056-444553540000}
   这个是CDROM的
   {4D36E965-E325-11CE-BFC1-08002BE10318}等等了。
这些值你即使把它改为别的GUID，我使用原来的GUID仍然可以实现我的功能。

		自动登录	找回密码
密码			注册