fix: bytearray may be passed into lru_cached func

sethmlarson · Sep 7, 2022 · caa0223 · caa0223
1 parent bad06ab
commit caa0223
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/socksio/socks4.py b/socksio/socks4.py
@@ -193,6 +193,9 @@ def loads(cls, data: bytes) -> "SOCKS4Reply":
         Raises:
             ProtocolError: If the data does not match the spec.
         """
+        if isinstance(data, bytearray):
+            data = bytes(data)
+
         if len(data) != 8 or data[0:1] != b"\x00":
             raise ProtocolError("Malformed reply")
 

diff --git a/socksio/socks5.py b/socksio/socks5.py
@@ -250,6 +250,9 @@ def loads(cls, data: bytes) -> "SOCKS5Reply":
         Raises:
             ProtocolError: If the data does not match the spec.
         """
+        if isinstance(data, bytearray):
+            data = bytes(data)
+
         if data[0:1] != b"\x05":
             raise ProtocolError("Malformed reply")