X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/99f2ed669601a61f4f0210e0b481ff877cd9bea7..35a7d633d3f16c1e0539af6ca5d8e7482926cd93:/lunaix-os/hal/gfxa/vga/vga_pci.c

diff --git a/lunaix-os/hal/gfxa/vga/vga_pci.c b/lunaix-os/hal/gfxa/vga/vga_pci.c
index 1109e5a..af4f228 100644
--- a/lunaix-os/hal/gfxa/vga/vga_pci.c
+++ b/lunaix-os/hal/gfxa/vga/vga_pci.c
@@ -58,22 +58,26 @@ static u32_t palette[] = {
 #define VGA_REG_OFF 0x0400
 
 static int
-vga_pci_bind(struct device_def* devdef, struct device* pcidev_base)
+vga_pci_bind(struct device_def* devdef, morph_t* obj)
 {
-    struct pci_device* pcidev = PCI_DEVICE(pcidev_base);
+    struct pci_probe* probe;
 
-    struct pci_base_addr* fb = &pcidev->bar[0];
-    struct pci_base_addr* mmio = &pcidev->bar[2];
+    struct pci_base_addr* fb;
+    struct pci_base_addr* mmio;
+
+    probe = changeling_reveal(obj, pci_probe_morpher);
+    fb    = &probe->bar[0];
+    mmio  = &probe->bar[2];
 
     if (PCI_BAR_ADDR_IO(mmio->type)) {
         return EINVAL;
     }
 
-    pci_reg_t cmd = pci_read_cspace(pcidev->cspace_base, PCI_REG_STATUS_CMD);
+    pci_reg_t cmd = pci_read_cspace(probe->cspace_base, PCI_REG_STATUS_CMD);
 
     cmd |= (PCI_RCMD_MM_ACCESS | PCI_RCMD_DISABLE_INTR | PCI_RCMD_BUS_MASTER);
 
-    pci_write_cspace(pcidev->cspace_base, PCI_REG_STATUS_CMD, cmd);
+    pci_write_cspace(probe->cspace_base, PCI_REG_STATUS_CMD, cmd);
 
     ptr_t fb_mapped = ioremap(fb->start, FB256K);
     ptr_t mmio_mapped = ioremap(mmio->start, mmio->size);
@@ -96,27 +100,27 @@ vga_pci_bind(struct device_def* devdef, struct device* pcidev_base)
     return 0;
 }
 
-static int
-vga_pci_init(struct device_def* def)
-{
-    return pci_bind_definition_all(pcidev_def(def));
-}
 
+static bool
+vga_pci_compat(struct pci_probe* probe)
+{
 #define VGA_PCI_CLASS 0x30000
+    return pci_device_class(probe) == VGA_PCI_CLASS;
+}
 
-static bool
-vga_pci_compat(struct pci_device_def* def, 
-                struct pci_device* pcidev)
+static int
+vga_pci_register(struct device_def* def)
 {
-    return pci_device_class(pcidev) == VGA_PCI_CLASS;
+    return !pci_register_driver(def, vga_pci_compat);
 }
 
+static struct device_def vga_pci_devdef = {
+    def_device_class(GENERIC, DISP, VGA),
+    def_device_name("Generic VGA"),
+    
+    def_on_register(vga_pci_register),
+    def_on_create(vga_pci_bind),
 
-static struct pci_device_def vga_pci_devdef = {
-    .devdef = { .class = DEVCLASS(DEVIF_PCI, DEVFN_DISP, DEV_VGA),
-                .name = "Generic VGA",
-                .init = vga_pci_init,
-                .bind = vga_pci_bind },
-    .test_compatibility = vga_pci_compat
+    def_non_trivial
 };
-EXPORT_PCI_DEVICE(vga_pci, &vga_pci_devdef, load_onboot);
\ No newline at end of file
+EXPORT_DEVICE(vga_pci, &vga_pci_devdef, load_onboot);
\ No newline at end of file