bcur 1.0.0

A Rust implementation of Uniform Resources
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

#![allow(
    unused_crate_dependencies,
    clippy::tests_outside_test_module,
    clippy::unwrap_used,
    clippy::expect_used,
    clippy::excessive_nesting,
    reason = "integration targets link full dev-deps and host unwraps by design"
)]

//! Adversarial multi-part decoder session behavior (public API).

use bcur::{Decoder, DecoderLimits, Encoder, Error, ResourceKind, UrType};

#[test]
fn uri_len_limit_poisons_session() {
    let data = b"Ten chars!".repeat(8);
    let mut enc = Encoder::bytes(&data, 5).unwrap();
    let part = enc.next_part().unwrap();

    let mut decoder = Decoder::with_limits(DecoderLimits {
        max_uri_len: 16,
        ..DecoderLimits::default()
    });
    assert!(matches!(
        decoder.receive(&part),
        Err(Error::ResourceLimit(ResourceKind::UriLen))
    ));
    assert!(decoder.is_poisoned());
    assert!(matches!(
        decoder.receive(&part),
        Err(Error::ResourceLimit(ResourceKind::UriLen))
    ));
    assert!(matches!(
        decoder.message(),
        Err(Error::ResourceLimit(ResourceKind::UriLen))
    ));
}

#[test]
fn fragment_count_limit_poisons() {
    let data = b"Ten chars!".repeat(16);
    let mut enc = Encoder::bytes(&data, 4).unwrap();
    assert!(enc.fragment_count() > 1);

    let mut decoder = Decoder::with_limits(DecoderLimits {
        max_fragment_count: 1,
        ..DecoderLimits::default()
    });
    let part = enc.next_part().unwrap();
    assert!(matches!(
        decoder.receive(&part),
        Err(Error::ResourceLimit(ResourceKind::FragmentCount))
    ));
    assert!(decoder.is_poisoned());
}

#[test]
fn message_length_limit_poisons() {
    let data = b"Ten chars!".repeat(16);
    let mut enc = Encoder::bytes(&data, 8).unwrap();
    let mut decoder = Decoder::with_limits(DecoderLimits {
        max_message_length: 8,
        ..DecoderLimits::default()
    });
    assert!(matches!(
        decoder.receive(&enc.next_part().unwrap()),
        Err(Error::ResourceLimit(ResourceKind::MessageLength))
    ));
    assert!(decoder.is_poisoned());
}

#[test]
fn type_stickiness_does_not_poison() {
    let data = b"Ten chars!".repeat(6);
    let mut a = Encoder::new(&data, 5, &UrType::new("alpha").unwrap()).unwrap();
    let mut b = Encoder::new(&data, 5, &UrType::new("beta").unwrap()).unwrap();
    let mut decoder = Decoder::default();
    decoder.receive(&a.next_part().unwrap()).unwrap();
    assert!(matches!(
        decoder.receive(&b.next_part().unwrap()),
        Err(Error::UnexpectedType { .. })
    ));
    assert!(!decoder.is_poisoned());
    // Same type continues to be accepted.
    decoder.receive(&a.next_part().unwrap()).unwrap();
}

#[test]
fn single_part_receive_does_not_poison() {
    let mut decoder = Decoder::default();
    decoder.receive("ur:bytes/iehsjyhspmwfwfia").unwrap();
    assert!(decoder.complete());
    assert!(!decoder.is_poisoned());
    assert_eq!(
        decoder.message().unwrap().as_deref(),
        Some(b"data".as_slice())
    );
}

#[test]
fn expected_type_mismatch_does_not_poison() {
    let data = b"Ten chars!".repeat(4);
    let mut enc = Encoder::new(&data, 5, &UrType::new("alpha").unwrap()).unwrap();
    let mut decoder = Decoder::default().with_expected_type(UrType::new("beta").unwrap());
    assert!(matches!(
        decoder.receive(&enc.next_part().unwrap()),
        Err(Error::UnexpectedType { .. })
    ));
    assert!(!decoder.is_poisoned());
}

#[test]
fn index_path_mismatch_does_not_poison() {
    let data = b"Ten chars!".repeat(4);
    let mut enc = Encoder::bytes(&data, 5).unwrap();
    let part = enc.next_part().unwrap();
    let corrupted = part.replacen("/1-", "/2-", 1);
    let mut decoder = Decoder::default();
    assert!(matches!(
        decoder.receive(&corrupted),
        Err(Error::InvalidIndices)
    ));
    assert!(!decoder.is_poisoned());
}