blob: c3c9fd20a2e6ae8980fc379844840100272424fb [file] [log] [blame]
/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
/*
* Copyright (c) 2011 University of California, Los Angeles
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation;
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
*/
#include "ccnx-pit.h"
#include "ns3/log.h"
#include "ns3/string.h"
#include "ns3/simulator.h"
#include "ccnx-interest-header.h"
#include "ccnx-content-object-header.h"
#include <boost/bind.hpp>
#include <boost/lambda/lambda.hpp>
NS_LOG_COMPONENT_DEFINE ("CcnxPit");
using namespace boost::tuples;
using namespace boost;
namespace ns3 {
NS_OBJECT_ENSURE_REGISTERED (CcnxPit);
using namespace __ccnx_private;
TypeId
CcnxPit::GetTypeId ()
{
static TypeId tid = TypeId ("ns3::CcnxPit")
.SetGroupName ("Ccnx")
.SetParent<Object> ()
.AddConstructor<CcnxPit> ()
.AddAttribute ("CleanupTimeout",
"Timeout defining how frequent RIT should be cleaned up",
StringValue ("1s"),
MakeTimeAccessor (&CcnxPit::GetCleanupTimeout, &CcnxPit::SetCleanupTimeout),
MakeTimeChecker ())
.AddAttribute ("PitEntryPruningTimout",
"Timeout for PIT entry to live after being satisfied. To make sure recently satisfied interest will not be satisfied again",
StringValue ("100ms"),
MakeTimeAccessor (&CcnxPit::m_PitEntryPruningTimout),
MakeTimeChecker ())
.AddAttribute ("PitEntryDefaultLifetime",
"Default lifetime of PIT entry (aka default Interest lifetime)",
StringValue("4s"),
MakeTimeAccessor (&CcnxPit::m_PitEntryDefaultLifetime),
MakeTimeChecker ())
;
return tid;
}
CcnxPit::CcnxPit ()
{
}
CcnxPit::~CcnxPit ()
{
DoDispose ();
}
void
CcnxPit::NotifyNewAggregate ()
{
}
void
CcnxPit::DoDispose ()
{
if (m_cleanupEvent.IsRunning ())
m_cleanupEvent.Cancel ();
clear ();
}
void
CcnxPit::SetCleanupTimeout (const Time &timeout)
{
m_cleanupTimeout = timeout;
if (m_cleanupEvent.IsRunning ())
m_cleanupEvent.Cancel (); // cancel any scheduled cleanup events
// schedule even with new timeout
m_cleanupEvent = Simulator::Schedule (m_cleanupTimeout,
&CcnxPit::CleanExpired, this);
}
Time
CcnxPit::GetCleanupTimeout () const
{
return m_cleanupTimeout;
}
void CcnxPit::CleanExpired ()
{
NS_LOG_LOGIC ("Cleaning PIT. Total: " << size ());
Time now = Simulator::Now ();
// uint32_t count = 0;
while (!empty ())
{
CcnxPit::index<i_timestamp>::type::iterator entry = get<i_timestamp> ().begin ();
if (entry->GetExpireTime () <= now) // is the record stale?
{
get<i_timestamp> ().erase (entry);
// count ++;
}
else
break; // nothing else to do. All later records will not be stale
}
// NS_LOG_LOGIC ("Cleaned " << count << " records. Total: " << size ());
// schedule next even
m_cleanupEvent = Simulator::Schedule (m_cleanupTimeout,
&CcnxPit::CleanExpired, this);
}
void
CcnxPit::SetFib (Ptr<CcnxFib> fib)
{
m_fib = fib;
}
const CcnxPitEntry&
CcnxPit::Lookup (const CcnxContentObjectHeader &header) const
{
// NS_LOG_FUNCTION_NOARGS ();
CcnxPitEntryContainer::type::iterator entry =
get<i_prefix> ().find (header.GetName ());
if (entry == end ())
throw CcnxPitEntryNotFound();
return *entry;
}
boost::tuple<const CcnxPitEntry&, bool, bool>
CcnxPit::Lookup (const CcnxInterestHeader &header)
{
NS_LOG_FUNCTION_NOARGS ();
NS_ASSERT_MSG (m_fib != 0, "FIB should be set");
bool isDuplicate = false;
bool isNew = true;
CcnxPitEntryContainer::type::iterator entry =
get<i_prefix> ().find (header.GetName ());
if (entry == end ())
{
CcnxFibEntryContainer::type::iterator fibEntry = m_fib->LongestPrefixMatch (header);
NS_ASSERT_MSG (fibEntry != m_fib->m_fib.end (),
"There should be at least default route set" << " Prefix = "<<header.GetName() << "NodeID == " << m_fib->GetObject<Node>()->GetId() << "\n" << *m_fib);
entry = insert (end (),
CcnxPitEntry (Create<CcnxNameComponents> (header.GetName ()),
header.GetInterestLifetime ().IsZero ()?m_PitEntryDefaultLifetime
: header.GetInterestLifetime (),
*fibEntry));
// isDuplicate = false; // redundant
// isNew = true; // also redundant
}
else
{
isNew = false;
isDuplicate = entry->IsNonceSeen (header.GetNonce ());
}
if (!isDuplicate)
{
modify (entry,
boost::bind(&CcnxPitEntry::AddSeenNonce, boost::lambda::_1, header.GetNonce ()));
}
return make_tuple (cref(*entry), isNew, isDuplicate);
}
std::ostream& operator<< (std::ostream& os, const CcnxPit &pit)
{
BOOST_FOREACH (const CcnxPitEntry &entry, pit)
{
os << entry;
}
return os;
}
} // namespace ns3