diff options
author | Nicolas Esteves <hamstahguru@gmail.com> | 2013-04-06 19:18:12 +0100 |
---|---|---|
committer | Nicolas Esteves <hamstahguru@gmail.com> | 2013-04-06 19:21:13 +0100 |
commit | d4cfbbb822f786b7e1f0ab9282dd76ee04ccb178 (patch) | |
tree | 2f810bf703379223be5994ce6f0c3c06e9f5bd39 /libmproxy/console/contentview.py | |
parent | 51b775cfd4165ed80dcb5458ddff942125fcba31 (diff) | |
download | mitmproxy-d4cfbbb822f786b7e1f0ab9282dd76ee04ccb178.tar.gz mitmproxy-d4cfbbb822f786b7e1f0ab9282dd76ee04ccb178.tar.bz2 mitmproxy-d4cfbbb822f786b7e1f0ab9282dd76ee04ccb178.zip |
Adds a new view for protocol buffers
The view uses protoc from the Google protocol buffer
tools. If the tool isn't installed, the view isn't
shown.
Google protobuf repo:
https://code.google.com/p/protobuf/
Diffstat (limited to 'libmproxy/console/contentview.py')
-rw-r--r-- | libmproxy/console/contentview.py | 37 |
1 files changed, 36 insertions, 1 deletions
diff --git a/libmproxy/console/contentview.py b/libmproxy/console/contentview.py index 2b46064a..61c7adb5 100644 --- a/libmproxy/console/contentview.py +++ b/libmproxy/console/contentview.py @@ -12,7 +12,7 @@ import netlib.utils import common from .. import utils, encoding, flow from ..contrib import jsbeautifier, html2text - +import subprocess try: import pyamf from pyamf import remoting, flex @@ -364,6 +364,38 @@ class ViewImage: ) return "%s image"%img.format, fmt +class ViewProtobuf: + """Human friendly view of protocol buffers + The view uses the protoc compiler to decode the binary + """ + + name = "Protocol Buffer" + prompt = ("protobuf", "p") + content_types = ["application/x-protobuf"] + + @staticmethod + def is_available(): + try: + p = subprocess.Popen(["protoc", "--version"], stdout=subprocess.PIPE) + out, _ = p.communicate() + return out.startswith("libprotoc") + except: + return False + + def decode_protobuf(self, content): + # if Popen raises OSError, it will be caught in + # get_content_view and fall back to Raw + p = subprocess.Popen(['protoc', '--decode_raw'], + stdin=subprocess.PIPE, + stdout=subprocess.PIPE, + stderr=subprocess.PIPE) + out, _ = p.communicate(input=content) + return out + + def __call__(self, hdrs, content, limit): + decoded = self.decode_protobuf(content) + txt = _view_text(decoded[:limit], len(decoded), limit) + return "Protobuf", txt views = [ ViewAuto(), @@ -381,6 +413,9 @@ views = [ if pyamf: views.append(ViewAMF()) +if ViewProtobuf.is_available(): + views.append(ViewProtobuf()) + content_types_map = {} for i in views: for ct in i.content_types: |