mirror of
https://github.com/boostorg/beast.git
synced 2026-01-26 17:22:20 +01:00
74891560ae
fix #297, fix #488 * Errors are checked and reported * More comments explaining what is going on * The connection is gracefully closed WebSocket: * Messages are drained before closing
99 lines
2.7 KiB
C++
99 lines
2.7 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)
|
|
//
|
|
|
|
//[example_websocket_client
|
|
|
|
#include <beast/core.hpp>
|
|
#include <beast/websocket.hpp>
|
|
#include <boost/asio.hpp>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
int main()
|
|
{
|
|
// A helper for reporting errors
|
|
auto const fail =
|
|
[](std::string what, beast::error_code ec)
|
|
{
|
|
std::cerr << what << ": " << ec.message() << std::endl;
|
|
std::cerr.flush();
|
|
return EXIT_FAILURE;
|
|
};
|
|
|
|
boost::system::error_code ec;
|
|
|
|
// Set up an asio socket
|
|
boost::asio::io_service ios;
|
|
boost::asio::ip::tcp::resolver r{ios};
|
|
boost::asio::ip::tcp::socket sock{ios};
|
|
|
|
// Look up the domain name
|
|
std::string const host = "echo.websocket.org";
|
|
auto const lookup = r.resolve(boost::asio::ip::tcp::resolver::query{host, "http"}, ec);
|
|
if(ec)
|
|
return fail("resolve", ec);
|
|
|
|
// Make the connection on the IP address we get from a lookup
|
|
boost::asio::connect(sock, lookup, ec);
|
|
if(ec)
|
|
return fail("connect", ec);
|
|
|
|
// Wrap the now-connected socket in a websocket stream
|
|
beast::websocket::stream<boost::asio::ip::tcp::socket&> ws{sock};
|
|
|
|
// Perform the websocket handhskae
|
|
ws.handshake(host, "/", ec);
|
|
if(ec)
|
|
return fail("handshake", ec);
|
|
|
|
// Send a message
|
|
ws.write(boost::asio::buffer(std::string("Hello, world!")), ec);
|
|
if(ec)
|
|
return fail("write", ec);
|
|
|
|
// This buffer will hold the incoming message
|
|
beast::multi_buffer b;
|
|
|
|
// Read the message into our buffer
|
|
ws.read(b, ec);
|
|
if(ec)
|
|
return fail("read", ec);
|
|
|
|
// Send a "close" frame to the other end, this is a websocket thing
|
|
ws.close(beast::websocket::close_code::normal, ec);
|
|
if(ec)
|
|
return fail("close", ec);
|
|
|
|
// The buffers() function helps print a ConstBufferSequence
|
|
std::cout << beast::buffers(b.data()) << std::endl;
|
|
|
|
// WebSocket says that to close a connection you have
|
|
// to keep reading messages until you receive a close frame.
|
|
// Beast delivers the close frame as an error from read.
|
|
//
|
|
beast::drain_buffer drain; // Throws everything away efficiently
|
|
for(;;)
|
|
{
|
|
// Keep reading messages...
|
|
ws.read(drain, ec);
|
|
|
|
// ...until we get the special error code
|
|
if(ec == beast::websocket::error::closed)
|
|
break;
|
|
|
|
// Some other error occurred, report it and exit.
|
|
if(ec)
|
|
return fail("close", ec);
|
|
}
|
|
|
|
// If we get here the connection was cleanly closed
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
//]
|