1
2// Copyright Oliver Kowalke 2009.
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#ifndef BOOST_COROUTINES_STANDARD_STACK_ALLOCATOR_H
8#define BOOST_COROUTINES_STANDARD_STACK_ALLOCATOR_H
9
10#if defined(BOOST_USE_VALGRIND)
11#include <valgrind/valgrind.h>
12#endif
13
14#include <cstddef>
15#include <cstdlib>
16#include <new>
17
18#include <boost/assert.hpp>
19#include <boost/config.hpp>
20
21#include <boost/coroutine/detail/config.hpp>
22#include <boost/coroutine/stack_context.hpp>
23#include <boost/coroutine/stack_traits.hpp>
24
25#if defined(BOOST_COROUTINES_USE_MAP_STACK)
26extern "C" {
27#include <sys/mman.h>
28}
29#endif
30
31#ifdef BOOST_HAS_ABI_HEADERS
32# include BOOST_ABI_PREFIX
33#endif
34
35namespace boost {
36namespace coroutines {
37
38template< typename traitsT >
39struct basic_standard_stack_allocator
40{
41 typedef traitsT traits_type;
42
43 void allocate( stack_context & ctx, std::size_t size = traits_type::minimum_size() )
44 {
45 BOOST_ASSERT( traits_type::minimum_size() <= size);
46 BOOST_ASSERT( traits_type::is_unbounded() || ( traits_type::maximum_size() >= size) );
47
48#if defined(BOOST_COROUTINES_USE_MAP_STACK)
49 void * limit = ::mmap(0, size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANON | MAP_STACK, -1, 0);
50 if ( limit == MAP_FAILED ) throw std::bad_alloc();
51#else
52 void * limit = std::malloc( size: size);
53 if ( ! limit) throw std::bad_alloc();
54#endif
55
56 ctx.size = size;
57 ctx.sp = static_cast< char * >( limit) + ctx.size;
58#if defined(BOOST_USE_VALGRIND)
59 ctx.valgrind_stack_id = VALGRIND_STACK_REGISTER( ctx.sp, limit);
60#endif
61 }
62
63 void deallocate( stack_context & ctx)
64 {
65 BOOST_ASSERT( ctx.sp);
66 BOOST_ASSERT( traits_type::minimum_size() <= ctx.size);
67 BOOST_ASSERT( traits_type::is_unbounded() || ( traits_type::maximum_size() >= ctx.size) );
68
69#if defined(BOOST_USE_VALGRIND)
70 VALGRIND_STACK_DEREGISTER( ctx.valgrind_stack_id);
71#endif
72
73 void * limit = static_cast< char * >( ctx.sp) - ctx.size;
74#if defined(BOOST_COROUTINES_USE_MAP_STACK)
75 munmap(limit, ctx.size);
76#else
77 std::free( ptr: limit);
78#endif
79 }
80};
81
82typedef basic_standard_stack_allocator< stack_traits > standard_stack_allocator;
83
84}}
85
86#ifdef BOOST_HAS_ABI_HEADERS
87# include BOOST_ABI_SUFFIX
88#endif
89
90#endif // BOOST_COROUTINES_STANDARD_STACK_ALLOCATOR_H
91

source code of boost/libs/coroutine/include/boost/coroutine/standard_stack_allocator.hpp