Home | History | Annotate | Download | only in src
      1 //===------------------------- thread.cpp----------------------------------===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is dual licensed under the MIT and the University of Illinois Open
      6 // Source Licenses. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 
     10 #include "__config"
     11 #ifndef _LIBCPP_HAS_NO_THREADS
     12 
     13 #include "thread"
     14 #include "exception"
     15 #include "vector"
     16 #include "future"
     17 #include "limits"
     18 #include <sys/types.h>
     19 
     20 #if defined(__unix__) || (defined(__APPLE__) && defined(__MACH__))
     21 # include <sys/param.h>
     22 # if defined(BSD)
     23 #   include <sys/sysctl.h>
     24 # endif // defined(BSD)
     25 #endif // defined(__unix__) || (defined(__APPLE__) && defined(__MACH__))
     26 
     27 #if !defined(_WIN32)
     28 # include <unistd.h>
     29 #endif // !_WIN32
     30 
     31 #if defined(__NetBSD__)
     32 #pragma weak pthread_create // Do not create libpthread dependency
     33 #endif
     34 #if defined(_WIN32)
     35 #include <windows.h>
     36 #endif
     37 
     38 _LIBCPP_BEGIN_NAMESPACE_STD
     39 
     40 thread::~thread()
     41 {
     42     if (__t_ != 0)
     43         terminate();
     44 }
     45 
     46 void
     47 thread::join()
     48 {
     49     int ec = __libcpp_thread_join(&__t_);
     50 #ifndef _LIBCPP_NO_EXCEPTIONS
     51     if (ec)
     52         throw system_error(error_code(ec, system_category()), "thread::join failed");
     53 #else
     54     (void)ec;
     55 #endif  // _LIBCPP_NO_EXCEPTIONS
     56     __t_ = 0;
     57 }
     58 
     59 void
     60 thread::detach()
     61 {
     62     int ec = EINVAL;
     63     if (__t_ != 0)
     64     {
     65         ec = __libcpp_thread_detach(&__t_);
     66         if (ec == 0)
     67             __t_ = 0;
     68     }
     69 #ifndef _LIBCPP_NO_EXCEPTIONS
     70     if (ec)
     71         throw system_error(error_code(ec, system_category()), "thread::detach failed");
     72 #endif  // _LIBCPP_NO_EXCEPTIONS
     73 }
     74 
     75 unsigned
     76 thread::hardware_concurrency() _NOEXCEPT
     77 {
     78 #if defined(CTL_HW) && defined(HW_NCPU)
     79     unsigned n;
     80     int mib[2] = {CTL_HW, HW_NCPU};
     81     std::size_t s = sizeof(n);
     82     sysctl(mib, 2, &n, &s, 0, 0);
     83     return n;
     84 #elif defined(_SC_NPROCESSORS_ONLN)
     85     long result = sysconf(_SC_NPROCESSORS_ONLN);
     86     // sysconf returns -1 if the name is invalid, the option does not exist or
     87     // does not have a definite limit.
     88     // if sysconf returns some other negative number, we have no idea
     89     // what is going on. Default to something safe.
     90     if (result < 0)
     91         return 0;
     92     return static_cast<unsigned>(result);
     93 #elif defined(_WIN32)
     94     SYSTEM_INFO info;
     95     GetSystemInfo(&info);
     96     return info.dwNumberOfProcessors;
     97 #else  // defined(CTL_HW) && defined(HW_NCPU)
     98     // TODO: grovel through /proc or check cpuid on x86 and similar
     99     // instructions on other architectures.
    100 #   if defined(_MSC_VER) && ! defined(__clang__)
    101         _LIBCPP_WARNING("hardware_concurrency not yet implemented")
    102 #   else
    103 #       warning hardware_concurrency not yet implemented
    104 #   endif
    105     return 0;  // Means not computable [thread.thread.static]
    106 #endif  // defined(CTL_HW) && defined(HW_NCPU)
    107 }
    108 
    109 namespace this_thread
    110 {
    111 
    112 void
    113 sleep_for(const chrono::nanoseconds& ns)
    114 {
    115     using namespace chrono;
    116     if (ns > nanoseconds::zero())
    117     {
    118         seconds s = duration_cast<seconds>(ns);
    119         timespec ts;
    120         typedef decltype(ts.tv_sec) ts_sec;
    121         _LIBCPP_CONSTEXPR ts_sec ts_sec_max = numeric_limits<ts_sec>::max();
    122         if (s.count() < ts_sec_max)
    123         {
    124             ts.tv_sec = static_cast<ts_sec>(s.count());
    125             ts.tv_nsec = static_cast<decltype(ts.tv_nsec)>((ns-s).count());
    126         }
    127         else
    128         {
    129             ts.tv_sec = ts_sec_max;
    130             ts.tv_nsec = giga::num - 1;
    131         }
    132 
    133         while (nanosleep(&ts, &ts) == -1 && errno == EINTR)
    134             ;
    135     }
    136 }
    137 
    138 }  // this_thread
    139 
    140 __thread_specific_ptr<__thread_struct>&
    141 __thread_local_data()
    142 {
    143     static __thread_specific_ptr<__thread_struct> __p;
    144     return __p;
    145 }
    146 
    147 // __thread_struct_imp
    148 
    149 template <class T>
    150 class _LIBCPP_HIDDEN __hidden_allocator
    151 {
    152 public:
    153     typedef T  value_type;
    154 
    155     T* allocate(size_t __n)
    156         {return static_cast<T*>(::operator new(__n * sizeof(T)));}
    157     void deallocate(T* __p, size_t) {::operator delete(static_cast<void*>(__p));}
    158 
    159     size_t max_size() const {return size_t(~0) / sizeof(T);}
    160 };
    161 
    162 class _LIBCPP_HIDDEN __thread_struct_imp
    163 {
    164     typedef vector<__assoc_sub_state*,
    165                           __hidden_allocator<__assoc_sub_state*> > _AsyncStates;
    166     typedef vector<pair<condition_variable*, mutex*>,
    167                __hidden_allocator<pair<condition_variable*, mutex*> > > _Notify;
    168 
    169     _AsyncStates async_states_;
    170     _Notify notify_;
    171 
    172     __thread_struct_imp(const __thread_struct_imp&);
    173     __thread_struct_imp& operator=(const __thread_struct_imp&);
    174 public:
    175     __thread_struct_imp() {}
    176     ~__thread_struct_imp();
    177 
    178     void notify_all_at_thread_exit(condition_variable* cv, mutex* m);
    179     void __make_ready_at_thread_exit(__assoc_sub_state* __s);
    180 };
    181 
    182 __thread_struct_imp::~__thread_struct_imp()
    183 {
    184     for (_Notify::iterator i = notify_.begin(), e = notify_.end();
    185             i != e; ++i)
    186     {
    187         i->second->unlock();
    188         i->first->notify_all();
    189     }
    190     for (_AsyncStates::iterator i = async_states_.begin(), e = async_states_.end();
    191             i != e; ++i)
    192     {
    193         (*i)->__make_ready();
    194         (*i)->__release_shared();
    195     }
    196 }
    197 
    198 void
    199 __thread_struct_imp::notify_all_at_thread_exit(condition_variable* cv, mutex* m)
    200 {
    201     notify_.push_back(pair<condition_variable*, mutex*>(cv, m));
    202 }
    203 
    204 void
    205 __thread_struct_imp::__make_ready_at_thread_exit(__assoc_sub_state* __s)
    206 {
    207     async_states_.push_back(__s);
    208     __s->__add_shared();
    209 }
    210 
    211 // __thread_struct
    212 
    213 __thread_struct::__thread_struct()
    214     : __p_(new __thread_struct_imp)
    215 {
    216 }
    217 
    218 __thread_struct::~__thread_struct()
    219 {
    220     delete __p_;
    221 }
    222 
    223 void
    224 __thread_struct::notify_all_at_thread_exit(condition_variable* cv, mutex* m)
    225 {
    226     __p_->notify_all_at_thread_exit(cv, m);
    227 }
    228 
    229 void
    230 __thread_struct::__make_ready_at_thread_exit(__assoc_sub_state* __s)
    231 {
    232     __p_->__make_ready_at_thread_exit(__s);
    233 }
    234 
    235 _LIBCPP_END_NAMESPACE_STD
    236 
    237 #endif // !_LIBCPP_HAS_NO_THREADS
    238