commit e3623920cf0649fd0b26952ddd27446b5966a138
author valord577 <valord577@gmail.com> Wed Feb 15 19:01:16 2023 +0800
committer Dave Rodgman <dave.rodgman@arm.com> Tue May 16 16:47:09 2023 +0100
tree a2f6252cb13ecdef47c16d90da974b0a0df18eb0
parent 40fe1cd364daaef821f1a9150718c2c16f0dd03e

send debug msg if contains '\n'

Signed-off-by: valord577 <valord577@gmail.com>

library/debug.c

diff --git a/library/debug.c b/library/debug.c
index eecde10..839ceba 100644
--- a/library/debug.c
+++ b/library/debug.c
@@ -68,6 +68,7 @@
     va_list argp;
     char str[DEBUG_BUF_SIZE];
     int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
+    int newline = -1;
 
     if (NULL == ssl              ||
         NULL == ssl->conf        ||
@@ -80,16 +81,26 @@
     ret = mbedtls_vsnprintf(str, DEBUG_BUF_SIZE, format, argp);
     va_end(argp);
 
-    if (ret >= 0 && ret < DEBUG_BUF_SIZE - 1) {
-        str[ret]     = '\n';
-        str[ret + 1] = '\0';
-    }
-    else
-    {
-        str[DEBUG_BUF_SIZE - 2] = '\n';
+    if (DEBUG_BUF_SIZE >= 2) {
+        if (ret < 0) {
+            newline = 0;
+        } else {
+            newline = ret;
+            if (ret >= DEBUG_BUF_SIZE - 1) {
+                newline = DEBUG_BUF_SIZE - 2;
+            }
+        }
     }
 
-    debug_send_line(ssl, level, file, line, str);
+    /*
+     * Send if str contains '\n'.
+     */
+    if (newline >= 0) {
+        str[newline]     = '\n';
+        str[newline + 1] = '\0';
+
+        debug_send_line(ssl, level, file, line, str);
+    }
 }
 
 void mbedtls_debug_print_ret(const mbedtls_ssl_context *ssl, int level,