// // Copyright (c) 2016-2017 Vinnie Falco (vinnie dot falco at gmail dot com) // // 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) // // Official repository: https://github.com/boostorg/beast // #ifndef BOOST_BEAST_WEBSOCKET_IMPL_ACCEPT_IPP #define BOOST_BEAST_WEBSOCKET_IMPL_ACCEPT_IPP #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace boost { namespace beast { namespace websocket { // Respond to an upgrade HTTP request template template class stream::response_op { struct data { stream& ws; response_type res; int step = 0; template data(Handler&, stream& ws_, http::request< Body, http::basic_fields> const& req, Decorator const& decorator) : ws(ws_) , res(ws_.build_response(req, decorator)) { } }; handler_ptr d_; public: response_op(response_op&&) = default; response_op(response_op const&) = default; template response_op(DeducedHandler&& h, stream& ws, Args&&... args) : d_(std::forward(h), ws, std::forward(args)...) { } template void operator()(Buffers const& buffers); void operator()(error_code ec); friend void* asio_handler_allocate( std::size_t size, response_op* op) { using boost::asio::asio_handler_allocate; return asio_handler_allocate( size, std::addressof(op->d_.handler())); } friend void asio_handler_deallocate( void* p, std::size_t size, response_op* op) { using boost::asio::asio_handler_deallocate; asio_handler_deallocate( p, size, std::addressof(op->d_.handler())); } friend bool asio_handler_is_continuation(response_op* op) { // VFALCO This will go away in Net-TS return false; } template friend void asio_handler_invoke(Function&& f, response_op* op) { using boost::asio::asio_handler_invoke; asio_handler_invoke( f, std::addressof(op->d_.handler())); } }; template template template void stream:: response_op:: operator()(Buffers const& buffers) { using boost::asio::buffer_copy; using boost::asio::buffer_size; auto& d = *d_; error_code ec; boost::optional mb; auto const len = buffer_size(buffers); try { mb.emplace(d.ws.rd_.buf.prepare(len)); } catch(std::length_error const&) { d.step = 2; ec = error::buffer_overflow; return d.ws.get_io_service().post( bind_handler(std::move(*this), ec)); } d.ws.rd_.buf.commit( buffer_copy(*mb, buffers)); (*this)(ec); } template template void stream:: response_op:: operator()(error_code ec) { auto& d = *d_; switch(d.step) { case 0: // send response d.step = 1; http::async_write(d.ws.next_layer(), d.res, std::move(*this)); return; // sent response case 1: if(d.res.result() != http::status::switching_protocols) ec = error::handshake_failed; if(! ec) { pmd_read(d.ws.pmd_config_, d.res); d.ws.open(role_type::server); } break; // call handler case 2: break; } d_.invoke(ec); } //------------------------------------------------------------------------------ // read and respond to an upgrade request // template template class stream::accept_op { struct data { stream& ws; Decorator decorator; http::request_parser p; int step = 0; data(Handler&, stream& ws_, Decorator const& decorator_) : ws(ws_) , decorator(decorator_) { } }; handler_ptr d_; public: accept_op(accept_op&&) = default; accept_op(accept_op const&) = default; template accept_op(DeducedHandler&& h, stream& ws, Args&&... args) : d_(std::forward(h), ws, std::forward(args)...) { } template void operator()(Buffers const& buffers); void operator()(error_code ec); friend void* asio_handler_allocate( std::size_t size, accept_op* op) { using boost::asio::asio_handler_allocate; return asio_handler_allocate( size, std::addressof(op->d_.handler())); } friend void asio_handler_deallocate( void* p, std::size_t size, accept_op* op) { using boost::asio::asio_handler_deallocate; asio_handler_deallocate( p, size, std::addressof(op->d_.handler())); } friend bool asio_handler_is_continuation(accept_op* op) { using boost::asio::asio_handler_is_continuation; return asio_handler_is_continuation( std::addressof(op->d_.handler())); } template friend void asio_handler_invoke(Function&& f, accept_op* op) { using boost::asio::asio_handler_invoke; asio_handler_invoke( f, std::addressof(op->d_.handler())); } }; template template template void stream:: accept_op:: operator()(Buffers const& buffers) { using boost::asio::buffer_copy; using boost::asio::buffer_size; auto& d = *d_; error_code ec; boost::optional mb; auto const len = buffer_size(buffers); try { mb.emplace(d.ws.rd_.buf.prepare(len)); } catch(std::length_error const&) { d.step = 2; ec = error::buffer_overflow; return d.ws.get_io_service().post( bind_handler(std::move(*this), ec)); } d.ws.rd_.buf.commit( buffer_copy(*mb, buffers)); (*this)(ec); } template template void stream:: accept_op:: operator()(error_code ec) { auto& d = *d_; switch(d.step) { case 0: d.step = 1; return http::async_read( d.ws.next_layer(), d.ws.rd_.buf, d.p, std::move(*this)); case 1: { if(ec) break; // Arguments from our step must be // moved to the stack before releasing // the handler. auto& ws = d.ws; auto const req = d.p.release(); auto const decorator = d.decorator; #if 1 return response_op{ d_.release_handler(), ws, req, decorator}(ec); #else // VFALCO This *should* work but breaks // coroutine invariants in the unit test. // Also it calls reset() when it shouldn't. return ws.async_accept_ex( req, decorator, d_.release_handler()); #endif } } d_.invoke(ec); } //------------------------------------------------------------------------------ template void stream:: accept() { static_assert(is_sync_stream::value, "SyncStream requirements not met"); error_code ec; accept(ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template void stream:: accept_ex(ResponseDecorator const& decorator) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); error_code ec; accept_ex(decorator, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template void stream:: accept(error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); reset(); do_accept(&default_decorate_res, ec); } template template void stream:: accept_ex(ResponseDecorator const& decorator, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); reset(); do_accept(decorator, ec); } template template typename std::enable_if::value>::type stream:: accept(ConstBufferSequence const& buffers) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); error_code ec; accept(buffers, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template< class ConstBufferSequence, class ResponseDecorator> typename std::enable_if::value>::type stream:: accept_ex(ConstBufferSequence const& buffers, ResponseDecorator const &decorator) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); error_code ec; accept_ex(buffers, decorator, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template typename std::enable_if::value>::type stream:: accept(ConstBufferSequence const& buffers, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); reset(); using boost::asio::buffer_copy; using boost::asio::buffer_size; rd_.buf.commit(buffer_copy( rd_.buf.prepare( buffer_size(buffers)), buffers)); do_accept(&default_decorate_res, ec); } template template< class ConstBufferSequence, class ResponseDecorator> typename std::enable_if::value>::type stream:: accept_ex(ConstBufferSequence const& buffers, ResponseDecorator const& decorator, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); reset(); using boost::asio::buffer_copy; using boost::asio::buffer_size; rd_.buf.commit(buffer_copy( rd_.buf.prepare( buffer_size(buffers)), buffers)); do_accept(decorator, ec); } template template void stream:: accept(http::request> const& req) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); error_code ec; accept(req, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template void stream:: accept_ex(http::request> const& req, ResponseDecorator const& decorator) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); error_code ec; accept_ex(req, decorator, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template void stream:: accept(http::request> const& req, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); reset(); do_accept(req, &default_decorate_res, ec); } template template void stream:: accept_ex(http::request> const& req, ResponseDecorator const& decorator, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); reset(); do_accept(req, decorator, ec); } template template void stream:: accept(http::request> const& req, ConstBufferSequence const& buffers) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); error_code ec; accept(req, buffers, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template void stream:: accept_ex(http::request> const& req, ConstBufferSequence const& buffers, ResponseDecorator const& decorator) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); error_code ec; accept_ex(req, buffers, decorator, ec); if(ec) BOOST_THROW_EXCEPTION(system_error{ec}); } template template void stream:: accept(http::request> const& req, ConstBufferSequence const& buffers, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); reset(); using boost::asio::buffer_copy; using boost::asio::buffer_size; rd_.buf.commit(buffer_copy( rd_.buf.prepare( buffer_size(buffers)), buffers)); do_accept(req, &default_decorate_res, ec); } template template void stream:: accept_ex(http::request> const& req, ConstBufferSequence const& buffers, ResponseDecorator const& decorator, error_code& ec) { static_assert(is_sync_stream::value, "SyncStream requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); reset(); using boost::asio::buffer_copy; using boost::asio::buffer_size; rd_.buf.commit(buffer_copy( rd_.buf.prepare( buffer_size(buffers)), buffers)); do_accept(req, decorator, ec); } //------------------------------------------------------------------------------ template template< class AcceptHandler> async_return_type stream:: async_accept( AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); async_completion init{handler}; reset(); accept_op< decltype(&default_decorate_res), handler_type>{ init.completion_handler, *this, &default_decorate_res}({}); return init.result.get(); } template template< class ResponseDecorator, class AcceptHandler> async_return_type stream:: async_accept_ex( ResponseDecorator const& decorator, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); async_completion init{handler}; reset(); accept_op< ResponseDecorator, handler_type>{ init.completion_handler, *this, decorator}({}); return init.result.get(); } template template< class ConstBufferSequence, class AcceptHandler> typename std::enable_if< ! http::detail::is_header::value, async_return_type>::type stream:: async_accept( ConstBufferSequence const& buffers, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); async_completion init{handler}; reset(); accept_op< decltype(&default_decorate_res), handler_type>{ init.completion_handler, *this, &default_decorate_res}(buffers); return init.result.get(); } template template< class ConstBufferSequence, class ResponseDecorator, class AcceptHandler> typename std::enable_if< ! http::detail::is_header::value, async_return_type>::type stream:: async_accept_ex( ConstBufferSequence const& buffers, ResponseDecorator const& decorator, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); async_completion init{handler}; reset(); accept_op< ResponseDecorator, handler_type>{ init.completion_handler, *this, decorator}(buffers); return init.result.get(); } template template< class Body, class Allocator, class AcceptHandler> async_return_type stream:: async_accept( http::request> const& req, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); async_completion init{handler}; reset(); using boost::asio::asio_handler_is_continuation; response_op< handler_type>{ init.completion_handler, *this, req, &default_decorate_res}({}); return init.result.get(); } template template< class Body, class Allocator, class ResponseDecorator, class AcceptHandler> async_return_type stream:: async_accept_ex( http::request> const& req, ResponseDecorator const& decorator, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); async_completion init{handler}; reset(); using boost::asio::asio_handler_is_continuation; response_op< handler_type>{ init.completion_handler, *this, req, decorator}({}); return init.result.get(); } template template< class Body, class Allocator, class ConstBufferSequence, class AcceptHandler> async_return_type stream:: async_accept( http::request> const& req, ConstBufferSequence const& buffers, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); async_completion init{handler}; reset(); using boost::asio::asio_handler_is_continuation; response_op< handler_type>{ init.completion_handler, *this, req, &default_decorate_res}(buffers); return init.result.get(); } template template< class Body, class Allocator, class ConstBufferSequence, class ResponseDecorator, class AcceptHandler> async_return_type< AcceptHandler, void(error_code)> stream:: async_accept_ex(http::request> const& req, ConstBufferSequence const& buffers, ResponseDecorator const& decorator, AcceptHandler&& handler) { static_assert(is_async_stream::value, "AsyncStream requirements requirements not met"); static_assert(is_const_buffer_sequence< ConstBufferSequence>::value, "ConstBufferSequence requirements not met"); static_assert(detail::is_ResponseDecorator< ResponseDecorator>::value, "ResponseDecorator requirements not met"); async_completion init{handler}; reset(); using boost::asio::asio_handler_is_continuation; response_op< handler_type>{ init.completion_handler, *this, req, decorator}(buffers); return init.result.get(); } //------------------------------------------------------------------------------ template template void stream:: do_accept( Decorator const& decorator, error_code& ec) { http::request_parser p; http::read(next_layer(), rd_.buf, p, ec); if(ec) return; do_accept(p.get(), decorator, ec); } template template void stream:: do_accept( http::request> const& req, Decorator const& decorator, error_code& ec) { auto const res = build_response(req, decorator); http::write(stream_, res, ec); if(ec) return; if(res.result() != http::status::switching_protocols) { ec = error::handshake_failed; // VFALCO TODO Respect keep alive setting, perform // teardown if Connection: close. return; } pmd_read(pmd_config_, req); open(role_type::server); } } // websocket } // beast } // boost #endif