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
use crate::mmio;
use bitflags::bitflags;
use core::ptr::{read_volatile, write_volatile};
const UART_REG: u8 = 0x10;
bitflags! {
struct UartStatus: u8 {
const TX_EN = 1;
const TX_BUSY = 2;
const RX_READY = 4;
const RX_CLEAR = 8;
}
}
unsafe fn wait_for_tx() -> UartStatus {
loop {
let reg = mmio::read(UART_REG);
let status = UartStatus::from_bits_unchecked(reg as u8);
if !status.contains(UartStatus::TX_BUSY) {
break status;
}
}
}
pub fn write(byte: u8) {
unsafe {
let mut status = wait_for_tx() | UartStatus::TX_EN;
let reg = (status.bits() as u32) | ((byte as u32) << 8);
mmio::write(UART_REG, reg);
wait_for_tx();
}
}
pub fn try_read() -> Option<u8> {
unsafe {
let reg = mmio::read(UART_REG);
let status = UartStatus::from_bits_unchecked(reg as u8);
if status.contains(UartStatus::RX_READY) {
mmio::write(UART_REG, (status | UartStatus::RX_CLEAR).bits() as u32);
Some((reg >> 16) as u8)
} else {
None
}
}
}
pub fn read() -> u8 {
loop {
if let Some(x) = try_read() {
break x;
}
}
}