image view: add fallback, catch all images but svgs

4 files changed, 29 insertions, 20 deletions

diff --git a/mitmproxy/contentviews/auto.py b/mitmproxy/contentviews/auto.py
index 7b3cbd78..d46a1bd3 100644
--- a/mitmproxy/contentviews/auto.py
+++ b/mitmproxy/contentviews/auto.py
@@ -18,6 +18,8 @@ class ViewAuto(base.View):
                 return contentviews.content_types_map[ct][0](data, **metadata)
             elif strutils.is_xml(data):
                 return contentviews.get("XML/HTML")(data, **metadata)
+            elif ct.startswith("image/"):
+                return contentviews.get("Image")(data, **metadata)
         if metadata.get("query"):
             return contentviews.get("Query")(data, **metadata)
         if data and strutils.is_mostly_bin(data):
diff --git a/mitmproxy/contentviews/image/__init__.py b/mitmproxy/contentviews/image/__init__.py
index 0d0f06e0..33356bd7 100644
--- a/mitmproxy/contentviews/image/__init__.py
+++ b/mitmproxy/contentviews/image/__init__.py
@@ -1 +1,3 @@
-from .view import ViewImage  # noqa
+from .view import ViewImage
+
+__all__ = ["ViewImage"]
diff --git a/mitmproxy/contentviews/image/image_parser.py b/mitmproxy/contentviews/image/image_parser.py
index 1ff3cff7..062fb38e 100644
--- a/mitmproxy/contentviews/image/image_parser.py
+++ b/mitmproxy/contentviews/image/image_parser.py
@@ -13,9 +13,9 @@ Metadata = typing.List[typing.Tuple[str, str]]
 def parse_png(data: bytes) -> Metadata:
     img = png.Png(KaitaiStream(io.BytesIO(data)))
     parts = [
-        ('Format', 'Portable network graphics')
+        ('Format', 'Portable network graphics'),
+        ('Size', "{0} x {1} px".format(img.ihdr.width, img.ihdr.height))
     ]
-    parts.append(('Size', "{0} x {1} px".format(img.ihdr.width, img.ihdr.height)))
     for chunk in img.chunks:
         if chunk.type == 'gAMA':
             parts.append(('gamma', str(chunk.body.gamma_int / 100000)))
@@ -34,13 +34,13 @@ def parse_png(data: bytes) -> Metadata:
 
 def parse_gif(data: bytes) -> Metadata:
     img = gif.Gif(KaitaiStream(io.BytesIO(data)))
+    descriptor = img.logical_screen_descriptor
     parts = [
-        ('Format', 'Compuserve GIF')
+        ('Format', 'Compuserve GIF'),
+        ('Version', "GIF{}".format(img.header.version.decode('ASCII'))),
+        ('Size', "{} x {} px".format(descriptor.screen_width, descriptor.screen_height)),
+        ('background', str(descriptor.bg_color_index))
     ]
-    parts.append(('version', "GIF{0}".format(img.header.version.decode('ASCII'))))
-    descriptor = img.logical_screen_descriptor
-    parts.append(('Size', "{0} x {1} px".format(descriptor.screen_width, descriptor.screen_height)))
-    parts.append(('background', str(descriptor.bg_color_index)))
     ext_blocks = []
     for block in img.blocks:
         if block.block_type.name == 'extension':
diff --git a/mitmproxy/contentviews/image/view.py b/mitmproxy/contentviews/image/view.py
index d77e1ecb..95ee1e43 100644
--- a/mitmproxy/contentviews/image/view.py
+++ b/mitmproxy/contentviews/image/view.py
@@ -8,26 +8,31 @@ from . import image_parser
 class ViewImage(base.View):
     name = "Image"
     prompt = ("image", "i")
+
+    # there is also a fallback in the auto view for image/*.
     content_types = [
         "image/png",
         "image/jpeg",
         "image/gif",
+        "image/vnd.microsoft.icon",
+        "image/x-icon",
+        "image/webp",
     ]
 
     def __call__(self, data, **metadata):
         image_type = imghdr.what('', h=data)
         if image_type == 'png':
-            f = "PNG"
-            parts = image_parser.parse_png(data)
-            fmt = base.format_dict(multidict.MultiDict(parts))
-            return "%s image" % f, fmt
+            image_metadata = image_parser.parse_png(data)
         elif image_type == 'gif':
-            f = "GIF"
-            parts = image_parser.parse_gif(data)
-            fmt = base.format_dict(multidict.MultiDict(parts))
-            return "%s image" % f, fmt
+            image_metadata = image_parser.parse_gif(data)
         elif image_type == 'jpeg':
-            f = "JPEG"
-            parts = image_parser.parse_jpeg(data)
-            fmt = base.format_dict(multidict.MultiDict(parts))
-            return "%s image" % f, fmt
+            image_metadata = image_parser.parse_jpeg(data)
+        else:
+            image_metadata = [
+                ("Image Format", image_type or "unknown")
+            ]
+        if image_type:
+            view_name = "{} Image".format(image_type.upper())
+        else:
+            view_name = "Unknown Image"
+        return view_name, base.format_dict(multidict.MultiDict(image_metadata))