forked from IDontCode/Theodosius
reloc_transform_pass are all done....3.0
_xeroxz
3 years ago
parent
df0ba19093
commit
c7eb3c83bf
@ -1,17 +0,0 @@
|
||||
#pragma once
|
||||
#include <obf/transform/operation.hpp>
|
||||
|
||||
namespace theo::obf::transform {
|
||||
class and_op_t : public operation_t {
|
||||
explicit and_op_t()
|
||||
: operation_t([&](std::size_t val,
|
||||
std::uint32_t imm) -> std::size_t { return val & imm; },
|
||||
XED_ICLASS_AND) {}
|
||||
|
||||
public:
|
||||
static and_op_t* get() {
|
||||
static and_op_t obj;
|
||||
return &obj;
|
||||
}
|
||||
};
|
||||
} // namespace theo::obf::transform
|
||||
@ -1,17 +0,0 @@
|
||||
#pragma once
|
||||
#include <obf/transform/operation.hpp>
|
||||
|
||||
namespace theo::obf::transform {
|
||||
class or_op_t : public operation_t {
|
||||
explicit or_op_t()
|
||||
: operation_t([&](std::size_t val,
|
||||
std::uint32_t imm) -> std::size_t { return val | imm; },
|
||||
XED_ICLASS_OR) {}
|
||||
|
||||
public:
|
||||
static or_op_t* get() {
|
||||
static or_op_t obj;
|
||||
return &obj;
|
||||
}
|
||||
};
|
||||
} // namespace theo::obf::transform
|
||||
@ -0,0 +1,53 @@
|
||||
#include <obf/passes/jcc_rewrite_pass.hpp>
|
||||
#include <obf/passes/next_inst_pass.hpp>
|
||||
|
||||
namespace theo::obf {
|
||||
jcc_rewrite_pass_t* jcc_rewrite_pass_t::get() {
|
||||
static jcc_rewrite_pass_t obj;
|
||||
return &obj;
|
||||
}
|
||||
|
||||
void jcc_rewrite_pass_t::run(decomp::symbol_t* sym) {
|
||||
std::int32_t disp = {};
|
||||
xed_decoded_inst_t inst;
|
||||
xed_state_t istate{XED_MACHINE_MODE_LONG_64, XED_ADDRESS_WIDTH_64b};
|
||||
xed_decoded_inst_zero_set_mode(&inst, &istate);
|
||||
xed_decode(&inst, sym->data().data(), XED_MAX_INSTRUCTION_BYTES);
|
||||
|
||||
// if the instruction is branching...
|
||||
if ((disp = xed_decoded_inst_get_branch_displacement(&inst))) {
|
||||
disp += xed_decoded_inst_get_length(&inst);
|
||||
|
||||
// update displacement...
|
||||
xed_decoded_inst_set_branch_displacement(
|
||||
&inst, sym->data().size() - xed_decoded_inst_get_length(&inst),
|
||||
xed_decoded_inst_get_branch_displacement_width(&inst));
|
||||
|
||||
xed_encoder_request_init_from_decode(&inst);
|
||||
xed_encoder_request_t* req = &inst;
|
||||
|
||||
// update jcc in the buffer...
|
||||
std::uint32_t len = {};
|
||||
xed_encode(req, sym->data().data(), xed_decoded_inst_get_length(&inst),
|
||||
&len);
|
||||
|
||||
// create a relocation to the instruction the branch would normally go
|
||||
// too...
|
||||
auto offset = disp < 0 ? sym->offset() - std::abs(disp)
|
||||
: sym->offset() + std::abs(disp);
|
||||
|
||||
auto sym_name =
|
||||
std::string(
|
||||
sym->sym()->name.to_string(sym->img()->get_strings()).data())
|
||||
.append("@")
|
||||
.append(std::to_string(offset));
|
||||
|
||||
sym->relocs().push_back(
|
||||
recomp::reloc_t(0, decomp::symbol_t::hash(sym_name), sym_name.data()));
|
||||
|
||||
// run next_inst_pass on this symbol to generate the transformations for the
|
||||
// relocation to the jcc branch dest instruction...
|
||||
next_inst_pass_t::get()->run(sym);
|
||||
}
|
||||
};
|
||||
} // namespace theo::obf
|
||||
@ -0,0 +1,73 @@
|
||||
#include <obf/passes/next_inst_pass.hpp>
|
||||
|
||||
namespace theo::obf {
|
||||
next_inst_pass_t* next_inst_pass_t::get() {
|
||||
static next_inst_pass_t obj;
|
||||
return &obj;
|
||||
}
|
||||
void next_inst_pass_t::run(decomp::symbol_t* sym) {
|
||||
std::optional<recomp::reloc_t*> reloc;
|
||||
if (!(reloc = has_next_inst_reloc(sym)).has_value())
|
||||
return;
|
||||
|
||||
xed_decoded_inst_t inst = m_tmp_inst;
|
||||
std::vector<std::uint8_t> new_inst_bytes =
|
||||
generate_transforms(&inst, reloc.value(), 3, 6);
|
||||
|
||||
// add a push [rip+offset] and update reloc->offset()...
|
||||
//
|
||||
std::uint32_t inst_len = {};
|
||||
std::uint8_t inst_buff[XED_MAX_INSTRUCTION_BYTES];
|
||||
|
||||
xed_error_enum_t err;
|
||||
xed_encoder_request_t req;
|
||||
xed_state_t istate{XED_MACHINE_MODE_LONG_64, XED_ADDRESS_WIDTH_64b};
|
||||
|
||||
xed_encoder_request_zero_set_mode(&req, &istate);
|
||||
xed_encoder_request_set_effective_operand_width(&req, 64);
|
||||
xed_encoder_request_set_iclass(&req, XED_ICLASS_PUSH);
|
||||
|
||||
xed_encoder_request_set_mem0(&req);
|
||||
xed_encoder_request_set_operand_order(&req, 0, XED_OPERAND_MEM0);
|
||||
|
||||
xed_encoder_request_set_base0(&req, XED_REG_RIP);
|
||||
xed_encoder_request_set_seg0(&req, XED_REG_INVALID);
|
||||
xed_encoder_request_set_index(&req, XED_REG_INVALID);
|
||||
xed_encoder_request_set_scale(&req, 0);
|
||||
|
||||
xed_encoder_request_set_memory_operand_length(&req, 8);
|
||||
xed_encoder_request_set_memory_displacement(&req, new_inst_bytes.size() + 1,
|
||||
1);
|
||||
|
||||
if ((err = xed_encode(&req, inst_buff, sizeof(inst_buff), &inst_len)) !=
|
||||
XED_ERROR_NONE) {
|
||||
spdlog::info("failed to encode instruction... reason: {}",
|
||||
xed_error_enum_t2str(err));
|
||||
|
||||
assert(err == XED_ERROR_NONE);
|
||||
}
|
||||
|
||||
new_inst_bytes.insert(new_inst_bytes.begin(), inst_buff,
|
||||
inst_buff + inst_len);
|
||||
|
||||
// put a return instruction at the end of the decrypt instructions...
|
||||
//
|
||||
new_inst_bytes.push_back(0xC3);
|
||||
|
||||
sym->data().insert(sym->data().end(), new_inst_bytes.begin(),
|
||||
new_inst_bytes.end());
|
||||
|
||||
reloc.value()->offset(sym->data().size());
|
||||
sym->data().resize(sym->data().size() + 8);
|
||||
}
|
||||
|
||||
std::optional<recomp::reloc_t*> next_inst_pass_t::has_next_inst_reloc(
|
||||
decomp::symbol_t* sym) {
|
||||
auto res = std::find_if(
|
||||
sym->relocs().begin(), sym->relocs().end(),
|
||||
[&](recomp::reloc_t& reloc) -> bool { return !reloc.offset(); });
|
||||
|
||||
return res != sym->relocs().end() ? res._Ptr
|
||||
: std::optional<recomp::reloc_t*>();
|
||||
}
|
||||
} // namespace theo::obf
|
||||
Loading…
Reference in new issue