Make the underlying TCP protocol accept ipv6 addresses

delivrance · delivrance · commit 56748ff39075 · 2018-06-13T13:36:26.000+02:00
diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py
@@ -33,8 +33,9 @@
 
 
 class TCP(socks.socksocket):
-    def __init__(self, proxy: dict):
-        super().__init__()
+    def __init__(self, ipv6: bool, proxy: dict):
+        super().__init__(family=socket.AF_INET6 if ipv6 else socket.AF_INET)
+
         self.settimeout(10)
         self.proxy_enabled = proxy.get("enabled", False)
 
diff --git a/pyrogram/connection/transport/tcp/tcp_abridged.py b/pyrogram/connection/transport/tcp/tcp_abridged.py
@@ -24,8 +24,8 @@
 
 
 class TCPAbridged(TCP):
-    def __init__(self, proxy: dict):
-        super().__init__(proxy)
+    def __init__(self, ipv6: bool, proxy: dict):
+        super().__init__(ipv6, proxy)
 
     def connect(self, address: tuple):
         super().connect(address)
diff --git a/pyrogram/connection/transport/tcp/tcp_abridged_o.py b/pyrogram/connection/transport/tcp/tcp_abridged_o.py
@@ -28,8 +28,9 @@
 class TCPAbridgedO(TCP):
     RESERVED = (b"HEAD", b"POST", b"GET ", b"OPTI", b"\xee" * 4)
 
-    def __init__(self, proxy: dict):
-        super().__init__(proxy)
+    def __init__(self, ipv6: bool, proxy: dict):
+        super().__init__(ipv6, proxy)
+
         self.encrypt = None
         self.decrypt = None
 
diff --git a/pyrogram/connection/transport/tcp/tcp_full.py b/pyrogram/connection/transport/tcp/tcp_full.py
@@ -26,8 +26,9 @@
 
 
 class TCPFull(TCP):
-    def __init__(self, proxy: dict):
-        super().__init__(proxy)
+    def __init__(self, ipv6: bool, proxy: dict):
+        super().__init__(ipv6, proxy)
+
         self.seq_no = None
 
     def connect(self, address: tuple):
diff --git a/pyrogram/connection/transport/tcp/tcp_intermediate.py b/pyrogram/connection/transport/tcp/tcp_intermediate.py
@@ -25,8 +25,8 @@
 
 
 class TCPIntermediate(TCP):
-    def __init__(self, proxy: dict):
-        super().__init__(proxy)
+    def __init__(self, ipv6: bool, proxy: dict):
+        super().__init__(ipv6, proxy)
 
     def connect(self, address: tuple):
         super().connect(address)
diff --git a/pyrogram/connection/transport/tcp/tcp_intermediate_o.py b/pyrogram/connection/transport/tcp/tcp_intermediate_o.py
@@ -29,8 +29,9 @@
 class TCPIntermediateO(TCP):
     RESERVED = (b"HEAD", b"POST", b"GET ", b"OPTI", b"\xee" * 4)
 
-    def __init__(self, proxy: dict):
-        super().__init__(proxy)
+    def __init__(self, ipv6: bool, proxy: dict):
+        super().__init__(ipv6, proxy)
+
         self.encrypt = None
         self.decrypt = None
 
