Add on_log_quit.

Author: guilt

src/bin/rustup-init.rs

@@ -13,13 +13,18 @@
 //! different name.
 
 #![recursion_limit = "1024"]
+#![feature(rt)]
 
 use std::path::PathBuf;
+#[cfg(feature="rt")]
+use std::rt;
 
 use cfg_if::cfg_if;
 use rs_tracing::*;
 
 use rustup::cli::common;
+#[cfg(feature="rt")]
+use rustup::cli::log;
 use rustup::cli::errors::*;
 use rustup::cli::proxy_mode;
 use rustup::cli::rustup_mode;
@@ -42,6 +47,10 @@ fn main() {
 }
 
 fn run_rustup() -> Result<utils::ExitCode> {
+    #[cfg(feature="rt")]
+    {
+        let _ = rt::at_exit(log::on_quit_log);
+    }
     if let Ok(dir) = process().var("RUSTUP_TRACE_DIR") {
         open_trace_file!(dir)?;
     }

src/cli/log.rs

@@ -23,6 +23,11 @@ macro_rules! debug {
     ( $ ( $ arg : tt ) * ) => ( $crate::cli::log::debug_fmt ( format_args ! ( $ ( $ arg ) * ) ) )
 }
 
+pub fn on_quit_log() {
+    let mut t = term2::stderr();
+    let _ = t.reset();
+}
+
 pub fn warn_fmt(args: fmt::Arguments<'_>) {
     let mut t = term2::stderr();
     let _ = t.fg(term2::color::YELLOW);