Forráskód Böngészése

WinTLS: Fix potential infinite loop

Tatsuhiro Tsujikawa 10 éve
szülő
commit
925b5cec07
1 módosított fájl, 4 hozzáadás és 1 törlés
  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;
     }
   }