Scan for multi-function PCI devices

This commit is contained in:
Josh Holtrop 2023-06-08 20:28:32 -04:00
parent dda1b6379f
commit 7a4edf38f0

View File

@ -26,9 +26,9 @@ struct Pci
} }
} }
private static uint read_config_register(uint bus_id, uint device_idx, uint register_id) private static uint read_config_register(uint bus_id, uint device_idx, uint function_idx, uint register_id)
{ {
uint addr = 0x8000_0000u | (bus_id << 16u) | (device_idx << 11u) | (register_id << 2u); uint addr = 0x8000_0000u | (bus_id << 16u) | (device_idx << 11u) | (function_idx << 8u) | (register_id << 2u);
out32(IO_ADDR_CONFIG_ADDRESS, addr); out32(IO_ADDR_CONFIG_ADDRESS, addr);
return in32(IO_ADDR_CONFIG_DATA); return in32(IO_ADDR_CONFIG_DATA);
} }
@ -59,22 +59,34 @@ struct Pci
} }
} }
private static void scan(uint bus_id, uint device_idx) private static void scan(uint bus_id, uint device_idx, uint function_idx)
{ {
ulong reg0 = read_config_register(bus_id, device_idx, 0u); ulong reg0 = read_config_register(bus_id, device_idx, function_idx, 0u);
if (reg0 != 0xFFFFFFFFu) if (reg0 != 0xFFFFFFFFu)
{ {
uint reg2 = read_config_register(bus_id, device_idx, 2u); uint reg2 = read_config_register(bus_id, device_idx, function_idx, 2u);
ushort vendor_id = reg0 & 0xFFFFu; ushort vendor_id = reg0 & 0xFFFFu;
ushort device_id = (reg0 >> 16u) & 0xFFFFu; ushort device_id = (reg0 >> 16u) & 0xFFFFu;
ubyte class_id = (reg2 >> 24u) & 0xFFu; ubyte class_id = (reg2 >> 24u) & 0xFFu;
ubyte subclass_id = (reg2 >> 16u) & 0xFFu; ubyte subclass_id = (reg2 >> 16u) & 0xFFu;
ubyte if_id = (reg2 >> 8u) & 0xFFu; ubyte if_id = (reg2 >> 8u) & 0xFFu;
Klog.writefln("Found PCI device %04x:%04x (%02x:%02x:%02x) at %02u:%02u", Klog.writefln("Found PCI device %04x:%04x (%02x:%02x:%02x) at %02u:%02u.%u",
vendor_id, device_id, vendor_id, device_id,
class_id, subclass_id, if_id, class_id, subclass_id, if_id,
bus_id, device_idx); bus_id, device_idx, function_idx);
init_device(class_id, subclass_id, if_id); init_device(class_id, subclass_id, if_id);
if (function_idx == 0u)
{
uint reg3 = read_config_register(bus_id, device_idx, function_idx, 3u);
if ((reg3 & (1u << 23u)) != 0u)
{
/* Multi-function device found. */
for (function_idx = 1u; function_idx < 8u; function_idx++)
{
scan(bus_id, device_idx, function_idx);
}
}
}
} }
} }
@ -85,7 +97,7 @@ struct Pci
{ {
for (uint device_idx = 0u; device_idx < 32u; device_idx++) for (uint device_idx = 0u; device_idx < 32u; device_idx++)
{ {
scan(bus_id, device_idx); scan(bus_id, device_idx, 0u);
} }
} }
Klog.writefln("PCI scan complete."); Klog.writefln("PCI scan complete.");