boost/coroutine/detail/pull_coroutine_object.hpp
// Copyright Oliver Kowalke 2009. // 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_COROUTINES_DETAIL_PULL_COROUTINE_OBJECT_H #define BOOST_COROUTINES_DETAIL_PULL_COROUTINE_OBJECT_H #include <boost/assert.hpp> #include <boost/config.hpp> #include <boost/context/detail/config.hpp> #include <boost/cstdint.hpp> #include <boost/exception_ptr.hpp> #include <boost/move/move.hpp> #include <boost/coroutine/detail/config.hpp> #include <boost/coroutine/detail/coroutine_context.hpp> #include <boost/coroutine/detail/flags.hpp> #include <boost/coroutine/detail/preallocated.hpp> #include <boost/coroutine/detail/pull_coroutine_impl.hpp> #include <boost/coroutine/detail/trampoline_pull.hpp> #include <boost/coroutine/exceptions.hpp> #include <boost/coroutine/flags.hpp> #include <boost/coroutine/stack_context.hpp> #ifdef BOOST_HAS_ABI_HEADERS # include BOOST_ABI_PREFIX #endif #if defined(BOOST_MSVC) # pragma warning(push) # pragma warning(disable:4355) #endif namespace boost { namespace coroutines { namespace detail { struct pull_coroutine_context { coroutine_context caller; coroutine_context callee; template< typename Coro > pull_coroutine_context( preallocated const& palloc, Coro *) : caller(), callee( trampoline_pull< Coro >, palloc) {} }; template< typename PushCoro, typename R, typename Fn, typename StackAllocator > class pull_coroutine_object : private pull_coroutine_context, public pull_coroutine_impl< R > { private: typedef pull_coroutine_context ctx_t; typedef pull_coroutine_impl< R > base_t; typedef pull_coroutine_object< PushCoro, R, Fn, StackAllocator > obj_t; Fn fn_; stack_context stack_ctx_; StackAllocator stack_alloc_; static void deallocate_( obj_t * obj) { stack_context stack_ctx( obj->stack_ctx_); StackAllocator stack_alloc( obj->stack_alloc_); obj->unwind_stack(); obj->~obj_t(); stack_alloc.deallocate( stack_ctx); } public: #ifdef BOOST_NO_CXX11_RVALUE_REFERENCES pull_coroutine_object( Fn fn, attributes const& attrs, preallocated const& palloc, StackAllocator const& stack_alloc) BOOST_NOEXCEPT : ctx_t( palloc, this), base_t( & this->caller, & this->callee, stack_unwind == attrs.do_unwind), fn_( fn), stack_ctx_( palloc.sctx), stack_alloc_( stack_alloc) {} #endif pull_coroutine_object( BOOST_RV_REF( Fn) fn, attributes const& attrs, preallocated const& palloc, StackAllocator const& stack_alloc) BOOST_NOEXCEPT : ctx_t( palloc, this), base_t( & this->caller, & this->callee, stack_unwind == attrs.do_unwind), #ifdef BOOST_NO_CXX11_RVALUE_REFERENCES fn_( fn), #else fn_( boost::forward< Fn >( fn) ), #endif stack_ctx_( palloc.sctx), stack_alloc_( stack_alloc) {} void run() { BOOST_ASSERT( ! base_t::unwind_requested() ); base_t::flags_ |= flag_started; base_t::flags_ |= flag_running; // create push_coroutine typename PushCoro::synth_type b( & this->callee, & this->caller, false); PushCoro push_coro( synthesized_t::syntesized, b); try { fn_( push_coro); } catch ( forced_unwind const&) {} #if defined( BOOST_CONTEXT_HAS_CXXABI_H ) catch ( abi::__forced_unwind const&) { throw; } #endif catch (...) { base_t::except_ = current_exception(); } base_t::flags_ |= flag_complete; base_t::flags_ &= ~flag_running; typename base_t::param_type to; this->callee.jump( this->caller, & to); BOOST_ASSERT_MSG( false, "pull_coroutine is complete"); } void destroy() { deallocate_( this); } }; template< typename PushCoro, typename R, typename Fn, typename StackAllocator > class pull_coroutine_object< PushCoro, R &, Fn, StackAllocator > : private pull_coroutine_context, public pull_coroutine_impl< R & > { private: typedef pull_coroutine_context ctx_t; typedef pull_coroutine_impl< R & > base_t; typedef pull_coroutine_object< PushCoro, R &, Fn, StackAllocator > obj_t; Fn fn_; stack_context stack_ctx_; StackAllocator stack_alloc_; static void deallocate_( obj_t * obj) { stack_context stack_ctx( obj->stack_ctx_); StackAllocator stack_alloc( obj->stack_alloc_); obj->unwind_stack(); obj->~obj_t(); stack_alloc.deallocate( stack_ctx); } public: #ifdef BOOST_NO_CXX11_RVALUE_REFERENCES pull_coroutine_object( Fn fn, attributes const& attrs, preallocated const& palloc, StackAllocator const& stack_alloc) BOOST_NOEXCEPT : ctx_t( palloc, this), base_t( & this->caller, & this->callee, stack_unwind == attrs.do_unwind), fn_( fn), stack_ctx_( palloc.sctx), stack_alloc_( stack_alloc) {} #endif pull_coroutine_object( BOOST_RV_REF( Fn) fn, attributes const& attrs, preallocated const& palloc, StackAllocator const& stack_alloc) BOOST_NOEXCEPT : ctx_t( palloc, this), base_t( & this->caller, & this->callee, stack_unwind == attrs.do_unwind), #ifdef BOOST_NO_CXX11_RVALUE_REFERENCES fn_( fn), #else fn_( boost::forward< Fn >( fn) ), #endif stack_ctx_( palloc.sctx), stack_alloc_( stack_alloc) {} void run() { BOOST_ASSERT( ! base_t::unwind_requested() ); base_t::flags_ |= flag_started; base_t::flags_ |= flag_running; // create push_coroutine typename PushCoro::synth_type b( & this->callee, & this->caller, false); PushCoro push_coro( synthesized_t::syntesized, b); try { fn_( push_coro); } catch ( forced_unwind const&) {} #if defined( BOOST_CONTEXT_HAS_CXXABI_H ) catch ( abi::__forced_unwind const&) { throw; } #endif catch (...) { base_t::except_ = current_exception(); } base_t::flags_ |= flag_complete; base_t::flags_ &= ~flag_running; typename base_t::param_type to; this->callee.jump( this->caller, & to); BOOST_ASSERT_MSG( false, "pull_coroutine is complete"); } void destroy() { deallocate_( this); } }; template< typename PushCoro, typename Fn, typename StackAllocator > class pull_coroutine_object< PushCoro, void, Fn, StackAllocator > : private pull_coroutine_context, public pull_coroutine_impl< void > { private: typedef pull_coroutine_context ctx_t; typedef pull_coroutine_impl< void > base_t; typedef pull_coroutine_object< PushCoro, void, Fn, StackAllocator > obj_t; Fn fn_; stack_context stack_ctx_; StackAllocator stack_alloc_; static void deallocate_( obj_t * obj) { stack_context stack_ctx( obj->stack_ctx_); StackAllocator stack_alloc( obj->stack_alloc_); obj->unwind_stack(); obj->~obj_t(); stack_alloc.deallocate( stack_ctx); } public: #ifdef BOOST_NO_CXX11_RVALUE_REFERENCES pull_coroutine_object( Fn fn, attributes const& attrs, preallocated const& palloc, StackAllocator const& stack_alloc) BOOST_NOEXCEPT : ctx_t( palloc, this), base_t( & this->caller, & this->callee, stack_unwind == attrs.do_unwind), fn_( fn), stack_ctx_( palloc.sctx), stack_alloc_( stack_alloc) {} #endif pull_coroutine_object( BOOST_RV_REF( Fn) fn, attributes const& attrs, preallocated const& palloc, StackAllocator const& stack_alloc) BOOST_NOEXCEPT : ctx_t( palloc, this), base_t( & this->caller, & this->callee, stack_unwind == attrs.do_unwind), #ifdef BOOST_NO_CXX11_RVALUE_REFERENCES fn_( fn), #else fn_( boost::forward< Fn >( fn) ), #endif stack_ctx_( palloc.sctx), stack_alloc_( stack_alloc) {} void run() { BOOST_ASSERT( ! base_t::unwind_requested() ); base_t::flags_ |= flag_started; base_t::flags_ |= flag_running; // create push_coroutine typename PushCoro::synth_type b( & this->callee, & this->caller, false); PushCoro push_coro( synthesized_t::syntesized, b); try { fn_( push_coro); } catch ( forced_unwind const&) {} #if defined( BOOST_CONTEXT_HAS_CXXABI_H ) catch ( abi::__forced_unwind const&) { throw; } #endif catch (...) { base_t::except_ = current_exception(); } base_t::flags_ |= flag_complete; base_t::flags_ &= ~flag_running; typename base_t::param_type to; this->callee.jump( this->caller, & to); BOOST_ASSERT_MSG( false, "pull_coroutine is complete"); } void destroy() { deallocate_( this); } }; }}} #if defined(BOOST_MSVC) # pragma warning(pop) #endif #ifdef BOOST_HAS_ABI_HEADERS # include BOOST_ABI_SUFFIX #endif #endif // BOOST_COROUTINES_DETAIL_PULL_COROUTINE_OBJECT_H