commit 08d5f46c83d0d7fcfbadd0e879db7da015046a88
author Yanray Wang <yanray.wang@arm.com> Mon Aug 21 15:15:19 2023 +0800
committer Yanray Wang <yanray.wang@arm.com> Mon Aug 21 15:15:24 2023 +0800
tree 9b9acbf3596a13521157300b947e87b272752c07
parent 0882828b5162f1a4e3e9776dc5ea8105650de314

pkwrite.c: call calloc and free properly

Signed-off-by: Yanray Wang <yanray.wang@arm.com>

library/pkwrite.c

diff --git a/library/pkwrite.c b/library/pkwrite.c
index cc58f46..225cde9 100644
--- a/library/pkwrite.c
+++ b/library/pkwrite.c
@@ -760,7 +760,7 @@
 {
     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
     unsigned char *output_buf = NULL;
-    output_buf = calloc(1, PUB_DER_MAX_BYTES);
+    output_buf = mbedtls_calloc(1, PUB_DER_MAX_BYTES);
     if (output_buf == NULL) {
         return MBEDTLS_ERR_PK_ALLOC_FAILED;
     }
@@ -779,7 +779,7 @@
 
     ret = 0;
 cleanup:
-    free(output_buf);
+    mbedtls_free(output_buf);
     return ret;
 }
 
@@ -787,7 +787,7 @@
 {
     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
     unsigned char *output_buf = NULL;
-    output_buf = calloc(1, PRV_DER_MAX_BYTES);
+    output_buf = mbedtls_calloc(1, PRV_DER_MAX_BYTES);
     if (output_buf == NULL) {
         return MBEDTLS_ERR_PK_ALLOC_FAILED;
     }
@@ -860,7 +860,7 @@
 
     ret = 0;
 cleanup:
-    free(output_buf);
+    mbedtls_free(output_buf);
     return ret;
 }
 #endif /* MBEDTLS_PEM_WRITE_C */