Introduce Uint::new_overflowing_shr

RustCrypto · Jan 24, 2025 · 6fdf08b · 6fdf08b
1 parent 1c98833
commit 6fdf08b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/benches/uint.rs b/benches/uint.rs
@@ -405,6 +405,14 @@ fn bench_shr(c: &mut Criterion) {
         )
     });
 
+    group.bench_function("new shr, U2048", |b| {
+        b.iter_batched(
+            || U2048::ONE,
+            |x| x.new_overflowing_shr(1024 + 10),
+            BatchSize::SmallInput,
+        )
+    });
+
     group.finish();
 }
 

diff --git a/src/uint/shr.rs b/src/uint/shr.rs
@@ -47,6 +47,12 @@ impl<const LIMBS: usize> Uint<LIMBS> {
         ConstCtOption::new(Uint::select(&result, &Self::ZERO, overflow), overflow.not())
     }
 
+    pub const fn new_overflowing_shr(&self, shift: u32) -> ConstCtOption<Self> {
+        let (intra_limb_shift, limb_shift) = (shift % Limb::BITS, shift / Limb::BITS);
+        self.intra_limb_carrying_shr_internal(intra_limb_shift)
+            .full_limb_shr(limb_shift)
+    }
+
     /// Computes `self >> shift`, for `shift < Limb::BITS`.
     ///
     /// Returns `None` if `shift >= Limb::BITS`.