adding tstore and tload. Split for the moment

costa-group · Jan 16, 2025 · bd1ab12 · bd1ab12
1 parent ba9edd0
commit bd1ab12
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/parser/constants.py b/src/parser/constants.py
@@ -1,5 +1,5 @@
 split_block = {'calldatacopy', 'create', 'codecopy', 'call', 'log4', 'gas', 'delegatecall', 'extcodecopy', 'create2',
-               'assignimmutable', 'returndatacopy', 'log2', 'log1', 'log3', 'log0', 'datacopy', 'staticcall'}
+               'assignimmutable', 'returndatacopy', 'log2', 'log1', 'log3', 'log0', 'datacopy', 'staticcall','tstore','tload'}
 
 terminal_ops = {"functionReturn", "functionReturn", "return", "revert"}
 

diff --git a/src/parser/opcodes.py b/src/parser/opcodes.py
@@ -67,6 +67,8 @@
     "JUMPDEST": [0x5b, 0, 0],
     "SLOADEXT": [0x5c, 2, 1],
     "SSTOREEXT": [0x5d, 3, 0],
+    "TLOAD": [0x5c, 1, 1],
+    "TSTORE": [0x5d, 2, 0],
     "SLOADBYTESEXT": [0x5c, 4, 0],
     "SSTOREBYTESEXT": [0x5d, 4, 0],
     "LOG0": [0xa0, 2, 0],
@@ -295,7 +297,10 @@ def get_ins_cost(opcode, params=None, already=False, store_changed_original_valu
         return (GCOST["Gcoldsload"] if not already else 0) + (GCOST["Gwarmaccess"] if store_changed_original_value else GCOST["Gsreset"])
     elif opcode == "KECCAK256":
         return GCOST["Gsha3"]
-
+    elif opcode == "TLOAD":
+        return 100
+    elif opcode == "TSTORE":
+        return 100
     elif opcode == "SELFDESTRUCT":
         return GCOST["Gsuicide"]
     return 0