X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/01a605198da7bc5883a3dfa96b38f4f7c04f67f9..b60166b327a9108b07e3069fa6568a451529ffd9:/lunaix-os/kernel/mm/mmio.c

diff --git a/lunaix-os/kernel/mm/mmio.c b/lunaix-os/kernel/mm/mmio.c
index 5e0d01e..ccb57d7 100644
--- a/lunaix-os/kernel/mm/mmio.c
+++ b/lunaix-os/kernel/mm/mmio.c
@@ -1,18 +1,27 @@
 #include <lunaix/mm/mmio.h>
 #include <lunaix/mm/pmm.h>
 #include <lunaix/mm/vmm.h>
+#include <lunaix/spike.h>
 
 void*
-ioremap(uintptr_t paddr, uint32_t size)
+ioremap(ptr_t paddr, u32_t size)
 {
-    return vmm_vmap(paddr, size, PG_PREM_RW | PG_DISABLE_CACHE);
+    void* ptr = vmap(paddr, size, PG_PREM_RW | PG_DISABLE_CACHE, 0);
+
+    if (ptr) {
+        pmm_mark_chunk_occupied(paddr >> PG_SIZE_BITS,
+                                CEIL(size, PG_SIZE_BITS),
+                                PP_FGLOCKED);
+    }
+
+    return ptr;
 }
 
-void*
-iounmap(uintptr_t vaddr, uint32_t size)
+void
+iounmap(ptr_t vaddr, u32_t size)
 {
     for (size_t i = 0; i < size; i += PG_SIZE) {
-        uintptr_t paddr = vmm_del_mapping(PD_REFERENCED, vaddr + i);
-        pmm_free_page(KERNEL_PID, paddr);
+        ptr_t paddr = vmm_del_mapping(VMS_SELF, vaddr + i);
+        pmm_free_page(paddr);
     }
 }
\ No newline at end of file