68 lines
1.7 KiB
C++
68 lines
1.7 KiB
C++
|
#include "script_patch.hpp"
|
||
|
|
#include "script_data.hpp"
|
||
|
|
|
||
|
|
namespace big
|
||
|
|
{
|
||
|
|
script_patch::script_patch(rage::joaat_t script, const memory::pattern pattern, int32_t offset, std::vector<std::uint8_t> patch, bool* enable_bool) :
|
||
|
|
m_script(script),
|
||
|
|
m_pattern(pattern),
|
||
|
|
m_offset(offset),
|
||
|
|
m_patch(std::move(patch)),
|
||
|
|
m_bool(enable_bool),
|
||
|
|
m_ip(0)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
std::uint8_t* script_patch::get_code_address(script_data* data, std::uint32_t index)
|
||
|
|
{
|
||
|
|
return &data->m_bytecode[index >> 14][index & 0x3FFF];
|
||
|
|
}
|
||
|
|
|
||
|
|
const std::optional<uint32_t> script_patch::get_code_location_by_pattern(script_data* data, const memory::pattern& pattern)
|
||
|
|
{
|
||
|
|
std::uint32_t code_size = data->m_code_size;
|
||
|
|
for (std::uint32_t i = 0; i < (code_size - pattern.m_bytes.size()); i++)
|
||
|
|
{
|
||
|
|
for (std::uint32_t j = 0; j < pattern.m_bytes.size(); j++)
|
||
|
|
if (pattern.m_bytes[j].has_value())
|
||
|
|
if (pattern.m_bytes[j].value() != *get_code_address(data, i + j))
|
||
|
|
goto incorrect;
|
||
|
|
|
||
|
|
return i;
|
||
|
|
incorrect:
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
return std::nullopt;
|
||
|
|
}
|
||
|
|
|
||
|
|
void script_patch::enable(script_data* data)
|
||
|
|
{
|
||
|
|
std::memcpy(get_code_address(data, m_ip), m_patch.data(), m_patch.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
void script_patch::disable(script_data* data)
|
||
|
|
{
|
||
|
|
std::memcpy(get_code_address(data, m_ip), m_original.data(), m_original.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
void script_patch::update(script_data* data)
|
||
|
|
{
|
||
|
|
if (m_ip == 0)
|
||
|
|
{
|
||
|
|
auto result = get_code_location_by_pattern(data, m_pattern);
|
||
|
|
if (!result.has_value())
|
||
|
|
LOG(FATAL) << "Failed to find pattern";
|
||
|
|
|
||
|
|
m_ip = result.value() + m_offset;
|
||
|
|
|
||
|
|
for (int i = 0; i < m_patch.size(); i++)
|
||
|
|
m_original.push_back(*get_code_address(data, m_ip + i));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!m_bool || *m_bool)
|
||
|
|
enable(data);
|
||
|
|
else
|
||
|
|
disable(data);
|
||
|
|
}
|
||
|
|
}
|