binary_codec/bitstream/
mod.rs

1pub mod reader;
2pub mod writer;
3
4/// Crypto Stream
5pub trait CryptoStream {
6    /// Applies the keystream byte to the given byte and returns the result.
7    fn apply_keystream_byte(&mut self, b: u8) -> u8;
8    /// Applies the keystream to the given slice and returns the result.
9    fn apply_keystream(&mut self, slice: &[u8]) -> &[u8];
10    /// Get the underlying original/encrypted buffer
11    /// `original` indicates whether to return the original (unencrypted) buffer or the encrypted buffer.
12    fn get_cached(&self, original: bool) -> &[u8];
13}
14
15#[cfg(test)]
16mod tests {
17    use crate::bitstream::{reader::BitStreamReader, writer::BitStreamWriter};
18
19    #[test]
20    fn can_serialize_and_deserialize_using_bitstream() {
21        let mut buffer = Vec::new();
22        let mut writer = BitStreamWriter::new(&mut buffer);
23
24        writer.write_bit(true);
25        writer.write_bit(false);
26        writer.write_bit(true);
27        writer.write_byte(12);
28        writer.write_bytes(&[34, 56]);
29
30        assert_eq!(writer.byte_pos(), 4);
31
32        let mut reader = BitStreamReader::new(&buffer);
33        assert_eq!(reader.read_bit().unwrap(), true);
34        assert_eq!(reader.read_bit().unwrap(), false);
35        assert_eq!(reader.read_bit().unwrap(), true);
36        assert_eq!(reader.read_byte().unwrap(), 12);
37        assert_eq!(reader.read_byte().unwrap(), 34);
38        assert_eq!(reader.read_byte().unwrap(), 56);
39
40        assert_eq!(buffer, [0b00000101, 12, 34, 56]);
41    }
42}
binary_codec/bitstream/mod.rs

binary_codec/bitstream/
mod.rs
