lib/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-07-30 22:43:08 +03:00
Code Activity

pkwrite.c: add a cleanup label to save code size

Browse Source 
Signed-off-by: Yanray Wang <yanray.wang@arm.com>
This commit is contained in:
Yanray Wang
2023-08-11 15:52:09 +08:00
parent 7bbca1363f
commit a8f00508fe
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
library/pkwrite.c
View File

@ -583,19 +583,19 @@ int mbedtls_pk_write_pubkey_pem(mbedtls_pk_context *key, unsigned char *buf, siz
if ((ret = mbedtls_pk_write_pubkey_der(key, output_buf, if ((ret = mbedtls_pk_write_pubkey_der(key, output_buf,
PUB_DER_MAX_BYTES)) < 0) { PUB_DER_MAX_BYTES)) < 0) {
free(output_buf); goto cleanup;
return ret;
} }
if ((ret = mbedtls_pem_write_buffer(PEM_BEGIN_PUBLIC_KEY, PEM_END_PUBLIC_KEY, if ((ret = mbedtls_pem_write_buffer(PEM_BEGIN_PUBLIC_KEY, PEM_END_PUBLIC_KEY,
output_buf + PUB_DER_MAX_BYTES - ret, output_buf + PUB_DER_MAX_BYTES - ret,
ret, buf, size, &olen)) != 0) { ret, buf, size, &olen)) != 0) {
free(output_buf); goto cleanup;
return ret;
} }
ret = 0;
cleanup:
free(output_buf); free(output_buf);
return 0; return ret;
} }
int mbedtls_pk_write_key_pem(mbedtls_pk_context *key, unsigned char *buf, size_t size) int mbedtls_pk_write_key_pem(mbedtls_pk_context *key, unsigned char *buf, size_t size)
@ -613,8 +613,7 @@ int mbedtls_pk_write_key_pem(mbedtls_pk_context *key, unsigned char *buf, size_t
PK_VALIDATE_RET(buf != NULL || size == 0); PK_VALIDATE_RET(buf != NULL || size == 0);
if ((ret = mbedtls_pk_write_key_der(key, output_buf, PRV_DER_MAX_BYTES)) < 0) { if ((ret = mbedtls_pk_write_key_der(key, output_buf, PRV_DER_MAX_BYTES)) < 0) {
free(output_buf); goto cleanup;
return ret;
} }
#if defined(MBEDTLS_RSA_C) #if defined(MBEDTLS_RSA_C)
@ -630,19 +629,20 @@ int mbedtls_pk_write_key_pem(mbedtls_pk_context *key, unsigned char *buf, size_t
} else } else
#endif #endif
{ {
free(output_buf); ret = MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE;
return MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE; goto cleanup;
} }
if ((ret = mbedtls_pem_write_buffer(begin, end, if ((ret = mbedtls_pem_write_buffer(begin, end,
output_buf + PRV_DER_MAX_BYTES - ret, output_buf + PRV_DER_MAX_BYTES - ret,
ret, buf, size, &olen)) != 0) { ret, buf, size, &olen)) != 0) {
free(output_buf); goto cleanup;
return ret;
} }
ret = 0;
cleanup:
free(output_buf); free(output_buf);
return 0; return ret;
} }
#endif /* MBEDTLS_PEM_WRITE_C */ #endif /* MBEDTLS_PEM_WRITE_C */