Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests 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):
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)

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

@ -24,8 +24,8 @@ log = logging.getLogger(__name__)
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)

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

@ -28,8 +28,9 @@ log = logging.getLogger(__name__)
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

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

@ -26,8 +26,9 @@ log = logging.getLogger(__name__)
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):

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

@ -25,8 +25,8 @@ log = logging.getLogger(__name__)
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)

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

@ -29,8 +29,9 @@ log = logging.getLogger(__name__)
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