tracy/common/TracySystem.cpp

250 lines
6.1 KiB
C++
Raw Normal View History

#if defined _MSC_VER || defined __CYGWIN__ || defined _WIN32
2018-08-01 12:07:30 +00:00
# ifndef WIN32_LEAN_AND_MEAN
# define WIN32_LEAN_AND_MEAN
# endif
# ifndef NOMINMAX
# define NOMINMAX
# endif
#endif
2020-01-20 21:49:03 +00:00
#ifdef _MSC_VER
# pragma warning(disable:4996)
#endif
2019-08-13 14:20:24 +00:00
#if defined _WIN32 || defined __CYGWIN__
2017-09-10 15:46:20 +00:00
# include <windows.h>
#else
# include <pthread.h>
2017-09-22 19:48:21 +00:00
# include <string.h>
2017-09-10 15:46:20 +00:00
# include <unistd.h>
#endif
#ifdef __linux__
# ifdef __ANDROID__
# include <sys/types.h>
# else
# include <sys/syscall.h>
2019-08-13 23:12:23 +00:00
# endif
# include <fcntl.h>
#elif defined __FreeBSD__
# include <sys/thr.h>
#elif defined __NetBSD__ || defined __DragonFly__
# include <sys/lwp.h>
#endif
2019-01-19 11:03:30 +00:00
#ifdef __MINGW32__
# define __STDC_FORMAT_MACROS
#endif
#include <inttypes.h>
2017-10-30 20:51:24 +00:00
#include <stdio.h>
#include <stdlib.h>
2017-09-10 15:46:20 +00:00
#include "TracySystem.hpp"
#if defined _WIN32 || defined __CYGWIN__
extern "C" typedef HRESULT (WINAPI *t_SetThreadDescription)( HANDLE, PCWSTR );
extern "C" typedef HRESULT (WINAPI *t_GetThreadDescription)( HANDLE, PWSTR* );
#endif
#ifdef TRACY_ENABLE
# include <atomic>
2017-11-02 11:57:02 +00:00
# include "TracyAlloc.hpp"
#endif
2017-09-10 15:46:20 +00:00
namespace tracy
{
namespace detail
{
TRACY_API uint64_t GetThreadHandleImpl()
{
#if defined _WIN32 || defined __CYGWIN__
static_assert( sizeof( decltype( GetCurrentThreadId() ) ) <= sizeof( uint64_t ), "Thread handle too big to fit in protocol" );
return uint64_t( GetCurrentThreadId() );
#elif defined __APPLE__
uint64_t id;
pthread_threadid_np( pthread_self(), &id );
return id;
#elif defined __ANDROID__
return (uint64_t)gettid();
#elif defined __linux__
return (uint64_t)syscall( SYS_gettid );
#elif defined __FreeBSD__
long id;
thr_self( &id );
return id;
#elif defined __NetBSD__
return _lwp_self();
#elif defined __DragonFly__
return lwp_gettid();
#elif defined __OpenBSD__
return getthrid();
#else
static_assert( sizeof( decltype( pthread_self() ) ) <= sizeof( uint64_t ), "Thread handle too big to fit in protocol" );
return uint64_t( pthread_self() );
#endif
}
}
#ifdef TRACY_ENABLE
struct ThreadNameData
{
uint64_t id;
const char* name;
ThreadNameData* next;
};
2020-01-25 15:17:03 +00:00
std::atomic<ThreadNameData*>& GetThreadNameData();
TRACY_API void InitRPMallocThread();
#endif
2020-01-25 15:17:03 +00:00
TRACY_API void SetThreadName( const char* name )
2017-09-10 15:46:20 +00:00
{
#if defined _WIN32 || defined __CYGWIN__
static auto _SetThreadDescription = (t_SetThreadDescription)GetProcAddress( GetModuleHandleA( "kernel32.dll" ), "SetThreadDescription" );
if( _SetThreadDescription )
{
wchar_t buf[256];
mbstowcs( buf, name, 256 );
_SetThreadDescription( GetCurrentThread(), buf );
}
else
2017-09-21 22:37:28 +00:00
{
# if defined _MSC_VER
const DWORD MS_VC_EXCEPTION=0x406D1388;
# pragma pack( push, 8 )
struct THREADNAME_INFO
{
DWORD dwType;
LPCSTR szName;
DWORD dwThreadID;
DWORD dwFlags;
};
2017-09-21 22:37:28 +00:00
# pragma pack(pop)
DWORD ThreadId = GetCurrentThreadId();
THREADNAME_INFO info;
info.dwType = 0x1000;
info.szName = name;
info.dwThreadID = ThreadId;
info.dwFlags = 0;
2017-09-21 22:37:28 +00:00
__try
{
RaiseException( MS_VC_EXCEPTION, 0, sizeof(info)/sizeof(ULONG_PTR), (ULONG_PTR*)&info );
}
__except(EXCEPTION_EXECUTE_HANDLER)
{
}
2017-09-21 22:37:28 +00:00
# endif
}
2019-08-13 14:20:24 +00:00
#elif defined _GNU_SOURCE && !defined __EMSCRIPTEN__ && !defined __CYGWIN__
{
const auto sz = strlen( name );
if( sz <= 15 )
{
pthread_setname_np( pthread_self(), name );
}
else
{
char buf[16];
memcpy( buf, name, 15 );
buf[15] = '\0';
pthread_setname_np( pthread_self(), buf );
}
}
2017-09-10 15:46:20 +00:00
#endif
#ifdef TRACY_ENABLE
{
InitRPMallocThread();
const auto sz = strlen( name );
char* buf = (char*)tracy_malloc( sz+1 );
memcpy( buf, name, sz );
buf[sz] = '\0';
2017-11-02 11:57:02 +00:00
auto data = (ThreadNameData*)tracy_malloc( sizeof( ThreadNameData ) );
data->id = detail::GetThreadHandleImpl();
data->name = buf;
2019-02-19 18:33:37 +00:00
data->next = GetThreadNameData().load( std::memory_order_relaxed );
while( !GetThreadNameData().compare_exchange_weak( data->next, data, std::memory_order_release, std::memory_order_relaxed ) ) {}
}
#endif
2017-09-10 15:46:20 +00:00
}
2020-01-25 15:17:03 +00:00
TRACY_API const char* GetThreadName( uint64_t id )
{
static char buf[256];
#ifdef TRACY_ENABLE
2019-02-19 18:33:37 +00:00
auto ptr = GetThreadNameData().load( std::memory_order_relaxed );
while( ptr )
{
if( ptr->id == id )
{
2018-03-30 23:38:57 +00:00
return ptr->name;
}
ptr = ptr->next;
}
#else
# if defined _WIN32 || defined __CYGWIN__
static auto _GetThreadDescription = (t_GetThreadDescription)GetProcAddress( GetModuleHandleA( "kernel32.dll" ), "GetThreadDescription" );
if( _GetThreadDescription )
{
auto hnd = OpenThread( THREAD_QUERY_LIMITED_INFORMATION, FALSE, (DWORD)id );
if( hnd != 0 )
2017-10-15 11:26:05 +00:00
{
PWSTR tmp;
_GetThreadDescription( hnd, &tmp );
auto ret = wcstombs( buf, tmp, 256 );
CloseHandle( hnd );
if( ret != 0 )
{
return buf;
}
2017-10-15 11:26:05 +00:00
}
}
# elif defined __linux__
int cs, fd;
char path[32];
2018-07-24 11:43:25 +00:00
# ifdef __ANDROID__
int tid = gettid();
# else
2018-07-14 11:46:25 +00:00
int tid = (int) syscall( SYS_gettid );
2018-07-24 11:43:25 +00:00
# endif
2018-07-14 11:46:25 +00:00
snprintf( path, sizeof( path ), "/proc/self/task/%d/comm", tid );
sprintf( buf, "%" PRIu64, id );
2018-09-07 15:21:27 +00:00
# ifndef __ANDROID__
pthread_setcancelstate( PTHREAD_CANCEL_DISABLE, &cs );
2018-09-07 15:21:27 +00:00
# endif
2018-07-14 11:46:25 +00:00
if ( ( fd = open( path, O_RDONLY ) ) > 0) {
int len = read( fd, buf, 255 );
if( len > 0 )
{
buf[len] = 0;
if( len > 1 && buf[len-1] == '\n' )
{
buf[len-1] = 0;
}
}
2018-07-14 11:46:25 +00:00
close( fd );
}
2018-09-07 15:21:27 +00:00
# ifndef __ANDROID__
2018-07-14 11:46:25 +00:00
pthread_setcancelstate( cs, 0 );
2018-09-07 15:21:27 +00:00
# endif
return buf;
# endif
#endif
sprintf( buf, "%" PRIu64, id );
return buf;
}
2017-09-22 19:48:21 +00:00
}
2020-08-03 15:31:00 +00:00
#ifdef __cplusplus
extern "C" {
#endif
TRACY_API void ___tracy_set_thread_name( const char* name ) { tracy::SetThreadName( name ); }
#ifdef __cplusplus
}
#endif