Fix zeroize impl

RustCrypto · Jul 31, 2024 · f5cec24 · f5cec24
1 parent 42728f6
commit f5cec24
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/aes/src/armv8.rs b/aes/src/armv8.rs
@@ -158,6 +158,16 @@ macro_rules! define_aes_impl {
             }
         }
 
+        impl Drop for $name {
+            #[inline]
+            fn drop(&mut self) {
+                #[cfg(feature = "zeroize")]
+                unsafe {
+                    zeroize::zeroize_flat_type(self);
+                }
+            }
+        }
+
         #[cfg(feature = "zeroize")]
         impl zeroize::ZeroizeOnDrop for $name {}
 
@@ -216,7 +226,7 @@ macro_rules! define_aes_impl {
             fn drop(&mut self) {
                 #[cfg(feature = "zeroize")]
                 unsafe {
-                    zeroize::zeroize_flat_type(&mut self.backend);
+                    zeroize::zeroize_flat_type(self);
                 }
             }
         }
@@ -294,7 +304,7 @@ macro_rules! define_aes_impl {
             fn drop(&mut self) {
                 #[cfg(feature = "zeroize")]
                 unsafe {
-                    zeroize::zeroize_flat_type(&mut self.backend);
+                    zeroize::zeroize_flat_type(self);
                 }
             }
         }