x86Cpu.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_X86CPU_HPP
#define TRITON_X86CPU_HPP
 
#include <set>
#include <string>
#include <unordered_map>
#include <vector>
 
#include <triton/archEnums.hpp>
#include <triton/callbacks.hpp>
#include <triton/cpuInterface.hpp>
#include <triton/dllexport.hpp>
#include <triton/instruction.hpp>
#include <triton/memoryAccess.hpp>
#include <triton/register.hpp>
#include <triton/tritonTypes.hpp>
#include <triton/x86Specifications.hpp>
 
 
 
namespace triton {
  namespace arch {
    namespace x86 {
 
      class x86Cpu : public CpuInterface, public x86Specifications {
 
        static const triton::arch::register_e pcId = triton::arch::ID_REG_X86_EIP;
        static const triton::arch::register_e spId = triton::arch::ID_REG_X86_ESP;
 
        private:
          triton::callbacks::Callbacks* callbacks;
 
          std::size_t handle;
 
          void copy(const x86Cpu& other);
 
          void disassInit(void);
 
        protected:
          std::unordered_map<triton::uint64, triton::uint8, IdentityHash<triton::uint64>> memory;
 
          triton::uint8 eax[triton::size::dword];
          triton::uint8 ebx[triton::size::dword];
          triton::uint8 ecx[triton::size::dword];
          triton::uint8 edx[triton::size::dword];
          triton::uint8 edi[triton::size::dword];
          triton::uint8 esi[triton::size::dword];
          triton::uint8 ebp[triton::size::dword];
          triton::uint8 esp[triton::size::dword];
          triton::uint8 eip[triton::size::dword];
          triton::uint8 eflags[triton::size::dword];
          triton::uint8 st0[triton::size::fword];
          triton::uint8 st1[triton::size::fword];
          triton::uint8 st2[triton::size::fword];
          triton::uint8 st3[triton::size::fword];
          triton::uint8 st4[triton::size::fword];
          triton::uint8 st5[triton::size::fword];
          triton::uint8 st6[triton::size::fword];
          triton::uint8 st7[triton::size::fword];
          triton::uint8 ymm0[triton::size::qqword];
          triton::uint8 ymm1[triton::size::qqword];
          triton::uint8 ymm2[triton::size::qqword];
          triton::uint8 ymm3[triton::size::qqword];
          triton::uint8 ymm4[triton::size::qqword];
          triton::uint8 ymm5[triton::size::qqword];
          triton::uint8 ymm6[triton::size::qqword];
          triton::uint8 ymm7[triton::size::qqword];
          triton::uint8 cr0[triton::size::dword];
          triton::uint8 cr1[triton::size::dword];
          triton::uint8 cr2[triton::size::dword];
          triton::uint8 cr3[triton::size::dword];
          triton::uint8 cr4[triton::size::dword];
          triton::uint8 cr5[triton::size::dword];
          triton::uint8 cr6[triton::size::dword];
          triton::uint8 cr7[triton::size::dword];
          triton::uint8 cr8[triton::size::dword];
          triton::uint8 cr9[triton::size::dword];
          triton::uint8 cr10[triton::size::dword];
          triton::uint8 cr11[triton::size::dword];
          triton::uint8 cr12[triton::size::dword];
          triton::uint8 cr13[triton::size::dword];
          triton::uint8 cr14[triton::size::dword];
          triton::uint8 cr15[triton::size::dword];
          triton::uint8 cs[triton::size::dword];
          triton::uint8 ds[triton::size::dword];
          triton::uint8 es[triton::size::dword];
          triton::uint8 fs[triton::size::dword];
          triton::uint8 gs[triton::size::dword];
          triton::uint8 ss[triton::size::dword];
          triton::uint8 dr0[triton::size::dword];
          triton::uint8 dr1[triton::size::dword];
          triton::uint8 dr2[triton::size::dword];
          triton::uint8 dr3[triton::size::dword];
          triton::uint8 dr6[triton::size::dword];
          triton::uint8 dr7[triton::size::dword];
          triton::uint8 fcw[triton::size::word];
          triton::uint8 fsw[triton::size::word];
          triton::uint8 ftw[triton::size::word];
          triton::uint8 fop[triton::size::word];
          triton::uint8 fip[triton::size::qword];
          triton::uint8 fcs[triton::size::word];
          triton::uint8 fdp[triton::size::qword];
          triton::uint8 fds[triton::size::word];
          triton::uint8 efer[triton::size::qword];
          triton::uint8 mxcsr[triton::size::dword];
          triton::uint8 mxcsr_mask[triton::size::dword];
          triton::uint8 tsc[triton::size::qword];
 
        public:
          TRITON_EXPORT x86Cpu(triton::callbacks::Callbacks* callbacks=nullptr);
 
          TRITON_EXPORT x86Cpu(const x86Cpu& other);
 
          TRITON_EXPORT virtual ~x86Cpu();
 
          TRITON_EXPORT x86Cpu& operator=(const x86Cpu& other);
 
          TRITON_EXPORT bool isGPR(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isMMX(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isSTX(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isSSECTL(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isSSE(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isFPU(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isEFER(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isTSC(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isAVX256(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isControl(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isDebug(triton::arch::register_e regId) const;
 
          TRITON_EXPORT bool isSegment(triton::arch::register_e regId) const;
 
          /* Virtual pure inheritance ================================================= */
          TRITON_EXPORT bool isFlag(triton::arch::register_e regId) const;
          TRITON_EXPORT bool isRegister(triton::arch::register_e regId) const;
          TRITON_EXPORT bool isRegisterValid(triton::arch::register_e regId) const;
          TRITON_EXPORT bool isThumb(void) const;
          TRITON_EXPORT bool isMemoryExclusive(const triton::arch::MemoryAccess& mem) const;
          TRITON_EXPORT const std::unordered_map<triton::arch::register_e, const triton::arch::Register>& getAllRegisters(void) const;
          TRITON_EXPORT const std::unordered_map<triton::uint64, triton::uint8, IdentityHash<triton::uint64>>& getConcreteMemory(void) const;
          TRITON_EXPORT const triton::arch::Register& getParentRegister(const triton::arch::Register& reg) const;
          TRITON_EXPORT const triton::arch::Register& getParentRegister(triton::arch::register_e id) const;
          TRITON_EXPORT const triton::arch::Register& getProgramCounter(void) const;
          TRITON_EXPORT const triton::arch::Register& getRegister(triton::arch::register_e id) const;
          TRITON_EXPORT const triton::arch::Register& getRegister(const std::string& name) const;
          TRITON_EXPORT const triton::arch::Register& getStackPointer(void) const;
          TRITON_EXPORT std::set<const triton::arch::Register*> getParentRegisters(void) const;
          TRITON_EXPORT std::vector<triton::uint8> getConcreteMemoryAreaValue(triton::uint64 baseAddr, triton::usize size, bool execCallbacks=true) const;
          TRITON_EXPORT triton::arch::endianness_e getEndianness(void) const;
          TRITON_EXPORT triton::uint32 numberOfRegisters(void) const;
          TRITON_EXPORT triton::uint32 gprBitSize(void) const;
          TRITON_EXPORT triton::uint32 gprSize(void) const;
          TRITON_EXPORT triton::uint512 getConcreteMemoryValue(const triton::arch::MemoryAccess& mem, bool execCallbacks=true) const;
          TRITON_EXPORT triton::uint512 getConcreteRegisterValue(const triton::arch::Register& reg, bool execCallbacks=true) const;
          TRITON_EXPORT triton::uint8 getConcreteMemoryValue(triton::uint64 addr, bool execCallbacks=true) const;
          TRITON_EXPORT void clear(void);
          TRITON_EXPORT void disassembly(triton::arch::Instruction& inst);
          TRITON_EXPORT void setConcreteMemoryAreaValue(triton::uint64 baseAddr, const std::vector<triton::uint8>& values, bool execCallbacks=true);
          TRITON_EXPORT void setConcreteMemoryAreaValue(triton::uint64 baseAddr, const void* area, triton::usize size, bool execCallbacks=true);
          TRITON_EXPORT void setConcreteMemoryValue(const triton::arch::MemoryAccess& mem, const triton::uint512& value, bool execCallbacks=true);
          TRITON_EXPORT void setConcreteMemoryValue(triton::uint64 addr, triton::uint8 value, bool execCallbacks=true);
          TRITON_EXPORT void setConcreteRegisterValue(const triton::arch::Register& reg, const triton::uint512& value, bool execCallbacks=true);
          TRITON_EXPORT void setThumb(bool state);
          TRITON_EXPORT void setMemoryExclusiveTag(const triton::arch::MemoryAccess& mem, bool tag);
          TRITON_EXPORT bool isConcreteMemoryValueDefined(const triton::arch::MemoryAccess& mem) const;
          TRITON_EXPORT bool isConcreteMemoryValueDefined(triton::uint64 baseAddr, triton::usize size=1) const;
          TRITON_EXPORT void clearConcreteMemoryValue(const triton::arch::MemoryAccess& mem);
          TRITON_EXPORT void clearConcreteMemoryValue(triton::uint64 baseAddr, triton::usize size=1);
          /* End of virtual pure inheritance ========================================== */
      };
 
    };
  };
};
 
#endif /* TRITON_X86CPU_HPP */