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

simplify streaming http layer

Browse Source
This commit is contained in:
Maximilian Hils 2015-09-10 10:20:11 +02:00
parent 61f4319491
commit cf2b2e0cc7
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
libmproxy/protocol/http.py
View File

@ -45,12 +45,23 @@ class _StreamingHttpLayer(_HttpLayer):
raise NotImplementedError() raise NotImplementedError()
yield "this is a generator" # pragma: no cover yield "this is a generator" # pragma: no cover
def read_response(self, request_method):
response = self.read_response_headers()
response.body = "".join(
self.read_response_body(response.headers, request_method, response.code)
)
return response
def send_response_headers(self, response): def send_response_headers(self, response):
raise NotImplementedError raise NotImplementedError
def send_response_body(self, response, chunks): def send_response_body(self, response, chunks):
raise NotImplementedError() raise NotImplementedError()
def send_response(self, response):
self.send_response_headers(response)
self.send_response_body(response, response.body)
class Http1Layer(_StreamingHttpLayer): class Http1Layer(_StreamingHttpLayer):
def __init__(self, ctx, mode): def __init__(self, ctx, mode):
@ -68,17 +79,6 @@ class Http1Layer(_StreamingHttpLayer):
def send_request(self, request): def send_request(self, request):
self.server_conn.send(self.server_protocol.assemble(request)) self.server_conn.send(self.server_protocol.assemble(request))
def read_response(self, request_method):
return HTTPResponse.from_protocol(
self.server_protocol,
request_method=request_method,
body_size_limit=self.config.body_size_limit,
include_body=True
)
def send_response(self, response):
self.client_conn.send(self.client_protocol.assemble(response))
def read_response_headers(self): def read_response_headers(self):
return HTTPResponse.from_protocol( return HTTPResponse.from_protocol(
self.server_protocol, self.server_protocol,