1
2// Copyright Oliver Kowalke 2013.
3// Distributed under the Boost Software License, Version 1.0.
4// (See accompanying file LICENSE_1_0.txt or copy at
5// http://www.boost.org/LICENSE_1_0.txt)
6//
7// based on tss.hpp from boost.thread
8
9#ifndef BOOST_FIBERS_FSS_H
10#define BOOST_FIBERS_FSS_H
11
12#include <boost/config.hpp>
13
14#include <boost/fiber/context.hpp>
15#include <boost/fiber/detail/fss.hpp>
16
17#ifdef BOOST_HAS_ABI_HEADERS
18# include BOOST_ABI_PREFIX
19#endif
20
21namespace boost {
22namespace fibers {
23
24template< typename T >
25class fiber_specific_ptr {
26private:
27 struct default_cleanup_function : public detail::fss_cleanup_function {
28 void operator()( void * data) noexcept override {
29 delete static_cast< T * >( data);
30 }
31 };
32
33 struct custom_cleanup_function : public detail::fss_cleanup_function {
34 void (*fn)(T*);
35
36 explicit custom_cleanup_function( void(*fn_)(T*) ) noexcept :
37 fn{ fn_ } {
38 }
39
40 void operator()( void * data) override {
41 if ( BOOST_LIKELY( nullptr != fn) ) {
42 fn( static_cast< T * >( data) );
43 }
44 }
45 };
46
47 detail::fss_cleanup_function::ptr_t cleanup_fn_;
48
49public:
50 using element_type = T;
51
52 fiber_specific_ptr() :
53 cleanup_fn_{ new default_cleanup_function() } {
54 }
55
56 explicit fiber_specific_ptr( void(*fn)(T*) ) :
57 cleanup_fn_{ new custom_cleanup_function( fn) } {
58 }
59
60 ~fiber_specific_ptr() {
61 context * active_ctx = context::active();
62 if ( nullptr != active_ctx) {
63 active_ctx->set_fss_data(
64 vp: this, cleanup_fn: cleanup_fn_, data: nullptr, cleanup_existing: true);
65 }
66 }
67
68 fiber_specific_ptr( fiber_specific_ptr const&) = delete;
69 fiber_specific_ptr & operator=( fiber_specific_ptr const&) = delete;
70
71 T * get() const noexcept {
72 BOOST_ASSERT( context::active() );
73 void * vp = context::active()->get_fss_data( vp: this);
74 return static_cast< T * >( vp);
75 }
76
77 T * operator->() const noexcept {
78 return get();
79 }
80
81 T & operator*() const noexcept {
82 return * get();
83 }
84
85 T * release() {
86 T * tmp = get();
87 context::active()->set_fss_data(
88 vp: this, cleanup_fn: cleanup_fn_, data: nullptr, cleanup_existing: false);
89 return tmp;
90 }
91
92 void reset( T * t) {
93 T * c = get();
94 if ( BOOST_LIKELY( c != t) ) {
95 context::active()->set_fss_data(
96 vp: this, cleanup_fn: cleanup_fn_, data: t, cleanup_existing: true);
97 }
98 }
99};
100
101}}
102
103#ifdef BOOST_HAS_ABI_HEADERS
104# include BOOST_ABI_SUFFIX
105#endif
106
107#endif // BOOST_FIBERS_FSS_H
108

source code of boost/libs/fiber/include/boost/fiber/fss.hpp