use bit_field::BitField;
use bitflags::bitflags;
bitflags! {
struct RedirectionEntry: u32 {
const DISABLED = 0x00010000;
const LEVEL = 0x00008000;
const ACTIVELOW = 0x00002000;
const LOGICAL = 0x00000800;
const NONE = 0x00000000;
}
}
pub struct IoApic {
reg: *mut u32,
data: *mut u32,
}
impl IoApic {
pub unsafe fn new(addr: usize) -> Self {
IoApic {
reg: addr as *mut u32,
data: (addr + 0x10) as *mut u32,
}
}
pub fn disable_all(&mut self) {
for i in 0..self.supported_interrupts() {
self.write_irq(i, RedirectionEntry::DISABLED, 0);
}
}
unsafe fn read(&mut self, reg: u8) -> u32 {
self.reg.write_volatile(reg as u32);
self.data.read_volatile()
}
unsafe fn write(&mut self, reg: u8, data: u32) {
self.reg.write_volatile(reg as u32);
self.data.write_volatile(data);
}
fn write_irq(&mut self, irq: u8, flags: RedirectionEntry, dest: u8) {
unsafe {
self.write(REG_TABLE + 2 * irq, (T_IRQ0 + irq) as u32 | flags.bits());
self.write(REG_TABLE + 2 * irq + 1, (dest as u32) << 24);
}
}
pub fn enable(&mut self, irq: u8, cpunum: u8) {
self.write_irq(irq, RedirectionEntry::NONE, cpunum);
}
pub fn id(&mut self) -> u8 {
unsafe { self.read(REG_ID).get_bits(24..28) as u8 }
}
pub fn version(&mut self) -> u8 {
unsafe { self.read(REG_VER).get_bits(0..8) as u8 }
}
pub fn supported_interrupts(&mut self) -> u8 {
unsafe { (self.read(REG_VER).get_bits(16..24) + 1) as u8 }
}
}
const REG_ID: u8 = 0x00;
const REG_VER: u8 = 0x01;
const REG_TABLE: u8 = 0x10;
const T_IRQ0: u8 = 32;