mirror of
https://github.com/RPCS3/rpcs3.git
synced 2025-07-10 17:01:24 +12:00
Fixed TLS for PPU threads
This commit is contained in:
parent
2be045593a
commit
4ea1b8cffc
4 changed files with 72 additions and 3 deletions
|
@ -23,11 +23,62 @@
|
|||
|
||||
Module *sysPrxForUser = nullptr;
|
||||
|
||||
u32 g_tls_size; // size of every thread's storage
|
||||
u32 g_tls_start; // start of TLS memory area
|
||||
u32 g_tls_image_addr; // address of TLS initialization area
|
||||
u32 g_tls_image_size; // size of TLS initialization area
|
||||
|
||||
const u32 TLS_MAX = 256;
|
||||
std::array<std::atomic<u32>, TLS_MAX> g_tls_owners;
|
||||
|
||||
void sys_initialize_tls()
|
||||
{
|
||||
sysPrxForUser->Log("sys_initialize_tls()");
|
||||
}
|
||||
|
||||
u32 ppu_get_tls(u32 thread)
|
||||
{
|
||||
if (!g_tls_start)
|
||||
{
|
||||
g_tls_size = vm::cast(Emu.GetTLSMemsz(), "Emu.GetTLSMemsz"); // (not an address for vm::cast, but fine)
|
||||
g_tls_start = vm::cast(Memory.Alloc(g_tls_size * TLS_MAX, 4096)); // memory for up to TLS_MAX threads
|
||||
g_tls_image_addr = vm::cast(Emu.GetTLSAddr(), "Emu.GetTLSAddr");
|
||||
g_tls_image_size = vm::cast(Emu.GetTLSFilesz(), "Emu.GetTLSFilesz");
|
||||
|
||||
sysPrxForUser->Warning("TLS initialized (g_tls_size=0x%x, g_tls_start=0x%x, g_tls_image_addr=0x%x, g_tls_image_size=0x%x)", g_tls_size, g_tls_start, g_tls_image_addr, g_tls_image_size);
|
||||
}
|
||||
|
||||
for (auto& v : g_tls_owners)
|
||||
{
|
||||
u32 old = 0;
|
||||
if (v.compare_exchange_strong(old, thread))
|
||||
{
|
||||
const u32 addr = g_tls_start + (&v - g_tls_owners.data()) * g_tls_size; // get TLS address
|
||||
memset(vm::get_ptr(addr), 0, g_tls_size); // fill TLS area with zeros
|
||||
memcpy(vm::get_ptr(addr), vm::get_ptr(g_tls_image_addr), g_tls_image_size); // initialize from TLS image
|
||||
return addr;
|
||||
}
|
||||
else if (old == thread)
|
||||
{
|
||||
return g_tls_start + (&v - g_tls_owners.data()) * g_tls_size; // if already initialized, return TLS address again
|
||||
}
|
||||
}
|
||||
|
||||
throw "Out of TLS memory";
|
||||
}
|
||||
|
||||
void ppu_free_tls(u32 thread)
|
||||
{
|
||||
for (auto& v : g_tls_owners)
|
||||
{
|
||||
u32 old = thread;
|
||||
if (v.compare_exchange_strong(old, 0))
|
||||
{
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int _sys_heap_create_heap(const u32 heap_addr, const u32 align, const u32 size)
|
||||
{
|
||||
sysPrxForUser->Warning("_sys_heap_create_heap(heap_addr=0x%x, align=0x%x, size=0x%x)", heap_addr, align, size);
|
||||
|
@ -365,6 +416,15 @@ void sysPrxForUser_init(Module *pxThis)
|
|||
{
|
||||
sysPrxForUser = pxThis;
|
||||
|
||||
g_tls_size = 0;
|
||||
g_tls_start = 0;
|
||||
g_tls_image_addr = 0;
|
||||
g_tls_image_size = 0;
|
||||
for (auto& v : g_tls_owners)
|
||||
{
|
||||
v.store(0, std::memory_order_relaxed);
|
||||
}
|
||||
|
||||
// Setup random number generator
|
||||
srand(time(NULL));
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue