summaryrefslogtreecommitdiff
path: root/zen/thread.cpp
blob: 08bfaa25414f486ba70b3cca7e243a8487271189 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// *****************************************************************************
// * This file is part of the FreeFileSync project. It is distributed under    *
// * GNU General Public License: https://www.gnu.org/licenses/gpl-3.0          *
// * Copyright (C) Zenju (zenju AT freefilesync DOT org) - All Rights Reserved *
// *****************************************************************************

#include "thread.h"
    #include <sys/prctl.h>
    #include <unistd.h>
    #include <sys/syscall.h>

using namespace zen;




void zen::setCurrentThreadName(const char* threadName)
{
    ::prctl(PR_SET_NAME, threadName, 0, 0, 0);

}


namespace
{
uint64_t getThreadIdNative()
{
    const pid_t tid = ::syscall(SYS_gettid); //no-fail
    //"Invalid thread and process IDs": https://blogs.msdn.microsoft.com/oldnewthing/20040223-00/?p=40503
    //if (tid == 0) -> not sure this holds on Linux, too!
    //    throw std::runtime_error(std::string(__FILE__) + "[" + numberTo<std::string>(__LINE__) + "] Failed to get thread ID.");
    static_assert(sizeof(uint64_t) >= sizeof(tid));
    return tid;
}


const uint64_t globalMainThreadId = getThreadId(); //avoid code-gen for "magic static"!
}


uint64_t zen::getThreadId()
{
    thread_local const uint64_t tid = getThreadIdNative(); //buffer to get predictable perf characteristics
    return tid;
}


uint64_t zen::getMainThreadId()
{
    //don't make this a function-scope static (avoid code-gen for "magic static")
    if (globalMainThreadId == 0) //might be called during static initialization
        return getThreadId();

    return globalMainThreadId;
}
bgstack15