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

// Copyright 2024 Ulvetanna Inc.

use super::{
	packed::{impl_broadcast, PackedPrimitiveType},
	packed_arithmetic::impl_tower_constants,
};
use crate::{
	arch::{PackedStrategy, PairwiseRecursiveStrategy, PairwiseStrategy, PairwiseTableStrategy},
	arithmetic_traits::{
		impl_invert_with, impl_mul_alpha_with, impl_mul_with, impl_square_with,
		impl_transformation_with_strategy,
	},
	AESTowerField16b, AESTowerField8b,
};

// Define 16 bit packed field types
pub type PackedAESBinaryField2x8b = PackedPrimitiveType<u16, AESTowerField8b>;
pub type PackedAESBinaryField1x16b = PackedPrimitiveType<u16, AESTowerField16b>;

// Define broadcast
impl_broadcast!(u16, AESTowerField8b);
impl_broadcast!(u16, AESTowerField16b);

// Define constants
impl_tower_constants!(AESTowerField8b, u16, 0x00d3);

// Define multiplication
impl_mul_with!(PackedAESBinaryField2x8b @ PairwiseTableStrategy);
impl_mul_with!(PackedAESBinaryField1x16b @ PairwiseRecursiveStrategy);

// Define square
impl_square_with!(PackedAESBinaryField2x8b @ PairwiseTableStrategy);
impl_square_with!(PackedAESBinaryField1x16b @ PairwiseRecursiveStrategy);

// Define invert
impl_invert_with!(PackedAESBinaryField2x8b @ PairwiseTableStrategy);
impl_invert_with!(PackedAESBinaryField1x16b @ PairwiseRecursiveStrategy);

// Define multiply by alpha
impl_mul_alpha_with!(PackedAESBinaryField2x8b @ PairwiseTableStrategy);
impl_mul_alpha_with!(PackedAESBinaryField1x16b @ PackedStrategy);

// Define linear transformations
impl_transformation_with_strategy!(PackedAESBinaryField2x8b, PackedStrategy);
impl_transformation_with_strategy!(PackedAESBinaryField1x16b, PairwiseStrategy);