HTML Diff local/stream

src/examples/cpp03/local/stream_server.cpp		src/examples/cpp11/local/stream_server.cpp
⋮		⋮
1	//	1	//
2	//·stream_server.cpp	2	//·stream_server.cpp
3	//·~~~~~~~~~~~~~~~~~	3	//·~~~~~~~~~~~~~~~~~
4	//	4	//
5	//·Copyright·(c)·2003-2020·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com)	5	//·Copyright·(c)·2003-2020·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com)
6	//	6	//
7	//·Distributed·under·the·Boost·Software·License,·Version·1.0.·(See·accompanying	7	//·Distributed·under·the·Boost·Software·License,·Version·1.0.·(See·accompanying
8	//·file·LICENSE_1_0.txt·or·copy·at·http://www.boost.org/LICENSE_1_0.txt)	8	//·file·LICENSE_1_0.txt·or·copy·at·http://www.boost.org/LICENSE_1_0.txt)
9	//	9	//
10		10
		11	#include·<array>
11	#include·<cstdio>	12	#include·<cstdio>
12	#include·<iostream>	13	#include·<iostream>
13	#include·<boost/array.hpp>	14	#include·<memory>
14	#include·<boost/bind/bind.hpp>
15	#include·<boost/enable_shared_from_this.hpp>
16	#include·<boost/shared_ptr.hpp>
17	#include·"asio.hpp"	15	#include·"asio.hpp"
18		16
19	#if·defined(ASIO_HAS_LOCAL_SOCKETS)	17	#if·defined(ASIO_HAS_LOCAL_SOCKETS)
20		18
21	using·asio::local::stream_protocol;	19	using·asio::local::stream_protocol;
22		20
23	class·session	21	class·session
24	··:·public·boost::enable_shared_from_this<session>	22	··:·public·std::enable_shared_from_this<session>
25	{	23	{
26	public:	24	public:
27	··session(asio::io_context&·io_context)	25	··session(stream_protocol::socket·sock)
28	····:·socket_(io_context)	26	····:·socket_(std::move(sock))
29	··{	27	··{
30	··}	28	··}
31		29
32	··stream_protocol::socket&·socket()
33	··{
34	····return·socket_;
35	··}
36
37	··void·start()	30	··void·start()
38	··{	31	··{
39	····socket_.async_read_some(asio::buffer(data_),	32	····do_read();
40	········boost::bind(&session::handle_read,
41	··········shared_from_this(),
42	··········asio::placeholders::error,
43	··········asio::placeholders::bytes_transferred));
44	··}
45
46	··void·handle_read(const·asio::error_code&·error,
47	······size_t·bytes_transferred)
48	··{
49	····if·(!error)
50	····{
51	······asio::async_write(socket_,
52	··········asio::buffer(data_,·bytes_transferred),
53	··········boost::bind(&session::handle_write,
54	············shared_from_this(),
55	············asio::placeholders::error));
56	····}
57	··}	33	··}
58		34
59	··void·handle_write(const·asio::error_code&·error)	35	private:
		36	··void·do_read()
60	··{	37	··{
61	····if·(!error)	38	····auto·self(shared_from_this());
62	····{	39	····socket_.async_read_some(asio::buffer(data_),
63	······socket_.async_read_some(asio::buffer(data_),	40	········[this,·self](std::error_code·ec,·std::size_t·length)
64	··········boost::bind(&session::handle_read,	41	········{
65	············shared_from_this(),	42	··········if·(!ec)
66	············asio::placeholders::error,	43	············do_write(length);
67	············asio::placeholders::bytes_transferred));	44	········});
68	····}	45	··}
		46
		47	··void·do_write(std::size_t·length)
		48	··{
		49	····auto·self(shared_from_this());
		50	····asio::async_write(socket_,
		51	········asio::buffer(data_,·length),
		52	········[this,·self](std::error_code·ec,·std::size_t·/length/)
		53	········{
		54	··········if·(!ec)
		55	············do_read();
		56	········});
69	··}	57	··}
70		58
71	private:
72	··//·The·socket·used·to·communicate·with·the·client.	59	··//·The·socket·used·to·communicate·with·the·client.
73	··stream_protocol::socket·socket_;	60	··stream_protocol::socket·socket_;
74		61
75	··//·Buffer·used·to·store·data·received·from·the·client.	62	··//·Buffer·used·to·store·data·received·from·the·client.
76	··boost::array<char,·1024>·data_;	63	··std::array<char,·1024>·data_;
77	};	64	};
78		65
79	typedef·boost::shared_ptr<session>·session_ptr;
80
81	class·server	66	class·server
82	{	67	{
83	public:	68	public:
84	··server(asio::io_context&·io_context,·const·std::string&·file)	69	··server(asio::io_context&·io_context,·const·std::string&·file)
85	····:·io_context_(io_context),	70	····:·acceptor_(io_context,·stream_protocol::endpoint(file))
86	······acceptor_(io_context,·stream_protocol::endpoint(file))
87	··{	71	··{
88	····session_ptr·new_session(new·session(io_context_));	72	····do_accept();
89	····acceptor_.async_accept(new_session->socket(),
90	········boost::bind(&server::handle_accept,·this,·new_session,
91	··········asio::placeholders::error));
92	··}	73	··}
93		74
94	··void·handle_accept(session_ptr·new_session,	75	private:
95	······const·asio::error_code&·error)	76	··void·do_accept()
96	··{	77	··{
97	····if·(!error)	78	····acceptor_.async_accept(
98	····{	79	········[this](std::error_code·ec,·stream_protocol::socket·socket)
99	······new_session->start();	80	········{
100	····}	81	··········if·(!ec)
		82	··········{
		83	············std::make_shared<session>(std::move(socket))->start();
		84	··········}
101		85
102	····new_session.reset(new·session(io_context_));	86	··········do_accept();
103	····acceptor_.async_accept(new_session->socket(),	87	········});
104	········boost::bind(&server::handle_accept,·this,·new_session,
105	··········asio::placeholders::error));
106	··}	88	··}
107		89
108	private:
109	··asio::io_context&·io_context_;
110	··stream_protocol::acceptor·acceptor_;	90	··stream_protocol::acceptor·acceptor_;
111	};	91	};
112		92
113	int·main(int·argc,·char*·argv[])	93	int·main(int·argc,·char*·argv[])
114	{	94	{
115	··try	95	··try
116	··{	96	··{
117	····if·(argc·!=·2)	97	····if·(argc·!=·2)
118	····{	98	····{
119	······std::cerr·<<·"Usage:·stream_server·<file>\n";	99	······std::cerr·<<·"Usage:·stream_server·<file>\n";
120	······std::cerr·<<·"*·WARNING:·existing·file·is·removed·*\n";	100	······std::cerr·<<·"*·WARNING:·existing·file·is·removed·*\n";
121	······return·1;	101	······return·1;
122	····}	102	····}
123		103
124	····asio::io_context·io_context;	104	····asio::io_context·io_context;
125		105
126	····std::remove(argv[1]);	106	····std::remove(argv[1]);
127	····server·s(io_context,·argv[1]);	107	····server·s(io_context,·argv[1]);
128		108
129	····io_context.run();	109	····io_context.run();
130	··}	110	··}
131	··catch·(std::exception&·e)	111	··catch·(std::exception&·e)
132	··{	112	··{
133	····std::cerr·<<·"Exception:·"·<<·e.what()·<<·"\n";	113	····std::cerr·<<·"Exception:·"·<<·e.what()·<<·"\n";
134	··}	114	··}
135		115
136	··return·0;	116	··return·0;
137	}	117	}
138		118
139	#else·//·defined(ASIO_HAS_LOCAL_SOCKETS)	119	#else·//·defined(ASIO_HAS_LOCAL_SOCKETS)
140	#·error·Local·sockets·not·available·on·this·platform.	120	#·error·Local·sockets·not·available·on·this·platform.
141	#endif·//·defined(ASIO_HAS_LOCAL_SOCKETS)	121	#endif·//·defined(ASIO_HAS_LOCAL_SOCKETS)