python-socks 2.5.0__tar.gz → 2.5.1__tar.gz

{python_socks-2.5.0/python_socks.egg-info → python_socks-2.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-socks
-Version: 2.5.0
+Version: 2.5.1
 Summary: Core proxy (SOCKS4, SOCKS5, HTTP tunneling) functionality for Python
 Home-page: https://github.com/romis2012/python-socks
 Author: Roman Snegirev

{python_socks-2.5.0 → python_socks-2.5.1}/python_socks/_connectors/socks5_async.py

@@ -3,7 +3,6 @@ import socket
 from .._abc import AsyncSocketStream, AsyncResolver
 from .abc import AsyncConnector
 
-from .._protocols.errors import ReplyError
 from .._protocols import socks5
 from .._helpers import is_ip_address
 
@@ -66,15 +65,28 @@ class Socks5AsyncConnector(AsyncConnector):
         data = conn.send(request)
         await stream.write_all(data)
 
+        data = await self._read_reply(stream)
+        reply: socks5.ConnectReply = conn.receive(data)
+        return reply
+
+    # noinspection PyMethodMayBeStatic
+    async def _read_reply(self, stream: AsyncSocketStream) -> bytes:
         data = await stream.read_exact(4)
-        if data[3] == socks5.AddressType.IPV4:
+        if data[0] != socks5.SOCKS_VER:
+            return data
+        if data[1] != socks5.ReplyCode.SUCCEEDED:
+            return data
+        if data[2] != socks5.RSV:
+            return data
+
+        addr_type = data[3]
+
+        if addr_type == socks5.AddressType.IPV4:
             data += await stream.read_exact(6)
-        elif data[3] == socks5.AddressType.IPV6:
+        elif addr_type == socks5.AddressType.IPV6:
             data += await stream.read_exact(18)
-        elif data[3] == socks5.AddressType.DOMAIN:
-            data += await stream.read_exact(int.from_bytes(await stream.read_exact(1)) + 2)
-        else:  # pragma: no cover
-            raise ReplyError(f'Invalid address type: {data[3]:#02X}')
+        elif addr_type == socks5.AddressType.DOMAIN:
+            host_len, *_ = await stream.read_exact(1)
+            data += await stream.read_exact(host_len + 2)
 
-        reply: socks5.ConnectReply = conn.receive(data)
-        return reply
+        return data

{python_socks-2.5.0 → python_socks-2.5.1}/python_socks/_connectors/socks5_sync.py

@@ -3,7 +3,6 @@ import socket
 from .._abc import SyncSocketStream, SyncResolver
 from .abc import SyncConnector
 
-from .._protocols.errors import ReplyError
 from .._protocols import socks5
 from .._helpers import is_ip_address
 
@@ -57,15 +56,28 @@ class Socks5SyncConnector(SyncConnector):
         data = conn.send(request)
         stream.write_all(data)
 
+        data = self._read_reply(stream)
+        reply: socks5.ConnectReply = conn.receive(data)
+        return reply
+
+    # noinspection PyMethodMayBeStatic
+    def _read_reply(self, stream: SyncSocketStream) -> bytes:
         data = stream.read_exact(4)
-        if data[3] == socks5.AddressType.IPV4:
+        if data[0] != socks5.SOCKS_VER:
+            return data
+        if data[1] != socks5.ReplyCode.SUCCEEDED:
+            return data
+        if data[2] != socks5.RSV:
+            return data
+
+        addr_type = data[3]
+
+        if addr_type == socks5.AddressType.IPV4:
             data += stream.read_exact(6)
-        elif data[3] == socks5.AddressType.IPV6:
+        elif addr_type == socks5.AddressType.IPV6:
             data += stream.read_exact(18)
-        elif data[3] == socks5.AddressType.DOMAIN:
-            data += stream.read_exact(int.from_bytes(stream.read_exact(1)) + 2)
-        else:  # pragma: no cover
-            raise ReplyError(f'Invalid address type: {data[3]:#02X}')
+        elif addr_type == socks5.AddressType.DOMAIN:
+            host_len, *_ = stream.read_exact(1)
+            data += stream.read_exact(host_len + 2)
 
-        reply: socks5.ConnectReply = conn.receive(data)
-        return reply
+        return data

{python_socks-2.5.0 → python_socks-2.5.1}/python_socks/_version.py

@@ -1,2 +1,2 @@
 __title__ = 'python-socks'
-__version__ = '2.5.0'
+__version__ = '2.5.1'

{python_socks-2.5.0 → python_socks-2.5.1/python_socks.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-socks
-Version: 2.5.0
+Version: 2.5.1
 Summary: Core proxy (SOCKS4, SOCKS5, HTTP tunneling) functionality for Python
 Home-page: https://github.com/romis2012/python-socks
 Author: Roman Snegirev