blob: e1bd2aac091f27e81c1a0b3274a173f3e29266ab [file] [log] [blame]
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
/*
* Copyright (c) 2013, Regents of the University of California
* Alexander Afanasyev
*
* GNU v3.0 license, See the LICENSE file for more information
*
* Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
*/
#include "ndnSIM-api.h"
#include "ns3/core-module.h"
#include "ns3/ndnSIM-module.h"
#include "ns3/point-to-point-module.h"
#include <boost/lexical_cast.hpp>
NS_LOG_COMPONENT_DEFINE ("ndn.ApiTest");
namespace ns3
{
class ApiTestClient : public Application
{
public:
static TypeId
GetTypeId ()
{
static TypeId tid = TypeId ("ns3::ndn::test::ApiTestClient")
.SetParent<Application> ()
.AddConstructor<ApiTestClient> ()
;
return tid;
}
ApiTestClient ()
: datas (0)
, timeouts (0)
{
}
protected:
void
StartApplication ()
{
m_face = Create<ndn::ApiFace> (GetNode ());
Simulator::Schedule (Seconds (0.1), &ApiTestClient::SendPacket, this, std::string ("/1"));
Simulator::Schedule (Seconds (5.0), &ApiTestClient::SendPacket, this, std::string ("/2"));
}
void
StopApplication ()
{
m_face->Shutdown ();
m_face = 0;
}
private:
void
GotData (Ptr<const ndn::Interest>, Ptr<const ndn::Data>)
{
datas++;
}
void
GotTimeout (Ptr<const ndn::Interest>)
{
timeouts++;
}
void
SendPacket (const std::string &prefix)
{
Ptr<ndn::Interest> interest = Create<ndn::Interest> ();
interest->SetName (Create<ndn::Name> (prefix));
interest->SetInterestLifetime (Seconds (0.5));
m_face->ExpressInterest (interest,
MakeCallback (&ApiTestClient::GotData, this),
MakeCallback (&ApiTestClient::GotTimeout, this));
}
public:
uint32_t datas;
uint32_t timeouts;
private:
Ptr<ndn::ApiFace> m_face;
};
NS_OBJECT_ENSURE_REGISTERED (ApiTestClient);
void
ApiTest::Check0 (Ptr<Application> app)
{
NS_TEST_ASSERT_MSG_EQ (DynamicCast<ApiTestClient> (app)->datas, 0, "");
NS_TEST_ASSERT_MSG_EQ (DynamicCast<ApiTestClient> (app)->timeouts, 0, "");
}
void
ApiTest::Check1 (Ptr<Application> app)
{
NS_TEST_ASSERT_MSG_EQ (DynamicCast<ApiTestClient> (app)->datas, 1, "");
NS_TEST_ASSERT_MSG_EQ (DynamicCast<ApiTestClient> (app)->timeouts, 0, "");
}
void
ApiTest::Check2 (Ptr<Application> app)
{
NS_TEST_ASSERT_MSG_EQ (DynamicCast<ApiTestClient> (app)->datas, 1, "");
NS_TEST_ASSERT_MSG_EQ (DynamicCast<ApiTestClient> (app)->timeouts, 1, "");
}
void
ApiTest::DoRun ()
{
Config::SetDefault ("ns3::PointToPointNetDevice::DataRate", StringValue ("1Mbps"));
Config::SetDefault ("ns3::PointToPointChannel::Delay", StringValue ("10ms"));
Config::SetDefault ("ns3::DropTailQueue::MaxPackets", StringValue ("20"));
// Creating nodes
NodeContainer nodes;
nodes.Create (3);
// Connecting nodes using two links
PointToPointHelper p2p;
p2p.Install (nodes.Get (0), nodes.Get (1));
p2p.Install (nodes.Get (1), nodes.Get (2));
// Install NDN stack on all nodes
ndn::StackHelper ndnHelper;
ndnHelper.SetDefaultRoutes (true);
ndnHelper.InstallAll ();
// Installing applications
// Consumer
ndn::AppHelper consumerHelper ("ns3::ndn::test::ApiTestClient");
ApplicationContainer apps = consumerHelper.Install (nodes.Get (0)); // first node
// Producer
ndn::AppHelper producerHelper ("ns3::ndn::Producer");
// Producer will reply to all requests starting with /prefix
producerHelper.SetPrefix ("/");
producerHelper.SetAttribute ("Postfix", StringValue ("/unique/postfix"));
producerHelper.SetAttribute ("PayloadSize", StringValue("1024"));
producerHelper.Install (nodes.Get (2)).Stop (Seconds (4.0)); // last node
Simulator::Schedule (Seconds (0.0001), &ApiTest::Check0, this, apps.Get (0));
Simulator::Schedule (Seconds (0.2000), &ApiTest::Check1, this, apps.Get (0));
Simulator::Schedule (Seconds (5.6100), &ApiTest::Check2, this, apps.Get (0));
Simulator::Stop (Seconds (20.0));
Simulator::Run ();
Simulator::Destroy ();
}
}