mirrors/shadPS4
1
0
Fork 0
mirror of https://github.com/shadps4-emu/shadPS4.git synced 2025-08-04 00:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

tls: Add caching for Windows exception handler.

Browse Source
This commit is contained in:
squidbus 2024-09-17 10:33:46 -07:00
parent c70014a992
commit 69472d79a4
1 changed files with 33 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/core/tls.cpp
View File

@ -28,10 +28,19 @@ namespace Core {
// Windows // Windows
static DWORD slot = 0; static DWORD slot = 0;
static std::set<void*> known_fs_accesses;
static std::mutex known_fs_accesses_lock;
static ZydisDecoder tls_instr_decoder; static ZydisDecoder tls_instr_decoder;
static std::once_flag slot_alloc_flag; static std::once_flag slot_alloc_flag;
static bool TlsAccessViolationHandler(void* code_address, void* fault_address, bool is_write) { static bool TlsAccessViolationHandler(void* code_address, void* fault_address, bool is_write) {
bool known;
{
std::unique_lock lock{known_fs_accesses_lock};
known = known_fs_accesses.contains(code_address);
}
if (!known) {
ZydisDecodedInstruction instruction; ZydisDecodedInstruction instruction;
ZydisDecodedOperand operands[ZYDIS_MAX_OPERAND_COUNT]; ZydisDecodedOperand operands[ZYDIS_MAX_OPERAND_COUNT];
const auto status = const auto status =
@ -43,12 +52,23 @@ static bool TlsAccessViolationHandler(void* code_address, void* fault_address, b
for (u32 i = 0; i < instruction.operand_count_visible; i++) { for (u32 i = 0; i < instruction.operand_count_visible; i++) {
if (operands[i].type == ZYDIS_OPERAND_TYPE_MEMORY && if (operands[i].type == ZYDIS_OPERAND_TYPE_MEMORY &&
operands[i].mem.segment == ZYDIS_REGISTER_FS) { operands[i].mem.segment == ZYDIS_REGISTER_FS) {
known = true;
break;
}
}
if (known) {
std::unique_lock lock{known_fs_accesses_lock};
known_fs_accesses.insert(code_address);
}
}
if (known) {
// Set the FS register and try again. // Set the FS register and try again.
const auto* tcb_base = GetTcbBase(); const auto* tcb_base = GetTcbBase();
asm volatile("wrfsbase %0" ::"r"(tcb_base) : "memory"); asm volatile("wrfsbase %0" ::"r"(tcb_base) : "memory");
return true; return true;
} }
}
return false; return false;
} }