Fix error codes (double fault handler added)

Author: katietheqt-old

src/protected/stage_3/src/interrupts.rs

@@ -8,7 +8,8 @@ lazy_static! {
 
         idt.divide_error.set_handler_fn(divide_handler);
         idt.breakpoint.set_handler_fn(breakpoint_handler);
-        //idt.double_fault.set_handler_fn(double_fault_handler);
+
+        idt.double_fault.set_handler_fn(double_fault_handler);
 
         idt
     };
@@ -33,9 +34,8 @@ extern "x86-interrupt" fn breakpoint_handler(
     println!("[Bootloader] [IDT] Breakpoint Hit");
 }
 
-/*extern "x86-interrupt" fn double_fault_handler(
-    stack_frame: &mut InterruptStackFrame, _error_code: u64) -> !
+extern "x86-interrupt" fn double_fault_handler(
+    stack_frame: &mut InterruptStackFrame, _error_code: u32) -> !
 {
-    panic!("EXCEPTION: DOUBLE FAULT\n{:#?}", stack_frame);
-    loop {};
-}*/
+    panic!("[Bootloader] [IDT] Double Fault!");
+}

src/protected/stage_3/src/lib.rs

@@ -29,9 +29,5 @@ pub extern "C" fn third_stage() -> ! {
 
     println!("[Bootloader] [32] Loaded IDT");
 
-    unsafe { asm!("int 3") };
-
-    println!("[Bootloader] [32] It didn't crash!");
-
     loop {};
 }

src/shared/src/structures/idt.rs

@@ -207,12 +207,12 @@ impl EntryOptions {
 pub type HandlerFunc = extern "x86-interrupt" fn(&mut InterruptStackFrame);
 /// A handler function for an exception that pushes an error code.
 pub type HandlerFuncWithErrCode =
-    extern "x86-interrupt" fn(&mut InterruptStackFrame, error_code: u64);
+    extern "x86-interrupt" fn(&mut InterruptStackFrame, error_code: u32);
 /// A handler function that must not return, e.g. for a machine check exception.
 pub type DivergingHandlerFunc = extern "x86-interrupt" fn(&mut InterruptStackFrame) -> !;
 /// A handler function with an error code that must not return, e.g. for a double fault exception.
 pub type DivergingHandlerFuncWithErrCode =
-    extern "x86-interrupt" fn(&mut InterruptStackFrame, error_code: u64) -> !;
+    extern "x86-interrupt" fn(&mut InterruptStackFrame, error_code: u32) -> !;
 
 /// Represents the interrupt stack frame pushed by the CPU on interrupt or exception entry.
 #[derive(Clone, Debug)]