mirror of
https://github.com/Jous99/F4MP.git
synced 2026-01-13 21:30:53 +01:00
98 lines
3 KiB
C++
98 lines
3 KiB
C++
|
|
||
|
|
#include <Games/Skyrim/BSSystem/BSThread.h>
|
||
|
|
#include <base/threading/ThreadUtils.h>
|
||
|
|
|
||
|
|
namespace
|
||
|
|
{
|
||
|
|
void (*BSThread_Initialize)(BSThread*, int, const char*){nullptr};
|
||
|
|
|
||
|
|
void Hook_BSThread_Initialize(BSThread* apThis, int aStackSize, const char* apName)
|
||
|
|
{
|
||
|
|
BSThread_Initialize(apThis, aStackSize, apName);
|
||
|
|
|
||
|
|
if (!apName && apThis->m_ThreadHandle)
|
||
|
|
spdlog::warn("Unnamed thread started: {}", apThis->m_ThreadID);
|
||
|
|
|
||
|
|
// this means the thread was successfully created
|
||
|
|
if (apThis->m_ThreadHandle)
|
||
|
|
{
|
||
|
|
bool result = Base::SetThreadName(apThis->m_ThreadHandle, apName);
|
||
|
|
if (!result)
|
||
|
|
{
|
||
|
|
spdlog::warn("Failed to set thread name for tid {} to {}", apThis->m_ThreadID, apName);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// bsthreadutils
|
||
|
|
// hook this in order to redirect the game to use the new naming apis (windows 10+)
|
||
|
|
void Hook_SetThreadName(uint32_t aThreadId, const char* apThreadName)
|
||
|
|
{
|
||
|
|
// query thread handle
|
||
|
|
if (auto hThread = ::OpenThread(THREAD_QUERY_INFORMATION, FALSE, aThreadId))
|
||
|
|
{
|
||
|
|
Base::SetThreadName(hThread, apThreadName);
|
||
|
|
|
||
|
|
::CloseHandle(hThread);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
spdlog::warn("Hook_SetThreadName(): Unable to query thread handle :(");
|
||
|
|
}
|
||
|
|
|
||
|
|
// experimental stuff..
|
||
|
|
#if 0
|
||
|
|
typedef struct
|
||
|
|
{
|
||
|
|
DWORD dwType; // Must be 0x1000.
|
||
|
|
LPCSTR szName; // Pointer to name (in user addr space).
|
||
|
|
DWORD dwThreadID; // Thread ID (-1=caller thread).
|
||
|
|
DWORD dwFlags; // Reserved for future use, must be zero.
|
||
|
|
} THREADNAME_INFO;
|
||
|
|
|
||
|
|
void Hook_RaiseException(DWORD, DWORD, DWORD, const THREADNAME_INFO* lpArguments)
|
||
|
|
{
|
||
|
|
base::SetCurrentThreadName(lpArguments->szName);
|
||
|
|
}
|
||
|
|
|
||
|
|
HANDLE WINAPI Hook_CreateThread(LPSECURITY_ATTRIBUTES lpThreadAttributes, SIZE_T dwStackSize,
|
||
|
|
LPTHREAD_START_ROUTINE lpStartAddress, LPVOID lpParameter, DWORD dwCreationFlags,
|
||
|
|
LPDWORD lpThreadId)
|
||
|
|
{
|
||
|
|
auto hThread =
|
||
|
|
CreateThread(lpThreadAttributes, dwStackSize, lpStartAddress, lpParameter, dwCreationFlags, lpThreadId);
|
||
|
|
|
||
|
|
if (hThread && lpThreadId)
|
||
|
|
{
|
||
|
|
auto name = fmt::format("BNetThread{}", *lpThreadId);
|
||
|
|
base::SetThreadName(hThread, name.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
return hThread;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
static TiltedPhoques::Initializer s_BSThreadInit(
|
||
|
|
[]()
|
||
|
|
{
|
||
|
|
const VersionDbPtr<uint8_t> threadInit(68261);
|
||
|
|
BSThread_Initialize = static_cast<decltype(BSThread_Initialize)>(threadInit.GetPtr());
|
||
|
|
// need to detour this for now :/
|
||
|
|
TP_HOOK_IMMEDIATE(&BSThread_Initialize, &Hook_BSThread_Initialize);
|
||
|
|
|
||
|
|
const VersionDbPtr<uint8_t> setThreadName(69066);
|
||
|
|
TiltedPhoques::Jump(setThreadName.Get(), &Hook_SetThreadName);
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
const VersionDbPtr<uint8_t> createHavokThread(57704);
|
||
|
|
// relatively safe to do, since this is unlikely to ever change, as beth wont update havok
|
||
|
|
TiltedPhoques::Nop(createHavokThread.Get() + 0x81, 6);
|
||
|
|
|
||
|
|
TiltedPhoques::PutCall(createHavokThread.Get() + 0x81, &Hook_RaiseException);
|
||
|
|
|
||
|
|
|
||
|
|
TiltedPhoques::Nop(0x1411F0FD4, 6);
|
||
|
|
TiltedPhoques::PutCall(0x1411F0FD4, &Hook_CreateThread);
|
||
|
|
#endif
|
||
|
|
});
|