Updated index pack constructor to improve generation.

atac · Jun 30, 2021 · 0412650 · 0412650
1 parent 64abc15
commit 0412650
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chapter10/computer.py b/chapter10/computer.py
@@ -177,7 +177,7 @@ def __repr__(self):
     def __init__(self, *args, **kwargs):
         packet.Packet.__init__(self, *args, **kwargs)
 
-        if self.file_size_present:
+        if self.file_size_present and self.buffer:
             self.file_size, = bitstruct.unpack('u64<', self.buffer.read(8))
 
         fmt = 'u64 ipts'
@@ -195,7 +195,7 @@ def __init__(self, *args, **kwargs):
 
         self.Message.FORMAT = BitFormat(fmt)
 
-        if self.index_type == 0:
+        if self.index_type == 0 and self.buffer:
             pos = self.buffer.tell()
             self.buffer.seek(self.data_length - 8)
             self.root_offset, = bitstruct.unpack('u64<', self.buffer.read(8))