Add sni_hostname extension

encode · May 22, 2023 · 1e4d1c9 · 1e4d1c9
1 parent 6528f56
commit 1e4d1c9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/httpcore/_async/connection.py b/httpcore/_async/connection.py
@@ -93,6 +93,7 @@ async def handle_async_request(self, request: Request) -> Response:
 
     async def _connect(self, request: Request) -> AsyncNetworkStream:
         timeouts = request.extensions.get("timeout", {})
+        sni_hostname = request.extensions.get("sni_hostname", None)
         timeout = timeouts.get("connect", None)
 
         retries_left = self._retries
@@ -138,7 +139,8 @@ async def _connect(self, request: Request) -> AsyncNetworkStream:
 
                     kwargs = {
                         "ssl_context": ssl_context,
-                        "server_hostname": self._origin.host.decode("ascii"),
+                        "server_hostname": sni_hostname
+                        or self._origin.host.decode("ascii"),
                         "timeout": timeout,
                     }
                     async with Trace("connection.start_tls", request, kwargs) as trace:

diff --git a/httpcore/_sync/connection.py b/httpcore/_sync/connection.py
@@ -93,6 +93,7 @@ def handle_request(self, request: Request) -> Response:
 
     def _connect(self, request: Request) -> NetworkStream:
         timeouts = request.extensions.get("timeout", {})
+        sni_hostname = request.extensions.get("sni_hostname", None)
         timeout = timeouts.get("connect", None)
 
         retries_left = self._retries
@@ -138,7 +139,8 @@ def _connect(self, request: Request) -> NetworkStream:
 
                     kwargs = {
                         "ssl_context": ssl_context,
-                        "server_hostname": self._origin.host.decode("ascii"),
+                        "server_hostname": sni_hostname
+                        or self._origin.host.decode("ascii"),
                         "timeout": timeout,
                     }
                     with Trace("connection.start_tls", request, kwargs) as trace: