aes 0.9.4

Pure Rust implementation of the Advanced Encryption Standard (a.k.a. Rijndael)
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

use crate::Block;
use cipher::{Array, array::ArraySize};

#[cfg(target_arch = "x86")]
use core::arch::x86::*;
#[cfg(target_arch = "x86_64")]
use core::arch::x86_64::*;

#[inline]
#[target_feature(enable = "sse2")]
pub(super) fn load_block(block: &Block) -> __m128i {
    let p: *const __m128i = block.as_ptr().cast();
    // SAFETY: sizes of `Block` and `__m128i` are equal and we use unaligned load instruction
    unsafe { _mm_loadu_si128(p) }
}

#[inline]
#[target_feature(enable = "sse2")]
pub(super) fn store_block(dst: &mut Block, block: __m128i) {
    let p: *mut __m128i = dst.as_mut_ptr().cast();
    // SAFETY: sizes of `Block` and `__m128i` are equal and we use unaligned store instruction
    unsafe { _mm_storeu_si128(p, block) }
}

#[inline]
#[target_feature(enable = "sse2")]
pub(super) fn load_batch_blocks<N: ArraySize>(blocks: &Array<Block, N>) -> Array<__m128i, N> {
    Array::from_fn(|i| load_block(&blocks[i]))
}

#[inline]
#[target_feature(enable = "sse2")]
pub(super) fn store_batch_blocks<N: ArraySize>(
    dst: &mut Array<Block, N>,
    blocks: Array<__m128i, N>,
) {
    for i in 0..N::USIZE {
        store_block(&mut dst[i], blocks[i]);
    }
}