Home | History | Annotate | Line # | Download | only in libpthread
      1 /*	$NetBSD: pthread_int.h,v 1.115 2026/05/05 09:49:10 yamt Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2001, 2002, 2003, 2006, 2007, 2008, 2020
      5  *     The NetBSD Foundation, Inc.
      6  * All rights reserved.
      7  *
      8  * This code is derived from software contributed to The NetBSD Foundation
      9  * by Nathan J. Williams and Andrew Doran.
     10  *
     11  * Redistribution and use in source and binary forms, with or without
     12  * modification, are permitted provided that the following conditions
     13  * are met:
     14  * 1. Redistributions of source code must retain the above copyright
     15  *    notice, this list of conditions and the following disclaimer.
     16  * 2. Redistributions in binary form must reproduce the above copyright
     17  *    notice, this list of conditions and the following disclaimer in the
     18  *    documentation and/or other materials provided with the distribution.
     19  *
     20  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     21  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     22  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     23  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     24  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     30  * POSSIBILITY OF SUCH DAMAGE.
     31  */
     32 
     33 #ifndef _LIB_PTHREAD_INT_H
     34 #define _LIB_PTHREAD_INT_H
     35 
     36 #include <sys/tls.h>
     37 
     38 /* #define PTHREAD__DEBUG */
     39 
     40 #include "pthread_types.h"
     41 #include "pthread_queue.h"
     42 #include "pthread_md.h"
     43 
     44 #include <sys/atomic.h>
     45 #include <sys/rbtree.h>
     46 #include <sys/param.h>
     47 
     48 #include <limits.h>
     49 #include <lwp.h>
     50 #include <signal.h>
     51 #include <stdbool.h>
     52 
     53 #include <machine/lwp_private.h>
     54 
     55 #ifdef __GNUC__
     56 #define	PTHREAD_HIDE	__attribute__ ((visibility("hidden")))
     57 #else
     58 #define	PTHREAD_HIDE	/* nothing */
     59 #endif
     60 
     61 #define	PTHREAD__UNPARK_MAX	128
     62 
     63 /*
     64  * The size of this structure needs to be no larger than struct
     65  * __pthread_cleanup_store, defined in pthread.h.
     66  */
     67 struct pt_clean_t {
     68 	PTQ_ENTRY(pt_clean_t)	ptc_next;
     69 	void	(*ptc_cleanup)(void *);
     70 	void	*ptc_arg;
     71 };
     72 
     73 /* Private data for pthread_attr_t */
     74 struct pthread_attr_private {
     75 	char ptap_name[PTHREAD_MAX_NAMELEN_NP];
     76 	void *ptap_namearg;
     77 	void *ptap_stackaddr;
     78 	size_t ptap_stacksize;
     79 	size_t ptap_guardsize;
     80 	struct sched_param ptap_sp;
     81 	int ptap_policy;
     82 };
     83 
     84 struct pthread_lock_ops {
     85 	void	(*plo_init)(__cpu_simple_lock_t *);
     86 	int	(*plo_try)(__cpu_simple_lock_t *);
     87 	void	(*plo_unlock)(__cpu_simple_lock_t *);
     88 	void	(*plo_lock)(__cpu_simple_lock_t *);
     89 };
     90 
     91 struct	__pthread_st {
     92 	pthread_t	pt_self;	/* Must be first. */
     93 #if defined(__HAVE_TLS_VARIANT_I) || defined(__HAVE_TLS_VARIANT_II)
     94 	struct tls_tcb	*pt_tls;	/* Thread Local Storage area */
     95 #endif
     96 	unsigned int	pt_magic;	/* Magic number */
     97 	int		pt_state;	/* running, blocked, etc. */
     98 	int		pt_flags;	/* see PT_FLAG_* below */
     99 	_Atomic unsigned int	pt_cancel;	/* Cancellation */
    100 	int		pt_errno;	/* Thread-specific errno. */
    101 	stack_t		pt_stack;	/* Our stack */
    102 	bool		pt_stack_allocated;
    103 	bool		pt_is_main;	/* is the main thread? */
    104 	size_t		pt_guardsize;
    105 	void		*pt_exitval;	/* Read by pthread_join() */
    106 	char		*pt_name;	/* Thread's name, set by the app. */
    107 	struct pthread_lock_ops pt_lockops;/* Cached to avoid PIC overhead */
    108 	void		*(*pt_func)(void *);/* Function to call at start. */
    109 	void		*pt_arg;	/* Argument to pass at start. */
    110 
    111 	/* Stack of cancellation cleanup handlers and their arguments */
    112 	PTQ_HEAD(, pt_clean_t)	pt_cleanup_stack;
    113 
    114 	/* LWP ID and entry on the list of all threads. */
    115 	lwpid_t		pt_lid;
    116 	PTQ_ENTRY(__pthread_st)	pt_deadq;
    117 
    118 	/*
    119 	 * rbtree node and entry on the list of all threads.  pt_alltree in
    120 	 * its own cacheline, so pthread__find() is not needlessly impacted
    121 	 * by threads going about their normal business.  pt_allq is
    122 	 * adjusted at the same time as pt_alltree.
    123 	 */
    124 	rb_node_t	pt_alltree __aligned(COHERENCY_UNIT);
    125 	PTQ_ENTRY(__pthread_st) pt_allq;
    126 
    127 	/* Lock on state also gets its own line. */
    128 	pthread_mutex_t	pt_lock __aligned(COHERENCY_UNIT);
    129 
    130 	/*
    131 	 * General synchronization data.  We try to align, as threads
    132 	 * on other CPUs will access this data frequently.
    133 	 */
    134 	int		pt_dummy1 __aligned(COHERENCY_UNIT);
    135 	struct lwpctl 	*pt_lwpctl;	/* Kernel/user comms area */
    136 	volatile int	pt_rwlocked;	/* Handed rwlock successfully */
    137 	void * volatile	pt_sleepobj;	/* Object slept on */
    138 	PTQ_ENTRY(__pthread_st) pt_sleep;
    139 
    140 	/* Thread-specific data.  Large so it sits close to the end. */
    141 	int		pt_havespecific __aligned(COHERENCY_UNIT);
    142 	struct pt_specific {
    143 		void *pts_value;
    144 		PTQ_ENTRY(pt_specific) pts_next;
    145 	} pt_specific[];
    146 };
    147 
    148 /* Thread states */
    149 #define PT_STATE_RUNNING	1
    150 #define PT_STATE_ZOMBIE		5
    151 #define PT_STATE_DEAD		6
    152 
    153 /* Flag values */
    154 
    155 #define PT_FLAG_DETACHED	0x0001
    156 #define PT_FLAG_SCOPE_SYSTEM	0x0040
    157 #define PT_FLAG_EXPLICIT_SCHED	0x0080
    158 #define PT_FLAG_SUSPENDED	0x0100	/* In the suspended queue */
    159 
    160 /* pt_cancel word */
    161 
    162 #define	PT_CANCEL_DISABLED	__BIT(0)
    163 #define	PT_CANCEL_ASYNC		__BIT(1)
    164 #define	PT_CANCEL_PENDING	__BIT(2)
    165 #define	PT_CANCEL_CANCELLED	__BIT(3)
    166 
    167 #define PT_MAGIC	0x11110001
    168 #define PT_DEAD		0xDEAD0001
    169 
    170 #define PT_ATTR_MAGIC	0x22220002
    171 #define PT_ATTR_DEAD	0xDEAD0002
    172 
    173 extern size_t	pthread__stacksize;
    174 extern size_t	pthread__guardsize;
    175 extern size_t	pthread__pagesize;
    176 extern int	pthread__nspins;
    177 extern int	pthread__concurrency;
    178 extern int 	pthread__osrev;
    179 extern size_t 	pthread__unpark_max;
    180 extern int	pthread_keys_max;
    181 
    182 extern int	__uselibcstub;
    183 
    184 struct pthread__waiter {
    185 	struct pthread__waiter	*volatile next;
    186 	lwpid_t			volatile lid;
    187 };
    188 
    189 /* Flag to be used in a ucontext_t's uc_flags indicating that
    190  * the saved register state is "user" state only, not full
    191  * trap state.
    192  */
    193 #define _UC_USER_BIT		30
    194 #define _UC_USER		(1LU << _UC_USER_BIT)
    195 
    196 /* Utility functions */
    197 void	pthread__unpark_all(pthread_queue_t *, pthread_t, pthread_mutex_t *)
    198     PTHREAD_HIDE;
    199 void	pthread__unpark(pthread_queue_t *, pthread_t, pthread_mutex_t *)
    200     PTHREAD_HIDE;
    201 int	pthread__park(pthread_t, pthread_mutex_t *, pthread_queue_t *,
    202 		      const struct timespec *, int) PTHREAD_HIDE;
    203 pthread_mutex_t *pthread__hashlock(volatile const void *) PTHREAD_HIDE;
    204 
    205 /* Internal locking primitives */
    206 void	pthread__lockprim_init(void) PTHREAD_HIDE;
    207 void	pthread_lockinit(pthread_spin_t *) PTHREAD_HIDE;
    208 
    209 static inline void pthread__spinlock(pthread_t, pthread_spin_t *)
    210     __attribute__((__always_inline__));
    211 static inline void
    212 pthread__spinlock(pthread_t self, pthread_spin_t *lock)
    213 {
    214 	if (__predict_true((*self->pt_lockops.plo_try)(lock)))
    215 		return;
    216 	(*self->pt_lockops.plo_lock)(lock);
    217 }
    218 
    219 static inline int pthread__spintrylock(pthread_t, pthread_spin_t *)
    220     __attribute__((__always_inline__));
    221 static inline int
    222 pthread__spintrylock(pthread_t self, pthread_spin_t *lock)
    223 {
    224 	return (*self->pt_lockops.plo_try)(lock);
    225 }
    226 
    227 static inline void pthread__spinunlock(pthread_t, pthread_spin_t *)
    228     __attribute__((__always_inline__));
    229 static inline void
    230 pthread__spinunlock(pthread_t self, pthread_spin_t *lock)
    231 {
    232 	(*self->pt_lockops.plo_unlock)(lock);
    233 }
    234 
    235 extern const struct pthread_lock_ops *pthread__lock_ops;
    236 
    237 int	pthread__simple_locked_p(__cpu_simple_lock_t *) PTHREAD_HIDE;
    238 #define	pthread__simple_lock_init(alp)	(*pthread__lock_ops->plo_init)(alp)
    239 #define	pthread__simple_lock_try(alp)	(*pthread__lock_ops->plo_try)(alp)
    240 #define	pthread__simple_unlock(alp)	(*pthread__lock_ops->plo_unlock)(alp)
    241 
    242 void	pthread__testcancel(pthread_t) PTHREAD_HIDE;
    243 int	pthread__find(pthread_t) PTHREAD_HIDE;
    244 
    245 #ifndef PTHREAD_MD_INIT
    246 #define PTHREAD_MD_INIT
    247 #endif
    248 
    249 #ifndef _INITCONTEXT_U_MD
    250 #define _INITCONTEXT_U_MD(ucp)
    251 #endif
    252 
    253 #define _INITCONTEXT_U(ucp) do {					\
    254 	(ucp)->uc_flags = _UC_CPU | _UC_STACK;				\
    255 	_INITCONTEXT_U_MD(ucp)						\
    256 	} while (0)
    257 
    258 
    259 #if !defined(__HAVE_TLS_VARIANT_I) && !defined(__HAVE_TLS_VARIANT_II)
    260 #error Either __HAVE_TLS_VARIANT_I or __HAVE_TLS_VARIANT_II must be defined
    261 #endif
    262 
    263 #ifdef _PTHREAD_GETTCB_EXT
    264 struct tls_tcb *_PTHREAD_GETTCB_EXT(void);
    265 #endif
    266 
    267 static inline pthread_t __constfunc
    268 pthread__self(void)
    269 {
    270 #if defined(_PTHREAD_GETTCB_EXT)
    271 	struct tls_tcb * const tcb = _PTHREAD_GETTCB_EXT();
    272 #elif defined(__HAVE___LWP_GETTCB_FAST)
    273 	struct tls_tcb * const tcb = __lwp_gettcb_fast();
    274 #else
    275 	struct tls_tcb * const tcb = __lwp_getprivate_fast();
    276 #endif
    277 	return (pthread_t)tcb->tcb_pthread;
    278 }
    279 
    280 #define pthread__abort()						\
    281 	pthread__assertfunc(__FILE__, __LINE__, __func__, "unreachable")
    282 
    283 #define pthread__assert(e) do {						\
    284 	if (__predict_false(!(e)))					\
    285        	       pthread__assertfunc(__FILE__, __LINE__, __func__, #e);	\
    286         } while (0)
    287 
    288 #define pthread__error(err, msg, e) do {				\
    289 	if (__predict_false(!(e))) {					\
    290        	       pthread__errorfunc(__FILE__, __LINE__, __func__, msg);	\
    291 	       return (err);						\
    292 	} 								\
    293         } while (0)
    294 
    295 void 	*pthread_tsd_init(size_t *) PTHREAD_HIDE;
    296 void	pthread__destroy_tsd(pthread_t) PTHREAD_HIDE;
    297 void	pthread__copy_tsd(pthread_t) PTHREAD_HIDE;
    298 
    299 __dead void	pthread__assertfunc(const char *, int, const char *, const char *)
    300 			    PTHREAD_HIDE;
    301 void	pthread__errorfunc(const char *, int, const char *, const char *, ...)
    302 			    __printflike(4, 5) PTHREAD_HIDE;
    303 char	*pthread__getenv(const char *) PTHREAD_HIDE;
    304 __dead void	pthread__cancelled(void) PTHREAD_HIDE;
    305 void	pthread__mutex_deferwake(pthread_t, pthread_mutex_t *,
    306     struct pthread__waiter *) PTHREAD_HIDE;
    307 int	pthread__checkpri(int) PTHREAD_HIDE;
    308 int	pthread__add_specific(pthread_t, pthread_key_t, const void *) PTHREAD_HIDE;
    309 
    310 #ifndef pthread__smt_pause
    311 #define	pthread__smt_pause()	__nothing
    312 #endif
    313 #ifndef pthread__smt_wait
    314 #define	pthread__smt_wait()	__nothing
    315 #endif
    316 #ifndef pthread__smt_wake
    317 #define	pthread__smt_wake()	__nothing
    318 #endif
    319 
    320 /*
    321  * Bits in the owner field of the lock that indicate lock state.  If the
    322  * WRITE_LOCKED bit is clear, then the owner field is actually a count of
    323  * the number of readers.
    324  */
    325 #define	RW_HAS_WAITERS		0x01	/* lock has waiters */
    326 #define	RW_WRITE_WANTED		0x02	/* >= 1 waiter is a writer */
    327 #define	RW_WRITE_LOCKED		0x04	/* lock is currently write locked */
    328 #define	RW_UNUSED		0x08	/* currently unused */
    329 
    330 #define	RW_FLAGMASK		0x0f
    331 
    332 #define	RW_READ_COUNT_SHIFT	4
    333 #define	RW_READ_INCR		(1 << RW_READ_COUNT_SHIFT)
    334 #define	RW_THREAD		((uintptr_t)-RW_READ_INCR)
    335 
    336 #endif /* _LIB_PTHREAD_INT_H */
    337