From 737c74b010cd6069beae092e0cc319ee4cb6e35b Mon Sep 17 00:00:00 2001 From: Ujjwal Verma Date: Thu, 11 Jan 2018 18:30:38 +0530 Subject: [PATCH] Update GIF parser --- mitmproxy/contrib/kaitaistruct/gif.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mitmproxy/contrib/kaitaistruct/gif.py b/mitmproxy/contrib/kaitaistruct/gif.py index 820df568c..76d7fc167 100644 --- a/mitmproxy/contrib/kaitaistruct/gif.py +++ b/mitmproxy/contrib/kaitaistruct/gif.py @@ -35,9 +35,11 @@ class Gif(KaitaiStruct): self.global_color_table = self._root.ColorTable(io, self, self._root) self.blocks = [] - while not self._io.is_eof(): - self.blocks.append(self._root.Block(self._io, self, self._root)) - + while True: + _ = self._root.Block(self._io, self, self._root) + self.blocks.append(_) + if ((self._io.is_eof()) or (_.block_type == self._root.BlockType.end_of_file)) : + break class ImageData(KaitaiStruct): def __init__(self, _io, _parent=None, _root=None):