1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#![no_std]
#![cfg_attr(test, no_main)]
#![feature(custom_test_frameworks)]
#![feature(abi_x86_interrupt)]
#![test_runner(crate::testexec)]
#![reexport_test_harness_main = "testmain"]
use core::panic::PanicInfo;
pub mod gdt;
pub mod intr;
pub mod ser;
pub mod vgabuff;
pub fn init()
{
gdt::init();
intr::idtinit();
unsafe
{
intr::PICS.lock().initialize()
};
x86_64::instructions::interrupts::enable();
}
pub fn hltloop() -> !
{
loop
{
x86_64::instructions::hlt();
}
}
pub trait CanTest
{
fn exec(&self) -> ();
}
impl<T> CanTest for T
where
T: Fn(),
{
fn exec(&self)
{
serprint!("{}...\t", core::any::type_name::<T>());
self();
serprintln!("[SUCCESS]");
}
}
pub fn testexec(tests: &[&dyn CanTest])
{
serprintln!("RUNNING {} TESTS:", tests.len());
for test in tests
{
test.exec();
}
exitqemu(QEMUExitCode::Success);
}
pub fn test_panic_handler(info: &PanicInfo) -> !
{
serprintln!("[FAILURE]\n");
serprintln!("[ERR]: {}\n", info);
exitqemu(QEMUExitCode::Failure);
hltloop();
}
#[cfg(test)]
#[no_mangle]
pub extern "C" fn _start() -> !
{
init();
testmain();
hltloop();
}
#[cfg(test)]
#[panic_handler]
fn panic(info: &PanicInfo) -> !
{
test_panic_handler(info)
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[repr(u32)]
pub enum QEMUExitCode
{
Success = 0x10,
Failure = 0x11,
}
pub fn exitqemu(exitcode: QEMUExitCode)
{
use x86_64::instructions::port::Port;
unsafe
{
let mut port = Port::new(0xf4);
port.write(exitcode as u32);
}
}