Skip to content

Commit

Permalink
Revert "der: have Reader::peek* methods take &mut self (#1418)"
Browse files Browse the repository at this point in the history
This reverts commit 42ec503.
  • Loading branch information
tarcieri committed Aug 17, 2024
1 parent f1f3301 commit 7a95bb0
Show file tree
Hide file tree
Showing 4 changed files with 23 additions and 21 deletions.
14 changes: 7 additions & 7 deletions der/src/reader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -23,16 +23,16 @@ pub trait Reader<'r>: Sized {
/// Get the length of the input.
fn input_len(&self) -> Length;

/// Peek at the next byte of input without modifying the position within the reader.
fn peek_byte(&mut self) -> Option<u8>;
/// Peek at the next byte of input without modifying the cursor.
fn peek_byte(&self) -> Option<u8>;

/// Peek forward in the input data, attempting to decode a [`Header`] from
/// the data at the current position in the decoder.
///
/// Does not modify the position within the reader.
fn peek_header(&mut self) -> Result<Header, Error>;
/// Does not modify the decoder's state.
fn peek_header(&self) -> Result<Header, Error>;

/// Get the position within the reader in bytes.
/// Get the position within the buffer.
fn position(&self) -> Length;

/// Attempt to read data borrowed directly from the input as a slice,
Expand Down Expand Up @@ -103,8 +103,8 @@ pub trait Reader<'r>: Sized {
/// Peek at the next byte in the decoder and attempt to decode it as a
/// [`Tag`] value.
///
/// Does not modify the position within the reader.
fn peek_tag(&mut self) -> Result<Tag, Error> {
/// Does not modify the decoder's state.
fn peek_tag(&self) -> Result<Tag, Error> {
match self.peek_byte() {
Some(byte) => byte.try_into(),
None => Err(Error::incomplete(self.input_len())),
Expand Down
4 changes: 2 additions & 2 deletions der/src/reader/nested.rs
Original file line number Diff line number Diff line change
Expand Up @@ -59,15 +59,15 @@ impl<'i, 'r, R: Reader<'r>> Reader<'r> for NestedReader<'i, R> {
self.input_len
}

fn peek_byte(&mut self) -> Option<u8> {
fn peek_byte(&self) -> Option<u8> {
if self.is_finished() {
None
} else {
self.inner.peek_byte()
}
}

fn peek_header(&mut self) -> Result<Header> {
fn peek_header(&self) -> Result<Header> {
if self.is_finished() {
Err(Error::incomplete(self.offset()))
} else {
Expand Down
18 changes: 10 additions & 8 deletions der/src/reader/pem.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
use super::Reader;
use crate::{Decode, EncodingRules, Error, ErrorKind, Header, Length, Result};
use core::cell::RefCell;

#[allow(clippy::arithmetic_side_effects)]
mod utils {
Expand Down Expand Up @@ -125,7 +126,7 @@ mod utils {
#[derive(Clone)]
pub struct PemReader<'i> {
/// Inner PEM decoder wrapped in a BufReader.
reader: utils::BufReader<'i>,
reader: RefCell<utils::BufReader<'i>>,

/// Encoding rules to apply when decoding the input.
encoding_rules: EncodingRules,
Expand All @@ -147,7 +148,7 @@ impl<'i> PemReader<'i> {
let input_len = Length::try_from(reader.remaining_len())?;

Ok(Self {
reader,
reader: RefCell::new(reader),
encoding_rules: EncodingRules::default(),
input_len,
position: Length::ZERO,
Expand All @@ -157,7 +158,7 @@ impl<'i> PemReader<'i> {
/// Get the PEM label which will be used in the encapsulation boundaries
/// for this document.
pub fn type_label(&self) -> &'i str {
self.reader.type_label()
self.reader.borrow().type_label()
}
}

Expand All @@ -171,15 +172,15 @@ impl<'i> Reader<'i> for PemReader<'i> {
self.input_len
}

fn peek_byte(&mut self) -> Option<u8> {
fn peek_byte(&self) -> Option<u8> {
if self.is_finished() {
None
} else {
self.reader.peek_byte()
self.reader.borrow().peek_byte()
}
}

fn peek_header(&mut self) -> Result<Header> {
fn peek_header(&self) -> Result<Header> {
if self.is_finished() {
Err(Error::incomplete(self.offset()))
} else {
Expand All @@ -197,12 +198,13 @@ impl<'i> Reader<'i> for PemReader<'i> {
}

fn read_into<'o>(&mut self, buf: &'o mut [u8]) -> Result<&'o [u8]> {
let bytes = self.reader.copy_to_slice(buf)?;
let bytes = self.reader.borrow_mut().copy_to_slice(buf)?;

self.position = (self.position + bytes.len())?;

debug_assert_eq!(
self.position,
(self.input_len - Length::try_from(self.reader.remaining_len())?)?
(self.input_len - Length::try_from(self.reader.borrow().remaining_len())?)?
);

Ok(bytes)
Expand Down
8 changes: 4 additions & 4 deletions der/src/reader/slice.rs
Original file line number Diff line number Diff line change
Expand Up @@ -77,13 +77,13 @@ impl<'a> Reader<'a> for SliceReader<'a> {
self.bytes.len()
}

fn peek_byte(&mut self) -> Option<u8> {
fn peek_byte(&self) -> Option<u8> {
self.remaining()
.ok()
.and_then(|bytes| bytes.first().cloned())
}

fn peek_header(&mut self) -> Result<Header, Error> {
fn peek_header(&self) -> Result<Header, Error> {
Header::decode(&mut self.clone())
}

Expand Down Expand Up @@ -212,15 +212,15 @@ mod tests {

#[test]
fn peek_tag() {
let mut reader = SliceReader::new(EXAMPLE_MSG).unwrap();
let reader = SliceReader::new(EXAMPLE_MSG).unwrap();
assert_eq!(reader.position(), Length::ZERO);
assert_eq!(reader.peek_tag().unwrap(), Tag::Integer);
assert_eq!(reader.position(), Length::ZERO); // Position unchanged
}

#[test]
fn peek_header() {
let mut reader = SliceReader::new(EXAMPLE_MSG).unwrap();
let reader = SliceReader::new(EXAMPLE_MSG).unwrap();
assert_eq!(reader.position(), Length::ZERO);

let header = reader.peek_header().unwrap();
Expand Down

0 comments on commit 7a95bb0

Please sign in to comment.