mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-09-03 15:55:37 +00:00
Code Issues Releases Wiki Activity

Implement Intermediate protocol using asyncio

Browse Source
This commit is contained in:
Dan
2018-06-11 12:25:30 +02:00
parent ead0b4f029
commit d64337bf90
1 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -28,19 +28,23 @@ class TCPIntermediate(TCP):
def __init__(self, proxy: dict):
super().__init__(proxy)
def connect(self, address: tuple):
super().connect(address)
super().sendall(b"\xee" * 4)
async def connect(self, address: tuple):
await super().connect(address)
await super().send(b"\xee" * 4)
log.info("Connected{}!".format(" with proxy" if self.proxy_enabled else ""))
log.info("Connected{}!".format(
" with proxy"
if self.proxy_enabled
else ""
))
def sendall(self, data: bytes, *args):
super().sendall(pack("<i", len(data)) + data)
async def send(self, data: bytes, *args):
await super().send(pack("<i", len(data)) + data)
def recvall(self, length: int = 0) -> bytes or None:
length = super().recvall(4)
async def recv(self, length: int = 0) -> bytes or None:
length = await super().recv(4)
if length is None:
return None
return super().recvall(unpack("<i", length)[0])
return await super().recv(unpack("<i", length)[0])