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
|
#![allow(dead_code)]
#![allow(unused_imports)]
#![no_std]
#![no_main]
#![feature(const_option)]
#![feature(try_with_capacity)]
mod arch;
mod defs;
mod ds;
mod io;
mod machine;
mod mm;
mod proc;
extern crate alloc;
use crate::proc::sched::*;
use alloc::vec::Vec;
use arch::x86_64::interrupt;
use arch::x86_64::interrupt::pic_8259;
use arch::x86_64::interrupt::pic_8259::PicDeviceInt;
use core::panic::PanicInfo;
use defs::*;
use machine::cgascr::CGAScreen;
use machine::key::Modifiers;
use machine::multiboot;
use machine::serial::Serial;
use proc::task::Task;
#[panic_handler]
fn panic(info: &PanicInfo) -> ! {
println!("{}", info);
loop {}
}
#[no_mangle]
pub extern "C" fn _entry() -> ! {
// init code
io::set_attr(0x1f);
io::clear_screen();
assert!(multiboot::check(), "bad multiboot info from grub!");
// check mbi now. This will be later used to initilize the allocator
let _mbi = multiboot::get_mb_info().expect("bad multiboot info flags");
// initialize the idt and re-program the pic. Must do this before enabling irq
// also must initialize the idt before mm, because the later may trigger page faults, which is
// fatal and we want to catch them during system initilization.
interrupt::init();
mm::init();
println!(
"[init] kernel mapped @ {:#X} - {:#X}",
unsafe { vmap_kernel_start() },
unsafe { vmap_kernel_end() },
);
println!(
"[init] BSS mapped @ {:#X} - {:#X}",
bss_start(),
bss_end()
);
let interval = interrupt::pit::PIT::set_interval(500000);
println!("[init] timer interrupt set to {} ns", interval);
// busy loop query keyboard
interrupt::interrupt_enable();
pic_8259::allow(PicDeviceInt::KEYBOARD);
let mut test_vec = Vec::<&str>::new();
_test_proc_switch_to();
// default test, should not reach
test_vec.push("hello ");
test_vec.push("world");
for s in test_vec.iter() {
println!("{s}");
}
Serial::print("hello from serial");
loop {
io::KBCTL_GLOBAL.lock().fetch_key();
if let Some(k) = io::KBCTL_GLOBAL.lock().consume_key() {
println! {"key: {:?}", k}
}
}
// test heap
}
pub unsafe fn _test_pf() {
// try a page fault
use core::arch::asm;
use core::slice;
let name_buf = slice::from_raw_parts_mut(0xffffffffffff0000 as *mut u64, 10);
asm!("mov [rdi], rax", in("rdi") name_buf.as_mut_ptr());
}
pub fn _test_proc_switch_to() {
SCHEDULER.lock().insert_task(Task::create_dummy(1));
SCHEDULER.lock().insert_task(Task::create_dummy(2));
SCHEDULER.lock().insert_task(Task::create_dummy(3));
SCHEDULER.lock().insert_task(Task::create_dummy(4));
SCHEDULER.lock().insert_task(Task::create_dummy(5));
pic_8259::allow(PicDeviceInt::TIMER);
Scheduler::kickoff();
}
|
