blob: b98107e752d494228ca7e3979a0d21264c78f1b3 [file] [log] [blame]
Junxiao Shi96dc0c42014-01-30 23:51:59 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
3 * Copyright (C) 2014 Named Data Networking Project
4 * See COPYING for copyright and distribution information.
5 */
6
Alexander Afanasyev0eb70652014-02-27 18:35:07 -08007#include "face/tcp-factory.hpp"
Junxiao Shi96dc0c42014-01-30 23:51:59 -07008#include <ndn-cpp-dev/security/key-chain.hpp>
9
Junxiao Shid9ee45c2014-02-27 15:38:11 -070010#include "tests/test-common.hpp"
Junxiao Shi7e2413b2014-03-02 11:15:09 -070011#include "tests/core/limited-io.hpp"
Junxiao Shi96dc0c42014-01-30 23:51:59 -070012
13namespace nfd {
Junxiao Shid9ee45c2014-02-27 15:38:11 -070014namespace tests {
Junxiao Shi96dc0c42014-01-30 23:51:59 -070015
Junxiao Shid9ee45c2014-02-27 15:38:11 -070016BOOST_FIXTURE_TEST_SUITE(FaceTcp, BaseFixture)
Junxiao Shi96dc0c42014-01-30 23:51:59 -070017
18BOOST_AUTO_TEST_CASE(ChannelMap)
19{
Alexander Afanasyev0eb70652014-02-27 18:35:07 -080020 TcpFactory factory;
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -080021
Alexander Afanasyevd6655302014-02-28 08:41:28 -080022 shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070");
23 shared_ptr<TcpChannel> channel1a = factory.createChannel("127.0.0.1", "20070");
Junxiao Shi96dc0c42014-01-30 23:51:59 -070024 BOOST_CHECK_EQUAL(channel1, channel1a);
Junxiao Shi61e3cc52014-03-03 20:40:28 -070025 BOOST_CHECK_EQUAL(channel1->getUri().toString(), "tcp4://127.0.0.1:20070");
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -080026
Alexander Afanasyevd6655302014-02-28 08:41:28 -080027 shared_ptr<TcpChannel> channel2 = factory.createChannel("127.0.0.1", "20071");
Junxiao Shi96dc0c42014-01-30 23:51:59 -070028 BOOST_CHECK_NE(channel1, channel2);
Junxiao Shi61e3cc52014-03-03 20:40:28 -070029
30 shared_ptr<TcpChannel> channel3 = factory.createChannel("::1", "20071");
31 BOOST_CHECK_NE(channel2, channel3);
32 BOOST_CHECK_EQUAL(channel3->getUri().toString(), "tcp6://[::1]:20071");
Junxiao Shi96dc0c42014-01-30 23:51:59 -070033}
34
Junxiao Shid9ee45c2014-02-27 15:38:11 -070035class EndToEndFixture : protected BaseFixture
Junxiao Shi96dc0c42014-01-30 23:51:59 -070036{
37public:
38 void
Alexander Afanasyevbd220a02014-02-20 00:29:56 -080039 channel1_onFaceCreated(const shared_ptr<Face>& newFace)
Junxiao Shi96dc0c42014-01-30 23:51:59 -070040 {
Junxiao Shi79494162014-04-02 18:25:11 -070041 BOOST_CHECK(!static_cast<bool>(face1));
42 face1 = newFace;
43 face1->onReceiveInterest +=
Junxiao Shi96dc0c42014-01-30 23:51:59 -070044 bind(&EndToEndFixture::face1_onReceiveInterest, this, _1);
Junxiao Shi79494162014-04-02 18:25:11 -070045 face1->onReceiveData +=
Junxiao Shi96dc0c42014-01-30 23:51:59 -070046 bind(&EndToEndFixture::face1_onReceiveData, this, _1);
Junxiao Shi79494162014-04-02 18:25:11 -070047 face1->onFail +=
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -080048 bind(&EndToEndFixture::face1_onFail, this);
Junxiao Shi96dc0c42014-01-30 23:51:59 -070049
Junxiao Shi79494162014-04-02 18:25:11 -070050 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -070051 }
52
53 void
54 channel1_onConnectFailed(const std::string& reason)
55 {
56 BOOST_CHECK_MESSAGE(false, reason);
57
Junxiao Shi79494162014-04-02 18:25:11 -070058 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -070059 }
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -080060
Junxiao Shi96dc0c42014-01-30 23:51:59 -070061 void
62 face1_onReceiveInterest(const Interest& interest)
63 {
Junxiao Shi79494162014-04-02 18:25:11 -070064 face1_receivedInterests.push_back(interest);
Junxiao Shi96dc0c42014-01-30 23:51:59 -070065
Junxiao Shi79494162014-04-02 18:25:11 -070066 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -070067 }
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -080068
Junxiao Shi96dc0c42014-01-30 23:51:59 -070069 void
70 face1_onReceiveData(const Data& data)
71 {
Junxiao Shi79494162014-04-02 18:25:11 -070072 face1_receivedDatas.push_back(data);
Junxiao Shi96dc0c42014-01-30 23:51:59 -070073
Junxiao Shi79494162014-04-02 18:25:11 -070074 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -070075 }
76
77 void
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -080078 face1_onFail()
79 {
Junxiao Shi79494162014-04-02 18:25:11 -070080 face1.reset();
81 limitedIo.afterOp();
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -080082 }
83
84 void
Alexander Afanasyevbd220a02014-02-20 00:29:56 -080085 channel2_onFaceCreated(const shared_ptr<Face>& newFace)
Junxiao Shi96dc0c42014-01-30 23:51:59 -070086 {
Junxiao Shi79494162014-04-02 18:25:11 -070087 BOOST_CHECK(!static_cast<bool>(face2));
88 face2 = newFace;
89 face2->onReceiveInterest +=
Junxiao Shi96dc0c42014-01-30 23:51:59 -070090 bind(&EndToEndFixture::face2_onReceiveInterest, this, _1);
Junxiao Shi79494162014-04-02 18:25:11 -070091 face2->onReceiveData +=
Junxiao Shi96dc0c42014-01-30 23:51:59 -070092 bind(&EndToEndFixture::face2_onReceiveData, this, _1);
Junxiao Shi79494162014-04-02 18:25:11 -070093 face2->onFail +=
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -080094 bind(&EndToEndFixture::face2_onFail, this);
Junxiao Shi96dc0c42014-01-30 23:51:59 -070095
Junxiao Shi79494162014-04-02 18:25:11 -070096 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -070097 }
98
99 void
100 channel2_onConnectFailed(const std::string& reason)
101 {
102 BOOST_CHECK_MESSAGE(false, reason);
103
Junxiao Shi79494162014-04-02 18:25:11 -0700104 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700105 }
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800106
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700107 void
108 face2_onReceiveInterest(const Interest& interest)
109 {
Junxiao Shi79494162014-04-02 18:25:11 -0700110 face2_receivedInterests.push_back(interest);
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700111
Junxiao Shi79494162014-04-02 18:25:11 -0700112 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700113 }
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800114
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700115 void
116 face2_onReceiveData(const Data& data)
117 {
Junxiao Shi79494162014-04-02 18:25:11 -0700118 face2_receivedDatas.push_back(data);
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700119
Junxiao Shi79494162014-04-02 18:25:11 -0700120 limitedIo.afterOp();
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700121 }
122
123 void
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800124 face2_onFail()
125 {
Junxiao Shi79494162014-04-02 18:25:11 -0700126 face2.reset();
127 limitedIo.afterOp();
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800128 }
129
130 void
Alexander Afanasyevbd220a02014-02-20 00:29:56 -0800131 channel_onFaceCreated(const shared_ptr<Face>& newFace)
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800132 {
Junxiao Shi79494162014-04-02 18:25:11 -0700133 faces.push_back(newFace);
134 limitedIo.afterOp();
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800135 }
136
137 void
138 channel_onConnectFailed(const std::string& reason)
139 {
140 BOOST_CHECK_MESSAGE(false, reason);
141
Junxiao Shi79494162014-04-02 18:25:11 -0700142 limitedIo.afterOp();
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800143 }
144
145 void
146 checkFaceList(size_t shouldBe)
147 {
Junxiao Shi79494162014-04-02 18:25:11 -0700148 BOOST_CHECK_EQUAL(faces.size(), shouldBe);
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800149 }
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800150
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700151public:
Junxiao Shi79494162014-04-02 18:25:11 -0700152 LimitedIo limitedIo;
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700153
Junxiao Shi79494162014-04-02 18:25:11 -0700154 shared_ptr<Face> face1;
155 std::vector<Interest> face1_receivedInterests;
156 std::vector<Data> face1_receivedDatas;
157 shared_ptr<Face> face2;
158 std::vector<Interest> face2_receivedInterests;
159 std::vector<Data> face2_receivedDatas;
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800160
Junxiao Shi79494162014-04-02 18:25:11 -0700161 std::list< shared_ptr<Face> > faces;
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700162};
163
164
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000165BOOST_FIXTURE_TEST_CASE(EndToEnd4, EndToEndFixture)
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700166{
Alexander Afanasyev0eb70652014-02-27 18:35:07 -0800167 TcpFactory factory;
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700168
Alexander Afanasyevd6655302014-02-28 08:41:28 -0800169 shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070");
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000170 factory.createChannel("127.0.0.1", "20071");
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800171
Alexander Afanasyev53a6fd32014-03-23 00:00:04 -0700172 BOOST_CHECK_EQUAL(channel1->isListening(), false);
173
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700174 channel1->listen(bind(&EndToEndFixture::channel1_onFaceCreated, this, _1),
175 bind(&EndToEndFixture::channel1_onConnectFailed, this, _1));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800176
Alexander Afanasyev53a6fd32014-03-23 00:00:04 -0700177 BOOST_CHECK_EQUAL(channel1->isListening(), true);
178
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000179 factory.createFace(FaceUri("tcp://127.0.0.1:20070"),
180 bind(&EndToEndFixture::channel2_onFaceCreated, this, _1),
181 bind(&EndToEndFixture::channel2_onConnectFailed, this, _1));
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700182
Junxiao Shi79494162014-04-02 18:25:11 -0700183 BOOST_CHECK_MESSAGE(limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700184 "TcpChannel error: cannot connect or cannot accept connection");
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700185
Junxiao Shi79494162014-04-02 18:25:11 -0700186 BOOST_REQUIRE(static_cast<bool>(face1));
187 BOOST_REQUIRE(static_cast<bool>(face2));
Alexander Afanasyevbd220a02014-02-20 00:29:56 -0800188
Junxiao Shi79494162014-04-02 18:25:11 -0700189 BOOST_CHECK(face1->isOnDemand());
190 BOOST_CHECK(!face2->isOnDemand());
Alexander Afanasyev355c0662014-03-20 18:08:17 -0700191
Junxiao Shi79494162014-04-02 18:25:11 -0700192 BOOST_CHECK_EQUAL(face2->getRemoteUri().toString(), "tcp4://127.0.0.1:20070");
193 BOOST_CHECK_EQUAL(face1->getLocalUri().toString(), "tcp4://127.0.0.1:20070");
194 // face1 has an unknown remoteUri, since the source port is automatically chosen by OS
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000195
Junxiao Shi79494162014-04-02 18:25:11 -0700196 BOOST_CHECK_EQUAL(face1->isLocal(), true);
197 BOOST_CHECK_EQUAL(face2->isLocal(), true);
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000198
Junxiao Shi79494162014-04-02 18:25:11 -0700199 BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(face1)), true);
200 BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(face2)), true);
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000201
202 // integrated tests needs to check that TcpFace for non-loopback fails these tests...
203
204 Interest interest1("ndn:/TpnzGvW9R");
205 Data data1 ("ndn:/KfczhUqVix");
206 data1.setContent(0, 0);
207 Interest interest2("ndn:/QWiIMfj5sL");
208 Data data2 ("ndn:/XNBV796f");
209 data2.setContent(0, 0);
210
211 ndn::SignatureSha256WithRsa fakeSignature;
212 fakeSignature.setValue(ndn::dataBlock(tlv::SignatureValue, reinterpret_cast<const uint8_t*>(0), 0));
213
214 // set fake signature on data1 and data2
215 data1.setSignature(fakeSignature);
216 data2.setSignature(fakeSignature);
217
Junxiao Shi79494162014-04-02 18:25:11 -0700218 face1->sendInterest(interest1);
219 face1->sendInterest(interest1);
220 face1->sendInterest(interest1);
221 face1->sendData (data1 );
222 face2->sendInterest(interest2);
223 face2->sendData (data2 );
224 face2->sendData (data2 );
225 face2->sendData (data2 );
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000226
Junxiao Shi79494162014-04-02 18:25:11 -0700227 BOOST_CHECK_MESSAGE(limitedIo.run(8, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000228 "TcpChannel error: cannot send or receive Interest/Data packets");
229
230
Junxiao Shi79494162014-04-02 18:25:11 -0700231 BOOST_REQUIRE_EQUAL(face1_receivedInterests.size(), 1);
232 BOOST_REQUIRE_EQUAL(face1_receivedDatas .size(), 3);
233 BOOST_REQUIRE_EQUAL(face2_receivedInterests.size(), 3);
234 BOOST_REQUIRE_EQUAL(face2_receivedDatas .size(), 1);
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000235
Junxiao Shi79494162014-04-02 18:25:11 -0700236 BOOST_CHECK_EQUAL(face1_receivedInterests[0].getName(), interest2.getName());
237 BOOST_CHECK_EQUAL(face1_receivedDatas [0].getName(), data2.getName());
238 BOOST_CHECK_EQUAL(face2_receivedInterests[0].getName(), interest1.getName());
239 BOOST_CHECK_EQUAL(face2_receivedDatas [0].getName(), data1.getName());
Alexander Afanasyev7e698e62014-03-07 16:48:35 +0000240
Junxiao Shi79494162014-04-02 18:25:11 -0700241 const FaceCounters& counters1 = face1->getCounters();
Alexander Afanasyev7e698e62014-03-07 16:48:35 +0000242 BOOST_CHECK_EQUAL(counters1.getInInterest() , 1);
243 BOOST_CHECK_EQUAL(counters1.getInData() , 3);
244 BOOST_CHECK_EQUAL(counters1.getOutInterest(), 3);
245 BOOST_CHECK_EQUAL(counters1.getOutData() , 1);
246
Junxiao Shi79494162014-04-02 18:25:11 -0700247 const FaceCounters& counters2 = face2->getCounters();
Alexander Afanasyev7e698e62014-03-07 16:48:35 +0000248 BOOST_CHECK_EQUAL(counters2.getInInterest() , 3);
249 BOOST_CHECK_EQUAL(counters2.getInData() , 1);
250 BOOST_CHECK_EQUAL(counters2.getOutInterest(), 1);
251 BOOST_CHECK_EQUAL(counters2.getOutData() , 3);
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000252}
253
254BOOST_FIXTURE_TEST_CASE(EndToEnd6, EndToEndFixture)
255{
256 TcpFactory factory;
257
258 shared_ptr<TcpChannel> channel1 = factory.createChannel("::1", "20070");
259 shared_ptr<TcpChannel> channel2 = factory.createChannel("::1", "20071");
260
261 channel1->listen(bind(&EndToEndFixture::channel1_onFaceCreated, this, _1),
262 bind(&EndToEndFixture::channel1_onConnectFailed, this, _1));
263
264 factory.createFace(FaceUri("tcp://[::1]:20070"),
265 bind(&EndToEndFixture::channel2_onFaceCreated, this, _1),
266 bind(&EndToEndFixture::channel2_onConnectFailed, this, _1));
267
Junxiao Shi79494162014-04-02 18:25:11 -0700268 BOOST_CHECK_MESSAGE(limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000269 "TcpChannel error: cannot connect or cannot accept connection");
270
Junxiao Shi79494162014-04-02 18:25:11 -0700271 BOOST_REQUIRE(static_cast<bool>(face1));
272 BOOST_REQUIRE(static_cast<bool>(face2));
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000273
Junxiao Shi79494162014-04-02 18:25:11 -0700274 BOOST_CHECK_EQUAL(face2->getRemoteUri().toString(), "tcp6://[::1]:20070");
275 BOOST_CHECK_EQUAL(face1->getLocalUri().toString(), "tcp6://[::1]:20070");
276 // face1 has an unknown remoteUri, since the source port is automatically chosen by OS
Alexander Afanasyeva39b90b2014-03-05 15:31:00 +0000277
Junxiao Shi79494162014-04-02 18:25:11 -0700278 BOOST_CHECK_EQUAL(face1->isLocal(), true);
279 BOOST_CHECK_EQUAL(face2->isLocal(), true);
Alexander Afanasyevbd220a02014-02-20 00:29:56 -0800280
Junxiao Shi79494162014-04-02 18:25:11 -0700281 BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(face1)), true);
282 BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(face2)), true);
Alexander Afanasyevbd220a02014-02-20 00:29:56 -0800283
284 // integrated tests needs to check that TcpFace for non-loopback fails these tests...
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800285
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700286 Interest interest1("ndn:/TpnzGvW9R");
287 Data data1 ("ndn:/KfczhUqVix");
288 data1.setContent(0, 0);
289 Interest interest2("ndn:/QWiIMfj5sL");
290 Data data2 ("ndn:/XNBV796f");
291 data2.setContent(0, 0);
292
293 ndn::SignatureSha256WithRsa fakeSignature;
294 fakeSignature.setValue(ndn::dataBlock(tlv::SignatureValue, reinterpret_cast<const uint8_t*>(0), 0));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800295
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700296 // set fake signature on data1 and data2
297 data1.setSignature(fakeSignature);
298 data2.setSignature(fakeSignature);
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800299
Junxiao Shi79494162014-04-02 18:25:11 -0700300 face1->sendInterest(interest1);
301 face1->sendData (data1 );
302 face2->sendInterest(interest2);
303 face2->sendData (data2 );
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700304
Junxiao Shi79494162014-04-02 18:25:11 -0700305 BOOST_CHECK_MESSAGE(limitedIo.run(4, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700306 "TcpChannel error: cannot send or receive Interest/Data packets");
307
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700308
Junxiao Shi79494162014-04-02 18:25:11 -0700309 BOOST_REQUIRE_EQUAL(face1_receivedInterests.size(), 1);
310 BOOST_REQUIRE_EQUAL(face1_receivedDatas .size(), 1);
311 BOOST_REQUIRE_EQUAL(face2_receivedInterests.size(), 1);
312 BOOST_REQUIRE_EQUAL(face2_receivedDatas .size(), 1);
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800313
Junxiao Shi79494162014-04-02 18:25:11 -0700314 BOOST_CHECK_EQUAL(face1_receivedInterests[0].getName(), interest2.getName());
315 BOOST_CHECK_EQUAL(face1_receivedDatas [0].getName(), data2.getName());
316 BOOST_CHECK_EQUAL(face2_receivedInterests[0].getName(), interest1.getName());
317 BOOST_CHECK_EQUAL(face2_receivedDatas [0].getName(), data1.getName());
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700318}
319
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800320BOOST_FIXTURE_TEST_CASE(MultipleAccepts, EndToEndFixture)
321{
Alexander Afanasyev0eb70652014-02-27 18:35:07 -0800322 TcpFactory factory;
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800323
Alexander Afanasyevd6655302014-02-28 08:41:28 -0800324 shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070");
325 shared_ptr<TcpChannel> channel2 = factory.createChannel("127.0.0.1", "20071");
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800326
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800327 channel1->listen(bind(&EndToEndFixture::channel_onFaceCreated, this, _1),
328 bind(&EndToEndFixture::channel_onConnectFailed, this, _1));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800329
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800330 channel2->connect("127.0.0.1", "20070",
331 bind(&EndToEndFixture::channel_onFaceCreated, this, _1),
332 bind(&EndToEndFixture::channel_onConnectFailed, this, _1),
Alexander Afanasyevc1e2ee02014-02-25 17:02:07 -0800333 time::seconds(4)); // very short timeout
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800334
Junxiao Shi79494162014-04-02 18:25:11 -0700335 BOOST_CHECK_MESSAGE(limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700336 "TcpChannel error: cannot connect or cannot accept connection");
337
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800338
Junxiao Shi79494162014-04-02 18:25:11 -0700339 BOOST_CHECK_EQUAL(faces.size(), 2);
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800340
Alexander Afanasyevd6655302014-02-28 08:41:28 -0800341 shared_ptr<TcpChannel> channel3 = factory.createChannel("127.0.0.1", "20072");
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800342 channel3->connect("127.0.0.1", "20070",
343 bind(&EndToEndFixture::channel_onFaceCreated, this, _1),
344 bind(&EndToEndFixture::channel_onConnectFailed, this, _1),
Alexander Afanasyevc1e2ee02014-02-25 17:02:07 -0800345 time::seconds(4)); // very short timeout
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800346
347
Alexander Afanasyevd6655302014-02-28 08:41:28 -0800348 shared_ptr<TcpChannel> channel4 = factory.createChannel("127.0.0.1", "20073");
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800349
350 BOOST_CHECK_NE(channel3, channel4);
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800351
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700352 scheduler::schedule(time::seconds(1),
353 bind(&TcpChannel::connect, channel4, "127.0.0.1", "20070",
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800354 // does not work without static_cast
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700355 static_cast<TcpChannel::FaceCreatedCallback>(
356 bind(&EndToEndFixture::channel_onFaceCreated, this, _1)),
357 static_cast<TcpChannel::ConnectFailedCallback>(
358 bind(&EndToEndFixture::channel_onConnectFailed, this, _1)),
359 time::seconds(4)));
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800360
Alexander Afanasyeveb3197f2014-03-17 19:28:18 -0700361 scheduler::schedule(time::milliseconds(500),
Alexander Afanasyev7329e022014-02-27 14:47:22 -0800362 bind(&EndToEndFixture::checkFaceList, this, 4));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800363
Junxiao Shi79494162014-04-02 18:25:11 -0700364 BOOST_CHECK_MESSAGE(limitedIo.run(4,// 2 connects and 2 accepts
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700365 time::seconds(10)) == LimitedIo::EXCEED_OPS,
366 "TcpChannel error: cannot connect or cannot accept multiple connections");
Alexander Afanasyev7329e022014-02-27 14:47:22 -0800367
Junxiao Shi79494162014-04-02 18:25:11 -0700368 BOOST_CHECK_EQUAL(faces.size(), 6);
Alexander Afanasyeva16abd22014-01-31 18:12:29 -0800369}
370
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800371
372BOOST_FIXTURE_TEST_CASE(FaceClosing, EndToEndFixture)
373{
Alexander Afanasyev0eb70652014-02-27 18:35:07 -0800374 TcpFactory factory;
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800375
Alexander Afanasyevd6655302014-02-28 08:41:28 -0800376 shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070");
377 shared_ptr<TcpChannel> channel2 = factory.createChannel("127.0.0.1", "20071");
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800378
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800379 channel1->listen(bind(&EndToEndFixture::channel1_onFaceCreated, this, _1),
380 bind(&EndToEndFixture::channel1_onConnectFailed, this, _1));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800381
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800382 channel2->connect("127.0.0.1", "20070",
383 bind(&EndToEndFixture::channel2_onFaceCreated, this, _1),
384 bind(&EndToEndFixture::channel2_onConnectFailed, this, _1),
Alexander Afanasyevc1e2ee02014-02-25 17:02:07 -0800385 time::seconds(4)); // very short timeout
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800386
Junxiao Shi79494162014-04-02 18:25:11 -0700387 BOOST_CHECK_MESSAGE(limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700388 "TcpChannel error: cannot connect or cannot accept connection");
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800389
390 BOOST_CHECK_EQUAL(channel1->size(), 1);
391 BOOST_CHECK_EQUAL(channel2->size(), 1);
392
Junxiao Shi79494162014-04-02 18:25:11 -0700393 BOOST_REQUIRE(static_cast<bool>(face1));
394 BOOST_CHECK(static_cast<bool>(face2));
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800395
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700396 // Face::close must be invoked during io run to be counted as an op
Junxiao Shi79494162014-04-02 18:25:11 -0700397 scheduler::schedule(time::milliseconds(100), bind(&Face::close, face1));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800398
Junxiao Shi79494162014-04-02 18:25:11 -0700399 BOOST_CHECK_MESSAGE(limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS,
Junxiao Shi7e2413b2014-03-02 11:15:09 -0700400 "FaceClosing error: cannot properly close faces");
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800401
402 // both faces should get closed
Junxiao Shi79494162014-04-02 18:25:11 -0700403 BOOST_CHECK(!static_cast<bool>(face1));
404 BOOST_CHECK(!static_cast<bool>(face2));
Alexander Afanasyev5f1ec252014-02-28 10:59:17 -0800405
Alexander Afanasyeva0a10fb2014-02-13 19:56:15 -0800406 BOOST_CHECK_EQUAL(channel1->size(), 0);
407 BOOST_CHECK_EQUAL(channel2->size(), 0);
408}
409
410
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700411BOOST_AUTO_TEST_SUITE_END()
412
Junxiao Shid9ee45c2014-02-27 15:38:11 -0700413} // namespace tests
Junxiao Shi96dc0c42014-01-30 23:51:59 -0700414} // namespace nfd