digest: implement BlockSizeUser for RtVariableCoreWrapper

RustCrypto · Oct 16, 2023 · b321709 · b321709
1 parent eeed993
commit b321709
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/digest/src/core_api/rt_variable.rs b/digest/src/core_api/rt_variable.rs
@@ -1,4 +1,4 @@
-use super::{AlgorithmName, TruncSide, UpdateCore, VariableOutputCore};
+use super::{AlgorithmName, BlockSizeUser, TruncSide, UpdateCore, VariableOutputCore};
 #[cfg(feature = "mac")]
 use crate::MacMarker;
 use crate::{HashMarker, InvalidBufferSize};
@@ -59,6 +59,15 @@ impl<T> HashMarker for RtVariableCoreWrapper<T> where T: VariableOutputCore + Ha
 #[cfg_attr(docsrs, doc(cfg(feature = "mac")))]
 impl<T> MacMarker for RtVariableCoreWrapper<T> where T: VariableOutputCore + MacMarker {}
 
+impl<T> BlockSizeUser for RtVariableCoreWrapper<T>
+where
+    T: VariableOutputCore,
+    T::BlockSize: IsLess<U256>,
+    Le<T::BlockSize, U256>: NonZero,
+{
+    type BlockSize = T::BlockSize;
+}
+
 impl<T> Reset for RtVariableCoreWrapper<T>
 where
     T: VariableOutputCore + UpdateCore + Reset,