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

Check endianess when unconditionally reading boolean values

Browse source
This commit is contained in:
Joseph Angelo 2022-04-30 08:44:30 -07:00
commit 1e575b8efc
2 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/readbuffer.rs
View file

@ -314,8 +314,13 @@ where
let bit_offset = position & 7; let bit_offset = position & 7;
let byte = self.slice.get_unchecked(byte_index); let byte = self.slice.get_unchecked(byte_index);
if E::is_le() {
let shifted = byte >> bit_offset; let shifted = byte >> bit_offset;
shifted & 1u8 == 1 shifted & 1u8 == 1
} else {
let shifted = byte << bit_offset;
shifted & 0b1000_0000u8 == 0b1000_0000u8
}
} }
/// Read a sequence of bits from the buffer as integer /// Read a sequence of bits from the buffer as integer

9
tests/roundtrip.rs
View file

@ -82,6 +82,13 @@ fn test_bare_enum() {
#[test] #[test]
fn test_field_enum() { fn test_field_enum() {
#[derive(Debug, PartialEq, BitRead, BitWrite)]
struct CompoundVariant(
#[size = 15]
u16,
bool,
);
#[derive(Debug, PartialEq, BitRead, BitWrite)] #[derive(Debug, PartialEq, BitRead, BitWrite)]
#[discriminant_bits = 4] #[discriminant_bits = 4]
enum Enum { enum Enum {
@ -89,11 +96,13 @@ fn test_field_enum() {
B(String), B(String),
C(f32), C(f32),
D(#[size = 15] i64), D(#[size = 15] i64),
E(CompoundVariant),
} }
roundtrip(Enum::A); roundtrip(Enum::A);
roundtrip(Enum::B("foobar".into())); roundtrip(Enum::B("foobar".into()));
roundtrip(Enum::C(12.0)); roundtrip(Enum::C(12.0));
roundtrip(Enum::D(-12345)); roundtrip(Enum::D(-12345));
roundtrip(Enum::E(CompoundVariant(6789, true)));
} }
#[test] #[test]