hax-lib 0.3.6

Hax-specific helpers for Rust programs
Documentation
//! This module provides an approximation of `BigInt` which is
//! copiable, via an big array of `u8` of an fixed arbitrary size
//! `BYTES`.
//! Its interface provides bridges to `num_bigint::BigInt`.

/// Maximal number of bytes stored in our copiable `BigInt`s.
const BYTES: usize = 1024;
#[derive(Debug, Copy, Clone)]
pub(super) struct BigInt {
    sign: num_bigint::Sign,
    data: [u8; BYTES],
}
impl BigInt {
    /// Construct a [`BigInt`] from a [`num_bigint::BigInt`]. This
    /// operation panics when the provided [`num_bigint::BigInt`]
    /// has more than [`BYTES`] bytes.
    pub(super) fn new(i: &num_bigint::BigInt) -> Self {
        let (sign, bytes) = i.to_bytes_be();
        if bytes.len() > BYTES {
            panic!("`copiable_bigint::BigInt::new`: too big, please consider increasing `BYTES`");
        }
        let mut data = [0; BYTES];
        data[BYTES - bytes.len()..].copy_from_slice(&bytes[..]);
        BigInt { sign, data }
    }

    /// Constructs a [`num_bigint::BigInt`] out of a [`BigInt`].
    pub(super) fn get(self) -> num_bigint::BigInt {
        num_bigint::BigInt::from_bytes_be(self.sign, &self.data)
    }
}

impl core::cmp::PartialEq for BigInt {
    fn eq(&self, other: &Self) -> bool {
        self.get() == other.get()
    }
}
impl core::cmp::Eq for BigInt {}
impl core::cmp::Ord for BigInt {
    fn cmp(&self, other: &Self) -> core::cmp::Ordering {
        self.get().cmp(&other.get())
    }
}
impl core::cmp::PartialOrd for BigInt {
    fn partial_cmp(&self, other: &Self) -> Option<core::cmp::Ordering> {
        self.get().partial_cmp(&other.get())
    }
}