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

// Copyright lowRISC contributors (OpenTitan project).
// Licensed under the Apache License, Version 2.0, see LICENSE for details.
// SPDX-License-Identifier: Apache-2.0

use std::cell::RefCell;
use std::cmp;
use std::io::{Read, Write};
use std::thread;
use std::time::{Duration, Instant};

use anyhow::{Context, Result};

use crate::io::uart::Uart;
use crate::transport::ultradebug::Ultradebug;

pub struct Inner {
    device: ftdi::Device,
    baudrate: u32,
}

/// Represents the Ultradebug UART.
pub struct UltradebugUart {
    inner: RefCell<Inner>,
}

impl UltradebugUart {
    pub fn open(ultradebug: &Ultradebug) -> Result<Self> {
        let mut device = ultradebug
            .from_interface(ftdi::Interface::C)
            .context("FTDI error")?;
        device
            .set_bitmode(0, ftdi::BitMode::Reset)
            .context("FTDI error")?;
        device.set_baud_rate(115200).context("FTDI error")?;
        Ok(UltradebugUart {
            inner: RefCell::new(Inner {
                device,
                baudrate: 115200,
            }),
        })
    }
}

impl Uart for UltradebugUart {
    fn get_baudrate(&self) -> Result<u32> {
        Ok(self.inner.borrow().baudrate)
    }

    fn set_baudrate(&self, baudrate: u32) -> Result<()> {
        let mut inner = self.inner.borrow_mut();
        inner.device.set_baud_rate(baudrate).context("FTDI error")?;
        inner.baudrate = baudrate;
        Ok(())
    }

    fn read_timeout(&self, buf: &mut [u8], timeout: Duration) -> Result<usize> {
        let now = Instant::now();
        let count = self.read(buf).context("UART read error")?;
        if count > 0 {
            return Ok(count);
        }
        let short_delay = cmp::min(timeout.mul_f32(0.1), Duration::from_millis(20));
        while now.elapsed() < timeout {
            thread::sleep(short_delay);
            let count = self.read(buf).context("UART read error")?;
            if count > 0 {
                return Ok(count);
            }
        }
        Ok(0)
    }

    fn read(&self, buf: &mut [u8]) -> Result<usize> {
        let n = self
            .inner
            .borrow_mut()
            .device
            .read(buf)
            .context("UART read error")?;
        Ok(n)
    }

    fn write(&self, mut buf: &[u8]) -> Result<()> {
        let mut inner = self.inner.borrow_mut();
        while !buf.is_empty() {
            let n = inner.device.write(buf).context("UART write error")?;
            buf = &buf[n..];
        }
        Ok(())
    }

    // TODO(#13290): Ideally, we would like to call `usb_purge_rx_buffer()` to
    // clear the UART RX buffer but our `libftdi` is too old.
}