Merge pull request #46 from rigens/fix-socks5-reply-read

Fix socks5 reply reading

Author: romis2012

python_socks/_connectors/socks5_async.py

@@ -72,14 +72,15 @@ async def connect(
 
     # noinspection PyMethodMayBeStatic
     async def _read_reply(self, stream: AsyncSocketStream) -> bytes:
-        data = await stream.read_exact(4)
+        data = await stream.read_exact(3)
         if data[0] != socks5.SOCKS_VER:
             return data
         if data[1] != socks5.ReplyCode.SUCCEEDED:
             return data
         if data[2] != socks5.RSV:
             return data
 
+        data += await stream.read_exact(1)
         addr_type = data[3]
 
         if addr_type == socks5.AddressType.IPV4:

python_socks/_connectors/socks5_sync.py

@@ -63,14 +63,15 @@ def connect(
 
     # noinspection PyMethodMayBeStatic
     def _read_reply(self, stream: SyncSocketStream) -> bytes:
-        data = stream.read_exact(4)
+        data = stream.read_exact(3)
         if data[0] != socks5.SOCKS_VER:
             return data
         if data[1] != socks5.ReplyCode.SUCCEEDED:
             return data
         if data[2] != socks5.RSV:
             return data
 
+        data += stream.read_exact(1)
         addr_type = data[3]
 
         if addr_type == socks5.AddressType.IPV4: