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

// Copyright 2025 Google LLC
// SPDX-License-Identifier: Apache-2.0
use safe_mmio::{UniqueMmioPointer, field, field_shared};

/// PL011 UART 的驱动程序。
#[derive(Debug)]
pub struct Uart<'a> {
    registers: UniqueMmioPointer<'a, Registers>,
}

impl<'a> Uart<'a> {
    /// 为 PL011 设备构建 UART 驱动程序的新实例
    /// 给定的一组寄存器。
    pub fn new(registers: UniqueMmioPointer<'a, Registers>) -> Self {
        Self { registers }
    }

    /// 将单个字节写入 UART。
    pub fn write_byte(&mut self, byte: u8) {
        // 等待，直到 TX 缓冲区中有空间。
        while self.read_flag_register().contains(Flags::TXFF) {}

        // 写入 TX 缓冲区。
        field!(self.registers, dr).write(byte.into());

        // 等待 UART 不再繁忙。
        while self.read_flag_register().contains(Flags::BUSY) {}
    }

    /// 读取并返回一个待处理字节，或者`None`如果什么都没有
    /// 已收到。
    pub fn read_byte(&mut self) -> Option<u8> {
        if self.read_flag_register().contains(Flags::RXFE) {
            None
        } else {
            let data = field!(self.registers, dr).read();
            // TODO：检查位 8-11 中的错误情况。
            Some(data as u8)
        }
    }

    fn read_flag_register(&self) -> Flags {
        field_shared!(self.registers, fr).read()
    }
}