commit 0aead12706a1d340c014bfd1edc93b1aa0746638
author Ronald Cron <ronald.cron@arm.com> Wed Feb 21 14:46:56 2024 +0100
committer Ronald Cron <ronald.cron@arm.com> Wed Feb 21 17:37:33 2024 +0100
tree ba839c6814c7ba53dc795c724b74849a9863e8ac
parent b4fd47e8976585dc948e84135685a225f31c9925

ssl_client2: Improve loop writing early data

Signed-off-by: Ronald Cron <ronald.cron@arm.com>

programs/ssl/ssl_client2.c

diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index bcf44b7..0597298 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -3050,7 +3050,7 @@
                 while ((ret = mbedtls_ssl_write_early_data(&ssl, buf + written,
                                                            len - written)) < 0) {
                     if (ret == MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA) {
-                        break;
+                        goto end_of_early_data;
                     }
                     if (ret != MBEDTLS_ERR_SSL_WANT_READ &&
                         ret != MBEDTLS_ERR_SSL_WANT_WRITE &&
@@ -3069,15 +3069,14 @@
 #endif
                     }
                 }
-                if (ret == MBEDTLS_ERR_SSL_CANNOT_WRITE_EARLY_DATA) {
-                    break;
-                }
 
                 frags++;
                 written += ret;
             } while (written < len);
         }
 
+end_of_early_data:
+
         buf[written] = '\0';
         mbedtls_printf(
             " %" MBEDTLS_PRINTF_SIZET " bytes of early data written in %" MBEDTLS_PRINTF_SIZET " fragments\n\n%s\n",