From ca80408349ac9d5430b27ea7d83e5ea83e3f912c Mon Sep 17 00:00:00 2001
From: TuxSH <tuxsh@sfr.fr>
Date: Sun, 4 Mar 2018 00:36:40 +0100
Subject: [PATCH] Fix latest commit and set MAILBOX_NX_SECMON_BOOT_TIME

---
 exosphere/src/coldboot_init.c | 40 +++++++++++++++++++++++++++--------
 exosphere/src/package2.h      |  2 ++
 2 files changed, 33 insertions(+), 9 deletions(-)

diff --git a/exosphere/src/coldboot_init.c b/exosphere/src/coldboot_init.c
index fe3267613..89b009a12 100644
--- a/exosphere/src/coldboot_init.c
+++ b/exosphere/src/coldboot_init.c
@@ -3,12 +3,16 @@
 #include "mmu.h"
 #include "memory_map.h"
 #include "arm.h"
+#include "package2.h"
+#include "timers.h"
+
+#undef  MAILBOX_NX_BOOTLOADER_BASE
+#undef  TIMERS_BASE
+#define MAILBOX_NX_BOOTLOADER_BASE  (MMIO_GET_DEVICE_PA(MMIO_DEVID_NXBOOTLOADER_MAILBOX))
+#define TIMERS_BASE                 (MMIO_GET_DEVICE_PA(MMIO_DEVID_TMRs_WDTs))
 
 extern const uint8_t __start_cold[];
 
-/* warmboot_init.c */
-void set_memory_registers_enable_mmu(void);
-
 static void identity_map_all_mappings(uintptr_t *mmu_l1_tbl, uintptr_t *mmu_l3_tbl) {
     static const uintptr_t addrs[]      =   { TUPLE_FOLD_LEFT_0(EVAL(IDENTIY_MAPPING_ID_MAX), _MMAPID, COMMA) };
     static const size_t sizes[]         =   { TUPLE_FOLD_LEFT_1(EVAL(IDENTIY_MAPPING_ID_MAX), _MMAPID, COMMA) };
@@ -93,14 +97,18 @@ static void configure_ttbls(void) {
     tzram_map_all_segments(mmu_l3_tbl);
 }
 
-static void translate_warmboot_func_list(coldboot_crt0_reloc_list_t *reloc_list, boot_func_list_t *func_list) {
+static void translate_func_list(coldboot_crt0_reloc_list_t *reloc_list, boot_func_list_t *func_list, bool in_tzram) {
     coldboot_crt0_reloc_t *warmboot_crt0_reloc = &reloc_list->relocs[0];
     coldboot_crt0_reloc_t *main_reloc = &reloc_list->relocs[reloc_list->nb_relocs_pre_mmu_init];
 
-    /* The main segment immediately follows the warmboot crt0 in TZRAM, in the same page. */
+    uintptr_t main_pa;
+    if (in_tzram) {
+        /* The main segment immediately follows the warmboot crt0 in TZRAM, in the same page. */
+        main_pa = (uintptr_t)warmboot_crt0_reloc->vma | ((uintptr_t)main_reloc->vma & 0xFFF);
+    } else {
+        main_pa = reloc_list->reloc_base + main_reloc->reloc_offset;
+    }
 
-    uintptr_t main_pa = (uintptr_t)warmboot_crt0_reloc->vma | ((uintptr_t)main_reloc->vma & 0xFFF);
-    
     for(size_t i = 0; i < func_list->nb_funcs; i++) {
         if(func_list->addrs[i] >= 0x1F0000000ull) {
             func_list->addrs[i] = main_pa + func_list->addrs[i] - (uintptr_t)main_reloc->vma;
@@ -123,8 +131,23 @@ uintptr_t get_coldboot_crt0_stack_address(void) {
 }
 
 void coldboot_init(coldboot_crt0_reloc_list_t *reloc_list, boot_func_list_t *func_list, boot_func_list_t *func_list_warmboot) {
+    MAILBOX_NX_SECMON_BOOT_TIME = TIMERUS_CNTR_1US_0;
+
     /* Custom approach */
     reloc_list->reloc_base = (uintptr_t)__start_cold;
+    translate_func_list(reloc_list, func_list, false);
+
+    /*
+        From https://events.static.linuxfound.org/sites/events/files/slides/slides_17.pdf :
+        Caches may write back dirty lines at any time:
+            - To make space for new allocations
+            - Even if MMU is off
+            - Even if Cacheable accesses are disabled (caches are never 'off')
+
+        It should be fine to clear that here and not before.
+    */
+    func_list->funcs.flush_dcache_all();
+    func_list->funcs.invalidate_icache_all();
 
     /* TODO: Set NX BOOTLOADER clock time field */
     
@@ -133,7 +156,6 @@ void coldboot_init(coldboot_crt0_reloc_list_t *reloc_list, boot_func_list_t *fun
         do_relocation(reloc_list, i);
     }
     /* At this point, we can (and will) access functions located in .warm_crt0 */
-    translate_warmboot_func_list(reloc_list, func_list);
 
     /* Initialize DMA controllers, and write to AHB_GIZMO_TZRAM. */
     /* TZRAM accesses should work normally after this point. */
@@ -151,5 +173,5 @@ void coldboot_init(coldboot_crt0_reloc_list_t *reloc_list, boot_func_list_t *fun
     func_list->funcs.invalidate_icache_all();
     /* At this point we can access all the mapped segments (all other functions, data...) normally */
 
-    *func_list_warmboot = *func_list;
+    translate_func_list(reloc_list, func_list_warmboot, true);
 }
diff --git a/exosphere/src/package2.h b/exosphere/src/package2.h
index fad333b79..c1113ecf0 100644
--- a/exosphere/src/package2.h
+++ b/exosphere/src/package2.h
@@ -14,6 +14,8 @@ static inline uintptr_t get_nx_bootloader_mailbox_base(void) {
 
 #define MAILBOX_NX_BOOTLOADER_BASE (get_nx_bootloader_mailbox_base())
 
+#define MAILBOX_NX_SECMON_BOOT_TIME       MAKE_REG32(MAILBOX_NX_BOOTLOADER_BASE + 0xE08ull)
+
 #define MAILBOX_NX_BOOTLOADER_SETUP_STATE MAKE_REG32(MAILBOX_NX_BOOTLOADER_BASE + 0xEF8ull)
 
 #define NX_BOOTLOADER_STATE_INIT 0