Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

move body assembly to netlib

Browse Source
This commit is contained in:
Maximilian Hils 2015-09-17 02:40:08 +02:00
parent 14a5f405fd
commit 12984f149c
1 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libmproxy/protocol/http.py
View File

@ -91,21 +91,12 @@ class Http1Layer(_StreamingHttpLayer):
return http1.read_body(self.server_conn.rfile, expected_size, self.config.body_size_limit) return http1.read_body(self.server_conn.rfile, expected_size, self.config.body_size_limit)
def send_response_headers(self, response): def send_response_headers(self, response):
raw = http1.assemble_response_head(response, preserve_transfer_encoding=True) raw = http1.assemble_response_head(response)
self.client_conn.wfile.write(raw) self.client_conn.wfile.write(raw)
self.client_conn.wfile.flush() self.client_conn.wfile.flush()
def send_response_body(self, response, chunks): def send_response_body(self, response, chunks):
if b"chunked" in response.headers.get(b"transfer-encoding", b"").lower(): for chunk in http1.assemble_body(response.headers, chunks):
# TODO: Move this into netlib.http.http1
chunks = itertools.chain(
(
"{:x}\r\n{}\r\n".format(len(chunk), chunk)
for chunk in chunks if chunk
),
("0\r\n\r\n",)
)
for chunk in chunks:
self.client_conn.wfile.write(chunk) self.client_conn.wfile.write(chunk)
self.client_conn.wfile.flush() self.client_conn.wfile.flush()