From 72b55c00fe66fffffc0b1538d2e0d366ad583f8e Mon Sep 17 00:00:00 2001
From: "496148326@qq.com" <496148326@qq.com>
Date: Wed, 20 Sep 2023 15:51:58 +0000
Subject: [PATCH] when use SSL_IO mode to encrypt channel, SSL_write return
 SSL_ERROR_WANT_WRITE will return 0 bytes and connection will be closed

---
 deps/ussl-hook/ssl/ssl_config.c | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/deps/ussl-hook/ssl/ssl_config.c b/deps/ussl-hook/ssl/ssl_config.c
index 7d30b3e29..efabaf441 100644
--- a/deps/ussl-hook/ssl/ssl_config.c
+++ b/deps/ussl-hook/ssl/ssl_config.c
@@ -813,9 +813,10 @@ ssize_t writev_regard_ssl(int fildes, const struct iovec *iov, int iovcnt)
       wbytes = 0;
       for (i = 0; (i < iovcnt) && (0 == ret); i++) {
         ERR_clear_error();
-        int ret = 0;
         n = SSL_write(ssl, iov[i].iov_base, iov[i].iov_len);
-        if (n <= 0) {
+        if (n > 0) {
+          wbytes += n;
+        } else {
           int ssl_error = SSL_get_error(ssl, n);
           if (SSL_ERROR_WANT_WRITE == ssl_error) {
             errno = EAGAIN;
@@ -827,10 +828,21 @@ ssize_t writev_regard_ssl(int fildes, const struct iovec *iov, int iovcnt)
             ussl_log_error("SSL_write failed, fd:%d, reason:%s", fildes,
                           ERR_error_string(ERR_get_error(), NULL));
           }
+          // errno: EIO, need destroy connection
+          // errno: EAGAIN:
+          // (1) wbytes larger than 0 (means already send some data successfully), just return wbytes
+          // (2) wbytes equal to zero (means send the first iov), wbytes equals to n (-1 means socket buffer
+          // temporarily unwritable, 0 means need destroy connection)
+          if (EIO == errno) {
+            wbytes = -1;
+          } else if (EAGAIN == errno) {
+            if (wbytes > 0) {
+            } else {
+              wbytes += n;
+            }
+          }
           ret = ENODATA;
-          continue;
         }
-        wbytes += n;
       }
     }
   }