Apply MBEDTLS_PUT_xyz
Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>
diff --git a/library/ssl_tls13_generic.c b/library/ssl_tls13_generic.c
index 7ec7423..4a0493a 100644
--- a/library/ssl_tls13_generic.c
+++ b/library/ssl_tls13_generic.c
@@ -62,10 +62,10 @@
unsigned char hs_hdr[4];
/* Build HS header for checksum update. */
- hs_hdr[0] = hs_type;
- hs_hdr[1] = (unsigned char)( total_hs_len >> 16 );
- hs_hdr[2] = (unsigned char)( total_hs_len >> 8 );
- hs_hdr[3] = (unsigned char)( total_hs_len >> 0 );
+ hs_hdr[0] = MBEDTLS_BYTE_0( hs_type );
+ hs_hdr[1] = MBEDTLS_BYTE_2( total_hs_len );
+ hs_hdr[2] = MBEDTLS_BYTE_1( total_hs_len );
+ hs_hdr[3] = MBEDTLS_BYTE_0( total_hs_len );
ssl->handshake->update_checksum( ssl, hs_hdr, sizeof( hs_hdr ) );
}