tcp_echo_server.cpp

Go to the documentation of this file.

#include <iostream>
#include <libwire/tcp/listener.hpp>

int main(int argc, char** argv) {
    using namespace libwire;
    using namespace std::literals::string_view_literals;

    if (argc != 2) {
        std::cerr << "Usage: echo-server <port>\n";
        return 1;
    }

    uint16_t port = std::stoi(argv[1]);

    tcp::listener listener;
    listener.listen({ipv4::any, port});

    std::cout << "Listening on port " << port << ".\n";

    std::string buf;
    while (true) {
        auto sock = listener.accept();
        endpoint source = sock.remote_endpoint();

        std::cout << "Accepted connection from " << source.to_string() << ".\n";

        while (true) {
            try {
                sock.read_until('\n', buf);
                std::cout << "< " << buf << '\n';
                sock.write(buf);
                sock.write("\n"sv);
                std::cout << "> " << buf << '\n';
            } catch (std::system_error&) {
                break;
            }
        }

        std::cout << "Disconnected.\n";
    }
}