contrib/libc++/src/thread.cpp

   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 = EINVAL;
  50     if (__t_ != 0)
  51     {
  52         ec = __libcpp_thread_join(&__t_);
  53         if (ec == 0)
  54             __t_ = 0;
  55     }
  56 #ifndef _LIBCPP_NO_EXCEPTIONS
  57     if (ec)
  58         throw system_error(error_code(ec, system_category()), "thread::join failed");
  59 #endif  // _LIBCPP_NO_EXCEPTIONS
  60 }
  61
  62 void
  63 thread::detach()
  64 {
  65     int ec = EINVAL;
  66     if (__t_ != 0)
  67     {
  68         ec = __libcpp_thread_detach(&__t_);
  69         if (ec == 0)
  70             __t_ = 0;
  71     }
  72 #ifndef _LIBCPP_NO_EXCEPTIONS
  73     if (ec)
  74         throw system_error(error_code(ec, system_category()), "thread::detach failed");
  75 #endif  // _LIBCPP_NO_EXCEPTIONS
  76 }
  77
  78 unsigned
  79 thread::hardware_concurrency() _NOEXCEPT
  80 {
  81 #if defined(CTL_HW) && defined(HW_NCPU)
  82     unsigned n;
  83     int mib[2] = {CTL_HW, HW_NCPU};
  84     std::size_t s = sizeof(n);
  85     sysctl(mib, 2, &n, &s, 0, 0);
  86     return n;
  87 #elif defined(_SC_NPROCESSORS_ONLN)
  88     long result = sysconf(_SC_NPROCESSORS_ONLN);
  89     // sysconf returns -1 if the name is invalid, the option does not exist or
  90     // does not have a definite limit.
  91     // if sysconf returns some other negative number, we have no idea
  92     // what is going on. Default to something safe.
  93     if (result < 0)
  94         return 0;
  95     return static_cast<unsigned>(result);
  96 #elif defined(_WIN32)
  97     SYSTEM_INFO info;
  98     GetSystemInfo(&info);
  99     return info.dwNumberOfProcessors;
 100 #else  // defined(CTL_HW) && defined(HW_NCPU)
 101     // TODO: grovel through /proc or check cpuid on x86 and similar
 102     // instructions on other architectures.
 103 #   if defined(_MSC_VER) && ! defined(__clang__)
 104         _LIBCPP_WARNING("hardware_concurrency not yet implemented")
 105 #   else
 106 #       warning hardware_concurrency not yet implemented
 107 #   endif
 108     return 0;  // Means not computable [thread.thread.static]
 109 #endif  // defined(CTL_HW) && defined(HW_NCPU)
 110 }
 111
 112 namespace this_thread
 113 {
 114
 115 void
 116 sleep_for(const chrono::nanoseconds& ns)
 117 {
 118     using namespace chrono;
 119     if (ns > nanoseconds::zero())
 120     {
 121         seconds s = duration_cast<seconds>(ns);
 122         timespec ts;
 123         typedef decltype(ts.tv_sec) ts_sec;
 124         _LIBCPP_CONSTEXPR ts_sec ts_sec_max = numeric_limits<ts_sec>::max();
 125         if (s.count() < ts_sec_max)
 126         {
 127             ts.tv_sec = static_cast<ts_sec>(s.count());
 128             ts.tv_nsec = static_cast<decltype(ts.tv_nsec)>((ns-s).count());
 129         }
 130         else
 131         {
 132             ts.tv_sec = ts_sec_max;
 133             ts.tv_nsec = giga::num - 1;
 134         }
 135
 136         while (nanosleep(&ts, &ts) == -1 && errno == EINTR)
 137             ;
 138     }
 139 }
 140
 141 }  // this_thread
 142
 143 __thread_specific_ptr<__thread_struct>&
 144 __thread_local_data()
 145 {
 146     static __thread_specific_ptr<__thread_struct> __p;
 147     return __p;
 148 }
 149
 150 // __thread_struct_imp
 151
 152 template <class T>
 153 class _LIBCPP_HIDDEN __hidden_allocator
 154 {
 155 public:
 156     typedef T  value_type;
 157
 158     T* allocate(size_t __n)
 159         {return static_cast<T*>(::operator new(__n * sizeof(T)));}
 160     void deallocate(T* __p, size_t) {::operator delete(static_cast<void*>(__p));}
 161
 162     size_t max_size() const {return size_t(~0) / sizeof(T);}
 163 };
 164
 165 class _LIBCPP_HIDDEN __thread_struct_imp
 166 {
 167     typedef vector<__assoc_sub_state*,
 168                           __hidden_allocator<__assoc_sub_state*> > _AsyncStates;
 169     typedef vector<pair<condition_variable*, mutex*>,
 170                __hidden_allocator<pair<condition_variable*, mutex*> > > _Notify;
 171
 172     _AsyncStates async_states_;
 173     _Notify notify_;
 174
 175     __thread_struct_imp(const __thread_struct_imp&);
 176     __thread_struct_imp& operator=(const __thread_struct_imp&);
 177 public:
 178     __thread_struct_imp() {}
 179     ~__thread_struct_imp();
 180
 181     void notify_all_at_thread_exit(condition_variable* cv, mutex* m);
 182     void __make_ready_at_thread_exit(__assoc_sub_state* __s);
 183 };
 184
 185 __thread_struct_imp::~__thread_struct_imp()
 186 {
 187     for (_Notify::iterator i = notify_.begin(), e = notify_.end();
 188             i != e; ++i)
 189     {
 190         i->second->unlock();
 191         i->first->notify_all();
 192     }
 193     for (_AsyncStates::iterator i = async_states_.begin(), e = async_states_.end();
 194             i != e; ++i)
 195     {
 196         (*i)->__make_ready();
 197         (*i)->__release_shared();
 198     }
 199 }
 200
 201 void
 202 __thread_struct_imp::notify_all_at_thread_exit(condition_variable* cv, mutex* m)
 203 {
 204     notify_.push_back(pair<condition_variable*, mutex*>(cv, m));
 205 }
 206
 207 void
 208 __thread_struct_imp::__make_ready_at_thread_exit(__assoc_sub_state* __s)
 209 {
 210     async_states_.push_back(__s);
 211     __s->__add_shared();
 212 }
 213
 214 // __thread_struct
 215
 216 __thread_struct::__thread_struct()
 217     : __p_(new __thread_struct_imp)
 218 {
 219 }
 220
 221 __thread_struct::~__thread_struct()
 222 {
 223     delete __p_;
 224 }
 225
 226 void
 227 __thread_struct::notify_all_at_thread_exit(condition_variable* cv, mutex* m)
 228 {
 229     __p_->notify_all_at_thread_exit(cv, m);
 230 }
 231
 232 void
 233 __thread_struct::__make_ready_at_thread_exit(__assoc_sub_state* __s)
 234 {
 235     __p_->__make_ready_at_thread_exit(__s);
 236 }
 237
 238 _LIBCPP_END_NAMESPACE_STD
 239
 240 #endif // !_LIBCPP_HAS_NO_THREADS