From 35f91da44210ff77458560ab6597b74d20127ef0 Mon Sep 17 00:00:00 2001
From: Takashi Kojo <kojo@wolfssl.com>
Date: Fri, 1 Nov 2024 09:26:41 +0900
Subject: [PATCH] tests for dilithium

---
 .github/workflows/ubuntu-check.yml            |  2 +-
 tests/genkey_sign_ver/genkey-sign-ver-test.sh | 18 ++++++++++--------
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/ubuntu-check.yml b/.github/workflows/ubuntu-check.yml
index 5ac78097..accf6825 100644
--- a/.github/workflows/ubuntu-check.yml
+++ b/.github/workflows/ubuntu-check.yml
@@ -21,7 +21,7 @@ jobs:
       run: ./autogen.sh
     - name: wolfssl configure
       working-directory: ./wolfssl 
-      run: ./configure --enable-wolfclu --enable-crl --enable-dsa
+      run: ./configure --enable-wolfclu --enable-crl --enable-dsa --enable-pkcs7 --enable-dilithium
     - name: wolfssl make
       working-directory: ./wolfssl
       run: make
diff --git a/tests/genkey_sign_ver/genkey-sign-ver-test.sh b/tests/genkey_sign_ver/genkey-sign-ver-test.sh
index d0195ad4..25f307e9 100755
--- a/tests/genkey_sign_ver/genkey-sign-ver-test.sh
+++ b/tests/genkey_sign_ver/genkey-sign-ver-test.sh
@@ -176,14 +176,15 @@ SIGOUTNAME="ed-signed.sig"
 DERPEMRAW="raw"
 gen_key_sign_ver_test ${ALGORITHM} ${KEYFILENAME} ${SIGOUTNAME} ${DERPEMRAW}
 
-ALGORITHM="dilithium"
-KEYFILENAME="mldsakey"
-SIGOUTNAME="mldsa-signed.sig"
-DERPEMRAW="der"
-for level in 2 3 5
-do
-    gen_key_sign_ver_test ${ALGORITHM} ${KEYFILENAME} ${SIGOUTNAME} ${DERPEMRAW} ${level}
-done
+if grep -q "#define HAVE_DILITHIUM" /usr/local/include/wolfssl/options.h; then
+    ALGORITHM="dilithium"
+    KEYFILENAME="mldsakey"
+    SIGOUTNAME="mldsa-signed.sig"
+    DERPEMRAW="der"
+    for level in 2 3 5
+    do
+        gen_key_sign_ver_test ${ALGORITHM} ${KEYFILENAME} ${SIGOUTNAME} ${DERPEMRAW} ${level}
+    done
 
 ALGORITHM="dilithium"
 KEYFILENAME="mldsakey"
@@ -193,5 +194,6 @@ for level in 2 3 5
 do
     gen_key_sign_ver_test ${ALGORITHM} ${KEYFILENAME} ${SIGOUTNAME} ${DERPEMRAW} ${level}
 done
+fi
 
 exit 0