fix zeroize support in jh

RustCrypto · Jan 11, 2024 · c080ac9 · c080ac9
1 parent 176d993
commit c080ac9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/jh/src/lib.rs b/jh/src/lib.rs
@@ -122,8 +122,12 @@ impl Drop for JhCore {
     fn drop(&mut self) {
         #[cfg(feature = "zeroize")]
         {
-            // TODO: implement Zeroize for Compressor
-            // self.state.zeroize();
+            const N: usize = core::mem::size_of::<Compressor>();
+            // TODO: remove this unsafe after migration from `ppv-lite86`
+            unsafe {
+                let p: *mut [u8; N] = (&mut self.state as *mut Compressor).cast();
+                core::ptr::write_volatile(p, [0u8; N]);
+            }
             self.block_len.zeroize();
         }
     }