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 reserve not setting bit len correctly

Browse source
This commit is contained in:
Robin Appelman 2021-07-16 21:04:07 +02:00
commit 9587c7b872
3 changed files with 117 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/roundtrip.rs
View file

@ -15,18 +15,22 @@ fn roundtrip<
) {
{
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write(&val).unwrap();
let size = stream.bit_len();
let size = {
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write(&val).unwrap();
stream.bit_len()
};
let mut read = BitReadStream::new(BitReadBuffer::new_owned(data, LittleEndian));
assert_eq!(val, read.read().unwrap());
assert_eq!(size, read.pos());
}
{
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, BigEndian);
stream.write(&val).unwrap();
let size = stream.bit_len();
let size = {
let mut stream = BitWriteStream::new(&mut data, BigEndian);
stream.write(&val).unwrap();
stream.bit_len()
};
let mut read = BitReadStream::new(BitReadBuffer::new_owned(data, BigEndian));
assert_eq!(val, read.read().unwrap());
assert_eq!(size, read.pos());

92
tests/write_tests.rs
View file

@ -3,12 +3,14 @@ use bitbuffer::{BigEndian, BitReadBuffer, BitReadStream, BitWriteStream, LittleE
#[test]
fn test_write_bool_le() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
{
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write_bool(true).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(false).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(false).unwrap();
stream.write_bool(true).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, LittleEndian));
@ -24,12 +26,14 @@ fn test_write_bool_le() {
#[test]
fn test_write_bool_be() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, BigEndian);
{
let mut stream = BitWriteStream::new(&mut data, BigEndian);
stream.write_bool(true).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(false).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(true).unwrap();
stream.write_bool(false).unwrap();
stream.write_bool(true).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, BigEndian));
@ -45,11 +49,13 @@ fn test_write_bool_be() {
#[test]
fn test_write_bool_number_le() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
{
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write_bool(true).unwrap();
stream.write_int(3253u16, 16).unwrap();
stream.write_int(13253u64, 64).unwrap();
stream.write_bool(true).unwrap();
stream.write_int(3253u16, 16).unwrap();
stream.write_int(13253u64, 64).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, LittleEndian));
@ -64,11 +70,13 @@ fn test_write_bool_number_le() {
#[test]
fn test_write_bool_number_be() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, BigEndian);
{
let mut stream = BitWriteStream::new(&mut data, BigEndian);
stream.write_bool(true).unwrap();
stream.write_int(3253u16, 16).unwrap();
stream.write_int(13253u64, 64).unwrap();
stream.write_bool(true).unwrap();
stream.write_int(3253u16, 16).unwrap();
stream.write_int(13253u64, 64).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, BigEndian));
@ -83,10 +91,12 @@ fn test_write_bool_number_be() {
#[test]
fn test_write_float_le() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
{
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write_bool(true).unwrap();
stream.write_float(3253.12f32).unwrap();
stream.write_bool(true).unwrap();
stream.write_float(3253.12f32).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, LittleEndian));
@ -100,10 +110,12 @@ fn test_write_float_le() {
#[test]
fn test_write_float_be() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, BigEndian);
{
let mut stream = BitWriteStream::new(&mut data, BigEndian);
stream.write_bool(true).unwrap();
stream.write_float(3253.12f32).unwrap();
stream.write_bool(true).unwrap();
stream.write_float(3253.12f32).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, BigEndian));
@ -117,11 +129,13 @@ fn test_write_float_be() {
#[test]
fn test_write_string_le() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
{
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write_string("null terminated", None).unwrap();
stream.write_string("fixed length1", Some(16)).unwrap();
stream.write_string("fixed length2", Some(16)).unwrap();
stream.write_string("null terminated", None).unwrap();
stream.write_string("fixed length1", Some(16)).unwrap();
stream.write_string("fixed length2", Some(16)).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, LittleEndian));
@ -133,12 +147,14 @@ fn test_write_string_le() {
#[test]
fn test_write_string_le_unaligned() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
{
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write_bool(true).unwrap();
stream.write_string("null terminated", None).unwrap();
stream.write_string("fixed length1", Some(16)).unwrap();
stream.write_string("fixed length2", Some(16)).unwrap();
stream.write_bool(true).unwrap();
stream.write_string("null terminated", None).unwrap();
stream.write_string("fixed length1", Some(16)).unwrap();
stream.write_string("fixed length2", Some(16)).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, LittleEndian));
@ -154,11 +170,13 @@ fn test_write_string_le_unaligned() {
#[test]
fn test_write_signed() {
let mut data = Vec::new();
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
{
let mut stream = BitWriteStream::new(&mut data, LittleEndian);
stream.write_bool(true).unwrap();
stream.write_int(-17i32, 32).unwrap();
stream.write_int(-9i32, 8).unwrap();
stream.write_bool(true).unwrap();
stream.write_int(-17i32, 32).unwrap();
stream.write_int(-9i32, 8).unwrap();
}
let mut read = BitReadStream::from(BitReadBuffer::new(&data, LittleEndian));