aboutsummaryrefslogtreecommitdiffstats
path: root/pathod/language/http2.py
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2016-02-18 13:03:40 +0100
committerMaximilian Hils <git@maximilianhils.com>2016-02-18 13:03:40 +0100
commitd33d3663ecb166461d9cb5a78a29b44ee7a8fbb7 (patch)
treefe8856f65d1dafa946150c5acbaf6e942ba3c026 /pathod/language/http2.py
parent294774d6f0dee95b02a93307ec493b111b7f171e (diff)
downloadmitmproxy-d33d3663ecb166461d9cb5a78a29b44ee7a8fbb7.tar.gz
mitmproxy-d33d3663ecb166461d9cb5a78a29b44ee7a8fbb7.tar.bz2
mitmproxy-d33d3663ecb166461d9cb5a78a29b44ee7a8fbb7.zip
combine projects
Diffstat (limited to 'pathod/language/http2.py')
-rw-r--r--pathod/language/http2.py299
1 files changed, 299 insertions, 0 deletions
diff --git a/pathod/language/http2.py b/pathod/language/http2.py
new file mode 100644
index 00000000..d5e3ca31
--- /dev/null
+++ b/pathod/language/http2.py
@@ -0,0 +1,299 @@
+import pyparsing as pp
+
+from netlib import http
+from netlib.http import user_agents, Headers
+from . import base, message
+
+"""
+ Normal HTTP requests:
+ <method>:<path>:<header>:<body>
+ e.g.:
+ GET:/
+ GET:/:h"foo"="bar"
+ POST:/:h"foo"="bar":b'content body payload'
+
+ Normal HTTP responses:
+ <code>:<header>:<body>
+ e.g.:
+ 200
+ 302:h"foo"="bar"
+ 404:h"foo"="bar":b'content body payload'
+
+ Individual HTTP/2 frames:
+ h2f:<payload_length>:<type>:<flags>:<stream_id>:<payload>
+ e.g.:
+ h2f:0:PING
+ h2f:42:HEADERS:END_HEADERS:0x1234567:foo=bar,host=example.com
+ h2f:42:DATA:END_STREAM,PADDED:0x1234567:'content body payload'
+"""
+
+def get_header(val, headers):
+ """
+ Header keys may be Values, so we have to "generate" them as we try the
+ match.
+ """
+ for h in headers:
+ k = h.key.get_generator({})
+ if len(k) == len(val) and k[:].lower() == val.lower():
+ return h
+ return None
+
+
+class _HeaderMixin(object):
+ unique_name = None
+
+ def values(self, settings):
+ return (
+ self.key.get_generator(settings),
+ self.value.get_generator(settings),
+ )
+
+class _HTTP2Message(message.Message):
+ @property
+ def actions(self):
+ return [] # self.toks(actions._Action)
+
+ @property
+ def headers(self):
+ headers = self.toks(_HeaderMixin)
+
+ if not self.raw:
+ if not get_header("content-length", headers):
+ if not self.body:
+ length = 0
+ else:
+ length = len(self.body.string())
+ headers.append(
+ Header(
+ base.TokValueLiteral("content-length"),
+ base.TokValueLiteral(str(length)),
+ )
+ )
+ return headers
+
+ @property
+ def raw(self):
+ return bool(self.tok(Raw))
+
+ @property
+ def body(self):
+ return self.tok(Body)
+
+ def resolve(self, settings):
+ return self
+
+
+class StatusCode(base.Integer):
+ pass
+
+
+class Method(base.OptionsOrValue):
+ options = [
+ "GET",
+ "HEAD",
+ "POST",
+ "PUT",
+ "DELETE",
+ ]
+
+
+class Path(base.Value):
+ pass
+
+
+class Header(_HeaderMixin, base.KeyValue):
+ preamble = "h"
+
+
+class ShortcutContentType(_HeaderMixin, base.Value):
+ preamble = "c"
+ key = base.TokValueLiteral("content-type")
+
+
+class ShortcutLocation(_HeaderMixin, base.Value):
+ preamble = "l"
+ key = base.TokValueLiteral("location")
+
+
+class ShortcutUserAgent(_HeaderMixin, base.OptionsOrValue):
+ preamble = "u"
+ options = [i[1] for i in user_agents.UASTRINGS]
+ key = base.TokValueLiteral("user-agent")
+
+ def values(self, settings):
+ value = self.value.val
+ if self.option_used:
+ value = user_agents.get_by_shortcut(value.lower())[2]
+
+ return (
+ self.key.get_generator(settings),
+ value
+ )
+
+
+class Raw(base.CaselessLiteral):
+ TOK = "r"
+
+
+class Body(base.Value):
+ preamble = "b"
+
+
+class Times(base.Integer):
+ preamble = "x"
+
+
+class Response(_HTTP2Message):
+ unique_name = None
+ comps = (
+ Header,
+ Body,
+ ShortcutContentType,
+ ShortcutLocation,
+ Raw,
+ )
+
+ def __init__(self, tokens):
+ super(Response, self).__init__(tokens)
+ self.rendered_values = None
+ self.stream_id = 2
+
+ @property
+ def status_code(self):
+ return self.tok(StatusCode)
+
+ @classmethod
+ def expr(cls):
+ parts = [i.expr() for i in cls.comps]
+ atom = pp.MatchFirst(parts)
+ resp = pp.And(
+ [
+ StatusCode.expr(),
+ pp.ZeroOrMore(base.Sep + atom)
+ ]
+ )
+ resp = resp.setParseAction(cls)
+ return resp
+
+ def values(self, settings):
+ if self.rendered_values:
+ return self.rendered_values
+ else:
+ headers = Headers([header.values(settings) for header in self.headers])
+
+ body = self.body
+ if body:
+ body = body.string()
+
+ resp = http.Response(
+ (2, 0),
+ self.status_code.string(),
+ '',
+ headers,
+ body,
+ )
+ resp.stream_id = self.stream_id
+
+ self.rendered_values = settings.protocol.assemble(resp)
+ return self.rendered_values
+
+ def spec(self):
+ return ":".join([i.spec() for i in self.tokens])
+
+
+class NestedResponse(base.NestedMessage):
+ preamble = "s"
+ nest_type = Response
+
+
+class Request(_HTTP2Message):
+ comps = (
+ Header,
+ ShortcutContentType,
+ ShortcutUserAgent,
+ Raw,
+ NestedResponse,
+ Body,
+ Times,
+ )
+ logattrs = ["method", "path"]
+
+ def __init__(self, tokens):
+ super(Request, self).__init__(tokens)
+ self.rendered_values = None
+ self.stream_id = 1
+
+ @property
+ def method(self):
+ return self.tok(Method)
+
+ @property
+ def path(self):
+ return self.tok(Path)
+
+ @property
+ def nested_response(self):
+ return self.tok(NestedResponse)
+
+ @property
+ def times(self):
+ return self.tok(Times)
+
+ @classmethod
+ def expr(cls):
+ parts = [i.expr() for i in cls.comps]
+ atom = pp.MatchFirst(parts)
+ resp = pp.And(
+ [
+ Method.expr(),
+ base.Sep,
+ Path.expr(),
+ pp.ZeroOrMore(base.Sep + atom)
+ ]
+ )
+ resp = resp.setParseAction(cls)
+ return resp
+
+ def values(self, settings):
+ if self.rendered_values:
+ return self.rendered_values
+ else:
+ path = self.path.string()
+ if self.nested_response:
+ path += self.nested_response.parsed.spec()
+
+ headers = Headers([header.values(settings) for header in self.headers])
+
+ body = self.body
+ if body:
+ body = body.string()
+
+ req = http.Request(
+ '',
+ self.method.string(),
+ '',
+ '',
+ '',
+ path,
+ (2, 0),
+ headers,
+ body,
+ )
+ req.stream_id = self.stream_id
+
+ self.rendered_values = settings.protocol.assemble(req)
+ return self.rendered_values
+
+ def spec(self):
+ return ":".join([i.spec() for i in self.tokens])
+
+def make_error_response(reason, body=None):
+ tokens = [
+ StatusCode("800"),
+ Body(base.TokValueLiteral("pathod error: " + (body or reason))),
+ ]
+ return Response(tokens)
+
+
+# class Frame(message.Message):
+# pass
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-03 15:37:39 +0000