VC++实现启用与停用设备

前些日子在研究USB设备，就顺便研究了一些如何在应用程序实现USB设备的启用与停用，然后稍微深入，于是有了本文。

想要实现类似设备管理器的功能，其实也不是很难，无非就是调用一些API函数，就像本文描述的，采用的API函数就是SetupDi系列的函数。不过这类函数有很多，具体的请参见MSDN，而实现设备启用、停用仅需要用到的就只有5个函数：
       SetupDiGetClassDevs                   // 获取设备信息集
       SetupDiEnumDeviceInfo                 // 从设备信息集中枚举每个设备的具体信息
       SetupDiGetDeviceRegistryProperty      // 从注册表中读取PnP设备的属性
       SetupDiSetClassInstallParams          // 设置（包括取消）设备类的安装参数
       SetupDiCallClassInstaller             // 安装指定设备
以上函数均在setupapi.h头文件中声明，该头文件包含在setupapi.lib函数库中（使用以上函数前需要声明这个头文件）。

接下来就是如何实现设备的启用与停用。
从原理上讲，设备的启用与停用其实就是对该设备进行重安装。

首先，我们需要声明两个变量用来保存指定设备类的属性信息：
    HDEVINFO        m_hDevInfo;              // 类似设备句柄，以下暂且称为设备句柄
    SP_DEVINFO_DATA m_DeviceInfoData;        // 设备详细属性信息

然后调用SetupDiGetClassDevs函数获取设备句柄的值。（在这个函数中，需要指定设备类的GUID，如果不清楚这个GUID，可以在相应的安装文件.inf中查找。注意：是设备类的GUID，不是设备的GUID！）

接着循环使用SetupDiEnumDeviceInfo函数枚举对应设备类中的设备，并使用SetupDiGetDeviceRegistryProperty函数获取得到的设备的详细信息，进行判断是否为所需的设备（判断的方式有多种，具体参考MSDN，本文采用设备描述进行判断）。
一旦枚举结束（即枚举不成功，而且用GetLastError()可以得到错误码259）即可退出循环。当然如果找到设备，即可break退出。

如果找到对应的设备，就调用SetupDiSetClassInstallParams函数设置安装的属性。这里有个注意的地方需要详细说明一下：
SetupDiSetClassInstallParams的函数原型如下：
WINSETUPAPI BOOL WINAPI
  SetupDiSetClassInstallParams(
    IN HDEVINFO  DeviceInfoSet,
    IN PSP_DEVINFO_DATA  DeviceInfoData,  OPTIONAL
    IN PSP_CLASSINSTALL_HEADER  ClassInstallParams,  OPTIONAL
    IN DWORD  ClassInstallParamsSize
    );
注意第三个参数PSP_CLASSINSTALL_HEADER  ClassInstallParams，
这里我们不采用这个结构，而是采用另外一个结构：SP_PROPCHANGE_PARAMS
并在这个结构中，
设置ClassInstallHeader字段中（我们发现这个字段也是一个结构，就是PSP_CLASSINSTALL_HEADER结构）的InstallFunction字段值为DIF_PROPERTYCHANGE，
设置StateChange值为DICS_ENABLE（该值为启用，若是停用则为DICS_DISABLE）
然后采用强行转换将其转为PSP_CLASSINSTALL_HEADER结构。

最后，调用SetupDiCallClassInstaller函数执行设备的安装（即：启用或者停用），注意该函数第一个参数值应为DIF_PROPERTYCHANGE。
从设备管理器中，可以验证我们的做法。

以下贴出我程序中的主要源码。

  BOOL               rlt            = FALSE;
  CString            csFriendlyName = myDevcieName;                           // 请修改该值
  GUID               devGUID        = {XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}; // 这个值不用我再说了吧？！
  HDEVINFO           m_hDevInfo     = INVALID_HANDLE_VALUE;
  SP_DEVINFO_DATA    m_DeviceInfoData;

  RtlZeroMemory(&m_DeviceInfoData, sizeof(SP_DEVINFO_DATA));                  // 初始化m_DeviceInfoData，当然这只是范例，
  m_DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);                          // 有其他初始化的方法，大家见仁见智

  m_hDevInfo = SetupDiGetClassDevs(&devGUID,0,0,DIGCF_PRESENT );
 
  for (DWORD i = 0; SetupDiEnumDeviceInfo ( m_hDevInfo, i, &m_DeviceInfoData ); i++ )
  {
         DWORD DataT;
         LPTSTR buffer  = NULL;
         DWORD buffersize = 0;
        
         while ( !SetupDiGetDeviceRegistryProperty
              (
                   m_hDevInfo,
                   &m_DeviceInfoData,
                   SPDRP_DEVICEDESC,
                   &DataT,
                   (PBYTE)buffer,
                   buffersize,
                   &buffersize
         ) )
        {
            if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
            {
                 // Change the buffer size.
                 if (buffer) 
                 {
                       LocalFree(buffer);
                 }
                 buffer = (char*)LocalAlloc(LPTR,buffersize);
            }
            else
            {
                 break;
            }
        }

        if ( ( buffer != NULL ) && ( csFriendlyName.Find ( buffer, 0 ) != ( -1 ) ) )
        {
            if (buffer) 
            {
                 LocalFree(buffer);
            }
            break;
        }
                
        if (buffer) 
        {
            LocalFree(buffer);
        }
  }

  SP_PROPCHANGE_PARAMS propChange = { sizeof ( SP_CLASSINSTALL_HEADER ) };
  propChange.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
  propChange.Scope = DICS_FLAG_GLOBAL;
  propChange.StateChange = DICS_ENABLE;   // 启用，或是停用，请使用DICS_DISABLE

  if (m_DeviceInfoData.DevInst != NULL)
  {
        rlt = SetupDiSetClassInstallParams
        (      
                m_hDevInfo, 
                &m_DeviceInfoData, 
                ( SP_CLASSINSTALL_HEADER * ) &propChange, 
                sizeof ( propChange ) 
        );
  }

  if ( rlt )
  {
        rlt = SetupDiCallClassInstaller ( DIF_PROPERTYCHANGE, m_hDevInfo, &m_DeviceInfoData );
  }