tests/face/tcp.cpp - ndnSIM/NFD - Gitiles

 /* -*- 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(EndToEnd, 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));

   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_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
	/* -- 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(EndToEnd, 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));

	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_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