summaryrefslogtreecommitdiff
path: root/kernel/src/interrupts
diff options
context:
space:
mode:
Diffstat (limited to 'kernel/src/interrupts')
-rw-r--r--kernel/src/interrupts/apic.rs67
-rw-r--r--kernel/src/interrupts/exception_handlers.rs17
-rw-r--r--kernel/src/interrupts/gdt.rs6
-rw-r--r--kernel/src/interrupts/idt.rs2
-rw-r--r--kernel/src/interrupts/interrupt_handlers.rs8
5 files changed, 66 insertions, 34 deletions
diff --git a/kernel/src/interrupts/apic.rs b/kernel/src/interrupts/apic.rs
index 0e89943..836af20 100644
--- a/kernel/src/interrupts/apic.rs
+++ b/kernel/src/interrupts/apic.rs
@@ -1,8 +1,11 @@
+use core::sync::atomic::{AtomicU64, Ordering};
use x86_64::registers::model_specific::Msr;
#[cfg(target_arch = "x86_64")]
const APIC_FLAG: u64 = 0x0800;
const APIC_BASE_MSR: u32 = 0x001b;
+const CPU_COUNT: usize = 255;
+const APIC_BASE_ADDR: u64 = 0xfee0_0000;
// APIC constant register values
const APIC_DISABLE: u32 = 0x0001_0000;
@@ -20,8 +23,13 @@ pub fn is_apic() -> bool {
info & (1 << 9) != 0
}
-pub fn set_apic_base(v: u64) {
- unsafe { Msr::new(APIC_BASE_MSR).write(v) }
+pub fn cpu_id() -> u32 {
+ let mut info: u32;
+ unsafe { asm!("cpuid", inout("eax") 1 => _, out("ebx") info) }
+ if !is_x2apic() {
+ info >>= 24;
+ }
+ info
}
#[repr(C)]
@@ -75,27 +83,48 @@ pub enum TimerDivideConfig {
Div128 = 10,
NoDivide = 11,
}
+const NONE_APIC: Option<Apic> = None;
-static mut LOCAL_APIC: Option<Apic> = None;
+static mut APIC_INIT: [Option<Apic>; CPU_COUNT] = [NONE_APIC; CPU_COUNT];
-pub unsafe fn set_local_apic(apic: Apic) -> &'static mut Apic {
- LOCAL_APIC = Some(apic);
- LOCAL_APIC.as_mut().unwrap()
+pub fn move_local_apic(v: u64) -> bool {
+ match get_local_apic() {
+ None => return false,
+ Some(apic) => apic.regs.store(v, Ordering::Release),
+ }
+ unsafe { Msr::new(APIC_BASE_MSR).write(v) };
+ true
}
-pub unsafe fn get_local_apic() -> Option<&'static mut Apic> {
- LOCAL_APIC.as_mut()
+/// Tries to initialize the local apic at it's default location.
+/// returns false if no apic is available
+///
+/// # Safety
+/// Calling this function in a running system might cause the loss of interrupts or other weird
+/// behaviour
+pub unsafe fn try_initialize_local_apic() -> &'static mut Option<Apic> {
+ if is_apic() {
+ let cpu_id = cpu_id() as usize;
+ if APIC_INIT[cpu_id].is_none() {
+ APIC_INIT[cpu_id] = Some(Apic::from_base_apic(APIC_BASE_ADDR));
+ Apic::disable_pic();
+ }
+ APIC_INIT[cpu_id].as_mut().map(|x| x.initialize());
+ }
+ get_local_apic()
+}
+
+pub fn get_local_apic() -> &'static mut Option<Apic> {
+ let cpu_id = cpu_id() as usize;
+ unsafe { &mut APIC_INIT[cpu_id] }
}
pub struct Apic {
- regs: *mut [u32; 1024],
+ regs: AtomicU64,
}
impl Apic {
- pub fn new() -> Option<Self> {
- if !is_apic() {
- return None;
- }
+ fn initialize(&mut self) -> Self {
Self::disable_pic();
let base_apic = unsafe { Msr::new(APIC_BASE_MSR).read() };
let mut apic = Self::from_base_apic(base_apic);
@@ -118,28 +147,28 @@ impl Apic {
apic.set(ApicRegister::SpuriousInterruptVec, 0x1ff);
- Some(apic)
+ apic
}
- pub fn disable_pic() {
+ fn disable_pic() {
unsafe {
x86_64::instructions::port::PortWriteOnly::new(0xa1).write(0xffu8);
x86_64::instructions::port::PortWriteOnly::new(0x21).write(0xffu8);
}
}
- fn from_base_apic(base_apic: u64) -> Self {
+ const fn from_base_apic(base_apic: u64) -> Self {
Self {
- regs: ((base_apic) & 0xfffffffff << 12) as *mut [u32; 1024],
+ regs: AtomicU64::new((base_apic) & 0xfffffffff << 12),
}
}
pub fn ptr(&self) -> *const u32 {
- self.regs as *const u32
+ self.regs.load(Ordering::Acquire) as *const u32
}
fn ptr_mut(&mut self) -> *mut u32 {
- self.regs as *mut u32
+ self.regs.load(Ordering::SeqCst) as *mut u32
}
pub fn get(&self, register: ApicRegister) -> u32 {
diff --git a/kernel/src/interrupts/exception_handlers.rs b/kernel/src/interrupts/exception_handlers.rs
index ce8f6c0..db081a7 100644
--- a/kernel/src/interrupts/exception_handlers.rs
+++ b/kernel/src/interrupts/exception_handlers.rs
@@ -15,6 +15,11 @@ macro_rules! exception_default {
static mut EXPECTED_FAULT: InterruptType = InterruptType::None;
+/// Set an exception that causes the kernel to exit with status success
+/// This is used for testing
+///
+/// # Safety
+/// This function may only be called before executing any kernel code
pub unsafe fn expect_fault(int: InterruptType) {
if EXPECTED_FAULT == InterruptType::None {
EXPECTED_FAULT = int;
@@ -25,7 +30,7 @@ fn get_expected_fault() -> InterruptType {
unsafe { EXPECTED_FAULT }
}
-pub extern "x86-interrupt" fn breakpoint_handler(stack_frame: &mut InterruptStackFrame) {
+pub extern "x86-interrupt" fn breakpoint_handler(stack_frame: InterruptStackFrame) {
match get_expected_fault() {
InterruptType::Breakpoint => qemu::exit_qemu(qemu::QemuExitCode::Success),
_ => panic!(
@@ -35,7 +40,7 @@ pub extern "x86-interrupt" fn breakpoint_handler(stack_frame: &mut InterruptStac
}
}
-pub extern "x86-interrupt" fn div_zero_handler(stack_frame: &mut InterruptStackFrame) {
+pub extern "x86-interrupt" fn div_zero_handler(stack_frame: InterruptStackFrame) {
match get_expected_fault() {
InterruptType::DivZero => qemu::exit_qemu(qemu::QemuExitCode::Success),
_ => panic!(
@@ -46,7 +51,7 @@ pub extern "x86-interrupt" fn div_zero_handler(stack_frame: &mut InterruptStackF
}
pub extern "x86-interrupt" fn general_protection_fault_handler(
- stack_frame: &mut InterruptStackFrame,
+ stack_frame: InterruptStackFrame,
error_code: u64,
) {
match get_expected_fault() {
@@ -60,7 +65,7 @@ pub extern "x86-interrupt" fn general_protection_fault_handler(
}
pub extern "x86-interrupt" fn page_fault_handler(
- stack_frame: &mut InterruptStackFrame,
+ stack_frame: InterruptStackFrame,
error_code: PageFaultErrorCode,
) {
use x86_64::registers::control::Cr2;
@@ -78,7 +83,7 @@ pub extern "x86-interrupt" fn page_fault_handler(
}
pub extern "x86-interrupt" fn segment_not_present_handler(
- stack_frame: &mut InterruptStackFrame,
+ stack_frame: InterruptStackFrame,
error_code: u64,
) {
match get_expected_fault() {
@@ -92,7 +97,7 @@ pub extern "x86-interrupt" fn segment_not_present_handler(
}
pub extern "x86-interrupt" fn double_fault_handler(
- stack_frame: &mut InterruptStackFrame,
+ stack_frame: InterruptStackFrame,
_error_code: u64, // code is always zero
) -> ! {
match get_expected_fault() {
diff --git a/kernel/src/interrupts/gdt.rs b/kernel/src/interrupts/gdt.rs
index d75ff68..c20eabe 100644
--- a/kernel/src/interrupts/gdt.rs
+++ b/kernel/src/interrupts/gdt.rs
@@ -14,16 +14,14 @@ lazy_static! {
static mut STACK: [u8; STACK_SIZE] = [0; STACK_SIZE];
let stack_start = VirtAddr::from_ptr(unsafe { &STACK });
- let stack_end = stack_start + STACK_SIZE;
- stack_end
+ stack_start + STACK_SIZE
};
tss.interrupt_stack_table[PAGE_FAULT_IST_INDEX as usize] = {
const STACK_SIZE: usize = 4096 * 8;
static mut STACK: [u8; STACK_SIZE] = [0; STACK_SIZE];
let stack_start = VirtAddr::from_ptr(unsafe { &STACK });
- let stack_end = stack_start + STACK_SIZE;
- stack_end
+ stack_start + STACK_SIZE
};
tss
};
diff --git a/kernel/src/interrupts/idt.rs b/kernel/src/interrupts/idt.rs
index 4e5dd77..6bc8149 100644
--- a/kernel/src/interrupts/idt.rs
+++ b/kernel/src/interrupts/idt.rs
@@ -34,7 +34,7 @@ lazy_static! {
}
pub extern "x86-interrupt" fn dummy_handler(
- _stack_frame: &mut x86_64::structures::idt::InterruptStackFrame,
+ _stack_frame: x86_64::structures::idt::InterruptStackFrame,
) {
panic!("unhandled interrupt recieved");
}
diff --git a/kernel/src/interrupts/interrupt_handlers.rs b/kernel/src/interrupts/interrupt_handlers.rs
index 831211b..8b3a7eb 100644
--- a/kernel/src/interrupts/interrupt_handlers.rs
+++ b/kernel/src/interrupts/interrupt_handlers.rs
@@ -2,18 +2,18 @@ use core::fmt::Write;
use x86_64::structures::idt::InterruptStackFrame;
use x86_64::structures::port;
-pub extern "x86-interrupt" fn timer_handler(_stack_frame: &mut InterruptStackFrame) {
+pub extern "x86-interrupt" fn timer_handler(_stack_frame: InterruptStackFrame) {
crate::io::vga_text::OStream::new().print(b"Timer");
- if let Some(apic) = unsafe { super::apic::get_local_apic() } {
+ if let Some(apic) = super::apic::get_local_apic() {
apic.end_of_interrupt()
}
}
-pub extern "x86-interrupt" fn keyboard_handler(_stack_frame: &mut InterruptStackFrame) {
+pub extern "x86-interrupt" fn keyboard_handler(_stack_frame: InterruptStackFrame) {
let code: u8 = unsafe { port::PortRead::read_from_port(0x60) };
let _ = write!(crate::io::vga_text::OStream::new(), "{}", code);
- if let Some(apic) = unsafe { super::apic::get_local_apic() } {
+ if let Some(apic) = super::apic::get_local_apic() {
apic.end_of_interrupt()
}
}
generated by cgit v1.2.3-54-g00ecf (git 2.45.0) at 2024-05-15 22:26:25 +0000