Skip to content

Commit

Permalink
Add BorrowedBuf::into_filled{,_mut} methods to allow returning buffer…
Browse files Browse the repository at this point in the history
… with original lifetime
  • Loading branch information
SUPERCILEX authored and Alex Saveau committed Nov 2, 2024
1 parent 771d77c commit e883a60
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions core/src/io/borrowed_buf.rs
Original file line number Diff line number Diff line change
Expand Up @@ -108,6 +108,26 @@ impl<'data> BorrowedBuf<'data> {
}
}

/// Returns a shared reference to the filled portion of the buffer with its original lifetime.
#[inline]
pub fn into_filled(self) -> &'data [u8] {
// SAFETY: We only slice the filled part of the buffer, which is always valid
unsafe {
let buf = self.buf.get_unchecked(..self.filled);
MaybeUninit::slice_assume_init_ref(buf)
}
}

/// Returns a mutable reference to the filled portion of the buffer with its original lifetime.
#[inline]
pub fn into_filled_mut(self) -> &'data mut [u8] {
// SAFETY: We only slice the filled part of the buffer, which is always valid
unsafe {
let buf = self.buf.get_unchecked_mut(..self.filled);
MaybeUninit::slice_assume_init_mut(buf)
}
}

/// Returns a cursor over the unfilled part of the buffer.
#[inline]
pub fn unfilled<'this>(&'this mut self) -> BorrowedCursor<'this> {
Expand Down

0 comments on commit e883a60

Please sign in to comment.