taintEngine.hpp

Go to the documentation of this file.

 
/*
**  Copyright (C) - Triton
**
**  This program is under the terms of the Apache License 2.0.
*/
 
#ifndef TRITON_TAINTENGINE_H
#define TRITON_TAINTENGINE_H
 
#include <unordered_set>
 
#include <triton/dllexport.hpp>
#include <triton/memoryAccess.hpp>
#include <triton/modes.hpp>
#include <triton/register.hpp>
#include <triton/symbolicEngine.hpp>
#include <triton/tritonTypes.hpp>
 
 
 
namespace triton {
  namespace engines {
    namespace taint {
      const bool TAINTED = true;
 
      const bool UNTAINTED = !TAINTED;
 
      class TaintEngine {
        private:
          triton::modes::SharedModes modes;
 
          triton::engines::symbolic::SymbolicEngine* symbolicEngine;
 
          triton::arch::CpuInterface& cpu;
 
        protected:
          std::unordered_set<triton::uint64> taintedMemory;
 
          std::unordered_set<triton::arch::register_e> taintedRegisters;
 
        public:
          TRITON_EXPORT TaintEngine(const triton::modes::SharedModes& modes, triton::engines::symbolic::SymbolicEngine* symbolicEngine, triton::arch::CpuInterface& cpu);
 
          TRITON_EXPORT TaintEngine(const TaintEngine& other);
 
          TRITON_EXPORT TaintEngine& operator=(const TaintEngine& other);
 
          TRITON_EXPORT const std::unordered_set<triton::uint64>& getTaintedMemory(void) const;
 
          TRITON_EXPORT std::unordered_set<const triton::arch::Register*> getTaintedRegisters(void) const;
 
          TRITON_EXPORT bool isMemoryTainted(triton::uint64 addr, triton::uint32 size=1) const;
 
          TRITON_EXPORT bool isMemoryTainted(const triton::arch::MemoryAccess& mem, bool mode=true) const;
 
          TRITON_EXPORT bool isRegisterTainted(const triton::arch::Register& reg) const;
 
          TRITON_EXPORT bool isTainted(const triton::arch::OperandWrapper& op) const;
 
          TRITON_EXPORT bool setTaint(const triton::arch::OperandWrapper& op, bool flag);
 
          TRITON_EXPORT bool setTaintMemory(const triton::arch::MemoryAccess& mem, bool flag);
 
          TRITON_EXPORT bool setTaintRegister(const triton::arch::Register& reg, bool flag);
 
          TRITON_EXPORT bool taintMemory(triton::uint64 addr);
 
          TRITON_EXPORT bool taintMemory(const triton::arch::MemoryAccess& mem);
 
          TRITON_EXPORT bool taintRegister(const triton::arch::Register& reg);
 
          TRITON_EXPORT bool untaintMemory(triton::uint64 addr);
 
          TRITON_EXPORT bool untaintMemory(const triton::arch::MemoryAccess& mem);
 
          TRITON_EXPORT bool untaintRegister(const triton::arch::Register& reg);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::OperandWrapper& op1, const triton::arch::OperandWrapper& op2);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::MemoryAccess& memDst, const triton::arch::Immediate& imm);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::MemoryAccess& memDst, const triton::arch::MemoryAccess& memSrc);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::MemoryAccess& memDst, const triton::arch::Register& regSrc);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::Register& regDst, const triton::arch::Immediate& imm);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::Register& regDst, const triton::arch::MemoryAccess& memSrc);
 
          TRITON_EXPORT bool taintUnion(const triton::arch::Register& regDst, const triton::arch::Register& regSrc);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::OperandWrapper& op1, const triton::arch::OperandWrapper& op2);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::MemoryAccess& memDst,  const triton::arch::Immediate& imm);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::MemoryAccess& memDst, const triton::arch::MemoryAccess& memSrc);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::MemoryAccess& memDst, const triton::arch::Register& regSrc);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::Register& regDst,  const triton::arch::Immediate& imm);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::Register& regDst, const triton::arch::MemoryAccess& memSrc);
 
          TRITON_EXPORT bool taintAssignment(const triton::arch::Register& regDst, const triton::arch::Register& regSrc);
 
        private:
          bool unionMemoryImmediate(const triton::arch::MemoryAccess& memDst);
 
          bool unionMemoryMemory(const triton::arch::MemoryAccess& memDst, const triton::arch::MemoryAccess& memSrc);
 
          bool unionMemoryRegister(const triton::arch::MemoryAccess& memDst, const triton::arch::Register& regSrc);
 
          bool unionRegisterImmediate(const triton::arch::Register& regDst);
 
          bool unionRegisterMemory(const triton::arch::Register& regDst, const triton::arch::MemoryAccess& memSrc);
 
          bool unionRegisterRegister(const triton::arch::Register& regDst, const triton::arch::Register& regSrc);
 
          bool assignmentMemoryImmediate(const triton::arch::MemoryAccess& memDst);
 
          bool assignmentMemoryMemory(const triton::arch::MemoryAccess& memDst, const triton::arch::MemoryAccess& memSrc);
 
          bool assignmentMemoryRegister(const triton::arch::MemoryAccess& memDst, const triton::arch::Register& regSrc);
 
          bool assignmentRegisterImmediate(const triton::arch::Register& regDst);
 
          bool assignmentRegisterMemory(const triton::arch::Register& regDst, const triton::arch::MemoryAccess& memSrc);
 
          bool assignmentRegisterRegister(const triton::arch::Register& regDst, const triton::arch::Register& regSrc);
      };
 
    };
  };
};
 
#endif /* !__TRITON_TAINTENGINE_H__ */