minor fix, leftover from previous pr

aws · Aug 26, 2024 · d4d2f11 · d4d2f11
1 parent 2bb8357
commit d4d2f11
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/crypto/fipsmodule/ml_kem/ml_kem_ref/indcpa.c b/crypto/fipsmodule/ml_kem/ml_kem_ref/indcpa.c
@@ -162,7 +162,7 @@ static unsigned int rej_uniform(int16_t *r,
 **************************************************/
 #define GEN_MATRIX_NBLOCKS ((12*KYBER_N/8*(1 << 12)/KYBER_Q + XOF_BLOCKBYTES)/XOF_BLOCKBYTES)
 // Not static for benchmarking
-void gen_matrix(ml_kem_params *params, polyvec *a, const uint8_t *seed, int transposed)
+void gen_matrix(ml_kem_params *params, polyvec *a, const uint8_t seed[KYBER_SYMBYTES], int transposed)
 {
   unsigned int ctr, i, j, k;
   unsigned int buflen, off;
@@ -267,7 +267,7 @@ void indcpa_enc(ml_kem_params *params,
                 uint8_t *c,
                 const uint8_t *m,
                 const uint8_t *pk,
-                const uint8_t *coins)
+                const uint8_t coins[KYBER_SYMBYTES])
 {
   unsigned int i;
   uint8_t seed[KYBER_SYMBYTES];

diff --git a/crypto/fipsmodule/ml_kem/ml_kem_ref/indcpa.h b/crypto/fipsmodule/ml_kem/ml_kem_ref/indcpa.h
@@ -6,7 +6,7 @@
 #include "polyvec.h"
 
 #define gen_matrix KYBER_NAMESPACE(gen_matrix)
-void gen_matrix(ml_kem_params *params, polyvec *a, const uint8_t *seed, int transposed);
+void gen_matrix(ml_kem_params *params, polyvec *a, const uint8_t seed[KYBER_SYMBYTES], int transposed);
 
 #define indcpa_keypair_derand KYBER_NAMESPACE(indcpa_keypair_derand)
 void indcpa_keypair_derand(ml_kem_params *params,
@@ -19,7 +19,7 @@ void indcpa_enc(ml_kem_params *params,
                 uint8_t *c,
                 const uint8_t *m,
                 const uint8_t *pk,
-                const uint8_t *coins);
+                const uint8_t coins[KYBER_SYMBYTES]);
 
 #define indcpa_dec KYBER_NAMESPACE(indcpa_dec)
 void indcpa_dec(ml_kem_params *params,