make sure initLeaAst does not update the sym engine

Author: Triton Library

src/libtriton/bindings/python/objects/pyTritonContext.cpp

@@ -1994,7 +1994,7 @@ namespace triton {
       static PyObject* TritonContext_initLeaAst(PyObject* self, PyObject* mem) {
         try {
           if (PyMemoryAccess_Check(mem)) {
-            PyTritonContext_AsTritonContext(self)->getSymbolicEngine()->initLeaAst(*PyMemoryAccess_AsMemoryAccess(mem), true);
+            PyTritonContext_AsTritonContext(self)->initLeaAst(*PyMemoryAccess_AsMemoryAccess(mem), true);
           }
           else
             return PyErr_Format(PyExc_TypeError, "TritonContext::initLeaAst(): Expects a MemoryAccess as argument.");

src/libtriton/context/context.cpp

@@ -776,13 +776,13 @@ namespace triton {
   }
 
 
-  triton::ast::SharedAbstractNode Context::getRegisterAst(const triton::arch::Register& reg) {
+  triton::ast::SharedAbstractNode Context::getRegisterAst(const triton::arch::Register& reg) const {
     this->checkSymbolic();
     return this->symbolic->getRegisterAst(reg);
   }
 
 
-  triton::ast::SharedAbstractNode Context::getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) {
+  triton::ast::SharedAbstractNode Context::getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) const {
     this->checkSymbolic();
     return this->symbolic->getRegisterAst(inst, reg);
   }
@@ -1054,6 +1054,12 @@ namespace triton {
   }
 
 
+  void Context::initLeaAst(triton::arch::MemoryAccess& mem, bool force) const {
+    this->checkSymbolic();
+    this->symbolic->initLeaAst(mem);
+  }
+
+
   std::unordered_map<triton::usize, triton::engines::symbolic::SharedSymbolicExpression> Context::sliceExpressions(const triton::engines::symbolic::SharedSymbolicExpression& expr) {
     this->checkSymbolic();
     return this->symbolic->sliceExpressions(expr);

src/libtriton/engines/symbolic/symbolicEngine.cpp

@@ -697,7 +697,7 @@ namespace triton {
       }
 
 
-      triton::ast::SharedAbstractNode SymbolicEngine::getShiftAst(const triton::arch::arm::ArmOperandProperties& shift, const triton::ast::SharedAbstractNode& node) {
+      triton::ast::SharedAbstractNode SymbolicEngine::getShiftAst(const triton::arch::arm::ArmOperandProperties& shift, const triton::ast::SharedAbstractNode& node) const {
         auto imm = shift.getShiftImmediate();
         auto reg = shift.getShiftRegister();
 
@@ -790,7 +790,7 @@ namespace triton {
         }
       }
 
-      triton::ast::SharedAbstractNode SymbolicEngine::getIndexAst(const triton::arch::arm::ArmOperandProperties& vas_index, const triton::ast::SharedAbstractNode& node) {
+      triton::ast::SharedAbstractNode SymbolicEngine::getIndexAst(const triton::arch::arm::ArmOperandProperties& vas_index, const triton::ast::SharedAbstractNode& node) const {
         auto vas_size = vas_index.getVASSize() * triton::bitsize::byte;
 
         auto low = vas_index.getVectorIndex() * vas_size;
@@ -800,7 +800,7 @@ namespace triton {
       }
 
 
-      triton::ast::SharedAbstractNode SymbolicEngine::getExtendAst(const triton::arch::arm::ArmOperandProperties& extend, const triton::ast::SharedAbstractNode& node) {
+      triton::ast::SharedAbstractNode SymbolicEngine::getExtendAst(const triton::arch::arm::ArmOperandProperties& extend, const triton::ast::SharedAbstractNode& node) const {
         triton::uint32 size = extend.getExtendSize();
 
         switch (extend.getExtendType()) {
@@ -925,7 +925,7 @@ namespace triton {
 
 
       /* Returns the AST corresponding to the register */
-      triton::ast::SharedAbstractNode SymbolicEngine::getRegisterAst(const triton::arch::Register& reg) {
+      triton::ast::SharedAbstractNode SymbolicEngine::getRegisterAst(const triton::arch::Register& reg) const {
         triton::ast::SharedAbstractNode node = nullptr;
         triton::uint32 bvSize                = reg.getBitSize();
         triton::uint32 high                  = reg.getHigh();
@@ -957,7 +957,7 @@ namespace triton {
 
 
       /* Returns the AST corresponding to the register and defines the register as input of the instruction */
-      triton::ast::SharedAbstractNode SymbolicEngine::getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) {
+      triton::ast::SharedAbstractNode SymbolicEngine::getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) const {
         triton::ast::SharedAbstractNode node = this->getRegisterAst(reg);
         inst.setReadRegister(reg, node);
         return node;
@@ -1259,7 +1259,7 @@ namespace triton {
 
 
       /* Initializes the memory access AST (LOAD and STORE) */
-      void SymbolicEngine::initLeaAst(triton::arch::MemoryAccess& mem, bool force) {
+      void SymbolicEngine::initLeaAst(triton::arch::MemoryAccess& mem, bool force) const {
         if (mem.getBitSize() >= bitsize::byte) {
           const triton::arch::Register& base  = mem.getConstBaseRegister();
           const triton::arch::Register& index = mem.getConstIndexRegister();

src/libtriton/includes/triton/context.hpp

@@ -397,10 +397,10 @@ namespace triton {
         TRITON_EXPORT triton::ast::SharedAbstractNode getMemoryAst(triton::arch::Instruction& inst, const triton::arch::MemoryAccess& mem);
 
         //! [**symbolic api**] - Returns the AST corresponding to the register.
-        TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(const triton::arch::Register& reg);
+        TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(const triton::arch::Register& reg) const;
 
         //! [**symbolic api**] - Returns the AST corresponding to the register and defines the register as input of the instruction.
-        TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg);
+        TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) const;
 
         //! [**symbolic api**] - Returns a new symbolic expression. Note that if there are simplification passes recorded, simplification will be applied.
         TRITON_EXPORT triton::engines::symbolic::SharedSymbolicExpression newSymbolicExpression(const triton::ast::SharedAbstractNode& node, const std::string& comment="");
@@ -513,6 +513,9 @@ namespace triton {
         //! [**symbolic api**] - Concretizes a symbolic register.
         TRITON_EXPORT void concretizeRegister(const triton::arch::Register& reg);
 
+        //! [**symbolic api**] - Initializes the effective address of a memory access.
+        TRITON_EXPORT void initLeaAst(triton::arch::MemoryAccess& mem, bool force=true) const;
+
         //! [**symbolic api**] - Slices all expressions from a given one.
         TRITON_EXPORT std::unordered_map<triton::usize, triton::engines::symbolic::SharedSymbolicExpression> sliceExpressions(const triton::engines::symbolic::SharedSymbolicExpression& expr);
 

src/libtriton/includes/triton/symbolicEngine.hpp

@@ -127,7 +127,7 @@ namespace triton {
           inline void addBitvectorMemory(triton::uint64 mem, const SharedSymbolicExpression& expr);
 
           //! Returns the AST corresponding to the extend operation. Mainly used for AArch64 operands.
-          triton::ast::SharedAbstractNode getExtendAst(const triton::arch::arm::ArmOperandProperties& extend, const triton::ast::SharedAbstractNode& node);
+          triton::ast::SharedAbstractNode getExtendAst(const triton::arch::arm::ArmOperandProperties& extend, const triton::ast::SharedAbstractNode& node) const;
 
           //! Returns the parent AST after inserting the subregister (node) in its AST.
           triton::ast::SharedAbstractNode insertSubRegisterInParent(const triton::arch::Register& reg, const triton::ast::SharedAbstractNode& node, bool zxForAssign=true);
@@ -221,16 +221,16 @@ namespace triton {
           TRITON_EXPORT triton::ast::SharedAbstractNode getMemoryAst(triton::arch::Instruction& inst, const triton::arch::MemoryAccess& mem);
 
           //! Returns the AST corresponding to the register.
-          TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(const triton::arch::Register& reg);
+          TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(const triton::arch::Register& reg) const;
 
           //! Returns the AST corresponding to the register and defines the register as input of the instruction.
-          TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg);
+          TRITON_EXPORT triton::ast::SharedAbstractNode getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) const;
 
           //! Returns the AST corresponding to the shift operation. Mainly used for Arm32 operands.
-          triton::ast::SharedAbstractNode getShiftAst(const triton::arch::arm::ArmOperandProperties& shift, const triton::ast::SharedAbstractNode& node);
+          triton::ast::SharedAbstractNode getShiftAst(const triton::arch::arm::ArmOperandProperties& shift, const triton::ast::SharedAbstractNode& node) const;
 
           //! Returns the AST corresponding to the VAS vector index operation. Mainly used for Arm Neon vector operands.
-          triton::ast::SharedAbstractNode getIndexAst(const triton::arch::arm::ArmOperandProperties& vas_index, const triton::ast::SharedAbstractNode& node);
+          triton::ast::SharedAbstractNode getIndexAst(const triton::arch::arm::ArmOperandProperties& vas_index, const triton::ast::SharedAbstractNode& node) const;
 
           //! Returns the new symbolic expression and links this expression to the instruction.
           TRITON_EXPORT const SharedSymbolicExpression& createSymbolicExpression(triton::arch::Instruction& inst, const triton::ast::SharedAbstractNode& node, const triton::arch::OperandWrapper& dst, const std::string& comment="");
@@ -302,7 +302,7 @@ namespace triton {
           TRITON_EXPORT bool isRegisterSymbolized(const triton::arch::Register& reg) const;
 
           //! Initializes the effective address of a memory access.
-          TRITON_EXPORT void initLeaAst(triton::arch::MemoryAccess& mem, bool force=true);
+          TRITON_EXPORT void initLeaAst(triton::arch::MemoryAccess& mem, bool force=true) const;
 
           //! Gets the concrete value of a symbolic variable.
           TRITON_EXPORT triton::uint512 getConcreteVariableValue(const SharedSymbolicVariable& symVar) const;