use rand::Rng;
use rand::distr::{Distribution, StandardUniform};
use std::ops::{Add, AddAssign, Div, Mul, Neg, Sub};
pub const GF256_ORDER: usize = u8::MAX as usize + 1;
#[cfg(any(target_arch = "x86", target_arch = "x86_64", target_arch = "aarch64"))]
pub const GF256_BIT_WIDTH: usize = u8::BITS as usize;
#[cfg(any(target_arch = "x86", target_arch = "x86_64", target_arch = "aarch64"))]
pub const GF256_HALF_ORDER: usize = 1usize << (GF256_BIT_WIDTH / 2);
const GF256_LOG_TABLE: [u8; GF256_ORDER] = [
0, 0, 25, 1, 50, 2, 26, 198, 75, 199, 27, 104, 51, 238, 223, 3, 100, 4, 224, 14, 52, 141, 129, 239, 76, 113, 8, 200, 248, 105, 28, 193, 125, 194, 29, 181,
249, 185, 39, 106, 77, 228, 166, 114, 154, 201, 9, 120, 101, 47, 138, 5, 33, 15, 225, 36, 18, 240, 130, 69, 53, 147, 218, 142, 150, 143, 219, 189, 54, 208,
206, 148, 19, 92, 210, 241, 64, 70, 131, 56, 102, 221, 253, 48, 191, 6, 139, 98, 179, 37, 226, 152, 34, 136, 145, 16, 126, 110, 72, 195, 163, 182, 30, 66,
58, 107, 40, 84, 250, 133, 61, 186, 43, 121, 10, 21, 155, 159, 94, 202, 78, 212, 172, 229, 243, 115, 167, 87, 175, 88, 168, 80, 244, 234, 214, 116, 79,
174, 233, 213, 231, 230, 173, 232, 44, 215, 117, 122, 235, 22, 11, 245, 89, 203, 95, 176, 156, 169, 81, 160, 127, 12, 246, 111, 23, 196, 73, 236, 216, 67,
31, 45, 164, 118, 123, 183, 204, 187, 62, 90, 251, 96, 177, 134, 59, 82, 161, 108, 170, 85, 41, 157, 151, 178, 135, 144, 97, 190, 220, 252, 188, 149, 207,
205, 55, 63, 91, 209, 83, 57, 132, 60, 65, 162, 109, 71, 20, 42, 158, 93, 86, 242, 211, 171, 68, 17, 146, 217, 35, 32, 46, 137, 180, 124, 184, 38, 119,
153, 227, 165, 103, 74, 237, 222, 197, 49, 254, 24, 13, 99, 140, 128, 192, 247, 112, 7,
];
const GF256_EXP_TABLE: [u8; 2 * GF256_ORDER - 2] = [
1, 3, 5, 15, 17, 51, 85, 255, 26, 46, 114, 150, 161, 248, 19, 53, 95, 225, 56, 72, 216, 115, 149, 164, 247, 2, 6, 10, 30, 34, 102, 170, 229, 52, 92, 228,
55, 89, 235, 38, 106, 190, 217, 112, 144, 171, 230, 49, 83, 245, 4, 12, 20, 60, 68, 204, 79, 209, 104, 184, 211, 110, 178, 205, 76, 212, 103, 169, 224, 59,
77, 215, 98, 166, 241, 8, 24, 40, 120, 136, 131, 158, 185, 208, 107, 189, 220, 127, 129, 152, 179, 206, 73, 219, 118, 154, 181, 196, 87, 249, 16, 48, 80,
240, 11, 29, 39, 105, 187, 214, 97, 163, 254, 25, 43, 125, 135, 146, 173, 236, 47, 113, 147, 174, 233, 32, 96, 160, 251, 22, 58, 78, 210, 109, 183, 194,
93, 231, 50, 86, 250, 21, 63, 65, 195, 94, 226, 61, 71, 201, 64, 192, 91, 237, 44, 116, 156, 191, 218, 117, 159, 186, 213, 100, 172, 239, 42, 126, 130,
157, 188, 223, 122, 142, 137, 128, 155, 182, 193, 88, 232, 35, 101, 175, 234, 37, 111, 177, 200, 67, 197, 84, 252, 31, 33, 99, 165, 244, 7, 9, 27, 45, 119,
153, 176, 203, 70, 202, 69, 207, 74, 222, 121, 139, 134, 145, 168, 227, 62, 66, 198, 81, 243, 14, 18, 54, 90, 238, 41, 123, 141, 140, 143, 138, 133, 148,
167, 242, 13, 23, 57, 75, 221, 124, 132, 151, 162, 253, 28, 36, 108, 180, 199, 82, 246, 1, 3, 5, 15, 17, 51, 85, 255, 26, 46, 114, 150, 161, 248, 19, 53,
95, 225, 56, 72, 216, 115, 149, 164, 247, 2, 6, 10, 30, 34, 102, 170, 229, 52, 92, 228, 55, 89, 235, 38, 106, 190, 217, 112, 144, 171, 230, 49, 83, 245, 4,
12, 20, 60, 68, 204, 79, 209, 104, 184, 211, 110, 178, 205, 76, 212, 103, 169, 224, 59, 77, 215, 98, 166, 241, 8, 24, 40, 120, 136, 131, 158, 185, 208,
107, 189, 220, 127, 129, 152, 179, 206, 73, 219, 118, 154, 181, 196, 87, 249, 16, 48, 80, 240, 11, 29, 39, 105, 187, 214, 97, 163, 254, 25, 43, 125, 135,
146, 173, 236, 47, 113, 147, 174, 233, 32, 96, 160, 251, 22, 58, 78, 210, 109, 183, 194, 93, 231, 50, 86, 250, 21, 63, 65, 195, 94, 226, 61, 71, 201, 64,
192, 91, 237, 44, 116, 156, 191, 218, 117, 159, 186, 213, 100, 172, 239, 42, 126, 130, 157, 188, 223, 122, 142, 137, 128, 155, 182, 193, 88, 232, 35, 101,
175, 234, 37, 111, 177, 200, 67, 197, 84, 252, 31, 33, 99, 165, 244, 7, 9, 27, 45, 119, 153, 176, 203, 70, 202, 69, 207, 74, 222, 121, 139, 134, 145, 168,
227, 62, 66, 198, 81, 243, 14, 18, 54, 90, 238, 41, 123, 141, 140, 143, 138, 133, 148, 167, 242, 13, 23, 57, 75, 221, 124, 132, 151, 162, 253, 28, 36, 108,
180, 199, 82, 246,
];
#[repr(transparent)]
#[derive(Default, Clone, Copy, Debug)]
pub struct Gf256 {
val: u8,
}
impl Gf256 {
pub const fn new(val: u8) -> Self {
Gf256 { val }
}
pub const fn get(&self) -> u8 {
self.val
}
pub const fn zero() -> Self {
Gf256::new(0)
}
pub const fn one() -> Self {
Gf256::new(1)
}
pub const fn primitive_element() -> Self {
Gf256::new(3)
}
pub const fn mul_const(a: u8, b: u8) -> u8 {
if a == 0 || b == 0 {
return 0;
}
let l = GF256_LOG_TABLE[a as usize] as usize;
let r = GF256_LOG_TABLE[b as usize] as usize;
GF256_EXP_TABLE[l + r]
}
pub const fn inv(self) -> Option<Self> {
if self.val == 0 {
return None;
}
Some(Gf256 {
val: GF256_EXP_TABLE[(GF256_ORDER - 1) - GF256_LOG_TABLE[self.val as usize] as usize],
})
}
}
impl Add for Gf256 {
type Output = Self;
#[allow(clippy::suspicious_arithmetic_impl)]
fn add(self, rhs: Self) -> Self::Output {
Gf256 { val: self.val ^ rhs.val }
}
}
impl AddAssign for Gf256 {
#[allow(clippy::suspicious_op_assign_impl)]
fn add_assign(&mut self, rhs: Self) {
self.val ^= rhs.val;
}
}
impl Neg for Gf256 {
type Output = Self;
fn neg(self) -> Self::Output {
Gf256 { val: self.val }
}
}
impl Sub for Gf256 {
type Output = Self;
#[allow(clippy::suspicious_arithmetic_impl)]
fn sub(self, rhs: Self) -> Self::Output {
Gf256 { val: self.val ^ rhs.val }
}
}
impl Mul for Gf256 {
type Output = Self;
fn mul(self, rhs: Self) -> Self::Output {
Gf256 {
val: Self::mul_const(self.val, rhs.val),
}
}
}
impl Div for Gf256 {
type Output = Option<Self>;
#[allow(clippy::suspicious_arithmetic_impl)]
fn div(self, rhs: Self) -> Self::Output {
rhs.inv().map(|rhs_inv| self * rhs_inv)
}
}
impl PartialEq for Gf256 {
fn eq(&self, other: &Self) -> bool {
self.val == other.val
}
}
impl Distribution<Gf256> for StandardUniform {
fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> Gf256 {
Gf256 { val: rng.random() }
}
}
#[cfg(test)]
mod test {
use super::Gf256;
use rand::Rng;
#[test]
fn prop_test_gf256_operations() {
const NUM_TEST_ITERATIONS: usize = 100_000;
let mut rng = rand::rng();
(0..NUM_TEST_ITERATIONS).for_each(|_| {
let a: Gf256 = rng.random();
let b: Gf256 = rng.random();
let sum = a + b;
let diff = sum - b;
assert_eq!(diff, a);
let mul = a * b;
let div = mul / b;
if b == Gf256::zero() {
assert_eq!(div, None);
assert_eq!(mul, Gf256::zero());
} else {
assert_eq!(a, div.unwrap());
}
});
}
}