Support gzip compression in ClientSession._ws_connect() if specifie…

…d by server
aio-libs · Nov 17, 2024 · 23a4480 · 23a4480
1 parent aab680d
commit 23a4480
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/CHANGES/9933.feature.rst b/CHANGES/9933.feature.rst
@@ -0,0 +1 @@
+Provided experimental support for communicating with WebSocket servers that require the `permessage-gzip` extension -- by :user:`jakob-keller`.
diff --git a/aiohttp/client.py b/aiohttp/client.py
@@ -9,6 +9,7 @@
 import sys
 import traceback
 import warnings
+import zlib
 from contextlib import suppress
 from types import TracebackType
 from typing import (
@@ -1005,7 +1006,9 @@ async def _ws_connect(
                 compress_hdrs = resp.headers.get(hdrs.SEC_WEBSOCKET_EXTENSIONS)
                 if compress_hdrs:
                     try:
-                        compress, notakeover = ws_ext_parse(compress_hdrs)
+                        compress, notakeover = ws_ext_parse(
+                            compress_hdrs, detect_gzip=True
+                        )
                     except WSHandshakeError as exc:
                         raise WSServerHandshakeError(
                             resp.request_info,
@@ -1036,7 +1039,12 @@ async def _ws_connect(
             transport = conn.transport
             assert transport is not None
             reader = WebSocketDataQueue(conn_proto, 2**16, loop=self._loop)
-            conn_proto.set_parser(WebSocketReader(reader, max_msg_size), reader)
+            conn_proto.set_parser(
+                WebSocketReader(
+                    reader, max_msg_size, compress=compress <= zlib.MAX_WBITS or "gzip"
+                ),
+                reader,
+            )
             writer = WebSocketWriter(
                 conn_proto,
                 transport,