context.cpp

Go to the documentation of this file.

 
/*
**  Copyright (C) - Triton
**
**  This program is under the terms of the Apache License 2.0.
*/
 
#include <triton/aarch64Cpu.hpp>
#include <triton/arm32Cpu.hpp>
#include <triton/config.hpp>
#include <triton/context.hpp>
#include <triton/exceptions.hpp>
#ifdef COMPILE_RISCV
#include <triton/riscv32Cpu.hpp>
#include <triton/riscv64Cpu.hpp>
#endif
#include <triton/x8664Cpu.hpp>
#include <triton/x86Cpu.hpp>
 
#include <list>
#include <map>
#include <memory>
#include <new>
 
 
namespace triton {
 
  Context::Context() :
    callbacks(*this),
    arch(&this->callbacks) {
    this->modes   = std::make_shared<triton::modes::Modes>();
    this->astCtxt = std::make_shared<triton::ast::AstContext>(this->modes);
  }
 
 
  Context::Context(triton::arch::architecture_e arch) :
    Context() {
    this->setArchitecture(arch);
  }
 
 
  Context::~Context() {
    this->removeEngines();
  }
 
 
  inline void Context::checkArchitecture(void) const {
    if (!this->isArchitectureValid())
      throw triton::exceptions::Context("Context::checkArchitecture(): You must define an architecture.");
  }
 
 
  inline void Context::checkIrBuilder(void) const {
    if (!this->irBuilder)
      throw triton::exceptions::Context("Context::checkIrBuilder(): IR builder is undefined, you should define an architecture first.");
  }
 
 
  inline void Context::checkSymbolic(void) const {
    if (!this->symbolic)
      throw triton::exceptions::Context("Context::checkSymbolic(): Symbolic engine is undefined, you should define an architecture first.");
  }
 
 
  inline void Context::checkSolver(void) const {
    if (!this->solver)
      throw triton::exceptions::Context("Context::checkSolver(): Solver engine is undefined, you should define an architecture first.");
  }
 
 
  inline void Context::checkTaint(void) const {
    if (!this->taint)
      throw triton::exceptions::Context("Context::checkTaint(): Taint engine is undefined, you should define an architecture first.");
  }
 
 
  inline void Context::checkLifting(void) const {
    if (!this->lifting)
      throw triton::exceptions::Context("Context::checkLifting(): Lifting engine is undefined, you should define an architecture first.");
  }
 
 
 
  /* Architecture Context ============================================================================== */
 
  bool Context::isArchitectureValid(void) const {
    return this->arch.isValid();
  }
 
 
  triton::arch::architecture_e Context::getArchitecture(void) const {
    return this->arch.getArchitecture();
  }
 
 
  triton::arch::endianness_e Context::getEndianness(void) const {
    return this->arch.getEndianness();
  }
 
 
  triton::arch::CpuInterface* Context::getCpuInstance(void) {
    if (!this->isArchitectureValid())
      throw triton::exceptions::Context("Context::checkArchitecture(): You must define an architecture.");
    return this->arch.getCpuInstance();
  }
 
 
  void Context::setArchitecture(triton::arch::architecture_e arch) {
    /* Setup and init the targeted architecture */
    this->arch.setArchitecture(arch);
 
    /* remove and re-init previous engines (when setArchitecture() has been called twice) */
    this->removeEngines();
    this->initEngines();
  }
 
 
  void Context::clearArchitecture(void) {
    this->checkArchitecture();
    this->arch.clearArchitecture();
  }
 
 
  bool Context::isFlag(triton::arch::register_e regId) const {
    return this->arch.isFlag(regId);
  }
 
 
  bool Context::isFlag(const triton::arch::Register& reg) const {
    return this->arch.isFlag(reg);
  }
 
 
  bool Context::isRegister(triton::arch::register_e regId) const {
    return this->arch.isRegister(regId);
  }
 
 
  bool Context::isRegister(const triton::arch::Register& reg) const {
    return this->arch.isRegister(reg);
  }
 
 
  const triton::arch::Register& Context::getRegister(triton::arch::register_e id) const {
    return this->arch.getRegister(id);
  }
 
 
  const triton::arch::Register& Context::getRegister(const std::string& name) const {
    return this->arch.getRegister(name);
  }
 
 
  const triton::arch::Register& Context::getParentRegister(const triton::arch::Register& reg) const {
    return this->arch.getParentRegister(reg);
  }
 
 
  const triton::arch::Register& Context::getParentRegister(triton::arch::register_e id) const {
    return this->arch.getParentRegister(id);
  }
 
 
  bool Context::isRegisterValid(triton::arch::register_e regId) const {
    return this->arch.isRegisterValid(regId);
  }
 
 
  bool Context::isRegisterValid(const triton::arch::Register& reg) const {
    return this->arch.isRegisterValid(reg);
  }
 
 
  bool Context::isThumb(void) const {
    return this->arch.isThumb();
  }
 
 
  void Context::setThumb(bool state) {
    this->arch.setThumb(state);
  }
 
 
  triton::uint32 Context::getGprBitSize(void) const {
    return this->arch.gprBitSize();
  }
 
 
  triton::uint32 Context::getGprSize(void) const {
    return this->arch.gprSize();
  }
 
 
  triton::uint32 Context::getNumberOfRegisters(void) const {
    return this->arch.numberOfRegisters();
  }
 
 
  const triton::arch::Instruction Context::getNopInstruction(void) const {
    return this->arch.getNopInstruction();
  }
 
 
  const std::unordered_map<triton::arch::register_e, const triton::arch::Register>& Context::getAllRegisters(void) const {
    this->checkArchitecture();
    return this->arch.getAllRegisters();
  }
 
  const std::unordered_map<triton::uint64, triton::uint8, IdentityHash<triton::uint64>>& Context::getConcreteMemory(void) const {
    this->checkArchitecture();
    return this->arch.getConcreteMemory();
  }
 
 
  std::set<const triton::arch::Register*> Context::getParentRegisters(void) const {
    this->checkArchitecture();
    return this->arch.getParentRegisters();
  }
 
 
  triton::uint8 Context::getConcreteMemoryValue(triton::uint64 addr, bool execCallbacks) const {
    this->checkArchitecture();
    return this->arch.getConcreteMemoryValue(addr, execCallbacks);
  }
 
 
  triton::uint512 Context::getConcreteMemoryValue(const triton::arch::MemoryAccess& mem, bool execCallbacks) const {
    this->checkArchitecture();
    return this->arch.getConcreteMemoryValue(mem, execCallbacks);
  }
 
 
  std::vector<triton::uint8> Context::getConcreteMemoryAreaValue(triton::uint64 baseAddr, triton::usize size, bool execCallbacks) const {
    this->checkArchitecture();
    return this->arch.getConcreteMemoryAreaValue(baseAddr, size, execCallbacks);
  }
 
 
  triton::uint512 Context::getConcreteRegisterValue(const triton::arch::Register& reg, bool execCallbacks) const {
    this->checkArchitecture();
    return this->arch.getConcreteRegisterValue(reg, execCallbacks);
  }
 
 
  void Context::setConcreteMemoryValue(triton::uint64 addr, triton::uint8 value, bool execCallbacks) {
    this->checkArchitecture();
    this->arch.setConcreteMemoryValue(addr, value, execCallbacks);
    /*
     * In order to synchronize the concrete state with the symbolic
     * one, the symbolic expression is concretized.
     */
    this->concretizeMemory(addr);
  }
 
 
  void Context::setConcreteMemoryValue(const triton::arch::MemoryAccess& mem, const triton::uint512& value, bool execCallbacks) {
    this->checkArchitecture();
    this->arch.setConcreteMemoryValue(mem, value, execCallbacks);
    /*
     * In order to synchronize the concrete state with the symbolic
     * one, the symbolic expression is concretized.
     */
    this->concretizeMemory(mem);
  }
 
 
  void Context::setConcreteMemoryAreaValue(triton::uint64 baseAddr, const std::vector<triton::uint8>& values, bool execCallbacks) {
    this->checkArchitecture();
    this->arch.setConcreteMemoryAreaValue(baseAddr, values, execCallbacks);
    /*
     * In order to synchronize the concrete state with the symbolic
     * one, the symbolic expression is concretized.
     */
    for (triton::usize index = 0 ; index < values.size() ; index++) {
      this->concretizeMemory(baseAddr + index);
    }
  }
 
 
  void Context::setConcreteMemoryAreaValue(triton::uint64 baseAddr, const void* area, triton::usize size, bool execCallbacks) {
    this->checkArchitecture();
    this->arch.setConcreteMemoryAreaValue(baseAddr, area, size, execCallbacks);
    /*
     * In order to synchronize the concrete state with the symbolic
     * one, the symbolic expression is concretized.
     */
    for (triton::usize index = 0 ; index < size ; index++) {
      this->concretizeMemory(baseAddr + index);
    }
  }
 
 
  void Context::setConcreteRegisterValue(const triton::arch::Register& reg, const triton::uint512& value, bool execCallbacks) {
    this->checkArchitecture();
    this->arch.setConcreteRegisterValue(reg, value, execCallbacks);
    /*
     * In order to synchronize the concrete state with the symbolic
     * one, the symbolic expression is concretized.
     */
    this->concretizeRegister(reg);
  }
 
 
  void Context::setConcreteState(triton::arch::Architecture& other) {
    if (this->getArchitecture() != other.getArchitecture()) {
      throw triton::exceptions::Engines("Context::setConcreteState(): Not the same architecture.");
    }
 
    switch (this->getArchitecture()) {
      case triton::arch::ARCH_X86_64:
        *static_cast<triton::arch::x86::x8664Cpu*>(this->getCpuInstance()) = *static_cast<triton::arch::x86::x8664Cpu*>(other.getCpuInstance());
        break;
      case triton::arch::ARCH_X86:
        *static_cast<triton::arch::x86::x86Cpu*>(this->getCpuInstance()) = *static_cast<triton::arch::x86::x86Cpu*>(other.getCpuInstance());
        break;
      case triton::arch::ARCH_ARM32:
        *static_cast<triton::arch::arm::arm32::Arm32Cpu*>(this->getCpuInstance()) = *static_cast<triton::arch::arm::arm32::Arm32Cpu*>(other.getCpuInstance());
        break;
      case triton::arch::ARCH_AARCH64:
        *static_cast<triton::arch::arm::aarch64::AArch64Cpu*>(this->getCpuInstance()) = *static_cast<triton::arch::arm::aarch64::AArch64Cpu*>(other.getCpuInstance());
        break;
      #ifdef COMPILE_RISCV
      case triton::arch::ARCH_RV64:
        *static_cast<triton::arch::riscv::riscv64Cpu*>(this->getCpuInstance()) = *static_cast<triton::arch::riscv::riscv64Cpu*>(other.getCpuInstance());
        break;
      case triton::arch::ARCH_RV32:
        *static_cast<triton::arch::riscv::riscv32Cpu*>(this->getCpuInstance()) = *static_cast<triton::arch::riscv::riscv32Cpu*>(other.getCpuInstance());
        break;
      #endif
      default:
        throw triton::exceptions::Engines("Context::setConcreteState(): Invalid architecture.");
    }
 
    this->concretizeAllMemory();
    this->concretizeAllRegister();
  }
 
 
  bool Context::isConcreteMemoryValueDefined(const triton::arch::MemoryAccess& mem) const {
    this->checkArchitecture();
    return this->arch.isConcreteMemoryValueDefined(mem);
  }
 
 
  bool Context::isConcreteMemoryValueDefined(triton::uint64 baseAddr, triton::usize size) const {
    this->checkArchitecture();
    return this->arch.isConcreteMemoryValueDefined(baseAddr, size);
  }
 
 
  void Context::clearConcreteMemoryValue(const triton::arch::MemoryAccess& mem) {
    this->checkArchitecture();
    this->arch.clearConcreteMemoryValue(mem);
  }
 
 
  void Context::clearConcreteMemoryValue(triton::uint64 baseAddr, triton::usize size) {
    this->checkArchitecture();
    this->arch.clearConcreteMemoryValue(baseAddr, size);
  }
 
 
  void Context::disassembly(triton::arch::Instruction& inst) const {
    this->checkArchitecture();
    this->arch.disassembly(inst);
  }
 
 
  void Context::disassembly(triton::arch::BasicBlock& block, triton::uint64 addr) const {
    this->checkArchitecture();
    this->arch.disassembly(block, addr);
  }
 
 
  std::vector<triton::arch::Instruction> Context::disassembly(triton::uint64 addr, triton::usize count) const {
    this->checkArchitecture();
    return this->arch.disassembly(addr, count);
  }
 
 
  triton::arch::BasicBlock Context::disassembly(triton::uint64 addr, bool(*filterCallback)(std::vector<triton::arch::Instruction>&)) const {
    this->checkArchitecture();
    return this->arch.disassembly(addr, filterCallback);
  }
 
 
  triton::arch::BasicBlock Context::disassembly(triton::uint64 addr) const {
    this->checkArchitecture();
    return this->arch.disassembly(addr);
  }
 
 
 
  /* Processing Context ================================================================================ */
 
  void Context::initEngines(void) {
    this->checkArchitecture();
 
    this->symbolic = new(std::nothrow) triton::engines::symbolic::SymbolicEngine(&this->arch, this->modes, this->astCtxt, &this->callbacks);
    if (this->symbolic == nullptr)
      throw triton::exceptions::Context("Context::initEngines(): Not enough memory.");
 
    this->solver = new(std::nothrow) triton::engines::solver::SolverEngine();
    if (this->solver == nullptr)
      throw triton::exceptions::Context("Context::initEngines(): Not enough memory.");
 
    this->taint = new(std::nothrow) triton::engines::taint::TaintEngine(this->modes, this->symbolic, *this->getCpuInstance());
    if (this->taint == nullptr)
      throw triton::exceptions::Context("Context::initEngines(): Not enough memory.");
 
    this->lifting = new(std::nothrow) triton::engines::lifters::LiftingEngine(this->astCtxt, this->symbolic);
    if (this->lifting == nullptr)
      throw triton::exceptions::Context("Context::initEngines(): Not enough memory.");
 
    this->irBuilder = new(std::nothrow) triton::arch::IrBuilder(&this->arch, this->modes, this->astCtxt, this->symbolic, this->taint);
    if (this->irBuilder == nullptr)
      throw triton::exceptions::Context("Context::initEngines(): Not enough memory.");
 
    /* Setup registers shortcut */
    this->registers.init(this->arch.getArchitecture());
  }
 
 
  void Context::removeEngines(void) {
    if (this->isArchitectureValid()) {
      delete this->irBuilder;
      delete this->lifting;
      delete this->solver;
      delete this->symbolic;
      delete this->taint;
 
      this->astCtxt   = nullptr;
      this->irBuilder = nullptr;
      this->lifting   = nullptr;
      this->solver    = nullptr;
      this->symbolic  = nullptr;
      this->taint     = nullptr;
    }
 
    // Clean up the ast context
    this->astCtxt = std::make_shared<triton::ast::AstContext>(this->modes);
 
    // Clean up the registers shortcut
    this->registers.clear();
  }
 
 
  void Context::reset(void) {
    if (this->isArchitectureValid()) {
      this->removeEngines();
      this->initEngines();
      this->clearArchitecture();
      this->clearCallbacks();
      this->clearModes();
    }
  }
 
 
  triton::arch::exception_e Context::processing(triton::arch::Instruction& inst) {
    this->checkArchitecture();
    this->arch.disassembly(inst);
    return this->irBuilder->buildSemantics(inst);
  }
 
 
  triton::arch::exception_e Context::processing(triton::arch::BasicBlock& block, triton::uint64 addr) {
    this->checkArchitecture();
    this->arch.disassembly(block, addr);
    return this->irBuilder->buildSemantics(block);
  }
 
 
 
  /* IR builder Context ================================================================================= */
 
  triton::arch::exception_e Context::buildSemantics(triton::arch::Instruction& inst) {
    this->checkIrBuilder();
    return this->irBuilder->buildSemantics(inst);
  }
 
 
  triton::arch::exception_e Context::buildSemantics(triton::arch::BasicBlock& block) {
    this->checkIrBuilder();
    return this->irBuilder->buildSemantics(block);
  }
 
 
  triton::ast::SharedAstContext Context::getAstContext(void) {
    return this->astCtxt;
  }
 
 
 
  /* AST representation Context ========================================================================= */
 
  triton::ast::representations::mode_e Context::getAstRepresentationMode(void) const {
    return this->astCtxt->getRepresentationMode();
  }
 
 
  void Context::setAstRepresentationMode(triton::ast::representations::mode_e mode) {
    this->astCtxt->setRepresentationMode(mode);
  }
 
 
 
  /* Callbacks Context ================================================================================= */
 
  template TRITON_EXPORT void Context::addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&)> cb);
  template TRITON_EXPORT void Context::addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&)> cb);
  template TRITON_EXPORT void Context::addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&, const triton::uint512& value)> cb);
  template TRITON_EXPORT void Context::addCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&, const triton::uint512& value)> cb);
  template TRITON_EXPORT void Context::addCallback(triton::callbacks::callback_e kind, ComparableFunctor<triton::ast::SharedAbstractNode(triton::Context&, const triton::ast::SharedAbstractNode&)> cb);
 
  template TRITON_EXPORT void Context::removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&)> cb);
  template TRITON_EXPORT void Context::removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&)> cb);
  template TRITON_EXPORT void Context::removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::MemoryAccess&, const triton::uint512& value)> cb);
  template TRITON_EXPORT void Context::removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<void(triton::Context&, const triton::arch::Register&, const triton::uint512& value)> cb);
  template TRITON_EXPORT void Context::removeCallback(triton::callbacks::callback_e kind, ComparableFunctor<triton::ast::SharedAbstractNode(triton::Context&, const triton::ast::SharedAbstractNode&)> cb);
 
 
  void Context::clearCallbacks(void) {
    this->callbacks.clearCallbacks();
  }
 
 
  triton::ast::SharedAbstractNode Context::processCallbacks(triton::callbacks::callback_e kind, triton::ast::SharedAbstractNode node) {
    if (this->callbacks.isDefined()) {
      return this->callbacks.processCallbacks(kind, node);
    }
    return node;
  }
 
 
  void Context::processCallbacks(triton::callbacks::callback_e kind, const triton::arch::MemoryAccess& mem) {
    if (this->callbacks.isDefined()) {
      this->callbacks.processCallbacks(kind, mem);
    }
  }
 
 
  void Context::processCallbacks(triton::callbacks::callback_e kind, const triton::arch::Register& reg) {
    if (this->callbacks.isDefined()) {
      this->callbacks.processCallbacks(kind, reg);
    }
  }
 
 
 
  /* Modes Context======================================================================================= */
 
  void Context::setMode(triton::modes::mode_e mode, bool flag) {
    this->modes->setMode(mode, flag);
  }
 
 
  bool Context::isModeEnabled(triton::modes::mode_e mode) const {
    return this->modes->isModeEnabled(mode);
  }
 
 
  void Context::clearModes(void) {
    this->modes->clearModes();
  }
 
 
 
  /* Symbolic engine Context ============================================================================ */
 
  triton::engines::symbolic::SymbolicEngine* Context::getSymbolicEngine(void) {
    this->checkSymbolic();
    return this->symbolic;
  }
 
 
  triton::engines::symbolic::SharedSymbolicVariable Context::symbolizeExpression(triton::usize exprId, triton::uint32 symVarSize, const std::string& symVarAlias) {
    this->checkSymbolic();
    return this->symbolic->symbolizeExpression(exprId, symVarSize, symVarAlias);
  }
 
 
  triton::engines::symbolic::SharedSymbolicVariable Context::symbolizeMemory(const triton::arch::MemoryAccess& mem, const std::string& symVarAlias) {
    this->checkSymbolic();
    return this->symbolic->symbolizeMemory(mem, symVarAlias);
  }
 
 
  void Context::symbolizeMemory(triton::uint64 addr, triton::usize size) {
    this->checkSymbolic();
    this->symbolic->symbolizeMemory(addr, size);
  }
 
 
  triton::engines::symbolic::SharedSymbolicVariable Context::symbolizeRegister(const triton::arch::Register& reg, const std::string& symVarAlias) {
    this->checkSymbolic();
    return this->symbolic->symbolizeRegister(reg, symVarAlias);
  }
 
 
  triton::ast::SharedAbstractNode Context::getOperandAst(const triton::arch::OperandWrapper& op) {
    this->checkSymbolic();
    return this->symbolic->getOperandAst(op);
  }
 
 
  triton::ast::SharedAbstractNode Context::getOperandAst(triton::arch::Instruction& inst, const triton::arch::OperandWrapper& op) {
    this->checkSymbolic();
    return this->symbolic->getOperandAst(inst, op);
  }
 
 
  triton::ast::SharedAbstractNode Context::getImmediateAst(const triton::arch::Immediate& imm) {
    this->checkSymbolic();
    return this->symbolic->getImmediateAst(imm);
  }
 
 
  triton::ast::SharedAbstractNode Context::getImmediateAst(triton::arch::Instruction& inst, const triton::arch::Immediate& imm) {
    this->checkSymbolic();
    return this->symbolic->getImmediateAst(inst, imm);
  }
 
 
  triton::ast::SharedAbstractNode Context::getMemoryAst(const triton::arch::MemoryAccess& mem) {
    this->checkSymbolic();
    return this->symbolic->getMemoryAst(mem);
  }
 
 
  triton::ast::SharedAbstractNode Context::getMemoryAst(triton::arch::Instruction& inst, const triton::arch::MemoryAccess& mem) {
    this->checkSymbolic();
    return this->symbolic->getMemoryAst(inst, mem);
  }
 
 
  triton::ast::SharedAbstractNode Context::getRegisterAst(const triton::arch::Register& reg) {
    this->checkSymbolic();
    return this->symbolic->getRegisterAst(reg);
  }
 
 
  triton::ast::SharedAbstractNode Context::getRegisterAst(triton::arch::Instruction& inst, const triton::arch::Register& reg) {
    this->checkSymbolic();
    return this->symbolic->getRegisterAst(inst, reg);
  }
 
 
  triton::engines::symbolic::SharedSymbolicExpression Context::newSymbolicExpression(const triton::ast::SharedAbstractNode& node, const std::string& comment) {
    this->checkSymbolic();
    return this->symbolic->newSymbolicExpression(node, triton::engines::symbolic::VOLATILE_EXPRESSION, comment);
  }
 
 
  triton::engines::symbolic::SharedSymbolicVariable Context::newSymbolicVariable(triton::uint32 varSize, const std::string& alias) {
    this->checkSymbolic();
    return this->symbolic->newSymbolicVariable(triton::engines::symbolic::UNDEFINED_VARIABLE, 0, varSize, alias);
  }
 
 
  void Context::removeSymbolicExpression(const triton::engines::symbolic::SharedSymbolicExpression& expr) {
    this->checkSymbolic();
    return this->symbolic->removeSymbolicExpression(expr);
  }
 
 
  const triton::engines::symbolic::SharedSymbolicExpression& Context::createSymbolicExpression(triton::arch::Instruction& inst, const triton::ast::SharedAbstractNode& node, const triton::arch::OperandWrapper& dst, const std::string& comment) {
    this->checkSymbolic();
    return this->symbolic->createSymbolicExpression(inst, node, dst, comment);
  }
 
 
  const triton::engines::symbolic::SharedSymbolicExpression& Context::createSymbolicMemoryExpression(triton::arch::Instruction& inst, const triton::ast::SharedAbstractNode& node, const triton::arch::MemoryAccess& mem, const std::string& comment) {
    this->checkSymbolic();
    return this->symbolic->createSymbolicMemoryExpression(inst, node, mem, comment);
  }
 
 
  const triton::engines::symbolic::SharedSymbolicExpression& Context::createSymbolicRegisterExpression(triton::arch::Instruction& inst, const triton::ast::SharedAbstractNode& node, const triton::arch::Register& reg, const std::string& comment) {
    this->checkSymbolic();
    return this->symbolic->createSymbolicRegisterExpression(inst, node, reg, comment);
  }
 
 
  const triton::engines::symbolic::SharedSymbolicExpression& Context::createSymbolicVolatileExpression(triton::arch::Instruction& inst, const triton::ast::SharedAbstractNode& node, const std::string& comment) {
    this->checkSymbolic();
    return this->symbolic->createSymbolicVolatileExpression(inst, node, comment);
  }
 
 
  void Context::assignSymbolicExpressionToMemory(const triton::engines::symbolic::SharedSymbolicExpression& se, const triton::arch::MemoryAccess& mem) {
    this->checkSymbolic();
    this->symbolic->assignSymbolicExpressionToMemory(se, mem);
  }
 
 
  void Context::assignSymbolicExpressionToRegister(const triton::engines::symbolic::SharedSymbolicExpression& se, const triton::arch::Register& reg) {
    this->checkSymbolic();
    this->symbolic->assignSymbolicExpressionToRegister(se, reg);
  }
 
 
  triton::engines::symbolic::SharedSymbolicExpression Context::getSymbolicMemory(triton::uint64 addr) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicMemory(addr);
  }
 
 
  std::unordered_map<triton::arch::register_e, triton::engines::symbolic::SharedSymbolicExpression> Context::getSymbolicRegisters(void) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicRegisters();
  }
 
 
  std::unordered_map<triton::uint64, triton::engines::symbolic::SharedSymbolicExpression> Context::getSymbolicMemory(void) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicMemory();
  }
 
 
  const triton::engines::symbolic::SharedSymbolicExpression& Context::getSymbolicRegister(const triton::arch::Register& reg) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicRegister(reg);
  }
 
 
  triton::uint8 Context::getSymbolicMemoryValue(triton::uint64 address) {
    this->checkSymbolic();
    return this->symbolic->getSymbolicMemoryValue(address);
  }
 
 
  triton::uint512 Context::getSymbolicMemoryValue(const triton::arch::MemoryAccess& mem) {
    this->checkSymbolic();
    return this->symbolic->getSymbolicMemoryValue(mem);
  }
 
 
  std::vector<triton::uint8> Context::getSymbolicMemoryAreaValue(triton::uint64 baseAddr, triton::usize size) {
    this->checkSymbolic();
    return this->symbolic->getSymbolicMemoryAreaValue(baseAddr, size);
  }
 
 
  triton::uint512 Context::getSymbolicRegisterValue(const triton::arch::Register& reg) {
    this->checkSymbolic();
    return this->symbolic->getSymbolicRegisterValue(reg);
  }
 
 
  triton::ast::SharedAbstractNode Context::simplify(const triton::ast::SharedAbstractNode& node, bool usingSolver, bool usingLLVM) const {
    if (usingSolver) {
      return this->simplifyAstViaSolver(node);
    }
    else if (usingLLVM) {
      return this->simplifyAstViaLLVM(node);
    }
    else {
      this->checkSymbolic();
      return this->symbolic->simplify(node);
    }
  }
 
 
  triton::arch::BasicBlock Context::simplify(const triton::arch::BasicBlock& block, bool padding) const {
    this->checkSymbolic();
    return this->symbolic->simplify(block, padding);
  }
 
 
  triton::engines::symbolic::SharedSymbolicExpression Context::getSymbolicExpression(triton::usize symExprId) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicExpression(symExprId);
  }
 
 
  triton::uint512 Context::getConcreteVariableValue(const triton::engines::symbolic::SharedSymbolicVariable& symVar) const {
    this->checkSymbolic();
    return this->symbolic->getConcreteVariableValue(symVar);
  }
 
 
  void Context::setConcreteVariableValue(const triton::engines::symbolic::SharedSymbolicVariable& symVar, const triton::uint512& value) {
    this->checkSymbolic();
    this->symbolic->setConcreteVariableValue(symVar, value);
  }
 
 
  triton::engines::symbolic::SharedSymbolicVariable Context::getSymbolicVariable(triton::usize symVarId) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicVariable(symVarId);
  }
 
 
  triton::engines::symbolic::SharedSymbolicVariable Context::getSymbolicVariable(const std::string& symVarName) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicVariable(symVarName);
  }
 
 
  const std::vector<triton::engines::symbolic::PathConstraint>& Context::getPathConstraints(void) const {
    this->checkSymbolic();
    return this->symbolic->getPathConstraints();
  }
 
 
  std::vector<triton::engines::symbolic::PathConstraint> Context::getPathConstraints(triton::usize start, triton::usize end) const {
    this->checkSymbolic();
    return this->symbolic->getPathConstraints(start, end);
  }
 
 
  std::vector<triton::engines::symbolic::PathConstraint> Context::getPathConstraintsOfThread(triton::uint32 threadId) const {
    this->checkSymbolic();
    return this->symbolic->getPathConstraintsOfThread(threadId);
  }
 
 
  triton::usize Context::getSizeOfPathConstraints(void) const {
    this->checkSymbolic();
    return this->symbolic->getSizeOfPathConstraints();
  }
 
 
  triton::ast::SharedAbstractNode Context::getPathPredicate(void) {
    this->checkSymbolic();
    return this->symbolic->getPathPredicate();
  }
 
 
  std::vector<triton::ast::SharedAbstractNode> Context::getPredicatesToReachAddress(triton::uint64 addr) {
    this->checkSymbolic();
    return this->symbolic->getPredicatesToReachAddress(addr);
  }
 
 
  void Context::pushPathConstraint(const triton::ast::SharedAbstractNode& node, const std::string& comment) {
    this->checkSymbolic();
    this->symbolic->pushPathConstraint(node, comment);
  }
 
 
  void Context::pushPathConstraint(const triton::engines::symbolic::PathConstraint& pco) {
    this->checkSymbolic();
    this->symbolic->pushPathConstraint(pco);
  }
 
 
  void Context::popPathConstraint(void) {
    this->checkSymbolic();
    this->symbolic->popPathConstraint();
  }
 
 
  void Context::clearPathConstraints(void) {
    this->checkSymbolic();
    this->symbolic->clearPathConstraints();
  }
 
 
  bool Context::isSymbolicExpressionExists(triton::usize symExprId) const {
    this->checkSymbolic();
    return this->symbolic->isSymbolicExpressionExists(symExprId);
  }
 
 
  bool Context::isMemorySymbolized(const triton::arch::MemoryAccess& mem) const {
    this->checkSymbolic();
    return this->symbolic->isMemorySymbolized(mem);
  }
 
 
  bool Context::isMemorySymbolized(triton::uint64 addr, triton::uint32 size) const {
    this->checkSymbolic();
    return this->symbolic->isMemorySymbolized(addr, size);
  }
 
 
  bool Context::isRegisterSymbolized(const triton::arch::Register& reg) const {
    this->checkSymbolic();
    return this->symbolic->isRegisterSymbolized(reg);
  }
 
 
  void Context::concretizeAllMemory(void) {
    this->checkSymbolic();
    this->symbolic->concretizeAllMemory();
  }
 
 
  void Context::concretizeAllRegister(void) {
    this->checkSymbolic();
    this->symbolic->concretizeAllRegister();
  }
 
 
  void Context::concretizeMemory(const triton::arch::MemoryAccess& mem) {
    this->checkSymbolic();
    this->symbolic->concretizeMemory(mem);
  }
 
 
  void Context::concretizeMemory(triton::uint64 addr) {
    this->checkSymbolic();
    this->symbolic->concretizeMemory(addr);
  }
 
 
  void Context::concretizeRegister(const triton::arch::Register& reg) {
    this->checkSymbolic();
    this->symbolic->concretizeRegister(reg);
  }
 
 
  std::unordered_map<triton::usize, triton::engines::symbolic::SharedSymbolicExpression> Context::sliceExpressions(const triton::engines::symbolic::SharedSymbolicExpression& expr) {
    this->checkSymbolic();
    return this->symbolic->sliceExpressions(expr);
  }
 
 
  std::vector<triton::engines::symbolic::SharedSymbolicExpression> Context::getTaintedSymbolicExpressions(void) const {
    this->checkSymbolic();
    return this->symbolic->getTaintedSymbolicExpressions();
  }
 
 
  std::unordered_map<triton::usize, triton::engines::symbolic::SharedSymbolicExpression> Context::getSymbolicExpressions(void) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicExpressions();
  }
 
 
  std::map<triton::usize, triton::engines::symbolic::SharedSymbolicVariable> Context::getSymbolicVariables(void) const {
    this->checkSymbolic();
    return this->symbolic->getSymbolicVariables();
  }
 
 
 
  /* Solver engine Context ============================================================================= */
 
  triton::engines::solver::solver_e Context::getSolver(void) const {
    this->checkSolver();
    return this->solver->getSolver();
  }
 
 
  const triton::engines::solver::SolverInterface* Context::getSolverInstance(void) const {
    this->checkSolver();
    return this->solver->getSolverInstance();
  }
 
 
  void Context::setSolver(triton::engines::solver::solver_e kind) {
    this->checkSolver();
    this->solver->setSolver(kind);
  }
 
 
  void Context::setCustomSolver(triton::engines::solver::SolverInterface* customSolver) {
    this->checkSolver();
    this->solver->setCustomSolver(customSolver);
  }
 
 
  bool Context::isSolverValid(void) const {
    this->checkSolver();
    return this->solver->isValid();
  }
 
 
  std::unordered_map<triton::usize, triton::engines::solver::SolverModel> Context::getModel(const triton::ast::SharedAbstractNode& node, triton::engines::solver::status_e* status, triton::uint32 timeout, triton::uint32* solvingTime) const {
    this->checkSolver();
    return this->solver->getModel(node, status, timeout, solvingTime);
  }
 
 
  std::vector<std::unordered_map<triton::usize, triton::engines::solver::SolverModel>> Context::getModels(const triton::ast::SharedAbstractNode& node, triton::uint32 limit, triton::engines::solver::status_e* status, triton::uint32 timeout, triton::uint32* solvingTime) const {
    this->checkSolver();
    return this->solver->getModels(node, limit, status, timeout, solvingTime);
  }
 
 
  bool Context::isSat(const triton::ast::SharedAbstractNode& node, triton::engines::solver::status_e* status, triton::uint32 timeout, triton::uint32* solvingTime) const {
    this->checkSolver();
    return this->solver->isSat(node, status, timeout, solvingTime);
  }
 
 
  triton::uint512 Context::evaluateAstViaSolver(const triton::ast::SharedAbstractNode& node) const {
    this->checkSolver();
    #ifdef TRITON_Z3_INTERFACE
    if (this->getSolver() == triton::engines::solver::SOLVER_Z3) {
      return reinterpret_cast<const triton::engines::solver::Z3Solver*>(this->getSolverInstance())->evaluate(node);
    }
    #endif
    #ifdef TRITON_BITWUZLA_INTERFACE
    if (this->getSolver() == triton::engines::solver::SOLVER_BITWUZLA) {
      return reinterpret_cast<const triton::engines::solver::BitwuzlaSolver*>(this->getSolverInstance())->evaluate(node);
    }
    #endif
    throw triton::exceptions::Context("Context::evaluateAstViaZ3(): Solver instance must be a SOLVER_Z3 or SOLVER_BITWUZLA.");
  }
 
 
  triton::ast::SharedAbstractNode Context::simplifyAstViaSolver(const triton::ast::SharedAbstractNode& node) const {
    this->checkSolver();
    #ifdef TRITON_Z3_INTERFACE
    if (this->getSolver() == triton::engines::solver::SOLVER_Z3) {
      return reinterpret_cast<const triton::engines::solver::Z3Solver*>(this->getSolverInstance())->simplify(node);
    }
    #endif
    throw triton::exceptions::Context("Context::simplifyAstViaSolver(): Solver instance must be a SOLVER_Z3.");
  }
 
 
  void Context::setSolverTimeout(triton::uint32 ms) {
    this->checkSolver();
    this->solver->setTimeout(ms);
  }
 
 
  void Context::setSolverMemoryLimit(triton::uint32 limit) {
    this->checkSolver();
    this->solver->setMemoryLimit(limit);
  }
 
 
 
  /* Taint engine Context ============================================================================== */
 
  triton::engines::taint::TaintEngine* Context::getTaintEngine(void) {
    this->checkTaint();
    return this->taint;
  }
 
 
  const std::unordered_set<triton::uint64>& Context::getTaintedMemory(void) const {
    this->checkTaint();
    return this->taint->getTaintedMemory();
  }
 
 
  std::unordered_set<const triton::arch::Register*> Context::getTaintedRegisters(void) const {
    this->checkTaint();
    return this->taint->getTaintedRegisters();
  }
 
 
  bool Context::isTainted(const triton::arch::OperandWrapper& op) const {
    this->checkTaint();
    return this->taint->isTainted(op);
  }
 
 
  bool Context::isMemoryTainted(triton::uint64 addr, uint32 size) const {
    this->checkTaint();
    return this->taint->isMemoryTainted(addr, size);
  }
 
 
  bool Context::isMemoryTainted(const triton::arch::MemoryAccess& mem) const {
    this->checkTaint();
    return this->taint->isMemoryTainted(mem);
  }
 
 
  bool Context::isRegisterTainted(const triton::arch::Register& reg) const {
    this->checkTaint();
    return this->taint->isRegisterTainted(reg);
  }
 
 
  bool Context::setTaint(const triton::arch::OperandWrapper& op, bool flag) {
    this->checkTaint();
    return this->taint->setTaint(op, flag);
  }
 
 
  bool Context::setTaintMemory(const triton::arch::MemoryAccess& mem, bool flag) {
    this->checkTaint();
    this->taint->setTaintMemory(mem, flag);
    return flag;
  }
 
 
  bool Context::setTaintRegister(const triton::arch::Register& reg, bool flag) {
    this->checkTaint();
    this->taint->setTaintRegister(reg, flag);
    return flag;
  }
 
 
  bool Context::taintMemory(triton::uint64 addr) {
    this->checkTaint();
    return this->taint->taintMemory(addr);
  }
 
 
  bool Context::taintMemory(const triton::arch::MemoryAccess& mem) {
    this->checkTaint();
    return this->taint->taintMemory(mem);
  }
 
 
  bool Context::taintRegister(const triton::arch::Register& reg) {
    this->checkTaint();
    return this->taint->taintRegister(reg);
  }
 
 
  bool Context::untaintMemory(triton::uint64 addr) {
    this->checkTaint();
    return this->taint->untaintMemory(addr);
  }
 
 
  bool Context::untaintMemory(const triton::arch::MemoryAccess& mem) {
    this->checkTaint();
    return this->taint->untaintMemory(mem);
  }
 
 
  bool Context::untaintRegister(const triton::arch::Register& reg) {
    this->checkTaint();
    return this->taint->untaintRegister(reg);
  }
 
 
  bool Context::taintUnion(const triton::arch::OperandWrapper& op1, const triton::arch::OperandWrapper& op2) {
    this->checkTaint();
    return this->taint->taintUnion(op1, op2);
  }
 
 
  bool Context::taintUnion(const triton::arch::MemoryAccess& memDst, const triton::arch::Immediate& imm) {
    this->checkTaint();
    return this->taint->taintUnion(memDst, imm);
  }
 
 
  bool Context::taintUnion(const triton::arch::MemoryAccess& memDst, const triton::arch::MemoryAccess& memSrc) {
    this->checkTaint();
    return this->taint->taintUnion(memDst, memSrc);
  }
 
 
  bool Context::taintUnion(const triton::arch::MemoryAccess& memDst, const triton::arch::Register& regSrc) {
    this->checkTaint();
    return this->taint->taintUnion(memDst, regSrc);
  }
 
 
  bool Context::taintUnion(const triton::arch::Register& regDst, const triton::arch::Immediate& imm) {
    this->checkTaint();
    return this->taint->taintUnion(regDst, imm);
  }
 
 
  bool Context::taintUnion(const triton::arch::Register& regDst, const triton::arch::MemoryAccess& memSrc) {
    this->checkTaint();
    return this->taint->taintUnion(regDst, memSrc);
  }
 
 
  bool Context::taintUnion(const triton::arch::Register& regDst, const triton::arch::Register& regSrc) {
    this->checkTaint();
    return this->taint->taintUnion(regDst, regSrc);
  }
 
 
  bool Context::taintAssignment(const triton::arch::OperandWrapper& op1, const triton::arch::OperandWrapper& op2) {
    this->checkTaint();
    return this->taint->taintAssignment(op1, op2);
  }
 
 
  bool Context::taintAssignment(const triton::arch::MemoryAccess& memDst, const triton::arch::Immediate& imm) {
    this->checkTaint();
    return this->taint->taintAssignment(memDst, imm);
  }
 
 
  bool Context::taintAssignment(const triton::arch::MemoryAccess& memDst, const triton::arch::MemoryAccess& memSrc) {
    this->checkTaint();
    return this->taint->taintAssignment(memDst, memSrc);
  }
 
 
  bool Context::taintAssignment(const triton::arch::MemoryAccess& memDst, const triton::arch::Register& regSrc) {
    this->checkTaint();
    return this->taint->taintAssignment(memDst, regSrc);
  }
 
 
  bool Context::taintAssignment(const triton::arch::Register& regDst, const triton::arch::Immediate& imm) {
    this->checkTaint();
    return this->taint->taintAssignment(regDst, imm);
  }
 
 
  bool Context::taintAssignment(const triton::arch::Register& regDst, const triton::arch::MemoryAccess& memSrc) {
    this->checkTaint();
    return this->taint->taintAssignment(regDst, memSrc);
  }
 
 
  bool Context::taintAssignment(const triton::arch::Register& regDst, const triton::arch::Register& regSrc) {
    this->checkTaint();
    return this->taint->taintAssignment(regDst, regSrc);
  }
 
 
 
  /* Synthesizer engine Context ============================================================================= */
 
  triton::engines::synthesis::SynthesisResult Context::synthesize(const triton::ast::SharedAbstractNode& node, bool constant, bool subexpr, bool opaque) {
    this->checkSymbolic();
    triton::engines::synthesis::Synthesizer synth(this->symbolic);
    return synth.synthesize(node, constant, subexpr, opaque);
  }
 
 
 
  /* Lifters engine Context ================================================================================= */
 
  std::ostream& Context::liftToLLVM(std::ostream& stream, const triton::ast::SharedAbstractNode& node, const char* fname, bool optimize) {
    this->checkLifting();
    #ifdef TRITON_LLVM_INTERFACE
    return this->lifting->liftToLLVM(stream, node, fname, optimize);
    #endif
    throw triton::exceptions::Context("Context::liftToLLVM(): Triton not built with LLVM");
  }
 
 
  std::ostream& Context::liftToLLVM(std::ostream& stream, const triton::engines::symbolic::SharedSymbolicExpression& expr, const char* fname, bool optimize) {
    return this->liftToLLVM(stream, expr->getAst(), fname, optimize);
  }
 
 
  std::ostream& Context::liftToPython(std::ostream& stream, const triton::engines::symbolic::SharedSymbolicExpression& expr, bool icomment) {
    this->checkLifting();
    return this->lifting->liftToPython(stream, expr, icomment);
  }
 
 
  std::ostream& Context::liftToSMT(std::ostream& stream, const triton::engines::symbolic::SharedSymbolicExpression& expr, bool assert_, bool icomment) {
    this->checkLifting();
    return this->lifting->liftToSMT(stream, expr, assert_, icomment);
  }
 
 
  std::ostream& Context::liftToDot(std::ostream& stream, const triton::ast::SharedAbstractNode& node) {
    this->checkLifting();
    return this->lifting->liftToDot(stream, node);
  }
 
 
  std::ostream& Context::liftToDot(std::ostream& stream, const triton::engines::symbolic::SharedSymbolicExpression& expr) {
    this->checkLifting();
    return this->lifting->liftToDot(stream, expr);
  }
 
 
  triton::ast::SharedAbstractNode Context::simplifyAstViaLLVM(const triton::ast::SharedAbstractNode& node) const {
    this->checkLifting();
    #ifdef TRITON_LLVM_INTERFACE
    return this->lifting->simplifyAstViaLLVM(node);
    #endif
    throw triton::exceptions::Context("Context::simplifyAstViaLLVM(): Triton not built with LLVM");
  }
 
}; /* triton namespace */