基于Visual C++之Windows核心编程代码分析（1）实现设备管理器枚举设备

我们进行Windows编程的时候，有些时候需要枚举设备，例如光盘，光驱，硬盘等等，

我们如何实现功能呢，请见代码分析

 

 

[cpp]  view plain  copy

1. #include <windows.h>  
2. #include <setupapi.h>  
3. #include <stdio.h>  
4. #include <devguid.h>  
5. #include <regstr.h>  
6. /* 函数声明 */  
7. BOOL EnumPresentDevice( const GUID * InterfaceClassGuid );  
8. BOOL EnumAllDevice();  
9. /************************************* 
10. * BOOL EnumClassDevice( const GUID * InterfaceClassGuid ) 
11. * 功能    根据类型列举当前存在的设备 
12. * 参数    InterfaceClassGuid，所需列举设备接口类的GUID 
13. **************************************/  
14. BOOL EnumClassDevice( const GUID * InterfaceClassGuid )  
15.   
16. {  
17.     HDEVINFO DeviceInfoSet;  
18.     HDEVINFO NewDeviceInfoSet;  
19.   
20.     SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;  
21.     PSP_DEVICE_INTERFACE_DETAIL_DATA lpDeviceInterfaceDetailData;  
22.   
23.     DWORD dwBufferSize = 0;  
24.     DWORD i;  
25.     // 创建空设备信息列表  
26.     DeviceInfoSet = SetupDiCreateDeviceInfoList(NULL, NULL);  
27.   
28.     if(DeviceInfoSet == INVALID_HANDLE_VALUE)   
29.     {  
30.         printf("CreateDeviceInfoList failed: %d\n", GetLastError());  
31.         return 0;  
32.     }  
33.   
34.     // 根据接口类型获得新的设备信息列表  
35.   
36.     NewDeviceInfoSet = SetupDiGetClassDevsEx(  
37.         InterfaceClassGuid,  
38.         NULL,  
39.         NULL,  
40.         DIGCF_PRESENT | DIGCF_DEVICEINTERFACE,  
41.         DeviceInfoSet,// 之前创建的设备信息列表  
42.         NULL,  
43.         NULL  
44.         );  
45.     if(NewDeviceInfoSet == INVALID_HANDLE_VALUE)  
46.     {  
47.         printf( "SetupDiGetClassDevsEx failed: %d\n", GetLastError() );  
48.         return 0;  
49.     }  
50.     // 设置 SP_DEVICE_INTERFACE_DATA 大小  
51.     DeviceInterfaceData.cbSize   
52.         = sizeof(SP_DEVICE_INTERFACE_DATA);  
53.   
54.     for (i=0; ;i++)  
55.     {  
56.         // 列举接口信息  
57.         BOOL bResult = SetupDiEnumDeviceInterfaces(  
58.             NewDeviceInfoSet,  
59.             NULL,  
60.             InterfaceClassGuid,  
61.             i,  
62.             &DeviceInterfaceData  
63.             );  
64.         if(!bResult)  
65.         {  
66.             if ( GetLastError()!=NO_ERROR &&  
67.                 GetLastError()!=ERROR_NO_MORE_ITEMS )  
68.             {  
69.                 printf("ERROR: (%d)",GetLastError());  
70.                 return FALSE;  
71.             }  
72.             break;  
73.         }  
74.         else  
75.         {  
76.             // 为PSP_DEVICE_INTERFACE_DETAIL_DATA结构分配内存，填充  
77.             lpDeviceInterfaceDetailData = HeapAlloc(  
78.                 GetProcessHeap(), 0,  
79.                 sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA));  
80.             lpDeviceInterfaceDetailData->cbSize   
81.                 = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);  
82.             dwBufferSize = lpDeviceInterfaceDetailData->cbSize;  
83.             // 获得接口详细信息  
84.             while(!SetupDiGetDeviceInterfaceDetail(  
85.                 NewDeviceInfoSet,  
86.                 &DeviceInterfaceData,  
87.                 lpDeviceInterfaceDetailData,  
88.                 dwBufferSize,  
89.                 &dwBufferSize,  
90.                 NULL))  
91.             {  
92.                 // 如果内存空间不足，再次分配，直到可以成功调用  
93.                 if(ERROR_INSUFFICIENT_BUFFER==GetLastError())  
94.                 {  
95.                     lpDeviceInterfaceDetailData = HeapReAlloc(  
96.                         GetProcessHeap(), 0,   
97.                         lpDeviceInterfaceDetailData, dwBufferSize);  
98.                     lpDeviceInterfaceDetailData->cbSize   
99.                         = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);  
100.                 }                 
101.             }  
102.             // 显示信息  
103.             printf("DevicePath: %s\n",lpDeviceInterfaceDetailData->DevicePath);  
104.             // lpDeviceInterfaceDetailData->DevicePath可作为CreateFile的参数，进行IO控制  
105.               
106.             // 释放内存  
107.             HeapFree(GetProcessHeap(),0,lpDeviceInterfaceDetailData);  
108.         }  
109.     }  
110.     SetupDiDestroyDeviceInfoList(DeviceInfoSet);  
111.     return TRUE;  
112. }  
113. /************************************* 
114. * BOOL EnumAllDevice( ) 
115. * 功能    列举当前存在的设备 
116. * 返回值   是否成功 
117. **************************************/  
118. BOOL EnumAllDevice()  
119. {  
120.     HDEVINFO hDevInfo;  
121.     SP_DEVINFO_DATA DeviceInfoData;  
122.     DWORD i;  
123.   
124.     printf("Displaying the Installed Devices\n\n");  
125.   
126.     // 得到所有设备 HDEVINFO   
127.     hDevInfo = SetupDiGetClassDevs(NULL,  
128.         0, // 无类型  
129.         0, // 无回调函数  
130.         DIGCF_PRESENT | DIGCF_ALLCLASSES );  
131.     if (hDevInfo == INVALID_HANDLE_VALUE)  
132.     {  
133.         return FALSE;  
134.     }  
135.     // 循环列举  
136.     DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);  
137.     for (i=0;SetupDiEnumDeviceInfo(hDevInfo,i,  
138.         &DeviceInfoData);i++)  
139.     {  
140.         DWORD DataT;  
141.         LPTSTR buffer = NULL;  
142.         DWORD buffersize = 0;  
143.   
144.         // 获取详细信息  
145.         while (!SetupDiGetDeviceRegistryProperty(  
146.             hDevInfo,  
147.             &DeviceInfoData,  
148.             SPDRP_DEVICEDESC,  
149.             &DataT,  
150.             (PBYTE)buffer,  
151.             buffersize,  
152.             &buffersize))  
153.         {  
154.             if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)  
155.             {  
156.                 // 内存不足  
157.                 if (buffer) HeapFree(GetProcessHeap(), 0, buffer);  
158.                 buffer = (LPTSTR)HeapAlloc(GetProcessHeap(), 0,  buffersize);  
159.             }  
160.             else  
161.                 break;  
162.         }  
163. // 输出  
164.         printf("GUID:{%.8X-%.4X-%.4X--%.2X%.2X-%.2X%.2X%.2X%.2X%.2X%.2X} "  
165.             "Device: %s\n",  
166.             DeviceInfoData.ClassGuid.Data1,  
167.             DeviceInfoData.ClassGuid.Data2,  
168.             DeviceInfoData.ClassGuid.Data3,  
169.             DeviceInfoData.ClassGuid.Data4[0],  
170.             DeviceInfoData.ClassGuid.Data4[1],  
171.             DeviceInfoData.ClassGuid.Data4[2],  
172.             DeviceInfoData.ClassGuid.Data4[3],  
173.             DeviceInfoData.ClassGuid.Data4[4],  
174.             DeviceInfoData.ClassGuid.Data4[5],  
175.             DeviceInfoData.ClassGuid.Data4[6],  
176.             DeviceInfoData.ClassGuid.Data4[7],buffer);  
177.   
178.         if (buffer) HeapFree(GetProcessHeap(), 0, buffer);  
179.     }  
180.   
181.     if ( GetLastError()!=NO_ERROR &&  
182.         GetLastError()!=ERROR_NO_MORE_ITEMS )  
183.     {  
184.         return FALSE;  
185.     }  
186.     //  释放  
187.     SetupDiDestroyDeviceInfoList(hDevInfo);  
188.     return TRUE;  
189. }  
190.   
191. int main( int argc, char *argv[ ], char *envp[ ] )  
192. {  
193.     // 列举所有设备  
194.     printf("Enumerating All Device\n\n");  
195.     EnumAllDevice();  
196.     // 列举磁盘分卷驱动器设备  
197.     printf("\n\nEnumerating Present Volume \n\n");  
198.     EnumClassDevice(&GUID_DEVINTERFACE_VOLUME);  
199.     return 0;  
200. }