106 lines
3.0 KiB
C++
106 lines
3.0 KiB
C++
//
|
|
// async_tcp_echo_server.cpp
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
//
|
|
// Copyright (c) 2003-2008 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 <cstdlib>
|
|
#include <iostream>
|
|
#include <boost/bind.hpp>
|
|
#include <boost/asio.hpp>
|
|
|
|
using boost::asio::ip::tcp;
|
|
|
|
class session {
|
|
public:
|
|
session(boost::asio::io_service& io_service) :
|
|
socket_(io_service) {
|
|
}
|
|
|
|
tcp::socket& socket() {
|
|
return socket_;
|
|
}
|
|
|
|
void start() {
|
|
socket_.async_read_some(boost::asio::buffer(data_, max_length),
|
|
boost::bind(&session::handle_read, this,
|
|
boost::asio::placeholders::error,
|
|
boost::asio::placeholders::bytes_transferred));
|
|
}
|
|
|
|
void handle_read(const boost::system::error_code& error,
|
|
size_t bytes_transferred) {
|
|
if (!error) {
|
|
boost::asio::async_write(socket_, boost::asio::buffer(data_,
|
|
bytes_transferred), boost::bind(&session::handle_write,
|
|
this, boost::asio::placeholders::error));
|
|
} else {
|
|
delete this;
|
|
}
|
|
}
|
|
|
|
void handle_write(const boost::system::error_code& error) {
|
|
if (!error) {
|
|
socket_.async_read_some(boost::asio::buffer(data_, max_length),
|
|
boost::bind(&session::handle_read, this,
|
|
boost::asio::placeholders::error,
|
|
boost::asio::placeholders::bytes_transferred));
|
|
} else {
|
|
delete this;
|
|
}
|
|
}
|
|
|
|
private:
|
|
tcp::socket socket_;
|
|
enum {
|
|
max_length = 1024
|
|
};
|
|
char data_[max_length];
|
|
};
|
|
|
|
class server {
|
|
public:
|
|
server(boost::asio::io_service& io_service, short port) :
|
|
io_service_(io_service), acceptor_(io_service, tcp::endpoint(tcp::v4(),
|
|
port)) {
|
|
session* new_session = new session(io_service_);
|
|
acceptor_.async_accept(new_session->socket(), boost::bind(
|
|
&server::handle_accept, this, new_session,
|
|
boost::asio::placeholders::error));
|
|
}
|
|
|
|
void handle_accept(session* new_session,
|
|
const boost::system::error_code& error) {
|
|
if (!error) {
|
|
new_session->start();
|
|
new_session = new session(io_service_);
|
|
acceptor_.async_accept(new_session->socket(), boost::bind(
|
|
&server::handle_accept, this, new_session,
|
|
boost::asio::placeholders::error));
|
|
} else {
|
|
delete new_session;
|
|
}
|
|
}
|
|
|
|
private:
|
|
boost::asio::io_service& io_service_;
|
|
tcp::acceptor acceptor_;
|
|
};
|
|
|
|
int main(int argc, char** argv) {
|
|
if (argc < 2) {
|
|
printf("Usage: cmd port\n");
|
|
return -10;
|
|
}
|
|
int port = atoi(argv[1]);
|
|
|
|
boost::asio::io_service io_service;
|
|
server s(io_service, port);
|
|
io_service.run();
|
|
|
|
return 0;
|
|
}
|