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

Extend pseudo-header treatment to :status on responses

This commit is contained in:
Aldo Cortesi 2016-05-31 15:12:20 +12:00
parent 9ea68ebd28
commit 4de4223b2d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mitmproxy/protocol/http2.py
View File

@ -387,12 +387,14 @@ class Http2SingleStreamLayer(_HttpTransmissionLayer, threading.Thread):
self.response_arrived.wait() self.response_arrived.wait()
status_code = int(self.response_headers.get(':status', 502)) status_code = int(self.response_headers.get(':status', 502))
headers = self.response_headers.copy()
headers.clear(":status")
return HTTPResponse( return HTTPResponse(
http_version=b"HTTP/2.0", http_version=b"HTTP/2.0",
status_code=status_code, status_code=status_code,
reason='', reason='',
headers=self.response_headers, headers=headers,
content=None, content=None,
timestamp_start=self.timestamp_start, timestamp_start=self.timestamp_start,
timestamp_end=self.timestamp_end, timestamp_end=self.timestamp_end,
@ -412,10 +414,12 @@ class Http2SingleStreamLayer(_HttpTransmissionLayer, threading.Thread):
raise Http2ProtocolException("Zombie Stream") raise Http2ProtocolException("Zombie Stream")
def send_response_headers(self, response): def send_response_headers(self, response):
headers = response.headers.copy()
headers.insert(0, ":status", str(response.status_code))
self.client_conn.h2.safe_send_headers( self.client_conn.h2.safe_send_headers(
self.is_zombie, self.is_zombie,
self.client_stream_id, self.client_stream_id,
response.headers headers
) )
if self.zombie: # pragma: no cover if self.zombie: # pragma: no cover
raise Http2ProtocolException("Zombie Stream") raise Http2ProtocolException("Zombie Stream")