136 lines
3.3 KiB
C++
136 lines
3.3 KiB
C++
|
#include <std_include.hpp>
|
||
|
#include "loader/component_loader.hpp"
|
||
|
#include "game/game.hpp"
|
||
|
|
||
|
#include "scheduler.hpp"
|
||
|
|
||
|
#include <utils/hook.hpp>
|
||
|
#include <utils/string.hpp>
|
||
|
#include <utils/io.hpp>
|
||
|
|
||
|
namespace gamesettings
|
||
|
{
|
||
|
namespace
|
||
|
{
|
||
|
// <name, path>
|
||
|
std::unordered_map<std::string, std::string> gamesettings_files;
|
||
|
|
||
|
const std::string get_gamesettings_name(const std::vector<std::string>& sub_strings)
|
||
|
{
|
||
|
if (sub_strings.size() > 2)
|
||
|
{
|
||
|
return sub_strings[sub_strings.size() - 2] + '/' + sub_strings[sub_strings.size() - 1];
|
||
|
}
|
||
|
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
const std::string get_gamesettings_path(const std::string& name)
|
||
|
{
|
||
|
const auto itr = gamesettings_files.find(name);
|
||
|
return (itr == gamesettings_files.end()) ? "" : itr->second;
|
||
|
}
|
||
|
|
||
|
void search_gamesettings_folder(const std::string& gamesettings_dir)
|
||
|
{
|
||
|
if (!utils::io::directory_exists(gamesettings_dir))
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
const auto files = utils::io::list_files(gamesettings_dir, true);
|
||
|
|
||
|
for (const auto& path : files)
|
||
|
{
|
||
|
if (!std::filesystem::is_directory(path))
|
||
|
{
|
||
|
auto sub_strings = utils::string::split(path.generic_string(), '/');
|
||
|
gamesettings_files.insert_or_assign(get_gamesettings_name(sub_strings), path.generic_string());
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool has_gamesettings_file_on_disk(const char* path)
|
||
|
{
|
||
|
if (!path)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
auto sub_strings = utils::string::split(path, '/');
|
||
|
auto gamesettings_name = get_gamesettings_name(sub_strings);
|
||
|
|
||
|
return get_gamesettings_path(gamesettings_name) != "";
|
||
|
}
|
||
|
|
||
|
void cmd_exec_stub(utils::hook::assembler& a)
|
||
|
{
|
||
|
const auto exec_from_fastfile = a.newLabel();
|
||
|
const auto exec_from_disk = a.newLabel();
|
||
|
|
||
|
a.pushad64();
|
||
|
|
||
|
a.mov(rcx, r10);
|
||
|
a.call_aligned(has_gamesettings_file_on_disk);
|
||
|
a.cmp(rax, 1);
|
||
|
;
|
||
|
a.popad64();
|
||
|
|
||
|
a.jnz(exec_from_fastfile);
|
||
|
|
||
|
a.bind(exec_from_disk);
|
||
|
a.jmp(game::select(0x1420ED087, 0x1404F855E));
|
||
|
|
||
|
a.bind(exec_from_fastfile);
|
||
|
a.lea(rdx, ptr(rsp, (game::is_server() ? 0x30 : 0x40)));
|
||
|
a.jmp(game::select(0x1420ED007, 0x1404F853F));
|
||
|
}
|
||
|
|
||
|
int read_file_stub(const char* qpath, void** buffer)
|
||
|
{
|
||
|
auto sub_strings = utils::string::split(qpath, '/');
|
||
|
auto game_settings_name = get_gamesettings_name(sub_strings);
|
||
|
|
||
|
std::string gamesettings_data;
|
||
|
utils::io::read_file(get_gamesettings_path(game_settings_name), &gamesettings_data);
|
||
|
|
||
|
if (!gamesettings_data.empty())
|
||
|
{
|
||
|
++(*game::fs_loadStack);
|
||
|
|
||
|
auto len = static_cast<int>(gamesettings_data.length());
|
||
|
auto buf = game::FS_AllocMem(len + 1);
|
||
|
|
||
|
*buffer = buf;
|
||
|
gamesettings_data.copy(reinterpret_cast<char*>(*buffer), len);
|
||
|
buf[len] = '\0';
|
||
|
|
||
|
return len;
|
||
|
}
|
||
|
|
||
|
return utils::hook::invoke<int>(game::select(0x1422A48D0, 0x140564F70), qpath, buffer);
|
||
|
}
|
||
|
|
||
|
void search_gamesettings_files_on_disk()
|
||
|
{
|
||
|
const utils::nt::library host{};
|
||
|
|
||
|
search_gamesettings_folder((host.get_folder() / "boiii/gamesettings").string());
|
||
|
search_gamesettings_folder((game::get_appdata_path() / "data/gamesettings").string());
|
||
|
}
|
||
|
}
|
||
|
|
||
|
struct component final : generic_component
|
||
|
{
|
||
|
void post_unpack() override
|
||
|
{
|
||
|
search_gamesettings_files_on_disk();
|
||
|
|
||
|
utils::hook::call(game::select(0x1420ED0A1, 0x1404F857D), read_file_stub);
|
||
|
utils::hook::jump(game::select(0x1420ED002, 0x1404F853A), utils::hook::assemble(cmd_exec_stub));
|
||
|
}
|
||
|
};
|
||
|
};
|
||
|
|
||
|
REGISTER_COMPONENT(gamesettings::component)
|