#ifndef BOOST_THREAD_TSS_HPP #define BOOST_THREAD_TSS_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-8 Anthony Williams #include <boost/thread/detail/config.hpp> #include <boost/shared_ptr.hpp> #include <boost/thread/detail/thread_heap_alloc.hpp> #include <boost/config/abi_prefix.hpp> namespace boost { namespace detail { struct tss_cleanup_function { virtual ~tss_cleanup_function() {} virtual void operator()(void* data)=0; }; BOOST_THREAD_DECL void set_tss_data(void const* key,boost::shared_ptr<tss_cleanup_function> func,void* tss_data,bool cleanup_existing); BOOST_THREAD_DECL void* get_tss_data(void const* key); } template <typename T> class thread_specific_ptr { private: thread_specific_ptr(thread_specific_ptr&); thread_specific_ptr& operator=(thread_specific_ptr&); struct delete_data: detail::tss_cleanup_function { void operator()(void* data) { delete static_cast<T*>(data); } }; struct run_custom_cleanup_function: detail::tss_cleanup_function { void (*cleanup_function)(T*); explicit run_custom_cleanup_function(void (*cleanup_function_)(T*)): cleanup_function(cleanup_function_) {} void operator()(void* data) { cleanup_function(static_cast<T*>(data)); } }; boost::shared_ptr<detail::tss_cleanup_function> cleanup; public: thread_specific_ptr(): cleanup(detail::heap_new<delete_data>(),detail::do_heap_delete<delete_data>()) {} explicit thread_specific_ptr(void (*func_)(T*)) { if(func_) { cleanup.reset(detail::heap_new<run_custom_cleanup_function>(func_),detail::do_heap_delete<run_custom_cleanup_function>()); } } ~thread_specific_ptr() { detail::set_tss_data(this,boost::shared_ptr<detail::tss_cleanup_function>(),0,true); } T* get() const { return static_cast<T*>(detail::get_tss_data(this)); } T* operator->() const { return get(); } T& operator*() const { return *get(); } T* release() { T* const temp=get(); detail::set_tss_data(this,boost::shared_ptr<detail::tss_cleanup_function>(),0,false); return temp; } void reset(T* new_value=0) { T* const current_value=get(); if(current_value!=new_value) { detail::set_tss_data(this,cleanup,new_value,true); } } }; } #include <boost/config/abi_suffix.hpp> #endif