migrate to hyperframe

1 files changed, 19 insertions, 2 deletions

diff --git a/netlib/utils.py b/netlib/utils.py
index 66225897..c537754a 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):
+    field = rfile.peek(3)
+    length = int(codecs.encode(field, 'hex_codec'), 16)
+
+    if length == 4740180:
+        raise ValueError("Length field looks more like HTTP/1.1: %s" % rfile.peek(20))
+
+    raw = rfile.safe_read(9 + length)
+    return raw
+
+def http2_read_frame(rfile):
+    raw = http2_read_raw_frame(rfile)
+    frame, length = hyperframe.frame.Frame.parse_frame_header(raw[:9])
+    frame.parse_body(memoryview(raw[9:]))
+    return frame