mirror of
https://github.com/diamante0018/MW3ServerFreezer.git
synced 2025-04-18 19:22:53 +00:00
build: add back clang
This commit is contained in:
parent
0541f9d095
commit
572b71da00
1
.github/workflows/build.yml
vendored
1
.github/workflows/build.yml
vendored
@ -27,6 +27,7 @@ jobs:
|
|||||||
- release
|
- release
|
||||||
compiler:
|
compiler:
|
||||||
- msvc
|
- msvc
|
||||||
|
- clang
|
||||||
steps:
|
steps:
|
||||||
- name: Check out files
|
- name: Check out files
|
||||||
uses: actions/checkout@main
|
uses: actions/checkout@main
|
||||||
|
@ -2,7 +2,6 @@
|
|||||||
#include "loader/component_loader.hpp"
|
#include "loader/component_loader.hpp"
|
||||||
|
|
||||||
#include <utils/hook.hpp>
|
#include <utils/hook.hpp>
|
||||||
#include <utils/thread.hpp>
|
|
||||||
#include <utils/flags.hpp>
|
#include <utils/flags.hpp>
|
||||||
#include <utils/concurrency.hpp>
|
#include <utils/concurrency.hpp>
|
||||||
|
|
||||||
@ -37,8 +36,7 @@ public:
|
|||||||
void post_start() override {
|
void post_start() override {
|
||||||
this->terminate_runner_ = false;
|
this->terminate_runner_ = false;
|
||||||
|
|
||||||
this->console_runner_ = utils::thread::create_named_thread(
|
this->console_runner_ = std::thread([this] { this->runner(); });
|
||||||
"Console IO", [this] { this->runner(); });
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void post_unpack() override {
|
void post_unpack() override {
|
||||||
@ -77,31 +75,30 @@ private:
|
|||||||
int handles_[2]{};
|
int handles_[2]{};
|
||||||
|
|
||||||
void initialize() {
|
void initialize() {
|
||||||
this->console_thread_ =
|
this->console_thread_ = std::thread([this] {
|
||||||
utils::thread::create_named_thread("Console", [this] {
|
if (!utils::flags::has_flag("noconsole")) {
|
||||||
if (!utils::flags::has_flag("noconsole")) {
|
game::Sys_ShowConsole();
|
||||||
game::Sys_ShowConsole();
|
}
|
||||||
|
|
||||||
|
messages.access(
|
||||||
|
[&](message_queue&) { this->console_initialized_ = true; });
|
||||||
|
|
||||||
|
MSG msg;
|
||||||
|
while (!this->terminate_runner_) {
|
||||||
|
if (PeekMessageA(&msg, nullptr, NULL, NULL, PM_REMOVE)) {
|
||||||
|
if (msg.message == WM_QUIT) {
|
||||||
|
command::execute("quit", false);
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
messages.access(
|
TranslateMessage(&msg);
|
||||||
[&](message_queue&) { this->console_initialized_ = true; });
|
DispatchMessageA(&msg);
|
||||||
|
} else {
|
||||||
MSG msg;
|
this->log_messages();
|
||||||
while (!this->terminate_runner_) {
|
std::this_thread::sleep_for(1ms);
|
||||||
if (PeekMessageA(&msg, nullptr, NULL, NULL, PM_REMOVE)) {
|
}
|
||||||
if (msg.message == WM_QUIT) {
|
}
|
||||||
command::execute("quit", false);
|
});
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
TranslateMessage(&msg);
|
|
||||||
DispatchMessageA(&msg);
|
|
||||||
} else {
|
|
||||||
this->log_messages();
|
|
||||||
std::this_thread::sleep_for(1ms);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void log_messages() const {
|
void log_messages() const {
|
||||||
|
@ -3,7 +3,6 @@
|
|||||||
|
|
||||||
#include <utils/concurrency.hpp>
|
#include <utils/concurrency.hpp>
|
||||||
#include <utils/hook.hpp>
|
#include <utils/hook.hpp>
|
||||||
#include <utils/thread.hpp>
|
|
||||||
|
|
||||||
#include "scheduler.hpp"
|
#include "scheduler.hpp"
|
||||||
|
|
||||||
@ -135,7 +134,7 @@ unsigned int thread_id;
|
|||||||
class component final : public component_interface {
|
class component final : public component_interface {
|
||||||
public:
|
public:
|
||||||
void post_unpack() override {
|
void post_unpack() override {
|
||||||
thread = utils::thread::create_named_thread("Async Scheduler", [] {
|
thread = std::thread([] {
|
||||||
while (!kill) {
|
while (!kill) {
|
||||||
execute(pipeline::async);
|
execute(pipeline::async);
|
||||||
std::this_thread::sleep_for(10ms);
|
std::this_thread::sleep_for(10ms);
|
||||||
|
@ -2,6 +2,19 @@
|
|||||||
#include "loader/component_loader.hpp"
|
#include "loader/component_loader.hpp"
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
std::string get_current_date() {
|
||||||
|
auto now = std::chrono::system_clock::now();
|
||||||
|
auto current_time = std::chrono::system_clock::to_time_t(now);
|
||||||
|
std::tm local_time{};
|
||||||
|
|
||||||
|
(void)localtime_s(&local_time, ¤t_time);
|
||||||
|
|
||||||
|
std::stringstream ss;
|
||||||
|
ss << std::put_time(&local_time, "%Y%m%d_%H%M%S");
|
||||||
|
|
||||||
|
return ss.str();
|
||||||
|
}
|
||||||
|
|
||||||
LONG WINAPI exception_handler(PEXCEPTION_POINTERS exception_info) {
|
LONG WINAPI exception_handler(PEXCEPTION_POINTERS exception_info) {
|
||||||
if (exception_info->ExceptionRecord->ExceptionCode == 0x406D1388) {
|
if (exception_info->ExceptionRecord->ExceptionCode == 0x406D1388) {
|
||||||
return EXCEPTION_CONTINUE_EXECUTION;
|
return EXCEPTION_CONTINUE_EXECUTION;
|
||||||
@ -21,8 +34,8 @@ LONG WINAPI exception_handler(PEXCEPTION_POINTERS exception_info) {
|
|||||||
| MiniDumpWithFullMemoryInfo //
|
| MiniDumpWithFullMemoryInfo //
|
||||||
| MiniDumpWithThreadInfo;
|
| MiniDumpWithThreadInfo;
|
||||||
|
|
||||||
const auto file_name = std::format("minidumps\\mw3-server-freezer_{}.dmp",
|
const auto file_name =
|
||||||
game::Sys_Milliseconds());
|
std::format("minidumps\\mw3-server-freezer_{}.dmp", get_current_date());
|
||||||
const auto file_handle = CreateFileA(
|
const auto file_handle = CreateFileA(
|
||||||
file_name.data(), GENERIC_WRITE | GENERIC_READ,
|
file_name.data(), GENERIC_WRITE | GENERIC_READ,
|
||||||
FILE_SHARE_READ | FILE_SHARE_WRITE, nullptr, CREATE_ALWAYS,
|
FILE_SHARE_READ | FILE_SHARE_WRITE, nullptr, CREATE_ALWAYS,
|
||||||
|
@ -10,13 +10,16 @@
|
|||||||
|
|
||||||
#include <DbgHelp.h>
|
#include <DbgHelp.h>
|
||||||
|
|
||||||
#include <algorithm>
|
|
||||||
#include <cassert>
|
#include <cassert>
|
||||||
#include <cctype>
|
#include <cctype>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <chrono>
|
||||||
|
#include <iomanip>
|
||||||
|
#include <iostream>
|
||||||
#include <format>
|
#include <format>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <iostream>
|
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
#include <queue>
|
#include <queue>
|
||||||
#include <source_location>
|
#include <source_location>
|
||||||
|
@ -103,24 +103,6 @@ void hook::quick() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool hook::iat(const nt::library module, const std::string& target_module,
|
|
||||||
const std::string& process, void* stub) {
|
|
||||||
if (!module.is_valid())
|
|
||||||
return false;
|
|
||||||
|
|
||||||
auto ptr = module.get_iat_entry(target_module, process);
|
|
||||||
if (!ptr)
|
|
||||||
return false;
|
|
||||||
|
|
||||||
DWORD protect;
|
|
||||||
VirtualProtect(ptr, sizeof(*ptr), PAGE_EXECUTE_READWRITE, &protect);
|
|
||||||
|
|
||||||
*ptr = stub;
|
|
||||||
|
|
||||||
VirtualProtect(ptr, sizeof(*ptr), protect, &protect);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
hook* hook::uninstall(const bool unprotect) {
|
hook* hook::uninstall(const bool unprotect) {
|
||||||
std::lock_guard _(this->state_mutex_);
|
std::lock_guard _(this->state_mutex_);
|
||||||
|
|
||||||
|
@ -75,9 +75,6 @@ public:
|
|||||||
void* get_original() const;
|
void* get_original() const;
|
||||||
void quick();
|
void quick();
|
||||||
|
|
||||||
static bool iat(nt::library module, const std::string& target_module,
|
|
||||||
const std::string& process, void* stub);
|
|
||||||
|
|
||||||
static void nop(void* place, size_t length);
|
static void nop(void* place, size_t length);
|
||||||
static void nop(DWORD place, size_t length);
|
static void nop(DWORD place, size_t length);
|
||||||
|
|
||||||
|
@ -135,62 +135,6 @@ void library::free() {
|
|||||||
|
|
||||||
HMODULE library::get_handle() const { return this->module_; }
|
HMODULE library::get_handle() const { return this->module_; }
|
||||||
|
|
||||||
void** library::get_iat_entry(const std::string& module_name,
|
|
||||||
const std::string& proc_name) const {
|
|
||||||
if (!this->is_valid())
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
const library other_module(module_name);
|
|
||||||
if (!other_module.is_valid())
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
auto* const target_function = other_module.get_proc<void*>(proc_name);
|
|
||||||
if (!target_function)
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
auto* header = this->get_optional_header();
|
|
||||||
if (!header)
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
auto* import_descriptor = reinterpret_cast<PIMAGE_IMPORT_DESCRIPTOR>(
|
|
||||||
this->get_ptr() +
|
|
||||||
header->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);
|
|
||||||
|
|
||||||
while (import_descriptor->Name) {
|
|
||||||
if (!_stricmp(
|
|
||||||
reinterpret_cast<char*>(this->get_ptr() + import_descriptor->Name),
|
|
||||||
module_name.data())) {
|
|
||||||
auto* original_thunk_data = reinterpret_cast<PIMAGE_THUNK_DATA>(
|
|
||||||
import_descriptor->OriginalFirstThunk + this->get_ptr());
|
|
||||||
auto* thunk_data = reinterpret_cast<PIMAGE_THUNK_DATA>(
|
|
||||||
import_descriptor->FirstThunk + this->get_ptr());
|
|
||||||
|
|
||||||
while (original_thunk_data->u1.AddressOfData) {
|
|
||||||
const size_t ordinal_number =
|
|
||||||
original_thunk_data->u1.AddressOfData & 0xFFFFFFF;
|
|
||||||
|
|
||||||
if (ordinal_number > 0xFFFF)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
if (GetProcAddress(other_module.module_,
|
|
||||||
reinterpret_cast<char*>(ordinal_number)) ==
|
|
||||||
target_function) {
|
|
||||||
return reinterpret_cast<void**>(&thunk_data->u1.Function);
|
|
||||||
}
|
|
||||||
|
|
||||||
++original_thunk_data;
|
|
||||||
++thunk_data;
|
|
||||||
}
|
|
||||||
|
|
||||||
// break;
|
|
||||||
}
|
|
||||||
|
|
||||||
++import_descriptor;
|
|
||||||
}
|
|
||||||
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string load_resource(const int id) {
|
std::string load_resource(const int id) {
|
||||||
auto* const res = FindResource(library(), MAKEINTRESOURCE(id), RT_RCDATA);
|
auto* const res = FindResource(library(), MAKEINTRESOURCE(id), RT_RCDATA);
|
||||||
if (!res)
|
if (!res)
|
||||||
|
@ -48,27 +48,12 @@ public:
|
|||||||
|
|
||||||
HMODULE get_handle() const;
|
HMODULE get_handle() const;
|
||||||
|
|
||||||
template <typename T> T get_proc(const std::string& process) const {
|
|
||||||
if (!this->is_valid())
|
|
||||||
return T{};
|
|
||||||
return reinterpret_cast<T>(GetProcAddress(this->module_, process.data()));
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T> std::function<T> get(const std::string& process) const {
|
|
||||||
if (!this->is_valid())
|
|
||||||
return std::function<T>();
|
|
||||||
return reinterpret_cast<T*>(this->get_proc<void*>(process));
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<PIMAGE_SECTION_HEADER> get_section_headers() const;
|
std::vector<PIMAGE_SECTION_HEADER> get_section_headers() const;
|
||||||
|
|
||||||
PIMAGE_NT_HEADERS get_nt_headers() const;
|
PIMAGE_NT_HEADERS get_nt_headers() const;
|
||||||
PIMAGE_DOS_HEADER get_dos_header() const;
|
PIMAGE_DOS_HEADER get_dos_header() const;
|
||||||
PIMAGE_OPTIONAL_HEADER get_optional_header() const;
|
PIMAGE_OPTIONAL_HEADER get_optional_header() const;
|
||||||
|
|
||||||
void** get_iat_entry(const std::string& module_name,
|
|
||||||
const std::string& proc_name) const;
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
HMODULE module_;
|
HMODULE module_;
|
||||||
};
|
};
|
||||||
|
@ -1,105 +0,0 @@
|
|||||||
#include <thread>
|
|
||||||
#include "nt.hpp"
|
|
||||||
#include "string.hpp"
|
|
||||||
|
|
||||||
#include "thread.hpp"
|
|
||||||
|
|
||||||
#include <TlHelp32.h>
|
|
||||||
|
|
||||||
#include <gsl/gsl>
|
|
||||||
|
|
||||||
namespace utils::thread {
|
|
||||||
bool set_name(const HANDLE t, const std::string& name) {
|
|
||||||
const nt::library kernel32("kernel32.dll");
|
|
||||||
if (!kernel32) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto set_description =
|
|
||||||
kernel32.get_proc<HRESULT(WINAPI*)(HANDLE, PCWSTR)>(
|
|
||||||
"SetThreadDescription");
|
|
||||||
if (!set_description) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return SUCCEEDED(set_description(t, string::convert(name).data()));
|
|
||||||
}
|
|
||||||
|
|
||||||
bool set_name(const DWORD id, const std::string& name) {
|
|
||||||
auto* const t = OpenThread(THREAD_SET_LIMITED_INFORMATION, FALSE, id);
|
|
||||||
if (!t)
|
|
||||||
return false;
|
|
||||||
|
|
||||||
const auto _ = gsl::finally([t]() { CloseHandle(t); });
|
|
||||||
|
|
||||||
return set_name(t, name);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool set_name(std::thread& t, const std::string& name) {
|
|
||||||
return set_name(t.native_handle(), name);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool set_name(const std::string& name) {
|
|
||||||
return set_name(GetCurrentThread(), name);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<DWORD> get_thread_ids() {
|
|
||||||
auto* const h =
|
|
||||||
CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, GetCurrentProcessId());
|
|
||||||
if (h == INVALID_HANDLE_VALUE) {
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto _ = gsl::finally([h]() { CloseHandle(h); });
|
|
||||||
|
|
||||||
THREADENTRY32 entry{};
|
|
||||||
entry.dwSize = sizeof(entry);
|
|
||||||
if (!Thread32First(h, &entry)) {
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<DWORD> ids;
|
|
||||||
|
|
||||||
do {
|
|
||||||
const auto check_size =
|
|
||||||
entry.dwSize < FIELD_OFFSET(THREADENTRY32, th32OwnerProcessID) +
|
|
||||||
sizeof(entry.th32OwnerProcessID);
|
|
||||||
entry.dwSize = sizeof(entry);
|
|
||||||
|
|
||||||
if (check_size && entry.th32OwnerProcessID == GetCurrentProcessId()) {
|
|
||||||
ids.emplace_back(entry.th32ThreadID);
|
|
||||||
}
|
|
||||||
} while (Thread32Next(h, &entry));
|
|
||||||
|
|
||||||
return ids;
|
|
||||||
}
|
|
||||||
|
|
||||||
void for_each_thread(const std::function<void(HANDLE)>& callback) {
|
|
||||||
const auto ids = get_thread_ids();
|
|
||||||
|
|
||||||
for (const auto& id : ids) {
|
|
||||||
auto* const thread = OpenThread(THREAD_ALL_ACCESS, FALSE, id);
|
|
||||||
if (thread != nullptr) {
|
|
||||||
const auto _ = gsl::finally([thread]() { CloseHandle(thread); });
|
|
||||||
|
|
||||||
callback(thread);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void suspend_other_threads() {
|
|
||||||
for_each_thread([](const HANDLE thread) {
|
|
||||||
if (GetThreadId(thread) != GetCurrentThreadId()) {
|
|
||||||
SuspendThread(thread);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
void resume_other_threads() {
|
|
||||||
for_each_thread([](const HANDLE thread) {
|
|
||||||
if (GetThreadId(thread) != GetCurrentThreadId()) {
|
|
||||||
ResumeThread(thread);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
} // namespace utils::thread
|
|
@ -1,21 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
namespace utils::thread {
|
|
||||||
bool set_name(HANDLE t, const std::string& name);
|
|
||||||
bool set_name(DWORD id, const std::string& name);
|
|
||||||
bool set_name(std::thread& t, const std::string& name);
|
|
||||||
bool set_name(const std::string& name);
|
|
||||||
|
|
||||||
template <typename... Args>
|
|
||||||
std::thread create_named_thread(const std::string& name, Args&&... args) {
|
|
||||||
auto t = std::thread(std::forward<Args>(args)...);
|
|
||||||
set_name(t, name);
|
|
||||||
return t;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<DWORD> get_thread_ids();
|
|
||||||
void for_each_thread(const std::function<void(HANDLE)>& callback);
|
|
||||||
|
|
||||||
void suspend_other_threads();
|
|
||||||
void resume_other_threads();
|
|
||||||
} // namespace utils::thread
|
|
Loading…
x
Reference in New Issue
Block a user