From 020cfb89c63642e0114f6081b9d477152ec09184 Mon Sep 17 00:00:00 2001
From: TuxSH <1922548+TuxSH@users.noreply.github.com>
Date: Wed, 11 Mar 2020 02:19:45 +0000
Subject: [PATCH] thermosphere: minor refactor

---
 thermosphere/src/hvisor_exception_dispatcher.cpp | 8 +++++---
 thermosphere/src/hvisor_memory_map.hpp           | 5 +++++
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/thermosphere/src/hvisor_exception_dispatcher.cpp b/thermosphere/src/hvisor_exception_dispatcher.cpp
index 4c3ccda2d..4313faa98 100644
--- a/thermosphere/src/hvisor_exception_dispatcher.cpp
+++ b/thermosphere/src/hvisor_exception_dispatcher.cpp
@@ -18,6 +18,8 @@
 #include "hvisor_irq_manager.hpp"
 #include "hvisor_fpu_register_cache.hpp"
 #include "hvisor_guest_timers.hpp"
+#include "hvisor_generic_timer.hpp"
+#include "hvisor_memory_map.hpp"
 
 #include "traps/hvisor_traps_data_abort.hpp"
 #include "traps/hvisor_traps_hvc.hpp"
@@ -50,10 +52,10 @@ namespace ams::hvisor {
         EnableGuestTimerTraps();
     }
 
-void DumpStackFrame(ExceptionStackFrame *frame, bool sameEl)
+    void DumpStackFrame(ExceptionStackFrame *frame, bool sameEl)
     {
 #ifndef NDEBUG
-        uintptr_t stackTop = memoryMapGetStackTop(currentCoreCtx->GetCoreId());
+        uintptr_t stackTop = MemoryMap::GetStackTopVa(currentCoreCtx->GetCoreId());
 
         for (u32 i = 0; i < 30; i += 2) {
             DEBUG("x%u\t\t%016llx\t\tx%u\t\t%016llx\n", i, frame->x[i], i + 1, frame->x[i + 1]);
@@ -111,7 +113,7 @@ void DumpStackFrame(ExceptionStackFrame *frame, bool sameEl)
             }
 
             // Update virtual counter
-            u64 ticksNow = 0;// TODO timerGetSystemTick();
+            u64 ticksNow = GenericTimer::GetSystemTick();
             currentCoreCtx->IncrementTotalTimeInHypervisor(ticksNow - frame->cntpct_el0);
             UpdateVirtualOffsetSysreg();
 
diff --git a/thermosphere/src/hvisor_memory_map.hpp b/thermosphere/src/hvisor_memory_map.hpp
index 28127c638..ec66cf368 100644
--- a/thermosphere/src/hvisor_memory_map.hpp
+++ b/thermosphere/src/hvisor_memory_map.hpp
@@ -80,6 +80,11 @@ namespace ams::hvisor {
             static void SetupMmu(const LoadImageLayout *layout);
             static std::array<uintptr_t, 2> EnableMmuGetStacks(const LoadImageLayout *layout, u32 coreId);
 
+            static constexpr uintptr_t GetStackTopVa(u32 coreId)
+            {
+                return stacksBottomVa + 0x2000 * coreId + 0x1000;
+            }
+
             // Caller is expected to invalidate TLB + barrier at some point
             static uintptr_t MapPlatformMmio(uintptr_t pa, size_t size);