forked from OSchip/llvm-project
567 lines
13 KiB
C++
567 lines
13 KiB
C++
// -*- C++ -*-
|
|
//===--------------------------- mutex ------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is dual licensed under the MIT and the University of Illinois Open
|
|
// Source Licenses. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#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 {};
|
|
struct try_to_lock_t {};
|
|
struct adopt_lock_t {};
|
|
|
|
constexpr defer_lock_t defer_lock{};
|
|
constexpr try_to_lock_t try_to_lock{};
|
|
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 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 <functional>
|
|
#ifndef _LIBCPP_HAS_NO_VARIADICS
|
|
#include <tuple>
|
|
#endif
|
|
|
|
#include <__undef_min_max>
|
|
|
|
#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
|
|
#pragma GCC system_header
|
|
#endif
|
|
|
|
_LIBCPP_BEGIN_NAMESPACE_STD
|
|
|
|
class _LIBCPP_TYPE_VIS recursive_mutex
|
|
{
|
|
pthread_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 pthread_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>
|
|
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_;
|
|
pthread_t __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>
|
|
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;
|
|
pthread_t __id = pthread_self();
|
|
unique_lock<mutex> lk(__m_);
|
|
if (pthread_equal(__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_HAS_NO_VARIADICS
|
|
|
|
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_HAS_NO_VARIADICS
|
|
|
|
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;
|
|
}
|
|
}
|
|
sched_yield();
|
|
{
|
|
unique_lock<_L1> __u1(__l1);
|
|
if (__l0.try_lock())
|
|
{
|
|
__u1.release();
|
|
break;
|
|
}
|
|
}
|
|
sched_yield();
|
|
}
|
|
}
|
|
|
|
#ifndef _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
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;
|
|
sched_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;
|
|
sched_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...);
|
|
}
|
|
|
|
#endif // _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
struct _LIBCPP_TYPE_VIS once_flag;
|
|
|
|
#ifndef _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
template<class _Callable, class... _Args>
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void call_once(once_flag&, _Callable&&, _Args&&...);
|
|
|
|
#else // _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
template<class _Callable>
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void call_once(once_flag&, _Callable);
|
|
|
|
#endif // _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
struct _LIBCPP_TYPE_VIS once_flag
|
|
{
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
_LIBCPP_CONSTEXPR
|
|
once_flag() _NOEXCEPT : __state_(0) {}
|
|
|
|
private:
|
|
once_flag(const once_flag&); // = delete;
|
|
once_flag& operator=(const once_flag&); // = delete;
|
|
|
|
unsigned long __state_;
|
|
|
|
#ifndef _LIBCPP_HAS_NO_VARIADICS
|
|
template<class _Callable, class... _Args>
|
|
friend
|
|
void call_once(once_flag&, _Callable&&, _Args&&...);
|
|
#else // _LIBCPP_HAS_NO_VARIADICS
|
|
template<class _Callable>
|
|
friend
|
|
void call_once(once_flag&, _Callable);
|
|
#endif // _LIBCPP_HAS_NO_VARIADICS
|
|
};
|
|
|
|
#ifndef _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
template <class _Fp>
|
|
class __call_once_param
|
|
{
|
|
_Fp __f_;
|
|
public:
|
|
#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit __call_once_param(_Fp&& __f) : __f_(_VSTD::move(__f)) {}
|
|
#else
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit __call_once_param(const _Fp& __f) : __f_(__f) {}
|
|
#endif
|
|
|
|
_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...>)
|
|
{
|
|
__invoke(_VSTD::move(_VSTD::get<0>(__f_)), _VSTD::move(_VSTD::get<_Indices>(__f_))...);
|
|
}
|
|
};
|
|
|
|
#else
|
|
|
|
template <class _Fp>
|
|
class __call_once_param
|
|
{
|
|
_Fp __f_;
|
|
public:
|
|
#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit __call_once_param(_Fp&& __f) : __f_(_VSTD::move(__f)) {}
|
|
#else
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit __call_once_param(const _Fp& __f) : __f_(__f) {}
|
|
#endif
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void operator()()
|
|
{
|
|
__f_();
|
|
}
|
|
};
|
|
|
|
#endif
|
|
|
|
template <class _Fp>
|
|
void
|
|
__call_once_proxy(void* __vp)
|
|
{
|
|
__call_once_param<_Fp>* __p = static_cast<__call_once_param<_Fp>*>(__vp);
|
|
(*__p)();
|
|
}
|
|
|
|
void __call_once(volatile unsigned long&, void*, void(*)(void*));
|
|
|
|
#ifndef _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
template<class _Callable, class... _Args>
|
|
inline _LIBCPP_INLINE_VISIBILITY
|
|
void
|
|
call_once(once_flag& __flag, _Callable&& __func, _Args&&... __args)
|
|
{
|
|
if (__flag.__state_ != ~0ul)
|
|
{
|
|
typedef tuple<typename decay<_Callable>::type, typename decay<_Args>::type...> _Gp;
|
|
__call_once_param<_Gp> __p(_Gp(__decay_copy(_VSTD::forward<_Callable>(__func)),
|
|
__decay_copy(_VSTD::forward<_Args>(__args))...));
|
|
__call_once(__flag.__state_, &__p, &__call_once_proxy<_Gp>);
|
|
}
|
|
}
|
|
|
|
#else // _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
template<class _Callable>
|
|
inline _LIBCPP_INLINE_VISIBILITY
|
|
void
|
|
call_once(once_flag& __flag, _Callable __func)
|
|
{
|
|
if (__flag.__state_ != ~0ul)
|
|
{
|
|
__call_once_param<_Callable> __p(__func);
|
|
__call_once(__flag.__state_, &__p, &__call_once_proxy<_Callable>);
|
|
}
|
|
}
|
|
|
|
#endif // _LIBCPP_HAS_NO_VARIADICS
|
|
|
|
_LIBCPP_END_NAMESPACE_STD
|
|
|
|
#endif // _LIBCPP_MUTEX
|