zeroize mat array on end of usage

Colfenor · Nov 10, 2024 · d03ab15 · d03ab15
1 parent ead1096
commit d03ab15
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/pk_gen.rs b/src/pk_gen.rs
@@ -1,5 +1,7 @@
 //! Generation of public key
 
+use zeroize::Zeroize;
+
 use crate::{
     gf::{gf_inv, gf_mul},
     macros::sub,
@@ -305,6 +307,10 @@ pub(crate) fn pk_gen(
             {
                 if row == PK_NROWS - 32 {
                     if mov_columns(&mut mat, pi, pivots) != 0 {
+                        #[cfg(feature = "alloc")]
+                        {
+                            mat.zeroize();
+                        }
                         return -1;
                     }
                 }
@@ -322,6 +328,10 @@ pub(crate) fn pk_gen(
             }
 
             if ((mat[row][i] >> j) & 1) == 0 {
+                #[cfg(feature = "alloc")]
+                {
+                    mat.zeroize();
+                }
                 return -1;
             }
 
@@ -361,6 +371,11 @@ pub(crate) fn pk_gen(
         }
     }
 
+    #[cfg(feature = "alloc")]
+    {
+        mat.zeroize();
+    }
+
     0
 }