]>
Commit | Line | Data |
---|---|---|
1 | #include <pci.h> | |
2 | #include <output.h> | |
3 | ||
4 | int pci_probe(pci_probe_fn_t probe, void *data) | |
5 | { | |
6 | int devsfound = 0; | |
7 | pci_dev_t pdev; | |
8 | unsigned int bus; | |
9 | unsigned int dev; | |
10 | unsigned int fn; | |
11 | ||
12 | for (bus = 0; bus < 0x100; bus++) | |
13 | for (dev = 0; dev < 0x20; dev++) | |
14 | { | |
15 | if (pci_read32(bus, dev, 0, 0) == 0xFFFFFFFF) | |
16 | continue; | |
17 | for (fn = 0; fn < 8; fn++) | |
18 | { | |
19 | int bar; | |
20 | ||
21 | if (pci_read32(bus, dev, fn, 0) == 0xFFFFFFFF) | |
22 | continue; | |
23 | ||
24 | if ((fn != 0) && !(pci_read8(bus, dev, 0, 0x0E) & 0x80)) | |
25 | continue; | |
26 | ||
27 | pdev.bus = bus; | |
28 | pdev.dev = dev; | |
29 | pdev.fn = fn; | |
30 | pdev.vid = pci_read16(bus, dev, fn, 0); | |
31 | pdev.did = pci_read16(bus, dev, fn, 2); | |
32 | ||
33 | for (bar = 0; bar < 6; bar++) | |
34 | { | |
35 | unsigned long bardat = pci_read32(bus, dev, fn, 0x10 + bar*4); | |
36 | if (bardat == 0) | |
37 | { | |
38 | pdev.bars[bar].type = PCI_BAR_NONE; | |
39 | continue; | |
40 | } | |
41 | if (bardat & 1) | |
42 | { | |
43 | pdev.bars[bar].type = PCI_BAR_IO; | |
44 | pdev.bars[bar].addr = bardat & ~0x3; | |
45 | } else { | |
46 | pdev.bars[bar].prefetchable = (bardat >> 3) & 1; | |
47 | switch ((bardat >> 1) & 0x3) | |
48 | { | |
49 | case 0: | |
50 | pdev.bars[bar].type = PCI_BAR_MEMORY32; | |
51 | pdev.bars[bar].addr = bardat & ~0xF; | |
52 | break; | |
53 | case 2: | |
54 | pdev.bars[bar].type = PCI_BAR_MEMORY64; | |
55 | bar++; | |
56 | pdev.bars[bar].type = PCI_BAR_NONE; | |
57 | break; | |
58 | default: | |
59 | pdev.bars[bar].type = PCI_BAR_NONE; | |
60 | continue; | |
61 | } | |
62 | } | |
63 | } | |
64 | ||
65 | devsfound += probe(&pdev, data); | |
66 | } | |
67 | } | |
68 | ||
69 | return devsfound; | |
70 | } | |
71 | ||
72 | static int _enumfn(pci_dev_t *pdev, void *data) | |
73 | { | |
74 | int bar; | |
75 | ||
76 | outputf("Found device: %02x:%02x.%1x: %04X:%04X", | |
77 | pdev->bus, pdev->dev, pdev->fn, | |
78 | pdev->vid, pdev->did); | |
79 | for (bar = 0; bar < 6; bar++) | |
80 | { | |
81 | switch (pdev->bars[bar].type) | |
82 | { | |
83 | case PCI_BAR_IO: | |
84 | outputf(" BAR %d: I/O, Addr %04x", bar, pdev->bars[bar].addr); | |
85 | break; | |
86 | case PCI_BAR_MEMORY32: | |
87 | outputf(" BAR %d: Mem32, Addr %04x", bar, pdev->bars[bar].addr); | |
88 | break; | |
89 | case PCI_BAR_MEMORY64: | |
90 | outputf(" BAR %d: Mem64, Addr %04x", bar, pdev->bars[bar].addr); | |
91 | break; | |
92 | default: | |
93 | break; | |
94 | } | |
95 | } | |
96 | return 0; | |
97 | } | |
98 | ||
99 | void pci_bus_enum() | |
100 | { | |
101 | pci_probe(_enumfn, 0x0); | |
102 | } | |
103 | ||
104 | static int _probefn(pci_dev_t *dev, void *d) | |
105 | { | |
106 | pci_driver_t *driver = d; | |
107 | int i; | |
108 | ||
109 | for (i = 0; i < driver->id_count; i++) | |
110 | if ((dev->vid == driver->ids[i].vid) && (dev->did == driver->ids[i].did)) | |
111 | { | |
112 | outputf("Probing PCI device: %s (%s)", driver->ids[i].name ? driver->ids[i].name : driver->name, driver->ids[i].friendlyname ? driver->ids[i].friendlyname : "no friendly name"); | |
113 | return driver->probe(dev, d); | |
114 | } | |
115 | return 0; | |
116 | } | |
117 | ||
118 | int pci_probe_driver(pci_driver_t driver) | |
119 | { | |
120 | return pci_probe(_probefn, &driver); | |
121 | } |