software_interrupt: Add additional testing (rust-osdev#260)

Followup to rust-osdev#259. Code previously merged as part of rust-osdev#227. Signed-off-by: Joe Richey <[email protected]>
josephlr · Jul 18, 2021 · 9b49646 · 9b49646
1 parent c42f3d2
commit 9b49646
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 6 deletions.
diff --git a/testing/Cargo.toml b/testing/Cargo.toml
@@ -5,7 +5,7 @@ authors = ["Philipp Oppermann <[email protected]>"]
 edition = "2018"
 
 [[test]]
-name = "breakpoint_exception"
+name = "interrupt_handling"
 harness = false
 
 [[test]]

diff --git a/testing/tests/breakpoint_exception.rs → testing/tests/interrupt_handling.rs b/testing/tests/breakpoint_exception.rs → testing/tests/interrupt_handling.rs
@@ -7,7 +7,10 @@ use core::sync::atomic::{AtomicUsize, Ordering};
 use lazy_static::lazy_static;
 use testing::{exit_qemu, serial_print, serial_println, QemuExitCode};
 
+use x86_64::instructions::interrupts;
+
 static BREAKPOINT_HANDLER_CALLED: AtomicUsize = AtomicUsize::new(0);
+static INTERRUPT_HANDLER_CALLED: AtomicUsize = AtomicUsize::new(0);
 
 #[no_mangle]
 pub extern "C" fn _start() -> ! {
@@ -16,13 +19,10 @@ pub extern "C" fn _start() -> ! {
     init_test_idt();
 
     // invoke a breakpoint exception
-    x86_64::instructions::interrupts::int3();
+    interrupts::int3();
 
     match BREAKPOINT_HANDLER_CALLED.load(Ordering::SeqCst) {
-        1 => {
-            serial_println!("[ok]");
-            exit_qemu(QemuExitCode::Success);
-        }
+        1 => {}
         0 => {
             serial_println!("[failed]");
             serial_println!("    Breakpoint handler was not called.");
@@ -35,6 +35,29 @@ pub extern "C" fn _start() -> ! {
         }
     }
 
+    serial_print!("interrupt 42... ");
+    unsafe { interrupts::software_interrupt::<42>() };
+    serial_print!("interrupt 77... ");
+    unsafe { interrupts::software_interrupt::<77>() };
+    serial_print!("interrupt 42... ");
+    unsafe { interrupts::software_interrupt::<42>() };
+
+    match INTERRUPT_HANDLER_CALLED.load(Ordering::SeqCst) {
+        3 => {}
+        0 => {
+            serial_println!("[failed]");
+            serial_println!("    Interrupt handler was not called.");
+            exit_qemu(QemuExitCode::Failed);
+        }
+        other => {
+            serial_println!("[failed]");
+            serial_println!("    Interrupt handler was called {} times", other);
+            exit_qemu(QemuExitCode::Failed);
+        }
+    }
+
+    serial_println!("[ok]");
+    exit_qemu(QemuExitCode::Success);
     loop {}
 }
 
@@ -49,6 +72,8 @@ lazy_static! {
     static ref TEST_IDT: InterruptDescriptorTable = {
         let mut idt = InterruptDescriptorTable::new();
         idt.breakpoint.set_handler_fn(breakpoint_handler);
+        idt[42].set_handler_fn(interrupt_handler);
+        idt[77].set_handler_fn(interrupt_handler);
         idt
     };
 }
@@ -60,3 +85,7 @@ pub fn init_test_idt() {
 extern "x86-interrupt" fn breakpoint_handler(_stack_frame: InterruptStackFrame) {
     BREAKPOINT_HANDLER_CALLED.fetch_add(1, Ordering::SeqCst);
 }
+
+extern "x86-interrupt" fn interrupt_handler(_stack_frame: InterruptStackFrame) {
+    INTERRUPT_HANDLER_CALLED.fetch_add(1, Ordering::SeqCst);
+}