소스 검색

WinTLS: Fix potential infinite loop

Tatsuhiro Tsujikawa 10 년 전
부모
커밋
925b5cec07
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  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;
     }
   }