1349cc55cSDimitry Andric //===----------------------------------------------------------------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric 981ad6265SDimitry Andric #include <__assert> 1006c3fb27SDimitry Andric #include <__thread/id.h> 115f757f3fSDimitry Andric #include <__utility/exception_guard.h> 1281ad6265SDimitry Andric #include <limits> 1381ad6265SDimitry Andric #include <mutex> 1481ad6265SDimitry Andric 150b57cec5SDimitry Andric #include "include/atomic_support.h" 160b57cec5SDimitry Andric 17480093f4SDimitry Andric #if defined(__ELF__) && defined(_LIBCPP_LINK_PTHREAD_LIB) 180b57cec5SDimitry Andric # pragma comment(lib, "pthread") 190b57cec5SDimitry Andric #endif 200b57cec5SDimitry Andric 2181ad6265SDimitry Andric _LIBCPP_PUSH_MACROS 2281ad6265SDimitry Andric #include <__undef_macros> 2381ad6265SDimitry Andric 240b57cec5SDimitry Andric _LIBCPP_BEGIN_NAMESPACE_STD 2581ad6265SDimitry Andric 260b57cec5SDimitry Andric // ~mutex is defined elsewhere 270b57cec5SDimitry Andric 28cb14a3feSDimitry Andric void mutex::lock() { 290b57cec5SDimitry Andric int ec = __libcpp_mutex_lock(&__m_); 300b57cec5SDimitry Andric if (ec) 310b57cec5SDimitry Andric __throw_system_error(ec, "mutex lock failed"); 320b57cec5SDimitry Andric } 330b57cec5SDimitry Andric 34cb14a3feSDimitry Andric bool mutex::try_lock() noexcept { return __libcpp_mutex_trylock(&__m_); } 350b57cec5SDimitry Andric 36cb14a3feSDimitry Andric void mutex::unlock() noexcept { 370b57cec5SDimitry Andric int ec = __libcpp_mutex_unlock(&__m_); 380b57cec5SDimitry Andric (void)ec; 3906c3fb27SDimitry Andric _LIBCPP_ASSERT_UNCATEGORIZED(ec == 0, "call to mutex::unlock failed"); 400b57cec5SDimitry Andric } 410b57cec5SDimitry Andric 420b57cec5SDimitry Andric // recursive_mutex 430b57cec5SDimitry Andric 44cb14a3feSDimitry Andric recursive_mutex::recursive_mutex() { 450b57cec5SDimitry Andric int ec = __libcpp_recursive_mutex_init(&__m_); 460b57cec5SDimitry Andric if (ec) 470b57cec5SDimitry Andric __throw_system_error(ec, "recursive_mutex constructor failed"); 480b57cec5SDimitry Andric } 490b57cec5SDimitry Andric 50cb14a3feSDimitry Andric recursive_mutex::~recursive_mutex() { 510b57cec5SDimitry Andric int e = __libcpp_recursive_mutex_destroy(&__m_); 520b57cec5SDimitry Andric (void)e; 5306c3fb27SDimitry Andric _LIBCPP_ASSERT_UNCATEGORIZED(e == 0, "call to ~recursive_mutex() failed"); 540b57cec5SDimitry Andric } 550b57cec5SDimitry Andric 56cb14a3feSDimitry Andric void recursive_mutex::lock() { 570b57cec5SDimitry Andric int ec = __libcpp_recursive_mutex_lock(&__m_); 580b57cec5SDimitry Andric if (ec) 590b57cec5SDimitry Andric __throw_system_error(ec, "recursive_mutex lock failed"); 600b57cec5SDimitry Andric } 610b57cec5SDimitry Andric 62cb14a3feSDimitry Andric void recursive_mutex::unlock() noexcept { 630b57cec5SDimitry Andric int e = __libcpp_recursive_mutex_unlock(&__m_); 640b57cec5SDimitry Andric (void)e; 6506c3fb27SDimitry Andric _LIBCPP_ASSERT_UNCATEGORIZED(e == 0, "call to recursive_mutex::unlock() failed"); 660b57cec5SDimitry Andric } 670b57cec5SDimitry Andric 68cb14a3feSDimitry Andric bool recursive_mutex::try_lock() noexcept { return __libcpp_recursive_mutex_trylock(&__m_); } 690b57cec5SDimitry Andric 700b57cec5SDimitry Andric // timed_mutex 710b57cec5SDimitry Andric 72cb14a3feSDimitry Andric timed_mutex::timed_mutex() : __locked_(false) {} 730b57cec5SDimitry Andric 74cb14a3feSDimitry Andric timed_mutex::~timed_mutex() { lock_guard<mutex> _(__m_); } 750b57cec5SDimitry Andric 76cb14a3feSDimitry Andric void timed_mutex::lock() { 770b57cec5SDimitry Andric unique_lock<mutex> lk(__m_); 780b57cec5SDimitry Andric while (__locked_) 790b57cec5SDimitry Andric __cv_.wait(lk); 800b57cec5SDimitry Andric __locked_ = true; 810b57cec5SDimitry Andric } 820b57cec5SDimitry Andric 83cb14a3feSDimitry Andric bool timed_mutex::try_lock() noexcept { 840b57cec5SDimitry Andric unique_lock<mutex> lk(__m_, try_to_lock); 85cb14a3feSDimitry Andric if (lk.owns_lock() && !__locked_) { 860b57cec5SDimitry Andric __locked_ = true; 870b57cec5SDimitry Andric return true; 880b57cec5SDimitry Andric } 890b57cec5SDimitry Andric return false; 900b57cec5SDimitry Andric } 910b57cec5SDimitry Andric 92cb14a3feSDimitry Andric void timed_mutex::unlock() noexcept { 930b57cec5SDimitry Andric lock_guard<mutex> _(__m_); 940b57cec5SDimitry Andric __locked_ = false; 950b57cec5SDimitry Andric __cv_.notify_one(); 960b57cec5SDimitry Andric } 970b57cec5SDimitry Andric 980b57cec5SDimitry Andric // recursive_timed_mutex 990b57cec5SDimitry Andric 100cb14a3feSDimitry Andric recursive_timed_mutex::recursive_timed_mutex() : __count_(0), __id_{} {} 1010b57cec5SDimitry Andric 102cb14a3feSDimitry Andric recursive_timed_mutex::~recursive_timed_mutex() { lock_guard<mutex> _(__m_); } 1030b57cec5SDimitry Andric 104cb14a3feSDimitry Andric void recursive_timed_mutex::lock() { 1050b57cec5SDimitry Andric __thread_id id = this_thread::get_id(); 1060b57cec5SDimitry Andric unique_lock<mutex> lk(__m_); 107cb14a3feSDimitry Andric if (id == __id_) { 1080b57cec5SDimitry Andric if (__count_ == numeric_limits<size_t>::max()) 1090b57cec5SDimitry Andric __throw_system_error(EAGAIN, "recursive_timed_mutex lock limit reached"); 1100b57cec5SDimitry Andric ++__count_; 1110b57cec5SDimitry Andric return; 1120b57cec5SDimitry Andric } 1130b57cec5SDimitry Andric while (__count_ != 0) 1140b57cec5SDimitry Andric __cv_.wait(lk); 1150b57cec5SDimitry Andric __count_ = 1; 1160b57cec5SDimitry Andric __id_ = id; 1170b57cec5SDimitry Andric } 1180b57cec5SDimitry Andric 119cb14a3feSDimitry Andric bool recursive_timed_mutex::try_lock() noexcept { 1200b57cec5SDimitry Andric __thread_id id = this_thread::get_id(); 1210b57cec5SDimitry Andric unique_lock<mutex> lk(__m_, try_to_lock); 122cb14a3feSDimitry Andric if (lk.owns_lock() && (__count_ == 0 || id == __id_)) { 1230b57cec5SDimitry Andric if (__count_ == numeric_limits<size_t>::max()) 1240b57cec5SDimitry Andric return false; 1250b57cec5SDimitry Andric ++__count_; 1260b57cec5SDimitry Andric __id_ = id; 1270b57cec5SDimitry Andric return true; 1280b57cec5SDimitry Andric } 1290b57cec5SDimitry Andric return false; 1300b57cec5SDimitry Andric } 1310b57cec5SDimitry Andric 132cb14a3feSDimitry Andric void recursive_timed_mutex::unlock() noexcept { 1330b57cec5SDimitry Andric unique_lock<mutex> lk(__m_); 134cb14a3feSDimitry Andric if (--__count_ == 0) { 1350b57cec5SDimitry Andric __id_.__reset(); 1360b57cec5SDimitry Andric lk.unlock(); 1370b57cec5SDimitry Andric __cv_.notify_one(); 1380b57cec5SDimitry Andric } 1390b57cec5SDimitry Andric } 1400b57cec5SDimitry Andric 1410b57cec5SDimitry Andric _LIBCPP_END_NAMESPACE_STD 14281ad6265SDimitry Andric 14381ad6265SDimitry Andric _LIBCPP_POP_MACROS 144