forked from boostorg/beast
Vinnie Falco
ff5e659545
fix #124 The http::header data members "method", "url", and "reason" are changed from data members, to pairs of get and set functions which forward the call to the Fields type used to instantiate the template. Previously, these data members were implemented using std::string. With this change, the implementation of the Fields type used to instantiate the template is now in control of the representation of those values. This permits custom memory allocation strategies including uniform use of the Allocator type already provided to beast::http::basic_fields.
385 lines
11 KiB
C++
385 lines
11 KiB
C++
//
|
|
// Copyright (c) 2013-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)
|
|
//
|
|
|
|
#ifndef BEAST_WEBSOCKET_IMPL_HANDSHAKE_IPP
|
|
#define BEAST_WEBSOCKET_IMPL_HANDSHAKE_IPP
|
|
|
|
#include <beast/websocket/detail/type_traits.hpp>
|
|
#include <beast/http/message.hpp>
|
|
#include <beast/http/read.hpp>
|
|
#include <beast/http/write.hpp>
|
|
#include <beast/core/handler_helpers.hpp>
|
|
#include <beast/core/handler_ptr.hpp>
|
|
#include <beast/core/stream_concepts.hpp>
|
|
#include <boost/assert.hpp>
|
|
#include <memory>
|
|
|
|
namespace beast {
|
|
namespace websocket {
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
// send the upgrade request and process the response
|
|
//
|
|
template<class NextLayer>
|
|
template<class Handler>
|
|
class stream<NextLayer>::handshake_op
|
|
{
|
|
struct data
|
|
{
|
|
bool cont;
|
|
stream<NextLayer>& ws;
|
|
response_type* res_p;
|
|
detail::sec_ws_key_type key;
|
|
request_type req;
|
|
response_type res;
|
|
int state = 0;
|
|
|
|
template<class Decorator>
|
|
data(Handler& handler, stream<NextLayer>& ws_,
|
|
response_type* res_p_,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
Decorator const& decorator)
|
|
: cont(beast_asio_helpers::
|
|
is_continuation(handler))
|
|
, ws(ws_)
|
|
, res_p(res_p_)
|
|
, req(ws.build_request(key,
|
|
host, target, decorator))
|
|
{
|
|
ws.reset();
|
|
}
|
|
};
|
|
|
|
handler_ptr<data, Handler> d_;
|
|
|
|
public:
|
|
handshake_op(handshake_op&&) = default;
|
|
handshake_op(handshake_op const&) = default;
|
|
|
|
template<class DeducedHandler, class... Args>
|
|
handshake_op(DeducedHandler&& h,
|
|
stream<NextLayer>& ws, Args&&... args)
|
|
: d_(std::forward<DeducedHandler>(h),
|
|
ws, std::forward<Args>(args)...)
|
|
{
|
|
(*this)(error_code{}, false);
|
|
}
|
|
|
|
void
|
|
operator()(error_code ec, bool again = true);
|
|
|
|
friend
|
|
void* asio_handler_allocate(
|
|
std::size_t size, handshake_op* op)
|
|
{
|
|
return beast_asio_helpers::
|
|
allocate(size, op->d_.handler());
|
|
}
|
|
|
|
friend
|
|
void asio_handler_deallocate(
|
|
void* p, std::size_t size, handshake_op* op)
|
|
{
|
|
return beast_asio_helpers::
|
|
deallocate(p, size, op->d_.handler());
|
|
}
|
|
|
|
friend
|
|
bool asio_handler_is_continuation(handshake_op* op)
|
|
{
|
|
return op->d_->cont;
|
|
}
|
|
|
|
template<class Function>
|
|
friend
|
|
void asio_handler_invoke(Function&& f, handshake_op* op)
|
|
{
|
|
return beast_asio_helpers::
|
|
invoke(f, op->d_.handler());
|
|
}
|
|
};
|
|
|
|
template<class NextLayer>
|
|
template<class Handler>
|
|
void
|
|
stream<NextLayer>::handshake_op<Handler>::
|
|
operator()(error_code ec, bool again)
|
|
{
|
|
auto& d = *d_;
|
|
d.cont = d.cont || again;
|
|
while(! ec && d.state != 99)
|
|
{
|
|
switch(d.state)
|
|
{
|
|
case 0:
|
|
{
|
|
// send http upgrade
|
|
d.state = 1;
|
|
// VFALCO Do we need the ability to move
|
|
// a message on the async_write?
|
|
//
|
|
pmd_read(
|
|
d.ws.pmd_config_, d.req.fields);
|
|
http::async_write(d.ws.stream_,
|
|
d.req, std::move(*this));
|
|
// TODO We don't need d.req now. Figure
|
|
// out a way to make it a parameter instead
|
|
// of a state variable to reduce footprint.
|
|
return;
|
|
}
|
|
|
|
// sent upgrade
|
|
case 1:
|
|
// read http response
|
|
d.state = 2;
|
|
http::async_read(d.ws.next_layer(),
|
|
d.ws.stream_.buffer(), d.res,
|
|
std::move(*this));
|
|
return;
|
|
|
|
// got response
|
|
case 2:
|
|
{
|
|
d.ws.do_response(d.res, d.key, ec);
|
|
// call handler
|
|
d.state = 99;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if(d.res_p)
|
|
swap(d.res, *d.res_p);
|
|
d_.invoke(ec);
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class HandshakeHandler>
|
|
typename async_completion<HandshakeHandler,
|
|
void(error_code)>::result_type
|
|
stream<NextLayer>::
|
|
async_handshake(boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
HandshakeHandler&& handler)
|
|
{
|
|
static_assert(is_AsyncStream<next_layer_type>::value,
|
|
"AsyncStream requirements not met");
|
|
beast::async_completion<HandshakeHandler,
|
|
void(error_code)> completion{handler};
|
|
handshake_op<decltype(completion.handler)>{
|
|
completion.handler, *this, nullptr,
|
|
host, target, &default_decorate_req};
|
|
return completion.result.get();
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class HandshakeHandler>
|
|
typename async_completion<HandshakeHandler,
|
|
void(error_code)>::result_type
|
|
stream<NextLayer>::
|
|
async_handshake(response_type& res,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
HandshakeHandler&& handler)
|
|
{
|
|
static_assert(is_AsyncStream<next_layer_type>::value,
|
|
"AsyncStream requirements not met");
|
|
beast::async_completion<HandshakeHandler,
|
|
void(error_code)> completion{handler};
|
|
handshake_op<decltype(completion.handler)>{
|
|
completion.handler, *this, &res,
|
|
host, target, &default_decorate_req};
|
|
return completion.result.get();
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class RequestDecorator, class HandshakeHandler>
|
|
typename async_completion<HandshakeHandler,
|
|
void(error_code)>::result_type
|
|
stream<NextLayer>::
|
|
async_handshake_ex(boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
RequestDecorator const& decorator,
|
|
HandshakeHandler&& handler)
|
|
{
|
|
static_assert(is_AsyncStream<next_layer_type>::value,
|
|
"AsyncStream requirements not met");
|
|
static_assert(detail::is_RequestDecorator<
|
|
RequestDecorator>::value,
|
|
"RequestDecorator requirements not met");
|
|
beast::async_completion<HandshakeHandler,
|
|
void(error_code)> completion{handler};
|
|
handshake_op<decltype(completion.handler)>{
|
|
completion.handler, *this, nullptr,
|
|
host, target, decorator};
|
|
return completion.result.get();
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class RequestDecorator, class HandshakeHandler>
|
|
typename async_completion<HandshakeHandler,
|
|
void(error_code)>::result_type
|
|
stream<NextLayer>::
|
|
async_handshake_ex(response_type& res,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
RequestDecorator const& decorator,
|
|
HandshakeHandler&& handler)
|
|
{
|
|
static_assert(is_AsyncStream<next_layer_type>::value,
|
|
"AsyncStream requirements not met");
|
|
static_assert(detail::is_RequestDecorator<
|
|
RequestDecorator>::value,
|
|
"RequestDecorator requirements not met");
|
|
beast::async_completion<HandshakeHandler,
|
|
void(error_code)> completion{handler};
|
|
handshake_op<decltype(completion.handler)>{
|
|
completion.handler, *this, &res,
|
|
host, target, decorator};
|
|
return completion.result.get();
|
|
}
|
|
|
|
template<class NextLayer>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake(boost::string_ref const& host,
|
|
boost::string_ref const& target)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
error_code ec;
|
|
handshake(
|
|
host, target, ec);
|
|
if(ec)
|
|
throw system_error{ec};
|
|
}
|
|
|
|
template<class NextLayer>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake(response_type& res,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
error_code ec;
|
|
handshake(res, host, target, ec);
|
|
if(ec)
|
|
throw system_error{ec};
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class RequestDecorator>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake_ex(boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
RequestDecorator const& decorator)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
static_assert(detail::is_RequestDecorator<
|
|
RequestDecorator>::value,
|
|
"RequestDecorator requirements not met");
|
|
error_code ec;
|
|
handshake_ex(host, target, decorator, ec);
|
|
if(ec)
|
|
throw system_error{ec};
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class RequestDecorator>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake_ex(response_type& res,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
RequestDecorator const& decorator)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
static_assert(detail::is_RequestDecorator<
|
|
RequestDecorator>::value,
|
|
"RequestDecorator requirements not met");
|
|
error_code ec;
|
|
handshake_ex(res, host, target, decorator, ec);
|
|
if(ec)
|
|
throw system_error{ec};
|
|
}
|
|
|
|
template<class NextLayer>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake(boost::string_ref const& host,
|
|
boost::string_ref const& target, error_code& ec)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
do_handshake(nullptr,
|
|
host, target, &default_decorate_req, ec);
|
|
}
|
|
|
|
template<class NextLayer>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake(response_type& res,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
error_code& ec)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
do_handshake(&res,
|
|
host, target, &default_decorate_req, ec);
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class RequestDecorator>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake_ex(boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
RequestDecorator const& decorator,
|
|
error_code& ec)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
static_assert(detail::is_RequestDecorator<
|
|
RequestDecorator>::value,
|
|
"RequestDecorator requirements not met");
|
|
do_handshake(nullptr,
|
|
host, target, decorator, ec);
|
|
}
|
|
|
|
template<class NextLayer>
|
|
template<class RequestDecorator>
|
|
void
|
|
stream<NextLayer>::
|
|
handshake_ex(response_type& res,
|
|
boost::string_ref const& host,
|
|
boost::string_ref const& target,
|
|
RequestDecorator const& decorator,
|
|
error_code& ec)
|
|
{
|
|
static_assert(is_SyncStream<next_layer_type>::value,
|
|
"SyncStream requirements not met");
|
|
static_assert(detail::is_RequestDecorator<
|
|
RequestDecorator>::value,
|
|
"RequestDecorator requirements not met");
|
|
do_handshake(&res,
|
|
host, target, decorator, ec);
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
} // websocket
|
|
} // beast
|
|
|
|
#endif
|