d2b3370b7c
* Fixed a race condition caused by disabling and enabling a Lua script with scr_patch more than once. * Fixed patch not being enabled if data is already created --------- Co-authored-by: Arthur <121949966+ShinyWasabi@users.noreply.github.com>
58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
#include "scr_patch.hpp"
|
|
|
|
#include "gta_util.hpp"
|
|
#include "lua/lua_module.hpp"
|
|
#include "lua/bindings/network.hpp" // For convert_sequence
|
|
#include "services/script_patcher/script_patcher_service.hpp"
|
|
|
|
namespace lua::scr_patch
|
|
{
|
|
scr_patch::scr_patch(const std::string& script_name, const std::string& patch_name, const std::string& pattern, const int offset, sol::table patch_, sol::this_state state) :
|
|
m_script(rage::joaat(script_name)),
|
|
m_patch_name(patch_name),
|
|
m_pattern(pattern),
|
|
m_offset(offset),
|
|
m_patch(patch_),
|
|
m_enable(true)
|
|
{
|
|
auto patch = convert_sequence<uint8_t>(m_patch);
|
|
|
|
big::g_script_patcher_service->add_patch({m_script, m_patch_name, ::memory::pattern(m_pattern), m_offset, patch, &m_enable});
|
|
if (auto program = big::gta_util::find_script_program(m_script))
|
|
big::g_script_patcher_service->on_script_load(program);
|
|
|
|
big::lua_module* module = sol::state_view(state)["!this"];
|
|
module->m_registered_script_patches.push_back(std::make_unique<scr_patch>(*this));
|
|
}
|
|
|
|
scr_patch::~scr_patch()
|
|
{
|
|
big::g_script_patcher_service->remove_patch(m_patch_name);
|
|
}
|
|
|
|
void scr_patch::enable()
|
|
{
|
|
if (!m_enable)
|
|
{
|
|
m_enable = true;
|
|
big::g_script_patcher_service->update();
|
|
}
|
|
}
|
|
|
|
void scr_patch::disable()
|
|
{
|
|
if (m_enable)
|
|
{
|
|
m_enable = false;
|
|
big::g_script_patcher_service->update();
|
|
}
|
|
}
|
|
|
|
void bind(sol::state& state)
|
|
{
|
|
auto scr_patch_ut = state.new_usertype<scr_patch>("scr_patch", sol::constructors<scr_patch(const std::string&, const std::string&, const std::string&, const int, sol::table, sol::this_state)>());
|
|
|
|
scr_patch_ut["enable_patch"] = &scr_patch::enable;
|
|
scr_patch_ut["disable_patch"] = &scr_patch::disable;
|
|
}
|
|
} |