opentitanlib/console/
spi.rs1use anyhow::Result;
6use std::cell::{Cell, RefCell};
7use std::collections::VecDeque;
8use std::rc::Rc;
9use std::time::Duration;
10
11use crate::io::console::ConsoleDevice;
12use crate::io::gpio::GpioPin;
13use crate::io::spi::Target;
14use crate::spiflash::flash::SpiFlash;
15
16pub struct SpiConsoleDevice<'a> {
17 spi: &'a dyn Target,
18 flash: SpiFlash,
19 console_next_frame_number: Cell<u32>,
20 rx_buf: RefCell<VecDeque<u8>>,
21 next_read_address: Cell<u32>,
22 device_tx_ready_pin: Option<&'a Rc<dyn GpioPin>>,
23}
24
25impl<'a> SpiConsoleDevice<'a> {
26 const SPI_FRAME_HEADER_SIZE: usize = 12;
27 const SPI_FLASH_READ_BUFFER_SIZE: u32 = 2048;
28 const SPI_MAX_DATA_LENGTH: usize = 2036;
29 const SPI_FRAME_MAGIC_NUMBER: u32 = 0xa5a5beef;
30 const SPI_FLASH_PAYLOAD_BUFFER_SIZE: usize = 256;
31 const SPI_TX_LAST_CHUNK_MAGIC_ADDRESS: u32 = 0x100;
32 const SPI_BOOT_MAGIC_PATTERN: u32 = 0xcafeb002;
33
34 pub fn new(
35 spi: &'a dyn Target,
36 device_tx_ready_pin: Option<&'a Rc<dyn GpioPin>>,
37 ) -> Result<Self> {
38 let flash = SpiFlash {
39 ..Default::default()
40 };
41 Ok(Self {
42 spi,
43 flash,
44 rx_buf: RefCell::new(VecDeque::new()),
45 console_next_frame_number: Cell::new(0),
46 next_read_address: Cell::new(0),
47 device_tx_ready_pin,
48 })
49 }
50
51 fn check_device_boot_up(&self, buf: &[u8]) -> Result<usize> {
52 for i in (0..buf.len()).step_by(4) {
53 let pattern: u32 = u32::from_le_bytes(buf[i..i + 4].try_into().unwrap());
54 if pattern != SpiConsoleDevice::SPI_BOOT_MAGIC_PATTERN {
55 return Ok(0);
56 }
57 }
58 self.flash.program(self.spi, 0, buf)?;
60 self.console_next_frame_number.set(0);
61 self.next_read_address.set(0);
62 Ok(0)
63 }
64
65 fn read_from_spi(&self) -> Result<usize> {
66 let read_address = self.next_read_address.get();
68 let mut header = vec![0u8; SpiConsoleDevice::SPI_FRAME_HEADER_SIZE];
69 self.read_data(read_address, &mut header)?;
70
71 let magic_number: u32 = u32::from_le_bytes(header[0..4].try_into().unwrap());
72 let frame_number: u32 = u32::from_le_bytes(header[4..8].try_into().unwrap());
73 let data_len_bytes: usize = u32::from_le_bytes(header[8..12].try_into().unwrap()) as usize;
74 if magic_number != SpiConsoleDevice::SPI_FRAME_MAGIC_NUMBER
75 || frame_number != self.console_next_frame_number.get()
76 || data_len_bytes > SpiConsoleDevice::SPI_MAX_DATA_LENGTH
77 {
78 if self.get_tx_ready_pin()?.is_none() {
79 self.check_device_boot_up(&header)?;
80 }
81 return Ok(0);
83 }
84 self.console_next_frame_number.set(frame_number + 1);
85
86 let data_len_bytes_w_pad = (data_len_bytes + 3) & !3;
88 let mut data = vec![0u8; data_len_bytes_w_pad];
89 let data_address: u32 = (read_address
90 + u32::try_from(SpiConsoleDevice::SPI_FRAME_HEADER_SIZE).unwrap())
91 % SpiConsoleDevice::SPI_FLASH_READ_BUFFER_SIZE;
92 self.read_data(data_address, &mut data)?;
93
94 if self.get_tx_ready_pin()?.is_some() {
95 self.next_read_address.set(0);
99 } else {
100 let next_read_address: u32 = (read_address
101 + u32::try_from(SpiConsoleDevice::SPI_FRAME_HEADER_SIZE + data_len_bytes_w_pad)
102 .unwrap())
103 % SpiConsoleDevice::SPI_FLASH_READ_BUFFER_SIZE;
104 self.next_read_address.set(next_read_address);
105 }
106
107 self.rx_buf.borrow_mut().extend(&data[..data_len_bytes]);
109 Ok(data_len_bytes)
110 }
111
112 fn read_data(&self, address: u32, buf: &mut [u8]) -> Result<&Self> {
113 let buf_len: usize = buf.len();
114 let space_to_end_of_buffer: usize =
115 usize::try_from(SpiConsoleDevice::SPI_FLASH_READ_BUFFER_SIZE - address).unwrap();
116 let first_part_size: usize = if buf_len > space_to_end_of_buffer {
117 space_to_end_of_buffer
118 } else {
119 buf_len
120 };
121 self.flash
122 .read(self.spi, address, &mut buf[0..first_part_size])?;
123 if first_part_size < buf_len {
125 self.flash
126 .read(self.spi, 0, &mut buf[first_part_size..buf_len])?;
127 }
128 Ok(self)
129 }
130}
131
132impl<'a> ConsoleDevice for SpiConsoleDevice<'a> {
133 fn console_poll_read(
134 &self,
135 cx: &mut std::task::Context<'_>,
136 buf: &mut [u8],
137 ) -> std::task::Poll<Result<usize>> {
138 const POLL_INTERVAL: Duration = Duration::from_millis(1);
142
143 if self.rx_buf.borrow().is_empty() {
144 if let Some(ready_pin) = self.get_tx_ready_pin()? {
145 if !ready_pin.read()? {
148 return crate::util::runtime::poll_later(cx, POLL_INTERVAL);
149 }
150 }
151
152 if self.read_from_spi()? == 0 {
153 return crate::util::runtime::poll_later(cx, POLL_INTERVAL);
154 }
155 }
156
157 let mut i: usize = 0;
159 while !self.rx_buf.borrow().is_empty() && i < buf.len() {
160 buf[i] = self.rx_buf.borrow_mut().pop_front().unwrap();
161 i += 1;
162 }
163
164 std::task::Poll::Ready(Ok(i))
165 }
166
167 fn console_write(&self, buf: &[u8]) -> Result<()> {
168 let buf_len: usize = buf.len();
169 let mut written_data_len: usize = 0;
170 while written_data_len < buf_len {
171 let mut write_address = SpiConsoleDevice::SPI_TX_LAST_CHUNK_MAGIC_ADDRESS;
172 let mut data_len: usize = buf_len - written_data_len;
173
174 if data_len > SpiConsoleDevice::SPI_FLASH_PAYLOAD_BUFFER_SIZE {
175 data_len = SpiConsoleDevice::SPI_FLASH_PAYLOAD_BUFFER_SIZE;
176 write_address = 0;
177 }
178
179 self.flash.program(
180 self.spi,
181 write_address,
182 &buf[written_data_len..written_data_len + data_len],
183 )?;
184 written_data_len += data_len;
185 }
186
187 Ok(())
188 }
189
190 fn get_tx_ready_pin(&self) -> Result<Option<&'a Rc<dyn GpioPin>>> {
191 Ok(self.device_tx_ready_pin)
192 }
193}