aws_lc_rs/cipher/
block.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
// Copyright 2018 Brian Smith.
// SPDX-License-Identifier: ISC
// Modifications copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0 OR ISC

/// An array of 16 bytes that can (in the `x86_64` and `AAarch64` ABIs, at least)
/// be efficiently passed by value and returned by value (i.e. in registers),
/// and which meets the alignment requirements of `u32` and `u64` (at least)
/// for the target.
#[repr(C)]
#[derive(Copy, Clone)]
pub(crate) struct Block {
    subblocks: [u64; 2],
}

/// Block length
pub(crate) const BLOCK_LEN: usize = 16;

impl Block {
    #[inline]
    pub(crate) fn zero() -> Self {
        Self { subblocks: [0, 0] }
    }
}

impl From<[u8; BLOCK_LEN]> for Block {
    #[inline]
    fn from(bytes: [u8; BLOCK_LEN]) -> Self {
        unsafe { core::mem::transmute(bytes) }
    }
}

impl AsRef<[u8; BLOCK_LEN]> for Block {
    #[allow(clippy::transmute_ptr_to_ptr)]
    #[inline]
    fn as_ref(&self) -> &[u8; BLOCK_LEN] {
        unsafe { core::mem::transmute(self) }
    }
}

impl AsMut<[u8; BLOCK_LEN]> for Block {
    #[allow(clippy::transmute_ptr_to_ptr)]
    #[inline]
    fn as_mut(&mut self) -> &mut [u8; BLOCK_LEN] {
        unsafe { core::mem::transmute(self) }
    }
}

#[cfg(test)]
mod tests {
    #[test]
    fn test_block_clone() {
        use super::{Block, BLOCK_LEN};
        let block_a = Block::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]);
        #[allow(clippy::clone_on_copy)]
        let block_b = block_a.clone();

        for i in 0..BLOCK_LEN {
            assert_eq!(block_a.as_ref()[i], block_b.as_ref()[i]);
        }
    }

    #[test]
    fn test_block_clone_mut_ref() {
        use super::{Block, BLOCK_LEN};
        let mut block_a = Block::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]);
        #[allow(clippy::clone_on_copy)]
        let mut block_b = block_a.clone();

        for i in 0..BLOCK_LEN {
            assert_eq!(block_a.as_mut()[i], block_b.as_mut()[i]);
        }
    }
}