Improve base64 decoding safety in byte responses

Author: diprog

async_tls_client/__version__.py

@@ -6,6 +6,6 @@
 
 __title__ = "async_tls_client"
 __description__ = "Asyncio fork of Python-TLS-Client with updated dependencies"
-__version__ = "1.0.10"
+__version__ = "1.1"
 __author__ = "Diprog (fork), Florian Zager (original)"
 __license__ = "MIT"

async_tls_client/response.py

@@ -72,8 +72,16 @@ def build_response(res: Union[dict, list], res_cookies: RequestsCookieJar) -> Re
     response.headers = response_headers
     # Add cookies
     response.cookies = res_cookies
+
+    # Decode the byte-response (base64 format)
+    try:
+        data_part = res["body"].split(',', 1)[1]
+    except IndexError:
+        import logging
+        logging.warning("Invalid base64 response format")
+        data_part = res["body"]
     # Add response body
-    response.text = base64.b64decode(res["body"].split(',')[1]).decode(errors='ignore')
+    response.text = base64.b64decode(data_part).decode(errors='ignore')
     # Add response content (bytes)
-    response._content = base64.b64decode(res["body"].split(',')[1])
+    response._content = base64.b64decode(data_part)
     return response