From 833d52612064b8967ce7aa777cdc211e57e4287a Mon Sep 17 00:00:00 2001
From: GuEe_GUI <2991707448@qq.com>
Date: Fri, 24 Dec 2021 11:44:58 +0800
Subject: [PATCH] fixup aarch64's bss clean and mmu early page memset to 0

---
 libcpu/aarch64/common/mmu.c           |  3 +++
 libcpu/aarch64/cortex-a/entry_point.S | 34 ++++++++++++++++++++-------
 2 files changed, 28 insertions(+), 9 deletions(-)

diff --git a/libcpu/aarch64/common/mmu.c b/libcpu/aarch64/common/mmu.c
index b02b42ff57..caabb59291 100644
--- a/libcpu/aarch64/common/mmu.c
+++ b/libcpu/aarch64/common/mmu.c
@@ -895,6 +895,9 @@ void rt_hw_mmu_setup_early(unsigned long *tbl0, unsigned long *tbl1, unsigned lo
     unsigned long count = (size + ARCH_SECTION_MASK) >> ARCH_SECTION_SHIFT;
     unsigned long normal_attr = MMU_MAP_CUSTOM(MMU_AP_KAUN, NORMAL_MEM);
 
+    mmu_memset((char *)tbl0, 0, sizeof(struct page_table));
+    mmu_memset((char *)tbl1, 0, sizeof(struct page_table));
+
     ret = armv8_init_map_2M(tbl1 , va, va + pv_off, count, normal_attr);
     if (ret != 0)
     {
diff --git a/libcpu/aarch64/cortex-a/entry_point.S b/libcpu/aarch64/cortex-a/entry_point.S
index 3c300bcae7..5ed887cedb 100644
--- a/libcpu/aarch64/cortex-a/entry_point.S
+++ b/libcpu/aarch64/cortex-a/entry_point.S
@@ -88,15 +88,31 @@ __start:
     msr     cpacr_el1, x1
 
     /* clear bss */
-    ldr     x1, =__bss_start
-    add     x1, x1, x9
-    ldr     w2, =__bss_size
-
-.L__clean_bss_loop:
-    cbz     w2, .L__jump_to_entry
-    str     xzr, [x1], #8
-    sub     w2, w2, #8
-    cbnz    w2, .L__clean_bss_loop
+    ldr     x0, =__bss_start
+    ldr     x1, =__bss_end
+    ldr     x2, =PV_OFFSET
+    add     x0, x0, x2
+    add     x1, x1, x2
+
+    sub     x2, x1, x0
+    mov     x3, x1
+    cmp     x2, #7
+    bls     .L__clean_bss_check
+
+.L__clean_bss_loop_quad:
+    str     xzr, [x0], #8
+    sub     x2, x3, x0
+    cmp     x2, #7
+    bhi     .L__clean_bss_loop_quad
+    cmp     x1, x0
+    bls     .L__jump_to_entry
+
+.L__clean_bss_loop_byte:
+    str     xzr, [x0], #1
+
+.L__clean_bss_check:
+    cmp     x1, x0
+    bhi     .L__clean_bss_loop_byte
 
 .L__jump_to_entry:          /* jump to C code, should not return */        
 
-- 
Gitee