dev/pci: Unmap memory on allocation failure

dev/pci/core.c

@@ -224,8 +224,14 @@ 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);
 

dev/pci/ecam.c

@@ -43,8 +43,14 @@ 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);