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

http2: implement basic flow control updates

Browse Source
This commit is contained in:
Thomas Kriechbaumer 2015-08-20 10:21:38 +02:00
parent eb34305518
commit 94b7beae2a
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
netlib/http/http2/protocol.py
View File

@ -251,6 +251,9 @@ class HTTP2Protocol(semantics.ProtocolMixin):
if isinstance(frm, frame.SettingsFrame) and not frm.flags & frame.Frame.FLAG_ACK:
self._apply_settings(frm.settings, hide)
if isinstance(frm, frame.DataFrame) and frm.length > 0:
self._update_flow_control_window(frm.stream_id, frm.length)
return frm
def check_alpn(self):
@ -309,6 +312,12 @@ class HTTP2Protocol(semantics.ProtocolMixin):
# be liberal in what we expect from the other end
# to be more strict use: self._read_settings_ack(hide)
def _update_flow_control_window(self, stream_id, increment):
frm = frame.WindowUpdateFrame(stream_id=0, window_size_increment=increment)
self.send_frame(frm)
frm = frame.WindowUpdateFrame(stream_id=stream_id, window_size_increment=increment)
self.send_frame(frm)
def _create_headers(self, headers, stream_id, end_stream=True):
def frame_cls(chunks):
for i in chunks:
@ -351,8 +360,6 @@ class HTTP2Protocol(semantics.ProtocolMixin):
payload=body[i:i+chunk_size]) for i in chunks]
frms[-1].flags = frame.Frame.FLAG_END_STREAM
# TODO: implement flow-control window
if self.dump_frames: # pragma no cover
for frm in frms:
print(frm.human_readable(">>"))
@ -369,8 +376,10 @@ class HTTP2Protocol(semantics.ProtocolMixin):
while True:
frm = self.read_frame()
if isinstance(frm, frame.HeadersFrame)\
or isinstance(frm, frame.ContinuationFrame):
if (
(isinstance(frm, frame.HeadersFrame) or isinstance(frm, frame.ContinuationFrame)) and
(stream_id == 0 or frm.stream_id == stream_id)
):
stream_id = frm.stream_id
header_block_fragment += frm.header_block_fragment
if frm.flags & frame.Frame.FLAG_END_STREAM:
@ -382,15 +391,13 @@ class HTTP2Protocol(semantics.ProtocolMixin):
while body_expected:
frm = self.read_frame()
if isinstance(frm, frame.DataFrame):
if isinstance(frm, frame.DataFrame) and frm.stream_id == stream_id:
body += frm.payload
if frm.flags & frame.Frame.FLAG_END_STREAM:
break
else:
self._handle_unexpected_frame(frm)
# TODO: implement window update & flow
headers = odict.ODictCaseless()
for header, value in self.decoder.decode(header_block_fragment):
headers.add(header, value)