vmprofiler/src/calc_jmp.cpp

#include <vmprofiler.hpp>

namespace vm
{
  namespace calc_jmp
  {
    bool get(const zydis_routine_t &vm_entry, zydis_routine_t &calc_jmp)
    {
      auto result = std::find_if(
	  vm_entry.begin(), vm_entry.end(),
	  [](const zydis_instr_t &instr_data) -> bool {
	    // mov/movsx/movzx rax/eax/ax/al, [rsi]
	    if (instr_data.instr.operand_count > 1 &&
		(instr_data.instr.mnemonic == ZYDIS_MNEMONIC_MOV ||
		 instr_data.instr.mnemonic == ZYDIS_MNEMONIC_MOVSX ||
		 instr_data.instr.mnemonic == ZYDIS_MNEMONIC_MOVZX) &&
		instr_data.instr.operands[0].type ==
		    ZYDIS_OPERAND_TYPE_REGISTER &&
		util::reg::to64(instr_data.instr.operands[0].reg.value) ==
		    ZYDIS_REGISTER_RAX &&
		instr_data.instr.operands[1].type ==
		    ZYDIS_OPERAND_TYPE_MEMORY &&
		instr_data.instr.operands[1].mem.base == ZYDIS_REGISTER_RSI)
	      return true;
	    return false;
	  });

      if (result == vm_entry.end())
	return false;

      calc_jmp.insert(calc_jmp.end(), result, vm_entry.end());
      return true;
    }
  } // namespace calc_jmp
} // namespace vm