forked from boostorg/mqtt5
Korina Šimičević
d19f466e3e
Summary: resolves T15243 Reviewers: ivica Reviewed By: ivica Subscribers: iljazovic, miljen Maniphest Tasks: T15243 Differential Revision: https://repo.mireo.local/D32561
55 lines
1.4 KiB
C++
55 lines
1.4 KiB
C++
//
|
|
// Copyright (c) 2023-2024 Ivica Siladic, Bruno Iljazovic, Korina Simicevic
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0.
|
|
// (See accompanying file LICENSE or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef ASYNC_MQTT5_WEBSOCKET_HPP
|
|
#define ASYNC_MQTT5_WEBSOCKET_HPP
|
|
|
|
#include <boost/beast/http/field.hpp>
|
|
#include <boost/beast/websocket/rfc6455.hpp>
|
|
#include <boost/beast/websocket/stream.hpp>
|
|
|
|
#include <async_mqtt5/types.hpp>
|
|
|
|
namespace async_mqtt5 {
|
|
|
|
// Trait definition for Beast
|
|
template <typename Stream>
|
|
struct ws_handshake_traits<boost::beast::websocket::stream<Stream>> {
|
|
|
|
template <typename CompletionToken>
|
|
static decltype(auto) async_handshake(
|
|
boost::beast::websocket::stream<Stream>& stream,
|
|
authority_path ap, CompletionToken&& token
|
|
) {
|
|
using namespace boost::beast;
|
|
|
|
// Set suggested timeout settings for the websocket
|
|
stream.set_option(
|
|
websocket::stream_base::timeout::suggested(role_type::client)
|
|
);
|
|
|
|
stream.binary(true);
|
|
|
|
// Set a decorator to change the User-Agent of the handshake
|
|
stream.set_option(websocket::stream_base::decorator(
|
|
[](websocket::request_type& req) {
|
|
req.set(http::field::sec_websocket_protocol, "mqtt");
|
|
req.set(http::field::user_agent, "boost.mqtt");
|
|
})
|
|
);
|
|
|
|
stream.async_handshake(
|
|
ap.host + ':' + ap.port, ap.path,
|
|
std::forward<CompletionToken>(token)
|
|
);
|
|
}
|
|
};
|
|
|
|
} // end namespace async_mqtt5
|
|
|
|
#endif // !ASYNC_MQTT5_WEBSOCKET_HPP
|