update trace

src/halmos/sevm.py

@@ -350,13 +350,15 @@ class StorageWrite:
     address: Address
     slot: Word
     value: Word
+    transient: bool
 
 
 @dataclass(frozen=True)
 class StorageRead:
     address: Address
     slot: Word
     value: Word
+    transient: bool
 
 
 @dataclass(frozen=True)
@@ -2072,15 +2074,15 @@ def mk_storagedata(self) -> StorageData:
     def sload(self, ex: Exec, addr: Any, loc: Word, transient: bool = False) -> Word:
         storage = ex.storage if not transient else ex.transient_storage
         val = self.storage_model.load(ex, storage, addr, loc)
-        ex.context.trace.append(StorageRead(addr, loc, val))
+        ex.context.trace.append(StorageRead(addr, loc, val, transient))
         return val
 
     def sstore(
         self, ex: Exec, addr: Any, loc: Any, val: Any, transient: bool = False
     ) -> None:
         storage = ex.storage if not transient else ex.transient_storage
 
-        ex.context.trace.append(StorageWrite(addr, loc, val))
+        ex.context.trace.append(StorageWrite(addr, loc, val, transient))
 
         if ex.message().is_static:
             raise WriteInStaticContext(ex.context_str())

src/halmos/traces.py

@@ -114,12 +114,14 @@ def rendered_slot(slot: Address) -> str:
 
 def rendered_sstore(update: StorageWrite) -> str:
     slot_str = rendered_slot(update.slot)
-    return f"{cyan('SSTORE')} @{slot_str} ← {hexify(update.value)}"
+    opcode = cyan("SSTORE" if not update.transient else "TSTORE")
+    return f"{opcode} @{slot_str} ← {hexify(update.value)}"
 
 
 def rendered_sload(read: StorageRead) -> str:
     slot_str = rendered_slot(read.slot)
-    return f"{cyan('SLOAD')}  @{slot_str} → {hexify(read.value)}"
+    opcode = cyan("SLOAD" if not read.transient else "TLOAD")
+    return f"{opcode}  @{slot_str} → {hexify(read.value)}"
 
 
 def rendered_trace(context: CallContext) -> str: