enforce a maximum header line length

This commit is contained in:
jprjr 2020-05-16 13:58:33 +00:00
parent 3769ddebc6
commit 0b79cd174f
1 changed files with 8 additions and 3 deletions

11
av98.py
View File

@ -334,9 +334,14 @@ you'll be able to transparently follow links to Gopherspace!""")
address, f = None, open(gi.path, "rb")
else:
address, f = self._send_request(gi)
# Read response header
header = f.readline()
header = header.decode("UTF-8").strip()
# Spec dictates <META> should not exceed 1024 bytes
# but does not dictate a total maximum header length.
header = f.readline(2048)
header = header.decode("UTF-8")
if header[-1] != '\n':
raise RuntimeError("Received invalid header from server!")
header = header.strip()
self._debug("Response header: %s." % header)
# Catch network errors which may happen on initial connection