Fix for repz ret

titech-caras · Oct 31, 2024 · 1a0c9f8 · 1a0c9f8
1 parent b118396
commit 1a0c9f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/teapot/passes/transient/transient_indirect_branch_check_dest_pass.py b/teapot/passes/transient/transient_indirect_branch_check_dest_pass.py
@@ -45,7 +45,7 @@ def visit_code_block(self, block: gtirb.CodeBlock, function: Function = None):
             instructions = list(self.decoder.get_instructions(block))
             last_inst = instructions[-1]
 
-            if last_inst.mnemonic == "ret":
+            if non_fallthrough_edges[0].label.type == gtirb.cfg.Edge.Type.Return:
                 operand_str = "[rsp]"
             else:
                 dest_operand = last_inst.operands[0]