dilithium: support building dilithium with wolfboot.

wolfSSL · Sep 21, 2024 · 0e7ce3c · 0e7ce3c
1 parent a9cc880
commit 0e7ce3c
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 3 deletions.
diff --git a/wolfcrypt/src/dilithium.c b/wolfcrypt/src/dilithium.c
@@ -70,7 +70,7 @@
  * but is slower.
  * WOLFSSL_DILITHIUM_SMALL_MEM_POLY64 Default: OFF
  * Compiles the small memory implementations to use a 64-bit polynomial.
- * Uses 2KB of memory but is slighlty quicker (2.75-7%).
+ * Uses 2KB of memory but is slightly quicker (2.75-7%).
  *
  * WOLFSSL_DILITHIUM_ALIGNMENT Default: 8
  * Use to indicate whether loading and storing of words needs to be aligned.

diff --git a/wolfcrypt/test/test.c b/wolfcrypt/test/test.c
@@ -42080,16 +42080,25 @@ static wc_test_ret_t dilithium_param_vfy_test(int param, const byte* pubKey,
 {
  byte msg[512];
  dilithium_key* key;
+ byte * pubExported = NULL;
  wc_test_ret_t ret;
  int i;
  int res = 0;
+ word32 lenExported = pubKeyLen;
+ int n_diff = 0;
 
  key = (dilithium_key*)XMALLOC(sizeof(*key), HEAP_HINT,
  DYNAMIC_TYPE_TMP_BUFFER);
  if (key == NULL) {
  ERROR_OUT(WC_TEST_RET_ENC_ERRNO, out);
  }
 
+ pubExported = (byte*)XMALLOC(pubKeyLen, HEAP_HINT,
+ DYNAMIC_TYPE_TMP_BUFFER);
+ if (pubExported == NULL) {
+ ERROR_OUT(WC_TEST_RET_ENC_ERRNO, out);
+ }
+
  /* make dummy msg */
  for (i = 0; i < (int)sizeof(msg); i++) {
  msg[i] = (byte)i;
@@ -42108,6 +42117,22 @@ static wc_test_ret_t dilithium_param_vfy_test(int param, const byte* pubKey,
  if (ret != 0)
  ERROR_OUT(WC_TEST_RET_ENC_EC(ret), out);
 
+ /* Now test the ExportPubRaw API, verify we recover the original pub. */
+ ret = wc_dilithium_export_public(key, pubExported, &lenExported);
+ if (ret != 0) {
+ ERROR_OUT(WC_TEST_RET_ENC_EC(ret), out);
+ }
+
+ if (lenExported <= 0 || lenExported != pubKeyLen) {
+ ERROR_OUT(WC_TEST_RET_ENC_EC(lenExported), out);
+ }
+
+ n_diff = XMEMCMP(pubExported, pubKey, pubKeyLen);
+
+ if (n_diff) {
+ ERROR_OUT(WC_TEST_RET_ENC_EC(n_diff), out);
+ }
+
 #ifndef WOLFSSL_DILITHIUM_FIPS204_DRAFT
  ret = wc_dilithium_verify_ctx_msg(sig, sigLen, NULL, 0, msg,
  (word32)sizeof(msg), &res, key);
@@ -42122,6 +42147,7 @@ static wc_test_ret_t dilithium_param_vfy_test(int param, const byte* pubKey,
 out:
  wc_dilithium_free(key);
  XFREE(key, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
+ XFREE(pubExported, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
  return ret;
 }
 

diff --git a/wolfssl/wolfcrypt/dilithium.h b/wolfssl/wolfcrypt/dilithium.h
@@ -726,6 +726,7 @@ int wc_dilithium_export_public(dilithium_key* key, byte* out, word32* outLen);
 #ifdef WOLFSSL_DILITHIUM_PRIVATE_KEY
 WOLFSSL_API
 int wc_dilithium_export_private(dilithium_key* key, byte* out, word32* outLen);
+#define wc_dilithium_export_private_only wc_dilithium_export_private
 #endif
 #ifdef WOLFSSL_DILITHIUM_PRIVATE_KEY
 WOLFSSL_API
@@ -798,15 +799,15 @@ WOLFSSL_API int wc_Dilithium_PrivateKeyToDer(dilithium_key* key, byte* output,
 #define wc_MlDsaKey_ExportPrivRaw(key, out, outLen) \
  wc_dilithium_export_private_only(key, out, outLen)
 #define wc_MlDsaKey_ImportPrivRaw(key, in, inLen) \
- wc_dilithium_import_private_only(out, outLen, key)
+ wc_dilithium_import_private_only(in, inLen, key)
 #define wc_MlDsaKey_Sign(key, sig, sigSz, msg, msgSz, rng) \
  wc_dilithium_sign_msg(msg, msgSz, sig, sigSz, key, rng)
 #define wc_MlDsaKey_Free(key) \
  wc_dilithium_free(key)
 #define wc_MlDsaKey_ExportPubRaw(key, out, outLen) \
  wc_dilithium_export_public(key, out, outLen)
 #define wc_MlDsaKey_ImportPubRaw(key, in, inLen) \
- wc_dilithium_import_public(out, outLen, key)
+ wc_dilithium_import_public(in, inLen, key)
 #define wc_MlDsaKey_Verify(key, sig, sigSz, msg, msgSz, res) \
  wc_dilithium_verify_msg(sig, sigSz, msg, msgSz, res, key)