3939 MLK_ADD_PARAM_SET(mlk_polyvec_permute_bitrev_to_custom)
4040#define mlk_polymat_permute_bitrev_to_custom \
4141 MLK_ADD_PARAM_SET(mlk_polymat_permute_bitrev_to_custom)
42- #define mlk_keypair_getnoise MLK_ADD_PARAM_SET(mlk_keypair_getnoise )
42+ #define mlk_keypair_getnoise_eta1 MLK_ADD_PARAM_SET(mlk_keypair_getnoise_eta1 )
4343/* End of parameter set namespacing */
4444
4545/*************************************************
@@ -367,10 +367,10 @@ __contract__(
367367}
368368
369369/*************************************************
370- * Name: mlk_keypair_getnoise
370+ * Name: mlk_keypair_getnoise_eta1
371371 *
372372 * Description: Computes and fills the pv and e polyvec
373- * structures needed by mlk_keypair_derand()
373+ * structures needed by mlk_keypair_derand().
374374 *
375375 * Arguments: - pv: Pointer to output polynomial vector
376376 * - e: Pointer to output polynomial vector
@@ -379,8 +379,8 @@ __contract__(
379379 * Specification: Implements @[FIPS203, Algorithm 13 (K-PKE.KeyGen)].
380380 * steps 8 - 15
381381 **************************************************/
382- static void mlk_keypair_getnoise (mlk_polyvec * pv , mlk_polyvec * e ,
383- const uint8_t seed [MLKEM_SYMBYTES ])
382+ static void mlk_keypair_getnoise_eta1 (mlk_polyvec * pv , mlk_polyvec * e ,
383+ const uint8_t seed [MLKEM_SYMBYTES ])
384384__contract__ (
385385 requires (memory_no_alias (pv , sizeof (mlk_polyvec )))
386386 requires (memory_no_alias (e , sizeof (mlk_polyvec )))
@@ -392,17 +392,19 @@ __contract__(
392392)
393393{
394394#if MLKEM_K == 2
395- mlk_poly_getnoise_eta1_4x (& pv -> vec [0 ], & pv -> vec [1 ], & e -> vec [0 ], & e -> vec [1 ],
395+ mlk_poly_getnoise_eta1_4x (& pv -> vec [0 ], & pv -> vec [1 ], /* Fill elements of pv */
396+ & e -> vec [0 ], & e -> vec [1 ], /* and two elements of e */
396397 seed , 0 , 1 , 2 , 3 );
397398#elif MLKEM_K == 3
398399 /*
399- * Only the first three output buffers are needed.
400+ * Only the first three output buffers are needed, so we pass NULL as
401+ * the fourth parameter, and 0xFF as its dummy nonce.
400402 */
401403 mlk_poly_getnoise_eta1_4x (& pv -> vec [0 ], & pv -> vec [1 ], & pv -> vec [2 ], NULL , seed ,
402- 0 , 1 , 2 , 0xFF /* irrelevant */ );
404+ 0 , 1 , 2 , 0xFF );
403405 /* Same here */
404406 mlk_poly_getnoise_eta1_4x (& e -> vec [0 ], & e -> vec [1 ], & e -> vec [2 ], NULL , seed , 3 ,
405- 4 , 5 , 0xFF /* irrelevant */ );
407+ 4 , 5 , 0xFF );
406408#elif MLKEM_K == 4
407409 mlk_poly_getnoise_eta1_4x (& pv -> vec [0 ], & pv -> vec [1 ], & pv -> vec [2 ], & pv -> vec [3 ],
408410 seed , 0 , 1 , 2 , 3 );
@@ -463,7 +465,7 @@ int mlk_indcpa_keypair_derand(uint8_t pk[MLKEM_INDCPA_PUBLICKEYBYTES],
463465
464466 mlk_gen_matrix (a , publicseed , 0 /* no transpose */ );
465467
466- mlk_keypair_getnoise (skpv , e , noiseseed );
468+ mlk_keypair_getnoise_eta1 (skpv , e , noiseseed );
467469
468470 mlk_polyvec_ntt (skpv );
469471 mlk_polyvec_ntt (e );
@@ -652,4 +654,4 @@ int mlk_indcpa_dec(uint8_t m[MLKEM_INDCPA_MSGBYTES],
652654#undef mlk_matvec_mul
653655#undef mlk_polyvec_permute_bitrev_to_custom
654656#undef mlk_polymat_permute_bitrev_to_custom
655- #undef mlk_keypair_getnoise
657+ #undef mlk_keypair_getnoise_eta1
0 commit comments