turbo-shake: fix zeroize

Author: newpavlov

turbo-shake/src/lib.rs

@@ -159,6 +159,17 @@ impl<Rate: BlockSizes, const DS: u8> fmt::Debug for TurboShake<Rate, DS> {
     }
 }
 
+impl<Rate: BlockSizes, const DS: u8> Drop for TurboShake<Rate, DS> {
+    fn drop(&mut self) {
+        #[cfg(feature = "zeroize")]
+        {
+            use digest::zeroize::Zeroize;
+            self.state.zeroize();
+            // self.buffer is zeroized by its `Drop`
+        }
+    }
+}
+
 #[cfg(feature = "zeroize")]
 impl<Rate: BlockSizes, const DS: u8> digest::zeroize::ZeroizeOnDrop for TurboShake<Rate, DS> {}
 
@@ -199,6 +210,20 @@ impl<Rate: BlockSizes> fmt::Debug for TurboShakeReader<Rate> {
     }
 }
 
+impl<Rate: BlockSizes> Drop for TurboShakeReader<Rate> {
+    fn drop(&mut self) {
+        #[cfg(feature = "zeroize")]
+        {
+            use digest::zeroize::Zeroize;
+            self.state.zeroize();
+            // self.buffer is zeroized by its `Drop`
+        }
+    }
+}
+
+#[cfg(feature = "zeroize")]
+impl<Rate: BlockSizes> digest::zeroize::ZeroizeOnDrop for TurboShakeReader<Rate> {}
+
 /// TurboSHAKE128 hasher with domain separator.
 pub type TurboShake128<const DS: u8 = DEFAULT_DS> = TurboShake<U168, DS>;
 /// TurboSHAKE256 hasher with domain separator.