#include <sys/cdefs.h>
__RCSID("$NetBSD: pthread_mutex.c,v 1.84 2026/05/06 09:03:08 yamt Exp $");
#include "../../common/lib/libc/atomic/atomic_op_namespace.h"
#include <sys/types.h>
#include <sys/lwpctl.h>
#include <sys/sched.h>
#include <sys/lock.h>
#include <errno.h>
#include <limits.h>
#include <stdlib.h>
#include <time.h>
#include <string.h>
#include <stdio.h>
#include "pthread.h"
#include "pthread_int.h"
#include "reentrant.h"
#define MUTEX_RECURSIVE_BIT ((uintptr_t)0x02)
#define MUTEX_PROTECT_BIT ((uintptr_t)0x08)
#define MUTEX_THREAD ((uintptr_t)~0x0f)
#define MUTEX_RECURSIVE(x) ((uintptr_t)(x) & MUTEX_RECURSIVE_BIT)
#define MUTEX_PROTECT(x) ((uintptr_t)(x) & MUTEX_PROTECT_BIT)
#define MUTEX_OWNER(x) ((uintptr_t)(x) & MUTEX_THREAD)
#define MUTEX_GET_TYPE(x) \
((int)(((uintptr_t)(x) & 0x000000ff) >> 0))
#define MUTEX_SET_TYPE(x, t) \
(x) = (void *)(((uintptr_t)(x) & ~0x000000ff) | ((t) << 0))
#define MUTEX_GET_PROTOCOL(x) \
((int)(((uintptr_t)(x) & 0x0000ff00) >> 8))
#define MUTEX_SET_PROTOCOL(x, p) \
(x) = (void *)(((uintptr_t)(x) & ~0x0000ff00) | ((p) << 8))
#define MUTEX_GET_CEILING(x) \
((int)(((uintptr_t)(x) & 0x00ff0000) >> 16))
#define MUTEX_SET_CEILING(x, c) \
(x) = (void *)(((uintptr_t)(x) & ~0x00ff0000) | ((c) << 16))
#if __GNUC_PREREQ__(3, 0)
#define NOINLINE __attribute ((noinline))
#else
#define NOINLINE
#endif
static void pthread__mutex_wakeup(pthread_t, struct pthread__waiter *);
static int pthread__mutex_lock_slow(pthread_mutex_t *,
const struct timespec *);
static void pthread__mutex_pause(void);
int _pthread_mutex_held_np(pthread_mutex_t *);
pthread_t _pthread_mutex_owner_np(pthread_mutex_t *);
__weak_alias(pthread_mutex_held_np,_pthread_mutex_held_np)
__weak_alias(pthread_mutex_owner_np,_pthread_mutex_owner_np)
__strong_alias(__libc_mutex_init,pthread_mutex_init)
__strong_alias(__libc_mutex_lock,pthread_mutex_lock)
__strong_alias(__libc_mutex_trylock,pthread_mutex_trylock)
__strong_alias(__libc_mutex_unlock,pthread_mutex_unlock)
__strong_alias(__libc_mutex_destroy,pthread_mutex_destroy)
__strong_alias(__libc_mutexattr_init,pthread_mutexattr_init)
__strong_alias(__libc_mutexattr_destroy,pthread_mutexattr_destroy)
__strong_alias(__libc_mutexattr_settype,pthread_mutexattr_settype)
int
pthread_mutex_init(pthread_mutex_t *ptm, const pthread_mutexattr_t *attr)
{
uintptr_t type, proto, val, ceil;
#if 0
if (__predict_false(__uselibcstub))
return __libc_mutex_init_stub(ptm, attr);
#endif
pthread__error(EINVAL, "Invalid mutes attribute",
attr == NULL || attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
if (attr == NULL) {
type = PTHREAD_MUTEX_NORMAL;
proto = PTHREAD_PRIO_NONE;
ceil = 0;
} else {
val = (uintptr_t)attr->ptma_private;
type = MUTEX_GET_TYPE(val);
proto = MUTEX_GET_PROTOCOL(val);
ceil = MUTEX_GET_CEILING(val);
}
switch (type) {
case PTHREAD_MUTEX_ERRORCHECK:
__cpu_simple_lock_set(&ptm->ptm_errorcheck);
ptm->ptm_owner = NULL;
break;
case PTHREAD_MUTEX_RECURSIVE:
__cpu_simple_lock_clear(&ptm->ptm_errorcheck);
ptm->ptm_owner = (void *)MUTEX_RECURSIVE_BIT;
break;
default:
__cpu_simple_lock_clear(&ptm->ptm_errorcheck);
ptm->ptm_owner = NULL;
break;
}
switch (proto) {
case PTHREAD_PRIO_PROTECT:
val = (uintptr_t)ptm->ptm_owner;
val |= MUTEX_PROTECT_BIT;
ptm->ptm_owner = (void *)val;
break;
}
ptm->ptm_magic = _PT_MUTEX_MAGIC;
ptm->ptm_waiters = NULL;
ptm->ptm_recursed = 0;
ptm->ptm_ceiling = (unsigned char)ceil;
return 0;
}
int
pthread_mutex_destroy(pthread_mutex_t *ptm)
{
if (__predict_false(__uselibcstub))
return __libc_mutex_destroy_stub(ptm);
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
pthread__error(EBUSY, "Destroying locked mutex",
MUTEX_OWNER(ptm->ptm_owner) == 0);
ptm->ptm_magic = _PT_MUTEX_DEAD;
return 0;
}
int
pthread_mutex_lock(pthread_mutex_t *ptm)
{
pthread_t self;
void *val;
if (__predict_false(__uselibcstub))
return __libc_mutex_lock_stub(ptm);
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
self = pthread__self();
val = atomic_cas_ptr(&ptm->ptm_owner, NULL, self);
if (__predict_true(val == NULL)) {
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
return 0;
}
return pthread__mutex_lock_slow(ptm, NULL);
}
int
pthread_mutex_timedlock(pthread_mutex_t* ptm, const struct timespec *ts)
{
pthread_t self;
void *val;
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
self = pthread__self();
val = atomic_cas_ptr(&ptm->ptm_owner, NULL, self);
if (__predict_true(val == NULL)) {
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
return 0;
}
return pthread__mutex_lock_slow(ptm, ts);
}
NOINLINE static void
pthread__mutex_pause(void)
{
pthread__smt_pause();
}
NOINLINE static void *
pthread__mutex_spin(pthread_mutex_t *ptm, pthread_t owner)
{
pthread_t thread;
unsigned int count, i;
for (count = 2;; owner = ptm->ptm_owner) {
thread = (pthread_t)MUTEX_OWNER(owner);
if (thread == NULL)
break;
if (thread->pt_lwpctl->lc_curcpu == LWPCTL_CPU_NONE)
break;
if (count < 128)
count += count;
for (i = count; i != 0; i--)
pthread__mutex_pause();
}
return owner;
}
NOINLINE static int
pthread__mutex_lock_slow(pthread_mutex_t *ptm, const struct timespec *ts)
{
void *newval, *owner, *next;
struct pthread__waiter waiter;
pthread_t self;
int serrno;
int error;
owner = ptm->ptm_owner;
self = pthread__self();
serrno = errno;
pthread__assert(self->pt_lid != 0);
if (MUTEX_OWNER(owner) == (uintptr_t)self) {
if (MUTEX_RECURSIVE(owner)) {
if (ptm->ptm_recursed == INT_MAX)
return EAGAIN;
ptm->ptm_recursed++;
return 0;
}
if (__SIMPLELOCK_LOCKED_P(&ptm->ptm_errorcheck))
return EDEADLK;
}
if (MUTEX_PROTECT(owner) && _sched_protect(ptm->ptm_ceiling) == -1) {
error = errno;
errno = serrno;
return error;
}
for (;;) {
if (MUTEX_OWNER(owner) == 0) {
newval = (void *)((uintptr_t)self | (uintptr_t)owner);
next = atomic_cas_ptr(&ptm->ptm_owner, owner, newval);
if (__predict_false(next != owner)) {
owner = next;
continue;
}
errno = serrno;
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
return 0;
} else if (MUTEX_OWNER(owner) != (uintptr_t)self) {
owner = pthread__mutex_spin(ptm, owner);
if (MUTEX_OWNER(owner) == 0) {
continue;
}
}
waiter.next = ptm->ptm_waiters;
waiter.lid = self->pt_lid;
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_producer();
#endif
next = atomic_cas_ptr(&ptm->ptm_waiters, waiter.next, &waiter);
if (next != waiter.next) {
owner = ptm->ptm_owner;
continue;
}
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
if (MUTEX_OWNER(ptm->ptm_owner) == 0) {
pthread__mutex_wakeup(self,
atomic_swap_ptr(&ptm->ptm_waiters, NULL));
}
while (waiter.lid != 0) {
error = _lwp_park(CLOCK_REALTIME, TIMER_ABSTIME,
__UNCONST(ts), 0, NULL, NULL);
if (error < 0 && errno == ETIMEDOUT) {
pthread__mutex_wakeup(self,
atomic_swap_ptr(&ptm->ptm_waiters, NULL));
while (waiter.lid != 0) {
(void)_lwp_park(CLOCK_MONOTONIC, 0,
NULL, 0, NULL, NULL);
}
if (MUTEX_PROTECT(owner))
(void)_sched_protect(-1);
errno = serrno;
return ETIMEDOUT;
}
}
owner = ptm->ptm_owner;
}
}
int
pthread_mutex_trylock(pthread_mutex_t *ptm)
{
pthread_t self;
void *val, *new, *next;
if (__predict_false(__uselibcstub))
return __libc_mutex_trylock_stub(ptm);
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
self = pthread__self();
val = atomic_cas_ptr(&ptm->ptm_owner, NULL, self);
if (__predict_true(val == NULL)) {
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
return 0;
}
if (MUTEX_RECURSIVE(val)) {
if (MUTEX_OWNER(val) == 0) {
new = (void *)((uintptr_t)self | (uintptr_t)val);
next = atomic_cas_ptr(&ptm->ptm_owner, val, new);
if (__predict_true(next == val)) {
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
return 0;
}
}
if (MUTEX_OWNER(val) == (uintptr_t)self) {
if (ptm->ptm_recursed == INT_MAX)
return EAGAIN;
ptm->ptm_recursed++;
return 0;
}
}
return EBUSY;
}
int
pthread_mutex_unlock(pthread_mutex_t *ptm)
{
pthread_t self;
void *val, *newval;
int error;
if (__predict_false(__uselibcstub))
return __libc_mutex_unlock_stub(ptm);
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_exit();
#endif
error = 0;
self = pthread__self();
newval = NULL;
val = atomic_cas_ptr(&ptm->ptm_owner, self, newval);
if (__predict_false(val != self)) {
bool weown = (MUTEX_OWNER(val) == (uintptr_t)self);
if (__SIMPLELOCK_LOCKED_P(&ptm->ptm_errorcheck)) {
if (!weown) {
error = EPERM;
newval = val;
} else {
newval = NULL;
}
} else if (MUTEX_RECURSIVE(val)) {
if (!weown) {
error = EPERM;
newval = val;
} else if (ptm->ptm_recursed) {
ptm->ptm_recursed--;
newval = val;
} else {
newval = (pthread_t)MUTEX_RECURSIVE_BIT;
}
} else {
pthread__error(EPERM,
"Unlocking unlocked mutex", (val != NULL));
pthread__error(EPERM,
"Unlocking mutex owned by another thread", weown);
newval = NULL;
}
if (newval != val) {
val = atomic_swap_ptr(&ptm->ptm_owner, newval);
if (__predict_false(MUTEX_PROTECT(val))) {
(void)_sched_protect(-1);
}
}
}
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_enter();
#endif
if (MUTEX_OWNER(newval) == 0 && ptm->ptm_waiters != NULL) {
pthread__mutex_wakeup(self,
atomic_swap_ptr(&ptm->ptm_waiters, NULL));
}
return error;
}
static void
pthread__mutex_wakeup(pthread_t self, struct pthread__waiter *cur)
{
lwpid_t lids[PTHREAD__UNPARK_MAX];
const size_t mlid = pthread__unpark_max;
struct pthread__waiter *next;
size_t nlid;
membar_datadep_consumer();
for (nlid = 0; cur != NULL; cur = next) {
if (nlid == mlid) {
(void)_lwp_unpark_all(lids, nlid, NULL);
nlid = 0;
}
next = cur->next;
pthread__assert(cur->lid != 0);
lids[nlid++] = cur->lid;
membar_exit();
cur->lid = 0;
}
if (nlid == 1) {
(void)_lwp_unpark(lids[0], NULL);
} else if (nlid > 1) {
(void)_lwp_unpark_all(lids, nlid, NULL);
}
}
int
pthread_mutexattr_init(pthread_mutexattr_t *attr)
{
#if 0
if (__predict_false(__uselibcstub))
return __libc_mutexattr_init_stub(attr);
#endif
attr->ptma_magic = _PT_MUTEXATTR_MAGIC;
attr->ptma_private = (void *)PTHREAD_MUTEX_DEFAULT;
return 0;
}
int
pthread_mutexattr_destroy(pthread_mutexattr_t *attr)
{
if (__predict_false(__uselibcstub))
return __libc_mutexattr_destroy_stub(attr);
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
attr->ptma_magic = _PT_MUTEXATTR_DEAD;
return 0;
}
int
pthread_mutexattr_gettype(const pthread_mutexattr_t *attr, int *typep)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
*typep = MUTEX_GET_TYPE(attr->ptma_private);
return 0;
}
int
pthread_mutexattr_settype(pthread_mutexattr_t *attr, int type)
{
if (__predict_false(__uselibcstub))
return __libc_mutexattr_settype_stub(attr, type);
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
switch (type) {
case PTHREAD_MUTEX_NORMAL:
case PTHREAD_MUTEX_ERRORCHECK:
case PTHREAD_MUTEX_RECURSIVE:
MUTEX_SET_TYPE(attr->ptma_private, type);
return 0;
default:
return EINVAL;
}
}
int
pthread_mutexattr_getprotocol(const pthread_mutexattr_t *attr, int*proto)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
*proto = MUTEX_GET_PROTOCOL(attr->ptma_private);
return 0;
}
int
pthread_mutexattr_setprotocol(pthread_mutexattr_t* attr, int proto)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
switch (proto) {
case PTHREAD_PRIO_NONE:
case PTHREAD_PRIO_PROTECT:
MUTEX_SET_PROTOCOL(attr->ptma_private, proto);
return 0;
case PTHREAD_PRIO_INHERIT:
return ENOTSUP;
default:
return EINVAL;
}
}
int
pthread_mutexattr_getprioceiling(const pthread_mutexattr_t *attr, int *ceil)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
*ceil = MUTEX_GET_CEILING(attr->ptma_private);
return 0;
}
int
pthread_mutexattr_setprioceiling(pthread_mutexattr_t *attr, int ceil)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
if (ceil & ~0xff)
return EINVAL;
MUTEX_SET_CEILING(attr->ptma_private, ceil);
return 0;
}
#ifdef _PTHREAD_PSHARED
int
pthread_mutexattr_getpshared(const pthread_mutexattr_t * __restrict attr,
int * __restrict pshared)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
*pshared = PTHREAD_PROCESS_PRIVATE;
return 0;
}
int
pthread_mutexattr_setpshared(pthread_mutexattr_t *attr, int pshared)
{
pthread__error(EINVAL, "Invalid mutex attribute",
attr->ptma_magic == _PT_MUTEXATTR_MAGIC);
switch(pshared) {
case PTHREAD_PROCESS_PRIVATE:
return 0;
case PTHREAD_PROCESS_SHARED:
return ENOSYS;
}
return EINVAL;
}
#endif
void
pthread__mutex_deferwake(pthread_t self, pthread_mutex_t *ptm,
struct pthread__waiter *head)
{
struct pthread__waiter *tail, *n, *o;
pthread__assert(head != NULL);
if (__predict_false(ptm == NULL ||
MUTEX_OWNER(ptm->ptm_owner) != (uintptr_t)self)) {
pthread__mutex_wakeup(self, head);
return;
}
if (atomic_cas_ptr(&ptm->ptm_waiters, NULL, head) == NULL) {
return;
}
for (tail = head; tail->next != NULL; tail = tail->next) {
}
for (o = ptm->ptm_waiters;; o = n) {
tail->next = o;
#ifndef PTHREAD__ATOMIC_IS_MEMBAR
membar_producer();
#endif
n = atomic_cas_ptr(&ptm->ptm_waiters, o, head);
if (__predict_true(n == o)) {
break;
}
}
}
int
pthread_mutex_getprioceiling(const pthread_mutex_t *ptm, int *ceil)
{
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
*ceil = ptm->ptm_ceiling;
return 0;
}
int
pthread_mutex_setprioceiling(pthread_mutex_t *ptm, int ceil, int *old_ceil)
{
int error;
pthread__error(EINVAL, "Invalid mutex",
ptm->ptm_magic == _PT_MUTEX_MAGIC);
error = pthread_mutex_lock(ptm);
if (error == 0) {
*old_ceil = ptm->ptm_ceiling;
ptm->ptm_ceiling = ceil;
pthread_mutex_unlock(ptm);
}
return error;
}
int
_pthread_mutex_held_np(pthread_mutex_t *ptm)
{
return MUTEX_OWNER(ptm->ptm_owner) == (uintptr_t)pthread__self();
}
pthread_t
_pthread_mutex_owner_np(pthread_mutex_t *ptm)
{
return (pthread_t)MUTEX_OWNER(ptm->ptm_owner);
}