1 files changed, 15 insertions, 0 deletions
diff --git a/screen_ui.cpp b/screen_ui.cpp
index eb3352ef..c5319ef1 100644
--- a/screen_ui.cpp
+++ b/screen_ui.cpp
@@ -251,6 +251,9 @@ void ScreenRecoveryUI::draw_dialog()
     }
     else {
         draw_background_locked(dialog_icon);
+        if (dialog_icon == HEADLESS) {
+            return;
+        }
     }
     draw_header_icon();
 
@@ -483,6 +486,7 @@ void ScreenRecoveryUI::Init()
     LoadBitmap("icon_info", &backgroundIcon[INFO]);
     LoadBitmap("icon_error", &backgroundIcon[ERROR]);
     backgroundIcon[NO_COMMAND] = backgroundIcon[ERROR];
+    LoadBitmap("icon_headless", &backgroundIcon[HEADLESS]);
 
     LoadBitmap("progress_empty", &progressBarEmpty);
     LoadBitmap("progress_fill", &progressBarFill);
@@ -663,6 +667,17 @@ void ScreenRecoveryUI::DialogDismiss()
     pthread_mutex_unlock(&updateMutex);
 }
 
+void ScreenRecoveryUI::SetHeadlessMode()
+{
+    pthread_mutex_lock(&updateMutex);
+    free(dialog_text);
+    dialog_text = strdup("");
+    dialog_show_log = false;
+    dialog_icon = HEADLESS;
+    update_screen_locked();
+    pthread_mutex_unlock(&updateMutex);
+}
+
 void ScreenRecoveryUI::StartMenu(const char* const * headers, const char* const * items,
                                  int initial_selection) {
     int i = 0;