dev/pci: Unmap memory on allocation failure

dev/block/nvme/pci.c

@@ -39,9 +39,15 @@ static errno_t map_bar(struct pci_bar* bar, void** out) {
 
     for (size_t i = 0; i < len; i += page_size) {
         errno_t status = pmap_map(&kernel_vma.pmap, vaddr + i, bar->address + i, PTE_READ | PTE_WRITE, CACHE_MMIO);
-        if (status)
+        if (status) {
+            while (i) {
+                i -= page_size;
+                pmap_unmap(&kernel_vma.pmap, vaddr + i);
+            }
+            vm_free(vaddr, len);
             return status;
         }
+    }
 
     *out = (void*)vaddr;
     return 0;
@@ -81,8 +87,10 @@ static errno_t nvme_pci_probe(struct pci_device* dev, const struct pci_match* ma
 
     if (irq_line) {
         struct nvme_irq_handler* handler = mem_alloc(sizeof(*handler), 0);
-        if (!handler)
+        if (!handler) {
+            mem_free(controller);
             return ENOMEM;
+        }
         handler->handler.thread = nvme_irq_thread;
         handler->controller = controller;
 

dev/pci/core.c

@@ -224,9 +224,15 @@ errno_t pci_setup_msix(struct pci_device* dev, struct irq_line** out) {
         return ENOMEM;
     for (size_t i = 0; i < map_len; i += page_size) {
         status = pmap_map(&kernel_vma.pmap, vaddr + i, page_base + i, PTE_READ | PTE_WRITE, CACHE_MMIO);
-        if (status)
+        if (status) {
+            while (i) {
+                i -= page_size;
+                pmap_unmap(&kernel_vma.pmap, vaddr + i);
+            }
+            vm_free(vaddr, map_len);
             return status;
         }
+    }
     void* entry = (void*)(vaddr + page_offset);
 
     struct msi_line* msi;

dev/pci/ecam.c

@@ -43,9 +43,15 @@ errno_t pci_ecam_register(phys_t base, uint16_t domain, uint8_t start_bus, uint8
 
     for (size_t i = 0; i < length; i += arch_mem_page_size()) {
         errno_t status = pmap_map(&kernel_vma.pmap, vaddr + i, phys + i, PTE_READ | PTE_WRITE, CACHE_MMIO);
-        if (status != 0)
+        if (status != 0) {
+            while (i) {
+                i -= arch_mem_page_size();
+                pmap_unmap(&kernel_vma.pmap, vaddr + i);
+            }
+            vm_free(vaddr, length);
             return status;
         }
+    }
 
     struct pci_ecam* ecam = mem_alloc(sizeof(*ecam), 0);
     if (ecam == nullptr)