From cc98f98aa75be5561d2db3a0d13f3db0d2968963 Mon Sep 17 00:00:00 2001
From: arlecchino <kolbma@users.noreply.github.com>
Date: Mon, 17 Jan 2022 01:33:59 +0100
Subject: [PATCH] Implement and log with panic-free 'write_out!'.

Resolves #2019.
---
 core/lib/src/log.rs | 40 +++++++++++++++++++++++++---------------
 1 file changed, 25 insertions(+), 15 deletions(-)

diff --git a/core/lib/src/log.rs b/core/lib/src/log.rs
index 9590c113..bfe26df5 100644
--- a/core/lib/src/log.rs
+++ b/core/lib/src/log.rs
@@ -35,6 +35,16 @@ define_log_macro!(trace, trace_);
 define_log_macro!(launch_info: info, "rocket::launch", $);
 define_log_macro!(launch_info_: info, "rocket::launch_", $);
 
+// `print!` panics when stdout isn't available, but this macro doesn't.
+// See SergioBenitez/Rocket#2019 and rust-lang/rust#46016 for more.
+macro_rules! write_out {
+    ($($arg:tt)*) => ({
+        use std::io::{Write, stdout, stderr};
+
+        let _ = write!(stdout(), $($arg)*).or_else(|e| write!(stderr(), "{}", e));
+    })
+}
+
 #[derive(Debug)]
 struct RocketLogger;
 
@@ -86,7 +96,7 @@ impl log::Log for RocketLogger {
         // In Rocket, we abuse targets with suffix "_" to indicate indentation.
         let indented = record.target().ends_with('_');
         if indented {
-            print!("   {} ", Paint::default(">>").bold());
+            write_out!("   {} ", Paint::default(">>").bold());
         }
 
         // Downgrade a physical launch `warn` to logical `info`.
@@ -96,30 +106,30 @@ impl log::Log for RocketLogger {
 
         match level {
             log::Level::Error if !indented => {
-                println!("{} {}",
-                         Paint::red("Error:").bold(),
-                         Paint::red(record.args()).wrap())
+                write_out!("{} {}\n",
+                    Paint::red("Error:").bold(),
+                    Paint::red(record.args()).wrap());
             }
             log::Level::Warn if !indented => {
-                println!("{} {}",
-                         Paint::yellow("Warning:").bold(),
-                         Paint::yellow(record.args()).wrap())
+                write_out!("{} {}\n",
+                    Paint::yellow("Warning:").bold(),
+                    Paint::yellow(record.args()).wrap());
             }
-            log::Level::Info => println!("{}", Paint::blue(record.args()).wrap()),
-            log::Level::Trace => println!("{}", Paint::magenta(record.args()).wrap()),
-            log::Level::Warn => println!("{}", Paint::yellow(record.args()).wrap()),
-            log::Level::Error => println!("{}", Paint::red(record.args()).wrap()),
+            log::Level::Info => write_out!("{}\n", Paint::blue(record.args()).wrap()),
+            log::Level::Trace => write_out!("{}\n", Paint::magenta(record.args()).wrap()),
+            log::Level::Warn => write_out!("{}\n", Paint::yellow(record.args()).wrap()),
+            log::Level::Error => write_out!("{}\n", Paint::red(record.args()).wrap()),
             log::Level::Debug => {
-                print!("\n{} ", Paint::blue("-->").bold());
+                write_out!("\n{} ", Paint::blue("-->").bold());
                 if let Some(file) = record.file() {
-                    print!("{}", Paint::blue(file));
+                    write_out!("{}", Paint::blue(file));
                 }
 
                 if let Some(line) = record.line() {
-                    println!(":{}", Paint::blue(line));
+                    write_out!(":{}\n", Paint::blue(line));
                 }
 
-                println!("\t{}", record.args());
+                write_out!("\t{}\n", record.args());
             }
         }
     }