|
|
|
#include <theo.hpp>
|
|
|
|
|
|
|
|
namespace theo {
|
|
|
|
theo_t::theo_t(std::vector<std::uint8_t>& lib,
|
|
|
|
lnk_fns_t lnkr_fns,
|
|
|
|
const std::string&& entry_sym)
|
|
|
|
: m_dcmp(lib, &m_sym_tbl), m_recmp(&m_dcmp), m_entry_sym(entry_sym) {
|
|
|
|
m_recmp.allocator(std::get<0>(lnkr_fns));
|
|
|
|
m_recmp.copier(std::get<1>(lnkr_fns));
|
|
|
|
m_recmp.resolver(std::get<2>(lnkr_fns));
|
|
|
|
}
|
|
|
|
|
|
|
|
std::optional<std::uint32_t> theo_t::decompose() {
|
|
|
|
auto res = m_dcmp.decompose(m_entry_sym);
|
|
|
|
if (!res.has_value()) {
|
|
|
|
spdlog::error("failed to decompose...\n");
|
|
|
|
return {};
|
|
|
|
}
|
|
|
|
|
|
|
|
spdlog::info("decompose successful... {} symbols", res.value()->size());
|
|
|
|
return res.value()->size();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::uintptr_t theo_t::compose() {
|
|
|
|
// run obfuscation engine on all symbols...
|
|
|
|
//
|
|
|
|
auto engine = obf::engine_t::get();
|
|
|
|
m_sym_tbl.for_each([&](decomp::symbol_t& sym) { engine->run(&sym); });
|
|
|
|
|
|
|
|
m_recmp.allocate();
|
|
|
|
m_recmp.resolve();
|
|
|
|
m_recmp.copy_syms();
|
|
|
|
return m_recmp.resolve(m_entry_sym.data());
|
|
|
|
}
|
|
|
|
|
|
|
|
std::uintptr_t theo_t::resolve(const std::string&& sym) {
|
|
|
|
auto val = m_sym_tbl.sym_from_hash(decomp::symbol_t::hash(sym));
|
|
|
|
if (!val.has_value())
|
|
|
|
return {};
|
|
|
|
|
|
|
|
return val.value()->allocated_at();
|
|
|
|
}
|
|
|
|
} // namespace theo
|