boost::asio编写的异步服务器和客户端

本文介绍使用 Boost.Asio 库实现 TCP 协议下的客户端和服务端程序。通过设置非阻塞选项,利用异步读写完成客户端与服务器之间的交互。客户端发送“Hello World!”消息到服务器并接收回应。

客户端


#include<boost/asio/io_service.hpp>
#include<boost/asio/ip/tcp.hpp>
#include<boost/bind.hpp>
#include<boost/shared_ptr.hpp>
#include<boost/enable_shared_from_this.hpp>
#include<string>
#include<iostream>
#include<boost/asio/streambuf.hpp>
#include<boost/asio/placeholders.hpp>
#include<boost/asio.hpp>

using boost::asio::ip::tcp;
using boost::asio::ip::address;

class client : public boost::enable_shared_from_this<client> {
public:
	client(boost::asio::io_service &io_service, tcp::endpoint &endpoint)
		: io_service_(io_service), socket_(io_service), endpoint_(endpoint)
	{
	}

	void start() {
		socket_.async_connect(endpoint_,
			boost::bind(&client::handle_connect,
			shared_from_this(),
			boost::asio::placeholders::error));
	}

private:
	void handle_connect(const boost::system::error_code &error) {
		if (error) {
			if (error.value() != boost::system::errc::operation_canceled) {
				std::cerr << boost::system::system_error(error).what() << std::endl;
			}

			socket_.close();
			return;
		}

		static tcp::no_delay option(true);
		socket_.set_option(option);

		strcpy(buf, "Hello World!\n");
		boost::asio::async_write(socket_,
			boost::asio::buffer(buf, strlen(buf)),
			boost::bind(&client::handle_write,
			shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
	}

	void handle_write(const boost::system::error_code& error, size_t bytes_transferred) {
		memset(buf, sizeof(buf), 0);
		boost::asio::async_read_until(socket_,	sbuf,"\n",
			boost::bind(&client::handle_read,
			shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
	}

	void handle_read(const boost::system::error_code& error, size_t bytes_transferred) {
		std::cout << buf << std::endl;
	}

private:
	boost::asio::io_service &io_service_;
	tcp::socket socket_;
	tcp::endpoint &endpoint_;
	char buf[1024];
	boost::asio::streambuf sbuf;
};

typedef boost::shared_ptr<client> client_ptr;

int main(int argc, char* argv[])
{
	boost::asio::io_service io_service;
	tcp::endpoint endpoint(address::from_string("127.0.0.1"), 1000);

	client_ptr new_session(new client(io_service, endpoint));
	new_session->start();
	io_service.run();
	std::string ss;
	std::cin >> ss;
	return 0;
}

服务器

#include <string.h>
#include <boost/asio.hpp>
#include <boost/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>

using boost::asio::ip::tcp;
using boost::asio::ip::address;

class session : public boost::enable_shared_from_this<session> {
public:
	session(boost::asio::io_service &io_service) : socket_(io_service)
	{
	}

	void start() {
		static tcp::no_delay option(true);
		socket_.set_option(option);

		boost::asio::async_read_until(socket_,
			sbuf_,
			"\n",
			boost::bind(&session::handle_read,
			shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
	}

	tcp::socket &socket() {
		return socket_;
	}

private:
	void handle_write(const boost::system::error_code& error, size_t bytes_transferred) {
		boost::asio::async_read_until(socket_,
			sbuf_,
			"\n",
			boost::bind(&session::handle_read,
			shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
	}

	void handle_read(const boost::system::error_code& error, size_t bytes_transferred) {
		boost::asio::async_write(socket_,
			sbuf_,
			boost::bind(&session::handle_write,
			shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
	}

private:
	tcp::socket socket_;
	boost::asio::streambuf sbuf_;
};

typedef boost::shared_ptr<session> session_ptr;

class server {
public:
	server(boost::asio::io_service &io_service, tcp::endpoint &endpoint)
		: io_service_(io_service), acceptor_(io_service, endpoint)
	{
		session_ptr 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_ptr new_session, const boost::system::error_code& error) {
		if (error) {
			return;
		}

		new_session->start();
		new_session.reset(new session(io_service_));
		acceptor_.async_accept(new_session->socket(),boost::bind(&server::handle_accept,this,new_session,
			boost::asio::placeholders::error));
	}

	void run() {
		io_service_.run();
	}

private:
	boost::asio::io_service &io_service_;
	tcp::acceptor acceptor_;
};

int main(int argc, char* argv[])
{
	boost::asio::io_service io_service;
	tcp::endpoint endpoint(tcp::v4(), 1000);

	server s(io_service, endpoint);
	s.run();
	return 0;
}


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值