use crate::devices::{clint::Clint, plic::Plic, uart::Uart, virtio_blk::Virtio};
use crate::dram::{Dram, DRAM_SIZE};
use crate::exception::Exception;
use crate::rom::Rom;
pub const MROM_BASE: u64 = 0x1000;
const MROM_END: u64 = MROM_BASE + 0xf000;
pub const CLINT_BASE: u64 = 0x200_0000;
const CLINT_END: u64 = CLINT_BASE + 0x10000;
pub const PLIC_BASE: u64 = 0xc00_0000;
const PLIC_END: u64 = PLIC_BASE + 0x208000;
pub const UART_BASE: u64 = 0x1000_0000;
pub const UART_SIZE: u64 = 0x100;
const UART_END: u64 = UART_BASE + 0x100;
pub const VIRTIO_BASE: u64 = 0x1000_1000;
const VIRTIO_END: u64 = VIRTIO_BASE + 0x1000;
pub const DRAM_BASE: u64 = 0x8000_0000;
const DRAM_END: u64 = DRAM_BASE + DRAM_SIZE;
pub struct Bus {
pub clint: Clint,
plic: Plic,
pub uart: Uart,
pub virtio: Virtio,
dram: Dram,
rom: Rom,
}
impl Bus {
pub fn new() -> Bus {
Self {
clint: Clint::new(),
plic: Plic::new(),
uart: Uart::new(),
virtio: Virtio::new(),
dram: Dram::new(),
rom: Rom::new(),
}
}
pub fn initialize_dram(&mut self, data: Vec<u8>) {
self.dram.initialize(data);
}
pub fn initialize_disk(&mut self, data: Vec<u8>) {
self.virtio.initialize(data);
}
pub fn read(&mut self, addr: u64, size: u8) -> Result<u64, Exception> {
match addr {
MROM_BASE..=MROM_END => self.rom.read(addr, size),
CLINT_BASE..=CLINT_END => self.clint.read(addr, size),
PLIC_BASE..=PLIC_END => self.plic.read(addr, size),
UART_BASE..=UART_END => self.uart.read(addr, size),
VIRTIO_BASE..=VIRTIO_END => self.virtio.read(addr, size),
DRAM_BASE..=DRAM_END => self.dram.read(addr, size),
_ => Err(Exception::LoadAccessFault),
}
}
pub fn write(&mut self, addr: u64, value: u64, size: u8) -> Result<(), Exception> {
match addr {
CLINT_BASE..=CLINT_END => self.clint.write(addr, value, size),
PLIC_BASE..=PLIC_END => self.plic.write(addr, value, size),
UART_BASE..=UART_END => self.uart.write(addr, value as u8, size),
VIRTIO_BASE..=VIRTIO_END => self.virtio.write(addr, value as u32, size),
DRAM_BASE..=DRAM_END => self.dram.write(addr, value, size),
_ => Err(Exception::StoreAMOAccessFault),
}
}
}