From 8b06dffafbdfcac24004d02aae1a0b5a36f7d931 Mon Sep 17 00:00:00 2001 From: Ujjwal Verma Date: Tue, 6 Jun 2017 03:52:36 +0530 Subject: Update png.py. Stop reading after IEND --- mitmproxy/contrib/kaitaistruct/png.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'mitmproxy/contrib') diff --git a/mitmproxy/contrib/kaitaistruct/png.py b/mitmproxy/contrib/kaitaistruct/png.py index 98a70693..45074d70 100644 --- a/mitmproxy/contrib/kaitaistruct/png.py +++ b/mitmproxy/contrib/kaitaistruct/png.py @@ -34,9 +34,11 @@ class Png(KaitaiStruct): self.ihdr = self._root.IhdrChunk(self._io, self, self._root) self.ihdr_crc = self._io.read_bytes(4) self.chunks = [] - while not self._io.is_eof(): - self.chunks.append(self._root.Chunk(self._io, self, self._root)) - + while True: + _ = self._root.Chunk(self._io, self, self._root) + self.chunks.append(_) + if ((_.type == u"IEND") or (self._io.is_eof())) : + break class Rgb(KaitaiStruct): def __init__(self, _io, _parent=None, _root=None): -- cgit v1.2.3