Added IDT and CPU exeption support

src/interrupts.rs

@@ -0,0 +1,25 @@
+use crate::println;
+use x86_64::structures::idt::{InterruptDescriptorTable, InterruptStackFrame};
+
+use lazy_static::lazy_static;
+
+lazy_static! {
+    static ref IDT: InterruptDescriptorTable = {
+        let mut idt = InterruptDescriptorTable::new();
+        idt.breakpoint.set_handler_fn(breakpoint_handler);
+        idt
+    };
+}
+
+pub fn init_idt() {
+    IDT.load();
+}
+
+extern "x86-interrupt" fn breakpoint_handler(stack_frame: InterruptStackFrame) {
+    println!("EXCEPTION: BREAKPOINT\n{:#?}", stack_frame);
+}
+
+#[test_case]
+fn test_breakpoint_exception() {
+    x86_64::instructions::interrupts::int3();
+}

src/lib.rs

@@ -3,12 +3,18 @@
 #![feature(custom_test_frameworks)]
 #![test_runner(crate::test_runner)]
 #![reexport_test_harness_main = "test_main"]
+#![feature(abi_x86_interrupt)]
 
 use core::panic::PanicInfo;
 
+pub mod interrupts;
 pub mod serial;
 pub mod vga_buffer;
 
+pub fn init() {
+    interrupts::init_idt();
+}
+
 pub trait Testable {
     fn run(&self) -> ();
 }
@@ -59,6 +65,7 @@ pub fn test_panic_handler(info: &PanicInfo) -> ! {
 #[cfg(test)]
 #[no_mangle]
 pub extern "C" fn _start() -> ! {
+    init();
     test_main();
     loop {}
 }

src/main.rs

@@ -9,7 +9,16 @@ use infinity_os::println;
 
 #[no_mangle]
 pub extern "C" fn _start() -> ! {
-    println!("Hello World{}", "!");
+    println!("  _____        __ _       _ _          ____   _____ ");
+    println!(" |_   _|      / _(_)     (_) |        / __ \\ / ____|");
+    println!("   | |  _ __ | |_ _ _ __  _| |_ _   _| |  | | (___  ");
+    println!("   | | | '_ \\|  _| | '_ \\| | __| | | | |  | |\\___ \\ ");
+    println!("  _| |_| | | | | | | | | | | |_| |_| | |__| |____) |");
+    println!(" |_____|_| |_|_| |_|_| |_|_|\\__|\\__, |\\____/|_____/ ");
+    println!("                                 __/ |              ");
+    println!("                                |___/               ");
+
+    infinity_os::init();
 
     #[cfg(test)]
     test_main();

src/serial.rs

@@ -34,4 +34,4 @@ macro_rules! serial_println {
     ($fmt:expr) => ($crate::serial_print!(concat!($fmt, "\n")));
     ($fmt:expr, $($arg:tt)*) => ($crate::serial_print!(
         concat!($fmt, "\n"), $($arg)*));
-}
+}