Fix concretize memory function in ProcessState

quarkslab · May 10, 2024 · 8420d45 · 8420d45
1 parent e3c1757
commit 8420d45
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tritondse/process_state.py b/tritondse/process_state.py
@@ -798,7 +798,11 @@ def concretize_memory_bytes(self, addr: Addr, size: ByteSize) -> None:
         """
         data = self.memory.read(addr, size)
         if self.is_memory_symbolic(addr, size):
-            self.push_constraint(self.read_symbolic_memory_bytes(addr, size).getAst() == data)
+            if isinstance(data, bytes):
+                data_ast = self.actx.concat([self.actx.bv(b, 8) for b in data])
+                self.push_constraint(self.read_symbolic_memory_bytes(addr, size).getAst() == data_ast)
+            else:
+                self.push_constraint(self.read_symbolic_memory_bytes(addr, size).getAst() == data)
         # else do not even push the constraint
 
     def concretize_memory_int(self, addr: Addr, size: ByteSize) -> None: