From 465966ac9597ce7879171268816171c75937ad4e Mon Sep 17 00:00:00 2001
From: Petar Maksimovic <petar.maksimovic@runtimeverification.com>
Date: Wed, 25 Sep 2024 18:44:27 +0100
Subject: [PATCH] adding is_loop

---
 kevm-pyk/src/kevm_pyk/kevm.py | 27 +++++++++------------------
 1 file changed, 9 insertions(+), 18 deletions(-)

diff --git a/kevm-pyk/src/kevm_pyk/kevm.py b/kevm-pyk/src/kevm_pyk/kevm.py
index 2023cd8db4..bf74746c3f 100644
--- a/kevm-pyk/src/kevm_pyk/kevm.py
+++ b/kevm-pyk/src/kevm_pyk/kevm.py
@@ -20,13 +20,12 @@
 from pyk.kast.manip import abstract_term_safely, flatten_label, set_cell
 from pyk.kast.pretty import paren
 from pyk.kcfg.kcfg import Step
-from pyk.kcfg.semantics import KCFGSemantics
+from pyk.kcfg.semantics import DefaultSemantics
 from pyk.kcfg.show import NodePrinter
 from pyk.ktool.kprove import KProve
 from pyk.ktool.krun import KRun
 from pyk.prelude.bytes import BYTES, pretty_bytes
-from pyk.prelude.kbool import notBool
-from pyk.prelude.kint import INT, eqInt, gtInt, intToken, ltInt
+from pyk.prelude.kint import INT, gtInt, intToken, ltInt
 from pyk.prelude.ml import mlEqualsFalse, mlEqualsTrue
 from pyk.prelude.string import stringToken
 from pyk.prelude.utils import token
@@ -50,7 +49,7 @@
 # KEVM class
 
 
-class KEVMSemantics(KCFGSemantics):
+class KEVMSemantics(DefaultSemantics):
     auto_abstract_gas: bool
     allow_symbolic_program: bool
     _cached_subst: Subst | None
@@ -98,6 +97,12 @@ def is_terminal(self, cterm: CTerm) -> bool:
 
         return False
 
+    def is_loop(self, cterm: CTerm) -> bool:
+        jumpi_pattern = KEVM.jumpi_applied(KVariable('###PCOUNT'), KVariable('###COND'))
+        pc_next_pattern = KEVM.pc_applied(KEVM.jumpi())
+        branch_pattern = KSequence([jumpi_pattern, pc_next_pattern, KEVM.sharp_execute(), KVariable('###CONTINUATION')])
+        return branch_pattern.match(cterm.cell('K_CELL')) is not None
+
     def same_loop(self, cterm1: CTerm, cterm2: CTerm) -> bool:
         # In the same program, at the same calldepth, at the same program counter
         for cell in ['PC_CELL', 'CALLDEPTH_CELL', 'PROGRAM_CELL']:
@@ -116,20 +121,6 @@ def same_loop(self, cterm1: CTerm, cterm2: CTerm) -> bool:
                 return True
         return False
 
-    def extract_branches(self, cterm: CTerm) -> list[KInner]:
-        k_cell = cterm.cell('K_CELL')
-        jumpi_pattern = KEVM.jumpi_applied(KVariable('###PCOUNT'), KVariable('###COND'))
-        pc_next_pattern = KEVM.pc_applied(KEVM.jumpi())
-        branch_pattern = KSequence([jumpi_pattern, pc_next_pattern, KEVM.sharp_execute(), KVariable('###CONTINUATION')])
-        if subst := branch_pattern.match(k_cell):
-            cond = subst['###COND']
-            if cond_subst := KEVM.bool_2_word(KVariable('###BOOL_2_WORD')).match(cond):
-                cond = cond_subst['###BOOL_2_WORD']
-            else:
-                cond = eqInt(cond, intToken(0))
-            return [mlEqualsTrue(cond), mlEqualsTrue(notBool(cond))]
-        return []
-
     def abstract_node(self, cterm: CTerm) -> CTerm:
         if not self.auto_abstract_gas:
             return cterm