// // Copyright (c) 2013-2016 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) // #ifndef BEAST_HTTP_IMPL_MESSAGE_IPP #define BEAST_HTTP_IMPL_MESSAGE_IPP #include #include #include #include #include #include #include #include namespace beast { namespace http { template void swap( header& m1, header& m2) { using std::swap; swap(m1.version, m2.version); swap(m1.method, m2.method); swap(m1.url, m2.url); swap(m1.fields, m2.fields); } template void swap( header& a, header& b) { using std::swap; swap(a.version, b.version); swap(a.status, b.status); swap(a.reason, b.reason); swap(a.fields, b.fields); } template void swap( message& m1, message& m2) { using std::swap; swap(m1.base(), m2.base()); swap(m1.body, m2.body); } template bool is_keep_alive(header const& msg) { BOOST_ASSERT(msg.version == 10 || msg.version == 11); if(msg.version == 11) { if(token_list{msg.fields["Connection"]}.exists("close")) return false; return true; } if(token_list{msg.fields["Connection"]}.exists("keep-alive")) return true; return false; } template bool is_upgrade(header const& msg) { BOOST_ASSERT(msg.version == 10 || msg.version == 11); if(msg.version == 10) return false; if(token_list{msg.fields["Connection"]}.exists("upgrade")) return true; return false; } namespace detail { struct prepare_info { boost::optional connection_value; boost::optional content_length; }; template inline void prepare_options(prepare_info& pi, message& msg) { beast::detail::ignore_unused(pi, msg); } template void prepare_option(prepare_info& pi, message& msg, connection value) { beast::detail::ignore_unused(msg); pi.connection_value = value; } template< bool isRequest, class Body, class Fields, class Opt, class... Opts> void prepare_options(prepare_info& pi, message& msg, Opt&& opt, Opts&&... opts) { prepare_option(pi, msg, opt); prepare_options(pi, msg, std::forward(opts)...); } template void prepare_content_length(prepare_info& pi, message const& msg, std::true_type) { typename Body::writer w(msg); // VFALCO This is a design problem! error_code ec; w.init(ec); if(ec) throw system_error{ec}; pi.content_length = w.content_length(); } template void prepare_content_length(prepare_info& pi, message const& msg, std::false_type) { beast::detail::ignore_unused(msg); pi.content_length = boost::none; } } // detail template< bool isRequest, class Body, class Fields, class... Options> void prepare(message& msg, Options&&... options) { using beast::detail::make_exception; // VFALCO TODO static_assert(is_Body::value, "Body requirements not met"); static_assert(has_writer::value, "Body has no writer"); static_assert(is_Writer>::value, "Writer requirements not met"); detail::prepare_info pi; detail::prepare_content_length(pi, msg, detail::has_content_length{}); detail::prepare_options(pi, msg, std::forward(options)...); if(msg.fields.exists("Connection")) throw make_exception( "prepare called with Connection field set", __FILE__, __LINE__); if(msg.fields.exists("Content-Length")) throw make_exception( "prepare called with Content-Length field set", __FILE__, __LINE__); if(token_list{msg.fields["Transfer-Encoding"]}.exists("chunked")) throw make_exception( "prepare called with Transfer-Encoding: chunked set", __FILE__, __LINE__); if(pi.connection_value != connection::upgrade) { if(pi.content_length) { struct set_field { void operator()(message& msg, detail::prepare_info const& pi) const { using beast::detail::ci_equal; if(*pi.content_length > 0 || ci_equal(msg.method, "POST")) { msg.fields.insert( "Content-Length", *pi.content_length); } } void operator()(message& msg, detail::prepare_info const& pi) const { if((msg.status / 100 ) != 1 && msg.status != 204 && msg.status != 304) { msg.fields.insert( "Content-Length", *pi.content_length); } } }; set_field{}(msg, pi); } else if(msg.version >= 11) { msg.fields.insert("Transfer-Encoding", "chunked"); } } auto const content_length = msg.fields.exists("Content-Length"); if(pi.connection_value) { switch(*pi.connection_value) { case connection::upgrade: msg.fields.insert("Connection", "upgrade"); break; case connection::keep_alive: if(msg.version < 11) { if(content_length) msg.fields.insert("Connection", "keep-alive"); } break; case connection::close: if(msg.version >= 11) msg.fields.insert("Connection", "close"); break; } } // rfc7230 6.7. if(msg.version < 11 && token_list{ msg.fields["Connection"]}.exists("upgrade")) throw make_exception( "invalid version for Connection: upgrade", __FILE__, __LINE__); } } // http } // beast #endif