commit 66d9e6f4058a925cd1c76c1ffa5f9b6d2e342d59
author Jerry Yu <jerry.h.yu@arm.com> Sat May 07 10:50:12 2022 +0800
committer Jerry Yu <jerry.h.yu@arm.com> Mon May 09 15:49:11 2022 +0800
tree f4f8b1243e7a2f8f20434dcbf944ed622897531b
parent 483305683322b1fd573ec578d97646cd29e60848

refactor next state of client hello

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

library/ssl_tls13_server.c

diff --git a/library/ssl_tls13_server.c b/library/ssl_tls13_server.c
index 3c57f81..baf8c61 100644
--- a/library/ssl_tls13_server.c
+++ b/library/ssl_tls13_server.c
@@ -743,10 +743,21 @@
     parse_client_hello_ret = ret;
     MBEDTLS_SSL_PROC_CHK( ssl_tls13_postprocess_client_hello( ssl ) );
 
-    if( parse_client_hello_ret == SSL_CLIENT_HELLO_OK )
-        mbedtls_ssl_handshake_set_state( ssl, MBEDTLS_SSL_SERVER_HELLO );
-    else
-        mbedtls_ssl_handshake_set_state( ssl, MBEDTLS_SSL_HELLO_RETRY_REQUEST );
+    switch( parse_client_hello_ret )
+    {
+        case SSL_CLIENT_HELLO_OK:
+            mbedtls_ssl_handshake_set_state( ssl, MBEDTLS_SSL_SERVER_HELLO );
+            break;
+
+        case SSL_CLIENT_HELLO_HRR_REQUIRED:
+            mbedtls_ssl_handshake_set_state( ssl, MBEDTLS_SSL_HELLO_RETRY_REQUEST );
+            break;
+
+        default:
+            MBEDTLS_SSL_DEBUG_MSG( 2, ( "should never happen" ) );
+            ret = parse_client_hello_ret;
+            break;
+    }
 
 cleanup: