commit 524314247686801c7fda1574ea2f00e6163e6e29
author Jerry Yu <jerry.h.yu@arm.com> Thu Sep 16 13:25:04 2021 +0800
committer Jerry Yu <jerry.h.yu@arm.com> Mon Sep 27 16:34:58 2021 +0800
tree 5629b32f0b08d4e92b1b4f06c2ca9a6731526b8d
parent e3131ef7f34055748d11e1e252124cf6c561a899

Add macro for length of input counter

Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>

library/ssl_msg.c

diff --git a/library/ssl_msg.c b/library/ssl_msg.c
index ea1d535..36a3e20 100644
--- a/library/ssl_msg.c
+++ b/library/ssl_msg.c
@@ -54,6 +54,8 @@
 #include "mbedtls/oid.h"
 #endif
 
+#define MBEDTLS_SSL_IN_CTR_LEN 8
+
 static uint32_t ssl_get_hs_total_len( mbedtls_ssl_context const *ssl );
 
 /*
@@ -4791,7 +4793,7 @@
     }
     else
 #endif /* MBEDTLS_SSL_PROTO_DTLS */
-    memset( ssl->in_ctr, 0, 8 );
+    mbedtls_platform_zeroize( ssl->in_ctr, MBEDTLS_SSL_IN_CTR_LEN );
 
     mbedtls_ssl_update_in_pointers( ssl );
 
@@ -5558,7 +5560,7 @@
         return;
 
     ssl->transform_in = transform;
-    mbedtls_platform_zeroize( ssl->in_ctr, 8 );
+    mbedtls_platform_zeroize( ssl->in_ctr, MBEDTLS_SSL_IN_CTR_LEN );
 }
 
 void mbedtls_ssl_set_outbound_transform( mbedtls_ssl_context *ssl,