mirror of
https://github.com/shadps4-emu/shadPS4.git
synced 2026-04-11 12:11:30 -06:00
84 lines
1.9 KiB
C++
84 lines
1.9 KiB
C++
// SPDX-FileCopyrightText: Copyright 2025 shadPS4 Emulator Project
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#pragma once
|
|
|
|
#include <cstring>
|
|
#include "common/types.h"
|
|
#ifdef _WIN32
|
|
#include <malloc.h>
|
|
#endif
|
|
|
|
namespace Xbyak {
|
|
class CodeGenerator;
|
|
}
|
|
|
|
namespace Libraries::Fiber {
|
|
struct OrbisFiberContext;
|
|
}
|
|
|
|
namespace Core {
|
|
|
|
union DtvEntry {
|
|
std::size_t counter;
|
|
u8* pointer;
|
|
};
|
|
|
|
struct Tcb {
|
|
Tcb* tcb_self;
|
|
DtvEntry* tcb_dtv;
|
|
void* tcb_thread;
|
|
::Libraries::Fiber::OrbisFiberContext* tcb_fiber;
|
|
};
|
|
|
|
#ifdef _WIN32
|
|
/// Gets the thread local storage key for the TCB block.
|
|
u32 GetTcbKey();
|
|
#endif
|
|
|
|
/// Sets the data pointer to the TCB block.
|
|
void SetTcbBase(void* image_address);
|
|
|
|
/// Retrieves Tcb structure for the calling thread.
|
|
Tcb* GetTcbBase();
|
|
|
|
/// Makes sure TLS is initialized for the thread before entering guest.
|
|
void EnsureThreadInitialized();
|
|
|
|
template <size_t size>
|
|
#ifdef __clang__
|
|
__attribute__((optnone))
|
|
#else
|
|
__attribute__((optimize("O0")))
|
|
#endif
|
|
void ClearStack() {
|
|
volatile void* buf = alloca(size);
|
|
memset(const_cast<void*>(buf), 0, size);
|
|
buf = nullptr;
|
|
}
|
|
|
|
template <class ReturnType, class... FuncArgs, class... CallArgs>
|
|
ReturnType ExecuteGuest(PS4_SYSV_ABI ReturnType (*func)(FuncArgs...), CallArgs&&... args) {
|
|
EnsureThreadInitialized();
|
|
// clear stack to avoid trash from EnsureThreadInitialized
|
|
auto* tcb = GetTcbBase();
|
|
if (tcb != nullptr && tcb->tcb_fiber == nullptr) {
|
|
ClearStack<12_KB>();
|
|
}
|
|
return func(std::forward<CallArgs>(args)...);
|
|
}
|
|
|
|
template <class F, F f>
|
|
struct HostCallWrapperImpl;
|
|
|
|
template <class ReturnType, class... Args, PS4_SYSV_ABI ReturnType (*func)(Args...)>
|
|
struct HostCallWrapperImpl<PS4_SYSV_ABI ReturnType (*)(Args...), func> {
|
|
static ReturnType PS4_SYSV_ABI wrap(Args... args) {
|
|
return func(args...);
|
|
}
|
|
};
|
|
|
|
#define HOST_CALL(func) (Core::HostCallWrapperImpl<decltype(&(func)), func>::wrap)
|
|
|
|
} // namespace Core
|