From 216fd7b74484b408a98ba46dcd775d0aa570c2b8 Mon Sep 17 00:00:00 2001
From: cheatfate <ka@hardcore.kiev.ua>
Date: Tue, 24 Jul 2018 18:43:30 +0300
Subject: [PATCH] Fix disconnect in process handling.

---
 asyncdispatch2/transports/stream.nim | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/asyncdispatch2/transports/stream.nim b/asyncdispatch2/transports/stream.nim
index 1889db2e..ab8140a0 100644
--- a/asyncdispatch2/transports/stream.nim
+++ b/asyncdispatch2/transports/stream.nim
@@ -309,7 +309,7 @@ when defined(windows):
         elif int(err) == ERROR_OPERATION_ABORTED:
           # CancelIO() interrupt
           transp.state.incl(ReadPaused)
-        elif int(err) == ERROR_NETNAME_DELETED:
+        elif int(err) in {ERROR_NETNAME_DELETED, WSAECONNABORTED}:
           transp.state.incl({ReadEof, ReadPaused})
         else:
           transp.setReadError(err)
@@ -340,9 +340,10 @@ when defined(windows):
                 # CancelIO() interrupt
                 transp.state.excl(ReadPending)
                 transp.state.incl(ReadPaused)
-              elif int32(err) in {WSAECONNRESET, WSAENETRESET}:
+              elif int32(err) in {WSAECONNRESET, WSAENETRESET, WSAECONNABORTED}:
+                transp.state.excl(ReadPending)
+                transp.state.incl({ReadEof, ReadPaused})
                 if not isNil(transp.reader):
-                  transp.state = {ReadEof, ReadPaused}
                   transp.reader.complete()
                   transp.reader = nil
               elif int32(err) != ERROR_IO_PENDING:
@@ -613,13 +614,13 @@ else:
         if int(err) == EINTR:
           continue
         elif int(err) in {ECONNRESET}:
-          transp.state = transp.state + {ReadEof, ReadPaused}
+          transp.state.incl({ReadEof, ReadPaused})
           cdata.fd.removeReader()
         else:
           transp.setReadError(err)
           cdata.fd.removeReader()
       elif res == 0:
-        transp.state = transp.state + {ReadEof, ReadPaused}
+        transp.state.incl({ReadEof, ReadPaused})
         cdata.fd.removeReader()
       else:
         transp.offset += res
@@ -943,6 +944,7 @@ proc readExactly*(transp: StreamTransport, pbytes: pointer,
   ## will raise ``TransportIncompleteError``.
   checkClosed(transp)
   checkPending(transp)
+
   var index = 0
   while true:
     if transp.offset == 0: