feat: probe more device info

[lunaix-os.git] / lunaix-os / hal / pci.c

diff --git a/lunaix-os/hal/pci.c b/lunaix-os/hal/pci.c
index 90011c6eb200a4ebcdca165afe2d74cda52e9d09..4bf4c56863dac16b6fa4b16e483ea97557199376 100644 (file)
--- a/lunaix-os/hal/pci.c
+++ b/lunaix-os/hal/pci.c
@@ -1,5 +1,18 @@
+/**
+ * @file pci.c
+ * @author Lunaixsky (zelong56@gmail.com)
+ * @brief A software implementation of PCI Local Bus Specification Revision 3.0
+ * @version 0.1
+ * @date 2022-06-28
+ *
+ * @copyright Copyright (c) 2022
+ *
+ */
+#include <hal/acpi/acpi.h>
+#include <hal/apic.h>
 #include <hal/pci.h>
 #include <lunaix/mm/kalloc.h>
+#include <lunaix/spike.h>
 #include <lunaix/syslog.h>
 
 LOG_MODULE("PCI")
@@ -23,7 +36,10 @@ pci_probe_device(int bus, int dev, int funct)
     pci_reg_t hdr_type = pci_read_cspace(base, 0xc);
     hdr_type = (hdr_type >> 16) & 0xff;
 
-    if ((hdr_type & 0x80)) {
+    // 防止堆栈溢出
+    // QEMU的ICH9/Q35实现似乎有点问题，对于多功能设备的每一个功能的header type
+    //  都将第七位置位。而virtualbox 就没有这个毛病。
+    if ((hdr_type & 0x80) && funct == 0) {
         hdr_type = hdr_type & ~0x80;
         // 探测多用途设备（multi-function device）
         for (int i = 1; i < 7; i++) {
@@ -31,7 +47,7 @@ pci_probe_device(int bus, int dev, int funct)
         }
     }
 
-    if (hdr_type != 0) {
+    if (hdr_type != PCI_TDEV) {
         // XXX: 目前忽略所有桥接设备，比如PCI-PCI桥接器，或者是CardBus桥接器
         return;
     }
@@ -65,6 +81,9 @@ pci_probe()
 void
 pci_probe_msi_info(struct pci_device* device)
 {
+    // Note that Virtualbox have to use ICH9 chipset for MSI support.
+    // Qemu seems ok with default PIIX3, Bochs is pending to test...
+    //    See https://www.virtualbox.org/manual/ch03.html (section 3.5.1)
     pci_reg_t status =
       pci_read_cspace(device->cspace_base, PCI_REG_STATUS_CMD) >> 16;
 
@@ -87,6 +106,8 @@ pci_probe_msi_info(struct pci_device* device)
     }
 }
 
+#define PCI_PRINT_BAR_LISTING
+
 void
 pci_print_device()
 {
@@ -104,13 +125,72 @@ pci_print_device()
         kprintf(KINFO "\t IRQ: %d, INT#x: %d\n",
                 PCI_INTR_IRQ(pos->intr_info),
                 PCI_INTR_PIN(pos->intr_info));
-
+#ifdef PCI_PRINT_BAR_LISTING
+        pci_reg_t bar;
+        for (size_t i = 1; i <= 6; i++) {
+            size_t size = pci_bar_sizing(pos, &bar, i);
+            if (!bar)
+                continue;
+            if (PCI_BAR_MMIO(bar)) {
+                kprintf(KINFO "\t BAR#%d (MMIO) %p [%d]\n",
+                        i,
+                        PCI_BAR_ADDR_MM(bar),
+                        size);
+            } else {
+                kprintf(KINFO "\t BAR#%d (I/O) %p [%d]\n",
+                        i,
+                        PCI_BAR_ADDR_IO(bar),
+                        size);
+            }
+        }
+#endif
         if (pos->msi_loc) {
             kprintf(KINFO "\t MSI supported (@%xh)\n", pos->msi_loc);
         }
     }
 }
 
+size_t
+pci_bar_sizing(struct pci_device* dev, uint32_t* bar_out, uint32_t bar_num)
+{
+    pci_reg_t bar = pci_read_cspace(dev->cspace_base, PCI_REG_BAR(bar_num));
+    if (!bar) {
+        *bar_out = 0;
+        return 0;
+    }
+
+    pci_write_cspace(dev->cspace_base, PCI_REG_BAR(bar_num), 0xffffffff);
+    pci_reg_t sized =
+      pci_read_cspace(dev->cspace_base, PCI_REG_BAR(bar_num)) & ~0x1;
+    if (PCI_BAR_MMIO(bar)) {
+        sized = PCI_BAR_ADDR_MM(sized);
+    }
+    *bar_out = bar;
+    pci_write_cspace(dev->cspace_base, PCI_REG_BAR(bar_num), bar);
+    return ~sized + 1;
+}
+
+void
+pci_setup_msi(struct pci_device* device, int vector)
+{
+    // Dest: APIC#0, Physical Destination, No redirection
+    uint32_t msi_addr = (__APIC_BASE_PADDR);
+
+    // Edge trigger, Fixed delivery
+    uint32_t msi_data = vector;
+
+    pci_write_cspace(
+      device->cspace_base, PCI_MSI_ADDR(device->msi_loc), msi_addr);
+    pci_write_cspace(
+      device->cspace_base, PCI_MSI_DATA(device->msi_loc), msi_data & 0xffff);
+
+    pci_reg_t reg1 = pci_read_cspace(device->cspace_base, device->msi_loc);
+
+    // manipulate the MSI_CTRL to allow device using MSI to request service.
+    reg1 = ((((reg1 >> 16) & ~0x70) | 0x1) << 16) | (reg1 & 0xffff);
+    pci_write_cspace(device->cspace_base, device->msi_loc, reg1);
+}
+
 struct pci_device*
 pci_get_device_by_id(uint16_t vendorId, uint16_t deviceId)
 {
@@ -144,5 +224,12 @@ void
 pci_init()
 {
     llist_init_head(&pci_devices);
+    acpi_context* acpi = acpi_get_context();
+    assert_msg(acpi, "ACPI not initialized.");
+    if (acpi->mcfg.alloc_num) {
+        // PCIe Enhanced Configuration Mechanism is supported.
+        // TODO: support PCIe addressing mechanism
+    }
+    // Otherwise, fallback to use legacy PCI 3.0 method.
     pci_probe();
 }
\ No newline at end of file