diff --git a/rocksock.py b/rocksock.py index 1a70476..1edac4c 100644 --- a/rocksock.py +++ b/rocksock.py @@ -325,7 +325,7 @@ class Rocksock(): if v4a: buf += '\0\0\0\x01' else: - af, sa = self._resolve(dest, True) + af, sa = self._resolve(dest.hostinfo, True) if af != socket.AF_INET: raise RocksockException(RS_E_SOCKS4_NO_IP6, failedproxy=-1) buf += self._ip_to_bytes(sa[0]) buf += '\0' @@ -406,7 +406,7 @@ class Rocksock(): self._connect_socks4(s4a, pnum) except RocksockException as e: if e.get_error() == RS_E_TARGETPROXY_CONNECT_FAILED: - s4 = self._setup_socks4_header(self, False, curr) + s4 = self._setup_socks4_header(False, curr) self._connect_socks4(s4a, pnum) else: raise e elif prev.type == RS_PT_SOCKS5: