blob: 86ec8ee593ea8cab8bfac2d650ead90eed39177e [file] [log] [blame]
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -07001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
2/*
3 * Copyright (c) 2011 University of California, Los Angeles
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation;
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 *
Alexander Afanasyev36b45772012-07-10 16:57:42 -070018 * Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070019 */
20
Alexander Afanasyev996b4872012-07-17 17:07:56 -070021#include "smart-flooding.h"
Alexander Afanasyeve3d126f2012-07-16 17:07:31 -070022
Alexander Afanasyevbd9c18e2012-11-19 15:23:41 -080023#include "ns3/ndn-interest.h"
Alexander Afanasyev4aac5572012-08-09 10:49:55 -070024#include "ns3/ndn-pit.h"
25#include "ns3/ndn-pit-entry.h"
Alexander Afanasyevf377b332011-12-16 15:32:12 -080026
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070027#include "ns3/assert.h"
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080028#include "ns3/log.h"
Alexander Afanasyeve67a97f2011-11-29 14:28:59 -080029#include "ns3/simulator.h"
Alexander Afanasyevbed75692012-04-06 13:01:25 -070030#include "ns3/boolean.h"
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070031
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080032#include <boost/ref.hpp>
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080033#include <boost/foreach.hpp>
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080034#include <boost/lambda/lambda.hpp>
35#include <boost/lambda/bind.hpp>
36namespace ll = boost::lambda;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070037
Alexander Afanasyev996b4872012-07-17 17:07:56 -070038namespace ns3 {
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070039namespace ndn {
40namespace fw {
Alexander Afanasyevbed75692012-04-06 13:01:25 -070041
Alexander Afanasyev996b4872012-07-17 17:07:56 -070042NS_OBJECT_ENSURE_REGISTERED (SmartFlooding);
Alexander Afanasyev042b4a72012-11-09 17:47:48 -080043
44LogComponent SmartFlooding::g_log = LogComponent (SmartFlooding::GetLogName ().c_str ());
45
46std::string
47SmartFlooding::GetLogName ()
48{
49 return super::GetLogName ()+".SmartFlooding";
50}
51
Alexander Afanasyev996b4872012-07-17 17:07:56 -070052TypeId
53SmartFlooding::GetTypeId (void)
54{
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070055 static TypeId tid = TypeId ("ns3::ndn::fw::SmartFlooding")
Alexander Afanasyev4aac5572012-08-09 10:49:55 -070056 .SetGroupName ("Ndn")
Alexander Afanasyev996b4872012-07-17 17:07:56 -070057 .SetParent <GreenYellowRed> ()
58 .AddConstructor <SmartFlooding> ()
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070059 ;
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080060 return tid;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070061}
Alexander Afanasyev06dba7c2013-02-21 11:36:26 -080062
Alexander Afanasyev996b4872012-07-17 17:07:56 -070063SmartFlooding::SmartFlooding ()
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070064{
65}
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080066
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070067bool
Alexander Afanasyev31cb4692012-08-17 13:08:20 -070068SmartFlooding::DoPropagateInterest (Ptr<Face> inFace,
Alexander Afanasyevb989b122013-07-10 17:15:46 -070069 Ptr<const Interest> interest,
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070070 Ptr<pit::Entry> pitEntry)
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070071{
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080072 NS_LOG_FUNCTION (this);
Alexander Afanasyev0a61c342011-12-06 12:48:55 -080073
Alexander Afanasyev996b4872012-07-17 17:07:56 -070074 // Try to work out with just green faces
Alexander Afanasyevb989b122013-07-10 17:15:46 -070075 bool greenOk = super::DoPropagateInterest (inFace, interest, pitEntry);
Alexander Afanasyev996b4872012-07-17 17:07:56 -070076 if (greenOk)
77 return true;
Alexander Afanasyev0a61c342011-12-06 12:48:55 -080078
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080079 int propagatedCount = 0;
Alexander Afanasyev0a61c342011-12-06 12:48:55 -080080
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070081 BOOST_FOREACH (const fib::FaceMetric &metricFace, pitEntry->GetFibEntry ()->m_faces.get<fib::i_metric> ())
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070082 {
Alexander Afanasyev23d2b542011-12-07 18:54:46 -080083 NS_LOG_DEBUG ("Trying " << boost::cref(metricFace));
Alexander Afanasyev06dba7c2013-02-21 11:36:26 -080084 if (metricFace.GetStatus () == fib::FaceMetric::NDN_FIB_RED) // all non-read faces are in the front of the list
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080085 break;
Alexander Afanasyev06dba7c2013-02-21 11:36:26 -080086
Alexander Afanasyevb989b122013-07-10 17:15:46 -070087 if (!TrySendOutInterest (inFace, metricFace.GetFace (), interest, pitEntry))
Alexander Afanasyeve67a97f2011-11-29 14:28:59 -080088 {
Alexander Afanasyeve67a97f2011-11-29 14:28:59 -080089 continue;
90 }
Alexander Afanasyev06dba7c2013-02-21 11:36:26 -080091
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080092 propagatedCount++;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070093 }
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080094
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080095 NS_LOG_INFO ("Propagated to " << propagatedCount << " faces");
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080096 return propagatedCount > 0;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070097}
Alexander Afanasyev996b4872012-07-17 17:07:56 -070098
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070099} // namespace fw
100} // namespace ndn
Alexander Afanasyev996b4872012-07-17 17:07:56 -0700101} // namespace ns3