用Setup系列函数完成驱动卸载安装[驱动安装卸载程序]
阅读量：6120 次
发布时间：2019-06-21
本文共 28893 字，大约阅读时间需要 96 分钟。
1 // InstallWDFDriver.cpp : Defines the entry point for the console application.  2 //  3   4 #include "stdafx.h"  5 #include "Shlwapi.h"  6   7 #pragma  comment(lib,"Shlwapi.lib")  8 #pragma comment (lib,"setupapi.lib")  9 #pragma comment(lib, "newdev.lib")   10 / 11  12 BOOL IsDeviceInstallInProgress(VOID); 13 int RemoveDriver(_TCHAR *HardwareID); 14 VOID UninstallWDMDriver(LPCTSTR theHardware); 15 BOOL UnicodeToAnsi(LPCWSTR Source, const WORD wLen, LPSTR Destination, const WORD sLen); 16 BOOL AnsiToUnicode(LPCSTR Source, const WORD sLen, LPWSTR Destination, const WORD wLen); 17 BOOL GetINFData(FILE *pFile); 18 VOID FindComma(LPSTR szData); 19 BOOL GetSectionData(FILE* pFile, const char* szKey, const char bIsVender); 20 BOOL IsInstalled(); 21 BOOL InstallClassDriver(LPCTSTR theINFName); 22 BOOL StartInstallWDMDriver(LPCTSTR theInfName) ; 23  24 BOOL FindExistingDevice(IN LPTSTR HardwareId); 25  26 VOID InitialGlobalVar(); 27  28  29 // 30  31 WORD g_wVender = 0;   32 WORD g_wHardware = 0;   33 TCHAR g_strVender[20][64] = {
    0};   34 TCHAR g_strHardware[20][64] = {
    0};   35 TCHAR g_strHID[MAX_PATH+1] = {
    0};   36 ///// 37  38 int _tmain(int argc, _TCHAR* argv[]) 39 { 40     WCHAR Wlp_USB_PATH[] = L"C:\\Windows\\System32\\drivers\\WLP_USB_Driver.sys"; 41     WCHAR New_Inf_Path[] = L"D:\\wlp driver\\WDF Driver _new\\win7\\x64\\cyusb3.inf"; 42     CHAR szInfPath[215] = {
    0};  43     FILE *fp = NULL; 44  45  46     printf("This Process is Driver Installtion Program...\n"); 47  48     if(PathFileExists(Wlp_USB_PATH)) 49     { 50        printf("Exists WLP_USB_Driver.sys!\n"); 51         52       UninstallWDMDriver(L"USB\\VID_0451&PID_AF32&REV_0000"); 53  54        55         56        if(DeleteFile(L"C:\\Windows\\System32\\drivers\\WLP_USB_Driver.sys")) 57        { 58            printf("WLP_USB_Driver.sys File has Deleted!\n"); 59        } 60  61       /* if(FindExistingDevice(L"USB\\VID_0451&PID_AF32&REV_0000")) 62        { 63           printf("找打了！\n"); 64  65           getch(); 66        } 67        else 68        { 69            printf("没找到！\n"); 70            getch(); 71        }*/ 72         73        printf("Get Started Install the New DLF Driver...\n"); 74  75       //  RemoveDriver(L"USB\\VID_0451&PID_AF32&REV_0000"); 76        while(IsDeviceInstallInProgress()) 77        { 78          printf("Has Driver Installtion Program is Processing!\n"); 79        } 80  81        printf("Get Started analyse inf File!\n"); 82  83        UnicodeToAnsi(New_Inf_Path, _tcslen(New_Inf_Path), szInfPath, 215);  84  85        if ((fopen_s(&fp, szInfPath, "r"))!=0)   86        {   87            _tprintf(_T("can not open file %s\n"), New_Inf_Path);   88            return 0;   89        }   90  91        GetINFData(fp);   92        fclose(fp);   93  94        // 安装WDM驱动   95  96        if (StartInstallWDMDriver(New_Inf_Path) == FALSE)   97        {   98            _tprintf(_T("Start Install WDF Driver failed\n"));  99            getch();100            return 0;  101        }  102 103        //if(CopyFile(L"D:\\wlp driver\\WDF Driver _new\\win7\\x64\\cyusb3.sys",L"C:\\Windows\\System32\\drivers\\cyusb3.sys",TRUE))104          //  printf("New DLF Driver successed!\n");105        getch();106        107 108     }109     else110     {111        printf("No Exists WLP_USB_Driver.sys!\n");112 113        printf("Get Started Install the New DLF Driver...\n");114 115       // RemoveDriver(L"USB\\VID_0451&PID_AF32&REV_0000");116        //while(IsDeviceInstallInProgress())117        //{118          //  printf("Has Driver Installtion Program is Processing!\n");119        //}120 121       printf("Get Started analyse inf File!\n");122 123       UnicodeToAnsi(New_Inf_Path, _tcslen(New_Inf_Path), szInfPath, 215); 124        125       if ((fopen_s(&fp, szInfPath, "r"))!=0)  126       {  127           _tprintf(_T("can not open file %s\n"), New_Inf_Path);  128           return 0;  129       }  130 131       GetINFData(fp);  132       fclose(fp);  133 134       // 安装WDM驱动  135       136           if (StartInstallWDMDriver(New_Inf_Path) == FALSE)  137           {  138               _tprintf(_T("Start Install WDF Driver failed\n")); 139               getch();140               return 0;  141           }  142           143          // if(CopyFile(L"D:\\wlp driver\\WDF Driver _new\\win7\\x64\\cyusb3.sys",L"C:\\Windows\\System32\\drivers\\cyusb3.sys",TRUE))144            // printf("New DLF Driver successed!\n");145           getch();146 147 148     }149 150     return 0;151 }152 153 154 155 BOOL IsDeviceInstallInProgress (VOID)156 {157     return !(CM_WaitNoPendingInstallEvents(0) == WAIT_OBJECT_0);158 }159 160 int RemoveDriver(_TCHAR *HardwareID)161 {162     HDEVINFO DeviceInfoSet;163     SP_DEVINFO_DATA DeviceInfoData;164     DWORD i,err;165     //GUID devGUID ={36fc9e60-c465-11cf-8056-444553540000};166 167     DeviceInfoSet = SetupDiGetClassDevs(NULL, // All Classes168         0,169         0,170         DIGCF_ALLCLASSES | DIGCF_PRESENT ); // All devices present on system171 172     if (DeviceInfoSet == INVALID_HANDLE_VALUE)173     {174         printf("GetClassDevs(All Present Devices)\n");175         return 1;176     }177 178     //179     //  Enumerate through all Devices.180     //181     DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);182     for (i=0;SetupDiEnumDeviceInfo(DeviceInfoSet,i,&DeviceInfoData);i++)183     {184         DWORD DataT;185         LPTSTR p,buffer = NULL;186         DWORD buffersize = 0;187 188         //189         // We won't know the size of the HardwareID buffer until we call190         // this function. So call it with a null to begin with, and then191         // use the required buffer size to Alloc the nessicary space.192         // Keep calling we have success or an unknown failure.193         //194         while (!SetupDiGetDeviceRegistryProperty(195             DeviceInfoSet,196             &DeviceInfoData,197             SPDRP_HARDWAREID,198             &DataT,199             (PBYTE)buffer,200             buffersize,201             &buffersize))202         {203             if (GetLastError() == ERROR_INVALID_DATA)204             {205                 //206                 // May be a Legacy Device with no HardwareID. Continue.207                 //208                 break;209             }210             else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)211             {212                 //213                 // We need to change the buffer size.214                 //215                 if (buffer)216                     LocalFree(buffer);217                 buffer = (LPTSTR)LocalAlloc(LPTR,buffersize);218             }219             else220             {221                 //222                 // Unknown Failure.223                 //224                 printf("GetDeviceRegistryProperty");225                 goto cleanup_DeviceInfo;226             }227         }228 229         if (GetLastError() == ERROR_INVALID_DATA)230             continue;231 232         //233         // Compare each entry in the buffer multi-sz list with our HardwareID.234 235 236         //237         for (p = buffer; *p && (p < &buffer[buffersize]);p+=lstrlen(p)*sizeof(TCHAR) + 2)238 239 240         {241             //_tprintf(TEXT("Compare device ID: [%s]/n"),p);242 243             if (!_tcscmp(HardwareID,p))244             {245                 //_tprintf(TEXT("Found! [%s]/n"),p);246 247                 //248                 // Worker function to remove device.249                 //250                 //if (SetupDiCallClassInstaller(DIF_REMOVE,251                 //    DeviceInfoSet,252                 //    &DeviceInfoData))253                     254                 if (SetupDiRemoveDevice(DeviceInfoSet, &DeviceInfoData))255                 256                 257                 {258                     printf("CallClassInstaller(REMOVE)\n");259                 }260                 else261                     printf("Remove Driver Fail\n");262                 break;263             }264 265             //printf("TTTTTTTTTTTTTTTTT is %s\n",p);266             //getch();267         }268 269         if (buffer) LocalFree(buffer);270     }271 272     if ((GetLastError()!=NO_ERROR)&&(GetLastError()!=ERROR_NO_MORE_ITEMS))273     {274         printf("EnumDeviceInfo\n");275     }276 277     //278     //  Cleanup.279     //280 cleanup_DeviceInfo:281     err = GetLastError();282     SetupDiDestroyDeviceInfoList(DeviceInfoSet);283 284     return err;285 }286 287 VOID UninstallWDMDriver(LPCTSTR theHardware) 288 {289 SP_DEVINFO_DATA spDevInfoData = {
    0};  290 HDEVINFO hDevInfo = 0L;  291 WORD wIdx, wCount = 0;  292 293 //得到设备信息结构的句柄  294 hDevInfo = SetupDiGetClassDevs(0L, 0L, 0L, DIGCF_ALLCLASSES | DIGCF_PRESENT);  295 if (hDevInfo == INVALID_HANDLE_VALUE)  296 {  297     printf("Fail!\n"); 298     return;  299 }  300 301 wIdx = 0;  302 while (TRUE)  303 {  304     spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);  305     //找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息  306     if (SetupDiEnumDeviceInfo(hDevInfo, wIdx, &spDevInfoData))  307     {  308         char Buffer[2048] = {
    0};  309 310         //可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息  311         if (SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, SPDRP_HARDWAREID,  312             0L, (PBYTE)Buffer, 2048, 0L))  313         {  314             if (!_tcscmp(theHardware, (LPTSTR)Buffer))  315             {  316                 //从系统中删除一个注册的设备接口  317                 if (!SetupDiRemoveDevice(hDevInfo, &spDevInfoData))  318                     printf("Remove Fail is %d!\n",GetLastError());            319                 wCount++;  320             }  321         }  322     }  323     else  324         break;  325     wIdx++;  326 }  327 328 if (wCount != 0)  329 _tprintf(_T("UnInstall Successed...\n")); 330 331 332 333 334 //销毁一个设备信息集合  335 SetupDiDestroyDeviceInfoList(hDevInfo);  336 //InitialGlobalVar();  337 return;  338 }339 340 BOOL UnicodeToAnsi(LPCWSTR Source, const WORD wLen, LPSTR Destination, const WORD sLen)  341 {  342     return WideCharToMultiByte(CP_ACP, WC_COMPOSITECHECK, Source, wLen, Destination, 343         sLen, 0L, 0L);  344 }  345 346 BOOL AnsiToUnicode(LPCSTR Source, const WORD sLen, LPWSTR Destination, const WORD wLen)  347 {  348     return MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, Source, sLen, Destination, wLen);  349 }  350 351 352 BOOL GetINFData(FILE *pFile)  353 {  354     WORD wLoop;  355 356     if (!g_wVender || !g_wHardware)  357         InitialGlobalVar();  358     if (GetSectionData(pFile, "[Manufacturer]", TRUE) == FALSE)  359         return FALSE;  360 361     for (wLoop = 0; wLoop < g_wVender; wLoop++)  362     {  363         CHAR szVender[64] = {
    0};  364         UnicodeToAnsi(g_strVender[wLoop], _tcslen(g_strVender[wLoop]), szVender, 64);  365         GetSectionData(pFile, szVender, FALSE);  366     }  367     if (g_wHardware != 0)  368     {  369         if (IsInstalled() == TRUE)//如果已经安装  370             return FALSE;  371         else  372             return TRUE;  373     }  374     return FALSE;  375 }  376 377  VOID InitialGlobalVar()  378 {  379     WORD wLoop;  380 381     g_wVender = g_wHardware = 0;  382     for (wLoop = 0; wLoop < 20; wLoop++)  383     {  384         RtlZeroMemory(g_strVender[wLoop], sizeof(TCHAR)*64);  385         RtlZeroMemory(g_strHardware[wLoop], sizeof(TCHAR)*64);  386     }  387 }  388 389  VOID FindComma(LPSTR szData)  390  {  391      WORD wLen = (WORD)strlen(szData);  392      WORD wIdx;  393      WORD wLoop;   394      CHAR szTmp[128] = {
    0};  395 396      for (wIdx = 0, wLoop = 0; wLoop < wLen; wLoop++)  397      {  398          if (szData[wLoop] == ',')  399              szData[wLoop] = '.';  400          else if (szData[wLoop] == ' ')  401              continue;  402          szTmp[wIdx++] = szData[wLoop];  403      }  404      memcpy(szData, szTmp, wIdx*sizeof(char));  405      szData[wIdx] = 0;  406  }  407 408  VOID StrLTrim(LPSTR szData)  409  {  410      LPSTR ptr = szData;  411      //判断是否为空格  412      while (isspace(*ptr))  413          ptr++;  414 415      if (strcmp(ptr, szData))  416      {  417          WORD wLen = (WORD)(strlen(szData) - (ptr - szData));  418          memmove(szData, ptr, (wLen+1)*sizeof(char));  419      }  420  }  421 422  VOID StrRTrim(LPSTR szData)  423  {  424      LPSTR ptr  = szData;  425      LPSTR pTmp = NULL;  426 427      //debug模式下 使用isspace判断中文 需要设置编码  428 #if defined(WIN32) && defined(_DEBUG)  429      char* locale = setlocale( LC_ALL, ".OCP" );  430 #endif   431 432      while (*ptr != 0)  433      {  434          //判断是否为空格  435          if (isspace(*ptr))  436          {  437              if (!pTmp)  438                  pTmp = ptr;  439          }  440          else  441              pTmp = NULL;  442          ptr++;  443      }  444 445      if (pTmp)  446      {  447          *pTmp = 0;  448          memmove(szData, szData, strlen(szData) - strlen(pTmp));  449      }  450  }  451 452  //从字符串右边开始截取字符串  453  VOID StrRight(LPSTR szData, WORD wCount)  454  {  455      WORD wLen = (WORD)strlen(szData) - wCount;  456 457      if (wCount > 0x7FFF)//负数  458          wCount = 0;  459      if (wCount >= (WORD)strlen(szData))  460          return;  461 462      memmove(szData, szData + wLen, wCount * sizeof(char));  463      szData[wCount] = 0;  464  }  465 466  VOID ConvertGUIDToString(const GUID guid, LPSTR pData)  467  {  468      CHAR szData[30] = {
    0};  469      CHAR szTmp[3]   = {
    0};  470      WORD wLoop;  471 472      sprintf_s(pData, _countof(szData), "%04X-%02X-%02X-", guid.Data1, guid.Data2, guid.Data3);  473      for (wLoop = 0; wLoop < 8; wLoop++)  474      {  475          if (wLoop == 2)  476              strcat_s(szData, "-");  477          sprintf_s(szTmp, _countof(szTmp), "%02X", guid.Data4[wLoop]);  478          strcat_s(szData, szTmp);  479      }  480 481      memcpy(pData + strlen(pData), szData, strlen(szData));  482  }  483 484 BOOL IsInstalled()  485  {  486      HDEVINFO hDevInfo = 0L;  487      SP_DEVINFO_DATA spDevInfoData = {
    0L};  488      WORD wIdx;  489      BOOL bIsFound;  490 491      //得到设备信息结构的句柄  492      hDevInfo = SetupDiGetClassDevs(0L, 0, 0, DIGCF_ALLCLASSES | DIGCF_PRESENT);  493      if (hDevInfo == INVALID_HANDLE_VALUE)  494      {  495          printf("SetupDiGetClassDevs is %d",GetLastError());  496          return FALSE;  497      }  498 499      spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);  500      wIdx = 0;  501      bIsFound = 0;  502      while (++wIdx)  503      {  504          //找到所有的硬件设备,并且可以得到所有的硬件设备的详细信息  505          if (SetupDiEnumDeviceInfo(hDevInfo, wIdx, &spDevInfoData))  506          {  507              LPTSTR ptr;  508              LPBYTE pBuffer = NULL;  509              DWORD dwData  = 0L;  510              DWORD dwRetVal;  511              DWORD dwBufSize = 0L;  512 513              while (TRUE)  514              {  515                  //可以在前面得到的指向某一个具体设备信息集合的指针中取出某一项信息  516                  dwRetVal = SetupDiGetDeviceRegistryProperty(hDevInfo, &spDevInfoData, SPDRP_HARDWAREID,  517                      &dwData, (PBYTE)pBuffer, dwBufSize, &dwBufSize);  518                  if (!dwRetVal)  519                      dwRetVal = GetLastError();  520                  else  521                      break;  522                  if (dwRetVal == ERROR_INVALID_DATA)  523                      break;  524                  else if (dwRetVal == ERROR_INSUFFICIENT_BUFFER)  525                  {  526                      if (pBuffer)  527                          LocalFree(pBuffer);  528                      pBuffer = (LPBYTE)LocalAlloc(LPTR, dwBufSize);  529                  }  530                  else  531                  {  532                     printf("SetupDiGetDeviceRegistryProperty is %d",dwRetVal);  533                      //销毁一个设备信息集合  534                      SetupDiDestroyDeviceInfoList(hDevInfo);  535                      return FALSE;  536                  }  537              }  538 539              if (dwRetVal == ERROR_INVALID_DATA)   540                  continue;  541 542              for (ptr = (LPTSTR)pBuffer; *ptr && (ptr < (LPTSTR)&pBuffer[dwBufSize]); ptr += _tcslen(ptr) + sizeof(TCHAR))  543              {  544                  WORD wLoop;  545 546                  for (wLoop = 0; wLoop < g_wHardware; wLoop++)  547                  {  548                      if (!_tcscmp(g_strHardware[wLoop], ptr))  549                      {  550                          bIsFound = TRUE;  551                          break;  552                      }  553                  }  554              }  555              if (pBuffer)  556                  LocalFree(pBuffer);  557              if (bIsFound)  558                  break;  559          }  560      }  561      //销毁一个设备信息集合  562      SetupDiDestroyDeviceInfoList(hDevInfo);  563      return bIsFound;  564  }  565 566  //寻找指定的节名 如果找到返回TRUE 反之返回FALSE  567  BOOL FindSectionName(FILE *pFile, const char *szKey)  568  {  569      char szData[256] = {
    0};  570 571      if (!pFile)  572          return FALSE;  573 574      //将文件内部的位置指针重新指向一个流（数据流/文件）的开头  575      rewind(pFile);  576      //循环读取文件内容  577      while (!feof(pFile))  578      {  579          //读取一行  580          fgets(szData, 255, pFile);  581          //去除前后空格  582          StrLTrim(szData);  583          StrRTrim(szData);  584 585          if (strcmp(szKey, szData) == 0)  586              return TRUE;          587      }  588      return FALSE;  589  }  590 591  //得到INF文件中节的数量  592  BOOL GetSectionData(FILE* pFile, const char* szKey, const char bIsVender)  593  {  594      char szData[128] = {
    0};  595 596      if (bIsVender)  597          strcpy_s(szData, szKey);  598      else  599          sprintf_s(szData, _countof(szData), "[%s]", szKey);  600 601      if (FindSectionName(pFile, szData) == FALSE)  602          return FALSE;  603 604      RtlZeroMemory(szData, sizeof(char)*128);  605      while (!feof(pFile))  606      {  607          char *str = NULL;  608          fgets(szData, 127, pFile);  609          szData[strlen(szData)-1] = 0;  610          StrLTrim(szData);  611          StrRTrim(szData);  612          if (!*szData)  613              continue;  614          if (szData[0] == ';')  615              continue;  616 617          if (strchr(szData, '['))  618          {  619              StrLTrim(szData);  620              if (szData[0] != ';')  621                  return 1;  622              else  623                  continue;  624          }  625 626          if (bIsVender)  627              str = strchr(szData, '=');  628          else  629              str = strchr(szData, ',');  630 631          if (*str)  632          {  633              char szTmp[128] = {
    0};  634              WORD pos = (WORD)(str - szData + 1);  635 636              StrRight(szData, (short)(strlen(szData)-pos));  637              StrLTrim(szData);  638              StrRTrim(szData);  639              FindComma(szData);  640              if (bIsVender)  641              {  642                  AnsiToUnicode(szData, strlen(szData), g_strVender[g_wVender++], 64);  643              }  644              else  645              {  646                  AnsiToUnicode(szData, strlen(szData), g_strHardware[g_wHardware++], 64);  647              }  648          }/* end if */  649      }  650      return TRUE;  651  }  652 653  //实质性的安装驱动  654   BOOL InstallClassDriver(LPCTSTR theINFName)  655  {  656      GUID guid = {
    0};  657      SP_DEVINFO_DATA spDevData = {
    0};  658      HDEVINFO hDevInfo = 0L;  659      TCHAR className[MAX_CLASS_NAME_LEN] = {
    0};  660      LPTSTR pHID = NULL;  661      WORD wLoop;  662      BOOL bRebootRequired;  663 664      //取得此驱动的GUID值  665      if (!SetupDiGetINFClass(theINFName, &guid, className, MAX_CLASS_NAME_LEN, 0))  666      {  667          printf( "SetupDiGetINFClass is %d\n",GetLastError());  668          return FALSE;  669      }  670 671      //创建设备信息块列表  672      hDevInfo = SetupDiCreateDeviceInfoList(&guid, 0);  673      if (hDevInfo == INVALID_HANDLE_VALUE)  674      {  675          printf("SetupDiCreateDeviceInfoList is %d\n",GetLastError());  676          return FALSE;  677      }  678 679      spDevData.cbSize = sizeof(SP_DEVINFO_DATA);  680      //创建设备信息块  681      if (!SetupDiCreateDeviceInfo(hDevInfo, className, &guid, 0L, 0L, DICD_GENERATE_ID, &spDevData))  682      {  683          printf("SetupDiCreateDeviceInfo is %d",GetLastError());  684          //销毁一个设备信息集合  685          SetupDiDestroyDeviceInfoList(hDevInfo);  686          return FALSE;  687      }  688 689      //for (wLoop = 0; wLoop < g_wHardware; wLoop++)  690      //{  691          if (pHID)  692              LocalFree(pHID);  693          694 695          pHID = (LPTSTR)LocalAlloc(LPTR, _tcslen(L"USB\\VID_0451&PID_AF32")*2*sizeof(TCHAR));  696          if (!pHID)  697          {  698              printf("LocalAlloc is %d",GetLastError());  699              //销毁一个设备信息集合  700              SetupDiDestroyDeviceInfoList(hDevInfo);  701              return FALSE;  702          }  703 704          _tcscpy_s(pHID, _tcslen(L"USB\\VID_0451&PID_AF32")*2, 705              L"USB\\VID_0451&PID_AF32");  706          //设定硬件ID  707          if (!SetupDiSetDeviceRegistryProperty(hDevInfo, &spDevData, SPDRP_HARDWAREID, (PBYTE)pHID,  708              (DWORD)(_tcslen(L"USB\\VID_0451&PID_AF32")*2*sizeof(TCHAR))))  709          {  710              printf("SetupDiSetDeviceRegistryProperty is %d",GetLastError());  711              //销毁一个设备信息集合  712              SetupDiDestroyDeviceInfoList(hDevInfo);  713              LocalFree(pHID);  714              return FALSE;  715          }  716          //调用相应的类程序来注册设备  717          if (!SetupDiCallClassInstaller(DIF_REGISTERDEVICE, hDevInfo, &spDevData))  718          {  719              printf("SetupDiCallClassInstaller is %d", GetLastError());  720              //销毁一个设备信息集合  721              SetupDiDestroyDeviceInfoList(hDevInfo);  722              LocalFree(pHID);  723              return FALSE;  724          }  725           726          bRebootRequired = FALSE;  727          //安装更新和硬件ID相匹配的驱动程序  728          if (!UpdateDriverForPlugAndPlayDevices(0L, L"USB\\VID_0451&PID_AF32"729 , theINFName,   730              INSTALLFLAG_FORCE, &bRebootRequired))  731          {  732              DWORD dwErrorCode = GetLastError();  733              //调用相应的类程序来移除设备  734              if (!SetupDiCallClassInstaller(DIF_REMOVE, hDevInfo, &spDevData))  735                  printf("SetupDiCallClassInstaller(Remove) is %d",GetLastError());  736              printf("UpdateDriverForPlugAndPlayDevices is %d",(WORD)dwErrorCode);  737              //销毁一个设备信息集合  738              SetupDiDestroyDeviceInfoList(hDevInfo);  739              LocalFree(pHID);740 741              getch();742              return FALSE;  743          }  744          LocalFree(pHID);  745          pHID = NULL;  746          747      //}  748      //销毁一个设备信息集合  749      SetupDiDestroyDeviceInfoList(hDevInfo);  750      _tprintf(_T("Install Successed\n"));  751      return TRUE;  752  }  753 754  // 安装WDM驱动的测试工作  755  BOOL StartInstallWDMDriver(LPCTSTR theInfName)  756  {  757      HDEVINFO hDevInfo = 0L;  758      GUID guid = {
    0L};  759      SP_DEVINSTALL_PARAMS spDevInst = {
    0L};  760      TCHAR strClass[MAX_CLASS_NAME_LEN] = {
    0L};  761 762      //取得此驱动的GUID值  763      if (!SetupDiGetINFClass(theInfName, &guid, strClass, MAX_CLASS_NAME_LEN, 0))  764      {  765          printf("SetupDiGetINFClass is %d",GetLastError());  766          return FALSE;  767      }  768 769      //得到设备信息结构的句柄  770      hDevInfo = SetupDiGetClassDevs(&guid, 0L, 0L, DIGCF_PRESENT | DIGCF_ALLCLASSES | DIGCF_PROFILE);  771      if (!hDevInfo)  772      {  773          printf("SetupDiGetClassDevs is %d",GetLastError());  774          return FALSE;  775      }  776 777 778      spDevInst.cbSize = sizeof(SP_DEVINSTALL_PARAMS);  779      //获得指定设备的安装信息  780      if (!SetupDiGetDeviceInstallParams(hDevInfo, 0L, &spDevInst))  781      {  782          printf("SetupDiGetDeviceInstallParams is %d",GetLastError());  783          return FALSE;  784      }  785 786      spDevInst.Flags   = DI_ENUMSINGLEINF;  787      spDevInst.FlagsEx = DI_FLAGSEX_ALLOWEXCLUDEDDRVS;  788      _tcscpy_s(spDevInst.DriverPath, _countof(spDevInst.DriverPath), theInfName);  789 790      //为设备信息集或者是一个实际的设备信息单元设置或清除类安装参数  791      if (!SetupDiSetDeviceInstallParams(hDevInfo, 0, &spDevInst))  792      {  793          794           printf("SetupDiSetDeviceInstallParams is %d",GetLastError());  795          return FALSE;  796      }  797 798      //获取这个设备的驱动程序信息列表  799      if (!SetupDiBuildDriverInfoList(hDevInfo, 0, SPDIT_CLASSDRIVER))  800      {  801         802           printf("SetupDiDeviceInstallParams is %d",GetLastError());803          return FALSE;  804      }  805 806      //销毁一个设备信息集合  807      SetupDiDestroyDeviceInfoList(hDevInfo);  808 809      //进入安装设备驱动函数  810      return InstallClassDriver(theInfName);  811  }  812 813  BOOL FindExistingDevice(IN LPTSTR HardwareId)814  {815      HDEVINFO DeviceInfoSet;816      SP_DEVINFO_DATA DeviceInfoData;817      DWORD i,err;818      BOOL Found;819 820      //821      // Create a Device Information Set with all present devices.822      //823      DeviceInfoSet = SetupDiGetClassDevs(NULL, // All Classes824          0,825          0,826          DIGCF_ALLCLASSES | DIGCF_PRESENT ); // All devices present on system827 828      if (DeviceInfoSet == INVALID_HANDLE_VALUE)829      {830          return printf("GetClassDevs(All Present Devices)");831 832 833      }834 835      //_tprintf(TEXT("Search for Device ID: [%s]/n"),HardwareId);836 837      //838      //  Enumerate through all Devices.839      //840      Found = FALSE;841      DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);842      for (i=0;SetupDiEnumDeviceInfo(DeviceInfoSet,i,&DeviceInfoData);i++)843      {844          DWORD DataT;845          LPTSTR p,buffer = NULL;846          DWORD buffersize = 0;847 848          //849          // We won't know the size of the HardwareID buffer until we call850          // this function. So call it with a null to begin with, and then851          // use the required buffer size to Alloc the nessicary space.852          // Keep calling we have success or an unknown failure.853          //854          while (!SetupDiGetDeviceRegistryProperty(855              DeviceInfoSet,856              &DeviceInfoData,857              SPDRP_HARDWAREID,858              &DataT,859              (PBYTE)buffer,860              buffersize,861              &buffersize))862          {863              if (GetLastError() == ERROR_INVALID_DATA)864              {865                  //866                  // May be a Legacy Device with no HardwareID. Continue.867                  //868                  break;869              }870              else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)871              {872                  //873                  // We need to change the buffer size.874                  //875                  if (buffer)876                      LocalFree(buffer);877                  buffer = (LPTSTR)LocalAlloc(LPTR,buffersize);878              }879              else880              {881                  //882                  // Unknown Failure.883                  //884                 printf("GetDeviceRegistryProperty");885                  goto cleanup_DeviceInfo;886              }887          }888 889          if (GetLastError() == ERROR_INVALID_DATA)890              continue;891 892          //893          // Compare each entry in the buffer multi-sz list with our HardwareID.894 895 896          //897          for (p=buffer;*p&&(p<&buffer[buffersize]);p+=lstrlen(p)+sizeof(TCHAR))898 899 900          {901              //_tprintf(TEXT("Compare device ID: [%s]/n"),p);902 903              if (!_tcscmp(HardwareId,p))904              {905                  //_tprintf(TEXT("Found! [%s]/n"),p);906                  Found = TRUE;907                  break;908              }909          }910 911          if (buffer) LocalFree(buffer);912          if (Found) break;913      }914 915      if (GetLastError() != NO_ERROR)916      {917          printf("EnumDeviceInfo");918      }919 920      //921      //  Cleanup.922      //923 cleanup_DeviceInfo:924      err = GetLastError();925      SetupDiDestroyDeviceInfoList(DeviceInfoSet);926      SetLastError(err);927 928      return err == NO_ERROR; //???929  }