X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/4bc10176c9f47063d8eddf8703832304b8151f5d..c7a60d29f54d6de10a8388d26af441c8bc48a803:/lunaix-os/arch/aarch64/boot/kremap.c

diff --git a/lunaix-os/arch/aarch64/boot/kremap.c b/lunaix-os/arch/aarch64/boot/kremap.c
index a230ece..b91d1d5 100644
--- a/lunaix-os/arch/aarch64/boot/kremap.c
+++ b/lunaix-os/arch/aarch64/boot/kremap.c
@@ -1,4 +1,6 @@
 #include <lunaix/sections.h>
+
+#include <asm/mm_defs.h>
 #include <asm-generic/init_pagetable.h>
 #include <asm/boot_stage.h>
 
@@ -17,10 +19,10 @@ kremap()
     init_pt_alloc(&alloc, to_kphysical(&kpt), sizeof(kpt));
     init_ptw_state(&ptw, &alloc, kpt_alloc_table(&alloc));
 
-    pte = mkpte(boot_start, KERNEL_DATA);
+    pte = mkpte(bootsec_start, KERNEL_DATA);
     pte = pte_mkexec(pte);
-    nr  = leaf_count(boot_end - boot_start);
-    kpt_set_ptes(&ptw, boot_start, pte, LFT_SIZE, nr);
+    nr  = leaf_count(bootsec_end - bootsec_start);
+    kpt_set_ptes(&ptw, bootsec_start, pte, LFT_SIZE, nr);
 
     kpt_mktable_at(&ptw, VMAP, L0T_SIZE);
     kpt_mktable_at(&ptw, PMAP, L2T_SIZE);