summaryrefslogtreecommitdiffstats
path: root/third_party/rust/digest/src/dev/fixed.rs
blob: 24f380112d8b79a419afbcd3e2e2cdddab6a31f7 (plain)
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
use crate::{Digest, FixedOutput, FixedOutputReset, HashMarker, Update};
use core::fmt::Debug;

/// Fixed-output resettable digest test via the `Digest` trait
pub fn fixed_reset_test<D>(input: &[u8], output: &[u8]) -> Option<&'static str>
where
    D: FixedOutputReset + Debug + Clone + Default + Update + HashMarker,
{
    let mut hasher = D::new();
    // Test that it works when accepting the message all at once
    hasher.update(input);
    let mut hasher2 = hasher.clone();
    if hasher.finalize()[..] != output[..] {
        return Some("whole message");
    }

    // Test if reset works correctly
    hasher2.reset();
    hasher2.update(input);
    if hasher2.finalize_reset()[..] != output[..] {
        return Some("whole message after reset");
    }

    // Test that it works when accepting the message in chunks
    for n in 1..core::cmp::min(17, input.len()) {
        let mut hasher = D::new();
        for chunk in input.chunks(n) {
            hasher.update(chunk);
            hasher2.update(chunk);
        }
        if hasher.finalize()[..] != output[..] {
            return Some("message in chunks");
        }
        if hasher2.finalize_reset()[..] != output[..] {
            return Some("message in chunks");
        }
    }

    None
}

/// Variable-output resettable digest test
pub fn fixed_test<D>(input: &[u8], output: &[u8]) -> Option<&'static str>
where
    D: FixedOutput + Default + Debug + Clone,
{
    let mut hasher = D::default();
    // Test that it works when accepting the message all at once
    hasher.update(input);
    if hasher.finalize_fixed()[..] != output[..] {
        return Some("whole message");
    }

    // Test that it works when accepting the message in chunks
    for n in 1..core::cmp::min(17, input.len()) {
        let mut hasher = D::default();
        for chunk in input.chunks(n) {
            hasher.update(chunk);
        }
        if hasher.finalize_fixed()[..] != output[..] {
            return Some("message in chunks");
        }
    }
    None
}