ceph/src/boost/boost/asio/detail/win_thread.hpp

   1 //
   2 // detail/win_thread.hpp
   3 // ~~~~~~~~~~~~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2019 Christopher M. Kohlhoff (chris at kohlhoff dot com)
   6 //
   7 // Distributed under the Boost Software License, Version 1.0. (See accompanying
   8 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   9 //
  10
  11 #ifndef BOOST_ASIO_DETAIL_WIN_THREAD_HPP
  12 #define BOOST_ASIO_DETAIL_WIN_THREAD_HPP
  13
  14 #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  15 # pragma once
  16 #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  17
  18 #include <boost/asio/detail/config.hpp>
  19
  20 #if defined(BOOST_ASIO_WINDOWS) \
  21   && !defined(BOOST_ASIO_WINDOWS_APP) \
  22   && !defined(UNDER_CE)
  23
  24 #include <cstddef>
  25 #include <boost/asio/detail/noncopyable.hpp>
  26 #include <boost/asio/detail/socket_types.hpp>
  27
  28 #include <boost/asio/detail/push_options.hpp>
  29
  30 namespace boost {
  31 namespace asio {
  32 namespace detail {
  33
  34 BOOST_ASIO_DECL unsigned int __stdcall win_thread_function(void* arg);
  35
  36 #if defined(WINVER) && (WINVER < 0x0500)
  37 BOOST_ASIO_DECL void __stdcall apc_function(ULONG data);
  38 #else
  39 BOOST_ASIO_DECL void __stdcall apc_function(ULONG_PTR data);
  40 #endif
  41
  42 template <typename T>
  43 class win_thread_base
  44 {
  45 public:
  46   static bool terminate_threads()
  47   {
  48     return ::InterlockedExchangeAdd(&terminate_threads_, 0) != 0;
  49   }
  50
  51   static void set_terminate_threads(bool b)
  52   {
  53     ::InterlockedExchange(&terminate_threads_, b ? 1 : 0);
  54   }
  55
  56 private:
  57   static long terminate_threads_;
  58 };
  59
  60 template <typename T>
  61 long win_thread_base<T>::terminate_threads_ = 0;
  62
  63 class win_thread
  64   : private noncopyable,
  65     public win_thread_base<win_thread>
  66 {
  67 public:
  68   // Constructor.
  69   template <typename Function>
  70   win_thread(Function f, unsigned int stack_size = 0)
  71     : thread_(0),
  72       exit_event_(0)
  73   {
  74     start_thread(new func<Function>(f), stack_size);
  75   }
  76
  77   // Destructor.
  78   BOOST_ASIO_DECL ~win_thread();
  79
  80   // Wait for the thread to exit.
  81   BOOST_ASIO_DECL void join();
  82
  83   // Get number of CPUs.
  84   BOOST_ASIO_DECL static std::size_t hardware_concurrency();
  85
  86 private:
  87   friend BOOST_ASIO_DECL unsigned int __stdcall win_thread_function(void* arg);
  88
  89 #if defined(WINVER) && (WINVER < 0x0500)
  90   friend BOOST_ASIO_DECL void __stdcall apc_function(ULONG);
  91 #else
  92   friend BOOST_ASIO_DECL void __stdcall apc_function(ULONG_PTR);
  93 #endif
  94
  95   class func_base
  96   {
  97   public:
  98     virtual ~func_base() {}
  99     virtual void run() = 0;
 100     ::HANDLE entry_event_;
 101     ::HANDLE exit_event_;
 102   };
 103
 104   struct auto_func_base_ptr
 105   {
 106     func_base* ptr;
 107     ~auto_func_base_ptr() { delete ptr; }
 108   };
 109
 110   template <typename Function>
 111   class func
 112     : public func_base
 113   {
 114   public:
 115     func(Function f)
 116       : f_(f)
 117     {
 118     }
 119
 120     virtual void run()
 121     {
 122       f_();
 123     }
 124
 125   private:
 126     Function f_;
 127   };
 128
 129   BOOST_ASIO_DECL void start_thread(func_base* arg, unsigned int stack_size);
 130
 131   ::HANDLE thread_;
 132   ::HANDLE exit_event_;
 133 };
 134
 135 } // namespace detail
 136 } // namespace asio
 137 } // namespace boost
 138
 139 #include <boost/asio/detail/pop_options.hpp>
 140
 141 #if defined(BOOST_ASIO_HEADER_ONLY)
 142 # include <boost/asio/detail/impl/win_thread.ipp>
 143 #endif // defined(BOOST_ASIO_HEADER_ONLY)
 144
 145 #endif // defined(BOOST_ASIO_WINDOWS)
 146        // && !defined(BOOST_ASIO_WINDOWS_APP)
 147        // && !defined(UNDER_CE)
 148
 149 #endif // BOOST_ASIO_DETAIL_WIN_THREAD_HPP