hal-common/atomic_ring_buffer: Add push_slice, pop_slice.

1 files changed, 16 insertions, 0 deletions

diff --git a/embassy-hal-common/src/atomic_ring_buffer.rs b/embassy-hal-common/src/atomic_ring_buffer.rs
index 3e90b0870..a8a6a2166 100644
--- a/embassy-hal-common/src/atomic_ring_buffer.rs
+++ b/embassy-hal-common/src/atomic_ring_buffer.rs
@@ -137,6 +137,14 @@ impl<'a> Writer<'a> {
 
     /// Get a buffer where data can be pushed to.
     ///
+    /// Equivalent to [`Self::push_buf`] but returns a slice.
+    pub fn push_slice(&mut self) -> &mut [u8] {
+        let (data, len) = self.push_buf();
+        unsafe { slice::from_raw_parts_mut(data, len) }
+    }
+
+    /// Get a buffer where data can be pushed to.
+    ///
     /// Write data to the start of the buffer, then call `push_done` with
     /// however many bytes you've pushed.
     ///
@@ -206,6 +214,14 @@ impl<'a> Reader<'a> {
 
     /// Get a buffer where data can be popped from.
     ///
+    /// Equivalent to [`Self::pop_buf`] but returns a slice.
+    pub fn pop_slice(&mut self) -> &mut [u8] {
+        let (data, len) = self.pop_buf();
+        unsafe { slice::from_raw_parts_mut(data, len) }
+    }
+
+    /// Get a buffer where data can be popped from.
+    ///
     /// Read data from the start of the buffer, then call `pop_done` with
     /// however many bytes you've processed.
     ///