From 4a2c5de5b6ea66cbe65f8f2a9911a80a9cd18f68 Mon Sep 17 00:00:00 2001 From: Max Wash Date: Fri, 9 Jun 2023 21:26:04 +0100 Subject: [PATCH] kexts: pci: implement device enumeration --- kexts/drivers/bus/pci/extension.yaml | 8 ++ kexts/drivers/bus/pci/include/socks/pci.h | 74 ++++++++++ kexts/drivers/bus/pci/main.c | 161 ++++++++++++++++++++++ 3 files changed, 243 insertions(+) create mode 100644 kexts/drivers/bus/pci/extension.yaml create mode 100644 kexts/drivers/bus/pci/include/socks/pci.h create mode 100644 kexts/drivers/bus/pci/main.c diff --git a/kexts/drivers/bus/pci/extension.yaml b/kexts/drivers/bus/pci/extension.yaml new file mode 100644 index 0000000..1d14dec --- /dev/null +++ b/kexts/drivers/bus/pci/extension.yaml @@ -0,0 +1,8 @@ +name: acpi +description: | + PCI bus driver. +id: net.doorstuck.socks.pci +license: BSD-3-Clause +copyright: Copyright © Max Wash 2023 +sources: + - main.c diff --git a/kexts/drivers/bus/pci/include/socks/pci.h b/kexts/drivers/bus/pci/include/socks/pci.h new file mode 100644 index 0000000..2f075df --- /dev/null +++ b/kexts/drivers/bus/pci/include/socks/pci.h @@ -0,0 +1,74 @@ +#ifndef SOCKS_PCI_H_ +#define SOCKS_PCI_H_ + +#include + +#define PCI_VENDOR_ID 0x00 // 2 +#define PCI_DEVICE_ID 0x02 // 2 +#define PCI_COMMAND 0x04 // 2 +#define PCI_STATUS 0x06 // 2 +#define PCI_REVISION_ID 0x08 // 1 + +#define PCI_PROG_IF 0x09 // 1 +#define PCI_SUBCLASS 0x0a // 1 +#define PCI_CLASS 0x0b // 1 +#define PCI_CACHE_LINE_SIZE 0x0c // 1 +#define PCI_LATENCY_TIMER 0x0d // 1 +#define PCI_HEADER_TYPE 0x0e // 1 +#define PCI_BIST 0x0f // 1 +#define PCI_BAR0 0x10 // 4 +#define PCI_BAR1 0x14 // 4 +#define PCI_BAR2 0x18 // 4 +#define PCI_BAR3 0x1C // 4 +#define PCI_BAR4 0x20 // 4 +#define PCI_BAR5 0x24 // 4 + +#define PCI_INTERRUPT_LINE 0x3C // 1 +#define PCI_INTERRUPT_PIN 0x3D + +#define PCI_SECONDARY_BUS 0x19 // 1 + +#define PCI_HEADER_TYPE_DEVICE 0 +#define PCI_HEADER_TYPE_BRIDGE 1 +#define PCI_HEADER_TYPE_CARDBUS 2 + +#define PCI_TYPE_BRIDGE 0x0604 +#define PCI_TYPE_SATA 0x0106 + +#define PCI_ADDRESS_PORT 0xCF8 +#define PCI_VALUE_PORT 0xCFC + +#define PCI_NONE 0xFFFF + +typedef void (*pci_func_t)(uint32_t device, uint16_t vendor_id, uint16_t device_id, void *arg); + +static inline int pci_get_bus(uint32_t device) +{ + return (uint8_t)((device >> 16)); +} + +static inline int pci_get_slot(uint32_t device) +{ + return (uint8_t)((device >> 8)); +} + +static inline int pci_get_func(uint32_t device) +{ + return (uint8_t)(device); +} + +static inline uint32_t pci_get_addr(uint32_t device, int field) +{ + return 0x80000000 | (pci_get_bus(device) << 16) | (pci_get_slot(device) << 11) | (pci_get_func(device) << 8) | ((field) & 0xFC); +} + +static inline uint32_t pci_box_device(int bus, int slot, int func) +{ + return (uint32_t)((bus << 16) | (slot << 8) | func); +} + +extern uint32_t pci_read_field(uint32_t device, int field, int size); +extern void pci_write_field(uint32_t device, int field, int size, uint32_t value); +extern uint16_t pci_find_type(uint32_t dev); + +#endif diff --git a/kexts/drivers/bus/pci/main.c b/kexts/drivers/bus/pci/main.c new file mode 100644 index 0000000..a4bb4bc --- /dev/null +++ b/kexts/drivers/bus/pci/main.c @@ -0,0 +1,161 @@ +#include +#include +#include +#include +#include +#include + +static struct driver *pci_driver = NULL; +static struct bus_device *pci_bus = NULL; + +static void pci_scan_bus(pci_func_t f, int type, int bus, void *arg); + +void pci_write_field(uint32_t device, int field, int size, uint32_t value) +{ + outportl(PCI_ADDRESS_PORT, pci_get_addr(device, field)); + outportl(PCI_VALUE_PORT, value); +} + +uint32_t pci_read_field(uint32_t device, int field, int size) +{ + outportl(PCI_ADDRESS_PORT, pci_get_addr(device, field)); + + if (size == 4) { + uint32_t t = inportl(PCI_VALUE_PORT); + return t; + } else if (size == 2) { + uint16_t t = inportw(PCI_VALUE_PORT + (field & 2)); + return t; + } else if (size == 1) { + uint8_t t = inportb(PCI_VALUE_PORT + (field & 3)); + return t; + } + + return 0xFFFF; +} + +uint16_t pci_find_type(uint32_t dev) +{ + return (pci_read_field(dev, PCI_CLASS, 1) << 8) | pci_read_field(dev, PCI_SUBCLASS, 1); +} + +static void pci_scan_hit(pci_func_t f, uint32_t dev, void *arg) +{ + int dev_vend = (int)pci_read_field(dev, PCI_VENDOR_ID, 2); + int dev_dvid = (int)pci_read_field(dev, PCI_DEVICE_ID, 2); + + f(dev, dev_vend, dev_dvid, arg); +} + +static void pci_scan_func(pci_func_t f, int type, int bus, int slot, int func, void *arg) +{ + uint32_t dev = pci_box_device(bus, slot, func); + if (type == -1 || type == pci_find_type(dev)) { + pci_scan_hit(f, dev, arg); + } + + if (pci_find_type(dev) == PCI_TYPE_BRIDGE) { + pci_scan_bus(f, type, pci_read_field(dev, PCI_SECONDARY_BUS, 1), arg); + } +} + +static void pci_scan_slot(pci_func_t f, int type, int bus, int slot, void *arg) +{ + uint32_t dev = pci_box_device(bus, slot, 0); + if (pci_read_field(dev, PCI_VENDOR_ID, 2) == PCI_NONE) { + return; + } + + pci_scan_func(f, type, bus, slot, 0, arg); + if (!pci_read_field(dev, PCI_HEADER_TYPE, 1)) { + return; + } + + for (int func = 1; func < 8; func++) { + uint32_t dev = pci_box_device(bus, slot, func); + if (pci_read_field(dev, PCI_VENDOR_ID, 2) != PCI_NONE) { + pci_scan_func(f, type, bus, slot, func, arg); + } + } +} + +static void pci_scan_bus(pci_func_t f, int type, int bus, void *arg) +{ + for (int slot = 0; slot < 32; ++slot) { + pci_scan_slot(f, type, bus, slot, arg); + } +} + +void pci_scan(pci_func_t f, int type, void *arg) +{ + if ((pci_read_field(0, PCI_HEADER_TYPE, 1) & 0x80) == 0) { + pci_scan_bus(f, type, 0, arg); + return; + } + + int hit = 0; + for (int func = 0; func < 8; ++func) { + uint32_t dev = pci_box_device(0, 0, func); + if (pci_read_field(dev, PCI_VENDOR_ID, 2) == PCI_NONE) { + break; + } + + hit = 1; + pci_scan_bus(f, type, func, arg); + } + + if (hit) { + return; + } + + for (int bus = 0; bus < 256; ++bus) { + for (int slot = 0; slot < 32; ++slot) { + pci_scan_slot(f, type, bus, slot, arg); + } + } +} + +static void init_pci_device(uint32_t device, uint16_t vendid, uint16_t devid, void *arg) +{ + struct device *dev = generic_device_create(); + snprintf(dev->dev_name, sizeof dev->dev_name, "%02x:%02x.%u", + pci_get_bus(device), + pci_get_slot(device), + pci_get_func(device)); + + device_register(dev, pci_driver, bus_device_base(pci_bus)); +} + +static kern_status_t pci_bus_scan(struct device *bus) +{ + printk("pci: scanning for devices..."); + pci_scan(init_pci_device, -1, NULL); + return KERN_OK; +} + +static struct bus_device_ops pci_bus_ops = { + .scan = pci_bus_scan, +}; + +static kern_status_t online(struct kext *self) +{ + pci_driver = driver_create(self, "pci"); + if (!pci_driver) { + return KERN_NO_MEMORY; + } + + driver_register(pci_driver); + + pci_bus = bus_device_create(); + struct device *pci_base = bus_device_base(pci_bus); + snprintf(pci_base->dev_name, sizeof pci_base->dev_name, "pci"); + pci_bus->b_ops = &pci_bus_ops; + + device_register(pci_base, pci_driver, root_device()); + + return KERN_OK; +} + +DEFINE_KEXT("net.doorstuck.socks.pci", + online, NULL, + KEXT_NO_DEPENDENCIES);