commit 0f49f836253244121feae78e81184af1087fc425
author Neil Armstrong <narmstrong@baylibre.com> Mon Feb 28 15:07:38 2022 +0100
committer Neil Armstrong <narmstrong@baylibre.com> Tue Mar 01 10:05:33 2022 +0100
tree d0ac19ce2ab37ff2f5197468f2d5fd9e17d251b2
parent e9ecd2789028e5d9cf96cb5f4aaa0996ab7b9571

Use now shared ECP_PUB_DER_MAX_BYTES define in pk_wrap.c

Signed-off-by: Neil Armstrong <narmstrong@baylibre.com>

library/pk_wrap.c

diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 0bb87a4..a9c9dd5 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -32,6 +32,10 @@
 #include "mbedtls/ecp.h"
 #endif
 
+#if defined(MBEDTLS_RSA_C) || defined(MBEDTLS_ECP_C)
+#include "pkwrite.h"
+#endif
+
 #if defined(MBEDTLS_ECDSA_C)
 #include "mbedtls/ecdsa.h"
 #endif
@@ -564,8 +568,7 @@
     psa_status_t status;
     mbedtls_pk_context key;
     int key_len;
-    /* see ECP_PUB_DER_MAX_BYTES in pkwrite.c */
-    unsigned char buf[30 + 2 * MBEDTLS_ECP_MAX_BYTES];
+    unsigned char buf[MBEDTLS_PK_ECP_PUB_DER_MAX_BYTES];
     unsigned char *p;
     mbedtls_pk_info_t pk_info = mbedtls_eckey_info;
     psa_algorithm_t psa_sig_md = PSA_ALG_ECDSA_ANY;