protocol: support URLs with port specified

This commit is contained in:
dece 2021-02-15 18:52:37 +01:00
parent 8cfc266e48
commit 1d51a2ca40

View file

@ -60,6 +60,11 @@ class Request:
self.state = Request.STATE_INVALID_URL self.state = Request.STATE_INVALID_URL
return False return False
hostname = url_parts["host"] hostname = url_parts["host"]
if ":" in hostname:
hostname, port = hostname.split(":", maxsplit=1)
port = int(port)
else:
port = 1965
try: try:
self.payload = self.url.encode() self.payload = self.url.encode()
@ -69,7 +74,7 @@ class Request:
self.payload += LINE_TERM self.payload += LINE_TERM
context = Request.get_ssl_context() context = Request.get_ssl_context()
sock = socket.create_connection((hostname, 1965)) sock = socket.create_connection((hostname, port))
self.ssock = context.wrap_socket(sock) self.ssock = context.wrap_socket(sock)
der = self.ssock.getpeercert(binary_form=True) der = self.ssock.getpeercert(binary_form=True)
self.cert_status, self.cert = \ self.cert_status, self.cert = \