LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-10 06:54:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

Avoid hardcoding the Host in HTTP mode

Browse Source
This commit is contained in:
Lonami Exo 2018-07-08 17:04:21 +02:00
parent 331b6a208f
commit e2c1efd0f1
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
telethon/network/connection/http.py
View File

@ -2,6 +2,17 @@ from .tcpfull import ConnectionTcpFull
class ConnectionHttp(ConnectionTcpFull): class ConnectionHttp(ConnectionTcpFull):
def __init__(self, *, loop, timeout, proxy=None):
super().__init__(loop=loop, timeout=timeout, proxy=proxy)
self._host = None
async def connect(self, ip, port):
if port != 80:
port = 80 # HTTP without TLS needs port 80
self._host = '{}:{}'.format(ip, port)
return await super().connect(ip, port)
async def recv(self): async def recv(self):
while True: while True:
line = await self._read_line() line = await self._read_line()
@ -20,10 +31,10 @@ class ConnectionHttp(ConnectionTcpFull):
async def send(self, message): async def send(self, message):
await self.write( await self.write(
'POST /api HTTP/1.1\r\n' 'POST /api HTTP/1.1\r\n'
'Host: 149.154.167.91:80\r\n' 'Host: {}\r\n'
'Content-Type: application/x-www-form-urlencoded\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n'
'Connection: keep-alive\r\n' 'Connection: keep-alive\r\n'
'Keep-Alive: timeout=100000, max=10000000\r\n' 'Keep-Alive: timeout=100000, max=10000000\r\n'
'Content-Length: {}\r\n\r\n'.format(len(message)) 'Content-Length: {}\r\n\r\n'.format(self._host, len(message))
.encode('ascii') + message .encode('ascii') + message
) )