1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// Copyright Oliver Kowalke 2014.
// 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)
#ifndef BOOST_CONTEXT_POOLED_pooled_fixedsize_H
#define BOOST_CONTEXT_POOLED_pooled_fixedsize_H
#include <atomic>
#include <cstddef>
#include <cstdlib>
#include <new>
#include <boost/assert.hpp>
#include <boost/config.hpp>
#include <boost/intrusive_ptr.hpp>
#include <boost/pool/pool.hpp>
#include <boost/context/detail/config.hpp>
#include <boost/context/stack_context.hpp>
#include <boost/context/stack_traits.hpp>
#if defined(BOOST_USE_VALGRIND)
#include <valgrind/valgrind.h>
#endif
#ifdef BOOST_HAS_ABI_HEADERS
# include BOOST_ABI_PREFIX
#endif
namespace boost {
namespace context {
template< typename traitsT >
class basic_pooled_fixedsize_stack {
private:
class storage {
private:
std::atomic< std::size_t > use_count_;
std::size_t stack_size_;
boost::pool< boost::default_user_allocator_malloc_free > storage_;
public:
storage( std::size_t stack_size, std::size_t next_size, std::size_t max_size) :
use_count_( 0),
stack_size_( stack_size),
storage_( stack_size, next_size, max_size) {
BOOST_ASSERT( traits_type::is_unbounded() || ( traits_type::maximum_size() >= stack_size_) );
}
stack_context allocate() {
void * vp = storage_.malloc();
if ( ! vp) {
throw std::bad_alloc();
}
stack_context sctx;
sctx.size = stack_size_;
sctx.sp = static_cast< char * >( vp) + sctx.size;
#if defined(BOOST_USE_VALGRIND)
sctx.valgrind_stack_id = VALGRIND_STACK_REGISTER( sctx.sp, vp);
#endif
return sctx;
}
void deallocate( stack_context & sctx) BOOST_NOEXCEPT_OR_NOTHROW {
BOOST_ASSERT( sctx.sp);
BOOST_ASSERT( traits_type::is_unbounded() || ( traits_type::maximum_size() >= sctx.size) );
#if defined(BOOST_USE_VALGRIND)
VALGRIND_STACK_DEREGISTER( sctx.valgrind_stack_id);
#endif
void * vp = static_cast< char * >( sctx.sp) - sctx.size;
storage_.free( vp);
}
friend void intrusive_ptr_add_ref( storage * s) noexcept {
++s->use_count_;
}
friend void intrusive_ptr_release( storage * s) noexcept {
if ( 0 == --s->use_count_) {
delete s;
}
}
};
intrusive_ptr< storage > storage_;
public:
typedef traitsT traits_type;
basic_pooled_fixedsize_stack( std::size_t stack_size = traits_type::default_size(),
std::size_t next_size = 32,
std::size_t max_size = 0) BOOST_NOEXCEPT_OR_NOTHROW :
storage_( new storage( stack_size, next_size, max_size) ) {
}
stack_context allocate() {
return storage_->allocate();
}
void deallocate( stack_context & sctx) BOOST_NOEXCEPT_OR_NOTHROW {
storage_->deallocate( sctx);
}
};
typedef basic_pooled_fixedsize_stack< stack_traits > pooled_fixedsize_stack;
}}
#ifdef BOOST_HAS_ABI_HEADERS
# include BOOST_ABI_SUFFIX
#endif
#endif // BOOST_CONTEXT_POOLED_pooled_fixedsize_H
|
