C++WEBSOCKET client and server examples (BOOST.BEAST)

官方的example中仅仅输出到控制台,而且不支持中文,这里我加入了ansi到utf8的转换,使用utf8就能正常解析中文。

server:

//
// 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>
#include <locale>
#include <codecvt>
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>
std::wstring string_to_wstring(const std::string &s)
{
    using default_convert = std::codecvt<wchar_t, char, std::mbstate_t>;
    static std::wstring_convert<default_convert>conv(new default_convert("CHS"));
    return conv.from_bytes(s);
}
std::string wstring_to_string(const std::wstring &s)
{
    using default_convert = std::codecvt<wchar_t, char, std::mbstate_t>;
    static std::wstring_convert<default_convert>conv(new default_convert("CHS"));
    return conv.to_bytes(s);
}
std::string ansi_to_utf8(const std::string &s) 
{
    static std::wstring_convert<std::codecvt_utf8<wchar_t> > conv;
    return conv.to_bytes(string_to_wstring(s));
}
std::string utf8_to_ansi(const std::string& s)
{
    static std::wstring_convert<std::codecvt_utf8<wchar_t> > conv;
    return wstring_to_string(conv.from_bytes(s));
}
// Sends a WebSocket message and prints the response
int main(int argc, char** argv)
{
    try
    {
        // 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 };
        auto const address = net::ip::make_address("192.168.0.27");
        auto const port = static_cast<unsigned short>(std::atoi("80"));
        tcp::endpoint endpoint{ address, port };
        // Look up the domain name
        auto const results = resolver.resolve(endpoint);
        // Make the connection on the IP address we get from a lookup
        net::connect(ws.next_layer(), results.begin(), results.end());
        //net::connect(ws.next_layer(), host, port);
        // 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("0.0.0.0", "/");
        while (true)
        {
            std::string text;
            std::cin >> text;
            // Send the message
            ws.write(net::buffer(ansi_to_utf8(text)));
            // This buffer will hold the incoming message
            beast::flat_buffer buffer;
            // Read a message into our buffer
            ws.read(buffer);
            std::string out;
            out = beast::buffers_to_string(buffer.cdata());
            std::cout << utf8_to_ansi(out) << std::endl;
        }
        // Close the WebSocket connection
        ws.close(websocket::close_code::normal);
    }
    catch (std::exception const& e)
    {
        std::cerr << "Error: " << e.what() << std::endl;
        return EXIT_FAILURE;
    }
    return EXIT_SUCCESS;
}
//]
client:

//
// 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 server, synchronous
//
//------------------------------------------------------------------------------
#include <boost/beast/core.hpp>
#include <boost/beast/websocket.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <cstdlib>
#include <functional>
#include <iostream>
#include <string>
#include <thread>
#include <codecvt>
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>
std::wstring string_to_wstring(const std::string& s)
{
    using default_convert = std::codecvt<wchar_t, char, std::mbstate_t>;
    static std::wstring_convert<default_convert>conv(new default_convert("CHS"));
    return conv.from_bytes(s);
}
std::string wstring_to_string(const std::wstring& s)
{
    using default_convert = std::codecvt<wchar_t, char, std::mbstate_t>;
    static std::wstring_convert<default_convert>conv(new default_convert("CHS"));
    return conv.to_bytes(s);
}
std::string ansi_to_utf8(const std::string& s)
{
    static std::wstring_convert<std::codecvt_utf8<wchar_t> > conv;
    return conv.to_bytes(string_to_wstring(s));
}
std::string utf8_to_ansi(const std::string& s)
{
    static std::wstring_convert<std::codecvt_utf8<wchar_t> > conv;
    return wstring_to_string(conv.from_bytes(s));
}
//------------------------------------------------------------------------------
// Echoes back all received WebSocket messages
void
do_session(tcp::socket& socket)
{
    try
    {
        // Construct the stream by moving in the socket
        websocket::stream<tcp::socket> ws{ std::move(socket) };
        // Set a decorator to change the Server of the handshake
        ws.set_option(websocket::stream_base::decorator(
            [](websocket::response_type& res)
            {
                res.set(http::field::server,
                    std::string(BOOST_BEAST_VERSION_STRING) +
                    " websocket-server-sync");
            }));
        // Accept the websocket handshake
        ws.accept();
        for (;;)
        {
            // This buffer will hold the incoming message
            beast::flat_buffer buffer;
            // Read a message
            ws.read(buffer);
            auto out = beast::buffers_to_string(buffer.cdata());
            std::cout << utf8_to_ansi(out) << std::endl;
            // Echo the message back
            ws.text(ws.got_text());
            ws.write(buffer.data());
        }
    }
    catch (beast::system_error const& se)
    {
        // This indicates that the session was closed
        if (se.code() != websocket::error::closed)
            std::cerr << "Error: " << se.code().message() << std::endl;
    }
    catch (std::exception const& e)
    {
        std::cerr << "Error: " << e.what() << std::endl;
    }
}
//------------------------------------------------------------------------------
int main(int argc, char* argv[])
{
    try
    {
        auto const address = net::ip::make_address("0.0.0.0");
        auto const port = static_cast<unsigned short>(std::atoi("80"));
        // The io_context is required for all I/O
        net::io_context ioc{ 1 };
        // The acceptor receives incoming connections
        tcp::acceptor acceptor{ ioc, {address, port} };
        for (;;)
        {
            // This will receive the new connection
            tcp::socket socket{ ioc };
            // Block until we get a connection
            acceptor.accept(socket);
            // Launch the session, transferring ownership of the socket
            std::thread{ std::bind(
                &do_session,
                std::move(socket)) }.detach();
        }
    }
    catch (const std::exception & e)
    {
        std::cerr << "Error: " << e.what() << std::endl;
        return EXIT_FAILURE;
    }
}
Copy the code