thirdpartymirrors/UnleashedRecomp
1
0
Fork 0
mirror of https://github.com/hedge-dev/UnleashedRecomp.git synced 2026-04-27 21:01:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cross-platform TLS.

Browse source
This commit is contained in:
Dario 2024-12-14 13:15:34 -03:00
parent 9b6e4406b7
commit d2ff8d9772
1 changed files with 30 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
UnleashedRecomp/kernel/imports.cpp
View file

@ -901,24 +901,50 @@ DWORD KeWaitForSingleObject(XDISPATCHER_HEADER* Object, DWORD WaitReason, DWORD
return WaitForSingleObjectEx(handle, timeout, Alertable); return WaitForSingleObjectEx(handle, timeout, Alertable);
} }
static thread_local std::vector<uint32_t> KeTlsValues;
static std::vector<size_t> KeTlsFreeIndices;
static size_t KeTlsNextIndex = 0;
static std::mutex KeTlsAllocationMutex;
static void KeTlsEnsureTlsCapacity(size_t index)
{
if (KeTlsValues.size() <= index)
{
KeTlsValues.resize(index + 1, 0);
}
}
uint32_t KeTlsGetValue(DWORD dwTlsIndex) uint32_t KeTlsGetValue(DWORD dwTlsIndex)
{ {
return (uint32_t)TlsGetValue(dwTlsIndex); KeTlsEnsureTlsCapacity(dwTlsIndex);
return KeTlsValues[dwTlsIndex];
} }
BOOL KeTlsSetValue(DWORD dwTlsIndex, DWORD lpTlsValue) BOOL KeTlsSetValue(DWORD dwTlsIndex, DWORD lpTlsValue)
{ {
return TlsSetValue(dwTlsIndex, (LPVOID)lpTlsValue); KeTlsEnsureTlsCapacity(dwTlsIndex);
KeTlsValues[dwTlsIndex] = lpTlsValue;
return TRUE;
} }
DWORD KeTlsAlloc() DWORD KeTlsAlloc()
{ {
return TlsAlloc(); std::lock_guard<std::mutex> lock(KeTlsAllocationMutex);
if (!KeTlsFreeIndices.empty())
{
size_t index = KeTlsFreeIndices.front();
KeTlsFreeIndices.pop_back();
return index;
}
return KeTlsNextIndex++;
} }
BOOL KeTlsFree(DWORD dwTlsIndex) BOOL KeTlsFree(DWORD dwTlsIndex)
{ {
return TlsFree(dwTlsIndex); std::lock_guard<std::mutex> lock(KeTlsAllocationMutex);
KeTlsFreeIndices.push_back(dwTlsIndex);
return TRUE;
} }
DWORD XMsgInProcessCall(uint32_t app, uint32_t message, XDWORD* param1, XDWORD* param2) DWORD XMsgInProcessCall(uint32_t app, uint32_t message, XDWORD* param1, XDWORD* param2)