i2c modules and defmt from OpenEMC

This commit is contained in:
Jörn-Michael Miehe 2024-03-03 15:49:38 +00:00
parent 4a7f688de9
commit fb40b4ace8
16 changed files with 1695 additions and 1 deletions

143
bluepill-rs/Cargo.lock generated
View file

@ -35,7 +35,11 @@ version = "0.1.0"
dependencies = [ dependencies = [
"cortex-m", "cortex-m",
"cortex-m-rt", "cortex-m-rt",
"defmt",
"defmt-ringbuf",
"defmt-rtt",
"embedded-hal 1.0.0", "embedded-hal 1.0.0",
"heapless",
"nb 1.1.0", "nb 1.1.0",
"panic-halt", "panic-halt",
"stm32f1xx-hal", "stm32f1xx-hal",
@ -53,6 +57,12 @@ dependencies = [
"vcell", "vcell",
] ]
[[package]]
name = "byteorder"
version = "1.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
[[package]] [[package]]
name = "cortex-m" name = "cortex-m"
version = "0.7.7" version = "0.7.7"
@ -83,7 +93,7 @@ checksum = "f0f6f3e36f203cfedbc78b357fb28730aa2c6dc1ab060ee5c2405e843988d3c7"
dependencies = [ dependencies = [
"proc-macro2", "proc-macro2",
"quote", "quote",
"syn", "syn 1.0.109",
] ]
[[package]] [[package]]
@ -92,6 +102,57 @@ version = "1.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7059fff8937831a9ae6f0fe4d658ffabf58f2ca96aa9dec1c889f936f705f216" checksum = "7059fff8937831a9ae6f0fe4d658ffabf58f2ca96aa9dec1c889f936f705f216"
[[package]]
name = "defmt"
version = "0.3.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3939552907426de152b3c2c6f51ed53f98f448babd26f28694c95f5906194595"
dependencies = [
"bitflags",
"defmt-macros",
]
[[package]]
name = "defmt-macros"
version = "0.3.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "18bdc7a7b92ac413e19e95240e75d3a73a8d8e78aa24a594c22cbb4d44b4bbda"
dependencies = [
"defmt-parser",
"proc-macro-error",
"proc-macro2",
"quote",
"syn 2.0.52",
]
[[package]]
name = "defmt-parser"
version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ff4a5fefe330e8d7f31b16a318f9ce81000d8e35e69b93eae154d16d2278f70f"
dependencies = [
"thiserror",
]
[[package]]
name = "defmt-ringbuf"
version = "0.2.0"
dependencies = [
"cortex-m",
"critical-section",
"defmt",
]
[[package]]
name = "defmt-rtt"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "609923761264dd99ed9c7d209718cda4631c5fe84668e0f0960124cbb844c49f"
dependencies = [
"critical-section",
"defmt",
]
[[package]] [[package]]
name = "embedded-dma" name = "embedded-dma"
version = "0.2.0" version = "0.2.0"
@ -142,6 +203,25 @@ version = "2.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1d758ba1b47b00caf47f24925c0074ecb20d6dfcffe7f6d53395c0465674841a" checksum = "1d758ba1b47b00caf47f24925c0074ecb20d6dfcffe7f6d53395c0465674841a"
[[package]]
name = "hash32"
version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "47d60b12902ba28e2730cd37e95b8c9223af2808df9e902d4df49588d1470606"
dependencies = [
"byteorder",
]
[[package]]
name = "heapless"
version = "0.8.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0bfb9eb618601c89945a70e254898da93b13be0388091d42117462b265bb3fad"
dependencies = [
"hash32",
"stable_deref_trait",
]
[[package]] [[package]]
name = "nb" name = "nb"
version = "0.1.3" version = "0.1.3"
@ -163,6 +243,30 @@ version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "de96540e0ebde571dc55c73d60ef407c653844e6f9a1e2fdbd40c07b9252d812" checksum = "de96540e0ebde571dc55c73d60ef407c653844e6f9a1e2fdbd40c07b9252d812"
[[package]]
name = "proc-macro-error"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "da25490ff9892aab3fcf7c36f08cfb902dd3e71ca0f9f9517bea02a73a5ce38c"
dependencies = [
"proc-macro-error-attr",
"proc-macro2",
"quote",
"syn 1.0.109",
"version_check",
]
[[package]]
name = "proc-macro-error-attr"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a1be40180e52ecc98ad80b184934baf3d0d29f979574e439af5a55274b35f869"
dependencies = [
"proc-macro2",
"quote",
"version_check",
]
[[package]] [[package]]
name = "proc-macro2" name = "proc-macro2"
version = "1.0.78" version = "1.0.78"
@ -265,6 +369,37 @@ dependencies = [
"unicode-ident", "unicode-ident",
] ]
[[package]]
name = "syn"
version = "2.0.52"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b699d15b36d1f02c3e7c69f8ffef53de37aefae075d8488d4ba1a7788d574a07"
dependencies = [
"proc-macro2",
"quote",
"unicode-ident",
]
[[package]]
name = "thiserror"
version = "1.0.57"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1e45bcbe8ed29775f228095caf2cd67af7a4ccf756ebff23a306bf3e8b47b24b"
dependencies = [
"thiserror-impl",
]
[[package]]
name = "thiserror-impl"
version = "1.0.57"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a953cb265bef375dae3de6663da4d3804eee9682ea80d8e2542529b73c531c81"
dependencies = [
"proc-macro2",
"quote",
"syn 2.0.52",
]
[[package]] [[package]]
name = "unicode-ident" name = "unicode-ident"
version = "1.0.12" version = "1.0.12"
@ -283,6 +418,12 @@ version = "0.1.3"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "77439c1b53d2303b20d9459b1ade71a83c716e3f9c34f3228c00e6f185d6c002" checksum = "77439c1b53d2303b20d9459b1ade71a83c716e3f9c34f3228c00e6f185d6c002"
[[package]]
name = "version_check"
version = "0.9.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "49874b5167b65d7193b8aba1567f5c7d93d001cafc34600cee003eda787e483f"
[[package]] [[package]]
name = "void" name = "void"
version = "1.0.2" version = "1.0.2"

View file

@ -6,10 +6,15 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies] [dependencies]
defmt-ringbuf = { path = "../defmt-ringbuf", optional = true }
embedded-hal = "1.0.0" embedded-hal = "1.0.0"
nb = "1" nb = "1"
cortex-m = { version = "0.7.6", features = ["critical-section-single-core"] } cortex-m = { version = "0.7.6", features = ["critical-section-single-core"] }
cortex-m-rt = { version = "0.7.1", features = ["device"] } cortex-m-rt = { version = "0.7.1", features = ["device"] }
defmt = { version = "0.3", features = ["encoding-rzcobs"] }
defmt-rtt = { version = "0.4", optional = true }
heapless = "0.8.0"
# Panic behaviour, see https://crates.io/keywords/panic-impl for alternatives # Panic behaviour, see https://crates.io/keywords/panic-impl for alternatives
panic-halt = "0.2.0" panic-halt = "0.2.0"

View file

@ -0,0 +1,360 @@
//! I2C register slave.
use core::ops::Deref;
use heapless::Vec;
use crate::i2c_slave::{self, Event as I2cEvent, I2cSlave, Instance};
/// I2C register slave error.
#[derive(Eq, PartialEq)]
#[non_exhaustive]
pub enum Error {
/// The buffer overflowed during reception.
BufferOverrun,
/// An I2C error occurred.
I2c(i2c_slave::Error),
}
impl From<i2c_slave::Error> for Error {
fn from(err: i2c_slave::Error) -> Self {
Self::I2c(err)
}
}
/// I2C slave using the register access model.
pub struct I2CRegSlave<I2C, PINS, const BUFFER: usize> {
slave: I2cSlave<I2C, PINS>,
reg: u8,
state: State,
buf: [u8; BUFFER],
pos: usize,
}
/// State of register slave.
#[derive(Clone, Copy)]
enum State {
/// Idle.
Idle,
/// Expecting to receive register.
ReceiveReg,
/// Register value received, waiting for write or read restart.
ReceivedReg,
/// Receiving register value.
Receiving,
/// Received read request, waiting for register value.
StartSend,
/// Sending register value.
Sending,
/// Waiting for end.
Clear,
}
impl<I2C, PINS, const BUFFER: usize> I2CRegSlave<I2C, PINS, BUFFER>
where
I2C: Instance,
{
/// Initializes the I2C register slave.
pub fn new(slave: I2cSlave<I2C, PINS>) -> Self {
Self {
slave,
reg: 0,
state: State::Idle,
buf: [0; BUFFER],
pos: 0,
}
}
/// Releases the I2C slave.
pub fn into_inner(self) -> I2cSlave<I2C, PINS> {
self.slave
}
/// Accesses the I2C slave.
///
/// This should not be used to process events, as doing so would
/// confuse the state of the I2C register model.
pub fn as_mut(&mut self) -> &mut I2cSlave<I2C, PINS> {
&mut self.slave
}
fn i2c_event(&mut self) -> nb::Result<i2c_slave::Event<I2C, PINS>, Error> {
match self.slave.event() {
Ok(evt) => Ok(evt),
Err(nb::Error::WouldBlock) => Err(nb::Error::WouldBlock),
Err(nb::Error::Other(err)) => {
self.state = State::Idle;
Err(nb::Error::Other(err.into()))
}
}
}
/// Gets the next event.
pub fn event(&mut self) -> nb::Result<Event<BUFFER>, Error> {
loop {
match &mut self.state {
State::Idle => match self.i2c_event()? {
I2cEvent::StartRead => self.state = State::ReceiveReg,
I2cEvent::StartWrite => {
self.state = State::StartSend;
self.pos = 0;
return Ok(Event::Read { reg: self.reg });
}
_ => defmt::unreachable!(),
},
State::ReceiveReg => match self.i2c_event()? {
I2cEvent::Read(value) => {
self.reg = value;
self.state = State::ReceivedReg;
}
I2cEvent::End | I2cEvent::Restart => self.state = State::Idle,
_ => defmt::unreachable!(),
},
State::ReceivedReg => match self.i2c_event()? {
I2cEvent::Read(value) => {
if !self.buf.is_empty() {
self.buf[0] = value;
self.pos = 1;
self.state = State::Receiving;
} else {
self.state = State::Clear;
return Err(nb::Error::Other(Error::BufferOverrun));
}
}
I2cEvent::End => {
self.state = State::Idle;
return Ok(Event::Write {
reg: self.reg,
value: Value(Vec::new()),
});
}
I2cEvent::Restart => self.state = State::Idle,
_ => defmt::unreachable!(),
},
State::Receiving => match self.i2c_event()? {
I2cEvent::Read(value) => {
if self.pos < self.buf.len() {
self.buf[self.pos] = value;
self.pos += 1;
} else {
return Err(nb::Error::Other(Error::BufferOverrun));
}
}
I2cEvent::End | I2cEvent::Restart => {
self.state = State::Idle;
return Ok(Event::Write {
reg: self.reg,
value: Value(defmt::unwrap!(Vec::try_from(&self.buf[..self.pos]))),
});
}
_ => defmt::unreachable!(),
},
State::StartSend => return Ok(Event::Read { reg: self.reg }),
State::Sending => {
let to_send = self.buf.get(self.pos).cloned().unwrap_or_default();
match self.i2c_event()? {
I2cEvent::Write(w) => {
w.write(to_send);
self.pos += 1;
}
I2cEvent::End | I2cEvent::Restart => self.state = State::Idle,
_ => defmt::unreachable!(),
}
}
State::Clear => match self.i2c_event()? {
I2cEvent::End => self.state = State::Idle,
I2cEvent::Write(w) => w.write(0),
_ => (),
},
}
}
}
/// Responds to a read register event.
pub fn respond(&mut self, response: Response<BUFFER>) {
if let State::StartSend = &self.state {
let n = response.0.len();
self.buf[..n].copy_from_slice(&response.0);
self.buf[n..].fill(0);
self.pos = 0;
self.state = State::Sending;
} else {
defmt::panic!("I2C register slave had no read register event")
}
}
}
/// I2C register slave event.
pub enum Event<const BUFFER: usize> {
/// Write register.
Write {
/// Register.
reg: u8,
/// Register value written by I2C master.
value: Value<BUFFER>,
},
/// Read register.
///
/// Call [`I2CRegSlave`] with your reply.
Read {
/// Register.
reg: u8,
},
}
/// The value written into an I2C register.
#[derive(Clone)]
pub struct Value<const BUFFER: usize>(pub Vec<u8, BUFFER>);
impl<const BUFFER: usize> Deref for Value<BUFFER> {
type Target = [u8];
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl<const BUFFER: usize> Value<BUFFER> {
fn get_or_zero(&self, idx: usize) -> u8 {
self.get(idx).cloned().unwrap_or_default()
}
/// Gets the register value as an u8.
///
/// Missing bytes are treated as zero and superflous bytes are ignored.
pub fn as_u8(&self) -> u8 {
self.get_or_zero(0)
}
/// Gets the register value as an u16 with LSB first.
///
/// Missing bytes are treated as zero and superflous bytes are ignored.
pub fn as_u16(&self) -> u16 {
let b = self.get_or_zero(0) as u16;
let a = self.get_or_zero(1) as u16;
(a << 8) | b
}
/// Gets the register value as an u32 with LSB first.
///
/// Missing bytes are treated as zero and superflous bytes are ignored.
pub fn as_u32(&self) -> u32 {
let d = self.get_or_zero(0) as u32;
let c = self.get_or_zero(1) as u32;
let b = self.get_or_zero(2) as u32;
let a = self.get_or_zero(3) as u32;
(a << 24) | (b << 16) | (c << 8) | d
}
/// Gets the register value as an u64 with LSB first.
///
/// Missing bytes are treated as zero and superflous bytes are ignored.
pub fn as_u64(&self) -> u64 {
let h = self.get_or_zero(0) as u64;
let g = self.get_or_zero(1) as u64;
let f = self.get_or_zero(2) as u64;
let e = self.get_or_zero(3) as u64;
let d = self.get_or_zero(4) as u64;
let c = self.get_or_zero(5) as u64;
let b = self.get_or_zero(6) as u64;
let a = self.get_or_zero(7) as u64;
(a << 56) | (b << 48) | (c << 40) | (d << 32) | (e << 24) | (f << 16) | (g << 8) | h
}
}
/// Response to register read request.
pub struct Response<const BUFFER: usize>(pub Vec<u8, BUFFER>);
impl<const BUFFER: usize> From<Vec<u8, BUFFER>> for Response<BUFFER> {
fn from(value: Vec<u8, BUFFER>) -> Self {
Self(value)
}
}
impl<const BUFFER: usize> Deref for Response<BUFFER> {
type Target = [u8];
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl<const BUFFER: usize> Response<BUFFER> {
/// Provides the register value.
///
/// The length of the value must not exceed `BUFFER`.
pub fn set(value: &[u8]) -> Self {
defmt::assert!(value.len() <= BUFFER);
Self(defmt::unwrap!(value.try_into()))
}
/// Provides an empty register value which will read as all zeros.
pub fn set_empty() -> Self {
Self::set(&[])
}
/// Provides the register value, clipping the provided value as necessary.
pub fn set_clipped(value: &[u8]) -> Self {
if value.len() <= BUFFER {
Self::set(value)
} else {
Self::set(&value[..BUFFER])
}
}
/// Sets the register value to an u8.
pub fn set_u8(value: u8) -> Self {
Self::set(&[value])
}
/// Sets the register value to an u16 with LSB first.
pub fn set_u16(value: u16) -> Self {
Self::set(&[(value & 0xff) as u8, ((value >> 8) & 0xff) as u8])
}
/// Sets the register value to an u32 with LSB first.
pub fn set_u32(value: u32) -> Self {
Self::set(&[
(value & 0xff) as u8,
((value >> 8) & 0xff) as u8,
((value >> 16) & 0xff) as u8,
((value >> 24) & 0xff) as u8,
])
}
/// Sets the register value to an u64 with LSB first.
pub fn set_u64(value: u64) -> Self {
Self::set(&[
(value & 0xff) as u8,
((value >> 8) & 0xff) as u8,
((value >> 16) & 0xff) as u8,
((value >> 24) & 0xff) as u8,
((value >> 32) & 0xff) as u8,
((value >> 40) & 0xff) as u8,
((value >> 48) & 0xff) as u8,
((value >> 56) & 0xff) as u8,
])
}
/// Sets the register value to an u128 with LSB first.
pub fn set_u128(value: u128) -> Self {
Self::set(&[
(value & 0xff) as u8,
((value >> 8) & 0xff) as u8,
((value >> 16) & 0xff) as u8,
((value >> 24) & 0xff) as u8,
((value >> 32) & 0xff) as u8,
((value >> 40) & 0xff) as u8,
((value >> 48) & 0xff) as u8,
((value >> 56) & 0xff) as u8,
((value >> 64) & 0xff) as u8,
((value >> 72) & 0xff) as u8,
((value >> 80) & 0xff) as u8,
((value >> 88) & 0xff) as u8,
((value >> 96) & 0xff) as u8,
((value >> 104) & 0xff) as u8,
((value >> 112) & 0xff) as u8,
((value >> 120) & 0xff) as u8,
])
}
}

View file

@ -0,0 +1,298 @@
//! I2C slave driver with interrupt support.
#![allow(unsafe_code)]
use core::ops::Deref;
use stm32f1xx_hal::{
afio::MAPR,
gpio::{self, Alternate, OpenDrain},
pac::{I2C1, I2C2, RCC},
rcc::{BusClock, Clocks, Enable, Reset},
time::Hertz,
};
/// I2C slave error.
#[derive(Eq, PartialEq)]
#[non_exhaustive]
pub enum Error {
/// Bus error
Bus,
}
/// Helper trait to ensure that the correct I2C pins are used for the corresponding interface
pub trait Pins<I2C> {
const REMAP: bool;
}
impl Pins<I2C1>
for (
gpio::PB6<Alternate<OpenDrain>>,
gpio::PB7<Alternate<OpenDrain>>,
)
{
const REMAP: bool = false;
}
impl Pins<I2C1>
for (
gpio::PB8<Alternate<OpenDrain>>,
gpio::PB9<Alternate<OpenDrain>>,
)
{
const REMAP: bool = true;
}
impl Pins<I2C2>
for (
gpio::PB10<Alternate<OpenDrain>>,
gpio::PB11<Alternate<OpenDrain>>,
)
{
const REMAP: bool = false;
}
/// I2C peripheral operating in slave mode.
///
/// Only a single 7-bit address is supported.
pub struct I2cSlave<I2C, PINS> {
i2c: I2C,
pins: PINS,
addr: u8,
pclk1: Hertz,
state: State,
}
#[derive(Clone, Copy)]
enum State {
Idle,
Receiving,
Sending,
WriteWaiting,
}
pub trait Instance:
Deref<Target = stm32f1xx_hal::pac::i2c1::RegisterBlock> + Enable + Reset + BusClock
{
}
impl Instance for I2C1 {}
impl Instance for I2C2 {}
impl<PINS> I2cSlave<I2C1, PINS> {
/// Creates a generic I2C1 object on pins PB6 and PB7 or PB8 and PB9 (if remapped)
pub fn i2c1(i2c: I2C1, pins: PINS, mapr: &mut MAPR, addr: u8, clocks: Clocks) -> Self
where
PINS: Pins<I2C1>,
{
mapr.modify_mapr(|_, w| w.i2c1_remap().bit(PINS::REMAP));
I2cSlave::<I2C1, _>::configure(i2c, pins, addr, clocks)
}
}
impl<PINS> I2cSlave<I2C2, PINS> {
/// Creates a generic I2C2 object on pins PB10 and PB11.
pub fn i2c2(i2c: I2C2, pins: PINS, addr: u8, clocks: Clocks) -> Self
where
PINS: Pins<I2C2>,
{
I2cSlave::<I2C2, _>::configure(i2c, pins, addr, clocks)
}
}
impl<I2C, PINS> I2cSlave<I2C, PINS>
where
I2C: Instance,
{
/// Configures the I2C peripheral to work in slave mode.
fn configure(i2c: I2C, pins: PINS, addr: u8, clocks: Clocks) -> Self {
let rcc = unsafe { &(*RCC::ptr()) };
I2C::enable(rcc);
I2C::reset(rcc);
let mut i2c = Self {
i2c,
pins,
addr,
pclk1: I2C::clock(&clocks),
state: State::Idle,
};
i2c.reset();
i2c
}
}
impl<I2C, PINS> I2cSlave<I2C, PINS>
where
I2C: Instance,
{
/// Initializes I2C.
///
/// Configures the `I2C_CRX` registers according to the system frequency.
fn init(&mut self) {
self.i2c.cr1.write(|w| w.pe().clear_bit());
let pclk1_mhz = self.pclk1.to_MHz() as u16;
self.i2c
.cr2
.write(|w| unsafe { w.freq().bits(pclk1_mhz as u8) });
self.i2c
.oar1
.write(|w| w.add().bits((self.addr as u16) << 1));
self.i2c.cr1.modify(|_, w| w.pe().enabled().ack().ack());
}
/// Perform an I2C software reset
fn reset(&mut self) {
self.i2c.cr1.write(|w| w.pe().set_bit().swrst().set_bit());
self.i2c.cr1.reset();
self.init();
}
/// Delay execution by reading SR1.
///
/// Workaround for I2C controller bugs.
fn sr1_delay(&mut self) {
for _ in 0..100 {
self.i2c.sr1.read();
}
}
/// Gets the next event.
pub fn event(&mut self) -> nb::Result<Event<I2C, PINS>, Error> {
let sr1 = self.i2c.sr1.read();
if sr1.berr().is_error() {
self.i2c.sr1.modify(|_, w| w.berr().clear_bit());
return Err(Error::Bus.into());
}
match self.state {
State::Idle => {
if sr1.addr().is_match() {
self.i2c.sr1.read();
let sr2 = self.i2c.sr2.read();
if sr2.tra().bit_is_set() {
self.state = State::Sending;
Ok(Event::StartWrite)
} else {
self.state = State::Receiving;
self.i2c.cr1.modify(|_, w| w.ack().ack());
Ok(Event::StartRead)
}
} else {
Err(nb::Error::WouldBlock)
}
}
State::Sending => {
if sr1.af().is_failure() {
self.i2c.sr1.modify(|_, w| w.af().no_failure());
self.state = State::Idle;
Ok(Event::End)
} else if sr1.tx_e().is_empty() {
self.state = State::WriteWaiting;
Ok(Event::Write(Writer(self)))
} else {
Err(nb::Error::WouldBlock)
}
}
State::WriteWaiting => defmt::panic!("write event was not handled"),
State::Receiving => {
if sr1.rx_ne().is_not_empty() {
let value = self.i2c.dr.read().dr().bits();
Ok(Event::Read(value))
} else if sr1.stopf().is_stop() || sr1.tx_e().is_empty() {
self.i2c.sr1.read();
self.i2c.cr1.modify(|_, w| w.pe().enabled().ack().ack());
self.state = State::Idle;
if sr1.tx_e().is_empty() {
// Workaround for I2C bus hanging after restart condition.
self.sr1_delay();
Ok(Event::Restart)
} else {
Ok(Event::End)
}
} else {
Err(nb::Error::WouldBlock)
}
}
}
}
/// Enables triggering the event interrupt (I2Cx_EV) when the transmit buffer is
/// empty or the receive buffer is not empty.
///
/// The event interrupt must be enabled separately using [`listen_event`](Self::listen_event).
pub fn listen_buffer(&mut self) {
self.i2c.cr2.modify(|_, w| w.itbufen().enabled());
}
/// Disables triggering the event interrupt (I2Cx_EV) when the transmit buffer is
/// empty or the receive buffer is not empty.
pub fn unlisten_buffer(&mut self) {
self.i2c.cr2.modify(|_, w| w.itbufen().disabled());
}
/// Enables triggering the event interrupt (I2Cx_EV) when the address is matched,
/// stop is detected, the byte transfer is finished or an event
/// described in [`listen_buffer`](Self::listen_buffer) occurrs.
pub fn listen_event(&mut self) {
self.i2c.cr2.modify(|_, w| w.itevten().enabled());
}
/// Disables triggering the event interrupt (I2Cx_EV) when the address is matched,
/// stop is detected, the byte transfer is finished or an event
/// described in [unlisten_buffer](Self::unlisten_buffer) occurrs.
pub fn unlisten_event(&mut self) {
self.i2c.cr2.modify(|_, w| w.itevten().disabled());
}
/// Enables triggering the error interrupt (I2Cx_ER) when an error is detected.
pub fn listen_error(&mut self) {
self.i2c.cr2.modify(|_, w| w.iterren().enabled());
}
/// Disables triggering the error interrupt (I2Cx_ER) when an error is detected.
pub fn unlisten_error(&mut self) {
self.i2c.cr2.modify(|_, w| w.iterren().disabled());
}
/// Releases the I2C peripheral and associated pins.
pub fn release(self) -> (I2C, PINS) {
(self.i2c, self.pins)
}
}
/// An I2C slave event.
pub enum Event<'a, I2C, PINS> {
/// Start of reception from the I2C master.
StartRead,
/// Start of transmission to the I2C master.
StartWrite,
/// A byte has been received from the I2C master.
Read(u8),
/// A byte must be transmitted to the I2C master.
Write(Writer<'a, I2C, PINS>),
/// End of reception and start of transmission.
Restart,
/// End of transfer.
End,
}
/// Transmits a byte to the I2C master.
pub struct Writer<'a, I2C, PINS>(&'a mut I2cSlave<I2C, PINS>);
impl<'a, I2C, PINS> Writer<'a, I2C, PINS>
where
I2C: Instance,
{
/// Sends the specified value to the I2C master.
pub fn write(self, value: u8) {
self.0.i2c.dr.write(|w| w.dr().bits(value));
self.0.state = State::Sending;
}
}

View file

@ -17,6 +17,9 @@ use nb::block;
use cortex_m_rt::entry; use cortex_m_rt::entry;
use stm32f1xx_hal::{pac, prelude::*, rcc, serial, timer::Timer}; use stm32f1xx_hal::{pac, prelude::*, rcc, serial, timer::Timer};
mod i2c_reg_slave;
mod i2c_slave;
#[entry] #[entry]
fn main() -> ! { fn main() -> ! {
// Get access to the core peripherals from the cortex-m crate // Get access to the core peripherals from the cortex-m crate

248
defmt-ringbuf/Cargo.lock generated Normal file
View file

@ -0,0 +1,248 @@
# This file is automatically @generated by Cargo.
# It is not intended for manual editing.
version = 3
[[package]]
name = "bare-metal"
version = "0.2.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5deb64efa5bd81e31fcd1938615a6d98c82eafcbcd787162b6f63b91d6bac5b3"
dependencies = [
"rustc_version",
]
[[package]]
name = "bitfield"
version = "0.13.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "46afbd2983a5d5a7bd740ccb198caf5b82f45c40c09c0eed36052d91cb92e719"
[[package]]
name = "bitflags"
version = "1.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
[[package]]
name = "cortex-m"
version = "0.7.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8ec610d8f49840a5b376c69663b6369e71f4b34484b9b2eb29fb918d92516cb9"
dependencies = [
"bare-metal",
"bitfield",
"embedded-hal",
"volatile-register",
]
[[package]]
name = "critical-section"
version = "1.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7059fff8937831a9ae6f0fe4d658ffabf58f2ca96aa9dec1c889f936f705f216"
[[package]]
name = "defmt"
version = "0.3.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3939552907426de152b3c2c6f51ed53f98f448babd26f28694c95f5906194595"
dependencies = [
"bitflags",
"defmt-macros",
]
[[package]]
name = "defmt-macros"
version = "0.3.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "18bdc7a7b92ac413e19e95240e75d3a73a8d8e78aa24a594c22cbb4d44b4bbda"
dependencies = [
"defmt-parser",
"proc-macro-error",
"proc-macro2",
"quote",
"syn 2.0.52",
]
[[package]]
name = "defmt-parser"
version = "0.3.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ff4a5fefe330e8d7f31b16a318f9ce81000d8e35e69b93eae154d16d2278f70f"
dependencies = [
"thiserror",
]
[[package]]
name = "defmt-ringbuf"
version = "0.2.0"
dependencies = [
"cortex-m",
"critical-section",
"defmt",
]
[[package]]
name = "embedded-hal"
version = "0.2.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "35949884794ad573cf46071e41c9b60efb0cb311e3ca01f7af807af1debc66ff"
dependencies = [
"nb 0.1.3",
"void",
]
[[package]]
name = "nb"
version = "0.1.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "801d31da0513b6ec5214e9bf433a77966320625a37860f910be265be6e18d06f"
dependencies = [
"nb 1.1.0",
]
[[package]]
name = "nb"
version = "1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8d5439c4ad607c3c23abf66de8c8bf57ba8adcd1f129e699851a6e43935d339d"
[[package]]
name = "proc-macro-error"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "da25490ff9892aab3fcf7c36f08cfb902dd3e71ca0f9f9517bea02a73a5ce38c"
dependencies = [
"proc-macro-error-attr",
"proc-macro2",
"quote",
"syn 1.0.109",
"version_check",
]
[[package]]
name = "proc-macro-error-attr"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a1be40180e52ecc98ad80b184934baf3d0d29f979574e439af5a55274b35f869"
dependencies = [
"proc-macro2",
"quote",
"version_check",
]
[[package]]
name = "proc-macro2"
version = "1.0.78"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2422ad645d89c99f8f3e6b88a9fdeca7fabeac836b1002371c4367c8f984aae"
dependencies = [
"unicode-ident",
]
[[package]]
name = "quote"
version = "1.0.35"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "291ec9ab5efd934aaf503a6466c5d5251535d108ee747472c3977cc5acc868ef"
dependencies = [
"proc-macro2",
]
[[package]]
name = "rustc_version"
version = "0.2.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "138e3e0acb6c9fb258b19b67cb8abd63c00679d2851805ea151465464fe9030a"
dependencies = [
"semver",
]
[[package]]
name = "semver"
version = "0.9.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1d7eb9ef2c18661902cc47e535f9bc51b78acd254da71d375c2f6720d9a40403"
dependencies = [
"semver-parser",
]
[[package]]
name = "semver-parser"
version = "0.7.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "388a1df253eca08550bef6c72392cfe7c30914bf41df5269b68cbd6ff8f570a3"
[[package]]
name = "syn"
version = "1.0.109"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
dependencies = [
"proc-macro2",
"unicode-ident",
]
[[package]]
name = "syn"
version = "2.0.52"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b699d15b36d1f02c3e7c69f8ffef53de37aefae075d8488d4ba1a7788d574a07"
dependencies = [
"proc-macro2",
"quote",
"unicode-ident",
]
[[package]]
name = "thiserror"
version = "1.0.57"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1e45bcbe8ed29775f228095caf2cd67af7a4ccf756ebff23a306bf3e8b47b24b"
dependencies = [
"thiserror-impl",
]
[[package]]
name = "thiserror-impl"
version = "1.0.57"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a953cb265bef375dae3de6663da4d3804eee9682ea80d8e2542529b73c531c81"
dependencies = [
"proc-macro2",
"quote",
"syn 2.0.52",
]
[[package]]
name = "unicode-ident"
version = "1.0.12"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3354b9ac3fae1ff6755cb6db53683adb661634f67557942dea4facebec0fee4b"
[[package]]
name = "vcell"
version = "0.1.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "77439c1b53d2303b20d9459b1ade71a83c716e3f9c34f3228c00e6f185d6c002"
[[package]]
name = "version_check"
version = "0.9.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "49874b5167b65d7193b8aba1567f5c7d93d001cafc34600cee003eda787e483f"
[[package]]
name = "void"
version = "1.0.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6a02e4885ed3bc0f2de90ea6dd45ebcbb66dacffe03547fadbb0eeae2770887d"
[[package]]
name = "volatile-register"
version = "0.2.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "de437e2a6208b014ab52972a27e59b33fa2920d3e00fe05026167a1c509d19cc"
dependencies = [
"vcell",
]

20
defmt-ringbuf/Cargo.toml Normal file
View file

@ -0,0 +1,20 @@
[package]
name = "defmt-ringbuf"
description = "defmt persistent ring buffer logger"
authors = ["Sebastian Urban <surban@surban.net>"]
categories = ["embedded", "no-std"]
keywords = ["defmt", "defmt-transport"]
readme = "README.md"
repository = "https://github.com/surban/openemc"
license = "MIT OR Apache-2.0"
version = "0.2.0"
edition = "2021"
[features]
default = ["logger"]
logger = []
[dependencies]
defmt = "0.3"
critical-section = "1"
cortex-m = "0.7"

View file

@ -0,0 +1,201 @@
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
1. Definitions.
"License" shall mean the terms and conditions for use, reproduction,
and distribution as defined by Sections 1 through 9 of this document.
"Licensor" shall mean the copyright owner or entity authorized by
the copyright owner that is granting the License.
"Legal Entity" shall mean the union of the acting entity and all
other entities that control, are controlled by, or are under common
control with that entity. For the purposes of this definition,
"control" means (i) the power, direct or indirect, to cause the
direction or management of such entity, whether by contract or
otherwise, or (ii) ownership of fifty percent (50%) or more of the
outstanding shares, or (iii) beneficial ownership of such entity.
"You" (or "Your") shall mean an individual or Legal Entity
exercising permissions granted by this License.
"Source" form shall mean the preferred form for making modifications,
including but not limited to software source code, documentation
source, and configuration files.
"Object" form shall mean any form resulting from mechanical
transformation or translation of a Source form, including but
not limited to compiled object code, generated documentation,
and conversions to other media types.
"Work" shall mean the work of authorship, whether in Source or
Object form, made available under the License, as indicated by a
copyright notice that is included in or attached to the work
(an example is provided in the Appendix below).
"Derivative Works" shall mean any work, whether in Source or Object
form, that is based on (or derived from) the Work and for which the
editorial revisions, annotations, elaborations, or other modifications
represent, as a whole, an original work of authorship. For the purposes
of this License, Derivative Works shall not include works that remain
separable from, or merely link (or bind by name) to the interfaces of,
the Work and Derivative Works thereof.
"Contribution" shall mean any work of authorship, including
the original version of the Work and any modifications or additions
to that Work or Derivative Works thereof, that is intentionally
submitted to Licensor for inclusion in the Work by the copyright owner
or by an individual or Legal Entity authorized to submit on behalf of
the copyright owner. For the purposes of this definition, "submitted"
means any form of electronic, verbal, or written communication sent
to the Licensor or its representatives, including but not limited to
communication on electronic mailing lists, source code control systems,
and issue tracking systems that are managed by, or on behalf of, the
Licensor for the purpose of discussing and improving the Work, but
excluding communication that is conspicuously marked or otherwise
designated in writing by the copyright owner as "Not a Contribution."
"Contributor" shall mean Licensor and any individual or Legal Entity
on behalf of whom a Contribution has been received by Licensor and
subsequently incorporated within the Work.
2. Grant of Copyright License. Subject to the terms and conditions of
this License, each Contributor hereby grants to You a perpetual,
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
copyright license to reproduce, prepare Derivative Works of,
publicly display, publicly perform, sublicense, and distribute the
Work and such Derivative Works in Source or Object form.
3. Grant of Patent License. Subject to the terms and conditions of
this License, each Contributor hereby grants to You a perpetual,
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
(except as stated in this section) patent license to make, have made,
use, offer to sell, sell, import, and otherwise transfer the Work,
where such license applies only to those patent claims licensable
by such Contributor that are necessarily infringed by their
Contribution(s) alone or by combination of their Contribution(s)
with the Work to which such Contribution(s) was submitted. If You
institute patent litigation against any entity (including a
cross-claim or counterclaim in a lawsuit) alleging that the Work
or a Contribution incorporated within the Work constitutes direct
or contributory patent infringement, then any patent licenses
granted to You under this License for that Work shall terminate
as of the date such litigation is filed.
4. Redistribution. You may reproduce and distribute copies of the
Work or Derivative Works thereof in any medium, with or without
modifications, and in Source or Object form, provided that You
meet the following conditions:
(a) You must give any other recipients of the Work or
Derivative Works a copy of this License; and
(b) You must cause any modified files to carry prominent notices
stating that You changed the files; and
(c) You must retain, in the Source form of any Derivative Works
that You distribute, all copyright, patent, trademark, and
attribution notices from the Source form of the Work,
excluding those notices that do not pertain to any part of
the Derivative Works; and
(d) If the Work includes a "NOTICE" text file as part of its
distribution, then any Derivative Works that You distribute must
include a readable copy of the attribution notices contained
within such NOTICE file, excluding those notices that do not
pertain to any part of the Derivative Works, in at least one
of the following places: within a NOTICE text file distributed
as part of the Derivative Works; within the Source form or
documentation, if provided along with the Derivative Works; or,
within a display generated by the Derivative Works, if and
wherever such third-party notices normally appear. The contents
of the NOTICE file are for informational purposes only and
do not modify the License. You may add Your own attribution
notices within Derivative Works that You distribute, alongside
or as an addendum to the NOTICE text from the Work, provided
that such additional attribution notices cannot be construed
as modifying the License.
You may add Your own copyright statement to Your modifications and
may provide additional or different license terms and conditions
for use, reproduction, or distribution of Your modifications, or
for any such Derivative Works as a whole, provided Your use,
reproduction, and distribution of the Work otherwise complies with
the conditions stated in this License.
5. Submission of Contributions. Unless You explicitly state otherwise,
any Contribution intentionally submitted for inclusion in the Work
by You to the Licensor shall be under the terms and conditions of
this License, without any additional terms or conditions.
Notwithstanding the above, nothing herein shall supersede or modify
the terms of any separate license agreement you may have executed
with Licensor regarding such Contributions.
6. Trademarks. This License does not grant permission to use the trade
names, trademarks, service marks, or product names of the Licensor,
except as required for reasonable and customary use in describing the
origin of the Work and reproducing the content of the NOTICE file.
7. Disclaimer of Warranty. Unless required by applicable law or
agreed to in writing, Licensor provides the Work (and each
Contributor provides its Contributions) on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
implied, including, without limitation, any warranties or conditions
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
PARTICULAR PURPOSE. You are solely responsible for determining the
appropriateness of using or redistributing the Work and assume any
risks associated with Your exercise of permissions under this License.
8. Limitation of Liability. In no event and under no legal theory,
whether in tort (including negligence), contract, or otherwise,
unless required by applicable law (such as deliberate and grossly
negligent acts) or agreed to in writing, shall any Contributor be
liable to You for damages, including any direct, indirect, special,
incidental, or consequential damages of any character arising as a
result of this License or out of the use or inability to use the
Work (including but not limited to damages for loss of goodwill,
work stoppage, computer failure or malfunction, or any and all
other commercial damages or losses), even if such Contributor
has been advised of the possibility of such damages.
9. Accepting Warranty or Additional Liability. While redistributing
the Work or Derivative Works thereof, You may choose to offer,
and charge a fee for, acceptance of support, warranty, indemnity,
or other liability obligations and/or rights consistent with this
License. However, in accepting such obligations, You may act only
on Your own behalf and on Your sole responsibility, not on behalf
of any other Contributor, and only if You agree to indemnify,
defend, and hold each Contributor harmless for any liability
incurred by, or claims asserted against, such Contributor by reason
of your accepting any such warranty or additional liability.
END OF TERMS AND CONDITIONS
APPENDIX: How to apply the Apache License to your work.
To apply the Apache License to your work, attach the following
boilerplate notice, with the fields enclosed by brackets "[]"
replaced with your own identifying information. (Don't include
the brackets!) The text should be enclosed in the appropriate
comment syntax for the file format. We also recommend that a
file or class name and description of purpose be included on the
same "printed page" as the copyright notice for easier
identification within third-party archives.
Copyright [yyyy] [name of copyright owner]
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

23
defmt-ringbuf/LICENSE-MIT Normal file
View file

@ -0,0 +1,23 @@
Permission is hereby granted, free of charge, to any
person obtaining a copy of this software and associated
documentation files (the "Software"), to deal in the
Software without restriction, including without
limitation the rights to use, copy, modify, merge,
publish, distribute, sublicense, and/or sell copies of
the Software, and to permit persons to whom the Software
is furnished to do so, subject to the following
conditions:
The above copyright notice and this permission notice
shall be included in all copies or substantial portions
of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
DEALINGS IN THE SOFTWARE.

21
defmt-ringbuf/README.md Normal file
View file

@ -0,0 +1,21 @@
# defmt-ringbuf
This crate stores [`defmt`] log messages in a simple ring buffer that is persisted across resets.
You still need to read the messages from the buffer and transfer them to a host for formatting.
[`defmt`]: https://github.com/knurling-rs/defmt
## License
Licensed under either of
- [Apache License, Version 2.0](LICENSE-APACHE)
- [MIT license](LICENSE-MIT)
at your option.
## Contribution
Unless you explicitly state otherwise, any contribution intentionally submitted
for inclusion in defmt-ringbuf by you shall be licensed as above,
without any additional terms or conditions.

View file

@ -0,0 +1,10 @@
[build]
target = "thumbv7m-none-eabi"
rustflags = [
"-C", "link-arg=-Tlink.x",
"-C", "link-arg=--nmagic",
"-C", "link-arg=-Tdefmt.x",
]
[target.'cfg(all(target_arch = "arm", target_os = "none"))']
runner = "probe-run --chip STM32F103RB"

View file

@ -0,0 +1,22 @@
[package]
name = "defmt-ringbuf-test"
authors = ["Sebastian Urban <surban@surban.net>"]
license = "GPL-3.0"
version = "0.0.0"
publish = false
edition = "2021"
[dependencies]
defmt-ringbuf = { path = "..", default-features = false }
cortex-m = { version = "0.7", features = ["critical-section-single-core"] }
cortex-m-rt = { version = "0.7", features = ["device"] }
stm32f1 = { version = "0.15", features = ["stm32f103", "rt"] }
panic-probe = "0.3"
defmt = "0.3"
defmt-rtt = "0.4"
[profile.release]
opt-level = 'z'
lto = true
debug = 2

View file

@ -0,0 +1,5 @@
MEMORY
{
FLASH : ORIGIN = 0x08000000, LENGTH = 64K
RAM : ORIGIN = 0x20000000, LENGTH = 20K
}

View file

@ -0,0 +1,42 @@
#![no_std]
#![no_main]
use core::mem;
use cortex_m::{asm::nop, peripheral::SCB};
use cortex_m_rt::entry;
use defmt_ringbuf::{RingBuf, RingBuffer};
use defmt_rtt as _;
use panic_probe as _;
use stm32f1::stm32f103::Peripherals;
#[no_mangle]
#[used]
#[link_section = ".uninit.BUFFER"]
pub static mut BUFFER: mem::MaybeUninit<defmt_ringbuf::RingBuffer<4096>> = mem::MaybeUninit::uninit();
#[entry]
fn main() -> ! {
let _dp = Peripherals::take().unwrap();
defmt::info!("Init");
let buffer = unsafe { RingBuffer::init(&mut BUFFER) };
let mut data = [0; 128];
let (n, lost) = buffer.read(&mut data);
let data = &data[..n];
defmt::info!("Buffer contents (lost={}): {}", lost, data);
let mut data = [0; 255];
for (i, d) in data.iter_mut().enumerate() {
*d = i as u8;
}
buffer.write(&data);
defmt::info!("Wrote {} bytes to buffer", data.len());
defmt::info!("Done");
for _ in 0..1_000_000 {
nop();
}
SCB::sys_reset();
}

140
defmt-ringbuf/src/lib.rs Normal file
View file

@ -0,0 +1,140 @@
//! defmt-ringbuf is a [`defmt`](https://github.com/knurling-rs/defmt) global logger that logs into
//! a persistent ring buffer.
//!
//! The ring buffer is not cleared at startup, i.e. if placed in a static global variable log messages
//! will still be available after a reset.
//!
//! To use this crate, link to it by importing it somewhere in your project.
//!
//! ```
//! use core::mem::MaybeUninit;
//! use defmt_ringbuf as _;
//!
//! static mut LOG: MaybeUninit<defmt_ringbuf::RingBuffer<1024>> = MaybeUninit::uninit();
//!
//! #[entry]
//! fn main() -> ! {
//! unsafe {
//! defmt_ringbuf::init(&mut LOG, || ());
//! }
//!
//! // ...
//! }
//! ```
//!
//! Call [init] to initialize logging and [read] to read buffered log data.
//!
//! # Critical section implementation
//!
//! This crate uses [`critical-section`](https://github.com/rust-embedded/critical-section) to ensure only one thread
//! is writing to the buffer at a time. You must import a crate that provides a `critical-section` implementation
//! suitable for the current target. See the `critical-section` README for details.
//!
//! For example, for single-core privileged-mode Cortex-M targets, you can add the following to your Cargo.toml.
//!
//! ```toml
//! [dependencies]
//! cortex-m = { version = "0.7.6", features = ["critical-section-single-core"]}
//! ```
#![no_std]
use core::{
mem::MaybeUninit,
sync::atomic::{AtomicBool, Ordering},
};
mod ring_buffer;
pub use ring_buffer::{RingBuf, RingBuffer};
#[cfg_attr(feature = "logger", defmt::global_logger)]
struct Logger;
/// Global logger lock.
static TAKEN: AtomicBool = AtomicBool::new(false);
/// Crticial section.
static mut CS_RESTORE: critical_section::RestoreState = critical_section::RestoreState::invalid();
/// Encoder.
static mut ENCODER: defmt::Encoder = defmt::Encoder::new();
/// Ring buffer.
static mut RING_BUFFER: Option<&'static mut dyn RingBuf> = None;
/// Callback when new log data is available.
static mut LOG_AVAILABLE: fn() = || ();
unsafe impl defmt::Logger for Logger {
fn acquire() {
let restore = unsafe { critical_section::acquire() };
if TAKEN.load(Ordering::Relaxed) {
panic!("defmt logger taken reentrantly")
}
TAKEN.store(true, Ordering::Relaxed);
unsafe { CS_RESTORE = restore };
unsafe { ENCODER.start_frame(do_write) }
}
unsafe fn flush() {
// Flush is a no-op.
}
unsafe fn release() {
ENCODER.end_frame(do_write);
TAKEN.store(false, Ordering::Relaxed);
let restore = CS_RESTORE;
critical_section::release(restore);
}
unsafe fn write(bytes: &[u8]) {
ENCODER.write(bytes, do_write);
}
}
fn do_write(data: &[u8]) {
unsafe {
if let Some(buffer) = RING_BUFFER.as_mut() {
buffer.write(data);
LOG_AVAILABLE();
}
}
}
/// Initializes logging to a ring buffer.
///
/// `ring_buffer` specifies the location of the log buffer.
/// It is not cleared if it contains vailid data from the previous boot.
///
/// `log_available` is called when new log messages are available.
///
/// This must be called exactly once.
/// Log messages received before initiailization are discarded.
pub unsafe fn init<const SIZE: usize>(
ring_buffer: &'static mut MaybeUninit<RingBuffer<SIZE>>,
log_available: fn(),
) {
defmt::assert!(RING_BUFFER.is_none());
let ring_buffer = RingBuffer::init(ring_buffer);
RING_BUFFER = Some(ring_buffer as &mut dyn RingBuf);
LOG_AVAILABLE = log_available;
}
/// Reads and removes data from the log buffer.
///
/// Returns the number of bytes read and whether data was lost.
pub fn read(data: &mut [u8]) -> (usize, bool) {
unsafe {
critical_section::with(|_cs| {
if let Some(buffer) = RING_BUFFER.as_mut() {
buffer.read(data)
} else {
(0, false)
}
})
}
}

View file

@ -0,0 +1,155 @@
//! Ring buffer.
use core::{
mem::MaybeUninit,
ptr::{addr_of, addr_of_mut},
sync::atomic::{AtomicBool, AtomicU32, AtomicU8, AtomicUsize, Ordering},
};
use cortex_m::Peripherals;
/// Ring buffer access methods.
pub trait RingBuf {
/// Write into the ring buffer.
fn write(&mut self, data: &[u8]);
/// Read from the ring buffer.
///
/// Returns the number of bytes read and whether data was lost.
fn read(&mut self, data: &mut [u8]) -> (usize, bool);
}
/// Ring buffer.
#[repr(C)]
pub struct RingBuffer<const SIZE: usize> {
/// Signature for validity check.
signature: AtomicU32,
/// Read position.
read_pos: AtomicUsize,
/// Write position.
write_pos: AtomicUsize,
/// Unread data overwritten?
overwritten: AtomicBool,
/// Buffer.
buf: [AtomicU8; SIZE],
}
impl<const SIZE: usize> RingBuffer<SIZE> {
/// Signature for validity check.
const SIGNATURE: u32 = 0xb0ffe300;
/// Initializes the ring buffer, keeping its data if it appears valid.
pub fn init(uninit: &mut MaybeUninit<Self>) -> &mut Self {
unsafe {
let mut scb = Peripherals::steal().SCB;
let ptr = uninit.as_mut_ptr();
let signature = (addr_of!((*ptr).signature) as *const u32).read_volatile();
let mut read_pos = (addr_of!((*ptr).read_pos) as *const usize).read_volatile();
let mut write_pos = (addr_of!((*ptr).write_pos) as *const usize).read_volatile();
let mut overwritten = (addr_of!((*ptr).overwritten) as *const u8).read_volatile();
let valid = signature == Self::SIGNATURE
&& read_pos < SIZE
&& write_pos < SIZE
&& (overwritten == 0 || overwritten == 1);
if !valid {
addr_of_mut!((*ptr).signature).write_volatile(AtomicU32::new(0));
scb.clean_dcache_by_ref(&(*ptr).signature);
read_pos = 0;
write_pos = 0;
overwritten = 0;
}
for i in 0..SIZE {
let b = if valid { (addr_of!((*ptr).buf[i]) as *const u8).read_volatile() } else { 0 };
addr_of_mut!((*ptr).buf[i]).write_volatile(AtomicU8::new(b));
}
scb.clean_dcache_by_slice(&(*ptr).buf);
addr_of_mut!((*ptr).read_pos).write_volatile(AtomicUsize::new(read_pos));
scb.clean_dcache_by_ref(&(*ptr).read_pos);
addr_of_mut!((*ptr).write_pos).write_volatile(AtomicUsize::new(write_pos));
scb.clean_dcache_by_ref(&(*ptr).write_pos);
addr_of_mut!((*ptr).overwritten).write_volatile(AtomicBool::new(overwritten != 0));
scb.clean_dcache_by_ref(&(*ptr).overwritten);
addr_of_mut!((*ptr).signature).write_volatile(AtomicU32::new(Self::SIGNATURE));
scb.clean_dcache_by_ref(&(*ptr).signature);
// SAFETY: all fields have been initialized with either default values or their previous contents.
uninit.assume_init_mut()
}
}
}
impl<const SIZE: usize> RingBuf for RingBuffer<SIZE> {
fn write(&mut self, mut data: &[u8]) {
let mut scb = unsafe { Peripherals::steal().SCB };
while !data.is_empty() {
// Split data into part that fits remaining buffer.
let write_pos = self.write_pos.load(Ordering::SeqCst);
let to_end = SIZE - write_pos;
let (part, rest) = data.split_at(to_end.min(data.len()));
data = rest;
// Calculate write boundaries.
let from = write_pos;
let to = write_pos + part.len();
// Update read position if we overwrite unread data.
let read_pos = self.read_pos.load(Ordering::SeqCst);
if from < read_pos && to >= read_pos {
self.overwritten.store(true, Ordering::SeqCst);
let mut new_read_pos = to + 1;
if new_read_pos >= SIZE {
new_read_pos = 0;
}
self.read_pos.store(new_read_pos, Ordering::SeqCst);
scb.clean_dcache_by_ref(&self.read_pos);
}
// Copy.
for (dst, src) in self.buf[from..to].iter_mut().zip(part.iter()) {
dst.store(*src, Ordering::SeqCst);
}
scb.clean_dcache_by_slice(&self.buf[from..to]);
// Update write position.
let new_write_pos = if to == SIZE { 0 } else { to };
self.write_pos.store(new_write_pos, Ordering::SeqCst);
scb.clean_dcache_by_ref(&self.write_pos);
}
}
fn read(&mut self, data: &mut [u8]) -> (usize, bool) {
let mut scb = unsafe { Peripherals::steal().SCB };
let read_pos = self.read_pos.load(Ordering::SeqCst);
let write_pos = self.write_pos.load(Ordering::SeqCst);
let overwritten = self.overwritten.load(Ordering::SeqCst);
// Calculate available data.
let avail = if read_pos > write_pos { SIZE - read_pos } else { write_pos - read_pos };
let n = avail.min(data.len());
// Copy.
let from = read_pos;
let to = read_pos + n;
for (dst, src) in data[..n].iter_mut().zip(self.buf[from..to].iter()) {
*dst = src.load(Ordering::SeqCst);
}
// Update read position and overwritten status.
let new_read_pos = if to == SIZE { 0 } else { to };
self.read_pos.store(new_read_pos, Ordering::SeqCst);
scb.clean_dcache_by_ref(&self.read_pos);
self.overwritten.store(false, Ordering::SeqCst);
scb.clean_dcache_by_ref(&self.overwritten);
(n, overwritten)
}
}