craft-codec 0.2.2

Cryptographic Random-Access Framing Toolkit
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

//! Roundtrips through all framing and transform combinations.
use craft_codec::{Compression, Config, Decoder, Encoder, Encryption, Framing, Metadata};

fn data(len: usize, mut seed: u64) -> Vec<u8> {
    (0..len)
        .map(|_| {
            seed ^= seed << 13;
            seed ^= seed >> 7;
            seed ^= seed << 17;
            seed as u8
        })
        .collect()
}

fn exercise(compression: Compression, encryption: Encryption) {
    for framing in [Framing::Fixed(256), Framing::Variable(256)] {
        let lengths = match framing {
            Framing::Fixed(_) => [256, 256, 256, 17],
            _ => [17, 128, 256, 13],
        };
        let frames: Vec<Vec<u8>> = lengths
            .into_iter()
            .enumerate()
            .map(|(i, n)| {
                if i % 2 == 0 {
                    vec![b'A' + i as u8; n]
                } else {
                    data(n, i as u64 + 1)
                }
            })
            .collect();
        let original = frames.concat();
        let mut metadata =
            Metadata::new(Config::new(framing, compression, encryption.clone()).unwrap());
        let mut encoder = Encoder::try_from(&metadata).unwrap();
        let mut stored = Vec::new();
        for (i, frame) in frames.into_iter().enumerate() {
            let mut buffer = frame;
            let sizes = encoder.encode_frame(i as u64, &mut buffer).unwrap();
            stored.extend_from_slice(&buffer);
            metadata.push(sizes).unwrap();
        }
        let metadata = Metadata::from_parts(metadata.into_parts()).unwrap();
        assert_eq!(stored.len() as u64, metadata.stored_len());
        let mut decoder = Decoder::try_from(&metadata).unwrap();
        let len = original.len();
        for (start, end) in [
            (0, len),
            (1, len - 1),
            (16, 18),
            (17, 17),
            (3, 9),
            (len - 1, len),
            (len, len),
        ] {
            let (query, frames) = metadata.range(start as u64, Some(end as u64)).unwrap();
            let mut offset = query.start as usize;
            let mut result = Vec::new();
            for frame in frames {
                let end = offset + frame.spec.stored_len();
                let mut buffer = stored[offset..end].to_vec();
                offset = end;
                decoder.decode_frame(frame.spec, &mut buffer).unwrap();
                result.extend_from_slice(&buffer[frame.selected]);
            }
            assert_eq!(offset as u64, query.end);
            assert_eq!(result, original[start..end]);
        }
    }
}

#[test]
fn raw_fixed_and_variable() {
    exercise(Compression::None, Encryption::None);
}

#[cfg(feature = "aes-gcm")]
#[test]
fn aes_fixed_and_variable() {
    exercise(Compression::None, Encryption::Aes256Gcm { key: [17; 32] });
}

#[cfg(feature = "lz4")]
#[test]
fn compression_fixed_and_variable() {
    exercise(Compression::Lz4, Encryption::None);
}

#[cfg(all(feature = "aes-gcm", feature = "lz4"))]
#[test]
fn compression_and_aes_fixed_and_variable() {
    exercise(Compression::Lz4, Encryption::Aes256Gcm { key: [23; 32] });
}

#[cfg(feature = "lz4")]
#[test]
fn compression_fallback_and_reused_contexts_are_independent() {
    let mut metadata = Metadata::new(
        Config::new(Framing::Variable(4096), Compression::Lz4, Encryption::None).unwrap(),
    );
    let mut encoder = Encoder::try_from(&metadata).unwrap();
    let mut records = Vec::new();
    for (i, raw) in [vec![0; 4096], vec![42], data(4096, 42), vec![3; 4096]]
        .into_iter()
        .enumerate()
    {
        let mut buffer = raw.clone();
        let sizes = encoder.encode_frame(i as u64, &mut buffer).unwrap();
        if i == 0 || i == 3 {
            assert!(sizes.payload_len() < sizes.raw_len());
        }
        if i == 1 || i == 2 {
            assert_eq!(sizes.payload_len(), sizes.raw_len());
        }
        metadata.push(sizes).unwrap();
        records.push((raw, buffer));
    }
    for (i, (raw, mut buffer)) in records.into_iter().enumerate().rev() {
        let mut decoder = Decoder::try_from(&metadata).unwrap();
        decoder
            .decode_frame(metadata.frame(i as u64).unwrap(), &mut buffer)
            .unwrap();
        assert_eq!(buffer, raw);
    }
}