somehal 0.4.12

Boot kernel code with mmu.
Documentation 
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156

use crate::lazy_static::LazyStatic;
use core::fmt::Write;
use spin::Mutex;

type RxFun = fn() -> Result<u8, RError>;

#[unsafe(link_section = ".data")]
static RX_FUN: LazyStatic<RxFun> = LazyStatic::with_default(empty_rx);

#[unsafe(link_section = ".data")]
static RX_MUTEX: Mutex<()> = Mutex::new(());

type TxFun = fn(u8) -> Result<(), TError>;

#[unsafe(link_section = ".data")]
static TX_FUN: LazyStatic<TxFun> = LazyStatic::with_default(empty_tx);
#[unsafe(link_section = ".data")]
static TX_MUTEX: Mutex<()> = Mutex::new(());

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RError {
    NoData,
    Timeout,
    ParityError,
    FrameError,
    Overrun,
    BufferFull,
    Other,
}

#[derive(Debug, Clone, Copy)]
pub enum TError {
    ReTry,
    Other,
}

fn empty_rx() -> Result<u8, RError> {
    Err(RError::NoData)
}

fn empty_tx(_: u8) -> Result<(), TError> {
    Ok(())
}

pub(crate) fn set_rx_fun(rx: RxFun) {
    RX_FUN.init(rx);
}

pub(crate) fn set_tx_fun(tx: TxFun) {
    TX_FUN.init(tx);
}

#[inline]
fn _read_byte() -> Result<u8, RError> {
    RX_FUN()
}

/// 读取一个字节
pub fn read_byte() -> Result<u8, RError> {
    let _lock = RX_MUTEX.lock();
    _read_byte()
}

/// 带超时的阻塞读取（max_spins 为自旋次数）
pub fn read_byte_timeout(max_spins: usize) -> Result<u8, RError> {
    let _lock = RX_MUTEX.lock();
    for _ in 0..max_spins {
        match _read_byte() {
            Ok(b) => return Ok(b),
            Err(RError::NoData) => {
                core::hint::spin_loop();
                continue;
            }
            Err(e) => return Err(e),
        }
    }
    Err(RError::Timeout)
}

/// 非阻塞读取多个字节，遇到 NoData 就停止
pub fn read_bytes(buffer: &mut [u8]) -> Result<usize, RError> {
    let _lock = RX_MUTEX.lock();
    let mut count = 0;

    for slot in buffer.iter_mut() {
        match _read_byte() {
            Ok(b) => {
                *slot = b;
                count += 1;
            }
            Err(RError::NoData) => break,
            Err(e) => return Err(e),
        }
    }

    Ok(count)
}

#[inline]
fn _write_byte(b: u8) -> Result<(), TError> {
    TX_FUN(b)
}

struct TX;

impl core::fmt::Write for TX {
    fn write_str(&mut self, s: &str) -> core::fmt::Result {
        for b in s.bytes() {
            loop {
                match _write_byte(b) {
                    Ok(_) => break,
                    Err(TError::ReTry) => continue,
                    Err(TError::Other) => break,
                }
            }
        }
        Ok(())
    }
}

pub fn write_str(s: &str) {
    let _lock = TX_MUTEX.lock();
    let mut tx = TX;
    let _ = tx.write_str(s);
}

pub fn write_fmt(args: core::fmt::Arguments) {
    let _lock = TX_MUTEX.lock();
    let mut tx = TX;
    let _ = tx.write_fmt(args);
}

pub fn write_bytes(bytes: &[u8]) -> Result<(), TError> {
    let _lock = TX_MUTEX.lock();
    for &b in bytes {
        loop {
            match _write_byte(b) {
                Ok(_) => break,
                Err(TError::ReTry) => continue,
                Err(TError::Other) => break,
            }
        }
    }
    Ok(())
}

#[macro_export]
macro_rules! println {
    () => {
        $crate::early_debug::write_fmt(format_args!("\r\n"));
    };
    ($($arg:tt)*) => {
        $crate::early_debug::write_fmt(format_args!($($arg)*));
        $crate::early_debug::write_fmt(format_args!("\r\n"));
    };
}