boot/zephyr: Load image to RAM on single loader

Now that's possible to load image to RAM on single loaders, add support on Zephyr port for that. Signed-off-by: Ederson de Souza <[email protected]> Signed-off-by: Tom Burdick <[email protected]>
mcu-tools · Oct 22, 2024 · 36f710e · 36f710e
1 parent f59cb85
commit 36f710e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/boot/zephyr/single_loader.c b/boot/zephyr/single_loader.c
@@ -111,14 +111,29 @@ boot_go(struct boot_rsp *rsp)
     if (rc != 0)
         goto out;
 
+#ifdef MCUBOOT_RAM_LOAD
+        static struct boot_loader_state state;
+        state.imgs[0][0].hdr = _hdr;
+
+        rc = boot_load_image_to_sram(&state);
+        if (rc != 0)
+            goto out;
+#endif
+
 #ifdef MCUBOOT_VALIDATE_PRIMARY_SLOT
     FIH_CALL(boot_image_validate, fih_rc, _fa_p, &_hdr);
     if (FIH_NOT_EQ(fih_rc, FIH_SUCCESS)) {
+#ifdef MCUBOOT_RAM_LOAD
+        boot_remove_image_from_sram(&state);
+#endif
         goto out;
     }
 #elif defined(MCUBOOT_VALIDATE_PRIMARY_SLOT_ONCE)
     FIH_CALL(boot_image_validate_once, fih_rc, _fa_p, &_hdr);
     if (FIH_NOT_EQ(fih_rc, FIH_SUCCESS)) {
+#ifdef MCUBOOT_RAM_LOAD
+        boot_remove_image_from_sram(&state);
+#endif
         goto out;
     }
 #else