222 lines · c
1// -*- C++ -*-2//===----------------------------------------------------------------------===//3//4// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.5// See https://llvm.org/LICENSE.txt for license information.6// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception7//8//===----------------------------------------------------------------------===//9 10#ifndef _LIBCPP___THREAD_SUPPORT_PTHREAD_H11#define _LIBCPP___THREAD_SUPPORT_PTHREAD_H12 13#include <__chrono/convert_to_timespec.h>14#include <__chrono/duration.h>15#include <__config>16#include <ctime>17#include <errno.h>18#include <pthread.h>19#include <sched.h>20 21#ifdef __MVS__22# include <__support/ibm/nanosleep.h>23#endif24 25// Some platforms require <bits/atomic_wide_counter.h> in order for26// PTHREAD_COND_INITIALIZER to be expanded. Normally that would come27// in via <pthread.h>, but it's a non-modular header on those platforms,28// so libc++'s <math.h> usually absorbs atomic_wide_counter.h into the29// module with <math.h> and makes atomic_wide_counter.h invisible.30// Include <math.h> here to work around that.31// This checks wheter a Clang module is built32#if __building_module(std)33# include <math.h>34#endif35 36#ifndef _LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER37# pragma GCC system_header38#endif39 40_LIBCPP_BEGIN_NAMESPACE_STD41 42using __libcpp_timespec_t _LIBCPP_NODEBUG = ::timespec;43 44//45// Mutex46//47typedef pthread_mutex_t __libcpp_mutex_t;48#define _LIBCPP_MUTEX_INITIALIZER PTHREAD_MUTEX_INITIALIZER49 50typedef pthread_mutex_t __libcpp_recursive_mutex_t;51 52inline _LIBCPP_HIDE_FROM_ABI int __libcpp_recursive_mutex_init(__libcpp_recursive_mutex_t* __m) {53 pthread_mutexattr_t __attr;54 int __ec = pthread_mutexattr_init(&__attr);55 if (__ec)56 return __ec;57 __ec = pthread_mutexattr_settype(&__attr, PTHREAD_MUTEX_RECURSIVE);58 if (__ec) {59 pthread_mutexattr_destroy(&__attr);60 return __ec;61 }62 __ec = pthread_mutex_init(__m, &__attr);63 if (__ec) {64 pthread_mutexattr_destroy(&__attr);65 return __ec;66 }67 __ec = pthread_mutexattr_destroy(&__attr);68 if (__ec) {69 pthread_mutex_destroy(__m);70 return __ec;71 }72 return 0;73}74 75inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS int76__libcpp_recursive_mutex_lock(__libcpp_recursive_mutex_t* __m) {77 return pthread_mutex_lock(__m);78}79 80inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS bool81__libcpp_recursive_mutex_trylock(__libcpp_recursive_mutex_t* __m) {82 return pthread_mutex_trylock(__m) == 0;83}84 85inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS int86__libcpp_recursive_mutex_unlock(__libcpp_recursive_mutex_t* __m) {87 return pthread_mutex_unlock(__m);88}89 90inline _LIBCPP_HIDE_FROM_ABI int __libcpp_recursive_mutex_destroy(__libcpp_recursive_mutex_t* __m) {91 return pthread_mutex_destroy(__m);92}93 94inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS int __libcpp_mutex_lock(__libcpp_mutex_t* __m) {95 return pthread_mutex_lock(__m);96}97 98inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS bool __libcpp_mutex_trylock(__libcpp_mutex_t* __m) {99 return pthread_mutex_trylock(__m) == 0;100}101 102inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS int __libcpp_mutex_unlock(__libcpp_mutex_t* __m) {103 return pthread_mutex_unlock(__m);104}105 106inline _LIBCPP_HIDE_FROM_ABI int __libcpp_mutex_destroy(__libcpp_mutex_t* __m) { return pthread_mutex_destroy(__m); }107 108//109// Condition Variable110//111typedef pthread_cond_t __libcpp_condvar_t;112#define _LIBCPP_CONDVAR_INITIALIZER PTHREAD_COND_INITIALIZER113 114inline _LIBCPP_HIDE_FROM_ABI int __libcpp_condvar_signal(__libcpp_condvar_t* __cv) { return pthread_cond_signal(__cv); }115 116inline _LIBCPP_HIDE_FROM_ABI int __libcpp_condvar_broadcast(__libcpp_condvar_t* __cv) {117 return pthread_cond_broadcast(__cv);118}119 120inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS int121__libcpp_condvar_wait(__libcpp_condvar_t* __cv, __libcpp_mutex_t* __m) {122 return pthread_cond_wait(__cv, __m);123}124 125inline _LIBCPP_HIDE_FROM_ABI _LIBCPP_NO_THREAD_SAFETY_ANALYSIS int126__libcpp_condvar_timedwait(__libcpp_condvar_t* __cv, __libcpp_mutex_t* __m, __libcpp_timespec_t* __ts) {127 return pthread_cond_timedwait(__cv, __m, __ts);128}129 130inline _LIBCPP_HIDE_FROM_ABI int __libcpp_condvar_destroy(__libcpp_condvar_t* __cv) {131 return pthread_cond_destroy(__cv);132}133 134//135// Execute once136//137typedef pthread_once_t __libcpp_exec_once_flag;138#define _LIBCPP_EXEC_ONCE_INITIALIZER PTHREAD_ONCE_INIT139 140inline _LIBCPP_HIDE_FROM_ABI int __libcpp_execute_once(__libcpp_exec_once_flag* __flag, void (*__init_routine)()) {141 return pthread_once(__flag, __init_routine);142}143 144//145// Thread id146//147#if defined(__MVS__)148typedef unsigned long long __libcpp_thread_id;149#else150typedef pthread_t __libcpp_thread_id;151#endif152 153// Returns non-zero if the thread ids are equal, otherwise 0154inline _LIBCPP_HIDE_FROM_ABI bool __libcpp_thread_id_equal(__libcpp_thread_id __t1, __libcpp_thread_id __t2) {155 return __t1 == __t2;156}157 158// Returns non-zero if t1 < t2, otherwise 0159inline _LIBCPP_HIDE_FROM_ABI bool __libcpp_thread_id_less(__libcpp_thread_id __t1, __libcpp_thread_id __t2) {160 return __t1 < __t2;161}162 163//164// Thread165//166#define _LIBCPP_NULL_THREAD ((__libcpp_thread_t()))167typedef pthread_t __libcpp_thread_t;168 169inline _LIBCPP_HIDE_FROM_ABI __libcpp_thread_id __libcpp_thread_get_id(const __libcpp_thread_t* __t) {170#if defined(__MVS__)171 return __t->__;172#else173 return *__t;174#endif175}176 177inline _LIBCPP_HIDE_FROM_ABI bool __libcpp_thread_isnull(const __libcpp_thread_t* __t) {178 return __libcpp_thread_get_id(__t) == 0;179}180 181inline _LIBCPP_HIDE_FROM_ABI int __libcpp_thread_create(__libcpp_thread_t* __t, void* (*__func)(void*), void* __arg) {182 return pthread_create(__t, nullptr, __func, __arg);183}184 185inline _LIBCPP_HIDE_FROM_ABI __libcpp_thread_id __libcpp_thread_get_current_id() {186 const __libcpp_thread_t __current_thread = pthread_self();187 return __libcpp_thread_get_id(&__current_thread);188}189 190inline _LIBCPP_HIDE_FROM_ABI int __libcpp_thread_join(__libcpp_thread_t* __t) { return pthread_join(*__t, nullptr); }191 192inline _LIBCPP_HIDE_FROM_ABI int __libcpp_thread_detach(__libcpp_thread_t* __t) { return pthread_detach(*__t); }193 194inline _LIBCPP_HIDE_FROM_ABI void __libcpp_thread_yield() { sched_yield(); }195 196inline _LIBCPP_HIDE_FROM_ABI void __libcpp_thread_sleep_for(const chrono::nanoseconds& __ns) {197 __libcpp_timespec_t __ts = std::__convert_to_timespec<__libcpp_timespec_t>(__ns);198 while (nanosleep(&__ts, &__ts) == -1 && errno == EINTR)199 ;200}201 202//203// Thread local storage204//205#define _LIBCPP_TLS_DESTRUCTOR_CC /* nothing */206 207typedef pthread_key_t __libcpp_tls_key;208 209inline _LIBCPP_HIDE_FROM_ABI int __libcpp_tls_create(__libcpp_tls_key* __key, void (*__at_exit)(void*)) {210 return pthread_key_create(__key, __at_exit);211}212 213inline _LIBCPP_HIDE_FROM_ABI void* __libcpp_tls_get(__libcpp_tls_key __key) { return pthread_getspecific(__key); }214 215inline _LIBCPP_HIDE_FROM_ABI int __libcpp_tls_set(__libcpp_tls_key __key, void* __p) {216 return pthread_setspecific(__key, __p);217}218 219_LIBCPP_END_NAMESPACE_STD220 221#endif // _LIBCPP___THREAD_SUPPORT_PTHREAD_H222