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

pathoc: Guess the Host header from the path if possible

This commit is contained in:
Aldo Cortesi 2016-10-19 11:41:42 +13:00
parent ceb8caee98
commit 85015fe561
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pathod/language/http.py
View File

@ -2,6 +2,7 @@ import abc
import pyparsing as pp
from netlib.http import url
import netlib.websockets
from netlib.http import status_codes, user_agents
from . import base, exceptions, actions, message
@ -318,7 +319,7 @@ class Request(_HTTPMessage):
)
)
if not self.raw:
if not get_header("Content-Length", self.headers):
if not get_header(b"Content-Length", self.headers):
if self.body:
length = sum(
len(i) for i in self.body.values(settings)
@ -330,11 +331,21 @@ class Request(_HTTPMessage):
)
)
if settings.request_host:
if not get_header("Host", self.headers):
if not get_header(b"Host", self.headers):
h = settings.request_host
if self.path:
path = b"".join(self.path.values({})).decode(
"ascii", errors="ignore"
)
try:
_, h, _, _ = url.parse(path)
h = h.decode("ascii", errors="ignore")
except ValueError:
pass
tokens.append(
Header(
base.TokValueLiteral("Host"),
base.TokValueLiteral(settings.request_host)
base.TokValueLiteral(h)
)
)
intermediate = self.__class__(tokens)