From c8c8e58f10488fba14adb85c8162fe1aeb467fe4 Mon Sep 17 00:00:00 2001
From: Hankai Zhang <hsamzhang@gmail.com>
Date: Sat, 10 Jun 2023 17:49:36 -0400
Subject: [PATCH] Colorize panic and program crash message

---
 kernel/src/interrupt.rs | 3 ++-
 kernel/src/main.rs      | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/kernel/src/interrupt.rs b/kernel/src/interrupt.rs
index 8c5f73f..49858e4 100644
--- a/kernel/src/interrupt.rs
+++ b/kernel/src/interrupt.rs
@@ -251,6 +251,7 @@ fn cpu_exception_common(frame: &InterruptStackFrame, id: u64, errcode: Option<u6
             // handler; this puts the interrupt flag and cli_count in agreement.
             interrupts::enable();
 
+            kprint!("\x1B[1;35m");
             kprintln!("Thread {} has crashed and was terminated.", thread::id());
             kprint!("[{}, stack frame: {:?}, ", exception_id_to_name(id), frame);
             if id == 14 {
@@ -264,7 +265,7 @@ fn cpu_exception_common(frame: &InterruptStackFrame, id: u64, errcode: Option<u6
             } else {
                 kprintln!("error code: {:?}]", errcode);
             }
-
+            kprint!("\x1B[m");
             thread::terminate();
             // No code afterwards will be run in this case
         }
diff --git a/kernel/src/main.rs b/kernel/src/main.rs
index 43d2eb6..1b08786 100644
--- a/kernel/src/main.rs
+++ b/kernel/src/main.rs
@@ -217,7 +217,7 @@ fn panic(panic_info: &PanicInfo) -> ! {
     if let Some(mut locked_term) = TERMINAL.as_inner().try_lock() {
         // The terminal may not yet be initialized.
         if let Some(ref mut term) = &mut *locked_term {
-            let _ = core::fmt::write(term, format_args!("{}", panic_info));
+            let _ = core::fmt::write(term, format_args!("\x1B[1;31m{}\x1B[m", panic_info));
         }
     }