// // server.cpp // ~~~~~~~~~~ // // Copyright (c) 2003-2018 Christopher M. Kohlhoff (chris at kohlhoff 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) // #include <ctime> #include <iostream> #include <string> #include <boost/array.hpp> #include <boost/bind.hpp> #include <boost/shared_ptr.hpp> #include <boost/enable_shared_from_this.hpp> #include <asio.hpp> using asio::ip::tcp; using asio::ip::udp; std::string make_daytime_string() { using namespace std; // For time_t, time and ctime; time_t now = time(0); return ctime(&now); } class tcp_connection : public boost::enable_shared_from_this<tcp_connection> { public: typedef boost::shared_ptr<tcp_connection> pointer; static pointer create(asio::io_context& io_context) { return pointer(new tcp_connection(io_context)); } tcp::socket& socket() { return socket_; } void start() { message_ = make_daytime_string(); asio::async_write(socket_, asio::buffer(message_), boost::bind(&tcp_connection::handle_write, shared_from_this())); } private: tcp_connection(asio::io_context& io_context) : socket_(io_context) { } void handle_write() { } tcp::socket socket_; std::string message_; }; class tcp_server { public: tcp_server(asio::io_context& io_context) : acceptor_(io_context, tcp::endpoint(tcp::v4(), 13)) { start_accept(); } private: void start_accept() { tcp_connection::pointer new_connection = tcp_connection::create(acceptor_.get_executor().context()); acceptor_.async_accept(new_connection->socket(), boost::bind(&tcp_server::handle_accept, this, new_connection, asio::placeholders::error)); } void handle_accept(tcp_connection::pointer new_connection, const asio::error_code& error) { if (!error) { new_connection->start(); } start_accept(); } tcp::acceptor acceptor_; }; class udp_server { public: udp_server(asio::io_context& io_context) : socket_(io_context, udp::endpoint(udp::v4(), 13)) { start_receive(); } private: void start_receive() { socket_.async_receive_from( asio::buffer(recv_buffer_), remote_endpoint_, boost::bind(&udp_server::handle_receive, this, asio::placeholders::error)); } void handle_receive(const asio::error_code& error) { if (!error) { boost::shared_ptr<std::string> message( new std::string(make_daytime_string())); socket_.async_send_to(asio::buffer(*message), remote_endpoint_, boost::bind(&udp_server::handle_send, this, message)); start_receive(); } } void handle_send(boost::shared_ptr<std::string> /*message*/) { } udp::socket socket_; udp::endpoint remote_endpoint_; boost::array<char, 1> recv_buffer_; }; int main() { try { asio::io_context io_context; tcp_server server1(io_context); udp_server server2(io_context); io_context.run(); } catch (std::exception& e) { std::cerr << e.what() << std::endl; } return 0; }
Return to Daytime.7 - A combined TCP/UDP asynchronous server