sha3: Add hazmat TSHK API new_with_round_count

Author: aewag

sha3/src/macros.rs

@@ -246,14 +246,27 @@ macro_rules! impl_turbo_shake {
         impl $name {
             /// Creates a new TurboSHAKE instance with the given domain separation.
             /// Note that the domain separation needs to be a byte with a value in
-            /// the range [0x01, . . . , 0x7F]
+            /// the range [0x01, . . . , 0x7F].
             pub fn new(domain_separation: u8) -> Self {
                 assert!((0x01..=0x7F).contains(&domain_separation));
                 Self {
                     domain_separation,
                     state: Sha3State::new(TURBO_SHAKE_ROUND_COUNT),
                 }
             }
+
+            /// Creates a new TurboSHAKE instance with the given domain separation
+            /// and round_count.
+            /// This is a low-level "hazmat" API.
+            /// Note that the domain separation needs to be a byte with a value in
+            /// the range [0x01, . . . , 0x7F].
+            pub fn new_with_round_count(domain_separation: u8, round_count: usize) -> Self {
+                assert!((0x01..=0x7F).contains(&domain_separation));
+                Self {
+                    domain_separation,
+                    state: Sha3State::new(round_count),
+                }
+            }
         }
 
         impl HashMarker for $name {}