summaryrefslogtreecommitdiffstats
path: root/vendor/crypto-bigint/tests/proptests.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-18 02:49:50 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-18 02:49:50 +0000
commit9835e2ae736235810b4ea1c162ca5e65c547e770 (patch)
tree3fcebf40ed70e581d776a8a4c65923e8ec20e026 /vendor/crypto-bigint/tests/proptests.rs
parentReleasing progress-linux version 1.70.0+dfsg2-1~progress7.99u1. (diff)
downloadrustc-9835e2ae736235810b4ea1c162ca5e65c547e770.tar.xz
rustc-9835e2ae736235810b4ea1c162ca5e65c547e770.zip
Merging upstream version 1.71.1+dfsg1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'vendor/crypto-bigint/tests/proptests.rs')
-rw-r--r--vendor/crypto-bigint/tests/proptests.rs84
1 files changed, 80 insertions, 4 deletions
diff --git a/vendor/crypto-bigint/tests/proptests.rs b/vendor/crypto-bigint/tests/proptests.rs
index 42a3066c6..9f489f0a8 100644
--- a/vendor/crypto-bigint/tests/proptests.rs
+++ b/vendor/crypto-bigint/tests/proptests.rs
@@ -1,7 +1,11 @@
//! Equivalence tests between `num-bigint` and `crypto-bigint`
-use crypto_bigint::{Encoding, U256};
+use crypto_bigint::{
+ modular::runtime_mod::{DynResidue, DynResidueParams},
+ Encoding, Limb, NonZero, Word, U256,
+};
use num_bigint::BigUint;
+use num_integer::Integer;
use num_traits::identities::Zero;
use proptest::prelude::*;
use std::mem;
@@ -15,9 +19,9 @@ fn to_biguint(uint: &U256) -> BigUint {
}
fn to_uint(big_uint: BigUint) -> U256 {
- let mut input = [0u8; U256::BYTE_SIZE];
+ let mut input = [0u8; U256::BYTES];
let encoded = big_uint.to_bytes_le();
- let l = encoded.len().min(U256::BYTE_SIZE);
+ let l = encoded.len().min(U256::BYTES);
input[..l].copy_from_slice(&encoded[..l]);
U256::from_le_slice(&input)
@@ -33,6 +37,11 @@ prop_compose! {
a.wrapping_rem(&p)
}
}
+prop_compose! {
+ fn nonzero_limb()(x in any::<Word>()) -> Limb {
+ if x == 0 { Limb::from(1u32) } else {Limb::from(x)}
+ }
+}
proptest! {
#[test]
@@ -41,6 +50,16 @@ proptest! {
}
#[test]
+ fn shl_vartime(a in uint(), shift in any::<u8>()) {
+ let a_bi = to_biguint(&a);
+
+ let expected = to_uint(a_bi << shift);
+ let actual = a.shl_vartime(shift as usize);
+
+ assert_eq!(expected, actual);
+ }
+
+ #[test]
fn wrapping_add(a in uint(), b in uint()) {
let a_bi = to_biguint(&a);
let b_bi = to_biguint(&b);
@@ -131,6 +150,30 @@ proptest! {
}
#[test]
+ fn div_rem_limb(a in uint(), b in nonzero_limb()) {
+ let a_bi = to_biguint(&a);
+ let b_bi = to_biguint(&U256::from(b));
+
+ let (expected_quo, expected_rem) = a_bi.div_rem(&b_bi);
+ let (actual_quo, actual_rem) = a.div_rem_limb(NonZero::new(b).unwrap());
+ assert_eq!(to_uint(expected_quo), actual_quo);
+ assert_eq!(to_uint(expected_rem), U256::from(actual_rem));
+ }
+
+ #[test]
+ fn div_rem_limb_min_max(a in uint()) {
+ let a_bi = to_biguint(&a);
+
+ for b in [Limb::from(1u32), Limb::MAX] {
+ let b_bi = to_biguint(&U256::from(b));
+ let (expected_quo, expected_rem) = a_bi.div_rem(&b_bi);
+ let (actual_quo, actual_rem) = a.div_rem_limb(NonZero::new(b).unwrap());
+ assert_eq!(to_uint(expected_quo), actual_quo);
+ assert_eq!(to_uint(expected_rem), U256::from(actual_rem));
+ }
+ }
+
+ #[test]
fn wrapping_rem(a in uint(), b in uint()) {
let a_bi = to_biguint(&a);
let b_bi = to_biguint(&b);
@@ -205,5 +248,38 @@ proptest! {
let mut bytes = a.to_le_bytes();
bytes.reverse();
assert_eq!(a, U256::from_be_bytes(bytes));
-}
+ }
+
+ #[test]
+ fn residue_pow(a in uint_mod_p(P), b in uint()) {
+ let a_bi = to_biguint(&a);
+ let b_bi = to_biguint(&b);
+ let p_bi = to_biguint(&P);
+
+ let expected = to_uint(a_bi.modpow(&b_bi, &p_bi));
+
+ let params = DynResidueParams::new(&P);
+ let a_m = DynResidue::new(&a, params);
+ let actual = a_m.pow(&b).retrieve();
+
+ assert_eq!(expected, actual);
+ }
+
+ #[test]
+ fn residue_pow_bounded_exp(a in uint_mod_p(P), b in uint(), exponent_bits in any::<u8>()) {
+
+ let b_masked = b & (U256::ONE << exponent_bits.into()).wrapping_sub(&U256::ONE);
+
+ let a_bi = to_biguint(&a);
+ let b_bi = to_biguint(&b_masked);
+ let p_bi = to_biguint(&P);
+
+ let expected = to_uint(a_bi.modpow(&b_bi, &p_bi));
+
+ let params = DynResidueParams::new(&P);
+ let a_m = DynResidue::new(&a, params);
+ let actual = a_m.pow_bounded_exp(&b, exponent_bits.into()).retrieve();
+
+ assert_eq!(expected, actual);
+ }
}