Merge 3.4 (asyncio)

Lib/asyncio/windows_events.py

@@ -126,14 +126,12 @@ class _BaseWaitHandleFuture(futures.Future):
             return
         self._registered = False
 
-        try:
+        wait_handle = self._wait_handle
-            _overlapped.UnregisterWait(self._wait_handle)
-        except OSError as exc:
         self._wait_handle = None
-            if exc.winerror == _overlapped.ERROR_IO_PENDING:
+        try:
-                # ERROR_IO_PENDING is not an error, the wait was unregistered
+            _overlapped.UnregisterWait(wait_handle)
-                self._unregister_wait_cb(None)
+        except OSError as exc:
-            elif exc.winerror != _overlapped.ERROR_IO_PENDING:
+            if exc.winerror != _overlapped.ERROR_IO_PENDING:
                 context = {
                     'message': 'Failed to unregister the wait handle',
                     'exception': exc,
@@ -142,8 +140,9 @@ class _BaseWaitHandleFuture(futures.Future):
                 if self._source_traceback:
                     context['source_traceback'] = self._source_traceback
                 self._loop.call_exception_handler(context)
-        else:
+                return
-            self._wait_handle = None
+            # ERROR_IO_PENDING means that the unregister is pending
+
         self._unregister_wait_cb(None)
 
     def cancel(self):
@@ -209,14 +208,12 @@ class _WaitHandleFuture(_BaseWaitHandleFuture):
             return
         self._registered = False
 
-        try:
+        wait_handle = self._wait_handle
-            _overlapped.UnregisterWaitEx(self._wait_handle, self._event)
-        except OSError as exc:
         self._wait_handle = None
-            if exc.winerror == _overlapped.ERROR_IO_PENDING:
+        try:
-                # ERROR_IO_PENDING is not an error, the wait was unregistered
+            _overlapped.UnregisterWaitEx(wait_handle, self._event)
-                self._unregister_wait_cb(None)
+        except OSError as exc:
-            elif exc.winerror != _overlapped.ERROR_IO_PENDING:
+            if exc.winerror != _overlapped.ERROR_IO_PENDING:
                 context = {
                     'message': 'Failed to unregister the wait handle',
                     'exception': exc,
@@ -225,10 +222,10 @@ class _WaitHandleFuture(_BaseWaitHandleFuture):
                 if self._source_traceback:
                     context['source_traceback'] = self._source_traceback
                 self._loop.call_exception_handler(context)
-        else:
+                return
-            self._wait_handle = None
+            # ERROR_IO_PENDING is not an error, the wait was unregistered
-            self._event_fut = self._proactor._wait_cancel(
+
-                                                self._event,
+        self._event_fut = self._proactor._wait_cancel(self._event,
                                                       self._unregister_wait_cb)
 
 
@@ -409,13 +406,21 @@ class IocpProactor:
         self._results = []
         return tmp
 
+    def _result(self, value):
+        fut = futures.Future(loop=self._loop)
+        fut.set_result(value)
+        return fut
+
     def recv(self, conn, nbytes, flags=0):
         self._register_with_iocp(conn)
         ov = _overlapped.Overlapped(NULL)
+        try:
             if isinstance(conn, socket.socket):
                 ov.WSARecv(conn.fileno(), nbytes, flags)
             else:
                 ov.ReadFile(conn.fileno(), nbytes)
+        except BrokenPipeError:
+            return self._result(b'')
 
         def finish_recv(trans, key, ov):
             try:
@@ -508,9 +513,7 @@ class IocpProactor:
             # ConnectNamePipe() failed with ERROR_PIPE_CONNECTED which means
             # that the pipe is connected. There is no need to wait for the
             # completion of the connection.
-            f = futures.Future(loop=self._loop)
+            return self._result(pipe)
-            f.set_result(pipe)
-            return f
 
         def finish_accept_pipe(trans, key, ov):
             ov.getresult()

Modules/overlapped.c

@@ -730,7 +730,7 @@ Overlapped_ReadFile(OverlappedObject *self, PyObject *args)
     switch (err) {
         case ERROR_BROKEN_PIPE:
             mark_as_completed(&self->overlapped);
-            Py_RETURN_NONE;
+            return SetFromWindowsErr(err);
         case ERROR_SUCCESS:
         case ERROR_MORE_DATA:
         case ERROR_IO_PENDING:
@@ -789,7 +789,7 @@ Overlapped_WSARecv(OverlappedObject *self, PyObject *args)
     switch (err) {
         case ERROR_BROKEN_PIPE:
             mark_as_completed(&self->overlapped);
-            Py_RETURN_NONE;
+            return SetFromWindowsErr(err);
         case ERROR_SUCCESS:
         case ERROR_MORE_DATA:
         case ERROR_IO_PENDING: