【C】windows驱动 扫描pci总线设备

Ayala

1. #include <ntdef.h>
   
2. #include <ntddk.h>
   
3. char*
   
4. fmt1="PCI_COMMON_CONFIG \n\
   
5.       USHORT VendorID   :%04X\n\
   
6.       USHORT DeviceID   :%04X\n\
   
7.       USHORT Command    :%04X\n\
   
8.       USHORT Status     :%04X\n\
   
9.       UCHAR  RevisionID :%02X\n\
   
10.       UCHAR  ProgIf     :%02X\n\
    
11.       UCHAR  SubClass   :%02X\n\
    
12.       UCHAR  BaseClass  :%02X\n\
    
13.       UCHAR  CacheLineSize :%02X\n\
    
14.       UCHAR  LatencyTimer  :%02X\n\
    
15.       UCHAR  HeaderType    :%02X\n\
    
16.       UCHAR  BIST          :%02X\n";
    
17. char*
    
18. fmt2="\
    
19.         ULONG BaseAddresses[6]:\n\
    
20.           0   %08X\n\
    
21.           1   %08X\n\
    
22.           2   %08X\n\
    
23.           3   %08X\n\
    
24.           4   %08X\n\
    
25.           5   %08X\n\
    
26.         ULONG Reserved1[2]\n\
    
27.         ULONG ROMBaseAddress:\n\
    
28.               %08X\n\
    
29.         ULONG Reserved2[2]\n\
    
30.         UCHAR InterruptLine:\n\
    
31.               %02X\n\
    
32.         UCHAR InterruptPin\n\
    
33.               %02X\n\
    
34.         UCHAR MinimumGrant\n\
    
35.               %02X\n\
    
36.         UCHAR MaximumLatency\n\
    
37.               %02X\n\
    
38.         ";
    
39. #define PCI_COMMON_CONFIG_s1(t) \
    
40.         t.VendorID,\
    
41.         t.DeviceID,\
    
42.         t.Command,\
    
43.         t.Status,\
    
44.         t.RevisionID,\
    
45.         t.ProgIf,\
    
46.         t.SubClass,\
    
47.         t.BaseClass,\
    
48.         t.CacheLineSize,\
    
49.         t.LatencyTimer,\
    
50.         t.HeaderType,\
    
51.         t.BIST
    
52. #define PCI_COMMON_CONFIG_s2(t) \
    
53.         t.u.type0.BaseAddresses[0],\
    
54.         t.u.type0.BaseAddresses[1],\
    
55.         t.u.type0.BaseAddresses[2],\
    
56.         t.u.type0.BaseAddresses[3],\
    
57.         t.u.type0.BaseAddresses[4],\
    
58.         t.u.type0.BaseAddresses[5],\
    
59.         t.u.type0.ROMBaseAddress,\
    
60.         t.u.type0.InterruptLine,\
    
61.         t.u.type0.InterruptPin,\
    
62.         t.u.type0.MinimumGrant,\
    
63.         t.u.type0.MaximumLatency
    
64.        
    
65. SacnPciConfiguration()
    
66. /*
    
67.         typedef struct _PCI_SLOT_NUMBER {
    
68.            union {
    
69.                    struct {
    
70.                            ULONG   DeviceNumber:5;
    
71.                            ULONG   FunctionNumber:3;
    
72.                            ULONG   Reserved:24;
    
73.                    } bits;
    
74.                    ULONG   AsULONG;
    
75.            } u;
    
76.         } PCI_SLOT_NUMBER, *PPCI_SLOT_NUMBER;
    
77. 
    
78.         typedef struct _PCI_COMMON_CONFIG {
    
79.           USHORT VendorID;
    
80.           USHORT DeviceID;
    
81.           USHORT Command;
    
82.           USHORT Status;
    
83.           UCHAR  RevisionID;
    
84.           UCHAR  ProgIf;
    
85.           UCHAR  SubClass;
    
86.           UCHAR  BaseClass;
    
87.           UCHAR  CacheLineSize;
    
88.           UCHAR  LatencyTimer;
    
89.           UCHAR  HeaderType;
    
90.           UCHAR  BIST;
    
91.           union {
    
92.                 struct {
    
93.                   ULONG BaseAddresses[PCI_TYPE0_ADDRESSES];
    
94.                   ULONG Reserved1[2];
    
95.                   ULONG ROMBaseAddress;
    
96.                   ULONG Reserved2[2];
    
97.                   UCHAR InterruptLine;
    
98.                   UCHAR InterruptPin;
    
99.                   UCHAR MinimumGrant;
    
100.                   UCHAR MaximumLatency;
     
101.                 } type0;
     
102.           } u;
     
103.           UCHAR  DeviceSpecific[192];
     
104.         } PCI_COMMON_CONFIG, *PPCI_COMMON_CONFIG;
     
105. */
     
106. {
     
107.     PCI_SLOT_NUMBER     slotData;
     
108.     PCI_COMMON_CONFIG   pciData;
     
109.     ULONG               pciBus;
     
110.     ULONG               slotNumber;
     
111.     ULONG               functionNumber;
     
112. 
     
113.     ULONG               length;
     
114. 
     
115.     slotData.u.AsULONG = 0;
     
116. 
     
117.     //
     
118.     // Search each PCI bus.
     
119.     //
     
120. 
     
121.     for (pciBus = 0;pciBus < 256; pciBus++) {
     
122. 
     
123.         //
     
124.         // Look at each device.
     
125.         //
     
126. 
     
127.         for (slotNumber = 0;slotNumber < 32;slotNumber++) {
     
128. 
     
129.             slotData.u.bits.DeviceNumber = slotNumber;
     
130. 
     
131.             //
     
132.             // Look at each function.
     
133.             //
     
134. 
     
135.             for (functionNumber = 0;functionNumber < 8;functionNumber++) {
     
136. 
     
137.                 slotData.u.bits.FunctionNumber = functionNumber;
     
138.                                         /*
     
139.                                 length = HalpPhase0GetPciDataByOffset(
     
140.                             pciBus,
     
141.                             slotData.u.AsULONG,
     
142.                             &pciData.ProgIf,
     
143.                             FIELD_OFFSET(PCI_COMMON_CONFIG,ProgIf),
     
144.                             sizeof(UCHAR)+sizeof(UCHAR)+sizeof(UCHAR));
     
145.                                         */
     
146.                 length = HalGetBusDataByOffset (
     
147.                             PCIConfiguration,
     
148.                             pciBus,
     
149.                             slotData.u.AsULONG,
     
150.                             &pciData.ProgIf,
     
151.                             FIELD_OFFSET(PCI_COMMON_CONFIG,ProgIf),
     
152.                             sizeof(UCHAR)+sizeof(UCHAR)+sizeof(UCHAR));
     
153.                                 #define DbgP(f,...) DbgPrintEx(DPFLTR_SETUP_ID,DPFLTR_INFO_LEVEL,f,__VA_ARGS__)
     
154. 
     
155.                 if (length == 0) {
     
156.                                         //DbgP("no more Slots");
     
157.                     break;
     
158.                 }
     
159. 
     
160.                                 switch (pciData.BaseClass)
     
161.                                 {
     
162.                                         char* s;
     
163.                                         case 1://STORAGE
     
164.                                                
     
165.                                                 switch (pciData.SubClass)
     
166.                                                 {
     
167.                                                         case 0://SCSI
     
168.                                                                 s="SCSI";
     
169.                                                                 break;
     
170.                                                         case 1://IDE
     
171.                                                                 s="IDE";
     
172.                                                                 break;
     
173.                                                         case 2://FLOPPY
     
174.                                                                 s="FLOPPY";
     
175.                                                                 break;
     
176.                                                         case 3://IPI
     
177.                                                                 s="IPI";
     
178.                                                                 break;
     
179.                                                         case 4://RAID
     
180.                                                                 s="RAID";
     
181.                                                                 break;
     
182.                                                         case 5://???
     
183.                                                                 s="???";
     
184.                                                                 break;
     
185.                                                         case 6://SATA
     
186.                                                                 s="SATA";
     
187.                                                                 if (pciData.ProgIf) s="SATA_AHCI";
     
188.                                                                 break;
     
189.                                                         case 7://SAS
     
190.                                                                 s="SAS";
     
191.                                                                 break;
     
192.                                                         case 8://NVME
     
193.                                                                 s="NVME";
     
194.                                                                 break;
     
195.                                                         case 80://OTHER
     
196.                                                                 s="OTHER";
     
197.                                                                 break;
     
198.                                                         default:break;
     
199.                                                                 s="????";
     
200.                                                 }
     
201.                                                 DbgP("\nSTORAGE:%s\n",s);
     
202.                                                 length = HalGetBusDataByOffset (
     
203.                                                                         PCIConfiguration,
     
204.                                                                         pciBus,
     
205.                                                                         slotData.u.AsULONG,
     
206.                                                                         &pciData,
     
207.                                                                         0,
     
208.                                                                         sizeof(PCI_COMMON_CONFIG));
     
209.                                                 if (length==sizeof(PCI_COMMON_CONFIG))
     
210.                                                 {
     
211.                                                        
     
212.                                                         DbgP(fmt1,PCI_COMMON_CONFIG_s1(pciData));
     
213.                                                         DbgP(fmt2,PCI_COMMON_CONFIG_s2(pciData));
     
214.                                                 }
     
215.                                                 break;
     
216.                                         case 0: //PRE_20
     
217.                                         case 2: //NETWORK
     
218.                                         case 3: //DISPLAY
     
219.                                         case 4: //MULTIMEDIA
     
220.                                         case 5: //MEMORY
     
221.                                         case 6: //BRIDGE
     
222.                                         case 7: //COMMUNICATION
     
223.                                         case 8: //SYSTEM
     
224.                                         case 9: //INPUT
     
225.                                         case 0xA : //DOCKING
     
226.                                         case 0xB : //PROCESSOR
     
227.                                         case 0xC : //SERIAL
     
228.                                         case 0xD : //WIRELESS
     
229.                                         case 0xE : //INTELLIGENT
     
230.                                         case 0xF : //SATELLITE
     
231.                                         case 0x10: //CRYPT
     
232.                                         case 0x11: //SIGNAL_PROCESSING
     
233.                                         case 0xFF: //OTHERS
     
234.                                         default:
     
235.                                                
     
236.                                                 break;
     
237.                                 }
     
238.                                
     
239. 
     
240.             }   // next PCI function
     
241. 
     
242. 
     
243.         }   // next PCI slot
     
244. 
     
245. 
     
246.     }   // next PCI bus
     
247. 
     
248.     return 0;
     
249. 
     
250. } // SacnPciConfiguration()
     
251. 
     
252. 
     
253. 
     
254. NTSTATUS
     
255. DriverEntry(a,b)
     
256. {
     
257.         SacnPciConfiguration();
     
258.         return STATUS_DEVICE_CONFIGURATION_ERROR;
     
259. }

复制代码

0xAA55

我看到了SacnPciConfiguration。。
Sacn，应该打错了，是Scan

Ayala

用了这么久 我自己都没发现

FFFFFFFE

很好. 正需要. 感谢开源