blob: cd2744d6ab1fc09d0c464982c1fa6aa6c92fe272 [file] [log] [blame]
Alexander Afanasyeva9034b02014-01-26 18:32:02 -08001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Junxiao Shia1937bf2014-11-06 11:43:40 -07003 * Copyright (c) 2014, Regents of the University of California,
4 * Arizona Board of Regents,
5 * Colorado State University,
6 * University Pierre & Marie Curie, Sorbonne University,
7 * Washington University in St. Louis,
8 * Beijing Institute of Technology,
9 * The University of Memphis
Alexander Afanasyev9bcbc7c2014-04-06 19:37:37 -070010 *
11 * This file is part of NFD (Named Data Networking Forwarding Daemon).
12 * See AUTHORS.md for complete list of NFD authors and contributors.
13 *
14 * NFD is free software: you can redistribute it and/or modify it under the terms
15 * of the GNU General Public License as published by the Free Software Foundation,
16 * either version 3 of the License, or (at your option) any later version.
17 *
18 * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
19 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
20 * PURPOSE. See the GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License along with
23 * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
Junxiao Shia1937bf2014-11-06 11:43:40 -070024 */
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080025
26#include "tcp-channel.hpp"
Junxiao Shi61e3cc52014-03-03 20:40:28 -070027#include "core/global-io.hpp"
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080028
Alexander Afanasyev18bbf812014-01-29 01:40:23 -080029namespace nfd {
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080030
Alexander Afanasyev3958b012014-01-31 15:06:13 -080031NFD_LOG_INIT("TcpChannel");
32
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080033using namespace boost::asio;
34
Junxiao Shi61e3cc52014-03-03 20:40:28 -070035TcpChannel::TcpChannel(const tcp::Endpoint& localEndpoint)
36 : m_localEndpoint(localEndpoint)
Alexander Afanasyev53a6fd32014-03-23 00:00:04 -070037 , m_isListening(false)
Junxiao Shi61e3cc52014-03-03 20:40:28 -070038{
39 this->setUri(FaceUri(localEndpoint));
40}
41
42TcpChannel::~TcpChannel()
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080043{
44}
45
46void
Alexander Afanasyev855a53c2014-01-27 12:03:00 -080047TcpChannel::listen(const FaceCreatedCallback& onFaceCreated,
48 const ConnectFailedCallback& onAcceptFailed,
49 int backlog/* = tcp::acceptor::max_connections*/)
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080050{
Alexander Afanasyevf6980282014-05-13 18:28:40 -070051 m_acceptor = make_shared<ip::tcp::acceptor>(ref(getGlobalIoService()));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080052 m_acceptor->open(m_localEndpoint.protocol());
53 m_acceptor->set_option(ip::tcp::acceptor::reuse_address(true));
Alexander Afanasyeveee71fd2014-03-13 17:40:33 -070054 if (m_localEndpoint.address().is_v6())
55 {
56 m_acceptor->set_option(ip::v6_only(true));
57 }
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080058 m_acceptor->bind(m_localEndpoint);
59 m_acceptor->listen(backlog);
60
61 shared_ptr<ip::tcp::socket> clientSocket =
Alexander Afanasyevf6980282014-05-13 18:28:40 -070062 make_shared<ip::tcp::socket>(ref(getGlobalIoService()));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080063 m_acceptor->async_accept(*clientSocket,
Alexander Afanasyeva16abd22014-01-31 18:12:29 -080064 bind(&TcpChannel::handleSuccessfulAccept, this, _1,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -080065 clientSocket,
Alexander Afanasyeva16abd22014-01-31 18:12:29 -080066 onFaceCreated, onAcceptFailed));
Alexander Afanasyev53a6fd32014-03-23 00:00:04 -070067
68 m_isListening = true;
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080069}
70
71void
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080072TcpChannel::connect(const tcp::Endpoint& remoteEndpoint,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -080073 const TcpChannel::FaceCreatedCallback& onFaceCreated,
74 const TcpChannel::ConnectFailedCallback& onConnectFailed,
Alexander Afanasyeveb3197f2014-03-17 19:28:18 -070075 const time::seconds& timeout/* = time::seconds(4)*/)
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080076{
Alexander Afanasyev334b7142014-01-28 12:59:39 -080077 ChannelFaceMap::iterator i = m_channelFaces.find(remoteEndpoint);
78 if (i != m_channelFaces.end()) {
79 onFaceCreated(i->second);
80 return;
81 }
82
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080083 shared_ptr<ip::tcp::socket> clientSocket =
Alexander Afanasyevf6980282014-05-13 18:28:40 -070084 make_shared<ip::tcp::socket>(ref(getGlobalIoService()));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080085
Alexander Afanasyev18861802014-06-13 17:49:03 -070086 EventId connectTimeoutEvent = scheduler::schedule(timeout,
87 bind(&TcpChannel::handleFailedConnect, this,
88 clientSocket, onConnectFailed));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080089
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080090 clientSocket->async_connect(remoteEndpoint,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -080091 bind(&TcpChannel::handleSuccessfulConnect, this, _1,
Alexander Afanasyev18861802014-06-13 17:49:03 -070092 clientSocket, connectTimeoutEvent,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -080093 onFaceCreated, onConnectFailed));
Alexander Afanasyeva9034b02014-01-26 18:32:02 -080094}
95
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -080096void
97TcpChannel::connect(const std::string& remoteHost, const std::string& remotePort,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -080098 const TcpChannel::FaceCreatedCallback& onFaceCreated,
99 const TcpChannel::ConnectFailedCallback& onConnectFailed,
Alexander Afanasyeveb3197f2014-03-17 19:28:18 -0700100 const time::seconds& timeout/* = time::seconds(4)*/)
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800101{
102 shared_ptr<ip::tcp::socket> clientSocket =
Alexander Afanasyevf6980282014-05-13 18:28:40 -0700103 make_shared<ip::tcp::socket>(ref(getGlobalIoService()));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800104
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800105 ip::tcp::resolver::query query(remoteHost, remotePort);
106 shared_ptr<ip::tcp::resolver> resolver =
Alexander Afanasyevf6980282014-05-13 18:28:40 -0700107 make_shared<ip::tcp::resolver>(ref(getGlobalIoService()));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800108
Alexander Afanasyev18861802014-06-13 17:49:03 -0700109 EventId connectTimeoutEvent = scheduler::schedule(timeout,
110 bind(&TcpChannel::handleFailedConnect, this,
111 clientSocket, onConnectFailed));
112
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800113 resolver->async_resolve(query,
Davide Pesavento855bf292014-02-27 03:58:01 +0100114 bind(&TcpChannel::handleEndpointResolution, this, _1, _2,
Alexander Afanasyev18861802014-06-13 17:49:03 -0700115 clientSocket, connectTimeoutEvent,
Alexander Afanasyev3958b012014-01-31 15:06:13 -0800116 onFaceCreated, onConnectFailed,
117 resolver));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800118}
119
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800120size_t
121TcpChannel::size() const
122{
123 return m_channelFaces.size();
124}
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800125
126void
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800127TcpChannel::createFace(const shared_ptr<ip::tcp::socket>& socket,
Alexander Afanasyev355c0662014-03-20 18:08:17 -0700128 const FaceCreatedCallback& onFaceCreated,
129 bool isOnDemand)
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800130{
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800131 tcp::Endpoint remoteEndpoint = socket->remote_endpoint();
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800132
Alexander Afanasyevbd220a02014-02-20 00:29:56 -0800133 shared_ptr<Face> face;
Steve DiBenedetto2aa4eca2014-06-20 10:47:40 -0600134
135 ChannelFaceMap::iterator faceMapPos = m_channelFaces.find(remoteEndpoint);
136 if (faceMapPos == m_channelFaces.end())
137 {
138 if (socket->local_endpoint().address().is_loopback())
139 face = make_shared<TcpLocalFace>(socket, isOnDemand);
140 else
141 face = make_shared<TcpFace>(socket, isOnDemand);
142
143 face->onFail += bind(&TcpChannel::afterFaceFailed, this, remoteEndpoint);
144
145 m_channelFaces[remoteEndpoint] = face;
146 }
Alexander Afanasyevbd220a02014-02-20 00:29:56 -0800147 else
Steve DiBenedetto2aa4eca2014-06-20 10:47:40 -0600148 {
149 // we've already created a a face for this endpoint, just reuse it
150 face = faceMapPos->second;
Junxiao Shi61e3cc52014-03-03 20:40:28 -0700151
Steve DiBenedetto2aa4eca2014-06-20 10:47:40 -0600152 boost::system::error_code error;
153 socket->shutdown(ip::tcp::socket::shutdown_both, error);
154 socket->close(error);
155 }
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800156
Steve DiBenedetto2aa4eca2014-06-20 10:47:40 -0600157 // Need to invoke the callback regardless of whether or not we have already created
158 // the face so that control responses and such can be sent.
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800159 onFaceCreated(face);
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800160}
161
162void
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800163TcpChannel::afterFaceFailed(tcp::Endpoint &remoteEndpoint)
164{
165 NFD_LOG_DEBUG("afterFaceFailed: " << remoteEndpoint);
166 m_channelFaces.erase(remoteEndpoint);
167}
168
169void
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800170TcpChannel::handleSuccessfulAccept(const boost::system::error_code& error,
171 const shared_ptr<boost::asio::ip::tcp::socket>& socket,
172 const FaceCreatedCallback& onFaceCreated,
173 const ConnectFailedCallback& onAcceptFailed)
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800174{
175 if (error) {
176 if (error == boost::system::errc::operation_canceled) // when socket is closed by someone
177 return;
178
Alexander Afanasyev3958b012014-01-31 15:06:13 -0800179 NFD_LOG_DEBUG("Connect to remote endpoint failed: "
180 << error.category().message(error.value()));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800181
182 if (static_cast<bool>(onAcceptFailed))
183 onAcceptFailed("Connect to remote endpoint failed: " +
184 error.category().message(error.value()));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800185 return;
186 }
187
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800188 // prepare accepting the next connection
189 shared_ptr<ip::tcp::socket> clientSocket =
Alexander Afanasyevf6980282014-05-13 18:28:40 -0700190 make_shared<ip::tcp::socket>(ref(getGlobalIoService()));
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800191 m_acceptor->async_accept(*clientSocket,
192 bind(&TcpChannel::handleSuccessfulAccept, this, _1,
193 clientSocket,
194 onFaceCreated, onAcceptFailed));
Alexander Afanasyev334b7142014-01-28 12:59:39 -0800195
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800196 NFD_LOG_DEBUG("[" << m_localEndpoint << "] "
197 "<< Connection from " << socket->remote_endpoint());
Alexander Afanasyev355c0662014-03-20 18:08:17 -0700198 createFace(socket, onFaceCreated, true);
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800199}
200
201void
202TcpChannel::handleSuccessfulConnect(const boost::system::error_code& error,
203 const shared_ptr<ip::tcp::socket>& socket,
Alexander Afanasyev18861802014-06-13 17:49:03 -0700204 const EventId& connectTimeoutEvent,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -0800205 const FaceCreatedCallback& onFaceCreated,
206 const ConnectFailedCallback& onConnectFailed)
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800207{
Alexander Afanasyev18861802014-06-13 17:49:03 -0700208 scheduler::cancel(connectTimeoutEvent);
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800209
Alexander Afanasyev7465d5f2014-07-07 10:19:42 -0700210#if (BOOST_VERSION == 105400)
211 // To handle regression in Boost 1.54
212 // https://svn.boost.org/trac/boost/ticket/8795
213 boost::system::error_code anotherErrorCode;
214 socket->remote_endpoint(anotherErrorCode);
215 if (error || anotherErrorCode) {
216#else
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800217 if (error) {
Alexander Afanasyev7465d5f2014-07-07 10:19:42 -0700218#endif
219
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800220 if (error == boost::system::errc::operation_canceled) // when socket is closed by someone
221 return;
222
223 socket->close();
Junxiao Shi61e3cc52014-03-03 20:40:28 -0700224
Alexander Afanasyev3958b012014-01-31 15:06:13 -0800225 NFD_LOG_DEBUG("Connect to remote endpoint failed: "
226 << error.category().message(error.value()));
Junxiao Shi61e3cc52014-03-03 20:40:28 -0700227
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800228 onConnectFailed("Connect to remote endpoint failed: " +
229 error.category().message(error.value()));
230 return;
231 }
232
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800233 NFD_LOG_DEBUG("[" << m_localEndpoint << "] "
234 ">> Connection to " << socket->remote_endpoint());
Junxiao Shi61e3cc52014-03-03 20:40:28 -0700235
Alexander Afanasyev355c0662014-03-20 18:08:17 -0700236 createFace(socket, onFaceCreated, false);
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800237}
238
239void
Alexander Afanasyev18861802014-06-13 17:49:03 -0700240TcpChannel::handleFailedConnect(const shared_ptr<ip::tcp::socket>& socket,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -0800241 const ConnectFailedCallback& onConnectFailed)
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800242{
Alexander Afanasyev18861802014-06-13 17:49:03 -0700243 NFD_LOG_DEBUG("Connect to remote endpoint timed out");
244 onConnectFailed("Connect to remote endpoint timed out");
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800245 socket->close(); // abort the connection
246}
247
248void
Davide Pesavento855bf292014-02-27 03:58:01 +0100249TcpChannel::handleEndpointResolution(const boost::system::error_code& error,
250 ip::tcp::resolver::iterator remoteEndpoint,
251 const shared_ptr<boost::asio::ip::tcp::socket>& socket,
Alexander Afanasyev18861802014-06-13 17:49:03 -0700252 const EventId& connectTimeoutEvent,
Davide Pesavento855bf292014-02-27 03:58:01 +0100253 const FaceCreatedCallback& onFaceCreated,
254 const ConnectFailedCallback& onConnectFailed,
255 const shared_ptr<ip::tcp::resolver>& resolver)
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800256{
257 if (error ||
258 remoteEndpoint == ip::tcp::resolver::iterator())
259 {
260 if (error == boost::system::errc::operation_canceled) // when socket is closed by someone
261 return;
262
263 socket->close();
Alexander Afanasyev18861802014-06-13 17:49:03 -0700264 scheduler::cancel(connectTimeoutEvent);
Alexander Afanasyev3958b012014-01-31 15:06:13 -0800265
266 NFD_LOG_DEBUG("Remote endpoint hostname or port cannot be resolved: "
267 << error.category().message(error.value()));
Junxiao Shi61e3cc52014-03-03 20:40:28 -0700268
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800269 onConnectFailed("Remote endpoint hostname or port cannot be resolved: " +
270 error.category().message(error.value()));
271 return;
272 }
273
274 // got endpoint, now trying to connect (only try the first resolution option)
275 socket->async_connect(*remoteEndpoint,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -0800276 bind(&TcpChannel::handleSuccessfulConnect, this, _1,
Alexander Afanasyev18861802014-06-13 17:49:03 -0700277 socket, connectTimeoutEvent,
Alexander Afanasyev855a53c2014-01-27 12:03:00 -0800278 onFaceCreated, onConnectFailed));
Alexander Afanasyev8ad71ba2014-01-27 00:07:14 -0800279}
280
Alexander Afanasyev18bbf812014-01-29 01:40:23 -0800281} // namespace nfd