From efbc6c08a0c37f88a6390c72eaf134b828e88419 Mon Sep 17 00:00:00 2001
From: Sarah GLINER <sarah.gliner@ledger.fr>
Date: Mon, 29 Jan 2024 12:16:37 +0100
Subject: [PATCH] Display: Fix processing screen not showing on nano devices

---
 src/boilerplate/io.c | 2 +-
 src/ui/display.c     | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/boilerplate/io.c b/src/boilerplate/io.c
index edac0d3c4..f9af7b555 100644
--- a/src/boilerplate/io.c
+++ b/src/boilerplate/io.c
@@ -47,7 +47,7 @@ struct {
 
 // set to true when the "Processing..." screen is shown, in order for the dispatcher to know if the
 // UX is not in idle state at the end of a command handler.
-bool G_was_processing_screen_shown;
+bool G_was_processing_screen_shown = false;
 
 uint16_t G_interruption_timeout_start_tick;
 uint16_t G_processing_timeout_start_tick;
diff --git a/src/ui/display.c b/src/ui/display.c
index 8c6ce766e..cc3eb8643 100644
--- a/src/ui/display.c
+++ b/src/ui/display.c
@@ -14,6 +14,8 @@
 // These globals are a workaround for a limitation of the UX library that
 // does not allow to pass proper callbacks and context.
 
+extern bool G_was_processing_screen_shown;
+
 static bool g_ux_flow_ended;
 static bool g_ux_flow_response;
 
@@ -35,6 +37,8 @@ void set_ux_flow_response(bool approved) {
 // so that the dispatcher refreshes resets the UI at the end of the command handler.
 // Returns true/false depending if the user accepted in the corresponding UX flow.
 static bool io_ui_process(dispatcher_context_t *context, bool set_dirty) {
+    G_was_processing_screen_shown = false;
+
     g_ux_flow_ended = false;
 
     if (set_dirty) {