refactor code

mandiant · Mar 1, 2023 · 7b9f498 · 7b9f498
1 parent 6663d3e
commit 7b9f498
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dncil/cil/body/__init__.py b/dncil/cil/body/__init__.py
@@ -48,7 +48,6 @@ def __init__(self, reader: CilMethodBodyReaderBase):
         self.parse_header(reader)
         self.parse_instructions(reader)
         self.parse_exception_handlers(reader)
-        self.parse_basic_blocks()
 
         # use initial offset + method body size to read method body bytes (not the most efficient)
         final_pos = reader.tell()
@@ -79,6 +78,8 @@ def get_exception_handler_bytes(self) -> bytes:
         return self.raw_bytes[self.header_size + self.code_size :]
 
     def get_basic_blocks(self) -> Iterator[BasicBlock]:
+        if not self.basic_blocks:
+            self.parse_basic_blocks()
         yield from self.basic_blocks
 
     def parse_header(self, reader: CilMethodBodyReaderBase):