highway/
internal.rs

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
#[cfg(any(
    target_arch = "x86_64",
    target_arch = "aarch64",
    all(target_family = "wasm", target_feature = "simd128")
))]
pub fn unordered_load3(from: &[u8]) -> u64 {
    if from.is_empty() {
        return 0;
    }

    let size_mod4 = from.len() % 4;

    u64::from(from[0])
        + (u64::from(from[size_mod4 >> 1]) << 8)
        + (u64::from(from[size_mod4 - 1]) << 16)
}

pub const PACKET_SIZE: usize = 32;

/// The c layout is needed as we'll be interpretting the buffer as different types and passing it
/// to simd instructions, so we need to subscribe to the whole "do what C does", else we will
/// segfault.
#[repr(C)]
#[derive(Default, Debug, Clone, Copy)]
pub struct HashPacket {
    pub(crate) buf: [u8; PACKET_SIZE],
    buf_index: usize,
}

impl HashPacket {
    #[inline]
    pub const fn len(&self) -> usize {
        self.buf_index
    }

    #[inline]
    pub const fn is_empty(&self) -> bool {
        self.buf_index == 0
    }

    #[inline]
    pub fn as_slice(&self) -> &[u8] {
        debug_assert!(self.buf_index <= self.buf.len(), "buf index too long");
        self.buf.get(..self.buf_index).unwrap_or(&self.buf)
    }

    #[inline]
    pub const fn inner(&self) -> &[u8; PACKET_SIZE] {
        &self.buf
    }

    #[inline]
    pub fn fill<'a>(&mut self, data: &'a [u8]) -> Option<&'a [u8]> {
        let dest = self.buf.get_mut(self.buf_index..).unwrap_or_default();
        if dest.len() > data.len() {
            dest[..data.len()].copy_from_slice(data);
            self.buf_index += data.len();
            None
        } else {
            let (head, tail) = data.split_at(dest.len());
            dest.copy_from_slice(head);
            self.buf_index = PACKET_SIZE;
            Some(tail)
        }
    }

    #[inline]
    pub fn set_to(&mut self, data: &[u8]) {
        debug_assert!(
            data.len() < PACKET_SIZE,
            "data large enough to process packet"
        );
        self.buf_index = data.len();
        if !data.is_empty() {
            self.buf[..data.len()].copy_from_slice(data);
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_hash_packet() {
        let mut packet: HashPacket = Default::default();
        for i in 0..31 {
            assert_eq!(&vec![0; i as usize][..], packet.as_slice());
            if let Some(_) = packet.fill(&[0]) {
                assert!(false);
            }

            assert_eq!(i + 1, packet.len() as u8);
            assert_eq!(&vec![0; (i + 1) as usize][..], packet.as_slice());
        }
    }

    #[test]
    fn test_hash_cusp_full_packet() {
        let mut packet: HashPacket = Default::default();
        assert_eq!(Some(&[][..]), packet.fill(&[0; 32]));
        assert_eq!(32, packet.len());
    }

    #[test]
    fn test_hash_packet_set_to() {
        let mut packet: HashPacket = Default::default();
        for i in 0..31 {
            let d = vec![0; i as usize];
            packet.set_to(&d[..]);
            assert_eq!(&d[..], packet.as_slice());
            assert_eq!(d.len(), packet.len());
        }
    }
}