From e8c1fca67c2ecce50010c1ff4b94dcf93ceb2c46 Mon Sep 17 00:00:00 2001 From: Jerry Yu <jerry.h.yu@arm.com> Date: Wed, 18 May 2022 14:48:56 +0800 Subject: [PATCH] move trafic set to generic Signed-off-by: Jerry Yu <jerry.h.yu@arm.com> --- library/ssl_tls13_client.c | 5 ----- library/ssl_tls13_generic.c | 7 +++++++ library/ssl_tls13_server.c | 3 --- 3 files changed, 7 insertions(+), 8 deletions(-) diff --git a/library/ssl_tls13_client.c b/library/ssl_tls13_client.c index 305ac0050b..657fb44d46 100644 --- a/library/ssl_tls13_client.c +++ b/library/ssl_tls13_client.c @@ -1844,11 +1844,6 @@ static int ssl_tls13_flush_buffers( mbedtls_ssl_context *ssl ) */ static int ssl_tls13_handshake_wrapup( mbedtls_ssl_context *ssl ) { - MBEDTLS_SSL_DEBUG_MSG( 1, ( "Switch to application keys for inbound traffic" ) ); - mbedtls_ssl_set_inbound_transform ( ssl, ssl->transform_application ); - - MBEDTLS_SSL_DEBUG_MSG( 1, ( "Switch to application keys for outbound traffic" ) ); - mbedtls_ssl_set_outbound_transform( ssl, ssl->transform_application ); mbedtls_ssl_tls13_handshake_wrapup( ssl ); diff --git a/library/ssl_tls13_generic.c b/library/ssl_tls13_generic.c index 9d924add62..ab2933e66f 100644 --- a/library/ssl_tls13_generic.c +++ b/library/ssl_tls13_generic.c @@ -1383,6 +1383,13 @@ void mbedtls_ssl_tls13_handshake_wrapup( mbedtls_ssl_context *ssl ) MBEDTLS_SSL_DEBUG_MSG( 3, ( "=> handshake wrapup" ) ); + MBEDTLS_SSL_DEBUG_MSG( 1, ( "Switch to application keys for inbound traffic" ) ); + mbedtls_ssl_set_inbound_transform ( ssl, ssl->transform_application ); + + MBEDTLS_SSL_DEBUG_MSG( 1, ( "Switch to application keys for outbound traffic" ) ); + mbedtls_ssl_set_outbound_transform( ssl, ssl->transform_application ); + + /* * Free the previous session and switch to the current one. */ diff --git a/library/ssl_tls13_server.c b/library/ssl_tls13_server.c index c6cedf08ed..c73b8bbcb3 100644 --- a/library/ssl_tls13_server.c +++ b/library/ssl_tls13_server.c @@ -1517,9 +1517,6 @@ static int ssl_tls13_handshake_wrapup( mbedtls_ssl_context *ssl ) { MBEDTLS_SSL_DEBUG_MSG( 2, ( "handshake: done" ) ); - MBEDTLS_SSL_DEBUG_MSG( 1, ( "Switch to application keys for all traffic" ) ); - mbedtls_ssl_set_inbound_transform ( ssl, ssl->transform_application ); - mbedtls_ssl_set_outbound_transform( ssl, ssl->transform_application ); mbedtls_ssl_tls13_handshake_wrapup( ssl ); mbedtls_ssl_handshake_set_state( ssl, MBEDTLS_SSL_HANDSHAKE_OVER ); return( 0 );