Chris@16: //
Chris@16: // detail/win_thread.hpp
Chris@16: // ~~~~~~~~~~~~~~~~~~~~~
Chris@16: //
Chris@101: // Copyright (c) 2003-2015 Christopher M. Kohlhoff (chris at kohlhoff dot com)
Chris@16: //
Chris@16: // Distributed under the Boost Software License, Version 1.0. (See accompanying
Chris@16: // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
Chris@16: //
Chris@16: 
Chris@16: #ifndef BOOST_ASIO_DETAIL_WIN_THREAD_HPP
Chris@16: #define BOOST_ASIO_DETAIL_WIN_THREAD_HPP
Chris@16: 
Chris@16: #if defined(_MSC_VER) && (_MSC_VER >= 1200)
Chris@16: # pragma once
Chris@16: #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
Chris@16: 
Chris@16: #include <boost/asio/detail/config.hpp>
Chris@16: 
Chris@16: #if defined(BOOST_ASIO_WINDOWS) && !defined(UNDER_CE)
Chris@16: 
Chris@16: #include <boost/asio/detail/noncopyable.hpp>
Chris@16: #include <boost/asio/detail/socket_types.hpp>
Chris@16: 
Chris@16: #include <boost/asio/detail/push_options.hpp>
Chris@16: 
Chris@16: namespace boost {
Chris@16: namespace asio {
Chris@16: namespace detail {
Chris@16: 
Chris@16: BOOST_ASIO_DECL unsigned int __stdcall win_thread_function(void* arg);
Chris@16: 
Chris@16: #if defined(WINVER) && (WINVER < 0x0500)
Chris@16: BOOST_ASIO_DECL void __stdcall apc_function(ULONG data);
Chris@16: #else
Chris@16: BOOST_ASIO_DECL void __stdcall apc_function(ULONG_PTR data);
Chris@16: #endif
Chris@16: 
Chris@16: template <typename T>
Chris@16: class win_thread_base
Chris@16: {
Chris@16: public:
Chris@16:   static bool terminate_threads()
Chris@16:   {
Chris@16:     return ::InterlockedExchangeAdd(&terminate_threads_, 0) != 0;
Chris@16:   }
Chris@16: 
Chris@16:   static void set_terminate_threads(bool b)
Chris@16:   {
Chris@16:     ::InterlockedExchange(&terminate_threads_, b ? 1 : 0);
Chris@16:   }
Chris@16: 
Chris@16: private:
Chris@16:   static long terminate_threads_;
Chris@16: };
Chris@16: 
Chris@16: template <typename T>
Chris@16: long win_thread_base<T>::terminate_threads_ = 0;
Chris@16: 
Chris@16: class win_thread
Chris@16:   : private noncopyable,
Chris@16:     public win_thread_base<win_thread>
Chris@16: {
Chris@16: public:
Chris@16:   // Constructor.
Chris@16:   template <typename Function>
Chris@16:   win_thread(Function f, unsigned int stack_size = 0)
Chris@16:     : thread_(0),
Chris@16:       exit_event_(0)
Chris@16:   {
Chris@16:     start_thread(new func<Function>(f), stack_size);
Chris@16:   }
Chris@16: 
Chris@16:   // Destructor.
Chris@16:   BOOST_ASIO_DECL ~win_thread();
Chris@16: 
Chris@16:   // Wait for the thread to exit.
Chris@16:   BOOST_ASIO_DECL void join();
Chris@16: 
Chris@16: private:
Chris@16:   friend BOOST_ASIO_DECL unsigned int __stdcall win_thread_function(void* arg);
Chris@16: 
Chris@16: #if defined(WINVER) && (WINVER < 0x0500)
Chris@16:   friend BOOST_ASIO_DECL void __stdcall apc_function(ULONG);
Chris@16: #else
Chris@16:   friend BOOST_ASIO_DECL void __stdcall apc_function(ULONG_PTR);
Chris@16: #endif
Chris@16: 
Chris@16:   class func_base
Chris@16:   {
Chris@16:   public:
Chris@16:     virtual ~func_base() {}
Chris@16:     virtual void run() = 0;
Chris@16:     ::HANDLE entry_event_;
Chris@16:     ::HANDLE exit_event_;
Chris@16:   };
Chris@16: 
Chris@16:   struct auto_func_base_ptr
Chris@16:   {
Chris@16:     func_base* ptr;
Chris@16:     ~auto_func_base_ptr() { delete ptr; }
Chris@16:   };
Chris@16: 
Chris@16:   template <typename Function>
Chris@16:   class func
Chris@16:     : public func_base
Chris@16:   {
Chris@16:   public:
Chris@16:     func(Function f)
Chris@16:       : f_(f)
Chris@16:     {
Chris@16:     }
Chris@16: 
Chris@16:     virtual void run()
Chris@16:     {
Chris@16:       f_();
Chris@16:     }
Chris@16: 
Chris@16:   private:
Chris@16:     Function f_;
Chris@16:   };
Chris@16: 
Chris@16:   BOOST_ASIO_DECL void start_thread(func_base* arg, unsigned int stack_size);
Chris@16: 
Chris@16:   ::HANDLE thread_;
Chris@16:   ::HANDLE exit_event_;
Chris@16: };
Chris@16: 
Chris@16: } // namespace detail
Chris@16: } // namespace asio
Chris@16: } // namespace boost
Chris@16: 
Chris@16: #include <boost/asio/detail/pop_options.hpp>
Chris@16: 
Chris@16: #if defined(BOOST_ASIO_HEADER_ONLY)
Chris@16: # include <boost/asio/detail/impl/win_thread.ipp>
Chris@16: #endif // defined(BOOST_ASIO_HEADER_ONLY)
Chris@16: 
Chris@16: #endif // defined(BOOST_ASIO_WINDOWS) && !defined(UNDER_CE)
Chris@16: 
Chris@16: #endif // BOOST_ASIO_DETAIL_WIN_THREAD_HPP