Use heapless crate instead of custom fixed string buffer

for panic string formatting

Signed-off-by: adamperlin <[email protected]>

Author: adamperlin

Cargo.lock

@@ -41,6 +41,4 @@ pub mod mem;
 pub mod outb;
 
 /// cbindgen:ignore
-pub mod resource;
-
-pub mod fixed_buf;
+pub mod resource;

src/hyperlight_common/src/fixed_buf.rs

@@ -27,6 +27,7 @@ hyperlight-guest-tracing = { workspace = true, default-features = false }
 buddy_system_allocator = "0.11.0"
 log = { version = "0.4", default-features = false }
 spin = "0.10.0"
+heapless = "0.9.1"
 
 [lints]
 workspace = true

src/hyperlight_common/src/lib.rs

@@ -26,7 +26,7 @@ use exceptions::{gdt::load_gdt, idtr::load_idt};
 use guest_function::call::dispatch_function;
 use guest_function::register::GuestFunctionRegister;
 use guest_logger::init_logger;
-use hyperlight_common::fixed_buf::FixedStringBuf;
+use heapless::{String, CString};
 use hyperlight_common::flatbuffer_wrappers::guest_error::ErrorCode;
 use hyperlight_common::mem::HyperlightPEB;
 #[cfg(feature = "mem_profile")]
@@ -143,7 +143,7 @@ fn panic(info: &core::panic::PanicInfo) -> ! {
     _panic_handler(info)
 }
 
-static PANIC_BUF: Mutex<FixedStringBuf<512>> = Mutex::new(FixedStringBuf::new());
+static PANIC_BUF: Mutex<String<512>> = Mutex::new(String::<512>::new());
 
 #[inline(always)]
 fn _panic_handler(info: &core::panic::PanicInfo) -> ! {
@@ -158,23 +158,22 @@ fn _panic_handler(info: &core::panic::PanicInfo) -> ! {
         }
     }
 
-    // create a CStr from the underlying array in PANIC_BUF using the as_cstr method.
-    // this wraps CStr::from_bytes_until_nul which takes a borrowed byte slice
-    // and does not allocate.
-    let c_string_res = panic_buf_guard.as_c_str();
-    if c_string_res.is_err() {
+    let mut c_string: CString<512> = CString::new();
+    let extend_ok = c_string.extend_from_bytes(panic_buf_guard.as_bytes());
+
+    if extend_ok.is_err() {
         unsafe {
             abort_with_code_and_message(
                 &[ErrorCode::UnknownError as u8],
-                c"panic: failed to convert to CStr".as_ptr(),
+                c"panic: failed to convert to CString".as_ptr(),
             )
         }
     }
 
     unsafe {
         abort_with_code_and_message(
             &[ErrorCode::UnknownError as u8],
-            c_string_res.unwrap().as_ptr(),
+            c_string.as_ptr(),
         )
     }
 }

src/hyperlight_guest_bin/Cargo.toml

@@ -552,7 +552,6 @@ fn guest_panic_no_alloc() {
             (),
         )
         .unwrap_err();
-    println!("{:?}", res);
 
     if let HyperlightError::StackOverflow() = res {
         panic!("panic on OOM caused stack overflow, this implies allocation in panic handler");