udp errors should result in protocol.error_received (#601)

MagicStack · Aug 26, 2024 · 3c3bbef · 3c3bbef
1 parent 4083a94
commit 3c3bbef
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/uvloop/handles/udp.pyx b/uvloop/handles/udp.pyx
@@ -244,16 +244,15 @@ cdef class UDPTransport(UVBaseTransport):
                 ctx.close()
 
                 exc = convert_error(err)
-                self._fatal_error(exc, True)
+                if isinstance(exc, OSError):
+                    run_in_context1(self.context.copy(), self._protocol.error_received, exc)
+                else:
+                    self._fatal_error(exc, True)
             else:
                 self._maybe_pause_protocol()
 
         else:
-            if err < 0:
-                exc = convert_error(err)
-                self._fatal_error(exc, True)
-            else:
-                self._on_sent(None, self.context.copy())
+            self._on_sent(convert_error(err) if err < 0 else None, self.context.copy())
 
     cdef _on_receive(self, bytes data, object exc, object addr):
         if exc is None: