diff --git a/include/vminstrs.hpp b/include/vminstrs.hpp
index 38c131f..f96e0e6 100644
--- a/include/vminstrs.hpp
+++ b/include/vminstrs.hpp
@@ -24,6 +24,7 @@ enum class mnemonic_t : uint8_t {
nop,
nor,
read,
+ readbzxw, //Read byte zero extend to word
write,
shl,
shld,
@@ -292,17 +293,22 @@ extern profiler_t nand;
extern profiler_t nop;
extern profiler_t nor;
extern profiler_t read;
+extern profiler_t readbzxw;
extern profiler_t write;
extern profiler_t imul;
+extern profiler_t shl;
+extern profiler_t shld;
extern profiler_t shr;
+extern profiler_t shrd;
+extern profiler_t shrd;
extern profiler_t vmexit;
///
/// unsorted vector of profiles... they get sorted once at runtime...
///
inline std::vector profiles = {
- &vmexit, &shr, &imul, &nor, &write, &svsp, &read, &nand,
- &lvsp, &add, &jmp, &sreg, &lreg, &lconst, &nop};
+ &vmexit, &shl, &shld, &shr, &shrd, &imul, &nor, &write, &svsp, &read, &readbzxw,
+ &nand, &lvsp, &add, &jmp, &sreg, &lreg, &lconst, &nop};
///
/// no i did not make this by hand, you cannot clown upon me!
diff --git a/src/vmprofiles/readbzxw.cpp b/src/vmprofiles/readbzxw.cpp
new file mode 100644
index 0000000..609acbe
--- /dev/null
+++ b/src/vmprofiles/readbzxw.cpp
@@ -0,0 +1,40 @@
+#include
+
+namespace vm::instrs {
+profiler_t readbzxw = {
+ "READ",
+ mnemonic_t::readbzxw,
+ {{// MOV REG, [VSP]
+ LOAD_VALUE,
+ // MOVZX REG, [REG]
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_MOVZX &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ instr.operands[0].size == 16 &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_MEMORY &&
+ instr.operands[1].size == 8 &&
+ instr.operands[1].mem.base != vsp;
+ },
+ // MOV [VSP], REG
+ STR_VALUE}},
+ [](zydis_reg_t& vip, zydis_reg_t& vsp,
+ hndlr_trace_t& hndlr) -> std::optional {
+ vinstr_t res{mnemonic_t::read};
+ res.imm.has_imm = false;
+
+ // MOV REG, [REG]
+ const auto mov_reg_reg = 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_MOVZX &&
+ i.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ i.operands[1].type == ZYDIS_OPERAND_TYPE_MEMORY &&
+ i.operands[1].mem.base != vsp;
+ });
+
+ res.stack_size = mov_reg_reg->m_instr.operands[0].size;
+ return res;
+ }};
+}
\ No newline at end of file
diff --git a/src/vmprofiles/shl.cpp b/src/vmprofiles/shl.cpp
new file mode 100644
index 0000000..d5ae9b1
--- /dev/null
+++ b/src/vmprofiles/shl.cpp
@@ -0,0 +1,63 @@
+#include
+
+namespace vm::instrs {
+profiler_t shl = {
+ "SHR",
+ mnemonic_t::shl,
+ {{// MOV REG, [VSP]
+ LOAD_VALUE,
+ // MOV REG, [VSP+OFFSET]
+ [](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 &&
+ instr.operands[1].mem.disp.has_displacement;
+ },
+ // SHL REG, REG
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_SHL &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ },
+ // MOV [VSP+OFFSET], 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 == vsp &&
+ instr.operands[0].mem.disp.has_displacement &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ },
+ // PUSHFQ
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_PUSHFQ;
+ },
+ // POP [VSP]
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_POP &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_MEMORY &&
+ instr.operands[0].mem.base == vsp;
+ }}},
+ [](zydis_reg_t& vip, zydis_reg_t& vsp,
+ hndlr_trace_t& hndlr) -> std::optional {
+ vinstr_t res{mnemonic_t::shr};
+ res.imm.has_imm = false;
+
+ const auto shr_reg = 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_SHL &&
+ i.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ i.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ });
+
+ res.stack_size = shr_reg->m_instr.operands[0].size;
+ return res;
+ }};
+}
\ No newline at end of file
diff --git a/src/vmprofiles/shld.cpp b/src/vmprofiles/shld.cpp
new file mode 100644
index 0000000..14a01e9
--- /dev/null
+++ b/src/vmprofiles/shld.cpp
@@ -0,0 +1,73 @@
+#include
+
+namespace vm::instrs {
+profiler_t shld = {
+ "SHR",
+ mnemonic_t::shld,
+ {{// MOV REG, [VSP]
+ LOAD_VALUE,
+ // MOV REG, [VSP+OFFSET]
+ [](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 &&
+ instr.operands[1].mem.disp.has_displacement;
+ },
+ // MOV REG, [VSP+OFFSET]
+ [](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 &&
+ instr.operands[1].mem.disp.has_displacement;
+ },
+ // SHL REG, REG
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_SHLD &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ },
+ // MOV [VSP+OFFSET], 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 == vsp &&
+ instr.operands[0].mem.disp.has_displacement &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ },
+ // PUSHFQ
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_PUSHFQ;
+ },
+ // POP [VSP]
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_POP &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_MEMORY &&
+ instr.operands[0].mem.base == vsp;
+ }}},
+ [](zydis_reg_t& vip, zydis_reg_t& vsp,
+ hndlr_trace_t& hndlr) -> std::optional {
+ vinstr_t res{mnemonic_t::shr};
+ res.imm.has_imm = false;
+
+ const auto shr_reg = 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_SHLD &&
+ i.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ i.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ });
+
+ res.stack_size = shr_reg->m_instr.operands[0].size;
+ return res;
+ }};
+
+}
\ No newline at end of file
diff --git a/src/vmprofiles/shrd.cpp b/src/vmprofiles/shrd.cpp
new file mode 100644
index 0000000..73d4aad
--- /dev/null
+++ b/src/vmprofiles/shrd.cpp
@@ -0,0 +1,73 @@
+#include
+
+namespace vm::instrs {
+profiler_t shrd = {
+ "SHR",
+ mnemonic_t::shrd,
+ {{// MOV REG, [VSP]
+ LOAD_VALUE,
+ // MOV REG, [VSP+OFFSET]
+ [](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 &&
+ instr.operands[1].mem.disp.has_displacement;
+ },
+ // MOV REG, [VSP+OFFSET]
+ [](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 &&
+ instr.operands[1].mem.disp.has_displacement;
+ },
+ // SHR REG, REG
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_SHRD &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ },
+ // MOV [VSP+OFFSET], 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 == vsp &&
+ instr.operands[0].mem.disp.has_displacement &&
+ instr.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ },
+ // PUSHFQ
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_PUSHFQ;
+ },
+ // POP [VSP]
+ [](const zydis_reg_t vip, const zydis_reg_t vsp,
+ const zydis_decoded_instr_t& instr) -> bool {
+ return instr.mnemonic == ZYDIS_MNEMONIC_POP &&
+ instr.operands[0].type == ZYDIS_OPERAND_TYPE_MEMORY &&
+ instr.operands[0].mem.base == vsp;
+ }}},
+ [](zydis_reg_t& vip, zydis_reg_t& vsp,
+ hndlr_trace_t& hndlr) -> std::optional {
+ vinstr_t res{mnemonic_t::shr};
+ res.imm.has_imm = false;
+
+ const auto shr_reg = 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_SHRD &&
+ i.operands[0].type == ZYDIS_OPERAND_TYPE_REGISTER &&
+ i.operands[1].type == ZYDIS_OPERAND_TYPE_REGISTER;
+ });
+
+ res.stack_size = shr_reg->m_instr.operands[0].size;
+ return res;
+ }};
+
+}
\ No newline at end of file