1 files changed, 27 insertions, 0 deletions
diff --git a/mitmproxy/contentviews/image/image_parser.py b/mitmproxy/contentviews/image/image_parser.py
index 0af58a88..b104d105 100644
--- a/mitmproxy/contentviews/image/image_parser.py
+++ b/mitmproxy/contentviews/image/image_parser.py
@@ -4,6 +4,7 @@ import typing
 from kaitaistruct import KaitaiStream
 
 from mitmproxy.contrib.kaitaistruct import png
+from mitmproxy.contrib.kaitaistruct import gif
 
 Metadata = typing.List[typing.Tuple[str, str]]
 
@@ -28,3 +29,29 @@ def parse_png(data: bytes) -> Metadata:
         elif chunk.type == 'zTXt':
             parts.append((chunk.body.keyword, chunk.body.text_datastream.decode('iso8859-1')))
     return parts
+
+
+def parse_gif(data: bytes) -> Metadata:
+    img = gif.Gif(KaitaiStream(io.BytesIO(data)))
+    parts = [
+        ('Format', 'Compuserve GIF')
+    ]
+    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':
+            ext_blocks.append(block)
+    comment_blocks = []
+    for block in ext_blocks:
+        if block.body.label._name_ == 'comment':
+            comment_blocks.append(block)
+    for block in comment_blocks:
+        entries = block.body.body.entries
+        for entry in entries:
+            comment = entry.bytes
+            if comment is not b'':
+                parts.append(('comment', str(comment)))
+    return parts