C++检查设备驱动状态

8427阅读　0评论2011-09-10　zpf1218
分类：C/C++

  1. bool IsDeviceDisabled(DWORD dwDevID, HDEVINFO hDevInfo, DWORD &dwStatus)
  2. {
  3.     SP_DEVINFO_DATA DevInfoData = {sizeof(SP_DEVINFO_DATA)};
  4.     DWORD dwDevStatus,dwProblem;
  5.     if(!SetupDiEnumDeviceInfo(hDevInfo,dwDevID,&DevInfoData))
  6.     {
  7.         return FALSE;
  8.     }

  10.     //查询设备状态
  11.     if(CM_Get_DevNode_Status(&dwDevStatus,&dwProblem,DevInfoData.DevInst,0)!=CR_SUCCESS)
  12.     {
  13.         return FALSE;
  14.     }
  15.     
  16.     dwStatus = dwProblem;
  17. //    return ( (dwProblem == CM_PROB_FAILED_INSTALL));

  19.     return true;
  20. }

  22. int IsInstallDriver()
  23. {
  24.     HDEVINFO hDevInfo;
  25.     SP_DEVINFO_DATA DeviceInfoData;
  26.     DWORD i;
  27.     bool bRet = false;
  28.     bool bOk = false;

  30.       //step1. Create a HDEVINFO with all present devices.
  31.     hDevInfo = SetupDiGetClassDevs(NULL,
  32.         0, // Enumerator
  33.         0,
  34.         DIGCF_PRESENT | DIGCF_ALLCLASSES );

  36.     if (hDevInfo == INVALID_HANDLE_VALUE)
  37.     {
  38.         // Insert error handling here.
  39.         return bRet;
  40.     }

  42.      
  43.     DWORD dwStatuts = -1;
  44.     //step2. Enumerate through all devices in Set.
  45.     DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
  46.     for (i=0; SetupDiEnumDeviceInfo(hDevInfo, i, &DeviceInfoData); i++)
  47.     {
  48.         DWORD DataT;
  49.         LPTSTR buffer = NULL;
  50.         DWORD buffersize = 0;
  51.    
  52.         // Call function with null to begin with,
  53.         // then use the returned buffer size
  54.         // to Alloc the buffer. Keep calling until
  55.         // success or an unknown failure.
  56.         //
  57.         while (!SetupDiGetDeviceRegistryProperty(
  58.             hDevInfo,
  59.             &DeviceInfoData,
  60.             SPDRP_HARDWAREID,
  61.             &DataT,
  62.             (PBYTE)buffer,
  63.             buffersize,
  64.             &buffersize))
  65.         {
  66.             if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
  67.             {
  68.                 // Change the buffer size.
  69.                 if (buffer) LocalFree(buffer);
  70.                 buffer = (LPTSTR)LocalAlloc(LPTR,buffersize);
  71.             }
  72.             else
  73.             {
  74.                 // Insert error handling here.
  75.                 break;
  76.             }
  77.         }

  79.         printf("%S\n", buffer);
  80.         //step3. find devices status
  81.         if (buffer && (!wcscmp(buffer,L"USB\\Vid_0955&Pid_7103") || !wcscmp(buffer,L"USB\\Vid_0955&Pid_7102")) )
  82.         {
  83.             
  84.             if (IsDeviceDisabled(i, hDevInfo, dwStatuts) && dwStatuts == 0)
  85.             {
  86.     
  87.             }
  88.             //printf( "SPDRP_DEVICEDESC:[%S] %d\n ",buffer, dwStatuts);

  90.             if (buffer)
  91.                 LocalFree(buffer);
  92.             break;
  93.         }

  95.         if (buffer)
  96.             LocalFree(buffer);
  97.     }

  99.     // step4. Cleanup
  100.     SetupDiDestroyDeviceInfoList(hDevInfo);
  101.     
  102.     return dwStatuts;
  103. }
上一篇：委托模式
下一篇：lsof