1 | /* Definition for thread-local data handling. nptl/x86_64 version. |
2 | Copyright (C) 2002-2017 Free Software Foundation, Inc. |
3 | This file is part of the GNU C Library. |
4 | |
5 | The GNU C Library is free software; you can redistribute it and/or |
6 | modify it under the terms of the GNU Lesser General Public |
7 | License as published by the Free Software Foundation; either |
8 | version 2.1 of the License, or (at your option) any later version. |
9 | |
10 | The GNU C Library is distributed in the hope that it will be useful, |
11 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
13 | Lesser General Public License for more details. |
14 | |
15 | You should have received a copy of the GNU Lesser General Public |
16 | License along with the GNU C Library; if not, see |
17 | <http://www.gnu.org/licenses/>. */ |
18 | |
19 | #ifndef _TLS_H |
20 | #define _TLS_H 1 |
21 | |
22 | #ifndef __ASSEMBLER__ |
23 | # include <asm/prctl.h> /* For ARCH_SET_FS. */ |
24 | # include <stdbool.h> |
25 | # include <stddef.h> |
26 | # include <stdint.h> |
27 | # include <stdlib.h> |
28 | # include <sysdep.h> |
29 | # include <libc-pointer-arith.h> /* For cast_to_integer. */ |
30 | # include <kernel-features.h> |
31 | # include <dl-dtv.h> |
32 | |
33 | /* Replacement type for __m128 since this file is included by ld.so, |
34 | which is compiled with -mno-sse. It must not change the alignment |
35 | of rtld_savespace_sse. */ |
36 | typedef struct |
37 | { |
38 | int i[4]; |
39 | } __128bits; |
40 | |
41 | |
42 | typedef struct |
43 | { |
44 | void *tcb; /* Pointer to the TCB. Not necessarily the |
45 | thread descriptor used by libpthread. */ |
46 | dtv_t *dtv; |
47 | void *self; /* Pointer to the thread descriptor. */ |
48 | int multiple_threads; |
49 | int gscope_flag; |
50 | uintptr_t sysinfo; |
51 | uintptr_t stack_guard; |
52 | uintptr_t pointer_guard; |
53 | unsigned long int vgetcpu_cache[2]; |
54 | # ifndef __ASSUME_PRIVATE_FUTEX |
55 | int private_futex; |
56 | # else |
57 | int __glibc_reserved1; |
58 | # endif |
59 | int __glibc_unused1; |
60 | /* Reservation of some values for the TM ABI. */ |
61 | void *__private_tm[4]; |
62 | /* GCC split stack support. */ |
63 | void *__private_ss; |
64 | long int __glibc_reserved2; |
65 | /* Must be kept even if it is no longer used by glibc since programs, |
66 | like AddressSanitizer, depend on the size of tcbhead_t. */ |
67 | __128bits __glibc_unused2[8][4] __attribute__ ((aligned (32))); |
68 | |
69 | void *__padding[8]; |
70 | } tcbhead_t; |
71 | |
72 | #else /* __ASSEMBLER__ */ |
73 | # include <tcb-offsets.h> |
74 | #endif |
75 | |
76 | |
77 | /* Alignment requirement for the stack. */ |
78 | #define STACK_ALIGN 16 |
79 | |
80 | |
81 | #ifndef __ASSEMBLER__ |
82 | /* Get system call information. */ |
83 | # include <sysdep.h> |
84 | |
85 | #ifndef LOCK_PREFIX |
86 | # ifdef UP |
87 | # define LOCK_PREFIX /* nothing */ |
88 | # else |
89 | # define LOCK_PREFIX "lock;" |
90 | # endif |
91 | #endif |
92 | |
93 | /* This is the size of the initial TCB. Can't be just sizeof (tcbhead_t), |
94 | because NPTL getpid, __libc_alloca_cutoff etc. need (almost) the whole |
95 | struct pthread even when not linked with -lpthread. */ |
96 | # define TLS_INIT_TCB_SIZE sizeof (struct pthread) |
97 | |
98 | /* Alignment requirements for the initial TCB. */ |
99 | # define TLS_INIT_TCB_ALIGN __alignof__ (struct pthread) |
100 | |
101 | /* This is the size of the TCB. */ |
102 | # define TLS_TCB_SIZE sizeof (struct pthread) |
103 | |
104 | /* Alignment requirements for the TCB. */ |
105 | # define TLS_TCB_ALIGN __alignof__ (struct pthread) |
106 | |
107 | /* The TCB can have any size and the memory following the address the |
108 | thread pointer points to is unspecified. Allocate the TCB there. */ |
109 | # define TLS_TCB_AT_TP 1 |
110 | # define TLS_DTV_AT_TP 0 |
111 | |
112 | /* Get the thread descriptor definition. */ |
113 | # include <nptl/descr.h> |
114 | |
115 | |
116 | /* Install the dtv pointer. The pointer passed is to the element with |
117 | index -1 which contain the length. */ |
118 | # define INSTALL_DTV(descr, dtvp) \ |
119 | ((tcbhead_t *) (descr))->dtv = (dtvp) + 1 |
120 | |
121 | /* Install new dtv for current thread. */ |
122 | # define INSTALL_NEW_DTV(dtvp) \ |
123 | ({ struct pthread *__pd; \ |
124 | THREAD_SETMEM (__pd, header.dtv, (dtvp)); }) |
125 | |
126 | /* Return dtv of given thread descriptor. */ |
127 | # define GET_DTV(descr) \ |
128 | (((tcbhead_t *) (descr))->dtv) |
129 | |
130 | |
131 | /* Code to initially initialize the thread pointer. This might need |
132 | special attention since 'errno' is not yet available and if the |
133 | operation can cause a failure 'errno' must not be touched. |
134 | |
135 | We have to make the syscall for both uses of the macro since the |
136 | address might be (and probably is) different. */ |
137 | # define TLS_INIT_TP(thrdescr) \ |
138 | ({ void *_thrdescr = (thrdescr); \ |
139 | tcbhead_t *_head = _thrdescr; \ |
140 | int _result; \ |
141 | \ |
142 | _head->tcb = _thrdescr; \ |
143 | /* For now the thread descriptor is at the same address. */ \ |
144 | _head->self = _thrdescr; \ |
145 | \ |
146 | /* It is a simple syscall to set the %fs value for the thread. */ \ |
147 | asm volatile ("syscall" \ |
148 | : "=a" (_result) \ |
149 | : "0" ((unsigned long int) __NR_arch_prctl), \ |
150 | "D" ((unsigned long int) ARCH_SET_FS), \ |
151 | "S" (_thrdescr) \ |
152 | : "memory", "cc", "r11", "cx"); \ |
153 | \ |
154 | _result ? "cannot set %fs base address for thread-local storage" : 0; \ |
155 | }) |
156 | |
157 | # define TLS_DEFINE_INIT_TP(tp, pd) void *tp = (pd) |
158 | |
159 | |
160 | /* Return the address of the dtv for the current thread. */ |
161 | # define THREAD_DTV() \ |
162 | ({ struct pthread *__pd; \ |
163 | THREAD_GETMEM (__pd, header.dtv); }) |
164 | |
165 | |
166 | /* Return the thread descriptor for the current thread. |
167 | |
168 | The contained asm must *not* be marked volatile since otherwise |
169 | assignments like |
170 | pthread_descr self = thread_self(); |
171 | do not get optimized away. */ |
172 | # define THREAD_SELF \ |
173 | ({ struct pthread *__self; \ |
174 | asm ("mov %%fs:%c1,%0" : "=r" (__self) \ |
175 | : "i" (offsetof (struct pthread, header.self))); \ |
176 | __self;}) |
177 | |
178 | /* Magic for libthread_db to know how to do THREAD_SELF. */ |
179 | # define DB_THREAD_SELF_INCLUDE <sys/reg.h> /* For the FS constant. */ |
180 | # define DB_THREAD_SELF CONST_THREAD_AREA (64, FS) |
181 | |
182 | /* Read member of the thread descriptor directly. */ |
183 | # define THREAD_GETMEM(descr, member) \ |
184 | ({ __typeof (descr->member) __value; \ |
185 | if (sizeof (__value) == 1) \ |
186 | asm volatile ("movb %%fs:%P2,%b0" \ |
187 | : "=q" (__value) \ |
188 | : "0" (0), "i" (offsetof (struct pthread, member))); \ |
189 | else if (sizeof (__value) == 4) \ |
190 | asm volatile ("movl %%fs:%P1,%0" \ |
191 | : "=r" (__value) \ |
192 | : "i" (offsetof (struct pthread, member))); \ |
193 | else \ |
194 | { \ |
195 | if (sizeof (__value) != 8) \ |
196 | /* There should not be any value with a size other than 1, \ |
197 | 4 or 8. */ \ |
198 | abort (); \ |
199 | \ |
200 | asm volatile ("movq %%fs:%P1,%q0" \ |
201 | : "=r" (__value) \ |
202 | : "i" (offsetof (struct pthread, member))); \ |
203 | } \ |
204 | __value; }) |
205 | |
206 | |
207 | /* Same as THREAD_GETMEM, but the member offset can be non-constant. */ |
208 | # define THREAD_GETMEM_NC(descr, member, idx) \ |
209 | ({ __typeof (descr->member[0]) __value; \ |
210 | if (sizeof (__value) == 1) \ |
211 | asm volatile ("movb %%fs:%P2(%q3),%b0" \ |
212 | : "=q" (__value) \ |
213 | : "0" (0), "i" (offsetof (struct pthread, member[0])), \ |
214 | "r" (idx)); \ |
215 | else if (sizeof (__value) == 4) \ |
216 | asm volatile ("movl %%fs:%P1(,%q2,4),%0" \ |
217 | : "=r" (__value) \ |
218 | : "i" (offsetof (struct pthread, member[0])), "r" (idx));\ |
219 | else \ |
220 | { \ |
221 | if (sizeof (__value) != 8) \ |
222 | /* There should not be any value with a size other than 1, \ |
223 | 4 or 8. */ \ |
224 | abort (); \ |
225 | \ |
226 | asm volatile ("movq %%fs:%P1(,%q2,8),%q0" \ |
227 | : "=r" (__value) \ |
228 | : "i" (offsetof (struct pthread, member[0])), \ |
229 | "r" (idx)); \ |
230 | } \ |
231 | __value; }) |
232 | |
233 | |
234 | /* Loading addresses of objects on x86-64 needs to be treated special |
235 | when generating PIC code. */ |
236 | #ifdef __pic__ |
237 | # define IMM_MODE "nr" |
238 | #else |
239 | # define IMM_MODE "ir" |
240 | #endif |
241 | |
242 | |
243 | /* Set member of the thread descriptor directly. */ |
244 | # define THREAD_SETMEM(descr, member, value) \ |
245 | ({ if (sizeof (descr->member) == 1) \ |
246 | asm volatile ("movb %b0,%%fs:%P1" : \ |
247 | : "iq" (value), \ |
248 | "i" (offsetof (struct pthread, member))); \ |
249 | else if (sizeof (descr->member) == 4) \ |
250 | asm volatile ("movl %0,%%fs:%P1" : \ |
251 | : IMM_MODE (value), \ |
252 | "i" (offsetof (struct pthread, member))); \ |
253 | else \ |
254 | { \ |
255 | if (sizeof (descr->member) != 8) \ |
256 | /* There should not be any value with a size other than 1, \ |
257 | 4 or 8. */ \ |
258 | abort (); \ |
259 | \ |
260 | asm volatile ("movq %q0,%%fs:%P1" : \ |
261 | : IMM_MODE ((uint64_t) cast_to_integer (value)), \ |
262 | "i" (offsetof (struct pthread, member))); \ |
263 | }}) |
264 | |
265 | |
266 | /* Same as THREAD_SETMEM, but the member offset can be non-constant. */ |
267 | # define THREAD_SETMEM_NC(descr, member, idx, value) \ |
268 | ({ if (sizeof (descr->member[0]) == 1) \ |
269 | asm volatile ("movb %b0,%%fs:%P1(%q2)" : \ |
270 | : "iq" (value), \ |
271 | "i" (offsetof (struct pthread, member[0])), \ |
272 | "r" (idx)); \ |
273 | else if (sizeof (descr->member[0]) == 4) \ |
274 | asm volatile ("movl %0,%%fs:%P1(,%q2,4)" : \ |
275 | : IMM_MODE (value), \ |
276 | "i" (offsetof (struct pthread, member[0])), \ |
277 | "r" (idx)); \ |
278 | else \ |
279 | { \ |
280 | if (sizeof (descr->member[0]) != 8) \ |
281 | /* There should not be any value with a size other than 1, \ |
282 | 4 or 8. */ \ |
283 | abort (); \ |
284 | \ |
285 | asm volatile ("movq %q0,%%fs:%P1(,%q2,8)" : \ |
286 | : IMM_MODE ((uint64_t) cast_to_integer (value)), \ |
287 | "i" (offsetof (struct pthread, member[0])), \ |
288 | "r" (idx)); \ |
289 | }}) |
290 | |
291 | |
292 | /* Atomic compare and exchange on TLS, returning old value. */ |
293 | # define THREAD_ATOMIC_CMPXCHG_VAL(descr, member, newval, oldval) \ |
294 | ({ __typeof (descr->member) __ret; \ |
295 | __typeof (oldval) __old = (oldval); \ |
296 | if (sizeof (descr->member) == 4) \ |
297 | asm volatile (LOCK_PREFIX "cmpxchgl %2, %%fs:%P3" \ |
298 | : "=a" (__ret) \ |
299 | : "0" (__old), "r" (newval), \ |
300 | "i" (offsetof (struct pthread, member))); \ |
301 | else \ |
302 | /* Not necessary for other sizes in the moment. */ \ |
303 | abort (); \ |
304 | __ret; }) |
305 | |
306 | |
307 | /* Atomic logical and. */ |
308 | # define THREAD_ATOMIC_AND(descr, member, val) \ |
309 | (void) ({ if (sizeof ((descr)->member) == 4) \ |
310 | asm volatile (LOCK_PREFIX "andl %1, %%fs:%P0" \ |
311 | :: "i" (offsetof (struct pthread, member)), \ |
312 | "ir" (val)); \ |
313 | else \ |
314 | /* Not necessary for other sizes in the moment. */ \ |
315 | abort (); }) |
316 | |
317 | |
318 | /* Atomic set bit. */ |
319 | # define THREAD_ATOMIC_BIT_SET(descr, member, bit) \ |
320 | (void) ({ if (sizeof ((descr)->member) == 4) \ |
321 | asm volatile (LOCK_PREFIX "orl %1, %%fs:%P0" \ |
322 | :: "i" (offsetof (struct pthread, member)), \ |
323 | "ir" (1 << (bit))); \ |
324 | else \ |
325 | /* Not necessary for other sizes in the moment. */ \ |
326 | abort (); }) |
327 | |
328 | |
329 | /* Set the stack guard field in TCB head. */ |
330 | # define THREAD_SET_STACK_GUARD(value) \ |
331 | THREAD_SETMEM (THREAD_SELF, header.stack_guard, value) |
332 | # define THREAD_COPY_STACK_GUARD(descr) \ |
333 | ((descr)->header.stack_guard \ |
334 | = THREAD_GETMEM (THREAD_SELF, header.stack_guard)) |
335 | |
336 | |
337 | /* Set the pointer guard field in the TCB head. */ |
338 | # define THREAD_SET_POINTER_GUARD(value) \ |
339 | THREAD_SETMEM (THREAD_SELF, header.pointer_guard, value) |
340 | # define THREAD_COPY_POINTER_GUARD(descr) \ |
341 | ((descr)->header.pointer_guard \ |
342 | = THREAD_GETMEM (THREAD_SELF, header.pointer_guard)) |
343 | |
344 | |
345 | /* Get and set the global scope generation counter in the TCB head. */ |
346 | # define THREAD_GSCOPE_FLAG_UNUSED 0 |
347 | # define THREAD_GSCOPE_FLAG_USED 1 |
348 | # define THREAD_GSCOPE_FLAG_WAIT 2 |
349 | # define THREAD_GSCOPE_RESET_FLAG() \ |
350 | do \ |
351 | { int __res; \ |
352 | asm volatile ("xchgl %0, %%fs:%P1" \ |
353 | : "=r" (__res) \ |
354 | : "i" (offsetof (struct pthread, header.gscope_flag)), \ |
355 | "0" (THREAD_GSCOPE_FLAG_UNUSED)); \ |
356 | if (__res == THREAD_GSCOPE_FLAG_WAIT) \ |
357 | lll_futex_wake (&THREAD_SELF->header.gscope_flag, 1, LLL_PRIVATE); \ |
358 | } \ |
359 | while (0) |
360 | # define THREAD_GSCOPE_SET_FLAG() \ |
361 | THREAD_SETMEM (THREAD_SELF, header.gscope_flag, THREAD_GSCOPE_FLAG_USED) |
362 | # define THREAD_GSCOPE_WAIT() \ |
363 | GL(dl_wait_lookup_done) () |
364 | |
365 | #endif /* __ASSEMBLER__ */ |
366 | |
367 | #endif /* tls.h */ |
368 | |