Add panic and collections

Author: jackpot51

uefi_std/src/lib.rs

@@ -1,6 +1,7 @@
 #![no_std]
 #![feature(alloc)]
 #![feature(core_intrinsics)]
+#![feature(lang_items)]
 #![feature(prelude_import)]
 #![feature(raw)]
 #![feature(slice_concat_ext)]
@@ -66,6 +67,7 @@ pub use alloc_crate::borrow;
 pub use alloc_crate::fmt;
 pub use alloc_crate::format;
 pub use core::pin;
+pub use alloc_crate::collections;
 pub use alloc_crate::slice;
 pub use alloc_crate::str;
 pub use alloc_crate::string;
@@ -81,13 +83,18 @@ pub use uefi_alloc;
 
 pub mod io;
 pub mod math;
+pub mod panic;
 pub mod rt;
 
 #[global_allocator]
 static ALLOCATOR: uefi_alloc::Allocator = uefi_alloc::Allocator;
 
-pub static mut HANDLE: uefi::Handle = uefi::Handle(0);
-pub static mut SYSTEM_TABLE: *mut uefi::system::SystemTable = 0 as *mut uefi::system::SystemTable;
+static mut HANDLE: uefi::Handle = uefi::Handle(0);
+static mut SYSTEM_TABLE: *mut uefi::system::SystemTable = 0 as *mut uefi::system::SystemTable;
+
+pub fn handle() -> uefi::Handle {
+    unsafe { HANDLE }
+}
 
 pub fn system_table() -> &'static uefi::system::SystemTable {
     unsafe { & *SYSTEM_TABLE }

uefi_std/src/panic.rs

@@ -0,0 +1,39 @@
+// These functions are used by the compiler, but not
+// for a bare-bones hello world. These are normally
+// provided by libstd.
+#[lang = "eh_personality"]
+#[no_mangle]
+pub extern fn rust_eh_personality() {}
+
+// This function may be needed based on the compilation target.
+#[lang = "eh_unwind_resume"]
+#[no_mangle]
+pub extern fn rust_eh_unwind_resume() {
+    loop {}
+}
+
+#[panic_handler]
+#[no_mangle]
+pub extern fn rust_begin_panic(pi: &::core::panic::PanicInfo) -> ! {
+    print!("SETUP PANIC: {}", pi);
+
+    loop {}
+}
+
+#[lang = "oom"]
+#[no_mangle]
+pub extern "C" fn rust_oom(layout: ::core::alloc::Layout) -> ! {
+    println!(
+        "SETUP OOM: {} bytes aligned to {} bytes\n",
+        layout.size(),
+        layout.align()
+    );
+
+    loop {}
+}
+
+#[allow(non_snake_case)]
+#[no_mangle]
+pub extern fn _Unwind_Resume() {
+    loop {}
+}