X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/bb5ae5c85c0812c52e8d53187f2c6b0ee9525c8c..ca2485429c19d3e5cdecbb39a5f3383b40477d94:/lunaix-os/arch/aarch64/vmutils.c

diff --git a/lunaix-os/arch/aarch64/vmutils.c b/lunaix-os/arch/aarch64/vmutils.c
index aef8c4c..405a32b 100644
--- a/lunaix-os/arch/aarch64/vmutils.c
+++ b/lunaix-os/arch/aarch64/vmutils.c
@@ -1,18 +1,19 @@
 #include <lunaix/mm/pagetable.h>
 #include <lunaix/mm/page.h>
+#include <lunaix/sections.h>
 
-struct leaflet*
-dup_leaflet(struct leaflet* leaflet)
+struct leaflet *
+dup_leaflet(struct leaflet *leaflet)
 {
     ptr_t dest_va, src_va;
-    struct leaflet* new_leaflet;
-    
+    struct leaflet *new_leaflet;
+
     new_leaflet = alloc_leaflet(leaflet_order(leaflet));
 
     src_va = leaflet_mount(leaflet);
     dest_va = vmap(new_leaflet, KERNEL_DATA);
 
-    memcpy((void*)dest_va, (void*)src_va, PAGE_SIZE);
+    memcpy((void *)dest_va, (void *)src_va, PAGE_SIZE);
 
     leaflet_unmount(leaflet);
     vunmap(dest_va, new_leaflet);
@@ -20,14 +21,51 @@ dup_leaflet(struct leaflet* leaflet)
     return new_leaflet;
 }
 
-ptr_t
-pmm_arch_init_remap(struct pmem* memory, struct boot_handoff* bctx)
+static inline struct ppage*
+__setup_pmem_map(ptr_t start, size_t plist_size)
 {
-    unsigned long plist_len;
+    pte_t pte, *ptep;
+    unsigned int nr_pages;
+
+    pte = mkpte(start, KERNEL_DATA);
+    ptep = mkl2tep_va(VMS_SELF, PMAP);
+    nr_pages = page_count(plist_size, L2T_SIZE);
+
+    vmm_set_ptes_contig(ptep, pte, L2T_SIZE, nr_pages);
 
-    plist_len = leaf_count(bctx->mem.size) * sizeof(struct ppage);
+    return (struct ppage*)PMAP;
+}
+
+ptr_t 
+pmm_arch_init_remap(struct pmem *memory, struct boot_handoff *bctx)
+{
+    unsigned long plist_len, plist_size, nr_huges;
+    struct boot_mmapent *ment;
 
-    for (int i = 0; i < bctx->mem.mmap_len; i++) {
-        
+    plist_len = leaf_count(bctx->mem.size);
+    plist_size = plist_len * sizeof(struct ppage);
+    nr_huges = page_count(plist_size, lnt_page_size(2));
+
+    ptr_t pkernel_end = to_kphysical(kernel_end);
+    pkernel_end = napot_upaligned(pkernel_end, L2T_SIZE);
+    pkernel_end += plist_size;
+
+    for (int i = 0; i < bctx->mem.mmap_len; i++)
+    {
+        ment = &bctx->mem.mmap[i];
+        if (ment->type != BOOT_MMAP_FREE) {
+            continue;
+        }
+
+        if (pkernel_end >= ment->start + ment->size) {
+            continue;
+        }
+
+        memory->pplist = __setup_pmem_map(pkernel_end, plist_size);;
+        memory->list_len = plist_len;
+        return pkernel_end;
     }
+
+    fail("no_mem, unable to map plist");
+    return NULL;
 }