blob: 53272e9df3947deee04432aa641d64342c3b29c9 [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 Afanasyev4aac5572012-08-09 10:49:55 -070023#include "ns3/ndn-interest-header.h"
24#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 Afanasyev2b4c9472012-08-09 15:00:38 -070038NS_LOG_COMPONENT_DEFINE ("ndn.fw.SmartFlooding");
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070039
Alexander Afanasyev996b4872012-07-17 17:07:56 -070040namespace ns3 {
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070041namespace ndn {
42namespace fw {
Alexander Afanasyevbed75692012-04-06 13:01:25 -070043
Alexander Afanasyev996b4872012-07-17 17:07:56 -070044NS_OBJECT_ENSURE_REGISTERED (SmartFlooding);
45
46TypeId
47SmartFlooding::GetTypeId (void)
48{
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070049 static TypeId tid = TypeId ("ns3::ndn::fw::SmartFlooding")
Alexander Afanasyev4aac5572012-08-09 10:49:55 -070050 .SetGroupName ("Ndn")
Alexander Afanasyev996b4872012-07-17 17:07:56 -070051 .SetParent <GreenYellowRed> ()
52 .AddConstructor <SmartFlooding> ()
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070053 ;
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080054 return tid;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070055}
56
Alexander Afanasyev996b4872012-07-17 17:07:56 -070057SmartFlooding::SmartFlooding ()
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070058{
59}
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080060
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070061bool
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070062SmartFlooding::DoPropagateInterest (const Ptr<Face> &incomingFace,
63 Ptr<InterestHeader> header,
Alexander Afanasyev996b4872012-07-17 17:07:56 -070064 const Ptr<const Packet> &packet,
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070065 Ptr<pit::Entry> pitEntry)
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070066{
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080067 NS_LOG_FUNCTION (this);
Alexander Afanasyev0a61c342011-12-06 12:48:55 -080068
Alexander Afanasyev996b4872012-07-17 17:07:56 -070069 // Try to work out with just green faces
70 bool greenOk = super::DoPropagateInterest (incomingFace, header, packet, pitEntry);
71 if (greenOk)
72 return true;
Alexander Afanasyev0a61c342011-12-06 12:48:55 -080073
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080074 int propagatedCount = 0;
Alexander Afanasyev0a61c342011-12-06 12:48:55 -080075
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070076 BOOST_FOREACH (const fib::FaceMetric &metricFace, pitEntry->GetFibEntry ()->m_faces.get<fib::i_metric> ())
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -070077 {
Alexander Afanasyev23d2b542011-12-07 18:54:46 -080078 NS_LOG_DEBUG ("Trying " << boost::cref(metricFace));
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070079 if (metricFace.m_status == fib::FaceMetric::NDN_FIB_RED) // all non-read faces are in the front of the list
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080080 break;
81
Alexander Afanasyev5a595072011-11-25 14:49:07 -080082 if (metricFace.m_face == incomingFace)
Alexander Afanasyev23d2b542011-12-07 18:54:46 -080083 {
84 NS_LOG_DEBUG ("continue (same as incoming)");
85 continue; // same face as incoming, don't forward
86 }
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080087
Alexander Afanasyevf249a192012-07-18 16:52:51 -070088 if (!WillSendOutInterest (metricFace.m_face, header, pitEntry))
Alexander Afanasyeve67a97f2011-11-29 14:28:59 -080089 {
Alexander Afanasyeve67a97f2011-11-29 14:28:59 -080090 continue;
91 }
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080092
Ilya Moiseenko1a8be032012-01-18 12:51:09 -080093 //transmission
Alexander Afanasyevf249a192012-07-18 16:52:51 -070094 Ptr<Packet> packetToSend = packet->Copy ();
Alexander Afanasyeve9c9d722012-01-19 16:59:30 -080095 metricFace.m_face->Send (packetToSend);
Alexander Afanasyevf249a192012-07-18 16:52:51 -070096
Alexander Afanasyev1c0248b2012-07-24 15:59:50 -070097 DidSendOutInterest (metricFace.m_face, header, packet, pitEntry);
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080098
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -080099 propagatedCount++;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -0700100 }
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -0800101
Alexander Afanasyev19426ef2011-11-23 20:55:28 -0800102 NS_LOG_INFO ("Propagated to " << propagatedCount << " faces");
Alexander Afanasyeva5bbe0e2011-11-22 17:28:39 -0800103 return propagatedCount > 0;
Ilya Moiseenko25f7d4d2011-09-29 18:41:06 -0700104}
Alexander Afanasyev996b4872012-07-17 17:07:56 -0700105
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -0700106} // namespace fw
107} // namespace ndn
Alexander Afanasyev996b4872012-07-17 17:07:56 -0700108} // namespace ns3