TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-17 21:22:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make the underlying TCP protocol accept ipv6 addresses

Browse Source
This commit is contained in:
Dan 2018-06-13 13:36:26 +02:00
parent efe26bcb19
commit 56748ff390
6 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyrogram/connection/transport/tcp/tcp.py
View File

@ -33,8 +33,9 @@ log = logging.getLogger(__name__)
class TCP(socks.socksocket): class TCP(socks.socksocket):
def __init__(self, proxy: dict): def __init__(self, ipv6: bool, proxy: dict):
super().__init__() super().__init__(family=socket.AF_INET6 if ipv6 else socket.AF_INET)
self.settimeout(10) self.settimeout(10)
self.proxy_enabled = proxy.get("enabled", False) self.proxy_enabled = proxy.get("enabled", False)

4
pyrogram/connection/transport/tcp/tcp_abridged.py
View File

@ -24,8 +24,8 @@ log = logging.getLogger(__name__)
class TCPAbridged(TCP): class TCPAbridged(TCP):
def __init__(self, proxy: dict): def __init__(self, ipv6: bool, proxy: dict):
super().__init__(proxy) super().__init__(ipv6, proxy)
def connect(self, address: tuple): def connect(self, address: tuple):
super().connect(address) super().connect(address)

5
pyrogram/connection/transport/tcp/tcp_abridged_o.py
View File

@ -28,8 +28,9 @@ log = logging.getLogger(__name__)
class TCPAbridgedO(TCP): class TCPAbridgedO(TCP):
RESERVED = (b"HEAD", b"POST", b"GET ", b"OPTI", b"\xee" * 4) RESERVED = (b"HEAD", b"POST", b"GET ", b"OPTI", b"\xee" * 4)
def __init__(self, proxy: dict): def __init__(self, ipv6: bool, proxy: dict):
super().__init__(proxy) super().__init__(ipv6, proxy)
self.encrypt = None self.encrypt = None
self.decrypt = None self.decrypt = None

5
pyrogram/connection/transport/tcp/tcp_full.py
View File

@ -26,8 +26,9 @@ log = logging.getLogger(__name__)
class TCPFull(TCP): class TCPFull(TCP):
def __init__(self, proxy: dict): def __init__(self, ipv6: bool, proxy: dict):
super().__init__(proxy) super().__init__(ipv6, proxy)
self.seq_no = None self.seq_no = None
def connect(self, address: tuple): def connect(self, address: tuple):

4
pyrogram/connection/transport/tcp/tcp_intermediate.py
View File

@ -25,8 +25,8 @@ log = logging.getLogger(__name__)
class TCPIntermediate(TCP): class TCPIntermediate(TCP):
def __init__(self, proxy: dict): def __init__(self, ipv6: bool, proxy: dict):
super().__init__(proxy) super().__init__(ipv6, proxy)
def connect(self, address: tuple): def connect(self, address: tuple):
super().connect(address) super().connect(address)

5
pyrogram/connection/transport/tcp/tcp_intermediate_o.py
View File

@ -29,8 +29,9 @@ log = logging.getLogger(__name__)
class TCPIntermediateO(TCP): class TCPIntermediateO(TCP):
RESERVED = (b"HEAD", b"POST", b"GET ", b"OPTI", b"\xee" * 4) RESERVED = (b"HEAD", b"POST", b"GET ", b"OPTI", b"\xee" * 4)
def __init__(self, proxy: dict): def __init__(self, ipv6: bool, proxy: dict):
super().__init__(proxy) super().__init__(ipv6, proxy)
self.encrypt = None self.encrypt = None
self.decrypt = None self.decrypt = None