use crate::internal::ByteVector;
use arrayvec;
use arrayvec::ArrayVec;
use gridiron::fp_256::Fp256;
use gridiron::fp_480::Fp480;
use quick_error::quick_error;
use std::convert::From;
use std::result::Result;
pub trait BytesDecoder
where
Self: Sized,
{
const ENCODED_SIZE_BYTES: usize;
fn decode(bytes: ByteVector) -> Result<Self, DecodeErr>;
}
impl BytesDecoder for gridiron::fp_256::Monty {
const ENCODED_SIZE_BYTES: usize = 32;
fn decode(bytes: ByteVector) -> Result<Self, DecodeErr> {
let byte_arr: ArrayVec<[u8; Self::ENCODED_SIZE_BYTES]> = bytes.into_iter().collect();
let byte_arr: [u8; Self::ENCODED_SIZE_BYTES] = byte_arr.into_inner()?;
Result::Ok(Fp256::from(byte_arr).to_monty())
}
}
impl BytesDecoder for gridiron::fp_480::Monty {
const ENCODED_SIZE_BYTES: usize = 60;
fn decode(bytes: Vec<u8>) -> Result<Self, DecodeErr> {
if bytes.len() == Self::ENCODED_SIZE_BYTES {
let mut byte_arr: [u8; Self::ENCODED_SIZE_BYTES] = [0u8; Self::ENCODED_SIZE_BYTES];
byte_arr.copy_from_slice(&bytes);
Result::Ok(Fp480::from(byte_arr).to_monty())
} else {
Result::Err(DecodeErr::BytesNotCorrectLength {
required_length: Self::ENCODED_SIZE_BYTES,
bad_bytes: bytes,
})
}
}
}
quick_error! {
#[derive(Debug, PartialEq, Eq)]
pub enum DecodeErr {
BytesNotCorrectLength {
required_length: usize,
bad_bytes: ByteVector,
} {
}
BytesInvalid {
message: String,
bad_bytes: ByteVector,
}{}
}
}
macro_rules! from_arrayvec_n { ($($n: expr), *) => {
$(impl From<arrayvec::ArrayVec<[u8; $n]>> for DecodeErr {
fn from(x: arrayvec::ArrayVec<[u8; $n]>) -> Self {
DecodeErr::BytesNotCorrectLength {
required_length: $n,
bad_bytes: x.to_vec(),
}
}
})+
}}
from_arrayvec_n! {32, 128, 384}