Rooba
/
proxy.py
mirror of https://github.com/abhinavsingh/proxy.py.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Address ipv6 parsing from Http requests #65

This commit is contained in:
Abhinav Singh 2019-09-02 22:08:54 -07:00
parent c3cd9be816
commit 95671e4905
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
proxy.py
View File

@ -406,7 +406,8 @@ class HttpParser(object):
def set_host_port(self): def set_host_port(self):
if self.type == HttpParser.types.REQUEST_PARSER: if self.type == HttpParser.types.REQUEST_PARSER:
if self.method == b'CONNECT': if self.method == b'CONNECT':
self.host, self.port = self.url.path.split(COLON) u = urlparse.urlsplit(b'//' + self.url.path)
self.host, self.port = u.hostname, u.port
elif self.url: elif self.url:
self.host, self.port = self.url.hostname, self.url.port \ self.host, self.port = self.url.hostname, self.url.port \
if self.url.port else 80 if self.url.port else 80
@ -1139,7 +1140,7 @@ def setup_logger(log_file=DEFAULT_LOG_FILE, log_level=DEFAULT_LOG_LEVEL, log_for
if log_file: if log_file:
logging.basicConfig(filename=log_file, filemode='a', level=ll, format=log_format) logging.basicConfig(filename=log_file, filemode='a', level=ll, format=log_format)
else: else:
logging.basicConfig(level=log_level, format=log_format) logging.basicConfig(level=ll, format=log_format)
def init_parser() -> argparse.ArgumentParser: def init_parser() -> argparse.ArgumentParser: