feat: 实现 DBCN

Signed-off-by: YdrMaster <[email protected]>

Author: YdrMaster

Cargo.lock

@@ -17,7 +17,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-rustsbi = { git = "https://github.com/rustsbi/rustsbi", rev = "7118c8c" }
+rustsbi = { version = "0.3.2-rc.2", features = ["sbi_2_0"] }
 sbi-spec = { version = "0.0.5", features = ["legacy"] }
 riscv = "0.10.1"
 spin = "0.9"

rustsbi-qemu/Cargo.toml

@@ -0,0 +1,58 @@
+use crate::UART;
+use core::ops::Range;
+use rustsbi::{spec::binary::SbiRet, Console, Physical};
+use spin::Once;
+
+pub(crate) struct DBCN(Range<usize>);
+
+static INSTANCE: Once<DBCN> = Once::new();
+
+pub(crate) fn init(memory: Range<usize>) {
+    INSTANCE.call_once(|| DBCN(memory));
+}
+
+pub(crate) fn get() -> &'static DBCN {
+    INSTANCE.wait()
+}
+
+impl Console for DBCN {
+    fn write(&self, bytes: Physical<&[u8]>) -> SbiRet {
+        let start = bytes.phys_addr_lo();
+        let end = start + bytes.num_bytes();
+        if self.0.contains(&start) && self.0.contains(&(end - 1)) {
+            let mut uart = UART.lock();
+            let uart = unsafe { uart.assume_init_mut() };
+            for ptr in start..end {
+                let c = ptr as *const u8;
+                unsafe { uart.send(c.read_volatile()) };
+            }
+            SbiRet::success(bytes.num_bytes())
+        } else {
+            SbiRet::invalid_param()
+        }
+    }
+
+    fn read(&self, bytes: Physical<&mut [u8]>) -> SbiRet {
+        let start = bytes.phys_addr_lo();
+        let end = start + bytes.num_bytes();
+        if self.0.contains(&start) && self.0.contains(&(end - 1)) {
+            let mut uart = UART.lock();
+            let uart = unsafe { uart.assume_init_mut() };
+            for ptr in start..end {
+                let c = ptr as *mut u8;
+                unsafe { c.write_volatile(uart.receive()) };
+            }
+            SbiRet::success(bytes.num_bytes())
+        } else {
+            SbiRet::invalid_param()
+        }
+    }
+
+    #[inline]
+    fn write_byte(&self, byte: u8) -> SbiRet {
+        let mut uart = UART.lock();
+        let uart = unsafe { uart.assume_init_mut() };
+        uart.send(byte);
+        SbiRet::success(0)
+    }
+}

rustsbi-qemu/src/dbcn.rs

@@ -4,6 +4,7 @@
 #![deny(warnings)]
 
 mod clint;
+mod dbcn;
 mod device_tree;
 mod hart_csr_utils;
 mod qemu_test;
@@ -87,6 +88,7 @@ extern "C" fn rust_main(hartid: usize, opaque: usize) {
         rcore_console::set_log_level(option_env!("LOG"));
         clint::init(board_info.clint.start);
         qemu_test::init(board_info.test.start);
+        dbcn::init(SUPERVISOR_ENTRY..board_info.mem.end);
         // 打印启动信息
         print!(
             "\
@@ -118,6 +120,7 @@ extern "C" fn rust_main(hartid: usize, opaque: usize) {
                     .with_timer(&clint::Clint)
                     .with_hsm(Hsm)
                     .with_reset(qemu_test::get())
+                    .with_console(dbcn::get())
                     .build(),
             );
         }
@@ -340,6 +343,7 @@ type FixedRustSBI<'a> = RustSBI<
     Hsm,
     &'a qemu_test::QemuTest,
     Infallible,
+    &'a dbcn::DBCN,
 >;
 
 struct Hsm;