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

prepare #4470

Browse Source
This commit is contained in:
Maximilian Hils 2021-03-07 16:16:49 +01:00
parent 93de96a720
commit 1e793ced24
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
mitmproxy/proxy/layers/http/__init__.py
View File

@ -200,18 +200,24 @@ class HttpStream(layer.Layer):
self.flow.request.headers.pop("expect")
if self.flow.request.stream:
if self.flow.response:
raise NotImplementedError("Can't set a response and enable streaming at the same time.")
yield HttpRequestHook(self.flow)
ok = yield from self.make_server_connection()
if not ok:
return
yield SendHttp(event, self.context.server)
self.client_state = self.state_stream_request_body
yield from self.start_request_stream()
else:
self.client_state = self.state_consume_request_body
self.server_state = self.state_wait_for_response_headers
def start_request_stream(self) -> layer.CommandGenerator[None]:
if self.flow.response:
raise NotImplementedError("Can't set a response and enable streaming at the same time.")
yield HttpRequestHook(self.flow)
ok = yield from self.make_server_connection()
if not ok:
return
yield SendHttp(
RequestHeaders(self.stream_id, self.flow.request, end_stream=False),
self.context.server
)
self.client_state = self.state_stream_request_body
@expect(RequestData, RequestEndOfMessage)
def state_stream_request_body(self, event: Union[RequestData, RequestEndOfMessage]) -> layer.CommandGenerator[None]:
if isinstance(event, RequestData):