Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-30 03:14:22 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #2379 from ujjwal96/kaitai

Browse Source 
Update png.py. Stop reading after IEND
This commit is contained in:
Thomas Kriechbaumer 2017-06-06 00:51:03 +02:00 committed by GitHub
commit 02232b5149
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mitmproxy/contrib/kaitaistruct/png.py
View File

@ -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):