robin/bitbuffer
1
0
Fork 0
mirror of https://codeberg.org/icewind/bitbuffer.git synced 2026-06-03 16:44:06 +02:00
Code Issues Releases Wiki Activity

fix read_bytes in BE mode

Browse source
This commit is contained in:
Robin Appelman 2019-02-28 22:18:38 +01:00
commit ea2640655b
2 changed files with 26 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/buffer.rs
View file

@ -275,15 +275,23 @@ where
///
/// [`ReadError::NotEnoughData`]: enum.ReadError.html#variant.NotEnoughData
pub fn read_bytes(&self, position: usize, byte_count: usize) -> Result<Vec<u8>> {
let mut data = vec![];
data.reserve_exact(byte_count);
let mut data = Vec::with_capacity(byte_count);
let mut byte_left = byte_count;
let max_read = size_of::<usize>() - 1;
let mut read_pos = position;
while byte_left > 0 {
let read = min(byte_left, max_read);
let bytes: [u8; USIZE_SIZE] = self.read_usize(read_pos, read * 8)?.to_le_bytes();
let usable_bytes = &bytes[0..read];
let raw_bytes = self.read_usize(read_pos, read * 8)?;
let bytes: [u8; USIZE_SIZE] = if E::is_le() {
raw_bytes.to_le_bytes()
} else {
raw_bytes.to_be_bytes()
};
let usable_bytes = if E::is_le() {
&bytes[0..read]
} else {
&bytes[8-read..8]
};
data.extend_from_slice(usable_bytes);
byte_left -= read;
read_pos += read * 8;