crypto_bigint/uint/modular/runtime_mod/
runtime_add.rs

1use core::ops::{Add, AddAssign};
2
3use crate::modular::add::add_montgomery_form;
4
5use super::DynResidue;
6
7impl<const LIMBS: usize> DynResidue<LIMBS> {
8    /// Adds `rhs`.
9    pub const fn add(&self, rhs: &Self) -> Self {
10        Self {
11            montgomery_form: add_montgomery_form(
12                &self.montgomery_form,
13                &rhs.montgomery_form,
14                &self.residue_params.modulus,
15            ),
16            residue_params: self.residue_params,
17        }
18    }
19}
20
21impl<const LIMBS: usize> Add<&DynResidue<LIMBS>> for &DynResidue<LIMBS> {
22    type Output = DynResidue<LIMBS>;
23    fn add(self, rhs: &DynResidue<LIMBS>) -> DynResidue<LIMBS> {
24        debug_assert_eq!(self.residue_params, rhs.residue_params);
25        self.add(rhs)
26    }
27}
28
29impl<const LIMBS: usize> Add<DynResidue<LIMBS>> for &DynResidue<LIMBS> {
30    type Output = DynResidue<LIMBS>;
31    #[allow(clippy::op_ref)]
32    fn add(self, rhs: DynResidue<LIMBS>) -> DynResidue<LIMBS> {
33        self + &rhs
34    }
35}
36
37impl<const LIMBS: usize> Add<&DynResidue<LIMBS>> for DynResidue<LIMBS> {
38    type Output = DynResidue<LIMBS>;
39    #[allow(clippy::op_ref)]
40    fn add(self, rhs: &DynResidue<LIMBS>) -> DynResidue<LIMBS> {
41        &self + rhs
42    }
43}
44
45impl<const LIMBS: usize> Add<DynResidue<LIMBS>> for DynResidue<LIMBS> {
46    type Output = DynResidue<LIMBS>;
47    fn add(self, rhs: DynResidue<LIMBS>) -> DynResidue<LIMBS> {
48        &self + &rhs
49    }
50}
51
52impl<const LIMBS: usize> AddAssign<&DynResidue<LIMBS>> for DynResidue<LIMBS> {
53    fn add_assign(&mut self, rhs: &DynResidue<LIMBS>) {
54        *self = *self + rhs;
55    }
56}
57
58impl<const LIMBS: usize> AddAssign<DynResidue<LIMBS>> for DynResidue<LIMBS> {
59    fn add_assign(&mut self, rhs: DynResidue<LIMBS>) {
60        *self += &rhs;
61    }
62}
63
64#[cfg(test)]
65mod tests {
66    use crate::{
67        modular::runtime_mod::{DynResidue, DynResidueParams},
68        U256,
69    };
70
71    #[test]
72    fn add_overflow() {
73        let params = DynResidueParams::new(&U256::from_be_hex(
74            "ffffffff00000000ffffffffffffffffbce6faada7179e84f3b9cac2fc632551",
75        ));
76
77        let x =
78            U256::from_be_hex("44acf6b7e36c1342c2c5897204fe09504e1e2efb1a900377dbc4e7a6a133ec56");
79        let mut x_mod = DynResidue::new(&x, params);
80
81        let y =
82            U256::from_be_hex("d5777c45019673125ad240f83094d4252d829516fac8601ed01979ec1ec1a251");
83        let y_mod = DynResidue::new(&y, params);
84
85        x_mod += &y_mod;
86
87        let expected =
88            U256::from_be_hex("1a2472fde50286541d97ca6a3592dd75beb9c9646e40c511b82496cfc3926956");
89
90        assert_eq!(expected, x_mod.retrieve());
91    }
92}
crypto_bigint/uint/modular/runtime_mod/runtime_add.rs

crypto_bigint/uint/modular/runtime_mod/
runtime_add.rs
