xref: /freebsd/contrib/llvm-project/libcxx/include/mutex (revision 5ffd83dbcc34f10e07f6d3e968ae6365869615f4) 
1// -*- C++ -*-
2//===--------------------------- mutex ------------------------------------===//
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-exception
7//
8//===----------------------------------------------------------------------===//
9
10#ifndef _LIBCPP_MUTEX
11#define _LIBCPP_MUTEX
12
13/*
14    mutex synopsis
15
16namespace std
17{
18
19class mutex
20{
21public:
22     constexpr mutex() noexcept;
23     ~mutex();
24
25    mutex(const mutex&) = delete;
26    mutex& operator=(const mutex&) = delete;
27
28    void lock();
29    bool try_lock();
30    void unlock();
31
32    typedef pthread_mutex_t* native_handle_type;
33    native_handle_type native_handle();
34};
35
36class recursive_mutex
37{
38public:
39     recursive_mutex();
40     ~recursive_mutex();
41
42    recursive_mutex(const recursive_mutex&) = delete;
43    recursive_mutex& operator=(const recursive_mutex&) = delete;
44
45    void lock();
46    bool try_lock() noexcept;
47    void unlock();
48
49    typedef pthread_mutex_t* native_handle_type;
50    native_handle_type native_handle();
51};
52
53class timed_mutex
54{
55public:
56     timed_mutex();
57     ~timed_mutex();
58
59    timed_mutex(const timed_mutex&) = delete;
60    timed_mutex& operator=(const timed_mutex&) = delete;
61
62    void lock();
63    bool try_lock();
64    template <class Rep, class Period>
65        bool try_lock_for(const chrono::duration<Rep, Period>& rel_time);
66    template <class Clock, class Duration>
67        bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time);
68    void unlock();
69};
70
71class recursive_timed_mutex
72{
73public:
74     recursive_timed_mutex();
75     ~recursive_timed_mutex();
76
77    recursive_timed_mutex(const recursive_timed_mutex&) = delete;
78    recursive_timed_mutex& operator=(const recursive_timed_mutex&) = delete;
79
80    void lock();
81    bool try_lock() noexcept;
82    template <class Rep, class Period>
83        bool try_lock_for(const chrono::duration<Rep, Period>& rel_time);
84    template <class Clock, class Duration>
85        bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time);
86    void unlock();
87};
88
89struct defer_lock_t { explicit defer_lock_t() = default; };
90struct try_to_lock_t { explicit try_to_lock_t() = default; };
91struct adopt_lock_t { explicit adopt_lock_t() = default; };
92
93inline constexpr defer_lock_t  defer_lock{};
94inline constexpr try_to_lock_t try_to_lock{};
95inline constexpr adopt_lock_t  adopt_lock{};
96
97template <class Mutex>
98class lock_guard
99{
100public:
101    typedef Mutex mutex_type;
102
103    explicit lock_guard(mutex_type& m);
104    lock_guard(mutex_type& m, adopt_lock_t);
105    ~lock_guard();
106
107    lock_guard(lock_guard const&) = delete;
108    lock_guard& operator=(lock_guard const&) = delete;
109};
110
111template <class... MutexTypes>
112class scoped_lock // C++17
113{
114public:
115    using mutex_type = Mutex;  // If MutexTypes... consists of the single type Mutex
116
117    explicit scoped_lock(MutexTypes&... m);
118    scoped_lock(adopt_lock_t, MutexTypes&... m);
119    ~scoped_lock();
120    scoped_lock(scoped_lock const&) = delete;
121    scoped_lock& operator=(scoped_lock const&) = delete;
122private:
123    tuple<MutexTypes&...> pm; // exposition only
124};
125
126template <class Mutex>
127class unique_lock
128{
129public:
130    typedef Mutex mutex_type;
131    unique_lock() noexcept;
132    explicit unique_lock(mutex_type& m);
133    unique_lock(mutex_type& m, defer_lock_t) noexcept;
134    unique_lock(mutex_type& m, try_to_lock_t);
135    unique_lock(mutex_type& m, adopt_lock_t);
136    template <class Clock, class Duration>
137        unique_lock(mutex_type& m, const chrono::time_point<Clock, Duration>& abs_time);
138    template <class Rep, class Period>
139        unique_lock(mutex_type& m, const chrono::duration<Rep, Period>& rel_time);
140    ~unique_lock();
141
142    unique_lock(unique_lock const&) = delete;
143    unique_lock& operator=(unique_lock const&) = delete;
144
145    unique_lock(unique_lock&& u) noexcept;
146    unique_lock& operator=(unique_lock&& u) noexcept;
147
148    void lock();
149    bool try_lock();
150
151    template <class Rep, class Period>
152        bool try_lock_for(const chrono::duration<Rep, Period>& rel_time);
153    template <class Clock, class Duration>
154        bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time);
155
156    void unlock();
157
158    void swap(unique_lock& u) noexcept;
159    mutex_type* release() noexcept;
160
161    bool owns_lock() const noexcept;
162    explicit operator bool () const noexcept;
163    mutex_type* mutex() const noexcept;
164};
165
166template <class Mutex>
167  void swap(unique_lock<Mutex>& x, unique_lock<Mutex>& y) noexcept;
168
169template <class L1, class L2, class... L3>
170  int try_lock(L1&, L2&, L3&...);
171template <class L1, class L2, class... L3>
172  void lock(L1&, L2&, L3&...);
173
174struct once_flag
175{
176    constexpr once_flag() noexcept;
177
178    once_flag(const once_flag&) = delete;
179    once_flag& operator=(const once_flag&) = delete;
180};
181
182template<class Callable, class ...Args>
183  void call_once(once_flag& flag, Callable&& func, Args&&... args);
184
185}  // std
186
187*/
188
189#include <__config>
190#include <__mutex_base>
191#include <cstdint>
192#include <functional>
193#include <memory>
194#ifndef _LIBCPP_CXX03_LANG
195#include <tuple>
196#endif
197#include <version>
198#include <__threading_support>
199
200#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
201#pragma GCC system_header
202#endif
203
204_LIBCPP_PUSH_MACROS
205#include <__undef_macros>
206
207
208_LIBCPP_BEGIN_NAMESPACE_STD
209
210#ifndef _LIBCPP_HAS_NO_THREADS
211
212class _LIBCPP_TYPE_VIS recursive_mutex
213{
214    __libcpp_recursive_mutex_t __m_;
215
216public:
217     recursive_mutex();
218     ~recursive_mutex();
219
220private:
221    recursive_mutex(const recursive_mutex&); // = delete;
222    recursive_mutex& operator=(const recursive_mutex&); // = delete;
223
224public:
225    void lock();
226    bool try_lock() _NOEXCEPT;
227    void unlock()  _NOEXCEPT;
228
229    typedef __libcpp_recursive_mutex_t* native_handle_type;
230
231    _LIBCPP_INLINE_VISIBILITY
232    native_handle_type native_handle() {return &__m_;}
233};
234
235class _LIBCPP_TYPE_VIS timed_mutex
236{
237    mutex              __m_;
238    condition_variable __cv_;
239    bool               __locked_;
240public:
241     timed_mutex();
242     ~timed_mutex();
243
244private:
245    timed_mutex(const timed_mutex&); // = delete;
246    timed_mutex& operator=(const timed_mutex&); // = delete;
247
248public:
249    void lock();
250    bool try_lock() _NOEXCEPT;
251    template <class _Rep, class _Period>
252        _LIBCPP_INLINE_VISIBILITY
253        bool try_lock_for(const chrono::duration<_Rep, _Period>& __d)
254            {return try_lock_until(chrono::steady_clock::now() + __d);}
255    template <class _Clock, class _Duration>
256        _LIBCPP_METHOD_TEMPLATE_IMPLICIT_INSTANTIATION_VIS
257        bool try_lock_until(const chrono::time_point<_Clock, _Duration>& __t);
258    void unlock() _NOEXCEPT;
259};
260
261template <class _Clock, class _Duration>
262bool
263timed_mutex::try_lock_until(const chrono::time_point<_Clock, _Duration>& __t)
264{
265    using namespace chrono;
266    unique_lock<mutex> __lk(__m_);
267    bool no_timeout = _Clock::now() < __t;
268    while (no_timeout && __locked_)
269        no_timeout = __cv_.wait_until(__lk, __t) == cv_status::no_timeout;
270    if (!__locked_)
271    {
272        __locked_ = true;
273        return true;
274    }
275    return false;
276}
277
278class _LIBCPP_TYPE_VIS recursive_timed_mutex
279{
280    mutex              __m_;
281    condition_variable __cv_;
282    size_t             __count_;
283    __thread_id        __id_;
284public:
285     recursive_timed_mutex();
286     ~recursive_timed_mutex();
287
288private:
289    recursive_timed_mutex(const recursive_timed_mutex&); // = delete;
290    recursive_timed_mutex& operator=(const recursive_timed_mutex&); // = delete;
291
292public:
293    void lock();
294    bool try_lock() _NOEXCEPT;
295    template <class _Rep, class _Period>
296        _LIBCPP_INLINE_VISIBILITY
297        bool try_lock_for(const chrono::duration<_Rep, _Period>& __d)
298            {return try_lock_until(chrono::steady_clock::now() + __d);}
299    template <class _Clock, class _Duration>
300        _LIBCPP_METHOD_TEMPLATE_IMPLICIT_INSTANTIATION_VIS
301        bool try_lock_until(const chrono::time_point<_Clock, _Duration>& __t);
302    void unlock() _NOEXCEPT;
303};
304
305template <class _Clock, class _Duration>
306bool
307recursive_timed_mutex::try_lock_until(const chrono::time_point<_Clock, _Duration>& __t)
308{
309    using namespace chrono;
310    __thread_id __id = this_thread::get_id();
311    unique_lock<mutex> lk(__m_);
312    if (__id == __id_)
313    {
314        if (__count_ == numeric_limits<size_t>::max())
315            return false;
316        ++__count_;
317        return true;
318    }
319    bool no_timeout = _Clock::now() < __t;
320    while (no_timeout && __count_ != 0)
321        no_timeout = __cv_.wait_until(lk, __t) == cv_status::no_timeout;
322    if (__count_ == 0)
323    {
324        __count_ = 1;
325        __id_ = __id;
326        return true;
327    }
328    return false;
329}
330
331template <class _L0, class _L1>
332int
333try_lock(_L0& __l0, _L1& __l1)
334{
335    unique_lock<_L0> __u0(__l0, try_to_lock);
336    if (__u0.owns_lock())
337    {
338        if (__l1.try_lock())
339        {
340            __u0.release();
341            return -1;
342        }
343        else
344            return 1;
345    }
346    return 0;
347}
348
349#ifndef _LIBCPP_CXX03_LANG
350
351template <class _L0, class _L1, class _L2, class... _L3>
352int
353try_lock(_L0& __l0, _L1& __l1, _L2& __l2, _L3&... __l3)
354{
355    int __r = 0;
356    unique_lock<_L0> __u0(__l0, try_to_lock);
357    if (__u0.owns_lock())
358    {
359        __r = try_lock(__l1, __l2, __l3...);
360        if (__r == -1)
361            __u0.release();
362        else
363            ++__r;
364    }
365    return __r;
366}
367
368#endif  // _LIBCPP_CXX03_LANG
369
370template <class _L0, class _L1>
371void
372lock(_L0& __l0, _L1& __l1)
373{
374    while (true)
375    {
376        {
377            unique_lock<_L0> __u0(__l0);
378            if (__l1.try_lock())
379            {
380                __u0.release();
381                break;
382            }
383        }
384        __libcpp_thread_yield();
385        {
386            unique_lock<_L1> __u1(__l1);
387            if (__l0.try_lock())
388            {
389                __u1.release();
390                break;
391            }
392        }
393        __libcpp_thread_yield();
394    }
395}
396
397#ifndef _LIBCPP_CXX03_LANG
398
399template <class _L0, class _L1, class _L2, class ..._L3>
400void
401__lock_first(int __i, _L0& __l0, _L1& __l1, _L2& __l2, _L3& ...__l3)
402{
403    while (true)
404    {
405        switch (__i)
406        {
407        case 0:
408            {
409                unique_lock<_L0> __u0(__l0);
410                __i = try_lock(__l1, __l2, __l3...);
411                if (__i == -1)
412                {
413                    __u0.release();
414                    return;
415                }
416            }
417            ++__i;
418            __libcpp_thread_yield();
419            break;
420        case 1:
421            {
422                unique_lock<_L1> __u1(__l1);
423                __i = try_lock(__l2, __l3..., __l0);
424                if (__i == -1)
425                {
426                    __u1.release();
427                    return;
428                }
429            }
430            if (__i == sizeof...(_L3) + 1)
431                __i = 0;
432            else
433                __i += 2;
434            __libcpp_thread_yield();
435            break;
436        default:
437            __lock_first(__i - 2, __l2, __l3..., __l0, __l1);
438            return;
439        }
440    }
441}
442
443template <class _L0, class _L1, class _L2, class ..._L3>
444inline _LIBCPP_INLINE_VISIBILITY
445void
446lock(_L0& __l0, _L1& __l1, _L2& __l2, _L3& ...__l3)
447{
448    __lock_first(0, __l0, __l1, __l2, __l3...);
449}
450
451template <class _L0>
452inline _LIBCPP_INLINE_VISIBILITY
453void __unlock(_L0& __l0) {
454    __l0.unlock();
455}
456
457template <class _L0, class _L1>
458inline _LIBCPP_INLINE_VISIBILITY
459void __unlock(_L0& __l0, _L1& __l1) {
460    __l0.unlock();
461    __l1.unlock();
462}
463
464template <class _L0, class _L1, class _L2, class ..._L3>
465inline _LIBCPP_INLINE_VISIBILITY
466void __unlock(_L0& __l0, _L1& __l1, _L2& __l2, _L3&... __l3) {
467    __l0.unlock();
468    __l1.unlock();
469    _VSTD::__unlock(__l2, __l3...);
470}
471
472#endif  // _LIBCPP_CXX03_LANG
473
474#if _LIBCPP_STD_VER > 14
475template <class ..._Mutexes>
476class _LIBCPP_TEMPLATE_VIS scoped_lock;
477
478template <>
479class _LIBCPP_TEMPLATE_VIS scoped_lock<> {
480public:
481    explicit scoped_lock() {}
482    ~scoped_lock() = default;
483
484    _LIBCPP_INLINE_VISIBILITY
485    explicit scoped_lock(adopt_lock_t) {}
486
487    scoped_lock(scoped_lock const&) = delete;
488    scoped_lock& operator=(scoped_lock const&) = delete;
489};
490
491template <class _Mutex>
492class _LIBCPP_TEMPLATE_VIS _LIBCPP_THREAD_SAFETY_ANNOTATION(scoped_lockable) scoped_lock<_Mutex> {
493public:
494    typedef _Mutex  mutex_type;
495private:
496    mutex_type& __m_;
497public:
498    explicit scoped_lock(mutex_type & __m) _LIBCPP_THREAD_SAFETY_ANNOTATION(acquire_capability(__m))
499        : __m_(__m) {__m_.lock();}
500
501    ~scoped_lock() _LIBCPP_THREAD_SAFETY_ANNOTATION(release_capability()) {__m_.unlock();}
502
503    _LIBCPP_INLINE_VISIBILITY
504    explicit scoped_lock(adopt_lock_t, mutex_type& __m) _LIBCPP_THREAD_SAFETY_ANNOTATION(requires_capability(__m))
505        : __m_(__m) {}
506
507    scoped_lock(scoped_lock const&) = delete;
508    scoped_lock& operator=(scoped_lock const&) = delete;
509};
510
511template <class ..._MArgs>
512class _LIBCPP_TEMPLATE_VIS scoped_lock
513{
514    static_assert(sizeof...(_MArgs) > 1, "At least 2 lock types required");
515    typedef tuple<_MArgs&...> _MutexTuple;
516
517public:
518    _LIBCPP_INLINE_VISIBILITY
519    explicit scoped_lock(_MArgs&... __margs)
520      : __t_(__margs...)
521    {
522        _VSTD::lock(__margs...);
523    }
524
525    _LIBCPP_INLINE_VISIBILITY
526    scoped_lock(adopt_lock_t, _MArgs&... __margs)
527        : __t_(__margs...)
528    {
529    }
530
531    _LIBCPP_INLINE_VISIBILITY
532    ~scoped_lock() {
533        typedef typename __make_tuple_indices<sizeof...(_MArgs)>::type _Indices;
534        __unlock_unpack(_Indices{}, __t_);
535    }
536
537    scoped_lock(scoped_lock const&) = delete;
538    scoped_lock& operator=(scoped_lock const&) = delete;
539
540private:
541    template <size_t ..._Indx>
542    _LIBCPP_INLINE_VISIBILITY
543    static void __unlock_unpack(__tuple_indices<_Indx...>, _MutexTuple& __mt) {
544        _VSTD::__unlock(_VSTD::get<_Indx>(__mt)...);
545    }
546
547    _MutexTuple __t_;
548};
549
550#endif // _LIBCPP_STD_VER > 14
551#endif // !_LIBCPP_HAS_NO_THREADS
552
553struct _LIBCPP_TEMPLATE_VIS once_flag;
554
555#ifndef _LIBCPP_CXX03_LANG
556
557template<class _Callable, class... _Args>
558_LIBCPP_INLINE_VISIBILITY
559void call_once(once_flag&, _Callable&&, _Args&&...);
560
561#else  // _LIBCPP_CXX03_LANG
562
563template<class _Callable>
564_LIBCPP_INLINE_VISIBILITY
565void call_once(once_flag&, _Callable&);
566
567template<class _Callable>
568_LIBCPP_INLINE_VISIBILITY
569void call_once(once_flag&, const _Callable&);
570
571#endif  // _LIBCPP_CXX03_LANG
572
573struct _LIBCPP_TEMPLATE_VIS once_flag
574{
575    _LIBCPP_INLINE_VISIBILITY
576    _LIBCPP_CONSTEXPR
577        once_flag() _NOEXCEPT : __state_(0) {}
578
579#if defined(_LIBCPP_ABI_MICROSOFT)
580   typedef uintptr_t _State_type;
581#else
582   typedef unsigned long _State_type;
583#endif
584
585
586private:
587    once_flag(const once_flag&); // = delete;
588    once_flag& operator=(const once_flag&); // = delete;
589
590    _State_type __state_;
591
592#ifndef _LIBCPP_CXX03_LANG
593    template<class _Callable, class... _Args>
594    friend
595    void call_once(once_flag&, _Callable&&, _Args&&...);
596#else  // _LIBCPP_CXX03_LANG
597    template<class _Callable>
598    friend
599    void call_once(once_flag&, _Callable&);
600
601    template<class _Callable>
602    friend
603    void call_once(once_flag&, const _Callable&);
604#endif  // _LIBCPP_CXX03_LANG
605};
606
607#ifndef _LIBCPP_CXX03_LANG
608
609template <class _Fp>
610class __call_once_param
611{
612    _Fp& __f_;
613public:
614    _LIBCPP_INLINE_VISIBILITY
615    explicit __call_once_param(_Fp& __f) : __f_(__f) {}
616
617    _LIBCPP_INLINE_VISIBILITY
618    void operator()()
619    {
620        typedef typename __make_tuple_indices<tuple_size<_Fp>::value, 1>::type _Index;
621        __execute(_Index());
622    }
623
624private:
625    template <size_t ..._Indices>
626    _LIBCPP_INLINE_VISIBILITY
627    void __execute(__tuple_indices<_Indices...>)
628    {
629        __invoke(_VSTD::get<0>(_VSTD::move(__f_)), _VSTD::get<_Indices>(_VSTD::move(__f_))...);
630    }
631};
632
633#else
634
635template <class _Fp>
636class __call_once_param
637{
638    _Fp& __f_;
639public:
640    _LIBCPP_INLINE_VISIBILITY
641    explicit __call_once_param(_Fp& __f) : __f_(__f) {}
642
643    _LIBCPP_INLINE_VISIBILITY
644    void operator()()
645    {
646        __f_();
647    }
648};
649
650#endif
651
652template <class _Fp>
653void _LIBCPP_INLINE_VISIBILITY
654__call_once_proxy(void* __vp)
655{
656    __call_once_param<_Fp>* __p = static_cast<__call_once_param<_Fp>*>(__vp);
657    (*__p)();
658}
659
660_LIBCPP_FUNC_VIS void __call_once(volatile once_flag::_State_type&, void*,
661                                  void (*)(void*));
662
663#ifndef _LIBCPP_CXX03_LANG
664
665template<class _Callable, class... _Args>
666inline _LIBCPP_INLINE_VISIBILITY
667void
668call_once(once_flag& __flag, _Callable&& __func, _Args&&... __args)
669{
670    if (__libcpp_acquire_load(&__flag.__state_) != ~once_flag::_State_type(0))
671    {
672        typedef tuple<_Callable&&, _Args&&...> _Gp;
673        _Gp __f(_VSTD::forward<_Callable>(__func), _VSTD::forward<_Args>(__args)...);
674        __call_once_param<_Gp> __p(__f);
675        __call_once(__flag.__state_, &__p, &__call_once_proxy<_Gp>);
676    }
677}
678
679#else  // _LIBCPP_CXX03_LANG
680
681template<class _Callable>
682inline _LIBCPP_INLINE_VISIBILITY
683void
684call_once(once_flag& __flag, _Callable& __func)
685{
686    if (__libcpp_acquire_load(&__flag.__state_) != ~once_flag::_State_type(0))
687    {
688        __call_once_param<_Callable> __p(__func);
689        __call_once(__flag.__state_, &__p, &__call_once_proxy<_Callable>);
690    }
691}
692
693template<class _Callable>
694inline _LIBCPP_INLINE_VISIBILITY
695void
696call_once(once_flag& __flag, const _Callable& __func)
697{
698    if (__libcpp_acquire_load(&__flag.__state_) != ~once_flag::_State_type(0))
699    {
700        __call_once_param<const _Callable> __p(__func);
701        __call_once(__flag.__state_, &__p, &__call_once_proxy<const _Callable>);
702    }
703}
704
705#endif  // _LIBCPP_CXX03_LANG
706
707_LIBCPP_END_NAMESPACE_STD
708
709_LIBCPP_POP_MACROS
710
711#endif  // _LIBCPP_MUTEX
712