forked from vmp3/vmprofiler
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
3.2 KiB
79 lines
3.2 KiB
#include <vminstrs.hpp>
|
|
|
|
namespace vm::instrs {
|
|
profiler_t sreg = {
|
|
"SREG",
|
|
mnemonic_t::sreg,
|
|
{{// MOV REG, [VSP]
|
|
[&](const zydis_reg_t vip,
|
|
const zydis_reg_t vsp,
|
|
const zydis_decoded_instr_t& instr) -> bool {
|
|
return instr.mnemonic == ZYDIS_MNEMONIC_MOV &&
|
|
instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
|
|
instr.operands[1].type == ZYDIS_OPERAND_TYPE_MEMORY &&
|
|
instr.operands[1].mem.base == vsp;
|
|
},
|
|
// ADD VSP, OFFSET
|
|
[&](const zydis_reg_t vip,
|
|
const zydis_reg_t vsp,
|
|
const zydis_decoded_instr_t& instr) -> bool {
|
|
return instr.mnemonic == ZYDIS_MNEMONIC_ADD &&
|
|
instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
|
|
instr.operands[0].reg.value == vsp &&
|
|
instr.operands[1].type == ZYDIS_OPERAND_TYPE_IMMEDIATE;
|
|
},
|
|
// MOV REG, [VIP]
|
|
[&](const zydis_reg_t vip,
|
|
const zydis_reg_t vsp,
|
|
const zydis_decoded_instr_t& instr) -> bool {
|
|
return (instr.mnemonic == ZYDIS_MNEMONIC_MOV ||
|
|
instr.mnemonic == ZYDIS_MNEMONIC_MOVSX ||
|
|
instr.mnemonic == ZYDIS_MNEMONIC_MOVZX) &&
|
|
instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
|
|
instr.operands[1].type == ZYDIS_OPERAND_TYPE_MEMORY &&
|
|
instr.operands[1].mem.base == vip;
|
|
},
|
|
// MOV [RSP+REG], REG
|
|
[&](const zydis_reg_t vip,
|
|
const zydis_reg_t vsp,
|
|
const zydis_decoded_instr_t& instr) -> bool {
|
|
return instr.mnemonic == ZYDIS_MNEMONIC_MOV &&
|
|
instr.operands[0].type == ZYDIS_OPERAND_TYPE_MEMORY &&
|
|
instr.operands[0].mem.base == ZYDIS_REGISTER_RSP &&
|
|
instr.operands[0].mem.index != ZYDIS_REGISTER_NONE &&
|
|
instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
|
|
}}},
|
|
[&](zydis_reg_t& vip,
|
|
zydis_reg_t& vsp,
|
|
hndlr_trace_t& hndlr) -> std::optional<vinstr_t> {
|
|
vinstr_t res;
|
|
res.mnemonic = mnemonic_t::sreg;
|
|
|
|
// locates ADD VSP, VALUE...
|
|
const auto add_vsp = std::find_if(
|
|
hndlr.m_instrs.begin(), hndlr.m_instrs.end(),
|
|
[&](emu_instr_t& instr) -> bool {
|
|
const auto& i = instr.m_instr;
|
|
return i.mnemonic == ZYDIS_MNEMONIC_ADD &&
|
|
i.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
|
|
i.operands[0].reg.value == vsp &&
|
|
i.operands[1].type == ZYDIS_OPERAND_TYPE_IMMEDIATE;
|
|
});
|
|
|
|
res.size = add_vsp->m_instr.operands[1].imm.value.u * 8;
|
|
|
|
// MOV [RSP+REG], REG...
|
|
const auto mov_vreg_value = std::find_if(
|
|
hndlr.m_instrs.begin(), hndlr.m_instrs.end(),
|
|
[&](emu_instr_t& instr) -> bool {
|
|
const auto& i = instr.m_instr;
|
|
return i.mnemonic == ZYDIS_MNEMONIC_MOV &&
|
|
i.operands[0].type == ZYDIS_OPERAND_TYPE_MEMORY &&
|
|
i.operands[0].mem.base == ZYDIS_REGISTER_RSP &&
|
|
i.operands[0].mem.index != ZYDIS_REGISTER_NONE &&
|
|
i.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
|
|
});
|
|
|
|
return res;
|
|
}};
|
|
} |