don't break everything.

wolfSSL · Sep 22, 2024 · 1d4602a · 1d4602a
1 parent 2a892e8
commit 1d4602a
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/include/wolfboot/wolfboot.h b/include/wolfboot/wolfboot.h
@@ -128,16 +128,21 @@ extern "C" {
  #define KEYSTORE_PUBKEY_SIZE_LMS     60
  #define KEYSTORE_PUBKEY_SIZE_XMSS    68
  /* ML-DSA pub key size is a function of parameters.
-  * This needs to be configurable. */
- #if ML_DSA_LEVEL == 2
-     #define KEYSTORE_PUBKEY_SIZE_ML_DSA  1312
- #elif ML_DSA_LEVEL == 3
-     #define KEYSTORE_PUBKEY_SIZE_ML_DSA  1952
- #elif ML_DSA_LEVEL == 5
-     #define KEYSTORE_PUBKEY_SIZE_ML_DSA  2592
+  * This needs to be configurable. Default to security
+  * category 2. */
+ #ifdef ML_DSA_LEVEL
+     #if ML_DSA_LEVEL == 2
+         #define KEYSTORE_PUBKEY_SIZE_ML_DSA  1312
+     #elif ML_DSA_LEVEL == 3
+         #define KEYSTORE_PUBKEY_SIZE_ML_DSA  1952
+     #elif ML_DSA_LEVEL == 5
+         #define KEYSTORE_PUBKEY_SIZE_ML_DSA  2592
+     #else
+         #error "Invalid ML_DSA_LEVEL!"
+     #endif
  #else
-     #error "Invalid ML_DSA_LEVEL!"
- #endif
+     #define KEYSTORE_PUBKEY_SIZE_ML_DSA  1312
+ #endif /* ML_DSA_LEVEL */
 
 /* Mask for key permissions */
  #define KEY_VERIFY_ALL         (0xFFFFFFFFU)