Debug: Adds RTT support to bootloader

bootloader.ld

@@ -110,6 +110,17 @@ SECTIONS
     . = ALIGN(4);
     _etext = .;
 
+    /* Place RTT allocations first in RAM in debug builds, so that they are
+     * aligned between bootloader and firmware */
+    .rtt (NOLOAD) :
+    {
+        . = ALIGN(4);
+        _srtt = .;
+        *(.segger_rtt);
+        *(.segger_rtt_buf);
+        _ertt = .;
+    } > ram
+
     .relocate :
     {
         . = ALIGN(4);

firmware.ld

@@ -109,6 +109,17 @@ SECTIONS
     . = ALIGN(4);
     _etext = .;
 
+    /* Place RTT allocations first in RAM in debug builds, so that they are
+     * aligned between bootloader and firmware */
+    .rtt (NOLOAD) :
+    {
+        . = ALIGN(4);
+        _srtt = .;
+        *(.segger_rtt);
+        *(.segger_rtt_buf);
+        _ertt = .;
+    } > ram
+
     .relocate :
     {
         . = ALIGN(4);

src/bootloader/bootloader.c

@@ -26,6 +26,7 @@
 #include <memory/memory_shared.h>
 #include <memory/nvmctrl.h>
 #include <pukcc/curve_p256.h>
+#include <rust/rust.h>
 #include <screen.h>
 #include <ui/components/ui_images.h>
 #include <ui/fonts/arial_fonts.h>
@@ -206,6 +207,8 @@ void _binExec(void* l_code_addr)
 
 static void _binary_exec(void)
 {
+    util_log("Jumping to firmware");
+    rust_rtt_flush();
     _render_bootloader_finished_marker();
 
     int i;
@@ -981,6 +984,7 @@ void bootloader_jump(void)
     }
 
     // App not entered. Start USB API to receive boot commands
+    util_log("Not jumping to firmware");
     _compute_is_app_flash_empty();
     _render_default_screen();
     if (usb_start(_api_setup) != ERR_NONE) {

src/hardfault.c

@@ -34,6 +34,7 @@ void MemManage_Handler(void)
 
 void Abort(const char* msg)
 {
+    util_log("%s", msg);
     screen_print_debug(msg, 0);
     usb_stop();
 #if !defined(TESTING)

src/platform/platform_init.c

@@ -20,16 +20,22 @@
 #endif
 #include "rust/rust.h"
 
+#if defined(BOOTLOADER)
+#define PREFIX "boot"
+#else
+#define PREFIX "app"
+#endif
+
 void platform_init(void)
 {
     oled_init();
-#if !defined(BOOTLOADER)
 // The factory setup image already has a c implementation of RTT.
 #if FACTORYSETUP != 1
     // these two functions are noops if "rtt" feature isn't enabled in rust
     rust_rtt_init();
-    util_log("platform_init");
+    util_log(PREFIX ": platform_init");
 #endif
+#if !defined(BOOTLOADER)
     sd_mmc_start();
 #endif
 }

src/rust/.cargo/config.toml

@@ -1,6 +1,11 @@
 [source.crates-io]
 replace-with = "vendored-sources"
 
+[source."git+https://github.com/BitBoxSwiss/rtt-target.git?branch=bitbox02"]
+git = "https://github.com/BitBoxSwiss/rtt-target.git"
+branch = "bitbox02"
+replace-with = "vendored-sources"
+
 [source."git+https://github.com/digitalbitbox/rust-bip32-ed25519?tag=v0.1.2"]
 git = "https://github.com/BitBoxSwiss/rust-bip32-ed25519"
 tag = "v0.2.0"

src/rust/Cargo.toml

@@ -36,6 +36,9 @@ sha2 = { version = "0.10.8", default-features = false }
 sha3 = { version = "0.10.8", default-features = false }
 zeroize = "1.7.0"
 
+[patch.crates-io]
+rtt-target = { git = "https://github.com/BitBoxSwiss/rtt-target.git", branch = "bitbox02" }
+
 [profile.release]
 # This only affects the .elf output. Debug info is stripped from the final .bin.
 # Paths to source code can still appear in the final bin, as they are part of the panic!() output.

src/rust/bitbox02-rust-c/src/lib.rs

@@ -56,6 +56,11 @@ pub extern "C" fn rust_rtt_init() {
     ::util::log::rtt_init();
 }
 
+#[no_mangle]
+pub extern "C" fn rust_rtt_flush() {
+    ::util::log::rtt_flush();
+}
+
 /// # Safety
 ///
 /// The pointer `ptr` must point to a null terminated string

src/rust/util/Cargo.toml

@@ -22,7 +22,7 @@ license = "Apache-2.0"
 
 [dependencies]
 num-bigint = { workspace = true, default-features = false }
-rtt-target = { version = "0.5.0", optional = true }
+rtt-target = { version = "0.6.1", optional = true }
 cortex-m = { version = "0.7.7", features = ["critical-section-single-core"], optional = true }
 
 [features]

src/rust/util/src/log.rs

@@ -13,6 +13,27 @@ pub use log;
 
 pub fn rtt_init() {
     #[cfg(feature = "rtt")]
-    rtt_target::rtt_init_print!();
-    log!("RTT Initialized");
+    {
+        let channels = rtt_target::rtt_init! {
+            up: {
+                0: {
+                    size: 1024,
+                    mode: rtt_target::ChannelMode::NoBlockSkip,
+                    name: "Terminal",
+                    section: ".segger_rtt_buf",
+                }
+            }
+            section_cb: ".segger_rtt"
+        };
+
+        rtt_target::set_print_channel(channels.up.0);
+
+        log!("RTT Initialized");
+    }
+}
+
+/// Wait until all messages have been read by host
+pub fn rtt_flush() {
+    #[cfg(feature = "rtt")]
+    rtt_target::with_terminal_channel(|c| c.flush());
 }