Crate embedded_hal_nb

source ·
Expand description

Non-blocking Hardware Abstraction Layer (HAL) traits for embedded systems, using the nb crate.

The embedded-hal-nb traits make use of the nb crate (please go read that crate documentation before continuing) to abstract over the asynchronous model and to also provide a blocking operation mode.

Here’s how a HAL trait may look like:

use embedded_hal_nb;

/// A serial interface
pub trait Serial {
    /// Error type associated to this serial interface
    type Error: core::fmt::Debug;

    /// Reads a single byte
    fn read(&mut self) -> nb::Result<u8, Self::Error>;

    /// Writes a single byte
    fn write(&mut self, byte: u8) -> nb::Result<(), Self::Error>;
}

The nb::Result enum is used to add a WouldBlock variant to the errors of the serial interface. As explained in the documentation of the nb crate this single API, when paired with the macros in the nb crate, can operate in a blocking manner, or be adapted to other asynchronous execution schemes.

Some traits, like the one shown below, may expose possibly blocking APIs that can’t fail. In those cases nb::Result<_, Infallible> is used.

use ::core::convert::Infallible;

/// A count down timer
pub trait CountDown {
    // ..

    /// "waits" until the count down is over
    fn wait(&mut self) -> nb::Result<(), Infallible>;
}

§Suggested implementation

The HAL traits should be implemented for device crates generated via svd2rust to maximize code reuse.

Shown below is an implementation of some of the HAL traits for the [stm32f1xx-hal] crate. This single implementation will work for any microcontroller in the STM32F1xx family.

// crate: stm32f1xx-hal
// An implementation of the `embedded-hal` traits for STM32F1xx microcontrollers

use embedded_hal_nb::serial;
use nb;

// device crate
use stm32f1::stm32f103::USART1;

/// A serial interface
// NOTE generic over the USART peripheral
pub struct Serial<USART> { usart: USART }

// convenience type alias
pub type Serial1 = Serial<USART1>;

impl serial::ErrorType for Serial<USART1> {
    type Error = serial::ErrorKind;
}

impl embedded_hal_nb::serial::Read<u8> for Serial<USART1> {
    fn read(&mut self) -> nb::Result<u8, Self::Error> {
        // read the status register
        let isr = self.usart.sr.read();

        if isr.ore().bit_is_set() {
            // Error: Buffer overrun
            Err(nb::Error::Other(Self::Error::Overrun))
        }
        // omitted: checks for other errors
        else if isr.rxne().bit_is_set() {
            // Data available: read the data register
            Ok(self.usart.dr.read().bits() as u8)
        } else {
            // No data available yet
            Err(nb::Error::WouldBlock)
        }
    }
}

impl embedded_hal_nb::serial::Write<u8> for Serial<USART1> {
    fn write(&mut self, byte: u8) -> nb::Result<(), Self::Error> {
        // Similar to the `read` implementation
    }

    fn flush(&mut self) -> nb::Result<(), Self::Error> {
        // Similar to the `read` implementation
    }
}

§Intended usage

Thanks to the nb crate the HAL API can be used in a blocking manner with the block! macro or with futures.

§Blocking mode

An example of writing a string over the serial interface in a blocking fashion:

use stm32f1xx_hal::Serial1;
use embedded_hal_nb::serial::Write;
use nb::block;

let mut serial: Serial1 = {
    // ..
};

for byte in b"Hello, world!" {
    // NOTE `block!` blocks until `serial.write()` completes and returns
    // `Result<(), Error>`
    block!(serial.write(*byte)).unwrap();
}

§Generic programming and higher level abstractions

The core of the HAL has been kept minimal on purpose to encourage building generic higher level abstractions on top of it. Some higher level abstractions that pick an asynchronous model or that have blocking behavior and that are deemed useful to build other abstractions can be found in the blocking module.

Some examples:

NOTE All the functions shown below could have been written as trait methods with default implementation to allow specialization, but they have been written as functions to keep things simple.

  • Write a whole buffer to a serial device in blocking a fashion.
use embedded_hal_nb::serial::Write;
use nb::block;

fn write_all<S>(serial: &mut S, buffer: &[u8]) -> Result<(), S::Error>
where
    S: Write<u8>
{
    for &byte in buffer {
        block!(serial.write(byte))?;
    }

    Ok(())
}
  • Buffered serial interface with periodic flushing in interrupt handler
use embedded_hal_nb::serial::{ErrorKind, Write};
use nb::block;

fn flush<S>(serial: &mut S, cb: &mut CircularBuffer)
where
    S: Write<u8, Error = ErrorKind>,
{
    loop {
        if let Some(byte) = cb.peek() {
            match serial.write(*byte) {
                Err(nb::Error::Other(_)) => unreachable!(),
                Err(nb::Error::WouldBlock) => return,
                Ok(()) => {}, // keep flushing data
            }
        }

        cb.pop();
    }
}

// The stuff below could be in some other crate

/// Global singleton
pub struct BufferedSerial1;

// NOTE private
static BUFFER1: Mutex<CircularBuffer> = {
    // ..
};
static SERIAL1: Mutex<Serial1> = {
    // ..
};

impl BufferedSerial1 {
    pub fn write(&self, byte: u8) {
        self.write_all(&[byte])
    }

    pub fn write_all(&self, bytes: &[u8]) {
        let mut buffer = BUFFER1.lock();
        for byte in bytes {
            buffer.push(*byte).expect("buffer overrun");
        }
        // omitted: pend / enable interrupt_handler
    }
}

fn interrupt_handler() {
    let mut serial = SERIAL1.lock();
    let mut buffer = BUFFER1.lock();

    flush(&mut *serial, &mut buffer);
}

Re-exports§

  • pub use nb;

Modules§

  • Serial interface.
  • SPI master mode traits using nb.