diff --git a/src/lib.rs b/src/lib.rs index 94ee9290..3e7673a5 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -2,6 +2,7 @@ #![cfg_attr(test, no_main)] #![feature(custom_test_frameworks)] #![feature(abi_x86_interrupt)] +#![feature(alloc_error_handler)] #![test_runner(crate::test_runner)] #![reexport_test_harness_main = "test_main"] @@ -82,3 +83,8 @@ fn test_kernel_main(_boot_info: &'static BootInfo) -> ! { fn panic(info: &PanicInfo) -> ! { test_panic_handler(info) } + +#[alloc_error_handler] +fn alloc_error_handler(layout: alloc::alloc::Layout) -> ! { + panic!("allocation error: {:?}", layout) +}