handle client connection errors

Signed-off-by: Achille Roussel <[email protected]>

Author: achille-roussel

src/dispatch/http.py

@@ -16,7 +16,7 @@
     overload,
 )
 
-from aiohttp import web
+from aiohttp import web, ClientConnectionError
 from http_message_signatures import InvalidSignature
 from typing_extensions import ParamSpec, TypeAlias
 
@@ -30,7 +30,10 @@
     parse_verification_key,
     verify_request,
 )
-from dispatch.status import Status
+from dispatch.status import Status, register_error_type
+
+# https://docs.aiohttp.org/en/stable/client_reference.html
+register_error_type(ClientConnectionError, Status.TCP_ERROR)
 
 logger = logging.getLogger(__name__)
 

src/dispatch/status.py

@@ -1,4 +1,5 @@
 import enum
+import ssl
 from typing import Any, Callable, Dict, Type, Union
 
 from dispatch.sdk.v1 import status_pb2 as status_pb
@@ -129,6 +130,8 @@ def status_for_error(error: BaseException) -> Status:
         # tend to be caused by invalid use of syscalls, which are
         # unlikely at higher abstraction levels.
         return Status.TEMPORARY_ERROR
+    elif isinstance(error, ssl.SSLError) or isinstance(error, ssl.CertificateError):
+        return Status.TLS_ERROR
     return Status.PERMANENT_ERROR