Commit | Line | Data |
---|---|---|
cb8d0d24 MJ |
1 | /* |
2 | * Copyright (C) 2020 Michael Jeanson <mjeanson@efficios.com> | |
3 | * | |
4 | * SPDX-License-Identifier: GPL-2.0-only | |
5 | * | |
6 | */ | |
7 | ||
8 | #ifndef _COMPAT_PTHREAD_H | |
9 | #define _COMPAT_PTHREAD_H | |
10 | ||
11 | #include <pthread.h> | |
edf4b93e | 12 | #include <common/compat/errno.h> |
014d7d3b MJ |
13 | #include <string.h> |
14 | ||
15 | #define LTTNG_PTHREAD_NAMELEN 16 | |
cb8d0d24 MJ |
16 | |
17 | #if defined(HAVE_PTHREAD_SETNAME_NP_WITH_TID) | |
18 | static inline | |
19 | int lttng_pthread_setname_np(const char *name) | |
20 | { | |
014d7d3b MJ |
21 | /* |
22 | * Some implementations don't error out, replicate this behavior for | |
23 | * consistency. | |
24 | */ | |
25 | if (strnlen(name, LTTNG_PTHREAD_NAMELEN) >= LTTNG_PTHREAD_NAMELEN) { | |
26 | return ERANGE; | |
27 | } | |
28 | ||
cb8d0d24 MJ |
29 | return pthread_setname_np(pthread_self(), name); |
30 | } | |
014d7d3b MJ |
31 | |
32 | static inline | |
33 | int lttng_pthread_getname_np(char *name, size_t len) | |
34 | { | |
35 | return pthread_getname_np(pthread_self(), name, len); | |
36 | } | |
cb8d0d24 MJ |
37 | #elif defined(HAVE_PTHREAD_SETNAME_NP_WITHOUT_TID) |
38 | static inline | |
39 | int lttng_pthread_setname_np(const char *name) | |
40 | { | |
41 | return pthread_setname_np(name); | |
42 | } | |
014d7d3b MJ |
43 | |
44 | static inline | |
45 | int lttng_pthread_getname_np(char *name, size_t len) | |
46 | { | |
47 | return pthread_getname_np(name, len); | |
48 | } | |
49 | #elif defined(HAVE_PTHREAD_SET_NAME_NP_WITH_TID) | |
50 | ||
51 | #include <pthread_np.h> | |
52 | static inline | |
53 | int lttng_pthread_setname_np(const char *name) | |
54 | { | |
55 | /* Replicate pthread_setname_np's behavior. */ | |
56 | if (strnlen(name, LTTNG_PTHREAD_NAMELEN) >= LTTNG_PTHREAD_NAMELEN) { | |
57 | return ERANGE; | |
58 | } | |
59 | ||
60 | pthread_set_name_np(pthread_self(), name); | |
61 | return 0; | |
62 | } | |
63 | ||
64 | static inline | |
65 | int lttng_pthread_getname_np(char *name, size_t len) | |
66 | { | |
67 | pthread_get_name_np(pthread_self(), name, len); | |
68 | return 0; | |
69 | } | |
70 | #elif defined(__linux__) | |
71 | ||
72 | /* Fallback on prtctl on Linux */ | |
73 | #include <sys/prctl.h> | |
74 | ||
75 | static inline | |
76 | int lttng_pthread_setname_np(const char *name) | |
77 | { | |
78 | /* Replicate pthread_setname_np's behavior. */ | |
79 | if (strnlen(name, LTTNG_UST_ABI_PROCNAME_LEN) >= LTTNG_UST_ABI_PROCNAME_LEN) { | |
80 | return ERANGE; | |
81 | } | |
82 | return prctl(PR_SET_NAME, name, 0, 0, 0); | |
83 | } | |
84 | ||
85 | static inline | |
86 | int lttng_pthread_getname_np(char *name, size_t len) | |
87 | { | |
88 | return prctl(PR_GET_NAME, name, 0, 0, 0); | |
89 | } | |
cb8d0d24 MJ |
90 | #else |
91 | /* | |
92 | * For platforms without thread name support, do nothing. | |
93 | */ | |
94 | static inline | |
95 | int lttng_pthread_setname_np(const char *name) | |
96 | { | |
97 | return -ENOSYS; | |
98 | } | |
014d7d3b MJ |
99 | |
100 | static inline | |
101 | int lttng_pthread_getname_np(char *name, size_t len) | |
102 | { | |
103 | return -ENOSYS; | |
104 | } | |
cb8d0d24 MJ |
105 | #endif |
106 | ||
107 | #endif /* _COMPAT_PTHREAD_H */ |