aboutsummaryrefslogtreecommitdiffstats
path: root/netlib
diff options
context:
space:
mode:
authorThomas Kriechbaumer <Kriechi@users.noreply.github.com>2016-02-01 20:28:14 +0100
committerThomas Kriechbaumer <Kriechi@users.noreply.github.com>2016-02-01 20:28:14 +0100
commit81b32cf42629dcbe8f59633dcb9b62816b781968 (patch)
treea0aa45506d52dc1cd273b77997022d04e7338f12 /netlib
parent1e203401261ab6b24ae193f81dc4256854de13d8 (diff)
parente98c729bb9b0d3debde6f61c948108bdc9dbafbe (diff)
downloadmitmproxy-81b32cf42629dcbe8f59633dcb9b62816b781968.tar.gz
mitmproxy-81b32cf42629dcbe8f59633dcb9b62816b781968.tar.bz2
mitmproxy-81b32cf42629dcbe8f59633dcb9b62816b781968.zip
Merge pull request #116 from Kriechi/hyperframe
migrate to hyperframe
Diffstat (limited to 'netlib')
-rw-r--r--netlib/http/http2/connections.py119
-rw-r--r--netlib/http/http2/frame.py651
-rw-r--r--netlib/utils.py21
3 files changed, 84 insertions, 707 deletions
diff --git a/netlib/http/http2/connections.py b/netlib/http/http2/connections.py
index c493abe6..91133121 100644
--- a/netlib/http/http2/connections.py
+++ b/netlib/http/http2/connections.py
@@ -5,7 +5,13 @@ import time
from hpack.hpack import Encoder, Decoder
from ... import utils
from .. import Headers, Response, Request
-from . import frame
+
+from hyperframe import frame
+
+# TODO: remove once hyperframe released a new version > 3.1.1
+# wrapper for deprecated name in old hyperframe release
+frame.SettingsFrame.MAX_FRAME_SIZE = frame.SettingsFrame.SETTINGS_MAX_FRAME_SIZE
+frame.SettingsFrame.MAX_HEADER_LIST_SIZE = frame.SettingsFrame.SETTINGS_MAX_HEADER_LIST_SIZE
class TCPHandler(object):
@@ -34,7 +40,16 @@ class HTTP2Protocol(object):
HTTP_1_1_REQUIRED=0xd
)
- CLIENT_CONNECTION_PREFACE = "PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n"
+ CLIENT_CONNECTION_PREFACE = b'PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n'
+
+ HTTP2_DEFAULT_SETTINGS = {
+ frame.SettingsFrame.HEADER_TABLE_SIZE: 4096,
+ frame.SettingsFrame.ENABLE_PUSH: 1,
+ frame.SettingsFrame.MAX_CONCURRENT_STREAMS: None,
+ frame.SettingsFrame.INITIAL_WINDOW_SIZE: 2 ** 16 - 1,
+ frame.SettingsFrame.MAX_FRAME_SIZE: 2 ** 14,
+ frame.SettingsFrame.MAX_HEADER_LIST_SIZE: None,
+ }
def __init__(
self,
@@ -54,7 +69,7 @@ class HTTP2Protocol(object):
self.decoder = decoder or Decoder()
self.unhandled_frame_cb = unhandled_frame_cb
- self.http2_settings = frame.HTTP2_DEFAULT_SETTINGS.copy()
+ self.http2_settings = self.HTTP2_DEFAULT_SETTINGS.copy()
self.current_stream_id = None
self.connection_preface_performed = False
@@ -84,7 +99,7 @@ class HTTP2Protocol(object):
timestamp_end = time.time()
- authority = headers.get(':authority', '')
+ authority = headers.get(':authority', b'')
method = headers.get(':method', 'GET')
scheme = headers.get(':scheme', 'https')
path = headers.get(':path', '/')
@@ -103,6 +118,8 @@ class HTTP2Protocol(object):
form_in = "absolute"
# FIXME: verify if path or :host contains what we need
scheme, host, port, _ = utils.parse_url(path)
+ scheme = scheme.decode('ascii')
+ host = host.decode('ascii')
if host is None:
host = 'localhost'
@@ -112,18 +129,17 @@ class HTTP2Protocol(object):
request = Request(
form_in,
- method,
- scheme,
- host,
+ method.encode('ascii'),
+ scheme.encode('ascii'),
+ host.encode('ascii'),
port,
- path,
- (2, 0),
+ path.encode('ascii'),
+ b'2.0',
headers,
body,
timestamp_start,
timestamp_end,
)
- # FIXME: We should not do this.
request.stream_id = stream_id
return request
@@ -131,7 +147,7 @@ class HTTP2Protocol(object):
def read_response(
self,
__rfile,
- request_method='',
+ request_method=b'',
body_size_limit=None,
include_body=True,
stream_id=None,
@@ -160,9 +176,9 @@ class HTTP2Protocol(object):
timestamp_end = None
response = Response(
- (2, 0),
+ b'2.0',
int(headers.get(':status', 502)),
- "",
+ b'',
headers,
body,
timestamp_start=timestamp_start,
@@ -190,13 +206,13 @@ class HTTP2Protocol(object):
headers = request.headers.copy()
if ':authority' not in headers:
- headers.fields.insert(0, (':authority', bytes(authority)))
+ headers.fields.insert(0, (b':authority', authority.encode('ascii')))
if ':scheme' not in headers:
- headers.fields.insert(0, (':scheme', bytes(request.scheme)))
+ headers.fields.insert(0, (b':scheme', request.scheme.encode('ascii')))
if ':path' not in headers:
- headers.fields.insert(0, (':path', bytes(request.path)))
+ headers.fields.insert(0, (b':path', request.path.encode('ascii')))
if ':method' not in headers:
- headers.fields.insert(0, (':method', bytes(request.method)))
+ headers.fields.insert(0, (b':method', request.method.encode('ascii')))
if hasattr(request, 'stream_id'):
stream_id = request.stream_id
@@ -213,7 +229,7 @@ class HTTP2Protocol(object):
headers = response.headers.copy()
if ':status' not in headers:
- headers.fields.insert(0, (':status', bytes(str(response.status_code))))
+ headers.fields.insert(0, (b':status', str(response.status_code).encode('ascii')))
if hasattr(response, 'stream_id'):
stream_id = response.stream_id
@@ -240,9 +256,9 @@ class HTTP2Protocol(object):
magic = self.tcp_handler.rfile.safe_read(magic_length)
assert magic == self.CLIENT_CONNECTION_PREFACE
- frm = frame.SettingsFrame(state=self, settings={
- frame.SettingsFrame.SETTINGS.SETTINGS_ENABLE_PUSH: 0,
- frame.SettingsFrame.SETTINGS.SETTINGS_MAX_CONCURRENT_STREAMS: 1,
+ frm = frame.SettingsFrame(settings={
+ frame.SettingsFrame.ENABLE_PUSH: 0,
+ frame.SettingsFrame.MAX_CONCURRENT_STREAMS: 1,
})
self.send_frame(frm, hide=True)
self._receive_settings(hide=True)
@@ -253,12 +269,12 @@ class HTTP2Protocol(object):
self.tcp_handler.wfile.write(self.CLIENT_CONNECTION_PREFACE)
- self.send_frame(frame.SettingsFrame(state=self), hide=True)
+ self.send_frame(frame.SettingsFrame(), hide=True)
self._receive_settings(hide=True) # server announces own settings
self._receive_settings(hide=True) # server acks my settings
def send_frame(self, frm, hide=False):
- raw_bytes = frm.to_bytes()
+ raw_bytes = frm.serialize()
self.tcp_handler.wfile.write(raw_bytes)
self.tcp_handler.wfile.flush()
if not hide and self.dump_frames: # pragma no cover
@@ -266,19 +282,19 @@ class HTTP2Protocol(object):
def read_frame(self, hide=False):
while True:
- frm = frame.Frame.from_file(self.tcp_handler.rfile, self)
+ frm = utils.http2_read_frame(self.tcp_handler.rfile)
if not hide and self.dump_frames: # pragma no cover
print(frm.human_readable("<<"))
if isinstance(frm, frame.PingFrame):
- raw_bytes = frame.PingFrame(flags=frame.Frame.FLAG_ACK, payload=frm.payload).to_bytes()
+ raw_bytes = frame.PingFrame(flags=['ACK'], payload=frm.payload).serialize()
self.tcp_handler.wfile.write(raw_bytes)
self.tcp_handler.wfile.flush()
continue
- if isinstance(frm, frame.SettingsFrame) and not frm.flags & frame.Frame.FLAG_ACK:
+ if isinstance(frm, frame.SettingsFrame) and 'ACK' not in frm.flags:
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)
+ if isinstance(frm, frame.DataFrame) and frm.flow_controlled_length > 0:
+ self._update_flow_control_window(frm.stream_id, frm.flow_controlled_length)
return frm
def check_alpn(self):
@@ -321,15 +337,13 @@ class HTTP2Protocol(object):
old_value = '-'
self.http2_settings[setting] = value
- frm = frame.SettingsFrame(
- state=self,
- flags=frame.Frame.FLAG_ACK)
+ frm = frame.SettingsFrame(flags=['ACK'])
self.send_frame(frm, hide)
def _update_flow_control_window(self, stream_id, increment):
- frm = frame.WindowUpdateFrame(stream_id=0, window_size_increment=increment)
+ frm = frame.WindowUpdateFrame(stream_id=0, window_increment=increment)
self.send_frame(frm)
- frm = frame.WindowUpdateFrame(stream_id=stream_id, window_size_increment=increment)
+ frm = frame.WindowUpdateFrame(stream_id=stream_id, window_increment=increment)
self.send_frame(frm)
def _create_headers(self, headers, stream_id, end_stream=True):
@@ -342,43 +356,40 @@ class HTTP2Protocol(object):
header_block_fragment = self.encoder.encode(headers.fields)
- chunk_size = self.http2_settings[frame.SettingsFrame.SETTINGS.SETTINGS_MAX_FRAME_SIZE]
+ chunk_size = self.http2_settings[frame.SettingsFrame.MAX_FRAME_SIZE]
chunks = range(0, len(header_block_fragment), chunk_size)
frms = [frm_cls(
- state=self,
- flags=frame.Frame.FLAG_NO_FLAGS,
+ flags=[],
stream_id=stream_id,
- header_block_fragment=header_block_fragment[i:i+chunk_size]) for frm_cls, i in frame_cls(chunks)]
+ data=header_block_fragment[i:i+chunk_size]) for frm_cls, i in frame_cls(chunks)]
- last_flags = frame.Frame.FLAG_END_HEADERS
+ frms[-1].flags.add('END_HEADERS')
if end_stream:
- last_flags |= frame.Frame.FLAG_END_STREAM
- frms[-1].flags = last_flags
+ frms[0].flags.add('END_STREAM')
if self.dump_frames: # pragma no cover
for frm in frms:
print(frm.human_readable(">>"))
- return [frm.to_bytes() for frm in frms]
+ return [frm.serialize() for frm in frms]
def _create_body(self, body, stream_id):
if body is None or len(body) == 0:
return b''
- chunk_size = self.http2_settings[frame.SettingsFrame.SETTINGS.SETTINGS_MAX_FRAME_SIZE]
+ chunk_size = self.http2_settings[frame.SettingsFrame.MAX_FRAME_SIZE]
chunks = range(0, len(body), chunk_size)
frms = [frame.DataFrame(
- state=self,
- flags=frame.Frame.FLAG_NO_FLAGS,
+ flags=[],
stream_id=stream_id,
- payload=body[i:i+chunk_size]) for i in chunks]
- frms[-1].flags = frame.Frame.FLAG_END_STREAM
+ data=body[i:i+chunk_size]) for i in chunks]
+ frms[-1].flags.add('END_STREAM')
if self.dump_frames: # pragma no cover
for frm in frms:
print(frm.human_readable(">>"))
- return [frm.to_bytes() for frm in frms]
+ return [frm.serialize() for frm in frms]
def _receive_transmission(self, stream_id=None, include_body=True):
if not include_body:
@@ -386,7 +397,7 @@ class HTTP2Protocol(object):
body_expected = True
- header_block_fragment = b''
+ header_blocks = b''
body = b''
while True:
@@ -396,10 +407,10 @@ class HTTP2Protocol(object):
(stream_id is None 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:
+ header_blocks += frm.data
+ if 'END_STREAM' in frm.flags:
body_expected = False
- if frm.flags & frame.Frame.FLAG_END_HEADERS:
+ if 'END_HEADERS' in frm.flags:
break
else:
self._handle_unexpected_frame(frm)
@@ -407,14 +418,14 @@ class HTTP2Protocol(object):
while body_expected:
frm = self.read_frame()
if isinstance(frm, frame.DataFrame) and frm.stream_id == stream_id:
- body += frm.payload
- if frm.flags & frame.Frame.FLAG_END_STREAM:
+ body += frm.data
+ if 'END_STREAM' in frm.flags:
break
else:
self._handle_unexpected_frame(frm)
headers = Headers(
- [[str(k), str(v)] for k, v in self.decoder.decode(header_block_fragment)]
+ [[k.encode('ascii'), v.encode('ascii')] for k, v in self.decoder.decode(header_blocks)]
)
return stream_id, headers, body
diff --git a/netlib/http/http2/frame.py b/netlib/http/http2/frame.py
deleted file mode 100644
index 188629d4..00000000
--- a/netlib/http/http2/frame.py
+++ /dev/null
@@ -1,651 +0,0 @@
-from __future__ import absolute_import, print_function, division
-import struct
-from hpack.hpack import Encoder, Decoder
-
-from ...utils import BiDi
-from ...exceptions import HttpSyntaxException
-
-
-ERROR_CODES = BiDi(
- NO_ERROR=0x0,
- PROTOCOL_ERROR=0x1,
- INTERNAL_ERROR=0x2,
- FLOW_CONTROL_ERROR=0x3,
- SETTINGS_TIMEOUT=0x4,
- STREAM_CLOSED=0x5,
- FRAME_SIZE_ERROR=0x6,
- REFUSED_STREAM=0x7,
- CANCEL=0x8,
- COMPRESSION_ERROR=0x9,
- CONNECT_ERROR=0xa,
- ENHANCE_YOUR_CALM=0xb,
- INADEQUATE_SECURITY=0xc,
- HTTP_1_1_REQUIRED=0xd
-)
-
-CLIENT_CONNECTION_PREFACE = b"PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n"
-
-class Frame(object):
-
- """
- Baseclass Frame
- contains header
- payload is defined in subclasses
- """
-
- FLAG_NO_FLAGS = 0x0
- FLAG_ACK = 0x1
- FLAG_END_STREAM = 0x1
- FLAG_END_HEADERS = 0x4
- FLAG_PADDED = 0x8
- FLAG_PRIORITY = 0x20
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=FLAG_NO_FLAGS,
- stream_id=0x0):
- valid_flags = 0
- for flag in self.VALID_FLAGS:
- valid_flags |= flag
- if flags | valid_flags != valid_flags:
- raise ValueError('invalid flags detected.')
-
- if state is None:
- class State(object):
- pass
-
- state = State()
- state.http2_settings = HTTP2_DEFAULT_SETTINGS.copy()
- state.encoder = Encoder()
- state.decoder = Decoder()
-
- self.state = state
-
- self.length = length
- self.type = self.TYPE
- self.flags = flags
- self.stream_id = stream_id
-
- @classmethod
- def _check_frame_size(cls, length, state):
- if state:
- settings = state.http2_settings
- else:
- settings = HTTP2_DEFAULT_SETTINGS.copy()
-
- max_frame_size = settings[
- SettingsFrame.SETTINGS.SETTINGS_MAX_FRAME_SIZE]
-
- if length > max_frame_size:
- raise HttpSyntaxException(
- "Frame size exceeded: %d, but only %d allowed." % (
- length, max_frame_size))
-
- @classmethod
- def from_file(cls, fp, state=None):
- """
- read a HTTP/2 frame sent by a server or client
- fp is a "file like" object that could be backed by a network
- stream or a disk or an in memory stream reader
- """
- raw_header = fp.safe_read(9)
-
- fields = struct.unpack("!HBBBL", raw_header)
- length = (fields[0] << 8) + fields[1]
- flags = fields[3]
- stream_id = fields[4]
-
- if raw_header[:4] == b'HTTP': # pragma no cover
- raise HttpSyntaxException("Expected HTTP2 Frame, got HTTP/1 connection")
-
- cls._check_frame_size(length, state)
-
- payload = fp.safe_read(length)
- return FRAMES[fields[2]].from_bytes(
- state,
- length,
- flags,
- stream_id,
- payload)
-
- def to_bytes(self):
- payload = self.payload_bytes()
- self.length = len(payload)
-
- self._check_frame_size(self.length, self.state)
-
- b = struct.pack('!HB', (self.length & 0xFFFF00) >> 8, self.length & 0x0000FF)
- b += struct.pack('!B', self.TYPE)
- b += struct.pack('!B', self.flags)
- b += struct.pack('!L', self.stream_id & 0x7FFFFFFF)
- b += payload
-
- return b
-
- def payload_bytes(self): # pragma: no cover
- raise NotImplementedError()
-
- def payload_human_readable(self): # pragma: no cover
- raise NotImplementedError()
-
- def human_readable(self, direction="-"):
- self.length = len(self.payload_bytes())
-
- return "\n".join([
- "%s: %s | length: %d | flags: %#x | stream_id: %d" % (
- direction, self.__class__.__name__, self.length, self.flags, self.stream_id),
- self.payload_human_readable(),
- "===============================================================",
- ])
-
- def __eq__(self, other):
- return self.to_bytes() == other.to_bytes()
-
-
-class DataFrame(Frame):
- TYPE = 0x0
- VALID_FLAGS = [Frame.FLAG_END_STREAM, Frame.FLAG_PADDED]
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- payload=b'',
- pad_length=0):
- super(DataFrame, self).__init__(state, length, flags, stream_id)
- self.payload = payload
- self.pad_length = pad_length
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- if f.flags & Frame.FLAG_PADDED:
- f.pad_length = struct.unpack('!B', payload[0])[0]
- f.payload = payload[1:-f.pad_length]
- else:
- f.payload = payload
-
- return f
-
- def payload_bytes(self):
- if self.stream_id == 0x0:
- raise ValueError('DATA frames MUST be associated with a stream.')
-
- b = b''
- if self.flags & self.FLAG_PADDED:
- b += struct.pack('!B', self.pad_length)
-
- b += bytes(self.payload)
-
- if self.flags & self.FLAG_PADDED:
- b += b'\0' * self.pad_length
-
- return b
-
- def payload_human_readable(self):
- return "payload: %s" % str(self.payload)
-
-
-class HeadersFrame(Frame):
- TYPE = 0x1
- VALID_FLAGS = [
- Frame.FLAG_END_STREAM,
- Frame.FLAG_END_HEADERS,
- Frame.FLAG_PADDED,
- Frame.FLAG_PRIORITY]
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- header_block_fragment=b'',
- pad_length=0,
- exclusive=False,
- stream_dependency=0x0,
- weight=0):
- super(HeadersFrame, self).__init__(state, length, flags, stream_id)
-
- self.header_block_fragment = header_block_fragment
- self.pad_length = pad_length
- self.exclusive = exclusive
- self.stream_dependency = stream_dependency
- self.weight = weight
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- if f.flags & Frame.FLAG_PADDED:
- f.pad_length = struct.unpack('!B', payload[0])[0]
- f.header_block_fragment = payload[1:-f.pad_length]
- else:
- f.header_block_fragment = payload[0:]
-
- if f.flags & Frame.FLAG_PRIORITY:
- f.stream_dependency, f.weight = struct.unpack(
- '!LB', f.header_block_fragment[:5])
- f.exclusive = bool(f.stream_dependency >> 31)
- f.stream_dependency &= 0x7FFFFFFF
- f.header_block_fragment = f.header_block_fragment[5:]
-
- return f
-
- def payload_bytes(self):
- if self.stream_id == 0x0:
- raise ValueError('HEADERS frames MUST be associated with a stream.')
-
- b = b''
- if self.flags & self.FLAG_PADDED:
- b += struct.pack('!B', self.pad_length)
-
- if self.flags & self.FLAG_PRIORITY:
- b += struct.pack('!LB',
- (int(self.exclusive) << 31) | self.stream_dependency,
- self.weight)
-
- b += self.header_block_fragment
-
- if self.flags & self.FLAG_PADDED:
- b += b'\0' * self.pad_length
-
- return b
-
- def payload_human_readable(self):
- s = []
-
- if self.flags & self.FLAG_PRIORITY:
- s.append("exclusive: %d" % self.exclusive)
- s.append("stream dependency: %#x" % self.stream_dependency)
- s.append("weight: %d" % self.weight)
-
- if self.flags & self.FLAG_PADDED:
- s.append("padding: %d" % self.pad_length)
-
- s.append(
- "header_block_fragment: %s" %
- self.header_block_fragment.encode('hex'))
-
- return "\n".join(s)
-
-
-class PriorityFrame(Frame):
- TYPE = 0x2
- VALID_FLAGS = []
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- exclusive=False,
- stream_dependency=0x0,
- weight=0):
- super(PriorityFrame, self).__init__(state, length, flags, stream_id)
- self.exclusive = exclusive
- self.stream_dependency = stream_dependency
- self.weight = weight
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- f.stream_dependency, f.weight = struct.unpack('!LB', payload)
- f.exclusive = bool(f.stream_dependency >> 31)
- f.stream_dependency &= 0x7FFFFFFF
-
- return f
-
- def payload_bytes(self):
- if self.stream_id == 0x0:
- raise ValueError(
- 'PRIORITY frames MUST be associated with a stream.')
-
- return struct.pack(
- '!LB',
- (int(
- self.exclusive) << 31) | self.stream_dependency,
- self.weight)
-
- def payload_human_readable(self):
- s = []
- s.append("exclusive: %d" % self.exclusive)
- s.append("stream dependency: %#x" % self.stream_dependency)
- s.append("weight: %d" % self.weight)
- return "\n".join(s)
-
-
-class RstStreamFrame(Frame):
- TYPE = 0x3
- VALID_FLAGS = []
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- error_code=0x0):
- super(RstStreamFrame, self).__init__(state, length, flags, stream_id)
- self.error_code = error_code
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
- f.error_code = struct.unpack('!L', payload)[0]
- return f
-
- def payload_bytes(self):
- if self.stream_id == 0x0:
- raise ValueError(
- 'RST_STREAM frames MUST be associated with a stream.')
-
- return struct.pack('!L', self.error_code)
-
- def payload_human_readable(self):
- return "error code: %#x" % self.error_code
-
-
-class SettingsFrame(Frame):
- TYPE = 0x4
- VALID_FLAGS = [Frame.FLAG_ACK]
-
- SETTINGS = BiDi(
- SETTINGS_HEADER_TABLE_SIZE=0x1,
- SETTINGS_ENABLE_PUSH=0x2,
- SETTINGS_MAX_CONCURRENT_STREAMS=0x3,
- SETTINGS_INITIAL_WINDOW_SIZE=0x4,
- SETTINGS_MAX_FRAME_SIZE=0x5,
- SETTINGS_MAX_HEADER_LIST_SIZE=0x6,
- )
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- settings=None):
- super(SettingsFrame, self).__init__(state, length, flags, stream_id)
-
- if settings is None:
- settings = {}
-
- self.settings = settings
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- for i in range(0, len(payload), 6):
- identifier, value = struct.unpack("!HL", payload[i:i + 6])
- f.settings[identifier] = value
-
- return f
-
- def payload_bytes(self):
- if self.stream_id != 0x0:
- raise ValueError(
- 'SETTINGS frames MUST NOT be associated with a stream.')
-
- b = b''
- for identifier, value in self.settings.items():
- b += struct.pack("!HL", identifier & 0xFF, value)
-
- return b
-
- def payload_human_readable(self):
- s = []
-
- for identifier, value in self.settings.items():
- s.append("%s: %#x" % (self.SETTINGS.get_name(identifier), value))
-
- if not s:
- return "settings: None"
- else:
- return "\n".join(s)
-
-
-class PushPromiseFrame(Frame):
- TYPE = 0x5
- VALID_FLAGS = [Frame.FLAG_END_HEADERS, Frame.FLAG_PADDED]
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- promised_stream=0x0,
- header_block_fragment=b'',
- pad_length=0):
- super(PushPromiseFrame, self).__init__(state, length, flags, stream_id)
- self.pad_length = pad_length
- self.promised_stream = promised_stream
- self.header_block_fragment = header_block_fragment
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- if f.flags & Frame.FLAG_PADDED:
- f.pad_length, f.promised_stream = struct.unpack('!BL', payload[:5])
- f.header_block_fragment = payload[5:-f.pad_length]
- else:
- f.promised_stream = int(struct.unpack("!L", payload[:4])[0])
- f.header_block_fragment = payload[4:]
-
- f.promised_stream &= 0x7FFFFFFF
-
- return f
-
- def payload_bytes(self):
- if self.stream_id == 0x0:
- raise ValueError(
- 'PUSH_PROMISE frames MUST be associated with a stream.')
-
- if self.promised_stream == 0x0:
- raise ValueError('Promised stream id not valid.')
-
- b = b''
- if self.flags & self.FLAG_PADDED:
- b += struct.pack('!B', self.pad_length)
-
- b += struct.pack('!L', self.promised_stream & 0x7FFFFFFF)
- b += bytes(self.header_block_fragment)
-
- if self.flags & self.FLAG_PADDED:
- b += b'\0' * self.pad_length
-
- return b
-
- def payload_human_readable(self):
- s = []
-
- if self.flags & self.FLAG_PADDED:
- s.append("padding: %d" % self.pad_length)
-
- s.append("promised stream: %#x" % self.promised_stream)
- s.append(
- "header_block_fragment: %s" %
- self.header_block_fragment.encode('hex'))
-
- return "\n".join(s)
-
-
-class PingFrame(Frame):
- TYPE = 0x6
- VALID_FLAGS = [Frame.FLAG_ACK]
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- payload=b''):
- super(PingFrame, self).__init__(state, length, flags, stream_id)
- self.payload = payload
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
- f.payload = payload
- return f
-
- def payload_bytes(self):
- if self.stream_id != 0x0:
- raise ValueError(
- 'PING frames MUST NOT be associated with a stream.')
-
- b = self.payload[0:8]
- b += b'\0' * (8 - len(b))
- return b
-
- def payload_human_readable(self):
- return "opaque data: %s" % str(self.payload)
-
-
-class GoAwayFrame(Frame):
- TYPE = 0x7
- VALID_FLAGS = []
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- last_stream=0x0,
- error_code=0x0,
- data=b''):
- super(GoAwayFrame, self).__init__(state, length, flags, stream_id)
- self.last_stream = last_stream
- self.error_code = error_code
- self.data = data
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- f.last_stream, f.error_code = struct.unpack("!LL", payload[:8])
- f.last_stream &= 0x7FFFFFFF
- f.data = payload[8:]
-
- return f
-
- def payload_bytes(self):
- if self.stream_id != 0x0:
- raise ValueError(
- 'GOAWAY frames MUST NOT be associated with a stream.')
-
- b = struct.pack('!LL', self.last_stream & 0x7FFFFFFF, self.error_code)
- b += bytes(self.data)
- return b
-
- def payload_human_readable(self):
- s = []
- s.append("last stream: %#x" % self.last_stream)
- s.append("error code: %d" % self.error_code)
- s.append("debug data: %s" % str(self.data))
- return "\n".join(s)
-
-
-class WindowUpdateFrame(Frame):
- TYPE = 0x8
- VALID_FLAGS = []
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- window_size_increment=0x0):
- super(WindowUpdateFrame, self).__init__(state, length, flags, stream_id)
- self.window_size_increment = window_size_increment
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
-
- f.window_size_increment = struct.unpack("!L", payload)[0]
- f.window_size_increment &= 0x7FFFFFFF
-
- return f
-
- def payload_bytes(self):
- if self.window_size_increment <= 0 or self.window_size_increment >= 2 ** 31:
- raise ValueError(
- 'Window Size Increment MUST be greater than 0 and less than 2^31.')
-
- return struct.pack('!L', self.window_size_increment & 0x7FFFFFFF)
-
- def payload_human_readable(self):
- return "window size increment: %#x" % self.window_size_increment
-
-
-class ContinuationFrame(Frame):
- TYPE = 0x9
- VALID_FLAGS = [Frame.FLAG_END_HEADERS]
-
- def __init__(
- self,
- state=None,
- length=0,
- flags=Frame.FLAG_NO_FLAGS,
- stream_id=0x0,
- header_block_fragment=b''):
- super(ContinuationFrame, self).__init__(state, length, flags, stream_id)
- self.header_block_fragment = header_block_fragment
-
- @classmethod
- def from_bytes(cls, state, length, flags, stream_id, payload):
- f = cls(state=state, length=length, flags=flags, stream_id=stream_id)
- f.header_block_fragment = payload
- return f
-
- def payload_bytes(self):
- if self.stream_id == 0x0:
- raise ValueError(
- 'CONTINUATION frames MUST be associated with a stream.')
-
- return self.header_block_fragment
-
- def payload_human_readable(self):
- s = []
- s.append(
- "header_block_fragment: %s" %
- self.header_block_fragment.encode('hex'))
- return "\n".join(s)
-
-_FRAME_CLASSES = [
- DataFrame,
- HeadersFrame,
- PriorityFrame,
- RstStreamFrame,
- SettingsFrame,
- PushPromiseFrame,
- PingFrame,
- GoAwayFrame,
- WindowUpdateFrame,
- ContinuationFrame
-]
-FRAMES = {cls.TYPE: cls for cls in _FRAME_CLASSES}
-
-
-HTTP2_DEFAULT_SETTINGS = {
- SettingsFrame.SETTINGS.SETTINGS_HEADER_TABLE_SIZE: 4096,
- SettingsFrame.SETTINGS.SETTINGS_ENABLE_PUSH: 1,
- SettingsFrame.SETTINGS.SETTINGS_MAX_CONCURRENT_STREAMS: None,
- SettingsFrame.SETTINGS.SETTINGS_INITIAL_WINDOW_SIZE: 2 ** 16 - 1,
- SettingsFrame.SETTINGS.SETTINGS_MAX_FRAME_SIZE: 2 ** 14,
- SettingsFrame.SETTINGS.SETTINGS_MAX_HEADER_LIST_SIZE: None,
-}
diff --git a/netlib/utils.py b/netlib/utils.py
index 66225897..1c1b617a 100644
--- a/netlib/utils.py
+++ b/netlib/utils.py
@@ -2,12 +2,12 @@ from __future__ import absolute_import, print_function, division
import os.path
import re
import string
+import codecs
import unicodedata
-
import six
from six.moves import urllib
-
+import hyperframe
def always_bytes(unicode_or_bytes, *encode_args):
if isinstance(unicode_or_bytes, six.text_type):
@@ -366,3 +366,20 @@ def multipartdecode(headers, content):
r.append((key, value))
return r
return []
+
+
+def http2_read_raw_frame(rfile):
+ header = rfile.safe_read(9)
+ length = int(codecs.encode(header[:3], 'hex_codec'), 16)
+
+ if length == 4740180:
+ raise ValueError("Length field looks more like HTTP/1.1: %s" % rfile.peek(20))
+
+ body = rfile.safe_read(length)
+ return [header, body]
+
+def http2_read_frame(rfile):
+ header, body = http2_read_raw_frame(rfile)
+ frame, length = hyperframe.frame.Frame.parse_frame_header(header)
+ frame.parse_body(memoryview(body))
+ return frame