| /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| /** |
| * Copyright (C) 2014 Named Data Networking Project |
| * See COPYING for copyright and distribution information. |
| */ |
| |
| #include "face/tcp-factory.hpp" |
| #include <ndn-cpp-dev/security/key-chain.hpp> |
| |
| #include "tests/test-common.hpp" |
| #include "tests/core/limited-io.hpp" |
| |
| namespace nfd { |
| namespace tests { |
| |
| BOOST_FIXTURE_TEST_SUITE(FaceTcp, BaseFixture) |
| |
| BOOST_AUTO_TEST_CASE(ChannelMap) |
| { |
| TcpFactory factory; |
| |
| shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070"); |
| shared_ptr<TcpChannel> channel1a = factory.createChannel("127.0.0.1", "20070"); |
| BOOST_CHECK_EQUAL(channel1, channel1a); |
| BOOST_CHECK_EQUAL(channel1->getUri().toString(), "tcp4://127.0.0.1:20070"); |
| |
| shared_ptr<TcpChannel> channel2 = factory.createChannel("127.0.0.1", "20071"); |
| BOOST_CHECK_NE(channel1, channel2); |
| |
| shared_ptr<TcpChannel> channel3 = factory.createChannel("::1", "20071"); |
| BOOST_CHECK_NE(channel2, channel3); |
| BOOST_CHECK_EQUAL(channel3->getUri().toString(), "tcp6://[::1]:20071"); |
| } |
| |
| class EndToEndFixture : protected BaseFixture |
| { |
| public: |
| void |
| channel1_onFaceCreated(const shared_ptr<Face>& newFace) |
| { |
| BOOST_CHECK(!static_cast<bool>(m_face1)); |
| m_face1 = newFace; |
| m_face1->onReceiveInterest += |
| bind(&EndToEndFixture::face1_onReceiveInterest, this, _1); |
| m_face1->onReceiveData += |
| bind(&EndToEndFixture::face1_onReceiveData, this, _1); |
| m_face1->onFail += |
| bind(&EndToEndFixture::face1_onFail, this); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| channel1_onConnectFailed(const std::string& reason) |
| { |
| BOOST_CHECK_MESSAGE(false, reason); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| face1_onReceiveInterest(const Interest& interest) |
| { |
| m_face1_receivedInterests.push_back(interest); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| face1_onReceiveData(const Data& data) |
| { |
| m_face1_receivedDatas.push_back(data); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| face1_onFail() |
| { |
| m_face1.reset(); |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| channel2_onFaceCreated(const shared_ptr<Face>& newFace) |
| { |
| BOOST_CHECK(!static_cast<bool>(m_face2)); |
| m_face2 = newFace; |
| m_face2->onReceiveInterest += |
| bind(&EndToEndFixture::face2_onReceiveInterest, this, _1); |
| m_face2->onReceiveData += |
| bind(&EndToEndFixture::face2_onReceiveData, this, _1); |
| m_face2->onFail += |
| bind(&EndToEndFixture::face2_onFail, this); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| channel2_onConnectFailed(const std::string& reason) |
| { |
| BOOST_CHECK_MESSAGE(false, reason); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| face2_onReceiveInterest(const Interest& interest) |
| { |
| m_face2_receivedInterests.push_back(interest); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| face2_onReceiveData(const Data& data) |
| { |
| m_face2_receivedDatas.push_back(data); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| face2_onFail() |
| { |
| m_face2.reset(); |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| channel_onFaceCreated(const shared_ptr<Face>& newFace) |
| { |
| m_faces.push_back(newFace); |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| channel_onConnectFailed(const std::string& reason) |
| { |
| BOOST_CHECK_MESSAGE(false, reason); |
| |
| m_limitedIo.afterOp(); |
| } |
| |
| void |
| checkFaceList(size_t shouldBe) |
| { |
| BOOST_CHECK_EQUAL(m_faces.size(), shouldBe); |
| } |
| |
| public: |
| LimitedIo m_limitedIo; |
| |
| shared_ptr<Face> m_face1; |
| std::vector<Interest> m_face1_receivedInterests; |
| std::vector<Data> m_face1_receivedDatas; |
| shared_ptr<Face> m_face2; |
| std::vector<Interest> m_face2_receivedInterests; |
| std::vector<Data> m_face2_receivedDatas; |
| |
| std::list< shared_ptr<Face> > m_faces; |
| }; |
| |
| |
| BOOST_FIXTURE_TEST_CASE(EndToEnd4, EndToEndFixture) |
| { |
| TcpFactory factory; |
| |
| shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070"); |
| factory.createChannel("127.0.0.1", "20071"); |
| |
| channel1->listen(bind(&EndToEndFixture::channel1_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel1_onConnectFailed, this, _1)); |
| |
| factory.createFace(FaceUri("tcp://127.0.0.1:20070"), |
| bind(&EndToEndFixture::channel2_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel2_onConnectFailed, this, _1)); |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot connect or cannot accept connection"); |
| |
| BOOST_REQUIRE(static_cast<bool>(m_face1)); |
| BOOST_REQUIRE(static_cast<bool>(m_face2)); |
| |
| BOOST_CHECK_EQUAL(m_face2->getUri().toString(), "tcp4://127.0.0.1:20070"); |
| // face1 has an unknown URI, since the source port is automatically chosen by OS |
| |
| BOOST_CHECK_EQUAL(m_face1->isLocal(), true); |
| BOOST_CHECK_EQUAL(m_face2->isLocal(), true); |
| |
| BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(m_face1)), true); |
| BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(m_face2)), true); |
| |
| // integrated tests needs to check that TcpFace for non-loopback fails these tests... |
| |
| Interest interest1("ndn:/TpnzGvW9R"); |
| Data data1 ("ndn:/KfczhUqVix"); |
| data1.setContent(0, 0); |
| Interest interest2("ndn:/QWiIMfj5sL"); |
| Data data2 ("ndn:/XNBV796f"); |
| data2.setContent(0, 0); |
| |
| ndn::SignatureSha256WithRsa fakeSignature; |
| fakeSignature.setValue(ndn::dataBlock(tlv::SignatureValue, reinterpret_cast<const uint8_t*>(0), 0)); |
| |
| // set fake signature on data1 and data2 |
| data1.setSignature(fakeSignature); |
| data2.setSignature(fakeSignature); |
| |
| m_face1->sendInterest(interest1); |
| m_face1->sendInterest(interest1); |
| m_face1->sendInterest(interest1); |
| m_face1->sendData (data1 ); |
| m_face2->sendInterest(interest2); |
| m_face2->sendData (data2 ); |
| m_face2->sendData (data2 ); |
| m_face2->sendData (data2 ); |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(8, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot send or receive Interest/Data packets"); |
| |
| |
| BOOST_REQUIRE_EQUAL(m_face1_receivedInterests.size(), 1); |
| BOOST_REQUIRE_EQUAL(m_face1_receivedDatas .size(), 3); |
| BOOST_REQUIRE_EQUAL(m_face2_receivedInterests.size(), 3); |
| BOOST_REQUIRE_EQUAL(m_face2_receivedDatas .size(), 1); |
| |
| BOOST_CHECK_EQUAL(m_face1_receivedInterests[0].getName(), interest2.getName()); |
| BOOST_CHECK_EQUAL(m_face1_receivedDatas [0].getName(), data2.getName()); |
| BOOST_CHECK_EQUAL(m_face2_receivedInterests[0].getName(), interest1.getName()); |
| BOOST_CHECK_EQUAL(m_face2_receivedDatas [0].getName(), data1.getName()); |
| |
| const FaceCounters& counters1 = m_face1->getCounters(); |
| BOOST_CHECK_EQUAL(counters1.getInInterest() , 1); |
| BOOST_CHECK_EQUAL(counters1.getInData() , 3); |
| BOOST_CHECK_EQUAL(counters1.getOutInterest(), 3); |
| BOOST_CHECK_EQUAL(counters1.getOutData() , 1); |
| |
| const FaceCounters& counters2 = m_face2->getCounters(); |
| BOOST_CHECK_EQUAL(counters2.getInInterest() , 3); |
| BOOST_CHECK_EQUAL(counters2.getInData() , 1); |
| BOOST_CHECK_EQUAL(counters2.getOutInterest(), 1); |
| BOOST_CHECK_EQUAL(counters2.getOutData() , 3); |
| } |
| |
| BOOST_FIXTURE_TEST_CASE(EndToEnd6, EndToEndFixture) |
| { |
| TcpFactory factory; |
| |
| shared_ptr<TcpChannel> channel1 = factory.createChannel("::1", "20070"); |
| shared_ptr<TcpChannel> channel2 = factory.createChannel("::1", "20071"); |
| |
| channel1->listen(bind(&EndToEndFixture::channel1_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel1_onConnectFailed, this, _1)); |
| |
| factory.createFace(FaceUri("tcp://[::1]:20070"), |
| bind(&EndToEndFixture::channel2_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel2_onConnectFailed, this, _1)); |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot connect or cannot accept connection"); |
| |
| BOOST_REQUIRE(static_cast<bool>(m_face1)); |
| BOOST_REQUIRE(static_cast<bool>(m_face2)); |
| |
| BOOST_CHECK_EQUAL(m_face2->getUri().toString(), "tcp6://[::1]:20070"); |
| // face1 has an unknown URI, since the source port is automatically chosen by OS |
| |
| BOOST_CHECK_EQUAL(m_face1->isLocal(), true); |
| BOOST_CHECK_EQUAL(m_face2->isLocal(), true); |
| |
| BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(m_face1)), true); |
| BOOST_CHECK_EQUAL(static_cast<bool>(dynamic_pointer_cast<LocalFace>(m_face2)), true); |
| |
| // integrated tests needs to check that TcpFace for non-loopback fails these tests... |
| |
| Interest interest1("ndn:/TpnzGvW9R"); |
| Data data1 ("ndn:/KfczhUqVix"); |
| data1.setContent(0, 0); |
| Interest interest2("ndn:/QWiIMfj5sL"); |
| Data data2 ("ndn:/XNBV796f"); |
| data2.setContent(0, 0); |
| |
| ndn::SignatureSha256WithRsa fakeSignature; |
| fakeSignature.setValue(ndn::dataBlock(tlv::SignatureValue, reinterpret_cast<const uint8_t*>(0), 0)); |
| |
| // set fake signature on data1 and data2 |
| data1.setSignature(fakeSignature); |
| data2.setSignature(fakeSignature); |
| |
| m_face1->sendInterest(interest1); |
| m_face1->sendData (data1 ); |
| m_face2->sendInterest(interest2); |
| m_face2->sendData (data2 ); |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(4, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot send or receive Interest/Data packets"); |
| |
| |
| BOOST_REQUIRE_EQUAL(m_face1_receivedInterests.size(), 1); |
| BOOST_REQUIRE_EQUAL(m_face1_receivedDatas .size(), 1); |
| BOOST_REQUIRE_EQUAL(m_face2_receivedInterests.size(), 1); |
| BOOST_REQUIRE_EQUAL(m_face2_receivedDatas .size(), 1); |
| |
| BOOST_CHECK_EQUAL(m_face1_receivedInterests[0].getName(), interest2.getName()); |
| BOOST_CHECK_EQUAL(m_face1_receivedDatas [0].getName(), data2.getName()); |
| BOOST_CHECK_EQUAL(m_face2_receivedInterests[0].getName(), interest1.getName()); |
| BOOST_CHECK_EQUAL(m_face2_receivedDatas [0].getName(), data1.getName()); |
| } |
| |
| BOOST_FIXTURE_TEST_CASE(MultipleAccepts, EndToEndFixture) |
| { |
| TcpFactory factory; |
| |
| shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070"); |
| shared_ptr<TcpChannel> channel2 = factory.createChannel("127.0.0.1", "20071"); |
| |
| channel1->listen(bind(&EndToEndFixture::channel_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel_onConnectFailed, this, _1)); |
| |
| channel2->connect("127.0.0.1", "20070", |
| bind(&EndToEndFixture::channel_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel_onConnectFailed, this, _1), |
| time::seconds(4)); // very short timeout |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot connect or cannot accept connection"); |
| |
| |
| BOOST_CHECK_EQUAL(m_faces.size(), 2); |
| |
| shared_ptr<TcpChannel> channel3 = factory.createChannel("127.0.0.1", "20072"); |
| channel3->connect("127.0.0.1", "20070", |
| bind(&EndToEndFixture::channel_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel_onConnectFailed, this, _1), |
| time::seconds(4)); // very short timeout |
| |
| |
| shared_ptr<TcpChannel> channel4 = factory.createChannel("127.0.0.1", "20073"); |
| |
| BOOST_CHECK_NE(channel3, channel4); |
| |
| scheduler::schedule(time::seconds(1), |
| bind(&TcpChannel::connect, channel4, "127.0.0.1", "20070", |
| // does not work without static_cast |
| static_cast<TcpChannel::FaceCreatedCallback>( |
| bind(&EndToEndFixture::channel_onFaceCreated, this, _1)), |
| static_cast<TcpChannel::ConnectFailedCallback>( |
| bind(&EndToEndFixture::channel_onConnectFailed, this, _1)), |
| time::seconds(4))); |
| |
| scheduler::schedule(time::seconds(0.5), |
| bind(&EndToEndFixture::checkFaceList, this, 4)); |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(4,// 2 connects and 2 accepts |
| time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot connect or cannot accept multiple connections"); |
| |
| BOOST_CHECK_EQUAL(m_faces.size(), 6); |
| } |
| |
| |
| BOOST_FIXTURE_TEST_CASE(FaceClosing, EndToEndFixture) |
| { |
| TcpFactory factory; |
| |
| shared_ptr<TcpChannel> channel1 = factory.createChannel("127.0.0.1", "20070"); |
| shared_ptr<TcpChannel> channel2 = factory.createChannel("127.0.0.1", "20071"); |
| |
| channel1->listen(bind(&EndToEndFixture::channel1_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel1_onConnectFailed, this, _1)); |
| |
| channel2->connect("127.0.0.1", "20070", |
| bind(&EndToEndFixture::channel2_onFaceCreated, this, _1), |
| bind(&EndToEndFixture::channel2_onConnectFailed, this, _1), |
| time::seconds(4)); // very short timeout |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "TcpChannel error: cannot connect or cannot accept connection"); |
| |
| BOOST_CHECK_EQUAL(channel1->size(), 1); |
| BOOST_CHECK_EQUAL(channel2->size(), 1); |
| |
| BOOST_REQUIRE(static_cast<bool>(m_face1)); |
| BOOST_CHECK(static_cast<bool>(m_face2)); |
| |
| // Face::close must be invoked during io run to be counted as an op |
| scheduler::schedule(time::seconds(0.1), bind(&Face::close, m_face1)); |
| |
| BOOST_CHECK_MESSAGE(m_limitedIo.run(2, time::seconds(10)) == LimitedIo::EXCEED_OPS, |
| "FaceClosing error: cannot properly close faces"); |
| |
| // both faces should get closed |
| BOOST_CHECK(!static_cast<bool>(m_face1)); |
| BOOST_CHECK(!static_cast<bool>(m_face2)); |
| |
| BOOST_CHECK_EQUAL(channel1->size(), 0); |
| BOOST_CHECK_EQUAL(channel2->size(), 0); |
| } |
| |
| |
| BOOST_AUTO_TEST_SUITE_END() |
| |
| } // namespace tests |
| } // namespace nfd |