adding STOP instruction for mainExit block

costa-group · Oct 22, 2024 · ff1c434 · ff1c434
1 parent 7f9476b
commit ff1c434
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/solution_generation/reconstruct_bytecode.py b/src/solution_generation/reconstruct_bytecode.py
@@ -219,6 +219,10 @@ def traverse_cfg(cfg_object, asm_dicts, tags_dict):
             asm_instructions += asm_block
             init_pos_dict += [block_id]*len(asm_block)
 
+        elif jump_type == "mainExit":
+            asm_instructions+=asm_block+[asm_from_op_info("STOP")]
+            init_pos_dict += [block_id]*len(asm_block+1)
+
         else:
             raise Exception("[ERROR]: Unknown jump type when generating asm output")