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.
123 lines
4.9 KiB
123 lines
4.9 KiB
// Copyright (c) 2022, _xeroxz
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are met:
|
|
//
|
|
// 1. Redistributions of source code must retain the above copyright notice,
|
|
// this list of conditions and the following disclaimer.
|
|
//
|
|
// 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
// this list of conditions and the following disclaimer in the documentation
|
|
// and/or other materials provided with the distribution.
|
|
//
|
|
// 3. Neither the name of the copyright holder nor the names of its
|
|
// contributors may be used to endorse or promote products derived from
|
|
// this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
// POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
|
|
#pragma once
|
|
#include <spdlog/spdlog.h>
|
|
#include <decomp/symbol.hpp>
|
|
#include <obf/transform/gen.hpp>
|
|
|
|
#define XED_ENCODER
|
|
extern "C" {
|
|
#include <xed-decode.h>
|
|
#include <xed-interface.h>
|
|
}
|
|
|
|
/// <summary>
|
|
/// this is the main namespace for obfuscation related things.
|
|
/// </summary>
|
|
namespace theo::obf {
|
|
|
|
/// <summary>
|
|
/// the pass_t class is a base clase for all passes made. you must override the
|
|
/// pass_t::run virtual function and declare the logic of your pass there.
|
|
///
|
|
/// in the constructor of your pass you must call the super constructor (the
|
|
/// pass_t constructor) and pass it the type of symbol which you are interesting
|
|
/// in receiving.
|
|
/// </summary>
|
|
class pass_t {
|
|
public:
|
|
/// <summary>
|
|
/// the explicit constructor of the pass_t base class.
|
|
/// </summary>
|
|
/// <param name="sym_type">the type of symbol in which the pass will run on.
|
|
/// every symbol passed to the virtual "run" instruction will be of this
|
|
/// type.</param>
|
|
explicit pass_t(decomp::sym_type_t sym_type) : m_sym_type(sym_type){};
|
|
|
|
/// <summary>
|
|
/// This virtual method is invoked before the recomp stage of Theodosius. This
|
|
/// allows you to manipulate symbols in a generic manner.
|
|
/// </summary>
|
|
/// <param name="sym">a symbol of the same type of m_sym_type.</param>
|
|
virtual void pre_recomp_pass(decomp::symbol_t* sym) = 0;
|
|
|
|
/// <summary>
|
|
/// This virtual method is invoked prior to calling the "allocator". This
|
|
/// allows you to manipulate the symbol prior to memory being allocated for
|
|
/// it.
|
|
/// </summary>
|
|
/// <param name="sym">The symbol that memory is being allocated for.</param>
|
|
/// <param name="size">The size of the symbol that memory is being allocated
|
|
/// for.</param>
|
|
/// <returns>returns the size of the symbol.</returns>
|
|
virtual std::uint32_t pre_allocation_pass(decomp::symbol_t* sym,
|
|
std::uint32_t size) = 0;
|
|
|
|
/// <summary>
|
|
/// This virtual method is invoked after the call to the "allocator". This
|
|
/// allows you to manipulate the symbol after memory has been allocated for
|
|
/// it.
|
|
/// </summary>
|
|
/// <param name="dest">Allocation address (can be linear virtual address or
|
|
/// file offset address)</param>
|
|
/// <param name="sym">The symbol for which allocation was made for.</param>
|
|
/// <param name="size">Size of the allocation.</param>
|
|
virtual void post_allocation_pass(decomp::symbol_t* sym,
|
|
std::uintptr_t dest,
|
|
std::uint32_t size) = 0;
|
|
|
|
/// <summary>
|
|
/// This virtual method is invoked prior to calling the "copier". This allows
|
|
/// you to manipulate the symbol prior to it being copied into memory.
|
|
/// </summary>
|
|
/// <param name="sym">Symbol being copied into memory...</param>
|
|
/// <param name="dest">Memory destination of the copy...</param>
|
|
/// <param name="size"></param>
|
|
/// <returns></returns>
|
|
virtual std::uintptr_t pre_copier_pass(decomp::symbol_t* sym,
|
|
std::uintptr_t dest,
|
|
std::uint32_t size) = 0;
|
|
|
|
virtual void post_copier_pass() = 0;
|
|
|
|
virtual void pre_resolver_pass() = 0;
|
|
virtual void post_resolver_pass() = 0;
|
|
|
|
/// <summary>
|
|
/// gets the passes symbol type.
|
|
/// </summary>
|
|
/// <returns>the passes symbol type.</returns>
|
|
decomp::sym_type_t sym_type() { return m_sym_type; }
|
|
|
|
private:
|
|
decomp::sym_type_t m_sym_type;
|
|
};
|
|
} // namespace theo::obf
|