Procházet zdrojové kódy

WinTLS: Fix potential infinite loop

Tatsuhiro Tsujikawa před 10 roky
rodič
revize
925b5cec07
1 změnil soubory, kde provedl 4 přidání a 1 odebrání
  1. 4 1
      src/WinTLSSession.cc

+ 4 - 1
src/WinTLSSession.cc

@@ -250,7 +250,10 @@ int WinTLSSession::closeConnection()
   // Send remaining data.
   while (writeBuf_.size()) {
     int rv = writeData(nullptr, 0);
-    if (rv == TLS_ERR_WOULDBLOCK) {
+    if (rv == 0) {
+      break;
+    }
+    if (rv < 0) {
       return rv;
     }
   }