2020-12-09 10:47:45 +03:00
#include "stdafx.h"
2020-10-30 21:26:22 +01:00
#include "Emu/System.h"
2019-02-17 17:53:38 +02:00
#include "Emu/Cell/SPUThread.h"
2019-06-06 21:32:35 +03:00
#include "Emu/Cell/PPUThread.h"
2017-10-08 15:37:54 -05:00
#include "Emu/Cell/lv2/sys_mmapper.h"
#include "Emu/Cell/lv2/sys_event.h"
2023-05-19 21:52:57 +03:00
#include "Emu/Cell/lv2/sys_process.h"
2021-01-15 20:28:45 +02:00
#include "Emu/RSX/RSXThread.h"
2012-11-15 01:39:56 +02:00
#include "Thread.h"
2020-11-13 16:59:56 +03:00
#include "Utilities/JIT.h"
2017-10-21 14:21:37 +03:00
#include <thread>
2021-03-29 23:05:24 +03:00
#include <cfenv>
2012-11-15 01:39:56 +02:00
2014-12-05 19:12:15 +03:00
#ifdef _WIN32
2016-08-06 15:09:44 +03:00
#include <Windows.h>
#include <Psapi.h>
2016-09-07 01:38:52 +03:00
#include <process.h>
2018-12-23 20:01:09 +02:00
#include <sysinfoapi.h>
2023-05-24 14:58:19 +03:00
#include "util/dyn_lib.hpp"
DYNAMIC_IMPORT_RENAME ( "Kernel32.dll" , SetThreadDescriptionImport , "SetThreadDescription" , HRESULT ( HANDLE hThread , PCWSTR lpThreadDescription ));
2015-01-05 00:46:31 +03:00
#else
2022-09-13 16:08:55 +03:00
#ifndef _GNU_SOURCE
2022-07-09 14:42:03 +03:00
#define _GNU_SOURCE
2022-09-13 16:08:55 +03:00
#endif
2015-02-14 00:45:36 +03:00
#ifdef __APPLE__
2015-02-10 23:17:39 -05:00
#define _XOPEN_SOURCE
#define __USE_GNU
2017-11-19 23:02:35 +01:00
#include <mach/thread_act.h>
#include <mach/thread_policy.h>
2015-02-14 00:45:36 +03:00
#endif
2018-03-12 20:13:49 +00:00
#if defined(__DragonFly__) || defined(__FreeBSD__) || defined(__OpenBSD__)
2017-11-20 21:56:25 +00:00
#include <pthread_np.h>
#define cpu_set_t cpuset_t
#endif
2016-05-25 21:04:08 +03:00
#include <errno.h>
2015-01-05 00:46:31 +03:00
#include <signal.h>
2019-03-19 00:02:21 +00:00
#ifndef __OpenBSD__
2015-01-05 00:46:31 +03:00
#include <ucontext.h>
2019-03-19 00:02:21 +00:00
#endif
2016-09-07 01:38:52 +03:00
#include <pthread.h>
#include <sys/time.h>
#include <sys/resource.h>
2018-05-24 22:22:07 +03:00
#include <time.h>
2014-12-05 19:12:15 +03:00
#endif
2019-10-09 19:03:34 +02:00
#ifdef __linux__
#include <sys/timerfd.h>
2020-11-13 16:59:56 +03:00
#include <unistd.h>
2019-10-09 19:03:34 +02:00
#endif
2014-12-05 19:12:15 +03:00
2020-03-28 01:57:41 +00:00
#if defined(__APPLE__) || defined(__DragonFly__) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)
# include <sys/sysctl.h>
# include <unistd.h>
# if defined(__DragonFly__) || defined(__FreeBSD__)
# include <sys/user.h>
# endif
# if defined(__OpenBSD__)
# include <sys/param.h>
# include <sys/proc.h>
# endif
# if defined(__NetBSD__)
# undef KERN_PROC
# define KERN_PROC KERN_PROC2
# define kinfo_proc kinfo_proc2
# endif
# if defined(__APPLE__)
# define KP_FLAGS kp_proc.p_flag
# elif defined(__DragonFly__)
# define KP_FLAGS kp_flags
# elif defined(__FreeBSD__)
# define KP_FLAGS ki_flag
# elif defined(__NetBSD__)
# define KP_FLAGS p_flag
# elif defined(__OpenBSD__)
# define KP_FLAGS p_psflags
# define P_TRACED PS_TRACED
# endif
#endif
2020-11-08 01:56:35 +03:00
#include "util/vm.hpp"
2020-03-07 12:29:23 +03:00
#include "util/logs.hpp"
2020-11-13 11:32:47 +03:00
#include "util/asm.hpp"
2020-12-21 17:12:05 +03:00
#include "util/v128.hpp"
2021-12-30 19:39:18 +03:00
#include "util/simd.hpp"
2020-12-18 12:55:54 +03:00
#include "util/sysinfo.hpp"
2020-11-03 07:18:42 +03:00
#include "Emu/Memory/vm_locking.h"
2016-09-07 01:38:52 +03:00
2020-03-28 15:25:39 +01:00
LOG_CHANNEL ( sig_log , "SIG" );
2020-02-22 12:58:42 +02:00
LOG_CHANNEL ( sys_log , "SYS" );
2020-02-01 10:43:43 +03:00
LOG_CHANNEL ( vm_log , "VM" );
2020-01-31 14:20:54 +03:00
2017-02-17 22:35:57 +03:00
thread_local u64 g_tls_fault_all = 0 ;
2016-09-07 01:38:52 +03:00
thread_local u64 g_tls_fault_rsx = 0 ;
thread_local u64 g_tls_fault_spu = 0 ;
2020-11-13 19:06:44 +03:00
thread_local u64 g_tls_wait_time = 0 ;
thread_local u64 g_tls_wait_fail = 0 ;
2020-11-21 07:56:54 +03:00
thread_local bool g_tls_access_violation_recovered = false ;
2018-09-25 15:21:04 +03:00
extern thread_local std :: string ( * g_tls_log_prefix )();
2016-09-07 01:38:52 +03:00
2020-12-09 18:04:52 +03:00
// Report error and call std::abort(), defined in main.cpp
2022-12-06 18:26:38 +03:00
[[noreturn]] void report_fatal_error ( std :: string_view text , bool is_html = false , bool include_help_text = true );
2020-12-09 18:04:52 +03:00
2021-01-15 20:28:45 +02:00
std :: string dump_useful_thread_info ()
{
std :: string result ;
if ( auto cpu = get_current_cpu_thread ())
{
2023-07-09 08:45:15 +03:00
cpu -> dump_all ( result );
2021-01-15 20:28:45 +02:00
}
return result ;
}
2020-03-10 11:31:11 +03:00
#ifndef _WIN32
bool IsDebuggerPresent ()
{
2020-03-28 01:57:41 +00:00
#if defined(__APPLE__) || defined(__DragonFly__) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)
int mib [] = {
CTL_KERN ,
KERN_PROC ,
KERN_PROC_PID ,
getpid (),
# if defined(__NetBSD__) || defined(__OpenBSD__)
sizeof ( struct kinfo_proc ),
1 ,
# endif
};
u_int miblen = std :: size ( mib );
struct kinfo_proc info ;
2020-12-18 10:39:54 +03:00
usz size = sizeof ( info );
2020-03-28 01:57:41 +00:00
if ( sysctl ( mib , miblen , & info , & size , NULL , 0 ))
{
return false ;
}
return info . KP_FLAGS & P_TRACED ;
#else
2020-03-10 11:31:11 +03:00
char buf [ 4096 ];
fs :: file status_fd ( "/proc/self/status" );
if ( ! status_fd )
{
std :: fprintf ( stderr , "Failed to open /proc/self/status \n " );
return false ;
}
const auto num_read = status_fd . read ( buf , sizeof ( buf ) - 1 );
if ( num_read == 0 || num_read == umax )
{
std :: fprintf ( stderr , "Failed to read /proc/self/status (%d) \n " , errno );
return false ;
}
buf [ num_read ] = '\0' ;
std :: string_view status = buf ;
const auto found = status . find ( "TracerPid:" );
if ( found == umax )
{
std :: fprintf ( stderr , "Failed to find 'TracerPid:' in /proc/self/status \n " );
return false ;
}
for ( const char * cp = status . data () + found + 10 ; cp <= status . data () + num_read ; ++ cp )
{
if ( ! std :: isspace ( * cp ))
{
return std :: isdigit ( * cp ) != 0 && * cp != '0' ;
}
}
return false ;
2020-03-28 01:57:41 +00:00
#endif
2020-03-10 11:31:11 +03:00
}
#endif
2022-03-25 20:51:55 +03:00
bool is_debugger_present ()
{
if ( g_cfg . core . external_debugger )
return true ;
return IsDebuggerPresent ();
}
2021-12-30 19:39:18 +03:00
#if defined(ARCH_X64)
2015-01-05 00:46:31 +03:00
enum x64_reg_t : u32
{
2015-02-17 03:08:23 +03:00
X64R_RAX = 0 ,
2015-02-15 15:31:42 +03:00
X64R_RCX ,
X64R_RDX ,
X64R_RBX ,
X64R_RSP ,
X64R_RBP ,
X64R_RSI ,
X64R_RDI ,
X64R_R8 ,
X64R_R9 ,
X64R_R10 ,
X64R_R11 ,
X64R_R12 ,
X64R_R13 ,
X64R_R14 ,
X64R_R15 ,
2015-01-05 00:46:31 +03:00
2015-02-17 03:08:23 +03:00
X64R_XMM0 = 0 ,
2015-02-15 15:31:42 +03:00
X64R_XMM1 ,
X64R_XMM2 ,
X64R_XMM3 ,
X64R_XMM4 ,
X64R_XMM5 ,
X64R_XMM6 ,
X64R_XMM7 ,
X64R_XMM8 ,
X64R_XMM9 ,
X64R_XMM10 ,
X64R_XMM11 ,
X64R_XMM12 ,
X64R_XMM13 ,
X64R_XMM14 ,
X64R_XMM15 ,
2018-02-09 15:24:46 +03:00
2015-02-15 15:31:42 +03:00
X64R_AL ,
X64R_CL ,
X64R_DL ,
X64R_BL ,
X64R_AH ,
X64R_CH ,
X64R_DH ,
X64R_BH ,
2018-02-09 15:24:46 +03:00
2015-02-15 15:31:42 +03:00
X64_NOT_SET ,
X64_IMM8 ,
X64_IMM16 ,
2015-01-05 00:46:31 +03:00
X64_IMM32 ,
2015-02-15 15:31:42 +03:00
2016-06-07 23:24:20 +03:00
X64_BIT_O = 0x90 ,
X64_BIT_NO ,
X64_BIT_C ,
X64_BIT_NC ,
X64_BIT_Z ,
X64_BIT_NZ ,
X64_BIT_BE ,
X64_BIT_NBE ,
X64_BIT_S ,
X64_BIT_NS ,
X64_BIT_P ,
X64_BIT_NP ,
X64_BIT_L ,
X64_BIT_NL ,
X64_BIT_LE ,
X64_BIT_NLE ,
2015-02-15 15:31:42 +03:00
X64R_ECX = X64R_CL ,
2015-01-05 00:46:31 +03:00
};
enum x64_op_t : u32
{
2015-02-15 20:13:06 +03:00
X64OP_NONE ,
2015-03-05 00:51:14 +03:00
X64OP_LOAD , // obtain and put the value into x64 register
2016-06-07 23:24:20 +03:00
X64OP_LOAD_BE ,
2016-06-23 00:39:39 +03:00
X64OP_LOAD_CMP ,
X64OP_LOAD_TEST ,
2015-03-05 00:51:14 +03:00
X64OP_STORE , // take the value from x64 register or an immediate and use it
2016-06-07 23:24:20 +03:00
X64OP_STORE_BE ,
2015-02-15 15:31:42 +03:00
X64OP_MOVS ,
2015-02-17 05:01:47 +03:00
X64OP_STOS ,
2015-02-15 15:31:42 +03:00
X64OP_XCHG ,
X64OP_CMPXCHG ,
2016-06-07 23:24:20 +03:00
X64OP_AND , // lock and [mem], ...
X64OP_OR , // lock or [mem], ...
X64OP_XOR , // lock xor [mem], ...
X64OP_INC , // lock inc [mem]
X64OP_DEC , // lock dec [mem]
X64OP_ADD , // lock add [mem], ...
X64OP_ADC , // lock adc [mem], ...
X64OP_SUB , // lock sub [mem], ...
X64OP_SBB , // lock sbb [mem], ...
2023-05-05 12:39:09 +03:00
X64OP_BEXTR ,
2015-01-05 00:46:31 +03:00
};
2023-05-05 12:39:09 +03:00
static thread_local x64_reg_t s_tls_reg3 {};
2020-12-18 10:39:54 +03:00
void decode_x64_reg_op ( const u8 * code , x64_op_t & out_op , x64_reg_t & out_reg , usz & out_size , usz & out_length )
2015-01-05 00:46:31 +03:00
{
// simple analysis of x64 code allows to reinterpret MOV or other instructions in any desired way
2015-02-15 15:31:42 +03:00
out_length = 0 ;
2015-01-05 00:46:31 +03:00
2015-02-15 15:31:42 +03:00
u8 rex = 0 , pg2 = 0 ;
bool oso = false , lock = false , repne = false , repe = false ;
enum : u8
{
LOCK = 0xf0 ,
REPNE = 0xf2 ,
REPE = 0xf3 ,
};
2015-01-05 00:46:31 +03:00
// check prefixes:
2015-02-15 15:31:42 +03:00
for (;; code ++ , out_length ++ )
2015-01-05 00:46:31 +03:00
{
switch ( const u8 prefix = * code )
{
2015-02-15 15:31:42 +03:00
case LOCK : // group 1
{
if ( lock )
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): LOCK prefix found twice" , code - out_length );
2015-02-15 15:31:42 +03:00
}
2018-02-09 15:24:46 +03:00
2015-02-15 15:31:42 +03:00
lock = true ;
continue ;
}
case REPNE : // group 1
{
if ( repne )
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): REPNE/REPNZ prefix found twice" , code - out_length );
2015-02-15 15:31:42 +03:00
}
2018-02-09 15:24:46 +03:00
2015-02-15 15:31:42 +03:00
repne = true ;
continue ;
}
case REPE : // group 1
{
if ( repe )
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): REP/REPE/REPZ prefix found twice" , code - out_length );
2015-02-15 15:31:42 +03:00
}
2018-02-09 15:24:46 +03:00
2015-02-15 15:31:42 +03:00
repe = true ;
continue ;
}
2015-01-05 00:46:31 +03:00
case 0x2e : // group 2
case 0x36 :
case 0x3e :
case 0x26 :
case 0x64 :
case 0x65 :
{
2015-02-15 15:31:42 +03:00
if ( pg2 )
2015-01-05 00:46:31 +03:00
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): 0x%02x (group 2 prefix) found after 0x%02x" , code - out_length , prefix , pg2 );
2015-01-05 00:46:31 +03:00
}
else
{
2015-02-15 15:31:42 +03:00
pg2 = prefix ; // probably, segment register
2015-01-05 00:46:31 +03:00
}
2015-02-15 15:31:42 +03:00
continue ;
2015-01-05 00:46:31 +03:00
}
2015-02-15 15:31:42 +03:00
case 0x66 : // group 3
{
if ( oso )
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): operand-size override prefix found twice" , code - out_length );
2015-02-15 15:31:42 +03:00
}
2018-02-09 15:24:46 +03:00
2015-02-15 15:31:42 +03:00
oso = true ;
continue ;
}
case 0x67 : // group 4
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): address-size override prefix found" , code - out_length , prefix );
2015-02-15 20:13:06 +03:00
out_op = X64OP_NONE ;
2015-02-15 15:31:42 +03:00
out_reg = X64_NOT_SET ;
out_size = 0 ;
out_length = 0 ;
return ;
}
2015-01-05 00:46:31 +03:00
default :
{
if (( prefix & 0xf0 ) == 0x40 ) // check REX prefix
{
if ( rex )
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "decode_x64_reg_op(%016llxh): 0x%02x (REX prefix) found after 0x%02x" , code - out_length , prefix , rex );
2015-01-05 00:46:31 +03:00
}
2015-02-15 15:31:42 +03:00
else
2015-01-05 00:46:31 +03:00
{
2015-02-15 15:31:42 +03:00
rex = prefix ;
2015-01-05 00:46:31 +03:00
}
continue ;
}
}
}
break ;
}
2015-02-15 15:31:42 +03:00
auto get_modRM_reg = []( const u8 * code , const u8 rex ) -> x64_reg_t
2015-01-05 00:46:31 +03:00
{
2019-11-28 21:18:37 +03:00
return x64_reg_t {(( * code & 0x38 ) >> 3 | ( /* check REX.R bit */ rex & 4 ? 8 : 0 )) + X64R_RAX };
2015-01-05 00:46:31 +03:00
};
2015-02-15 15:31:42 +03:00
auto get_modRM_reg_xmm = []( const u8 * code , const u8 rex ) -> x64_reg_t
{
2019-11-28 21:18:37 +03:00
return x64_reg_t {(( * code & 0x38 ) >> 3 | ( /* check REX.R bit */ rex & 4 ? 8 : 0 )) + X64R_XMM0 };
2015-02-15 15:31:42 +03:00
};
auto get_modRM_reg_lh = []( const u8 * code ) -> x64_reg_t
{
2019-11-28 21:18:37 +03:00
return x64_reg_t {(( * code & 0x38 ) >> 3 ) + X64R_AL };
2015-02-15 15:31:42 +03:00
};
2020-12-18 10:39:54 +03:00
auto get_op_size = []( const u8 rex , const bool oso ) -> usz
2015-02-15 15:31:42 +03:00
{
return rex & 8 ? 8 : ( oso ? 2 : 4 );
};
2020-12-18 10:39:54 +03:00
auto get_modRM_size = []( const u8 * code ) -> usz
2015-01-05 00:46:31 +03:00
{
switch ( * code >> 6 ) // check Mod
{
case 0 : return ( * code & 0x07 ) == 4 ? 2 : 1 ; // check SIB
case 1 : return ( * code & 0x07 ) == 4 ? 3 : 2 ; // check SIB (disp8)
case 2 : return ( * code & 0x07 ) == 4 ? 6 : 5 ; // check SIB (disp32)
default : return 1 ;
}
};
2015-02-15 15:31:42 +03:00
const u8 op1 = ( out_length ++ , * code ++ ), op2 = code [ 0 ], op3 = code [ 1 ];
switch ( op1 )
2015-01-05 00:46:31 +03:00
{
2015-02-15 15:31:42 +03:00
case 0x0f :
2015-01-05 00:46:31 +03:00
{
2015-02-15 15:31:42 +03:00
out_length ++ , code ++ ;
switch ( op2 )
{
2015-07-27 14:40:22 +03:00
case 0x11 :
2016-06-25 08:16:15 +03:00
case 0x29 :
2015-07-27 14:40:22 +03:00
{
2016-06-25 08:16:15 +03:00
if ( ! repe && ! repne ) // MOVUPS/MOVAPS/MOVUPD/MOVAPD xmm/m, xmm
2015-07-27 14:40:22 +03:00
{
out_op = X64OP_STORE ;
out_reg = get_modRM_reg_xmm ( code , rex );
out_size = 16 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
2015-02-15 15:31:42 +03:00
case 0x7f :
{
2021-04-14 20:44:37 +01:00
if ( repe != oso ) // MOVDQU/MOVDQA xmm/m, xmm
2015-02-15 15:31:42 +03:00
{
out_op = X64OP_STORE ;
out_reg = get_modRM_reg_xmm ( code , rex );
out_size = 16 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0xb0 :
{
if ( ! oso ) // CMPXCHG r8/m8, r8
{
out_op = X64OP_CMPXCHG ;
out_reg = rex & 8 ? get_modRM_reg ( code , rex ) : get_modRM_reg_lh ( code );
out_size = 1 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0xb1 :
{
if ( true ) // CMPXCHG r/m, r (16, 32, 64)
{
out_op = X64OP_CMPXCHG ;
out_reg = get_modRM_reg ( code , rex );
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code );
return ;
}
break ;
}
2016-06-07 23:24:20 +03:00
case 0x90 :
case 0x91 :
case 0x92 :
case 0x93 :
case 0x94 :
case 0x95 :
case 0x96 :
case 0x97 :
case 0x98 :
case 0x9a :
case 0x9b :
case 0x9c :
case 0x9d :
case 0x9e :
case 0x9f :
{
if ( ! lock ) // SETcc
{
out_op = X64OP_STORE ;
out_reg = x64_reg_t ( X64_BIT_O + op2 - 0x90 ); // 0x90 .. 0x9f
out_size = 1 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x38 :
{
out_length ++ , code ++ ;
switch ( op3 )
{
case 0xf0 :
case 0xf1 :
{
if ( ! repne ) // MOVBE
{
out_op = op3 == 0xf0 ? X64OP_LOAD_BE : X64OP_STORE_BE ;
out_reg = get_modRM_reg ( code , rex );
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code );
return ;
}
break ;
}
}
break ;
}
2015-02-15 15:31:42 +03:00
}
break ;
2015-01-05 00:46:31 +03:00
}
2015-03-07 01:10:04 +03:00
case 0x20 :
{
if ( ! oso )
{
2016-06-07 23:24:20 +03:00
out_op = X64OP_AND ;
2015-03-07 01:10:04 +03:00
out_reg = rex & 8 ? get_modRM_reg ( code , rex ) : get_modRM_reg_lh ( code );
out_size = 1 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x21 :
{
if ( true )
{
2016-06-07 23:24:20 +03:00
out_op = X64OP_AND ;
2015-03-07 01:10:04 +03:00
out_reg = get_modRM_reg ( code , rex );
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code );
return ;
}
break ;
}
2016-06-07 23:24:20 +03:00
case 0x80 :
{
2021-01-12 13:01:06 +03:00
switch ( get_modRM_reg ( code , 0 ))
2016-06-07 23:24:20 +03:00
{
//case 0: out_op = X64OP_ADD; break; // TODO: strange info in instruction manual
case 1 : out_op = X64OP_OR ; break ;
case 2 : out_op = X64OP_ADC ; break ;
case 3 : out_op = X64OP_SBB ; break ;
case 4 : out_op = X64OP_AND ; break ;
case 5 : out_op = X64OP_SUB ; break ;
case 6 : out_op = X64OP_XOR ; break ;
2016-06-23 00:39:39 +03:00
default : out_op = X64OP_LOAD_CMP ; break ;
2016-06-07 23:24:20 +03:00
}
out_reg = X64_IMM8 ;
out_size = 1 ;
out_length += get_modRM_size ( code ) + 1 ;
return ;
}
case 0x81 :
{
2021-01-12 13:01:06 +03:00
switch ( get_modRM_reg ( code , 0 ))
2016-06-07 23:24:20 +03:00
{
case 0 : out_op = X64OP_ADD ; break ;
case 1 : out_op = X64OP_OR ; break ;
case 2 : out_op = X64OP_ADC ; break ;
case 3 : out_op = X64OP_SBB ; break ;
case 4 : out_op = X64OP_AND ; break ;
case 5 : out_op = X64OP_SUB ; break ;
case 6 : out_op = X64OP_XOR ; break ;
2016-06-23 00:39:39 +03:00
default : out_op = X64OP_LOAD_CMP ; break ;
2016-06-07 23:24:20 +03:00
}
out_reg = oso ? X64_IMM16 : X64_IMM32 ;
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code ) + ( oso ? 2 : 4 );
return ;
}
case 0x83 :
{
2021-01-12 13:01:06 +03:00
switch ( get_modRM_reg ( code , 0 ))
2016-06-07 23:24:20 +03:00
{
case 0 : out_op = X64OP_ADD ; break ;
case 1 : out_op = X64OP_OR ; break ;
case 2 : out_op = X64OP_ADC ; break ;
case 3 : out_op = X64OP_SBB ; break ;
case 4 : out_op = X64OP_AND ; break ;
case 5 : out_op = X64OP_SUB ; break ;
case 6 : out_op = X64OP_XOR ; break ;
2016-06-23 00:39:39 +03:00
default : out_op = X64OP_LOAD_CMP ; break ;
2016-06-07 23:24:20 +03:00
}
out_reg = X64_IMM8 ;
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code ) + 1 ;
return ;
}
2015-02-15 15:31:42 +03:00
case 0x86 :
2015-01-05 00:46:31 +03:00
{
2015-02-15 15:31:42 +03:00
if ( ! oso ) // XCHG r8/m8, r8
{
out_op = X64OP_XCHG ;
out_reg = rex & 8 ? get_modRM_reg ( code , rex ) : get_modRM_reg_lh ( code );
out_size = 1 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x87 :
{
if ( true ) // XCHG r/m, r (16, 32, 64)
{
out_op = X64OP_XCHG ;
out_reg = get_modRM_reg ( code , rex );
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x88 :
{
if ( ! lock && ! oso ) // MOV r8/m8, r8
{
out_op = X64OP_STORE ;
out_reg = rex & 8 ? get_modRM_reg ( code , rex ) : get_modRM_reg_lh ( code );
out_size = 1 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x89 :
{
if ( ! lock ) // MOV r/m, r (16, 32, 64)
{
out_op = X64OP_STORE ;
out_reg = get_modRM_reg ( code , rex );
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x8a :
{
if ( ! lock && ! oso ) // MOV r8, r8/m8
{
out_op = X64OP_LOAD ;
out_reg = rex & 8 ? get_modRM_reg ( code , rex ) : get_modRM_reg_lh ( code );
out_size = 1 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x8b :
{
if ( ! lock ) // MOV r, r/m (16, 32, 64)
{
out_op = X64OP_LOAD ;
out_reg = get_modRM_reg ( code , rex );
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0xa4 :
{
if ( ! oso && ! lock && ! repe && ! rex ) // MOVS
{
out_op = X64OP_MOVS ;
out_reg = X64_NOT_SET ;
out_size = 1 ;
return ;
}
if ( ! oso && ! lock && repe ) // REP MOVS
{
out_op = X64OP_MOVS ;
out_reg = rex & 8 ? X64R_RCX : X64R_ECX ;
out_size = 1 ;
return ;
}
break ;
}
2015-02-17 05:01:47 +03:00
case 0xaa :
{
if ( ! oso && ! lock && ! repe && ! rex ) // STOS
{
out_op = X64OP_STOS ;
out_reg = X64_NOT_SET ;
out_size = 1 ;
return ;
}
if ( ! oso && ! lock && repe ) // REP STOS
{
out_op = X64OP_STOS ;
out_reg = rex & 8 ? X64R_RCX : X64R_ECX ;
out_size = 1 ;
return ;
}
break ;
}
2016-07-07 21:42:39 +03:00
case 0xc4 : // 3-byte VEX prefix
case 0xc5 : // 2-byte VEX prefix
{
// Last prefix byte: op2 or op3
const u8 opx = op1 == 0xc5 ? op2 : op3 ;
// Implied prefixes
rex |= op2 & 0x80 ? 0 : 0x4 ; // REX.R
rex |= op1 == 0xc4 && op3 & 0x80 ? 0x8 : 0 ; // REX.W ???
oso = ( opx & 0x3 ) == 0x1 ;
repe = ( opx & 0x3 ) == 0x2 ;
repne = ( opx & 0x3 ) == 0x3 ;
const u8 vopm = op1 == 0xc5 ? 1 : op2 & 0x1f ;
const u8 vop1 = op1 == 0xc5 ? op3 : code [ 2 ];
const u8 vlen = ( opx & 0x4 ) ? 32 : 16 ;
2023-05-05 12:39:09 +03:00
const u8 vreg = ( ~ opx >> 3 ) & 0xf ;
2016-07-07 21:42:39 +03:00
out_length += op1 == 0xc5 ? 2 : 3 ;
code += op1 == 0xc5 ? 2 : 3 ;
2023-05-05 12:39:09 +03:00
s_tls_reg3 = x64_reg_t { vreg };
2016-07-07 21:42:39 +03:00
if ( vopm == 0x1 ) switch ( vop1 ) // Implied leading byte 0x0F
{
case 0x11 :
case 0x29 :
{
if ( ! repe && ! repne ) // VMOVAPS/VMOVAPD/VMOVUPS/VMOVUPD mem,reg
{
out_op = X64OP_STORE ;
out_reg = get_modRM_reg_xmm ( code , rex );
out_size = vlen ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
case 0x7f :
{
if ( repe || oso ) // VMOVDQU/VMOVDQA mem,reg
{
out_op = X64OP_STORE ;
out_reg = get_modRM_reg_xmm ( code , rex );
out_size = vlen ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
}
2023-05-05 12:39:09 +03:00
if ( vopm == 0x2 ) switch ( vop1 ) // Implied leading bytes 0x0F 0x38
{
case 0xf7 :
{
if ( ! repe && ! repne && vlen == 16 ) // BEXTR r32,mem,r32
{
out_op = X64OP_BEXTR ;
out_reg = get_modRM_reg_xmm ( code , rex );
out_size = opx & 0x80 ? 8 : 4 ;
out_length += get_modRM_size ( code );
return ;
}
break ;
}
}
2016-07-07 21:42:39 +03:00
break ;
}
2015-02-15 15:31:42 +03:00
case 0xc6 :
{
2016-06-07 23:24:20 +03:00
if ( ! lock && ! oso && get_modRM_reg ( code , 0 ) == 0 ) // MOV r8/m8, imm8
2015-02-15 15:31:42 +03:00
{
out_op = X64OP_STORE ;
out_reg = X64_IMM8 ;
out_size = 1 ;
out_length += get_modRM_size ( code ) + 1 ;
return ;
}
break ;
2015-01-05 00:46:31 +03:00
}
case 0xc7 :
{
2016-06-07 23:24:20 +03:00
if ( ! lock && get_modRM_reg ( code , 0 ) == 0 ) // MOV r/m, imm16/imm32 (16, 32, 64)
2015-01-05 00:46:31 +03:00
{
2015-02-15 15:31:42 +03:00
out_op = X64OP_STORE ;
out_reg = oso ? X64_IMM16 : X64_IMM32 ;
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code ) + ( oso ? 2 : 4 );
2015-01-05 00:46:31 +03:00
return ;
}
2015-02-15 15:31:42 +03:00
break ;
2015-01-05 00:46:31 +03:00
}
2016-06-23 00:39:39 +03:00
case 0xf6 :
{
2021-01-12 13:01:06 +03:00
switch ( get_modRM_reg ( code , 0 ))
2016-06-23 00:39:39 +03:00
{
case 0 : out_op = X64OP_LOAD_TEST ; break ;
default : out_op = X64OP_NONE ; break ; // TODO...
}
out_reg = X64_IMM8 ;
out_size = 1 ;
out_length += get_modRM_size ( code ) + 1 ;
return ;
}
case 0xf7 :
{
2021-01-12 13:01:06 +03:00
switch ( get_modRM_reg ( code , 0 ))
2016-06-23 00:39:39 +03:00
{
case 0 : out_op = X64OP_LOAD_TEST ; break ;
default : out_op = X64OP_NONE ; break ; // TODO...
}
out_reg = oso ? X64_IMM16 : X64_IMM32 ;
out_size = get_op_size ( rex , oso );
out_length += get_modRM_size ( code ) + ( oso ? 2 : 4 );
return ;
}
2015-01-05 00:46:31 +03:00
}
2015-02-15 15:31:42 +03:00
2015-02-15 20:13:06 +03:00
out_op = X64OP_NONE ;
2015-02-15 15:31:42 +03:00
out_reg = X64_NOT_SET ;
out_size = 0 ;
out_length = 0 ;
2015-01-05 00:46:31 +03:00
}
#ifdef _WIN32
2015-01-18 16:57:39 +03:00
typedef CONTEXT x64_context ;
2021-12-30 19:39:18 +03:00
typedef CONTEXT ucontext_t ;
2015-01-18 16:57:39 +03:00
2015-02-17 03:08:23 +03:00
#define X64REG(context, reg) (&(&(context)->Rax)[reg])
2015-08-06 16:31:13 +03:00
#define XMMREG(context, reg) (reinterpret_cast<v128*>(&(&(context)->Xmm0)[reg]))
2015-02-17 03:08:23 +03:00
#define EFLAGS(context) ((context)->EFlags)
2015-01-18 16:57:39 +03:00
2015-12-16 17:20:46 +03:00
#define ARG1(context) RCX(context)
#define ARG2(context) RDX(context)
2015-01-18 16:57:39 +03:00
#else
typedef ucontext_t x64_context ;
2015-02-10 23:17:39 -05:00
#ifdef __APPLE__
#define X64REG(context, reg) (darwin_x64reg(context, reg))
2015-08-06 16:31:13 +03:00
#define XMMREG(context, reg) (reinterpret_cast<v128*>(&(context)->uc_mcontext->__fs.__fpu_xmm0.__xmm_reg[reg]))
2015-02-28 22:37:31 +03:00
#define EFLAGS(context) ((context)->uc_mcontext->__ss.__rflags)
2015-02-10 23:17:39 -05:00
2020-12-18 11:43:50 +03:00
u64 * darwin_x64reg ( x64_context * context , int reg )
2015-02-10 23:17:39 -05:00
{
2015-02-13 23:24:18 +03:00
auto * state = & context -> uc_mcontext -> __ss ;
switch ( reg )
{
2015-12-06 18:45:34 +03:00
case 0 : return & state -> __rax ;
case 1 : return & state -> __rcx ;
case 2 : return & state -> __rdx ;
case 3 : return & state -> __rbx ;
case 4 : return & state -> __rsp ;
case 5 : return & state -> __rbp ;
case 6 : return & state -> __rsi ;
case 7 : return & state -> __rdi ;
case 8 : return & state -> __r8 ;
case 9 : return & state -> __r9 ;
case 10 : return & state -> __r10 ;
case 11 : return & state -> __r11 ;
case 12 : return & state -> __r12 ;
case 13 : return & state -> __r13 ;
case 14 : return & state -> __r14 ;
case 15 : return & state -> __r15 ;
case 16 : return & state -> __rip ;
default :
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid register index: %d" , reg );
2015-12-06 18:45:34 +03:00
return nullptr ;
2015-02-13 23:24:18 +03:00
}
2015-02-10 23:17:39 -05:00
}
2017-06-22 21:03:41 +03:00
#elif defined(__DragonFly__) || defined(__FreeBSD__)
#define X64REG(context, reg) (freebsd_x64reg(context, reg))
#ifdef __DragonFly__
2019-12-04 23:56:19 +03:00
# define XMMREG(context, reg) (reinterpret_cast<v128*>((reinterpret_cast<union savefpu*>(context)->uc_mcontext.mc_fpregs)->sv_xmm.sv_xmm[reg]))
2017-06-22 21:03:41 +03:00
#else
2019-12-04 23:56:19 +03:00
# define XMMREG(context, reg) (reinterpret_cast<v128*>((reinterpret_cast<struct savefpu*>(context)->uc_mcontext.mc_fpstate)->sv_xmm[reg]))
2017-06-22 21:03:41 +03:00
#endif
#define EFLAGS(context) ((context)->uc_mcontext.mc_rflags)
register_t * freebsd_x64reg ( x64_context * context , int reg )
{
auto * state = & context -> uc_mcontext ;
switch ( reg )
{
case 0 : return & state -> mc_rax ;
case 1 : return & state -> mc_rcx ;
case 2 : return & state -> mc_rdx ;
case 3 : return & state -> mc_rbx ;
case 4 : return & state -> mc_rsp ;
case 5 : return & state -> mc_rbp ;
case 6 : return & state -> mc_rsi ;
case 7 : return & state -> mc_rdi ;
case 8 : return & state -> mc_r8 ;
case 9 : return & state -> mc_r9 ;
case 10 : return & state -> mc_r10 ;
case 11 : return & state -> mc_r11 ;
case 12 : return & state -> mc_r12 ;
case 13 : return & state -> mc_r13 ;
case 14 : return & state -> mc_r14 ;
case 15 : return & state -> mc_r15 ;
case 16 : return & state -> mc_rip ;
default :
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid register index: %d" , reg );
2017-06-22 21:03:41 +03:00
return nullptr ;
}
}
#elif defined(__OpenBSD__)
#define X64REG(context, reg) (openbsd_x64reg(context, reg))
#define XMMREG(context, reg) (reinterpret_cast<v128*>((context)->sc_fpstate->fx_xmm[reg]))
#define EFLAGS(context) ((context)->sc_rflags)
long * openbsd_x64reg ( x64_context * context , int reg )
{
2019-03-19 00:02:21 +00:00
auto * state = & context ;
2017-06-22 21:03:41 +03:00
switch ( reg )
{
case 0 : return & state -> sc_rax ;
case 1 : return & state -> sc_rcx ;
case 2 : return & state -> sc_rdx ;
case 3 : return & state -> sc_rbx ;
case 4 : return & state -> sc_rsp ;
case 5 : return & state -> sc_rbp ;
case 6 : return & state -> sc_rsi ;
case 7 : return & state -> sc_rdi ;
case 8 : return & state -> sc_r8 ;
case 9 : return & state -> sc_r9 ;
case 10 : return & state -> sc_r10 ;
case 11 : return & state -> sc_r11 ;
case 12 : return & state -> sc_r12 ;
case 13 : return & state -> sc_r13 ;
case 14 : return & state -> sc_r14 ;
case 15 : return & state -> sc_r15 ;
case 16 : return & state -> sc_rip ;
default :
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid register index: %d" , reg );
2017-06-22 21:03:41 +03:00
return nullptr ;
}
}
#elif defined(__NetBSD__)
static const decltype ( _REG_RAX ) reg_table [] =
{
_REG_RAX , _REG_RCX , _REG_RDX , _REG_RBX , _REG_RSP , _REG_RBP , _REG_RSI , _REG_RDI ,
_REG_R8 , _REG_R9 , _REG_R10 , _REG_R11 , _REG_R12 , _REG_R13 , _REG_R14 , _REG_R15 , _REG_RIP
};
#define X64REG(context, reg) (&(context)->uc_mcontext.__gregs[reg_table[reg]])
#define XMM_sig(context, reg) (reinterpret_cast<v128*>(((struct fxsave64*)(context)->uc_mcontext.__fpregs)->fx_xmm[reg]))
#define EFLAGS(context) ((context)->uc_mcontext.__gregs[_REG_RFL])
2015-02-10 23:17:39 -05:00
#else
2015-02-13 23:24:18 +03:00
2018-12-29 17:12:01 -08:00
static const int reg_table [] =
2015-01-18 16:57:39 +03:00
{
REG_RAX , REG_RCX , REG_RDX , REG_RBX , REG_RSP , REG_RBP , REG_RSI , REG_RDI ,
REG_R8 , REG_R9 , REG_R10 , REG_R11 , REG_R12 , REG_R13 , REG_R14 , REG_R15 , REG_RIP
};
2015-02-17 03:08:23 +03:00
#define X64REG(context, reg) (&(context)->uc_mcontext.gregs[reg_table[reg]])
2017-06-22 21:03:41 +03:00
#ifdef __sun
#define XMMREG(context, reg) (reinterpret_cast<v128*>(&(context)->uc_mcontext.fpregs.fp_reg_set.fpchip_state.xmm[reg_table[reg]]))
#else
2015-08-06 16:31:13 +03:00
#define XMMREG(context, reg) (reinterpret_cast<v128*>(&(context)->uc_mcontext.fpregs->_xmm[reg]))
2017-06-22 21:03:41 +03:00
#endif // __sun
2015-02-17 03:40:10 +03:00
#define EFLAGS(context) ((context)->uc_mcontext.gregs[REG_EFL])
2015-02-13 23:24:18 +03:00
2015-02-10 23:17:39 -05:00
#endif // __APPLE__
2015-01-18 16:57:39 +03:00
2015-12-16 17:20:46 +03:00
#define ARG1(context) RDI(context)
#define ARG2(context) RSI(context)
2015-01-18 16:57:39 +03:00
#endif
2015-02-15 20:13:06 +03:00
#define RAX(c) (*X64REG((c), 0))
#define RCX(c) (*X64REG((c), 1))
#define RDX(c) (*X64REG((c), 2))
2015-12-16 17:20:46 +03:00
#define RSP(c) (*X64REG((c), 4))
2015-02-15 20:13:06 +03:00
#define RSI(c) (*X64REG((c), 6))
#define RDI(c) (*X64REG((c), 7))
#define RIP(c) (*X64REG((c), 16))
2020-12-18 10:39:54 +03:00
bool get_x64_reg_value ( x64_context * context , x64_reg_t reg , usz d_size , usz i_size , u64 & out_value )
2015-02-15 20:13:06 +03:00
{
// get x64 reg value (for store operations)
if ( reg - X64R_RAX < 16 )
{
// load the value from x64 register
const u64 reg_value = * X64REG ( context , reg - X64R_RAX );
switch ( d_size )
{
2019-11-28 21:18:37 +03:00
case 1 : out_value = static_cast < u8 > ( reg_value ); return true ;
case 2 : out_value = static_cast < u16 > ( reg_value ); return true ;
case 4 : out_value = static_cast < u32 > ( reg_value ); return true ;
2015-02-15 20:13:06 +03:00
case 8 : out_value = reg_value ; return true ;
}
}
else if ( reg - X64R_AL < 4 && d_size == 1 )
{
2019-11-28 21:18:37 +03:00
out_value = static_cast < u8 > ( * X64REG ( context , reg - X64R_AL ));
2015-02-16 04:53:53 +03:00
return true ;
2015-02-15 20:13:06 +03:00
}
else if ( reg - X64R_AH < 4 && d_size == 1 )
{
2019-11-28 21:18:37 +03:00
out_value = static_cast < u8 > ( * X64REG ( context , reg - X64R_AH ) >> 8 );
2015-02-16 04:53:53 +03:00
return true ;
2015-02-15 20:13:06 +03:00
}
2015-02-19 23:38:25 +03:00
else if ( reg == X64_IMM8 )
{
2015-02-20 16:58:40 +03:00
// load the immediate value (assuming it's at the end of the instruction)
2019-11-28 21:18:37 +03:00
const s8 imm_value = * reinterpret_cast < s8 *> ( RIP ( context ) + i_size - 1 );
2015-02-19 23:38:25 +03:00
switch ( d_size )
{
2019-11-28 21:18:37 +03:00
case 1 : out_value = static_cast < u8 > ( imm_value ); return true ;
case 2 : out_value = static_cast < u16 > ( imm_value ); return true ; // sign-extended
case 4 : out_value = static_cast < u32 > ( imm_value ); return true ; // sign-extended
case 8 : out_value = static_cast < u64 > ( imm_value ); return true ; // sign-extended
2015-02-19 23:38:25 +03:00
}
}
else if ( reg == X64_IMM16 )
{
2019-11-28 21:18:37 +03:00
const s16 imm_value = * reinterpret_cast < s16 *> ( RIP ( context ) + i_size - 2 );
2015-02-19 23:38:25 +03:00
switch ( d_size )
{
2019-11-28 21:18:37 +03:00
case 2 : out_value = static_cast < u16 > ( imm_value ); return true ;
2015-02-19 23:38:25 +03:00
}
}
2015-02-15 20:13:06 +03:00
else if ( reg == X64_IMM32 )
{
2019-11-28 21:18:37 +03:00
const s32 imm_value = * reinterpret_cast < s32 *> ( RIP ( context ) + i_size - 4 );
2018-02-09 15:24:46 +03:00
2015-02-15 20:13:06 +03:00
switch ( d_size )
{
2019-11-28 21:18:37 +03:00
case 4 : out_value = static_cast < u32 > ( imm_value ); return true ;
case 8 : out_value = static_cast < u64 > ( imm_value ); return true ; // sign-extended
2015-02-15 20:13:06 +03:00
}
}
else if ( reg == X64R_ECX )
{
2019-11-28 21:18:37 +03:00
out_value = static_cast < u32 > ( RCX ( context ));
2015-02-16 04:53:53 +03:00
return true ;
2015-02-15 20:13:06 +03:00
}
2016-06-07 23:24:20 +03:00
else if ( reg >= X64_BIT_O && reg <= X64_BIT_NLE )
{
const u32 _cf = EFLAGS ( context ) & 0x1 ;
const u32 _zf = EFLAGS ( context ) & 0x40 ;
const u32 _sf = EFLAGS ( context ) & 0x80 ;
const u32 _of = EFLAGS ( context ) & 0x800 ;
const u32 _pf = EFLAGS ( context ) & 0x4 ;
const u32 _l = ( _sf << 4 ) ^ _of ; // SF != OF
switch ( reg & ~ 1 )
{
case X64_BIT_O : out_value = !! _of ^ ( reg & 1 ); break ;
case X64_BIT_C : out_value = !! _cf ^ ( reg & 1 ); break ;
case X64_BIT_Z : out_value = !! _zf ^ ( reg & 1 ); break ;
case X64_BIT_BE : out_value = !! ( _cf | _zf ) ^ ( reg & 1 ); break ;
case X64_BIT_S : out_value = !! _sf ^ ( reg & 1 ); break ;
case X64_BIT_P : out_value = !! _pf ^ ( reg & 1 ); break ;
case X64_BIT_L : out_value = !! _l ^ ( reg & 1 ); break ;
case X64_BIT_LE : out_value = !! ( _l | _zf ) ^ ( reg & 1 ); break ;
}
return true ;
}
2015-02-15 20:13:06 +03:00
2020-01-31 14:20:54 +03:00
sig_log . error ( "get_x64_reg_value(): invalid arguments (reg=%d, d_size=%lld, i_size=%lld)" , + reg , d_size , i_size );
2015-02-15 20:13:06 +03:00
return false ;
}
2020-12-18 10:39:54 +03:00
bool put_x64_reg_value ( x64_context * context , x64_reg_t reg , usz d_size , u64 value )
2015-02-15 20:13:06 +03:00
{
// save x64 reg value (for load operations)
if ( reg - X64R_RAX < 16 )
{
2015-02-17 03:08:23 +03:00
// save the value into x64 register
switch ( d_size )
{
2017-11-01 23:27:57 -07:00
case 1 : * X64REG ( context , reg - X64R_RAX ) = ( value & 0xff ) | ( * X64REG ( context , reg - X64R_RAX ) & 0xffffff00 ); return true ;
case 2 : * X64REG ( context , reg - X64R_RAX ) = ( value & 0xffff ) | ( * X64REG ( context , reg - X64R_RAX ) & 0xffff0000 ); return true ;
2015-02-17 03:08:23 +03:00
case 4 : * X64REG ( context , reg - X64R_RAX ) = value & 0xffffffff ; return true ;
case 8 : * X64REG ( context , reg - X64R_RAX ) = value ; return true ;
}
2015-02-15 20:13:06 +03:00
}
2020-01-31 14:20:54 +03:00
sig_log . error ( "put_x64_reg_value(): invalid destination (reg=%d, d_size=%lld, value=0x%llx)" , + reg , d_size , value );
2015-02-15 20:13:06 +03:00
return false ;
}
2020-12-18 10:39:54 +03:00
bool set_x64_cmp_flags ( x64_context * context , usz d_size , u64 x , u64 y , bool carry = true )
2015-02-17 03:08:23 +03:00
{
switch ( d_size )
{
case 1 : break ;
case 2 : break ;
case 4 : break ;
case 8 : break ;
2020-01-31 14:20:54 +03:00
default : sig_log . error ( "set_x64_cmp_flags(): invalid d_size (%lld)" , d_size ); return false ;
2015-02-17 03:08:23 +03:00
}
const u64 sign = 1ull << ( d_size * 8 - 1 ); // sign mask
const u64 diff = x - y ;
const u64 summ = x + y ;
2016-06-07 23:24:20 +03:00
if ( carry && (( x & y ) | (( x ^ y ) & ~ summ )) & sign )
2015-02-17 03:08:23 +03:00
{
EFLAGS ( context ) |= 0x1 ; // set CF
}
2016-06-07 23:24:20 +03:00
else if ( carry )
2015-02-17 03:08:23 +03:00
{
EFLAGS ( context ) &= ~ 0x1 ; // clear CF
}
if ( x == y )
{
EFLAGS ( context ) |= 0x40 ; // set ZF
}
else
{
EFLAGS ( context ) &= ~ 0x40 ; // clear ZF
}
if ( diff & sign )
{
EFLAGS ( context ) |= 0x80 ; // set SF
}
else
{
EFLAGS ( context ) &= ~ 0x80 ; // clear SF
}
if (( x ^ summ ) & ( y ^ summ ) & sign )
{
EFLAGS ( context ) |= 0x800 ; // set OF
}
else
{
EFLAGS ( context ) &= ~ 0x800 ; // clear OF
}
2019-11-28 21:18:37 +03:00
const u8 p1 = static_cast < u8 > ( diff ) ^ ( static_cast < u8 > ( diff ) >> 4 );
2015-02-17 03:08:23 +03:00
const u8 p2 = p1 ^ ( p1 >> 2 );
const u8 p3 = p2 ^ ( p2 >> 1 );
if (( p3 & 1 ) == 0 )
{
EFLAGS ( context ) |= 0x4 ; // set PF
}
else
{
EFLAGS ( context ) &= ~ 0x4 ; // clear PF
}
if ((( x & y ) | (( x ^ y ) & ~ summ )) & 0x8 )
{
EFLAGS ( context ) |= 0x10 ; // set AF
}
else
{
EFLAGS ( context ) &= ~ 0x10 ; // clear AF
}
return true ;
}
2020-12-18 10:39:54 +03:00
usz get_x64_access_size ( x64_context * context , x64_op_t op , x64_reg_t reg , usz d_size , usz i_size )
2015-02-15 20:13:06 +03:00
{
2015-02-17 18:27:15 +03:00
if ( op == X64OP_MOVS || op == X64OP_STOS )
2015-02-15 20:13:06 +03:00
{
2015-02-17 18:27:15 +03:00
if ( EFLAGS ( context ) & 0x400 /* direction flag */ )
2015-02-15 20:13:06 +03:00
{
2017-02-17 22:35:57 +03:00
// TODO
2015-02-17 18:27:15 +03:00
return 0 ;
2015-02-15 20:13:06 +03:00
}
2015-02-17 18:27:15 +03:00
if ( reg != X64_NOT_SET ) // get "full" access size from RCX register
{
2021-02-17 22:58:10 +03:00
u64 counter = 1 ;
2015-02-17 18:27:15 +03:00
if ( ! get_x64_reg_value ( context , reg , 8 , i_size , counter ))
{
return - 1 ;
}
return d_size * counter ;
}
}
2015-02-17 03:08:23 +03:00
return d_size ;
2015-02-15 20:13:06 +03:00
}
2021-12-30 19:39:18 +03:00
#elif defined(ARCH_ARM64)
2022-04-04 12:37:10 -07:00
#if defined(__APPLE__)
// https://github.com/bombela/backward-cpp/issues/200
#define RIP(context) ((context)->uc_mcontext->__ss.__pc)
2022-04-24 10:16:51 +00:00
#elif defined(__FreeBSD__)
#define RIP(context) ((context)->uc_mcontext.mc_gpregs.gp_elr)
#elif defined(__NetBSD__)
#define RIP(context) ((context)->uc_mcontext.__gregs[_REG_PC])
#elif defined(__OpenBSD__)
#define RIP(context) ((context)->sc_elr)
2022-04-04 12:37:10 -07:00
#else
2021-12-30 19:39:18 +03:00
#define RIP(context) ((context)->uc_mcontext.pc)
2022-04-04 12:37:10 -07:00
#endif
2021-12-30 19:39:18 +03:00
#endif /* ARCH_ */
2016-01-20 15:46:58 +03:00
namespace rsx
{
extern std :: function < bool ( u32 addr , bool is_writing ) > g_access_violation_handler ;
}
2015-07-02 19:05:27 +02:00
2021-12-30 19:39:18 +03:00
bool handle_access_violation ( u32 addr , bool is_writing , ucontext_t * context ) noexcept
2015-01-18 16:57:39 +03:00
{
2017-02-17 22:35:57 +03:00
g_tls_fault_all ++ ;
2017-03-13 13:35:20 +03:00
const auto cpu = get_current_cpu_thread ();
2022-08-31 20:56:23 +03:00
if ( addr < RAW_SPU_BASE_ADDR && vm :: check_addr ( addr ) && rsx :: g_access_violation_handler )
2018-09-06 22:20:39 +01:00
{
2022-08-19 12:02:45 +03:00
bool state_changed = false ;
2020-03-09 19:18:39 +03:00
if ( cpu )
2017-03-13 13:35:20 +03:00
{
2022-08-19 12:02:45 +03:00
state_changed = vm :: temporary_unlock ( * cpu );
2018-09-06 22:20:39 +01:00
}
2018-10-11 01:17:19 +03:00
2020-03-09 19:18:39 +03:00
bool handled = rsx :: g_access_violation_handler ( addr , is_writing );
2017-03-13 13:35:20 +03:00
2022-08-19 12:02:45 +03:00
if ( state_changed && ( cpu -> state += cpu_flag :: temp , cpu -> test_stopped ()))
2021-09-17 20:18:12 +03:00
{
//
}
2018-09-06 22:20:39 +01:00
if ( handled )
{
g_tls_fault_rsx ++ ;
return true ;
}
2016-01-20 15:46:58 +03:00
}
2021-12-30 19:39:18 +03:00
#if defined(ARCH_X64)
2019-11-28 21:18:37 +03:00
const u8 * const code = reinterpret_cast < u8 *> ( RIP ( context ));
2015-02-15 20:13:06 +03:00
2015-02-15 15:31:42 +03:00
x64_op_t op ;
x64_reg_t reg ;
2020-12-18 10:39:54 +03:00
usz d_size ;
usz i_size ;
2015-02-15 15:31:42 +03:00
// decode single x64 instruction that causes memory access
2015-02-15 20:13:06 +03:00
decode_x64_reg_op ( code , op , reg , d_size , i_size );
2015-02-15 15:31:42 +03:00
2015-07-28 19:01:07 +03:00
auto report_opcode = [ = ]()
{
if ( op == X64OP_NONE )
{
2021-03-08 23:41:23 +03:00
be_t < v128 > dump ;
std :: memcpy ( & dump , code , sizeof ( dump ));
sig_log . error ( "decode_x64_reg_op(%p): unsupported opcode: %s" , code , dump );
2015-07-28 19:01:07 +03:00
}
};
2020-10-11 14:59:18 +03:00
if ( 0x1'0000'0000ull - addr < d_size )
2015-02-16 04:53:53 +03:00
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid d_size (0x%llx)" , d_size );
2015-07-28 19:01:07 +03:00
report_opcode ();
2015-02-16 04:53:53 +03:00
return false ;
}
2015-02-17 03:08:23 +03:00
// get length of data being accessed
2020-12-18 10:39:54 +03:00
usz a_size = get_x64_access_size ( context , op , reg , d_size , i_size );
2015-02-16 04:53:53 +03:00
2020-10-11 14:59:18 +03:00
if ( 0x1'0000'0000ull - addr < a_size )
2015-02-17 03:08:23 +03:00
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid a_size (0x%llx)" , a_size );
2015-07-28 19:01:07 +03:00
report_opcode ();
2015-02-17 03:08:23 +03:00
return false ;
2015-02-16 04:53:53 +03:00
}
2015-02-07 02:39:51 +03:00
// check if address is RawSPU MMIO register
2020-10-11 14:59:18 +03:00
do if ( addr - RAW_SPU_BASE_ADDR < ( 6 * RAW_SPU_OFFSET ) && ( addr % RAW_SPU_OFFSET ) >= RAW_SPU_PROB_OFFSET )
2015-01-18 16:57:39 +03:00
{
2018-10-11 01:17:19 +03:00
auto thread = idm :: get < named_thread < spu_thread >> ( spu_thread :: find_raw_spu (( addr - RAW_SPU_BASE_ADDR ) / RAW_SPU_OFFSET ));
2015-03-05 00:51:14 +03:00
2015-07-03 19:07:36 +03:00
if ( ! thread )
2015-03-05 00:51:14 +03:00
{
2020-10-11 14:59:18 +03:00
break ;
2015-03-05 00:51:14 +03:00
}
2020-10-11 14:59:18 +03:00
if ( ! a_size || ! d_size || ! i_size )
2015-02-15 15:31:42 +03:00
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid or unsupported instruction (op=%d, reg=%d, d_size=%lld, a_size=0x%llx, i_size=%lld)" , + op , + reg , d_size , a_size , i_size );
2015-07-28 19:01:07 +03:00
report_opcode ();
2015-02-15 15:31:42 +03:00
return false ;
}
2015-01-18 16:57:39 +03:00
2020-10-11 14:59:18 +03:00
if ( a_size != 4 )
{
// Might be unimplemented, such as writing MFC proxy EAL+EAH using 64-bit store
break ;
}
2015-01-18 16:57:39 +03:00
switch ( op )
{
case X64OP_LOAD :
2016-06-07 23:24:20 +03:00
case X64OP_LOAD_BE :
2016-06-23 00:39:39 +03:00
case X64OP_LOAD_CMP :
case X64OP_LOAD_TEST :
2015-01-18 16:57:39 +03:00
{
2015-02-15 20:13:06 +03:00
u32 value ;
2016-06-23 00:39:39 +03:00
if ( is_writing || ! thread -> read_reg ( addr , value ))
{
return false ;
}
if ( op != X64OP_LOAD_BE )
{
2020-12-13 16:34:45 +03:00
value = stx :: se_storage < u32 >:: swap ( value );
2016-06-23 00:39:39 +03:00
}
if ( op == X64OP_LOAD_CMP )
{
u64 rvalue ;
if ( ! get_x64_reg_value ( context , reg , d_size , i_size , rvalue ) || ! set_x64_cmp_flags ( context , d_size , value , rvalue ))
{
return false ;
}
break ;
}
if ( op == X64OP_LOAD_TEST )
{
u64 rvalue ;
if ( ! get_x64_reg_value ( context , reg , d_size , i_size , rvalue ) || ! set_x64_cmp_flags ( context , d_size , value & rvalue , 0 ))
{
return false ;
}
break ;
}
if ( ! put_x64_reg_value ( context , reg , d_size , value ))
2015-02-15 20:13:06 +03:00
{
return false ;
}
2015-01-18 16:57:39 +03:00
break ;
}
2023-05-05 12:39:09 +03:00
case X64OP_BEXTR :
{
u32 value ;
if ( is_writing || ! thread -> read_reg ( addr , value ))
{
return false ;
}
value = stx :: se_storage < u32 >:: swap ( value );
u64 ctrl ;
if ( ! get_x64_reg_value ( context , s_tls_reg3 , d_size , i_size , ctrl ))
{
return false ;
}
u8 start = ctrl & 0xff ;
u8 _len = ( ctrl & 0xff00 ) >> 8 ;
if ( _len > 32 )
_len = 32 ;
if ( start > 32 )
start = 32 ;
value = ( u64 { value } >> start ) & ~ ( u64 { umax } << _len );
if ( ! put_x64_reg_value ( context , reg , d_size , value ) || ! set_x64_cmp_flags ( context , d_size , value , 0 ))
{
return false ;
}
break ;
}
2015-01-18 16:57:39 +03:00
case X64OP_STORE :
2016-06-07 23:24:20 +03:00
case X64OP_STORE_BE :
2015-01-18 16:57:39 +03:00
{
2015-02-15 20:13:06 +03:00
u64 reg_value ;
2016-06-23 00:39:39 +03:00
if ( ! is_writing || ! get_x64_reg_value ( context , reg , d_size , i_size , reg_value ))
{
return false ;
}
2019-11-28 21:18:37 +03:00
u32 val32 = static_cast < u32 > ( reg_value );
2020-12-13 16:34:45 +03:00
if ( ! thread -> write_reg ( addr , op == X64OP_STORE ? stx :: se_storage < u32 >:: swap ( val32 ) : val32 ))
2015-02-15 20:13:06 +03:00
{
return false ;
}
2015-01-18 16:57:39 +03:00
break ;
}
2015-02-17 05:01:47 +03:00
case X64OP_MOVS : // possibly, TODO
case X64OP_STOS :
2015-02-15 15:31:42 +03:00
default :
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "Invalid or unsupported operation (op=%d, reg=%d, d_size=%lld, i_size=%lld)" , + op , + reg , d_size , i_size );
2015-07-28 19:01:07 +03:00
report_opcode ();
2015-02-15 15:31:42 +03:00
return false ;
}
2015-01-18 16:57:39 +03:00
}
2015-02-15 20:13:06 +03:00
// skip processed instruction
RIP ( context ) += i_size ;
2016-09-07 01:38:52 +03:00
g_tls_fault_spu ++ ;
2015-01-18 16:57:39 +03:00
return true ;
2020-10-11 14:59:18 +03:00
} while ( 0 );
2021-12-30 19:39:18 +03:00
#else
static_cast < void > ( context );
#endif /* ARCH_ */
2015-01-18 16:57:39 +03:00
2020-11-11 13:03:03 +02:00
if ( vm :: check_addr ( addr , is_writing ? vm :: page_writable : vm :: page_readable ))
2017-03-11 02:14:48 +03:00
{
return true ;
}
2020-02-08 03:07:23 +02:00
// Hack: allocate memory in case the emulator is stopping
const auto hack_alloc = [ & ]()
{
2020-11-21 07:56:54 +03:00
g_tls_access_violation_recovered = true ;
2020-02-08 03:07:23 +02:00
2021-10-16 12:13:29 +03:00
if ( vm :: check_addr ( addr , is_writing ? vm :: page_writable : vm :: page_readable ))
{
return true ;
}
2020-02-08 03:07:23 +02:00
const auto area = vm :: reserve_map ( vm :: any , addr & - 0x10000 , 0x10000 );
if ( ! area )
{
return false ;
}
2021-10-16 12:13:29 +03:00
if ( vm :: writer_lock mlock ; area -> flags & vm :: preallocated || vm :: check_addr ( addr , 0 ))
2020-02-08 03:07:23 +02:00
{
2021-03-23 12:33:07 +02:00
// For allocated memory with protection lower than required (such as protection::no or read-only while writing to it)
2020-02-08 03:07:23 +02:00
utils :: memory_protect ( vm :: base ( addr & - 0x1000 ), 0x1000 , utils :: protection :: rw );
return true ;
}
2021-10-16 12:13:29 +03:00
return area -> falloc ( addr & - 0x10000 , 0x10000 ) || vm :: check_addr ( addr , is_writing ? vm :: page_writable : vm :: page_readable );
2020-02-08 03:07:23 +02:00
};
2021-01-22 10:11:54 +02:00
if ( cpu && ( cpu -> id_type () == 1 || cpu -> id_type () == 2 ))
2017-02-17 22:35:57 +03:00
{
2020-05-21 12:46:49 +03:00
vm :: temporary_unlock ( * cpu );
2019-02-17 17:53:38 +02:00
u32 pf_port_id = 0 ;
2021-03-02 14:59:19 +03:00
if ( auto & pf_entries = g_fxo -> get < page_fault_notification_entries > (); true )
2017-10-08 15:37:54 -05:00
{
2019-03-12 17:45:15 +02:00
if ( auto mem = vm :: get ( vm :: any , addr ))
2017-10-08 15:37:54 -05:00
{
2021-03-02 14:59:19 +03:00
reader_lock lock ( pf_entries . mutex );
2019-03-12 17:45:15 +02:00
2021-03-02 14:59:19 +03:00
for ( const auto & entry : pf_entries . entries )
2017-10-08 15:37:54 -05:00
{
2019-03-12 17:45:15 +02:00
if ( entry . start_addr == mem -> addr )
2017-10-08 15:37:54 -05:00
{
2019-02-17 17:53:38 +02:00
pf_port_id = entry . port_id ;
break ;
2017-10-08 15:37:54 -05:00
}
}
}
}
2023-05-19 21:52:57 +03:00
if ( auto pf_port = idm :: get < lv2_obj , lv2_event_port > ( pf_port_id ); pf_port && pf_port -> queue )
2019-02-17 17:53:38 +02:00
{
// We notify the game that a page fault occurred so it can rectify it.
// Note, for data3, were the memory readable AND we got a page fault, it must be due to a write violation since reads are allowed.
u64 data1 = addr ;
2021-02-17 22:58:10 +03:00
u64 data2 = 0 ;
2019-02-17 17:53:38 +02:00
2021-05-20 07:00:22 +03:00
if ( cpu -> try_get < ppu_thread > ())
2019-02-17 17:53:38 +02:00
{
data2 = ( SYS_MEMORY_PAGE_FAULT_TYPE_PPU_THREAD << 32 ) | cpu -> id ;
}
2021-05-20 07:00:22 +03:00
else if ( auto spu = cpu -> try_get < spu_thread > ())
2019-02-17 17:53:38 +02:00
{
2021-05-20 07:00:22 +03:00
const u64 type = spu -> get_type () == spu_type :: threaded ?
2019-12-10 19:17:03 +03:00
SYS_MEMORY_PAGE_FAULT_TYPE_SPU_THREAD :
2019-11-03 01:44:02 +02:00
SYS_MEMORY_PAGE_FAULT_TYPE_RAW_SPU ;
2021-05-20 07:00:22 +03:00
data2 = ( type << 32 ) | spu -> lv2_id ;
2019-02-17 17:53:38 +02:00
}
u64 data3 ;
{
2022-04-30 16:51:52 +03:00
vm :: writer_lock rlock ;
2020-11-11 13:03:03 +02:00
if ( vm :: check_addr ( addr , is_writing ? vm :: page_writable : vm :: page_readable ))
2019-02-17 17:53:38 +02:00
{
// Memory was allocated inbetween, retry
return true ;
}
2020-11-11 13:03:03 +02:00
else if ( vm :: check_addr ( addr ))
2019-02-17 17:53:38 +02:00
{
data3 = SYS_MEMORY_PAGE_FAULT_CAUSE_READ_ONLY ; // TODO
}
else
{
data3 = SYS_MEMORY_PAGE_FAULT_CAUSE_NON_MAPPED ;
}
}
2020-05-21 19:47:47 +03:00
2019-02-17 17:53:38 +02:00
// Now, place the page fault event onto table so that other functions [sys_mmapper_free_address and pagefault recovery funcs etc]
// know that this thread is page faulted and where.
2021-03-02 14:59:19 +03:00
auto & pf_events = g_fxo -> get < page_fault_event_entries > ();
2023-05-19 21:52:57 +03:00
// De-schedule
if ( cpu -> id_type () == 1 )
2019-02-17 17:53:38 +02:00
{
2023-05-19 21:52:57 +03:00
cpu -> state -= cpu_flag :: signal ; // Cannot use check_state here and signal must be removed if exists
lv2_obj :: sleep ( * cpu );
2019-02-17 17:53:38 +02:00
}
2023-05-19 21:52:57 +03:00
auto send_event = [ & ]() -> error_code
{
lv2_obj :: notify_all_t notify_later {};
std :: lock_guard pf_lock ( pf_events . pf_mutex );
if ( auto error = pf_port -> queue -> send ( pf_port -> name ? pf_port -> name : (( u64 { process_getpid () + 0u } << 32 ) | u64 { pf_port_id }), data1 , data2 , data3 ))
{
return error ;
}
pf_events . events . emplace ( cpu , addr );
return {};
};
2020-04-02 22:20:56 +03:00
sig_log . warning ( "Page_fault %s location 0x%x because of %s memory" , is_writing ? "writing" : "reading" ,
2019-02-17 17:53:38 +02:00
addr , data3 == SYS_MEMORY_PAGE_FAULT_CAUSE_READ_ONLY ? "writing read-only" : "using unmapped" );
2020-04-02 22:20:56 +03:00
if ( cpu -> id_type () == 1 )
{
if ( const auto func = static_cast < ppu_thread *> ( cpu ) -> current_function )
{
sig_log . warning ( "Page_fault while in function %s" , func );
}
}
2023-05-19 21:52:57 +03:00
error_code sending_error = not_an_error ( CELL_EBUSY );
2019-02-17 17:53:38 +02:00
// If we fail due to being busy, wait a bit and try again.
2023-05-19 21:52:57 +03:00
for (; static_cast < u32 > ( sending_error ) == CELL_EBUSY ; thread_ctrl :: wait_for ( 1000 ))
2019-02-17 17:53:38 +02:00
{
2023-05-19 21:52:57 +03:00
sending_error = send_event ();
2022-07-07 10:35:24 +03:00
2020-05-21 19:47:47 +03:00
if ( cpu -> is_stopped ())
2019-02-17 17:53:38 +02:00
{
2020-05-21 19:47:47 +03:00
sending_error = {};
break ;
2019-02-17 17:53:38 +02:00
}
}
2020-02-08 03:07:23 +02:00
if ( sending_error )
{
2021-01-27 17:50:39 +02:00
vm_log . error ( "Unknown error 0x%x while trying to pass page fault." , + sending_error );
return false ;
2020-02-08 03:07:23 +02:00
}
2020-05-21 19:47:47 +03:00
else
2019-02-17 17:53:38 +02:00
{
2020-05-21 19:47:47 +03:00
// Wait until the thread is recovered
2021-02-13 16:50:07 +02:00
while ( auto state = cpu -> state . fetch_sub ( cpu_flag :: signal ))
2019-02-17 17:53:38 +02:00
{
2021-02-13 16:50:07 +02:00
if ( is_stopped ( state ) || state & cpu_flag :: signal )
2020-05-21 19:47:47 +03:00
{
break ;
}
2019-02-17 17:53:38 +02:00
2021-02-13 16:50:07 +02:00
thread_ctrl :: wait_on ( cpu -> state , state );
2020-05-21 19:47:47 +03:00
}
2019-02-17 17:53:38 +02:00
}
2020-02-08 03:07:23 +02:00
// Reschedule, test cpu state and try recovery if stopped
if ( cpu -> test_stopped () && ! hack_alloc ())
2019-06-06 21:27:49 -07:00
{
2021-01-27 17:50:39 +02:00
return false ;
2019-02-17 17:53:38 +02:00
}
return true ;
}
2021-01-29 08:45:58 +02:00
if ( cpu -> id_type () == 2 )
2019-03-13 12:12:25 +02:00
{
2020-11-21 07:56:54 +03:00
if ( ! g_tls_access_violation_recovered )
2020-02-08 03:07:23 +02:00
{
2021-01-15 20:28:45 +02:00
vm_log . notice ( " \n %s" , dump_useful_thread_info ());
2023-07-31 11:34:23 +03:00
vm_log . error ( "[%s] Access violation %s location 0x%x (%s)" , cpu -> get_name (), is_writing ? "writing" : "reading" , addr , ( is_writing && vm :: check_addr ( addr )) ? "read-only memory" : "unmapped memory" );
2020-02-08 03:07:23 +02:00
}
2019-03-13 12:12:25 +02:00
// TODO:
// RawSPU: Send appropriate interrupt
// SPUThread: Send sys_spu exception event
cpu -> state += cpu_flag :: dbg_pause ;
2020-02-08 03:07:23 +02:00
if ( cpu -> check_state () && ! hack_alloc ())
2019-03-13 12:12:25 +02:00
{
2021-01-27 17:50:39 +02:00
return false ;
2019-03-13 12:12:25 +02:00
}
2020-02-08 03:07:23 +02:00
return true ;
2019-03-13 12:12:25 +02:00
}
else
{
2019-04-14 21:28:23 +03:00
if ( auto last_func = static_cast < ppu_thread *> ( cpu ) -> current_function )
{
2020-02-01 11:31:27 +03:00
ppu_log . fatal ( "Function aborted: %s" , last_func );
2019-04-14 21:28:23 +03:00
}
2019-03-13 12:12:25 +02:00
lv2_obj :: sleep ( * cpu );
}
2019-01-28 14:05:37 +02:00
}
2018-10-11 01:17:19 +03:00
2023-06-24 16:56:49 +03:00
if ( cpu )
{
cpu -> state += cpu_flag :: wait ;
}
2021-09-07 19:42:08 +03:00
Emu . Pause ( true );
2019-07-09 20:44:07 +03:00
2021-01-15 20:28:45 +02:00
if ( ! g_tls_access_violation_recovered )
2019-07-09 20:44:07 +03:00
{
2021-01-15 20:28:45 +02:00
vm_log . notice ( " \n %s" , dump_useful_thread_info ());
2019-07-09 20:44:07 +03:00
}
2020-02-08 03:07:23 +02:00
// Note: a thread may access violate more than once after hack_alloc recovery
// Do not log any further access violations in this case.
2020-11-21 07:56:54 +03:00
if ( ! g_tls_access_violation_recovered )
2020-02-08 03:07:23 +02:00
{
2021-12-30 19:39:18 +03:00
vm_log . fatal ( "Access violation %s location 0x%x (%s)" , is_writing ? "writing" : ( cpu && cpu -> id_type () == 1 && cpu -> get_pc () == addr ? "executing" : "reading" ), addr , ( is_writing && vm :: check_addr ( addr )) ? "read-only memory" : "unmapped memory" );
2020-02-08 03:07:23 +02:00
}
2019-01-28 14:05:37 +02:00
while ( Emu . IsPaused ())
{
thread_ctrl :: wait ();
}
2020-02-08 03:07:23 +02:00
if ( Emu . IsStopped () && ! hack_alloc ())
2019-01-28 14:05:37 +02:00
{
2021-01-27 17:50:39 +02:00
return false ;
2017-02-17 22:35:57 +03:00
}
return true ;
2015-01-18 16:57:39 +03:00
}
2021-03-01 13:58:13 +02:00
static void append_thread_name ( std :: string & msg )
{
if ( thread_ctrl :: get_current ())
{
fmt :: append ( msg , "Emu Thread Name: '%s'. \n " , thread_ctrl :: get_name ());
}
else if ( thread_ctrl :: is_main ())
{
2021-03-15 17:38:07 +02:00
fmt :: append ( msg , "Thread: Main Thread. \n " );
2021-03-01 13:58:13 +02:00
}
else
{
2022-07-09 14:42:03 +03:00
fmt :: append ( msg , "Thread id = %u. \n " , thread_ctrl :: get_tid ());
2021-03-01 13:58:13 +02:00
}
}
2015-12-16 17:20:46 +03:00
#ifdef _WIN32
2020-02-08 18:35:13 +02:00
static LONG exception_handler ( PEXCEPTION_POINTERS pExp ) noexcept
2015-12-16 17:20:46 +03:00
{
2022-03-25 20:51:55 +03:00
if ( pExp -> ExceptionRecord -> ExceptionCode == EXCEPTION_BREAKPOINT )
2020-02-29 08:14:01 +02:00
{
return EXCEPTION_CONTINUE_SEARCH ;
}
2020-12-26 12:04:45 +02:00
const auto ptr = reinterpret_cast < u8 *> ( pExp -> ExceptionRecord -> ExceptionInformation [ 1 ]);
2021-01-23 18:13:30 +03:00
const bool is_writing = pExp -> ExceptionRecord -> ExceptionInformation [ 0 ] == 1 ;
const bool is_executing = pExp -> ExceptionRecord -> ExceptionInformation [ 0 ] == 8 ;
2015-02-07 02:39:51 +03:00
2021-01-23 18:13:30 +03:00
if ( pExp -> ExceptionRecord -> ExceptionCode == EXCEPTION_ACCESS_VIOLATION && ! is_executing )
2015-02-07 02:39:51 +03:00
{
2020-12-26 12:04:45 +02:00
u32 addr = 0 ;
2016-02-02 00:55:43 +03:00
2020-12-26 12:04:45 +02:00
if ( auto [ addr0 , ok ] = vm :: try_get_addr ( ptr ); ok )
{
addr = addr0 ;
}
2021-05-22 10:35:15 +03:00
else if ( const usz exec64 = ( ptr - vm :: g_exec_addr ) / 2 ; exec64 <= u32 { umax })
2020-12-26 12:04:45 +02:00
{
addr = static_cast < u32 > ( exec64 );
}
else
{
return EXCEPTION_CONTINUE_SEARCH ;
}
if ( thread_ctrl :: get_current () && handle_access_violation ( addr , is_writing , pExp -> ContextRecord ))
2017-03-25 18:53:45 +03:00
{
return EXCEPTION_CONTINUE_EXECUTION ;
}
}
2015-02-07 02:39:51 +03:00
2023-09-04 19:30:23 +03:00
switch ( pExp -> ExceptionRecord -> ExceptionCode )
{
case EXCEPTION_ACCESS_VIOLATION :
case EXCEPTION_ARRAY_BOUNDS_EXCEEDED :
case EXCEPTION_DATATYPE_MISALIGNMENT :
case EXCEPTION_ILLEGAL_INSTRUCTION :
case EXCEPTION_IN_PAGE_ERROR :
case EXCEPTION_INT_DIVIDE_BY_ZERO :
case EXCEPTION_NONCONTINUABLE_EXCEPTION :
case EXCEPTION_PRIV_INSTRUCTION :
//case EXCEPTION_STACK_OVERFLOW:
{
sys_log . notice ( " \n %s" , dump_useful_thread_info ());
logs :: listener :: sync_all ();
break ;
}
default :
{
break ;
}
}
2020-02-29 18:36:51 +03:00
return EXCEPTION_CONTINUE_SEARCH ;
}
static LONG exception_filter ( PEXCEPTION_POINTERS pExp ) noexcept
{
2015-12-18 14:11:18 +03:00
std :: string msg = fmt :: format ( "Unhandled Win32 exception 0x%08X. \n " , pExp -> ExceptionRecord -> ExceptionCode );
2015-07-01 01:25:52 +03:00
2015-12-16 17:20:46 +03:00
if ( pExp -> ExceptionRecord -> ExceptionCode == EXCEPTION_ACCESS_VIOLATION )
{
2021-01-23 18:13:30 +03:00
const auto cause =
pExp -> ExceptionRecord -> ExceptionInformation [ 0 ] == 8 ? "executing" :
pExp -> ExceptionRecord -> ExceptionInformation [ 0 ] == 1 ? "writing" : "reading" ;
2015-12-16 17:20:46 +03:00
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "Segfault %s location %p at %p. \n " , cause , pExp -> ExceptionRecord -> ExceptionInformation [ 1 ], pExp -> ExceptionRecord -> ExceptionAddress );
2015-12-16 17:20:46 +03:00
}
else
{
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "Exception address: %p. \n " , pExp -> ExceptionRecord -> ExceptionAddress );
2015-12-16 17:20:46 +03:00
for ( DWORD i = 0 ; i < pExp -> ExceptionRecord -> NumberParameters ; i ++ )
{
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "ExceptionInformation[0x%x]: %p. \n " , i , pExp -> ExceptionRecord -> ExceptionInformation [ i ]);
2015-12-16 17:20:46 +03:00
}
}
2021-03-01 13:58:13 +02:00
append_thread_name ( msg );
2020-02-22 12:58:42 +02:00
2016-08-06 15:09:44 +03:00
std :: vector < HMODULE > modules ;
2016-08-10 14:17:07 +03:00
for ( DWORD size = 256 ; modules . size () != size ; size /= sizeof ( HMODULE ))
2016-08-06 15:09:44 +03:00
{
modules . resize ( size );
if ( ! EnumProcessModules ( GetCurrentProcess (), modules . data (), size * sizeof ( HMODULE ), & size ))
{
modules . clear ();
break ;
}
}
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "Instruction address: %p. \n " , pExp -> ContextRecord -> Rip );
2016-08-06 15:09:44 +03:00
DWORD64 unwind_base ;
if ( const auto rtf = RtlLookupFunctionEntry ( pExp -> ContextRecord -> Rip , & unwind_base , nullptr ))
{
// Get function address
const DWORD64 func_addr = rtf -> BeginAddress + unwind_base ;
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "Function address: %p (base+0x%x). \n " , func_addr , rtf -> BeginAddress );
2016-08-06 15:09:44 +03:00
// Access UNWIND_INFO structure
//const auto uw = (u8*)(unwind_base + rtf->UnwindData);
}
2020-05-06 18:18:30 +03:00
for ( HMODULE _module : modules )
2016-08-06 15:09:44 +03:00
{
MODULEINFO info ;
2020-05-06 18:18:30 +03:00
if ( GetModuleInformation ( GetCurrentProcess (), _module , & info , sizeof ( info )))
2016-08-06 15:09:44 +03:00
{
2019-11-28 21:18:37 +03:00
const DWORD64 base = reinterpret_cast < DWORD64 > ( info . lpBaseOfDll );
2016-08-06 15:09:44 +03:00
if ( pExp -> ContextRecord -> Rip >= base && pExp -> ContextRecord -> Rip < base + info . SizeOfImage )
{
std :: string module_name ;
2016-08-10 14:17:07 +03:00
for ( DWORD size = 15 ; module_name . size () != size ;)
2016-08-06 15:09:44 +03:00
{
module_name . resize ( size );
2020-05-06 18:18:30 +03:00
size = GetModuleBaseNameA ( GetCurrentProcess (), _module , & module_name . front (), size + 1 );
2016-08-06 15:09:44 +03:00
if ( ! size )
{
2016-08-10 14:17:07 +03:00
module_name . clear ();
2016-08-06 15:09:44 +03:00
break ;
}
}
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "Module name: '%s'. \n " , module_name );
fmt :: append ( msg , "Module base: %p. \n " , info . lpBaseOfDll );
2016-08-06 15:09:44 +03:00
}
}
}
2020-02-22 12:58:42 +02:00
fmt :: append ( msg , "RPCS3 image base: %p. \n " , GetModuleHandle ( NULL ));
2016-02-02 00:55:43 +03:00
2016-01-06 02:52:48 +03:00
// TODO: print registers and the callstack
2023-09-04 19:30:23 +03:00
sys_log . fatal ( " \n %s" , msg );
logs :: listener :: sync_all ();
2021-01-27 17:50:39 +02:00
thread_ctrl :: emergency_exit ( msg );
2016-01-06 02:52:48 +03:00
}
2016-02-02 00:55:43 +03:00
const bool s_exception_handler_set = []() -> bool
2016-01-06 02:52:48 +03:00
{
2023-04-12 23:47:14 +02:00
#ifdef USE_ASAN
2023-07-11 20:40:30 +02:00
if ( ! AddVectoredExceptionHandler ( FALSE , static_cast < PVECTORED_EXCEPTION_HANDLER > ( exception_handler )))
2023-04-12 23:47:14 +02:00
#else
2023-07-11 20:40:30 +02:00
if ( ! AddVectoredExceptionHandler ( 1 , static_cast < PVECTORED_EXCEPTION_HANDLER > ( exception_handler )))
2023-04-12 23:47:14 +02:00
#endif
2016-01-06 02:52:48 +03:00
{
report_fatal_error ( "AddVectoredExceptionHandler() failed." );
}
2023-07-11 20:40:30 +02:00
if ( ! SetUnhandledExceptionFilter ( static_cast < LPTOP_LEVEL_EXCEPTION_FILTER > ( exception_filter )))
2020-02-29 18:36:51 +03:00
{
report_fatal_error ( "SetUnhandledExceptionFilter() failed." );
}
2016-01-06 02:52:48 +03:00
return true ;
}();
2015-07-01 01:25:52 +03:00
2015-01-05 00:46:31 +03:00
#else
2021-03-05 22:05:37 +03:00
static void signal_handler ( int /*sig*/ , siginfo_t * info , void * uct ) noexcept
2015-01-05 00:46:31 +03:00
{
2021-12-30 19:39:18 +03:00
ucontext_t * context = static_cast < ucontext_t *> ( uct );
2015-02-10 23:17:39 -05:00
2021-12-30 19:39:18 +03:00
#if defined(ARCH_X64)
2015-02-10 23:17:39 -05:00
#ifdef __APPLE__
2021-01-23 18:13:30 +03:00
const u64 err = context -> uc_mcontext -> __es . __err ;
2017-06-22 21:03:41 +03:00
#elif defined(__DragonFly__) || defined(__FreeBSD__)
2021-01-23 18:13:30 +03:00
const u64 err = context -> uc_mcontext . mc_err ;
2017-06-22 21:03:41 +03:00
#elif defined(__OpenBSD__)
2021-01-23 18:13:30 +03:00
const u64 err = context -> sc_err ;
2017-06-22 21:03:41 +03:00
#elif defined(__NetBSD__)
2021-01-23 18:13:30 +03:00
const u64 err = context -> uc_mcontext . __gregs [ _REG_ERR ];
2015-02-10 23:17:39 -05:00
#else
2021-01-23 18:13:30 +03:00
const u64 err = context -> uc_mcontext . gregs [ REG_ERR ];
2015-02-10 23:17:39 -05:00
#endif
2015-02-08 16:38:08 +03:00
2021-01-23 18:13:30 +03:00
const bool is_executing = err & 0x10 ;
const bool is_writing = err & 0x2 ;
2021-12-30 19:39:18 +03:00
#elif defined(ARCH_ARM64)
2022-11-02 21:39:57 +00:00
const bool is_executing = uptr ( info -> si_addr ) == uptr ( RIP ( context ));
2021-12-30 19:39:18 +03:00
const u32 insn = is_executing ? 0 : * reinterpret_cast < u32 *> ( RIP ( context ));
const bool is_writing = ( insn & 0xbfff0000 ) == 0x0c000000
|| ( insn & 0xbfe00000 ) == 0x0c800000
|| ( insn & 0xbfdf0000 ) == 0x0d000000
|| ( insn & 0xbfc00000 ) == 0x0d800000
|| ( insn & 0x3f400000 ) == 0x08000000
|| ( insn & 0x3bc00000 ) == 0x39000000
|| ( insn & 0x3fc00000 ) == 0x3d800000
|| ( insn & 0x3bc00000 ) == 0x38000000
|| ( insn & 0x3fe00000 ) == 0x3c800000
|| ( insn & 0x3a400000 ) == 0x28000000 ;
#else
#error "signal_handler not implemented"
#endif
2015-12-16 17:20:46 +03:00
2021-01-23 18:13:30 +03:00
const u64 exec64 = ( reinterpret_cast < u64 > ( info -> si_addr ) - reinterpret_cast < u64 > ( vm :: g_exec_addr )) / 2 ;
const auto cause = is_executing ? "executing" : is_writing ? "writing" : "reading" ;
if ( auto [ addr , ok ] = vm :: try_get_addr ( info -> si_addr ); ok && ! is_executing )
2015-01-05 00:46:31 +03:00
{
2015-12-16 17:20:46 +03:00
// Try to process access violation
2020-12-26 12:04:45 +02:00
if ( thread_ctrl :: get_current () && handle_access_violation ( addr , is_writing , context ))
2017-03-25 18:53:45 +03:00
{
return ;
}
}
2021-01-23 18:13:30 +03:00
if ( exec64 < 0x100000000ull && ! is_executing )
2017-03-25 18:53:45 +03:00
{
2019-11-28 21:18:37 +03:00
if ( thread_ctrl :: get_current () && handle_access_violation ( static_cast < u32 > ( exec64 ), is_writing , context ))
2015-01-05 00:46:31 +03:00
{
2017-02-17 22:35:57 +03:00
return ;
2015-01-05 00:46:31 +03:00
}
}
2017-02-17 22:35:57 +03:00
2020-02-29 08:14:01 +02:00
std :: string msg = fmt :: format ( "Segfault %s location %p at %p. \n " , cause , info -> si_addr , RIP ( context ));
2020-02-22 12:58:42 +02:00
2021-03-01 13:58:13 +02:00
append_thread_name ( msg );
2020-02-22 12:58:42 +02:00
2023-09-04 19:30:23 +03:00
sys_log . fatal ( " \n %s" , msg );
sys_log . notice ( " \n %s" , dump_useful_thread_info ());
logs :: listener :: sync_all ();
2020-03-10 14:06:06 +03:00
if ( IsDebuggerPresent ())
{
2020-11-26 08:33:50 +03:00
// Convert to SIGTRAP
raise ( SIGTRAP );
return ;
2020-03-10 14:06:06 +03:00
}
2021-01-27 17:50:39 +02:00
thread_ctrl :: emergency_exit ( msg );
2015-01-05 00:46:31 +03:00
}
2021-12-30 19:39:18 +03:00
static void sigill_handler ( int /*sig*/ , siginfo_t * info , void * /*uct*/ ) noexcept
{
std :: string msg = fmt :: format ( "Illegal instruction at %p (%s). \n " , info -> si_addr , * reinterpret_cast < be_t < u128 >*> ( info -> si_addr ));
append_thread_name ( msg );
2023-09-04 19:30:23 +03:00
sys_log . fatal ( " \n %s" , msg );
sys_log . notice ( " \n %s" , dump_useful_thread_info ());
logs :: listener :: sync_all ();
2021-12-30 19:39:18 +03:00
if ( IsDebuggerPresent ())
{
// Convert to SIGTRAP
raise ( SIGTRAP );
return ;
}
thread_ctrl :: emergency_exit ( msg );
}
2020-11-11 22:59:30 +01:00
void sigpipe_signaling_handler ( int )
{
}
2016-02-02 00:55:43 +03:00
const bool s_exception_handler_set = []() -> bool
2015-01-05 00:46:31 +03:00
{
2016-01-06 02:52:48 +03:00
struct :: sigaction sa ;
2015-01-05 00:46:31 +03:00
sa . sa_flags = SA_SIGINFO ;
sigemptyset ( & sa . sa_mask );
sa . sa_sigaction = signal_handler ;
2015-12-16 17:20:46 +03:00
2016-01-06 02:52:48 +03:00
if ( :: sigaction ( SIGSEGV , & sa , NULL ) == - 1 )
{
2020-03-09 19:18:39 +03:00
std :: fprintf ( stderr , "sigaction(SIGSEGV) failed (%d). \n " , errno );
2016-07-16 20:58:42 +03:00
std :: abort ();
}
2021-12-12 21:35:56 +01:00
#ifdef __APPLE__
if ( :: sigaction ( SIGBUS , & sa , NULL ) == - 1 )
{
std :: fprintf ( stderr , "sigaction(SIGBUS) failed (%d). \n " , errno );
std :: abort ();
}
#endif
2021-12-30 19:39:18 +03:00
sa . sa_sigaction = sigill_handler ;
if ( :: sigaction ( SIGILL , & sa , NULL ) == - 1 )
{
std :: fprintf ( stderr , "sigaction(SIGILL) failed (%d). \n " , errno );
std :: abort ();
}
2020-11-11 22:59:30 +01:00
sa . sa_handler = sigpipe_signaling_handler ;
if ( :: sigaction ( SIGPIPE , & sa , NULL ) == - 1 )
{
std :: fprintf ( stderr , "sigaction(SIGPIPE) failed (%d). \n " , errno );
std :: abort ();
}
2020-03-10 11:31:11 +03:00
std :: printf ( "Debugger: %d \n " , + IsDebuggerPresent ());
2016-01-06 02:52:48 +03:00
return true ;
}();
2015-01-05 00:46:31 +03:00
#endif
2020-03-09 19:18:39 +03:00
const bool s_terminate_handler_set = []() -> bool
{
std :: set_terminate ([]()
{
if ( IsDebuggerPresent ())
2022-11-11 18:35:52 +02:00
{
logs :: listener :: sync_all ();
2021-12-30 19:39:18 +03:00
utils :: trap ();
2022-11-11 18:35:52 +02:00
}
2020-03-09 19:18:39 +03:00
report_fatal_error ( "RPCS3 has abnormally terminated." );
});
return true ;
}();
2016-09-07 01:38:52 +03:00
thread_local DECLARE ( thread_ctrl :: g_tls_this_thread ) = nullptr ;
2020-03-20 20:18:08 +03:00
thread_local DECLARE ( thread_ctrl :: g_tls_error_callback ) = nullptr ;
2017-10-21 14:21:37 +03:00
DECLARE ( thread_ctrl :: g_native_core_layout ) { native_core_arrangement :: undefined };
2020-11-23 11:32:34 +03:00
void thread_base :: start ()
2018-09-25 15:21:04 +03:00
{
2023-12-14 22:20:21 +02:00
m_sync . atomic_op ([ & ]( u32 & v )
{
v &= ~ static_cast < u32 > ( thread_state :: mask );
v |= static_cast < u32 > ( thread_state :: created );
});
2018-09-25 15:21:04 +03:00
#ifdef _WIN32
2020-11-23 11:32:34 +03:00
m_thread = :: _beginthreadex ( nullptr , 0 , entry_point , this , CREATE_SUSPENDED , nullptr );
2020-12-09 10:47:45 +03:00
ensure ( m_thread );
2023-07-11 20:40:30 +02:00
ensure ( :: ResumeThread ( reinterpret_cast < HANDLE > ( + m_thread )) != static_cast < DWORD > ( - 1 ));
2022-01-10 14:14:14 +01:00
#elif defined(__APPLE__)
pthread_attr_t stack_size_attr ;
pthread_attr_init ( & stack_size_attr );
2022-01-11 04:04:58 +01:00
pthread_attr_setstacksize ( & stack_size_attr , 0x800000 );
2022-01-10 14:14:14 +01:00
ensure ( pthread_create ( reinterpret_cast < pthread_t *> ( & m_thread . raw ()), & stack_size_attr , entry_point , this ) == 0 );
2018-09-25 15:21:04 +03:00
#else
2020-12-09 10:47:45 +03:00
ensure ( pthread_create ( reinterpret_cast < pthread_t *> ( & m_thread . raw ()), nullptr , entry_point , this ) == 0 );
2018-09-25 15:21:04 +03:00
#endif
}
2020-11-13 19:06:44 +03:00
void thread_base :: initialize ( void ( * error_cb )())
2015-07-01 01:25:52 +03:00
{
2020-11-21 07:56:54 +03:00
#ifndef _WIN32
m_thread . release ( reinterpret_cast < u64 > ( pthread_self ()));
#endif
2020-03-20 20:18:08 +03:00
// Initialize TLS variables
2018-09-25 15:21:04 +03:00
thread_ctrl :: g_tls_this_thread = this ;
2016-04-25 13:49:12 +03:00
2020-03-20 20:18:08 +03:00
thread_ctrl :: g_tls_error_callback = error_cb ;
2016-04-27 01:27:24 +03:00
g_tls_log_prefix = []
{
2020-02-28 10:43:37 +03:00
return thread_ctrl :: get_name_cached ();
2016-04-27 01:27:24 +03:00
};
2020-11-13 19:06:44 +03:00
atomic_wait_engine :: set_wait_callback ([]( const void * , u64 attempts , u64 stamp0 ) -> bool
{
if ( attempts == umax )
{
2021-12-30 19:39:18 +03:00
g_tls_wait_time += utils :: get_tsc () - stamp0 ;
2020-11-13 19:06:44 +03:00
}
else if ( attempts > 1 )
{
g_tls_wait_fail += attempts - 1 ;
}
return true ;
});
2020-11-12 23:24:35 +03:00
set_name ( thread_ctrl :: get_name_cached ());
}
2020-02-28 10:43:37 +03:00
2020-11-12 23:24:35 +03:00
void thread_base :: set_name ( std :: string name )
{
2023-05-24 14:58:19 +03:00
#ifdef _WIN32
if ( SetThreadDescriptionImport )
{
SetThreadDescriptionImport ( GetCurrentThread (), utf8_to_wchar ( name ). c_str ());
}
#endif
2016-09-07 01:38:52 +03:00
#ifdef _MSC_VER
2016-04-25 13:49:12 +03:00
struct THREADNAME_INFO
{
DWORD dwType ;
LPCSTR szName ;
DWORD dwThreadID ;
DWORD dwFlags ;
};
// Set thread name for VS debugger
2020-02-28 10:43:37 +03:00
if ( IsDebuggerPresent ()) [ & ]() NEVER_INLINE
2016-04-25 13:49:12 +03:00
{
THREADNAME_INFO info ;
info . dwType = 0x1000 ;
2020-02-28 10:43:37 +03:00
info . szName = name . c_str ();
2016-04-25 13:49:12 +03:00
info . dwThreadID = - 1 ;
info . dwFlags = 0 ;
__try
{
RaiseException ( 0x406D1388 , 0 , sizeof ( info ) / sizeof ( ULONG_PTR ), ( ULONG_PTR * ) & info );
}
__except ( EXCEPTION_EXECUTE_HANDLER )
{
}
2020-02-28 10:43:37 +03:00
}();
2016-04-25 13:49:12 +03:00
#endif
2018-03-11 19:30:44 +03:00
2018-03-12 20:13:49 +00:00
#if defined(__APPLE__)
2020-12-18 10:39:54 +03:00
name . resize ( std :: min < usz > ( 15 , name . size ()));
2020-02-28 10:43:37 +03:00
pthread_setname_np ( name . c_str ());
2018-03-12 20:13:49 +00:00
#elif defined(__DragonFly__) || defined(__FreeBSD__) || defined(__OpenBSD__)
2020-02-28 10:43:37 +03:00
pthread_set_name_np ( pthread_self (), name . c_str ());
2018-03-12 20:13:49 +00:00
#elif defined(__NetBSD__)
2020-02-28 10:43:37 +03:00
pthread_setname_np ( pthread_self (), "%s" , name . data ());
2018-03-12 20:13:49 +00:00
#elif !defined(_WIN32)
2020-12-18 10:39:54 +03:00
name . resize ( std :: min < usz > ( 15 , name . size ()));
2020-02-28 10:43:37 +03:00
pthread_setname_np ( pthread_self (), name . c_str ());
2018-03-11 19:30:44 +03:00
#endif
2015-11-26 11:06:29 +03:00
}
2020-11-13 11:32:47 +03:00
u64 thread_base :: finalize ( thread_state result_state ) noexcept
2015-11-26 11:06:29 +03:00
{
2018-09-15 14:42:14 +03:00
// Report pending errors
2022-11-26 12:01:55 +02:00
error_code :: error_report ( 0 , nullptr , nullptr , nullptr , nullptr );
2016-09-07 01:38:52 +03:00
#ifdef _WIN32
2021-08-01 09:40:42 +03:00
static thread_local ULONG64 tls_cycles {};
static thread_local u64 tls_time {};
2016-09-07 01:38:52 +03:00
ULONG64 cycles {};
QueryThreadCycleTime ( GetCurrentThread (), & cycles );
2021-08-01 09:40:42 +03:00
cycles -= tls_cycles ;
tls_cycles += cycles ;
2016-09-07 01:38:52 +03:00
FILETIME ctime , etime , ktime , utime ;
GetThreadTimes ( GetCurrentThread (), & ctime , & etime , & ktime , & utime );
2023-07-11 20:40:30 +02:00
const u64 time = (( ktime . dwLowDateTime | static_cast < u64 > ( ktime . dwHighDateTime ) << 32 ) + ( utime . dwLowDateTime | static_cast < u64 > ( utime . dwHighDateTime ) << 32 )) * 100ull - tls_time ;
2021-08-01 09:40:42 +03:00
tls_time += time ;
2020-02-23 15:41:48 +03:00
const u64 fsoft = 0 ;
const u64 fhard = 0 ;
const u64 ctxvol = 0 ;
const u64 ctxinv = 0 ;
2017-06-22 21:03:41 +03:00
#elif defined(RUSAGE_THREAD)
2021-08-01 09:40:42 +03:00
static thread_local u64 tls_time {}, tls_fsoft {}, tls_fhard {}, tls_ctxvol {}, tls_ctxinv {};
2016-09-07 01:38:52 +03:00
const u64 cycles = 0 ; // Not supported
struct :: rusage stats {};
:: getrusage ( RUSAGE_THREAD , & stats );
2021-08-01 09:40:42 +03:00
const u64 time = ( stats . ru_utime . tv_sec + stats . ru_stime . tv_sec ) * 1000000000ull + ( stats . ru_utime . tv_usec + stats . ru_stime . tv_usec ) * 1000ull - tls_time ;
tls_time += time ;
const u64 fsoft = stats . ru_minflt - tls_fsoft ;
tls_fsoft += fsoft ;
const u64 fhard = stats . ru_majflt - tls_fhard ;
tls_fhard += fhard ;
const u64 ctxvol = stats . ru_nvcsw - tls_ctxvol ;
tls_ctxvol += ctxvol ;
const u64 ctxinv = stats . ru_nivcsw - tls_ctxinv ;
tls_ctxinv += ctxinv ;
2016-09-07 01:38:52 +03:00
#else
const u64 cycles = 0 ;
const u64 time = 0 ;
2020-02-23 15:41:48 +03:00
const u64 fsoft = 0 ;
const u64 fhard = 0 ;
const u64 ctxvol = 0 ;
const u64 ctxinv = 0 ;
2016-09-07 01:38:52 +03:00
#endif
2017-03-13 13:14:34 +03:00
g_tls_log_prefix = []
{
2020-02-28 10:43:37 +03:00
return thread_ctrl :: get_name_cached ();
2017-03-13 13:14:34 +03:00
};
2020-11-13 19:06:44 +03:00
sig_log . notice ( "Thread time: %fs (%fGc); Faults: %u [rsx:%u, spu:%u]; [soft:%u hard:%u]; Switches:[vol:%u unvol:%u]; Wait:[%.3fs, spur:%u]" ,
2016-09-07 01:38:52 +03:00
time / 1000000000. ,
cycles / 1000000000. ,
2017-02-17 22:35:57 +03:00
g_tls_fault_all ,
2016-09-07 01:38:52 +03:00
g_tls_fault_rsx ,
2020-02-23 15:41:48 +03:00
g_tls_fault_spu ,
2020-11-13 19:06:44 +03:00
fsoft , fhard , ctxvol , ctxinv ,
g_tls_wait_time / ( utils :: get_tsc_freq () / 1. ),
g_tls_wait_fail );
atomic_wait_engine :: set_wait_callback ( nullptr );
2016-04-25 13:49:12 +03:00
2020-11-26 16:30:17 +03:00
// Avoid race with the destructor
const u64 _self = m_thread ;
// Set result state (errored or finalized)
2023-07-31 23:57:26 +03:00
m_sync . fetch_op ([ & ]( u32 & v )
2020-11-13 19:06:44 +03:00
{
v &= - 4 ;
v |= static_cast < u32 > ( result_state );
2021-01-12 13:01:06 +03:00
});
2018-09-25 15:21:04 +03:00
// Signal waiting threads
2023-07-31 23:57:26 +03:00
m_sync . notify_all ();
2020-02-28 20:45:27 +03:00
2020-11-26 16:30:17 +03:00
return _self ;
2018-09-25 15:21:04 +03:00
}
2020-11-23 11:32:34 +03:00
thread_base :: native_entry thread_base :: finalize ( u64 _self ) noexcept
2018-09-25 15:21:04 +03:00
{
2020-11-21 07:56:54 +03:00
g_tls_fault_all = 0 ;
g_tls_fault_rsx = 0 ;
g_tls_fault_spu = 0 ;
g_tls_wait_time = 0 ;
g_tls_wait_fail = 0 ;
g_tls_access_violation_recovered = false ;
2018-09-25 15:21:04 +03:00
g_tls_log_prefix = []() -> std :: string { return {}; };
2020-11-12 23:24:35 +03:00
2023-10-24 11:56:09 +03:00
if ( _self == umax )
2020-11-13 16:59:56 +03:00
{
2023-10-24 11:56:09 +03:00
thread_ctrl :: g_tls_this_thread = nullptr ;
2020-11-23 11:32:34 +03:00
return nullptr ;
2020-11-13 16:59:56 +03:00
}
2020-11-23 09:28:03 +03:00
#ifdef _WIN32
2023-07-31 23:43:06 +03:00
_endthreadex ( 0 );
2020-11-23 09:28:03 +03:00
#else
2023-07-31 23:43:06 +03:00
pthread_exit ( nullptr );
2020-11-23 09:28:03 +03:00
#endif
2023-07-31 23:43:06 +03:00
return nullptr ;
2020-11-13 16:59:56 +03:00
}
2020-11-21 07:56:54 +03:00
thread_base :: native_entry thread_base :: make_trampoline ( u64 ( * entry )( thread_base * _base ))
2020-11-13 16:59:56 +03:00
{
2022-01-23 15:20:07 +03:00
return build_function_asm < native_entry > ( "" , [ & ]( native_asm & c , auto & args )
2020-11-13 16:59:56 +03:00
{
using namespace asmjit ;
2021-12-30 19:39:18 +03:00
#if defined(ARCH_X64)
2020-11-21 07:56:54 +03:00
Label _ret = c . newLabel ();
c . push ( x86 :: rbp );
c . sub ( x86 :: rsp , 0x20 );
// Call entry point (TODO: support for detached threads missing?)
2022-01-17 19:24:53 +03:00
c . call ( entry );
2020-11-21 07:56:54 +03:00
// Call finalize, return if zero
c . mov ( args [ 0 ], x86 :: rax );
2022-01-17 19:24:53 +03:00
c . call ( static_cast < native_entry ( * )( u64 ) > ( & finalize ));
2020-11-21 07:56:54 +03:00
c . test ( x86 :: rax , x86 :: rax );
c . jz ( _ret );
// Otherwise, call it as an entry point with first arg = new current thread
c . mov ( x86 :: rbp , x86 :: rax );
2022-01-17 19:24:53 +03:00
c . call ( thread_ctrl :: get_current );
2020-11-21 07:56:54 +03:00
c . mov ( args [ 0 ], x86 :: rax );
c . add ( x86 :: rsp , 0x28 );
c . jmp ( x86 :: rbp );
c . bind ( _ret );
c . add ( x86 :: rsp , 0x28 );
c . ret ();
2021-12-30 19:39:18 +03:00
#endif
2020-11-13 16:59:56 +03:00
});
2016-04-25 13:49:12 +03:00
}
2021-02-28 14:39:04 +03:00
thread_state thread_ctrl :: state ()
{
auto _this = g_tls_this_thread ;
2021-03-01 18:02:25 +03:00
// Guard for recursive calls (TODO: may be more effective to reuse one of m_sync bits)
static thread_local bool s_tls_exec = false ;
2021-02-28 14:39:04 +03:00
// Drain execution queue
2021-03-01 18:02:25 +03:00
if ( ! s_tls_exec )
{
s_tls_exec = true ;
_this -> exec ();
s_tls_exec = false ;
}
2021-02-28 14:39:04 +03:00
return static_cast < thread_state > ( _this -> m_sync & 3 );
}
2020-06-23 16:41:16 +03:00
void thread_ctrl :: wait_for ( u64 usec , [[maybe_unused]] bool alert /* true */ )
2016-09-07 01:38:52 +03:00
{
2023-07-31 23:57:26 +03:00
if ( ! usec )
{
return ;
}
2016-09-07 01:38:52 +03:00
auto _this = g_tls_this_thread ;
2023-07-31 23:57:26 +03:00
if ( ! alert && usec > 50000 )
{
usec = 50000 ;
}
2019-10-09 19:03:34 +02:00
#ifdef __linux__
2020-03-03 16:36:59 +03:00
static thread_local struct linux_timer_handle_t
{
2020-06-23 16:41:16 +03:00
// Allocate timer only if needed (i.e. someone calls wait_for with alert and short period)
2020-03-03 16:36:59 +03:00
const int m_timer = timerfd_create ( CLOCK_MONOTONIC , 0 );
linux_timer_handle_t () noexcept
{
if ( m_timer == - 1 )
{
sig_log . error ( "Linux timer allocation failed, using the fallback instead." );
}
}
operator int () const
{
return m_timer ;
}
~ linux_timer_handle_t ()
{
if ( m_timer != - 1 )
{
close ( m_timer );
}
}
} fd_timer ;
2023-07-31 23:57:26 +03:00
if ( ! alert && fd_timer != - 1 )
2019-10-09 19:03:34 +02:00
{
struct itimerspec timeout ;
u64 missed ;
2023-07-31 23:57:26 +03:00
timeout . it_value . tv_nsec = usec % 1'000'000 * 1'000ull ;
timeout . it_value . tv_sec = usec / 1'000'000 ;
2019-10-09 19:03:34 +02:00
timeout . it_interval . tv_sec = 0 ;
timeout . it_interval . tv_nsec = 0 ;
2020-03-03 16:36:59 +03:00
timerfd_settime ( fd_timer , 0 , & timeout , NULL );
if ( read ( fd_timer , & missed , sizeof ( missed )) != sizeof ( missed ))
2020-01-31 14:20:54 +03:00
sig_log . error ( "timerfd: read() failed" );
2019-10-09 19:03:34 +02:00
return ;
}
#endif
2023-07-31 23:57:26 +03:00
if ( alert )
2016-09-07 01:38:52 +03:00
{
2023-07-31 23:57:26 +03:00
if ( _this -> m_sync . bit_test_reset ( 2 ) || _this -> m_taskq )
{
return ;
}
2019-09-09 11:09:30 +03:00
}
2020-03-03 17:53:27 +03:00
2020-11-13 19:06:44 +03:00
// Wait for signal and thread state abort
2021-02-28 14:39:04 +03:00
atomic_wait :: list < 2 > list {};
2023-07-31 23:57:26 +03:00
if ( alert )
{
list . set < 0 > ( _this -> m_sync , 0 );
list . set < 1 > ( utils :: bless < atomic_t < u32 >> ( & _this -> m_taskq )[ 1 ], 0 );
}
else
{
list . set < 0 > ( _this -> m_dummy , 0 );
}
2021-02-28 14:39:04 +03:00
list . wait ( atomic_wait_timeout { usec <= 0xffff'ffff'ffff'ffff / 1000 ? usec * 1000 : 0xffff'ffff'ffff'ffff });
2016-09-07 01:38:52 +03:00
}
2020-06-23 16:41:16 +03:00
void thread_ctrl :: wait_for_accurate ( u64 usec )
{
if ( ! usec )
{
return ;
}
2023-07-31 23:57:26 +03:00
if ( usec > 50000 )
{
fmt :: throw_exception ( "thread_ctrl::wait_for_accurate: unsupported amount" );
}
#ifdef __linux__
return wait_for ( usec , false );
#else
2020-06-23 16:41:16 +03:00
using namespace std :: chrono_literals ;
const auto until = std :: chrono :: steady_clock :: now () + 1u s * usec ;
while ( true )
{
// Host scheduler quantum for windows (worst case)
constexpr u64 host_min_quantum = 500 ;
2023-07-31 23:57:26 +03:00
2020-06-23 16:41:16 +03:00
if ( usec >= host_min_quantum )
{
// Wait on multiple of min quantum for large durations to avoid overloading low thread cpus
wait_for ( usec - ( usec % host_min_quantum ), false );
}
// TODO: Determine best value for yield delay
else if ( usec >= host_min_quantum / 2 )
{
std :: this_thread :: yield ();
}
else
{
busy_wait ( 100 );
}
const auto current = std :: chrono :: steady_clock :: now ();
if ( current >= until )
{
break ;
}
usec = ( until - current ). count ();
}
2023-07-31 23:57:26 +03:00
#endif
2020-06-23 16:41:16 +03:00
}
2020-02-28 10:43:37 +03:00
std :: string thread_ctrl :: get_name_cached ()
{
auto _this = thread_ctrl :: g_tls_this_thread ;
if ( ! _this )
{
return {};
}
2020-11-26 12:30:51 +03:00
static thread_local shared_ptr < std :: string > name_cache ;
2020-02-28 10:43:37 +03:00
if ( ! _this -> m_tname . is_equal ( name_cache )) [[unlikely]]
{
2020-12-06 11:13:34 +03:00
_this -> m_tname . peek_op ([ & ]( const shared_ptr < std :: string >& ptr )
{
if ( ptr != name_cache )
{
name_cache = ptr ;
}
});
2020-02-28 10:43:37 +03:00
}
return * name_cache ;
}
2021-05-01 09:34:52 +03:00
thread_base :: thread_base ( native_entry entry , std :: string name )
2020-11-23 11:32:34 +03:00
: entry_point ( entry )
2021-05-01 09:34:52 +03:00
, m_tname ( make_single_value ( std :: move ( name )))
2016-05-13 17:01:48 +03:00
{
2016-04-25 13:49:12 +03:00
}
2018-09-25 15:21:04 +03:00
thread_base ::~ thread_base ()
2016-04-25 13:49:12 +03:00
{
2021-02-28 14:39:04 +03:00
// Cleanup abandoned tasks: initialize default results and signal
this -> exec ();
2023-10-24 11:56:09 +03:00
// Cleanup
2020-11-13 11:32:47 +03:00
{
#ifdef _WIN32
2020-11-23 09:28:03 +03:00
const HANDLE handle0 = reinterpret_cast < HANDLE > ( m_thread . load ());
WaitForSingleObject ( handle0 , INFINITE );
CloseHandle ( handle0 );
2020-11-13 11:32:47 +03:00
#else
2020-11-21 07:56:54 +03:00
pthread_join ( reinterpret_cast < pthread_t > ( m_thread . load ()), nullptr );
2020-11-13 11:32:47 +03:00
#endif
}
2016-04-25 13:49:12 +03:00
}
2020-12-16 11:07:02 +03:00
bool thread_base :: join ( bool dtor ) const
2016-04-25 13:49:12 +03:00
{
2020-12-16 11:07:02 +03:00
// Check if already finished
if ( m_sync & 2 )
{
return ( m_sync & 3 ) == 3 ;
}
2020-11-13 19:06:44 +03:00
// Hacked for too sleepy threads (1ms) TODO: make sure it's unneeded and remove
2020-12-16 11:07:02 +03:00
const auto timeout = dtor && Emu . IsStopped () ? atomic_wait_timeout { 1'000'000 } : atomic_wait_timeout :: inf ;
2020-11-13 19:06:44 +03:00
2021-12-30 19:39:18 +03:00
auto stamp0 = utils :: get_tsc ();
2020-11-13 19:06:44 +03:00
for ( u64 i = 0 ; ( m_sync & 3 ) <= 1 ; i ++ )
2018-09-25 15:21:04 +03:00
{
2023-07-31 23:57:26 +03:00
m_sync . wait ( m_sync & ~ 2 , timeout );
2020-11-13 19:06:44 +03:00
if ( m_sync & 2 )
{
break ;
}
2021-03-01 00:06:48 +02:00
if ( i >= 16 && ! ( i & ( i - 1 )) && timeout != atomic_wait_timeout :: inf )
2020-11-13 19:06:44 +03:00
{
2021-12-30 19:39:18 +03:00
sig_log . error ( u8 "Thread [%s] is too sleepy. Waiting for it %.3fµs already!" , * m_tname . load (), ( utils :: get_tsc () - stamp0 ) / ( utils :: get_tsc_freq () / 1000000. ));
2020-11-13 19:06:44 +03:00
}
2016-04-25 13:49:12 +03:00
}
2020-03-20 20:18:08 +03:00
2020-11-13 19:06:44 +03:00
return ( m_sync & 3 ) == 3 ;
2018-09-25 15:21:04 +03:00
}
2016-04-25 13:49:12 +03:00
2018-09-25 15:21:04 +03:00
void thread_base :: notify ()
2016-04-25 13:49:12 +03:00
{
2020-11-13 19:06:44 +03:00
// Set notification
m_sync |= 4 ;
2023-07-31 23:57:26 +03:00
m_sync . notify_all ();
2020-11-13 19:06:44 +03:00
}
u64 thread_base :: get_native_id () const
{
#ifdef _WIN32
return GetThreadId ( reinterpret_cast < HANDLE > ( m_thread . load ()));
#else
return m_thread . load ();
#endif
2016-04-25 13:49:12 +03:00
}
2018-09-25 15:21:04 +03:00
u64 thread_base :: get_cycles ()
2018-05-24 22:22:07 +03:00
{
2020-11-21 07:56:54 +03:00
u64 cycles = 0 ;
const u64 handle = m_thread ;
2018-05-24 22:22:07 +03:00
#ifdef _WIN32
2020-11-21 07:56:54 +03:00
if ( QueryThreadCycleTime ( reinterpret_cast < HANDLE > ( handle ), & cycles ))
2018-05-24 22:22:07 +03:00
{
2018-11-10 17:43:38 +03:00
#elif __APPLE__
2020-11-21 07:56:54 +03:00
mach_port_name_t port = pthread_mach_thread_np ( reinterpret_cast < pthread_t > ( handle ));
2018-11-10 17:43:38 +03:00
mach_msg_type_number_t count = THREAD_BASIC_INFO_COUNT ;
thread_basic_info_data_t info ;
2019-12-04 23:56:19 +03:00
kern_return_t ret = thread_info ( port , THREAD_BASIC_INFO , reinterpret_cast < thread_info_t > ( & info ), & count );
2018-11-10 17:43:38 +03:00
if ( ret == KERN_SUCCESS )
{
cycles = static_cast < u64 > ( info . user_time . seconds + info . system_time . seconds ) * 1'000'000'000 +
static_cast < u64 > ( info . user_time . microseconds + info . system_time . microseconds ) * 1'000 ;
2018-05-24 22:22:07 +03:00
#else
2018-10-11 13:57:43 +03:00
clockid_t _clock ;
2018-05-24 22:22:07 +03:00
struct timespec thread_time ;
2020-11-21 07:56:54 +03:00
if ( ! pthread_getcpuclockid ( reinterpret_cast < pthread_t > ( handle ), & _clock ) && ! clock_gettime ( _clock , & thread_time ))
2018-05-24 22:22:07 +03:00
{
cycles = static_cast < u64 > ( thread_time . tv_sec ) * 1'000'000'000 + thread_time . tv_nsec ;
#endif
2023-07-31 23:57:26 +03:00
if ( const u64 old_cycles = m_cycles . exchange ( cycles ))
2018-05-24 22:22:07 +03:00
{
2018-10-11 01:17:19 +03:00
return cycles - old_cycles ;
2018-05-24 22:22:07 +03:00
}
2018-10-11 01:17:19 +03:00
// Report 0 the first time this function is called
return 0 ;
2018-05-24 22:22:07 +03:00
}
else
{
2023-07-31 23:57:26 +03:00
return m_cycles ;
2018-05-24 22:22:07 +03:00
}
}
2021-02-28 14:39:04 +03:00
void thread_base :: push ( shared_ptr < thread_future > task )
{
const auto next = & task -> next ;
m_taskq . push_head ( * next , std :: move ( task ));
m_taskq . notify_one ();
}
void thread_base :: exec ()
{
if ( ! m_taskq ) [[likely]]
{
return ;
}
while ( shared_ptr < thread_future > head = m_taskq . exchange ( null_ptr ))
{
// TODO: check if adapting reverse algorithm is feasible here
2023-04-26 22:39:26 +03:00
thread_future * prev_head { head . get ()};
2021-02-28 14:39:04 +03:00
2023-04-26 22:39:26 +03:00
for ( thread_future * prev {};;)
2021-02-28 14:39:04 +03:00
{
2023-04-26 22:39:26 +03:00
utils :: prefetch_exec ( prev_head -> exec . load ());
2021-02-28 14:39:04 +03:00
2023-04-26 22:39:26 +03:00
if ( auto next = prev_head -> next . get ())
2021-02-28 14:39:04 +03:00
{
2023-04-26 22:39:26 +03:00
prev = std :: exchange ( prev_head , next );
prev_head -> prev = prev ;
}
else
{
break ;
2021-02-28 14:39:04 +03:00
}
}
2023-04-26 22:39:26 +03:00
for ( auto ptr = prev_head ; ptr ; ptr = ptr -> prev )
2021-02-28 14:39:04 +03:00
{
if ( auto task = ptr -> exec . load ()) [[likely]]
{
// Execute or discard (if aborting)
if (( m_sync & 3 ) == 0 ) [[likely]]
{
task ( this , ptr );
}
else
{
task ( nullptr , ptr );
}
// Notify waiters
2023-07-31 23:57:26 +03:00
ptr -> done . release ( 1 );
ptr -> done . notify_all ();
2021-02-28 14:39:04 +03:00
}
if ( ptr -> next )
{
// Partial cleanup
ptr -> next . reset ();
}
}
if ( ! m_taskq ) [[likely]]
{
return ;
}
}
}
2020-12-28 17:05:35 -08:00
[[noreturn]] void thread_ctrl :: emergency_exit ( std :: string_view reason )
2020-03-08 14:48:06 +03:00
{
2021-01-15 20:28:45 +02:00
if ( std :: string info = dump_useful_thread_info (); ! info . empty ())
{
2021-01-26 19:58:47 +03:00
sys_log . notice ( " \n %s" , info );
2021-01-15 20:28:45 +02:00
}
2020-03-08 14:48:06 +03:00
sig_log . fatal ( "Thread terminated due to fatal error: %s" , reason );
2020-03-10 11:31:11 +03:00
if ( IsDebuggerPresent ())
2022-11-11 18:35:52 +02:00
{
logs :: listener :: sync_all ();
2021-12-30 19:39:18 +03:00
utils :: trap ();
2022-11-11 18:35:52 +02:00
}
2020-03-10 11:31:11 +03:00
2020-03-08 14:48:06 +03:00
if ( const auto _this = g_tls_this_thread )
{
2020-03-20 20:18:08 +03:00
g_tls_error_callback ();
2020-11-13 16:59:56 +03:00
u64 _self = _this -> finalize ( thread_state :: errored );
2023-10-24 11:56:09 +03:00
if ( _self == umax )
2020-03-08 14:48:06 +03:00
{
2020-11-21 07:56:54 +03:00
// Unused, detached thread support remnant
2020-03-08 14:48:06 +03:00
delete _this ;
}
2023-10-24 11:56:09 +03:00
thread_base :: finalize ( umax );
2020-03-08 14:48:06 +03:00
#ifdef _WIN32
2020-11-21 07:56:54 +03:00
_endthreadex ( 0 );
2020-03-08 14:48:06 +03:00
#else
2020-12-28 17:08:37 -08:00
pthread_exit ( nullptr );
2020-03-08 14:48:06 +03:00
#endif
}
2021-03-01 13:58:13 +02:00
report_fatal_error ( reason );
2020-03-08 14:48:06 +03:00
}
2017-10-21 14:21:37 +03:00
void thread_ctrl :: detect_cpu_layout ()
{
if ( ! g_native_core_layout . compare_and_swap_test ( native_core_arrangement :: undefined , native_core_arrangement :: generic ))
return ;
2020-04-30 19:06:05 +03:00
const auto system_id = utils :: get_cpu_brand ();
2020-03-05 14:05:23 +03:00
if ( system_id . find ( "Ryzen" ) != umax )
2017-10-21 14:21:37 +03:00
{
g_native_core_layout . store ( native_core_arrangement :: amd_ccx );
}
2020-03-05 14:05:23 +03:00
else if ( system_id . find ( "Intel" ) != umax )
2018-12-23 20:01:09 +02:00
{
#ifdef _WIN32
const LOGICAL_PROCESSOR_RELATIONSHIP relationship = LOGICAL_PROCESSOR_RELATIONSHIP :: RelationProcessorCore ;
DWORD buffer_size = 0 ;
2018-05-29 13:19:01 +03:00
2018-12-23 20:01:09 +02:00
// If buffer size is set to 0 bytes, it will be overwritten with the required size
if ( GetLogicalProcessorInformationEx ( relationship , nullptr , & buffer_size ))
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "GetLogicalProcessorInformationEx returned 0 bytes" );
2018-12-23 20:01:09 +02:00
return ;
}
DWORD error_code = GetLastError ();
if ( error_code != ERROR_INSUFFICIENT_BUFFER )
{
2020-01-31 14:20:54 +03:00
sig_log . error ( "Unexpected windows error code when detecting CPU layout: %u" , error_code );
2018-12-23 20:01:09 +02:00
return ;
}
std :: vector < u8 > buffer ( buffer_size );
if ( ! GetLogicalProcessorInformationEx ( relationship ,
reinterpret_cast < SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX *> ( buffer . data ()), & buffer_size ))
{
2023-03-04 11:05:32 +01:00
sig_log . error ( "GetLogicalProcessorInformationEx failed (size=%u, error=%s)" , buffer_size , fmt :: win_error { GetLastError (), nullptr });
2018-12-23 20:01:09 +02:00
}
else
{
// Iterate through the buffer until a core with hyperthreading is found
2020-12-12 16:29:55 +03:00
auto ptr = reinterpret_cast < uptr > ( buffer . data ());
const uptr end = ptr + buffer_size ;
2018-12-23 20:01:09 +02:00
while ( ptr < end )
{
auto info = reinterpret_cast < SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX *> ( ptr );
if ( info -> Relationship == relationship && info -> Processor . Flags == LTP_PC_SMT )
{
g_native_core_layout . store ( native_core_arrangement :: intel_ht );
break ;
}
ptr += info -> Size ;
}
}
#else
2020-01-31 14:20:54 +03:00
sig_log . todo ( "Thread scheduler is not implemented for Intel and this OS" );
2018-12-23 20:01:09 +02:00
#endif
}
2017-10-21 14:21:37 +03:00
}
2019-07-20 15:57:23 +03:00
u64 thread_ctrl :: get_affinity_mask ( thread_class group )
2017-10-21 14:21:37 +03:00
{
detect_cpu_layout ();
2021-01-28 20:33:50 +02:00
if ( const auto thread_count = utils :: get_thread_count ())
2017-10-21 14:21:37 +03:00
{
2021-01-25 21:50:14 +03:00
const u64 all_cores_mask = process_affinity_mask ;
2017-10-21 14:21:37 +03:00
switch ( g_native_core_layout )
{
default :
case native_core_arrangement :: generic :
{
return all_cores_mask ;
}
case native_core_arrangement :: amd_ccx :
{
2020-12-09 21:29:48 +03:00
if ( thread_count <= 8 )
2017-10-21 14:21:37 +03:00
{
2020-12-09 21:29:48 +03:00
// Single CCX or not enough threads, do nothing
return all_cores_mask ;
}
u64 spu_mask , ppu_mask , rsx_mask ;
spu_mask = ppu_mask = rsx_mask = all_cores_mask ; // Fallback, in case someone is messing with core config
const auto system_id = utils :: get_cpu_brand ();
const auto family_id = utils :: get_cpu_family ();
const auto model_id = utils :: get_cpu_model ();
switch ( family_id )
{
case 0x17 : // Zen, Zen+, Zen2
case 0x18 : // Dhyana core (Zen)
{
if ( model_id > 0x30 )
2019-07-10 00:47:08 +02:00
{
2020-12-09 21:29:48 +03:00
// Zen2 (models 49, 96, 113, 144)
// Much improved inter-CCX latency
switch ( thread_count )
{
case 128 :
case 64 :
case 48 :
case 32 :
// TR 3000 series, or R9 3950X, Assign threads 9-32
ppu_mask = 0b11111111000000000000000000000000 ;
spu_mask = 0b00000000111111110000000000000000 ;
rsx_mask = 0b00000000000000001111111100000000 ;
break ;
case 24 :
2021-01-02 20:24:53 +03:00
// 3900X, Assign threads 7-24
2020-12-09 21:29:48 +03:00
ppu_mask = 0b111111000000000000000000 ;
spu_mask = 0b000000111111000000000000 ;
rsx_mask = 0b000000000000111111000000 ;
break ;
case 16 :
// 3700, 3800 family, Assign threads 1-16
ppu_mask = 0b0000000011110000 ;
spu_mask = 0b1111111100000000 ;
rsx_mask = 0b0000000000001111 ;
break ;
case 12 :
// 3600 family, Assign threads 1-12
ppu_mask = 0b000000111000 ;
spu_mask = 0b111111000000 ;
rsx_mask = 0b000000000111 ;
break ;
default :
break ;
}
2019-07-10 00:47:08 +02:00
}
else
{
2020-12-09 21:29:48 +03:00
// Zen, Zen+ (models 1, 8(+), 17, 24(+), 32)
switch ( thread_count )
{
case 64 :
// TR 2990WX, Assign threads 17-32
ppu_mask = 0b00000000111111110000000000000000 ;
spu_mask = ppu_mask ;
rsx_mask = 0b11111111000000000000000000000000 ;
break ;
case 48 :
// TR 2970WX, Assign threads 9-24
ppu_mask = 0b000000111111000000000000 ;
spu_mask = ppu_mask ;
rsx_mask = 0b111111000000000000000000 ;
break ;
case 32 :
// TR 2950X, TR 1950X, Assign threads 17-32
ppu_mask = 0b00000000111111110000000000000000 ;
spu_mask = ppu_mask ;
rsx_mask = 0b11111111000000000000000000000000 ;
break ;
case 24 :
// TR 1920X, 2920X, Assign threads 13-24
ppu_mask = 0b000000111111000000000000 ;
spu_mask = ppu_mask ;
rsx_mask = 0b111111000000000000000000 ;
break ;
case 16 :
// 1700, 1800, 2700, TR 1900X family
2021-03-17 18:28:03 +02:00
if ( g_cfg . core . thread_scheduler == thread_scheduler_mode :: alt )
{
ppu_mask = 0b0010000010000000 ;
spu_mask = 0b0000101010101010 ;
rsx_mask = 0b1000000000000000 ;
}
else
{
ppu_mask = 0b1111111100000000 ;
spu_mask = ppu_mask ;
rsx_mask = 0b0000000000111100 ;
}
2020-12-09 21:29:48 +03:00
break ;
case 12 :
// 1600, 2600 family, Assign threads 3-12
ppu_mask = 0b111111000000 ;
spu_mask = ppu_mask ;
rsx_mask = 0b000000111100 ;
break ;
default :
break ;
}
2019-07-10 00:47:08 +02:00
}
2020-12-09 21:29:48 +03:00
break ;
2019-07-10 00:47:08 +02:00
}
2020-12-09 21:29:48 +03:00
case 0x19 : // Zen3
2019-07-10 00:47:08 +02:00
{
2020-12-09 21:29:48 +03:00
// Single-CCX architecture, just disable SMT if wide enough
// CCX now holds upto 16 threads
// Lack of hw availability makes testing difficult
switch ( thread_count )
2019-07-10 00:47:08 +02:00
{
2020-12-09 21:29:48 +03:00
case 24 :
// 5900X, Use same scheduler as 3900X
// Unverified on windows, may be worse than just disabling SMT and scheduler
2019-07-10 00:47:08 +02:00
ppu_mask = 0b111111000000000000000000 ;
spu_mask = 0b000000111111000000000000 ;
rsx_mask = 0b000000000000111111000000 ;
2020-12-09 21:29:48 +03:00
break ;
2021-03-17 18:28:03 +02:00
case 16 :
// 5800X
if ( g_cfg . core . thread_scheduler == thread_scheduler_mode :: alt )
{
ppu_mask = 0b0000000011110000 ;
spu_mask = 0b1111111100000000 ;
rsx_mask = 0b0000000000001111 ;
}
else
2020-12-09 21:29:48 +03:00
{
// Verified by more than one windows user on 16-thread CPU
ppu_mask = spu_mask = rsx_mask = ( 0b10101010101010101010101010101010 & all_cores_mask );
}
2021-04-09 21:12:47 +02:00
break ;
2021-03-17 18:28:03 +02:00
case 12 :
// 5600X
if ( g_cfg . core . thread_scheduler == thread_scheduler_mode :: alt )
{
ppu_mask = 0b000000001100 ;
spu_mask = 0b111111110000 ;
rsx_mask = 0b000000000011 ;
}
2020-12-09 21:29:48 +03:00
else
{
ppu_mask = spu_mask = rsx_mask = all_cores_mask ;
}
2021-04-09 21:12:47 +02:00
break ;
2021-03-17 18:28:03 +02:00
default :
if ( thread_count > 24 )
{
ppu_mask = spu_mask = rsx_mask = ( 0b10101010101010101010101010101010 & all_cores_mask );
}
2020-12-09 21:29:48 +03:00
break ;
2019-07-10 00:47:08 +02:00
}
2020-12-09 21:29:48 +03:00
break ;
2019-07-10 00:47:08 +02:00
}
2020-12-09 21:29:48 +03:00
default :
2019-07-10 00:47:08 +02:00
{
2020-12-09 21:29:48 +03:00
break ;
2018-01-19 14:32:15 +03:00
}
2017-10-21 14:21:37 +03:00
}
switch ( group )
{
default :
case thread_class :: general :
return all_cores_mask ;
case thread_class :: rsx :
2018-01-19 14:32:15 +03:00
return rsx_mask ;
2017-10-21 14:21:37 +03:00
case thread_class :: ppu :
2018-01-19 14:32:15 +03:00
return ppu_mask ;
2017-10-21 14:21:37 +03:00
case thread_class :: spu :
2018-01-19 14:32:15 +03:00
return spu_mask ;
2017-10-21 14:21:37 +03:00
}
}
case native_core_arrangement :: intel_ht :
{
2021-03-17 18:28:03 +02:00
if ( thread_count >= 12 && g_cfg . core . thread_scheduler == thread_scheduler_mode :: alt )
return ( 0b10101010101010101010101010101010 & all_cores_mask ); // Potentially improves performance by mimicking HT off
2017-10-21 14:21:37 +03:00
return all_cores_mask ;
}
}
}
2021-05-22 10:35:15 +03:00
return - 1 ;
2017-10-21 14:21:37 +03:00
}
2017-06-24 18:36:49 +03:00
void thread_ctrl :: set_native_priority ( int priority )
{
#ifdef _WIN32
HANDLE _this_thread = GetCurrentThread ();
INT native_priority = THREAD_PRIORITY_NORMAL ;
2017-11-01 01:49:54 +03:00
if ( priority > 0 )
2017-06-24 18:36:49 +03:00
native_priority = THREAD_PRIORITY_ABOVE_NORMAL ;
2017-11-01 01:49:54 +03:00
if ( priority < 0 )
2017-06-24 18:36:49 +03:00
native_priority = THREAD_PRIORITY_BELOW_NORMAL ;
2017-11-01 01:49:54 +03:00
if ( ! SetThreadPriority ( _this_thread , native_priority ))
{
2023-03-04 11:05:32 +01:00
sig_log . error ( "SetThreadPriority() failed: %s" , fmt :: win_error { GetLastError (), nullptr });
2017-11-01 01:49:54 +03:00
}
#else
int policy ;
struct sched_param param ;
pthread_getschedparam ( pthread_self (), & policy , & param );
if ( priority > 0 )
param . sched_priority = sched_get_priority_max ( policy );
if ( priority < 0 )
param . sched_priority = sched_get_priority_min ( policy );
if ( int err = pthread_setschedparam ( pthread_self (), policy , & param ))
{
2022-01-10 09:43:54 +09:00
sig_log . error ( "pthread_setschedparam() failed: %d" , err );
2017-11-01 01:49:54 +03:00
}
2017-06-24 18:36:49 +03:00
#endif
}
2020-04-29 22:11:59 +03:00
u64 thread_ctrl :: get_process_affinity_mask ()
{
static const u64 mask = []() -> u64
{
#ifdef _WIN32
DWORD_PTR res , _sys ;
if ( ! GetProcessAffinityMask ( GetCurrentProcess (), & res , & _sys ))
{
sig_log . error ( "Failed to get process affinity mask." );
return 0 ;
}
return res ;
#else
// Assume it's called from the main thread (this is a bit shaky)
return thread_ctrl :: get_thread_affinity_mask ();
#endif
}();
return mask ;
}
DECLARE ( thread_ctrl :: process_affinity_mask ) = get_process_affinity_mask ();
2019-07-20 15:57:23 +03:00
void thread_ctrl :: set_thread_affinity_mask ( u64 mask )
2017-06-24 18:36:49 +03:00
{
2021-01-26 19:59:48 +03:00
sig_log . trace ( "set_thread_affinity_mask called with mask=0x%x" , mask );
2017-06-24 18:36:49 +03:00
#ifdef _WIN32
HANDLE _this_thread = GetCurrentThread ();
2021-01-26 16:45:36 +03:00
if ( ! SetThreadAffinityMask ( _this_thread , ! mask ? process_affinity_mask : mask ))
2021-01-25 21:50:14 +03:00
{
2023-03-04 11:05:32 +01:00
sig_log . error ( "Failed to set thread affinity 0x%x: error: %s" , mask , fmt :: win_error { GetLastError (), nullptr });
2021-01-25 21:50:14 +03:00
}
2017-11-19 23:02:35 +01:00
#elif __APPLE__
2019-07-20 15:57:23 +03:00
// Supports only one core
2020-04-13 15:31:41 +03:00
thread_affinity_policy_data_t policy = { static_cast < integer_t > ( std :: countr_zero ( mask )) };
2017-11-19 23:02:35 +01:00
thread_port_t mach_thread = pthread_mach_thread_np ( pthread_self ());
2021-01-26 16:45:36 +03:00
thread_policy_set ( mach_thread , THREAD_AFFINITY_POLICY , reinterpret_cast < thread_policy_t > ( & policy ), ! mask ? 0 : 1 );
2017-11-20 21:56:25 +00:00
#elif defined(__linux__) || defined(__DragonFly__) || defined(__FreeBSD__)
2021-01-26 16:45:36 +03:00
if ( ! mask )
{
// Reset affinity mask
mask = process_affinity_mask ;
}
2017-11-07 15:30:13 +03:00
cpu_set_t cs ;
CPU_ZERO ( & cs );
2017-10-21 14:21:37 +03:00
2019-07-20 15:57:23 +03:00
for ( u32 core = 0 ; core < 64u ; ++ core )
2017-10-21 14:21:37 +03:00
{
2019-07-20 15:57:23 +03:00
const u64 shifted = mask >> core ;
if ( shifted & 1 )
2017-10-21 14:21:37 +03:00
{
2020-02-04 21:37:00 +03:00
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wold-style-cast"
2017-10-21 14:21:37 +03:00
CPU_SET ( core , & cs );
2020-02-04 21:37:00 +03:00
#pragma GCC diagnostic pop
2017-10-21 14:21:37 +03:00
}
2019-07-20 15:57:23 +03:00
if ( shifted <= 1 )
{
break ;
}
2017-10-21 14:21:37 +03:00
}
2020-04-28 21:49:13 +03:00
if ( int err = pthread_setaffinity_np ( pthread_self (), sizeof ( cpu_set_t ), & cs ))
{
sig_log . error ( "Failed to set thread affinity 0x%x: error %d." , mask , err );
}
#endif
}
u64 thread_ctrl :: get_thread_affinity_mask ()
{
#ifdef _WIN32
2021-01-25 21:50:14 +03:00
const u64 res = process_affinity_mask ;
2020-04-28 21:49:13 +03:00
if ( DWORD_PTR result = SetThreadAffinityMask ( GetCurrentThread (), res ))
{
if ( res != result )
{
SetThreadAffinityMask ( GetCurrentThread (), result );
}
return result ;
}
sig_log . error ( "Failed to get thread affinity mask." );
return 0 ;
#elif defined(__linux__) || defined(__DragonFly__) || defined(__FreeBSD__)
cpu_set_t cs ;
CPU_ZERO ( & cs );
if ( int err = pthread_getaffinity_np ( pthread_self (), sizeof ( cpu_set_t ), & cs ))
{
sig_log . error ( "Failed to get thread affinity mask: error %d." , err );
return 0 ;
}
2020-05-05 21:37:47 +03:00
u64 result = 0 ;
2020-04-28 21:49:13 +03:00
for ( u32 core = 0 ; core < 64u ; core ++ )
{
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wold-style-cast"
if ( CPU_ISSET ( core , & cs ))
#pragma GCC diagnostic pop
{
result |= 1ull << core ;
}
}
if ( result == 0 )
{
sig_log . error ( "Thread affinity mask is out of u64 range." );
return 0 ;
}
return result ;
#else
return - 1 ;
2017-06-24 18:36:49 +03:00
#endif
}
2020-11-16 13:57:15 +03:00
2020-12-18 10:39:54 +03:00
std :: pair < void * , usz > thread_ctrl :: get_thread_stack ()
2020-11-16 13:57:15 +03:00
{
#ifdef _WIN32
ULONG_PTR _min = 0 ;
ULONG_PTR _max = 0 ;
GetCurrentThreadStackLimits ( & _min , & _max );
2020-12-18 10:39:54 +03:00
const usz ssize = _max - _min ;
2020-11-16 13:57:15 +03:00
const auto saddr = reinterpret_cast < void *> ( _min );
#else
void * saddr = 0 ;
2020-12-18 10:39:54 +03:00
usz ssize = 0 ;
2021-04-18 11:26:49 +01:00
#if defined(__linux__)
2021-05-09 08:05:48 +01:00
pthread_attr_t attr ;
2020-11-16 13:57:15 +03:00
pthread_getattr_np ( pthread_self (), & attr );
pthread_attr_getstack ( & attr , & saddr , & ssize );
2021-04-18 11:26:49 +01:00
#elif defined(__APPLE__)
saddr = pthread_get_stackaddr_np ( pthread_self ());
ssize = pthread_get_stacksize_np ( pthread_self ());
2020-11-16 13:57:15 +03:00
#else
2021-05-09 08:05:48 +01:00
pthread_attr_t attr ;
2020-11-16 13:57:15 +03:00
pthread_attr_get_np ( pthread_self (), & attr );
pthread_attr_getstackaddr ( & attr , & saddr );
pthread_attr_getstacksize ( & attr , & ssize );
#endif
#endif
return { saddr , ssize };
}
2021-03-06 20:10:48 +03:00
u64 thread_ctrl :: get_tid ()
{
#ifdef _WIN32
return GetCurrentThreadId ();
2022-07-09 14:42:03 +03:00
#elif defined(__linux__)
return syscall ( SYS_gettid );
2021-03-06 20:10:48 +03:00
#else
return reinterpret_cast < u64 > ( pthread_self ());
#endif
}
2021-03-06 21:02:37 +03:00
bool thread_ctrl :: is_main ()
{
return get_tid () == utils :: main_tid ;
}