PCodeConverter: Handle upper case space names. (#231)

angr · Aug 28, 2024 · da36db5 · da36db5
1 parent 697a40e
commit da36db5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/ailment/converter_pcode.py b/ailment/converter_pcode.py
@@ -419,8 +419,9 @@ def _convert_load(self) -> None:
         """
         spc = self._current_op.inputs[0].getSpaceFromConst()
         out = self._current_op.output
-        assert spc.name in {"ram", "mem", "register"}
-        if spc.name == "register":
+        spc_name = spc.name.lower()
+        assert spc_name in {"ram", "mem", "register"}
+        if spc_name == "register":
             # load from register
             res = self._get_value(self._current_op.inputs[1])
             stmt = self._set_value(out, res)
@@ -442,8 +443,9 @@ def _convert_store(self) -> None:
         Convert a p-code store operation
         """
         spc = self._current_op.inputs[0].getSpaceFromConst()
-        assert spc.name in {"ram", "mem", "register"}
-        if spc.name == "register":
+        spc_name = spc.name.lower()
+        assert spc_name in {"ram", "mem", "register"}
+        if spc_name == "register":
             # store to register
             out = self._current_op.inputs[2]
             res = self._get_value(self._current_op.inputs[1])