thread_data.hpp

#ifndef AUTOBOOST_THREAD_PTHREAD_THREAD_DATA_HPP
#define AUTOBOOST_THREAD_PTHREAD_THREAD_DATA_HPP
// Distributed under the Boost Software License, Version 1.0. (See
// accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
// (C) Copyright 2007 Anthony Williams
// (C) Copyright 2011-2012 Vicente J. Botet Escriba

#include <autoboost/thread/detail/config.hpp>
#include <autoboost/thread/exceptions.hpp>
#include <autoboost/thread/lock_guard.hpp>
#include <autoboost/thread/lock_types.hpp>
#include <autoboost/thread/mutex.hpp>
#include <autoboost/thread/pthread/condition_variable_fwd.hpp>

#include <autoboost/shared_ptr.hpp>
#include <autoboost/enable_shared_from_this.hpp>
#include <autoboost/assert.hpp>
#ifdef AUTOBOOST_THREAD_USES_CHRONO
#include <autoboost/chrono/system_clocks.hpp>
#endif

#include <map>
#include <vector>
#include <utility>


#include <pthread.h>
#include <unistd.h>

#include <autoboost/config/abi_prefix.hpp>

namespace autoboost
{
    class thread_attributes {
    public:
        thread_attributes() AUTOBOOST_NOEXCEPT {
            int res = pthread_attr_init(&val_);
            AUTOBOOST_VERIFY(!res && "pthread_attr_init failed");
        }
        ~thread_attributes() {
          int res = pthread_attr_destroy(&val_);
          AUTOBOOST_VERIFY(!res && "pthread_attr_destroy failed");
        }
        // stack
        void set_stack_size(std::size_t size) AUTOBOOST_NOEXCEPT {
          if (size==0) return;
          std::size_t page_size = getpagesize();
#ifdef PTHREAD_STACK_MIN
          if (size<PTHREAD_STACK_MIN) size=PTHREAD_STACK_MIN;
#endif
          size = ((size+page_size-1)/page_size)*page_size;
          int res = pthread_attr_setstacksize(&val_, size);
          AUTOBOOST_VERIFY(!res && "pthread_attr_setstacksize failed");
        }

        std::size_t get_stack_size() const AUTOBOOST_NOEXCEPT {
            std::size_t size;
            int res = pthread_attr_getstacksize(&val_, &size);
            AUTOBOOST_VERIFY(!res && "pthread_attr_getstacksize failed");
            return size;
        }
#define AUTOBOOST_THREAD_DEFINES_THREAD_ATTRIBUTES_NATIVE_HANDLE

        typedef pthread_attr_t native_handle_type;
        native_handle_type* native_handle() AUTOBOOST_NOEXCEPT {
          return &val_;
        }
        const native_handle_type* native_handle() const AUTOBOOST_NOEXCEPT {
          return &val_;
        }

    private:
        pthread_attr_t val_;
    };

    class thread;

    namespace detail
    {
        struct shared_state_base;
        struct tss_cleanup_function;
        struct thread_exit_callback_node;
        struct tss_data_node
        {
            autoboost::shared_ptr<autoboost::detail::tss_cleanup_function> func;
            void* value;

            tss_data_node(autoboost::shared_ptr<autoboost::detail::tss_cleanup_function> func_,
                          void* value_):
                func(func_),value(value_)
            {}
        };

        struct thread_data_base;
        typedef autoboost::shared_ptr<thread_data_base> thread_data_ptr;

        struct AUTOBOOST_THREAD_DECL thread_data_base:
            enable_shared_from_this<thread_data_base>
        {
            thread_data_ptr self;
            pthread_t thread_handle;
            autoboost::mutex data_mutex;
            autoboost::condition_variable done_condition;
            autoboost::mutex sleep_mutex;
            autoboost::condition_variable sleep_condition;
            bool done;
            bool join_started;
            bool joined;
            autoboost::detail::thread_exit_callback_node* thread_exit_callbacks;
            std::map<void const*,autoboost::detail::tss_data_node> tss_data;

            pthread_mutex_t* cond_mutex;
            pthread_cond_t* current_cond;
            typedef std::vector<std::pair<condition_variable*, mutex*>
            //, hidden_allocator<std::pair<condition_variable*, mutex*> >
            > notify_list_t;
            notify_list_t notify;

            typedef std::vector<shared_ptr<shared_state_base> > async_states_t;
            async_states_t async_states_;

//#if defined AUTOBOOST_THREAD_PROVIDES_INTERRUPTIONS
            // These data must be at the end so that the access to the other fields doesn't change
            // when AUTOBOOST_THREAD_PROVIDES_INTERRUPTIONS is defined.
            // Another option is to have them always
            bool interrupt_enabled;
            bool interrupt_requested;
//#endif
            thread_data_base():
                thread_handle(0),
                done(false),join_started(false),joined(false),
                thread_exit_callbacks(0),
                cond_mutex(0),
                current_cond(0),
                notify(),
                async_states_()
//#if defined AUTOBOOST_THREAD_PROVIDES_INTERRUPTIONS
                , interrupt_enabled(true)
                , interrupt_requested(false)
//#endif
            {}
            virtual ~thread_data_base();

            typedef pthread_t native_handle_type;

            virtual void run()=0;
            virtual void notify_all_autoboostat_thread_exit(condition_variable* cv, mutex* m)
            {
              notify.push_back(std::pair<condition_variable*, mutex*>(cv, m));
            }

            void make_ready_autoboostat_thread_exit(shared_ptr<shared_state_base> as)
            {
              async_states_.push_back(as);
            }

        };

        AUTOBOOST_THREAD_DECL thread_data_base* get_current_thread_data();

#if defined AUTOBOOST_THREAD_PROVIDES_INTERRUPTIONS
        class interruption_checker
        {
            thread_data_base* const thread_info;
            pthread_mutex_t* m;
            bool set;

            void check_for_interruption()
            {
#ifndef AUTOBOOST_NO_EXCEPTIONS
                if(thread_info->interrupt_requested)
                {
                    thread_info->interrupt_requested=false;
                    throw thread_interrupted(); // AUTOBOOST_NO_EXCEPTIONS protected
                }
#endif
            }

            void operator=(interruption_checker&);
        public:
            explicit interruption_checker(pthread_mutex_t* cond_mutex,pthread_cond_t* cond):
                thread_info(detail::get_current_thread_data()),m(cond_mutex),
                set(thread_info && thread_info->interrupt_enabled)
            {
                if(set)
                {
                    lock_guard<mutex> guard(thread_info->data_mutex);
                    check_for_interruption();
                    thread_info->cond_mutex=cond_mutex;
                    thread_info->current_cond=cond;
                    AUTOBOOST_VERIFY(!pthread_mutex_lock(m));
                }
                else
                {
                    AUTOBOOST_VERIFY(!pthread_mutex_lock(m));
                }
            }
            ~interruption_checker()
            {
                if(set)
                {
                    AUTOBOOST_VERIFY(!pthread_mutex_unlock(m));
                    lock_guard<mutex> guard(thread_info->data_mutex);
                    thread_info->cond_mutex=NULL;
                    thread_info->current_cond=NULL;
                }
                else
                {
                    AUTOBOOST_VERIFY(!pthread_mutex_unlock(m));
                }
            }
        };
#endif
    }

    namespace this_thread
    {
        namespace hiden
        {
          void AUTOBOOST_THREAD_DECL sleep_for(const timespec& ts);
          void AUTOBOOST_THREAD_DECL sleep_until(const timespec& ts);
        }

#ifdef AUTOBOOST_THREAD_USES_CHRONO
#ifdef AUTOBOOST_THREAD_SLEEP_FOR_IS_STEADY

        inline
        void AUTOBOOST_SYMBOL_VISIBLE sleep_for(const chrono::nanoseconds& ns)
        {
            return autoboost::this_thread::hiden::sleep_for(autoboost::detail::to_timespec(ns));
        }
#endif
#endif // AUTOBOOST_THREAD_USES_CHRONO

        namespace no_interruption_point
        {
          namespace hiden
          {
            void AUTOBOOST_THREAD_DECL sleep_for(const timespec& ts);
            void AUTOBOOST_THREAD_DECL sleep_until(const timespec& ts);
          }

    #ifdef AUTOBOOST_THREAD_USES_CHRONO
    #ifdef AUTOBOOST_THREAD_SLEEP_FOR_IS_STEADY

          inline
          void AUTOBOOST_SYMBOL_VISIBLE sleep_for(const chrono::nanoseconds& ns)
          {
              return autoboost::this_thread::hiden::sleep_for(autoboost::detail::to_timespec(ns));
          }
    #endif
    #endif // AUTOBOOST_THREAD_USES_CHRONO

        } // no_interruption_point

        void AUTOBOOST_THREAD_DECL yield() AUTOBOOST_NOEXCEPT;

#if defined AUTOBOOST_THREAD_USES_DATETIME
#ifdef __DECXXX
        /// Workaround of DECCXX issue of incorrect template substitution
        template<>
#endif
        inline void sleep(system_time const& abs_time)
        {
          return autoboost::this_thread::hiden::sleep_until(autoboost::detail::to_timespec(abs_time));
        }

        template<typename TimeDuration>
        inline AUTOBOOST_SYMBOL_VISIBLE void sleep(TimeDuration const& rel_time)
        {
            this_thread::sleep(get_system_time()+rel_time);
        }
#endif // AUTOBOOST_THREAD_USES_DATETIME
    } // this_thread
}

#include <autoboost/config/abi_suffix.hpp>

#endif