commit c7875b5f11d400f00f4ee6f8ccc83ffd4756dd08
author Jerry Yu <jerry.h.yu@arm.com> Sun Sep 05 21:05:50 2021 +0800
committer Jerry Yu <jerry.h.yu@arm.com> Mon Sep 27 16:34:58 2021 +0800
tree d51b73adeed456e0f7a82025580d5a30c33b546a
parent d3f73349a7b54507bc4becf960bea9ffc2132a3b

add set in/out transform utils

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

library/ssl_misc.h

diff --git a/library/ssl_misc.h b/library/ssl_misc.h
index 6206c6f..8b2d50d 100644
--- a/library/ssl_misc.h
+++ b/library/ssl_misc.h
@@ -986,6 +986,14 @@
  */
 void mbedtls_ssl_handshake_free( mbedtls_ssl_context *ssl );
 
+/* set inbound transform of ssl context */
+void mbedtls_ssl_set_inbound_transform( mbedtls_ssl_context *ssl,
+                                        mbedtls_ssl_transform *transform );
+
+/* set outbound transform of ssl context */
+void mbedtls_ssl_set_outbound_transform( mbedtls_ssl_context *ssl,
+                                         mbedtls_ssl_transform *transform );
+
 int mbedtls_ssl_handshake_client_step( mbedtls_ssl_context *ssl );
 int mbedtls_ssl_handshake_server_step( mbedtls_ssl_context *ssl );
 void mbedtls_ssl_handshake_wrapup( mbedtls_ssl_context *ssl );

library/ssl_msg.c

diff --git a/library/ssl_msg.c b/library/ssl_msg.c
index 2fe801a..fcdd024 100644
--- a/library/ssl_msg.c
+++ b/library/ssl_msg.c
@@ -5551,6 +5551,23 @@
     mbedtls_platform_zeroize( transform, sizeof( mbedtls_ssl_transform ) );
 }
 
+void mbedtls_ssl_set_inbound_transform( mbedtls_ssl_context *ssl,
+                                        mbedtls_ssl_transform *transform )
+{
+    if( ssl->transform_in == transform )
+        return;
+
+    ssl->transform_in = transform;
+    mbedtls_platform_zeroize( ssl->in_ctr, 8 );
+}
+
+void mbedtls_ssl_set_outbound_transform( mbedtls_ssl_context *ssl,
+                                         mbedtls_ssl_transform *transform )
+{
+    ssl->transform_out = transform;
+    mbedtls_platform_zeroize( ssl->cur_out_ctr, 8 );
+}
+
 #if defined(MBEDTLS_SSL_PROTO_DTLS)
 
 void mbedtls_ssl_buffering_free( mbedtls_ssl_context *ssl )