fix various issues
Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>
diff --git a/library/pk.c b/library/pk.c
index 4c3c774..0a7a40b 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -545,14 +545,10 @@
sig, sig_size, sig_len, f_rng, p_rng ) );
}
-#if defined(MBEDTLS_X509_RSASSA_PSS_SUPPORT)
return( mbedtls_pk_psa_sign_ext( PSA_ALG_RSA_PSS(
mbedtls_psa_translate_md( md_alg ) ),
ctx->pk_ctx, hash, hash_len,
sig, sig_size, sig_len ) );
-#else /* MBEDTLS_X509_RSASSA_PSS_SUPPORT */
- return( MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE );
-#endif /* !MBEDTLS_X509_RSASSA_PSS_SUPPORT */
}
#endif /* MBEDTLS_PSA_CRYPTO_C */
diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 4c66e5f..76b7a1b 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -251,7 +251,7 @@
ret = mbedtls_pk_error_from_psa( status );
return( ret );
}
-#endif /* MBEDTLS_USE_PSA_CRYPTO */
+#endif /* MBEDTLS_PSA_CRYPTO_C */
#if defined(MBEDTLS_USE_PSA_CRYPTO)
static int rsa_sign_wrap( void *ctx, mbedtls_md_type_t md_alg,
diff --git a/library/pk_wrap.h b/library/pk_wrap.h
index d4a90ab..e310005 100644
--- a/library/pk_wrap.h
+++ b/library/pk_wrap.h
@@ -155,6 +155,6 @@
unsigned char *sig, size_t sig_size,
size_t *sig_len );
-#endif /* MBEDTLS_USE_PSA_CRYPTO */
+#endif /* MBEDTLS_PSA_CRYPTO_C */
#endif /* MBEDTLS_PK_WRAP_H */
diff --git a/library/ssl_tls13_generic.c b/library/ssl_tls13_generic.c
index a8c3570..91c5448 100644
--- a/library/ssl_tls13_generic.c
+++ b/library/ssl_tls13_generic.c
@@ -1129,7 +1129,6 @@
pk_type = MBEDTLS_PK_ECDSA;
break;
#endif /* MBEDTLS_ECDSA_C */
-#if defined(MBEDTLS_X509_RSASSA_PSS_SUPPORT)
case MBEDTLS_TLS1_3_SIG_RSA_PSS_RSAE_SHA256:
md_alg = MBEDTLS_MD_SHA256;
pk_type = MBEDTLS_PK_RSASSA_PSS;
@@ -1142,7 +1141,6 @@
md_alg = MBEDTLS_MD_SHA512;
pk_type = MBEDTLS_PK_RSASSA_PSS;
break;
-#endif /* MBEDTLS_X509_RSASSA_PSS_SUPPORT */
default:
break;
}