diff options
| author | Sahn Lam <sahn@pobox.com> | 2012-08-16 23:27:47 -0700 |
|---|---|---|
| committer | Sahn Lam <sahn@pobox.com> | 2012-08-17 18:45:26 -0700 |
| commit | 3189d144a521fcc98695dd079fb3dd4304de2eee (patch) | |
| tree | f0d8198a5e1094be6ecf7cefe96aac95b26ea778 /libmproxy/console/contentview.py | |
| parent | a66d018363e6d0c597577ed459308d4c80cbc2cc (diff) | |
| download | mitmproxy-3189d144a521fcc98695dd079fb3dd4304de2eee.tar.gz mitmproxy-3189d144a521fcc98695dd079fb3dd4304de2eee.tar.bz2 mitmproxy-3189d144a521fcc98695dd079fb3dd4304de2eee.zip | |
Optional AMF decoding support
If PyAMF is installed, enable AMF decoding.
Diffstat (limited to 'libmproxy/console/contentview.py')
| -rw-r--r-- | libmproxy/console/contentview.py | 28 |
1 files changed, 26 insertions, 2 deletions
diff --git a/libmproxy/console/contentview.py b/libmproxy/console/contentview.py index 2443c188..4efbb2b1 100644 --- a/libmproxy/console/contentview.py +++ b/libmproxy/console/contentview.py @@ -20,6 +20,7 @@ VIEW_RAW = 7 VIEW_HEX = 8 VIEW_HTML = 9 VIEW_OUTLINE = 10 +VIEW_AMF = 11 VIEW_NAMES = { VIEW_AUTO: "Auto", @@ -36,7 +37,7 @@ VIEW_NAMES = { } -VIEW_PROMPT = ( +VIEW_PROMPT = [ ("auto detect", "a"), ("hex", "e"), ("html", "h"), @@ -48,7 +49,7 @@ VIEW_PROMPT = ( ("multipart", "m"), ("urlencoded", "u"), ("xml", "x"), -) +] VIEW_SHORTCUTS = { "a": VIEW_AUTO, @@ -285,6 +286,10 @@ def view_image(hdrs, content, limit): ) return "%s image"%img.format, fmt +def view_amf(hdrs, content, limit): + s = utils.pretty_amf(content) + if s: + return "AMF", _view_text(s[:limit], len(s), limit) PRETTY_FUNCTION_MAP = { VIEW_XML: view_xml, @@ -344,3 +349,22 @@ def get_content_view(viewmode, hdrItems, content, limit): else: msg.append(ret[0]) return " ".join(msg), ret[1] + + +# +# Enable optional decoding methods at runtime +# + +# AMF decoding requires pyamf +try: + import pyamf + + VIEW_SHORTCUTS["f"] = VIEW_AMF + VIEW_PROMPT.append(("amf", "f")) + VIEW_NAMES[VIEW_AMF] = "AMF" + CONTENT_TYPES_MAP["application/x-amf"] = VIEW_AMF + PRETTY_FUNCTION_MAP[VIEW_AMF] = view_amf +except ImportError: + pass + + |