uefi: Fix crash when pressing 'q' during -b pagination

The uefi crate's println! macro calls .expect() on write results. When
the user presses 'q' during -b page break mode, the UEFI shell's
ConsoleLogger disables output and returns EFI_DEVICE_ERROR for all
subsequent writes. This causes the .expect() to panic, resulting in a
page fault crash.

Replace the uefi crate's print!/println! macros with non-panicking
versions that silently ignore write errors, allowing the application to
exit cleanly after the user breaks out of pagination.

Signed-off-by: Daniel Schaefer <dhs@frame.work>

Author: JohnAZoidberg

framework_lib/src/fw_uefi/mod.rs

@@ -11,6 +11,19 @@ use uefi_raw::protocol::shell::ShellProtocol;
 
 pub mod fs;
 
+/// Non-panicking print helper for UEFI.
+///
+/// The uefi crate's `println!` calls `.expect()` on write results, which
+/// panics when the UEFI shell's ConsoleLogger returns an error (e.g. after
+/// the user presses 'q' during `-b` page break mode). This version silently
+/// ignores write errors to avoid crashing.
+#[doc(hidden)]
+pub fn _print_safe(args: core::fmt::Arguments) {
+    uefi::system::with_stdout(|stdout| {
+        let _ = core::fmt::Write::write_fmt(stdout, args);
+    });
+}
+
 fn get_shell_protocol() -> &'static ShellProtocol {
     let handle = boot::get_handle_for_protocol::<Shell>().expect("No Shell handles");
 

framework_lib/src/lib.rs

@@ -31,9 +31,23 @@ pub mod touchscreen_win;
 pub mod usbhub;
 
 #[cfg(feature = "uefi")]
-#[macro_use]
 extern crate uefi;
 
+// Override uefi crate's print!/println! macros with non-panicking versions.
+// The uefi crate's versions call .expect() on write results, which crashes
+// when the UEFI shell returns an error after 'q' is pressed during -b pagination.
+#[cfg(feature = "uefi")]
+macro_rules! print {
+    ($($arg:tt)*) => ($crate::fw_uefi::_print_safe(core::format_args!($($arg)*)));
+}
+#[cfg(feature = "uefi")]
+macro_rules! println {
+    () => ($crate::fw_uefi::_print_safe(core::format_args!("\n")));
+    ($($arg:tt)*) => ($crate::fw_uefi::_print_safe(
+        core::format_args!("{}\n", core::format_args!($($arg)*))
+    ));
+}
+
 pub mod capsule;
 pub mod capsule_content;
 pub mod ccgx;