diff options
author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
---|---|---|
committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /contrib/libs/cxxsupp/libcxx/include/mutex | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'contrib/libs/cxxsupp/libcxx/include/mutex')
-rw-r--r-- | contrib/libs/cxxsupp/libcxx/include/mutex | 737 |
1 files changed, 737 insertions, 0 deletions
diff --git a/contrib/libs/cxxsupp/libcxx/include/mutex b/contrib/libs/cxxsupp/libcxx/include/mutex new file mode 100644 index 0000000000..e815dda502 --- /dev/null +++ b/contrib/libs/cxxsupp/libcxx/include/mutex @@ -0,0 +1,737 @@ +// -*- C++ -*- +//===--------------------------- mutex ------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef _LIBCPP_MUTEX +#define _LIBCPP_MUTEX + +/* + mutex synopsis + +namespace std +{ + +class mutex +{ +public: + constexpr mutex() noexcept; + ~mutex(); + + mutex(const mutex&) = delete; + mutex& operator=(const mutex&) = delete; + + void lock(); + bool try_lock(); + void unlock(); + + typedef pthread_mutex_t* native_handle_type; + native_handle_type native_handle(); +}; + +class recursive_mutex +{ +public: + recursive_mutex(); + ~recursive_mutex(); + + recursive_mutex(const recursive_mutex&) = delete; + recursive_mutex& operator=(const recursive_mutex&) = delete; + + void lock(); + bool try_lock() noexcept; + void unlock(); + + typedef pthread_mutex_t* native_handle_type; + native_handle_type native_handle(); +}; + +class timed_mutex +{ +public: + timed_mutex(); + ~timed_mutex(); + + timed_mutex(const timed_mutex&) = delete; + timed_mutex& operator=(const timed_mutex&) = delete; + + void lock(); + bool try_lock(); + template <class Rep, class Period> + bool try_lock_for(const chrono::duration<Rep, Period>& rel_time); + template <class Clock, class Duration> + bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time); + void unlock(); +}; + +class recursive_timed_mutex +{ +public: + recursive_timed_mutex(); + ~recursive_timed_mutex(); + + recursive_timed_mutex(const recursive_timed_mutex&) = delete; + recursive_timed_mutex& operator=(const recursive_timed_mutex&) = delete; + + void lock(); + bool try_lock() noexcept; + template <class Rep, class Period> + bool try_lock_for(const chrono::duration<Rep, Period>& rel_time); + template <class Clock, class Duration> + bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time); + void unlock(); +}; + +struct defer_lock_t { explicit defer_lock_t() = default; }; +struct try_to_lock_t { explicit try_to_lock_t() = default; }; +struct adopt_lock_t { explicit adopt_lock_t() = default; }; + +inline constexpr defer_lock_t defer_lock{}; +inline constexpr try_to_lock_t try_to_lock{}; +inline constexpr adopt_lock_t adopt_lock{}; + +template <class Mutex> +class lock_guard +{ +public: + typedef Mutex mutex_type; + + explicit lock_guard(mutex_type& m); + lock_guard(mutex_type& m, adopt_lock_t); + ~lock_guard(); + + lock_guard(lock_guard const&) = delete; + lock_guard& operator=(lock_guard const&) = delete; +}; + +template <class... MutexTypes> +class scoped_lock // C++17 +{ +public: + using mutex_type = Mutex; // If MutexTypes... consists of the single type Mutex + + explicit scoped_lock(MutexTypes&... m); + scoped_lock(adopt_lock_t, MutexTypes&... m); + ~scoped_lock(); + scoped_lock(scoped_lock const&) = delete; + scoped_lock& operator=(scoped_lock const&) = delete; +private: + tuple<MutexTypes&...> pm; // exposition only +}; + +template <class Mutex> +class unique_lock +{ +public: + typedef Mutex mutex_type; + unique_lock() noexcept; + explicit unique_lock(mutex_type& m); + unique_lock(mutex_type& m, defer_lock_t) noexcept; + unique_lock(mutex_type& m, try_to_lock_t); + unique_lock(mutex_type& m, adopt_lock_t); + template <class Clock, class Duration> + unique_lock(mutex_type& m, const chrono::time_point<Clock, Duration>& abs_time); + template <class Rep, class Period> + unique_lock(mutex_type& m, const chrono::duration<Rep, Period>& rel_time); + ~unique_lock(); + + unique_lock(unique_lock const&) = delete; + unique_lock& operator=(unique_lock const&) = delete; + + unique_lock(unique_lock&& u) noexcept; + unique_lock& operator=(unique_lock&& u) noexcept; + + void lock(); + bool try_lock(); + + template <class Rep, class Period> + bool try_lock_for(const chrono::duration<Rep, Period>& rel_time); + template <class Clock, class Duration> + bool try_lock_until(const chrono::time_point<Clock, Duration>& abs_time); + + void unlock(); + + void swap(unique_lock& u) noexcept; + mutex_type* release() noexcept; + + bool owns_lock() const noexcept; + explicit operator bool () const noexcept; + mutex_type* mutex() const noexcept; +}; + +template <class Mutex> + void swap(unique_lock<Mutex>& x, unique_lock<Mutex>& y) noexcept; + +template <class L1, class L2, class... L3> + int try_lock(L1&, L2&, L3&...); +template <class L1, class L2, class... L3> + void lock(L1&, L2&, L3&...); + +struct once_flag +{ + constexpr once_flag() noexcept; + + once_flag(const once_flag&) = delete; + once_flag& operator=(const once_flag&) = delete; +}; + +template<class Callable, class ...Args> + void call_once(once_flag& flag, Callable&& func, Args&&... args); + +} // std + +*/ + +#include <__config> +#include <__mutex_base> +#include <__threading_support> +#include <__utility/forward.h> +#include <cstdint> +#include <functional> +#include <memory> +#ifndef _LIBCPP_CXX03_LANG +# include <tuple> +#endif +#include <version> + +#if !defined(_LIBCPP_HAS_NO_THREADS) && !defined(_LIBCPP_CXX03_LANG) +#include <atomic> +#endif + +#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) +#pragma GCC system_header +#endif + +_LIBCPP_PUSH_MACROS +#include <__undef_macros> + + +_LIBCPP_BEGIN_NAMESPACE_STD + +#ifndef _LIBCPP_HAS_NO_THREADS + +class _LIBCPP_TYPE_VIS recursive_mutex +{ + __libcpp_recursive_mutex_t __m_; + +public: + recursive_mutex(); + ~recursive_mutex(); + +private: + recursive_mutex(const recursive_mutex&); // = delete; + recursive_mutex& operator=(const recursive_mutex&); // = delete; + +public: + void lock(); + bool try_lock() _NOEXCEPT; + void unlock() _NOEXCEPT; + + typedef __libcpp_recursive_mutex_t* native_handle_type; + + _LIBCPP_INLINE_VISIBILITY + native_handle_type native_handle() {return &__m_;} +}; + +class _LIBCPP_TYPE_VIS timed_mutex +{ + mutex __m_; + condition_variable __cv_; + bool __locked_; +public: + timed_mutex(); + ~timed_mutex(); + +private: + timed_mutex(const timed_mutex&); // = delete; + timed_mutex& operator=(const timed_mutex&); // = delete; + +public: + void lock(); + bool try_lock() _NOEXCEPT; + template <class _Rep, class _Period> + _LIBCPP_INLINE_VISIBILITY + bool try_lock_for(const chrono::duration<_Rep, _Period>& __d) + {return try_lock_until(chrono::steady_clock::now() + __d);} + template <class _Clock, class _Duration> + _LIBCPP_METHOD_TEMPLATE_IMPLICIT_INSTANTIATION_VIS + bool try_lock_until(const chrono::time_point<_Clock, _Duration>& __t); + void unlock() _NOEXCEPT; +}; + +template <class _Clock, class _Duration> +bool +timed_mutex::try_lock_until(const chrono::time_point<_Clock, _Duration>& __t) +{ + using namespace chrono; + unique_lock<mutex> __lk(__m_); + bool no_timeout = _Clock::now() < __t; + while (no_timeout && __locked_) + no_timeout = __cv_.wait_until(__lk, __t) == cv_status::no_timeout; + if (!__locked_) + { + __locked_ = true; + return true; + } + return false; +} + +class _LIBCPP_TYPE_VIS recursive_timed_mutex +{ + mutex __m_; + condition_variable __cv_; + size_t __count_; + __thread_id __id_; +public: + recursive_timed_mutex(); + ~recursive_timed_mutex(); + +private: + recursive_timed_mutex(const recursive_timed_mutex&); // = delete; + recursive_timed_mutex& operator=(const recursive_timed_mutex&); // = delete; + +public: + void lock(); + bool try_lock() _NOEXCEPT; + template <class _Rep, class _Period> + _LIBCPP_INLINE_VISIBILITY + bool try_lock_for(const chrono::duration<_Rep, _Period>& __d) + {return try_lock_until(chrono::steady_clock::now() + __d);} + template <class _Clock, class _Duration> + _LIBCPP_METHOD_TEMPLATE_IMPLICIT_INSTANTIATION_VIS + bool try_lock_until(const chrono::time_point<_Clock, _Duration>& __t); + void unlock() _NOEXCEPT; +}; + +template <class _Clock, class _Duration> +bool +recursive_timed_mutex::try_lock_until(const chrono::time_point<_Clock, _Duration>& __t) +{ + using namespace chrono; + __thread_id __id = this_thread::get_id(); + unique_lock<mutex> lk(__m_); + if (__id == __id_) + { + if (__count_ == numeric_limits<size_t>::max()) + return false; + ++__count_; + return true; + } + bool no_timeout = _Clock::now() < __t; + while (no_timeout && __count_ != 0) + no_timeout = __cv_.wait_until(lk, __t) == cv_status::no_timeout; + if (__count_ == 0) + { + __count_ = 1; + __id_ = __id; + return true; + } + return false; +} + +template <class _L0, class _L1> +int +try_lock(_L0& __l0, _L1& __l1) +{ + unique_lock<_L0> __u0(__l0, try_to_lock); + if (__u0.owns_lock()) + { + if (__l1.try_lock()) + { + __u0.release(); + return -1; + } + else + return 1; + } + return 0; +} + +#ifndef _LIBCPP_CXX03_LANG + +template <class _L0, class _L1, class _L2, class... _L3> +int +try_lock(_L0& __l0, _L1& __l1, _L2& __l2, _L3&... __l3) +{ + int __r = 0; + unique_lock<_L0> __u0(__l0, try_to_lock); + if (__u0.owns_lock()) + { + __r = try_lock(__l1, __l2, __l3...); + if (__r == -1) + __u0.release(); + else + ++__r; + } + return __r; +} + +#endif // _LIBCPP_CXX03_LANG + +template <class _L0, class _L1> +void +lock(_L0& __l0, _L1& __l1) +{ + while (true) + { + { + unique_lock<_L0> __u0(__l0); + if (__l1.try_lock()) + { + __u0.release(); + break; + } + } + __libcpp_thread_yield(); + { + unique_lock<_L1> __u1(__l1); + if (__l0.try_lock()) + { + __u1.release(); + break; + } + } + __libcpp_thread_yield(); + } +} + +#ifndef _LIBCPP_CXX03_LANG + +template <class _L0, class _L1, class _L2, class ..._L3> +void +__lock_first(int __i, _L0& __l0, _L1& __l1, _L2& __l2, _L3& ...__l3) +{ + while (true) + { + switch (__i) + { + case 0: + { + unique_lock<_L0> __u0(__l0); + __i = try_lock(__l1, __l2, __l3...); + if (__i == -1) + { + __u0.release(); + return; + } + } + ++__i; + __libcpp_thread_yield(); + break; + case 1: + { + unique_lock<_L1> __u1(__l1); + __i = try_lock(__l2, __l3..., __l0); + if (__i == -1) + { + __u1.release(); + return; + } + } + if (__i == sizeof...(_L3) + 1) + __i = 0; + else + __i += 2; + __libcpp_thread_yield(); + break; + default: + __lock_first(__i - 2, __l2, __l3..., __l0, __l1); + return; + } + } +} + +template <class _L0, class _L1, class _L2, class ..._L3> +inline _LIBCPP_INLINE_VISIBILITY +void +lock(_L0& __l0, _L1& __l1, _L2& __l2, _L3& ...__l3) +{ + __lock_first(0, __l0, __l1, __l2, __l3...); +} + +template <class _L0> +inline _LIBCPP_INLINE_VISIBILITY +void __unlock(_L0& __l0) { + __l0.unlock(); +} + +template <class _L0, class _L1> +inline _LIBCPP_INLINE_VISIBILITY +void __unlock(_L0& __l0, _L1& __l1) { + __l0.unlock(); + __l1.unlock(); +} + +template <class _L0, class _L1, class _L2, class ..._L3> +inline _LIBCPP_INLINE_VISIBILITY +void __unlock(_L0& __l0, _L1& __l1, _L2& __l2, _L3&... __l3) { + __l0.unlock(); + __l1.unlock(); + _VSTD::__unlock(__l2, __l3...); +} + +#endif // _LIBCPP_CXX03_LANG + +#if _LIBCPP_STD_VER > 14 +template <class ..._Mutexes> +class _LIBCPP_TEMPLATE_VIS scoped_lock; + +template <> +class _LIBCPP_TEMPLATE_VIS scoped_lock<> { +public: + explicit scoped_lock() {} + ~scoped_lock() = default; + + _LIBCPP_INLINE_VISIBILITY + explicit scoped_lock(adopt_lock_t) {} + + scoped_lock(scoped_lock const&) = delete; + scoped_lock& operator=(scoped_lock const&) = delete; +}; + +template <class _Mutex> +class _LIBCPP_TEMPLATE_VIS _LIBCPP_THREAD_SAFETY_ANNOTATION(scoped_lockable) scoped_lock<_Mutex> { +public: + typedef _Mutex mutex_type; +private: + mutex_type& __m_; +public: + explicit scoped_lock(mutex_type & __m) _LIBCPP_THREAD_SAFETY_ANNOTATION(acquire_capability(__m)) + : __m_(__m) {__m_.lock();} + + ~scoped_lock() _LIBCPP_THREAD_SAFETY_ANNOTATION(release_capability()) {__m_.unlock();} + + _LIBCPP_INLINE_VISIBILITY + explicit scoped_lock(adopt_lock_t, mutex_type& __m) _LIBCPP_THREAD_SAFETY_ANNOTATION(requires_capability(__m)) + : __m_(__m) {} + + scoped_lock(scoped_lock const&) = delete; + scoped_lock& operator=(scoped_lock const&) = delete; +}; + +template <class ..._MArgs> +class _LIBCPP_TEMPLATE_VIS scoped_lock +{ + static_assert(sizeof...(_MArgs) > 1, "At least 2 lock types required"); + typedef tuple<_MArgs&...> _MutexTuple; + +public: + _LIBCPP_INLINE_VISIBILITY + explicit scoped_lock(_MArgs&... __margs) + : __t_(__margs...) + { + _VSTD::lock(__margs...); + } + + _LIBCPP_INLINE_VISIBILITY + scoped_lock(adopt_lock_t, _MArgs&... __margs) + : __t_(__margs...) + { + } + + _LIBCPP_INLINE_VISIBILITY + ~scoped_lock() { + typedef typename __make_tuple_indices<sizeof...(_MArgs)>::type _Indices; + __unlock_unpack(_Indices{}, __t_); + } + + scoped_lock(scoped_lock const&) = delete; + scoped_lock& operator=(scoped_lock const&) = delete; + +private: + template <size_t ..._Indx> + _LIBCPP_INLINE_VISIBILITY + static void __unlock_unpack(__tuple_indices<_Indx...>, _MutexTuple& __mt) { + _VSTD::__unlock(_VSTD::get<_Indx>(__mt)...); + } + + _MutexTuple __t_; +}; + +#endif // _LIBCPP_STD_VER > 14 +#endif // !_LIBCPP_HAS_NO_THREADS + +struct _LIBCPP_TEMPLATE_VIS once_flag; + +#ifndef _LIBCPP_CXX03_LANG + +template<class _Callable, class... _Args> +_LIBCPP_INLINE_VISIBILITY +void call_once(once_flag&, _Callable&&, _Args&&...); + +#else // _LIBCPP_CXX03_LANG + +template<class _Callable> +_LIBCPP_INLINE_VISIBILITY +void call_once(once_flag&, _Callable&); + +template<class _Callable> +_LIBCPP_INLINE_VISIBILITY +void call_once(once_flag&, const _Callable&); + +#endif // _LIBCPP_CXX03_LANG + +struct _LIBCPP_TEMPLATE_VIS once_flag +{ + _LIBCPP_INLINE_VISIBILITY + _LIBCPP_CONSTEXPR + once_flag() _NOEXCEPT : __state_(0) {} + +#if defined(_LIBCPP_ABI_MICROSOFT) + typedef uintptr_t _State_type; +#else + typedef unsigned long _State_type; +#endif + + +private: + once_flag(const once_flag&); // = delete; + once_flag& operator=(const once_flag&); // = delete; + +#if defined(_LIBCPP_ABI_MICROSOFT) + atomic<_State_type> __state_; +#else + _State_type __state_; +#endif + +#ifndef _LIBCPP_CXX03_LANG + template<class _Callable, class... _Args> + friend + void call_once(once_flag&, _Callable&&, _Args&&...); +#else // _LIBCPP_CXX03_LANG + template<class _Callable> + friend + void call_once(once_flag&, _Callable&); + + template<class _Callable> + friend + void call_once(once_flag&, const _Callable&); +#endif // _LIBCPP_CXX03_LANG +}; + +#ifndef _LIBCPP_CXX03_LANG + +template <class _Fp> +class __call_once_param +{ + _Fp& __f_; +public: + _LIBCPP_INLINE_VISIBILITY + explicit __call_once_param(_Fp& __f) : __f_(__f) {} + + _LIBCPP_INLINE_VISIBILITY + void operator()() + { + typedef typename __make_tuple_indices<tuple_size<_Fp>::value, 1>::type _Index; + __execute(_Index()); + } + +private: + template <size_t ..._Indices> + _LIBCPP_INLINE_VISIBILITY + void __execute(__tuple_indices<_Indices...>) + { + _VSTD::__invoke(_VSTD::get<0>(_VSTD::move(__f_)), _VSTD::get<_Indices>(_VSTD::move(__f_))...); + } +}; + +#else + +template <class _Fp> +class __call_once_param +{ + _Fp& __f_; +public: + _LIBCPP_INLINE_VISIBILITY + explicit __call_once_param(_Fp& __f) : __f_(__f) {} + + _LIBCPP_INLINE_VISIBILITY + void operator()() + { + __f_(); + } +}; + +#endif + +template <class _Fp> +void _LIBCPP_INLINE_VISIBILITY +__call_once_proxy(void* __vp) +{ + __call_once_param<_Fp>* __p = static_cast<__call_once_param<_Fp>*>(__vp); + (*__p)(); +} + +#ifdef _LIBCPP_ABI_MICROSOFT +_LIBCPP_FUNC_VIS void __call_once(volatile atomic<once_flag::_State_type>&, void*, + void (*)(void*)); +#else +_LIBCPP_FUNC_VIS void __call_once(volatile once_flag::_State_type&, void*, + void (*)(void*)); +#endif + +#ifndef _LIBCPP_CXX03_LANG + +template<class _Callable, class... _Args> +inline _LIBCPP_INLINE_VISIBILITY +void +call_once(once_flag& __flag, _Callable&& __func, _Args&&... __args) +{ +#if defined(_LIBCPP_ABI_MICROSOFT) + if (__flag.__state_.load(memory_order_acquire) != ~once_flag::_State_type(0)) +#else + if (__libcpp_acquire_load(&__flag.__state_) != ~once_flag::_State_type(0)) +#endif + { + typedef tuple<_Callable&&, _Args&&...> _Gp; + _Gp __f(_VSTD::forward<_Callable>(__func), _VSTD::forward<_Args>(__args)...); + __call_once_param<_Gp> __p(__f); + __call_once(__flag.__state_, &__p, &__call_once_proxy<_Gp>); + } +} + +#else // _LIBCPP_CXX03_LANG + +template<class _Callable> +inline _LIBCPP_INLINE_VISIBILITY +void +call_once(once_flag& __flag, _Callable& __func) +{ +#if defined(_LIBCPP_ABI_MICROSOFT) + if (__flag.__state_.load(memory_order_acquire) != ~once_flag::_State_type(0)) +#else + if (__libcpp_acquire_load(&__flag.__state_) != ~once_flag::_State_type(0)) +#endif + { + __call_once_param<_Callable> __p(__func); + __call_once(__flag.__state_, &__p, &__call_once_proxy<_Callable>); + } +} + +template<class _Callable> +inline _LIBCPP_INLINE_VISIBILITY +void +call_once(once_flag& __flag, const _Callable& __func) +{ +#if defined(_LIBCPP_ABI_MICROSOFT) + if (__flag.__state_.load(memory_order_relaxed) != ~once_flag::_State_type(0)) +#else + if (__flag.__state_ != ~once_flag::_State_type(0)) +#endif + { + __call_once_param<const _Callable> __p(__func); + __call_once(__flag.__state_, &__p, &__call_once_proxy<const _Callable>); + } +} + +#endif // _LIBCPP_CXX03_LANG + +_LIBCPP_END_NAMESPACE_STD + +_LIBCPP_POP_MACROS + +#endif // _LIBCPP_MUTEX |