101 lines
3.2 KiB
C++
101 lines
3.2 KiB
C++
//
|
|
// Copyright (c) 2016-2019 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
|
|
//
|
|
|
|
//------------------------------------------------------------------------------
|
|
//
|
|
// Example: WebSocket client, synchronous
|
|
//
|
|
//------------------------------------------------------------------------------
|
|
|
|
//[example_websocket_client
|
|
|
|
#include <boost/beast/core.hpp>
|
|
#include <boost/beast/websocket.hpp>
|
|
#include <boost/asio/connect.hpp>
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
namespace beast = boost::beast; // from <boost/beast.hpp>
|
|
namespace http = beast::http; // from <boost/beast/http.hpp>
|
|
namespace websocket = beast::websocket; // from <boost/beast/websocket.hpp>
|
|
namespace net = boost::asio; // from <boost/asio.hpp>
|
|
using tcp = boost::asio::ip::tcp; // from <boost/asio/ip/tcp.hpp>
|
|
|
|
// Sends a WebSocket message and prints the response
|
|
int main(int argc, char** argv)
|
|
{
|
|
try
|
|
{
|
|
// Check command line arguments.
|
|
if(argc != 4)
|
|
{
|
|
std::cerr <<
|
|
"Usage: websocket-client-sync <host> <port> <text>\n" <<
|
|
"Example:\n" <<
|
|
" websocket-client-sync echo.websocket.org 80 \"Hello, world!\"\n";
|
|
return EXIT_FAILURE;
|
|
}
|
|
auto const host = argv[1];
|
|
auto const port = argv[2];
|
|
auto const text = argv[3];
|
|
|
|
// The io_context is required for all I/O
|
|
net::io_context ioc;
|
|
|
|
// These objects perform our I/O
|
|
tcp::resolver resolver{ioc};
|
|
websocket::stream<tcp::socket> ws{ioc};
|
|
|
|
// Look up the domain name
|
|
auto const results = resolver.resolve(host, port);
|
|
|
|
// Make the connection on the IP address we get from a lookup
|
|
net::connect(ws.next_layer(), results.begin(), results.end());
|
|
|
|
// Set a decorator to change the User-Agent of the handshake
|
|
ws.set_option(websocket::stream_base::decorator(
|
|
[](websocket::request_type& req)
|
|
{
|
|
req.set(http::field::user_agent,
|
|
std::string(BOOST_BEAST_VERSION_STRING) +
|
|
" websocket-client-coro");
|
|
}));
|
|
|
|
// Perform the websocket handshake
|
|
ws.handshake(host, "/");
|
|
|
|
// Send the message
|
|
ws.write(net::buffer(std::string(text)));
|
|
|
|
// This buffer will hold the incoming message
|
|
beast::flat_buffer buffer;
|
|
|
|
// Read a message into our buffer
|
|
ws.read(buffer);
|
|
|
|
// Close the WebSocket connection
|
|
ws.close(websocket::close_code::normal);
|
|
|
|
// If we get here then the connection is closed gracefully
|
|
|
|
// The make_printable() function helps print a ConstBufferSequence
|
|
std::cout << beast::make_printable(buffer.data()) << std::endl;
|
|
}
|
|
catch(std::exception const& e)
|
|
{
|
|
std::cerr << "Error: " << e.what() << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
//]
|