@@ -1913,23 +1913,54 @@ int test_tls13_rpk_handshake(void)
19131913
19141914
19151915#if defined(HAVE_IO_TESTS_DEPENDENCIES ) && defined(WOLFSSL_TLS13 ) && \
1916- defined(WOLFSSL_HAVE_MLKEM )
1916+ defined(WOLFSSL_HAVE_MLKEM ) && !defined(WOLFSSL_MLKEM_NO_ENCAPSULATE ) && \
1917+ !defined(WOLFSSL_MLKEM_NO_DECAPSULATE ) && \
1918+ !defined(WOLFSSL_MLKEM_NO_MAKE_KEY )
19171919static void test_tls13_pq_groups_ctx_ready (WOLFSSL_CTX * ctx )
19181920{
1921+ #ifndef WOLFSSL_NO_ML_KEM_1024
19191922#ifdef WOLFSSL_MLKEM_KYBER
19201923 int group = WOLFSSL_KYBER_LEVEL5 ;
19211924#else
19221925 int group = WOLFSSL_ML_KEM_1024 ;
1926+ #endif /* WOLFSSL_MLKEM_KYBER */
1927+ #elif !defined(WOLFSSL_NO_ML_KEM_768 )
1928+ #ifdef WOLFSSL_MLKEM_KYBER
1929+ int group = WOLFSSL_KYBER_LEVEL3 ;
1930+ #else
1931+ int group = WOLFSSL_ML_KEM_768 ;
1932+ #endif /* WOLFSSL_MLKEM_KYBER */
1933+ #else
1934+ #ifdef WOLFSSL_MLKEM_KYBER
1935+ int group = WOLFSSL_KYBER_LEVEL1 ;
1936+ #else
1937+ int group = WOLFSSL_ML_KEM_512 ;
1938+ #endif /* WOLFSSL_MLKEM_KYBER */
19231939#endif
1940+
19241941 AssertIntEQ (wolfSSL_CTX_set_groups (ctx , & group , 1 ), WOLFSSL_SUCCESS );
19251942}
19261943
19271944static void test_tls13_pq_groups_on_result (WOLFSSL * ssl )
19281945{
1946+ #ifndef WOLFSSL_NO_ML_KEM_1024
19291947#ifdef WOLFSSL_MLKEM_KYBER
19301948 AssertStrEQ (wolfSSL_get_curve_name (ssl ), "KYBER_LEVEL5" );
19311949#else
19321950 AssertStrEQ (wolfSSL_get_curve_name (ssl ), "ML_KEM_1024" );
1951+ #endif /* WOLFSSL_MLKEM_KYBER */
1952+ #elif !defined(WOLFSSL_NO_ML_KEM_768 )
1953+ #ifdef WOLFSSL_MLKEM_KYBER
1954+ AssertStrEQ (wolfSSL_get_curve_name (ssl ), "KYBER_LEVEL3" );
1955+ #else
1956+ AssertStrEQ (wolfSSL_get_curve_name (ssl ), "ML_KEM_768" );
1957+ #endif /* WOLFSSL_MLKEM_KYBER */
1958+ #else
1959+ #ifdef WOLFSSL_MLKEM_KYBER
1960+ AssertStrEQ (wolfSSL_get_curve_name (ssl ), "KYBER_LEVEL1" );
1961+ #else
1962+ AssertStrEQ (wolfSSL_get_curve_name (ssl ), "ML_KEM_512" );
1963+ #endif /* WOLFSSL_MLKEM_KYBER */
19331964#endif
19341965}
19351966#endif
@@ -1938,7 +1969,9 @@ int test_tls13_pq_groups(void)
19381969{
19391970 EXPECT_DECLS ;
19401971#if defined(HAVE_IO_TESTS_DEPENDENCIES ) && defined(WOLFSSL_TLS13 ) && \
1941- defined(WOLFSSL_HAVE_MLKEM )
1972+ defined(WOLFSSL_HAVE_MLKEM ) && !defined(WOLFSSL_MLKEM_NO_ENCAPSULATE ) && \
1973+ !defined(WOLFSSL_MLKEM_NO_DECAPSULATE ) && \
1974+ !defined(WOLFSSL_MLKEM_NO_MAKE_KEY )
19421975 callback_functions func_cb_client ;
19431976 callback_functions func_cb_server ;
19441977
0 commit comments