blob: cb7fd89f21744cb19763bef863e4f77b50607e2e [file] [log] [blame]
Alexander Afanasyev996b4872012-07-17 17:07:56 -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 *
18 * Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
19 * Ilya Moiseenko <iliamo@cs.ucla.edu>
20 */
21
22#include "green-yellow-red.h"
23
Alexander Afanasyev4aac5572012-08-09 10:49:55 -070024#include "ns3/ndn-pit.h"
25#include "ns3/ndn-pit-entry.h"
Alexander Afanasyevbd9c18e2012-11-19 15:23:41 -080026#include "ns3/ndn-interest.h"
27#include "ns3/ndn-content-object.h"
Alexander Afanasyev4aac5572012-08-09 10:49:55 -070028#include "ns3/ndn-pit.h"
29#include "ns3/ndn-fib.h"
30#include "ns3/ndn-content-store.h"
Alexander Afanasyev996b4872012-07-17 17:07:56 -070031
32#include "ns3/assert.h"
33#include "ns3/ptr.h"
34#include "ns3/log.h"
35#include "ns3/simulator.h"
36#include "ns3/boolean.h"
37#include "ns3/string.h"
38
39#include <boost/ref.hpp>
40#include <boost/foreach.hpp>
41#include <boost/lambda/lambda.hpp>
42#include <boost/lambda/bind.hpp>
43#include <boost/tuple/tuple.hpp>
44namespace ll = boost::lambda;
45
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070046NS_LOG_COMPONENT_DEFINE ("ndn.fw.GreenYellowRed");
Alexander Afanasyev996b4872012-07-17 17:07:56 -070047
48namespace ns3 {
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070049namespace ndn {
50namespace fw {
Alexander Afanasyev996b4872012-07-17 17:07:56 -070051
52NS_OBJECT_ENSURE_REGISTERED (GreenYellowRed);
53
54TypeId
55GreenYellowRed::GetTypeId (void)
56{
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070057 static TypeId tid = TypeId ("ns3::ndn::fw::GreenYellowRed")
Alexander Afanasyev4aac5572012-08-09 10:49:55 -070058 .SetGroupName ("Ndn")
Alexander Afanasyev786936a2012-07-17 19:48:15 -070059 .SetParent<Nacks> ()
Alexander Afanasyev996b4872012-07-17 17:07:56 -070060
61 ;
62 return tid;
63}
64
65bool
Alexander Afanasyev31cb4692012-08-17 13:08:20 -070066GreenYellowRed::DoPropagateInterest (Ptr<Face> inFace,
Alexander Afanasyevb989b122013-07-10 17:15:46 -070067 Ptr<const Interest> interest,
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070068 Ptr<pit::Entry> pitEntry)
Alexander Afanasyev996b4872012-07-17 17:07:56 -070069{
70 NS_LOG_FUNCTION (this);
71 NS_ASSERT_MSG (m_pit != 0, "PIT should be aggregated with forwarding strategy");
72
73 int propagatedCount = 0;
Alexander Afanasyevb14f0bc2013-03-03 20:30:00 -080074
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070075 BOOST_FOREACH (const fib::FaceMetric &metricFace, pitEntry->GetFibEntry ()->m_faces.get<fib::i_metric> ())
Alexander Afanasyev996b4872012-07-17 17:07:56 -070076 {
Alexander Afanasyev06dba7c2013-02-21 11:36:26 -080077 if (metricFace.GetStatus () == fib::FaceMetric::NDN_FIB_RED ||
78 metricFace.GetStatus () == fib::FaceMetric::NDN_FIB_YELLOW)
Alexander Afanasyev996b4872012-07-17 17:07:56 -070079 break; //propagate only to green faces
80
Alexander Afanasyevb989b122013-07-10 17:15:46 -070081 if (!TrySendOutInterest (inFace, metricFace.GetFace (), interest, pitEntry))
Alexander Afanasyev996b4872012-07-17 17:07:56 -070082 {
Alexander Afanasyev996b4872012-07-17 17:07:56 -070083 continue;
84 }
Alexander Afanasyevb14f0bc2013-03-03 20:30:00 -080085
Alexander Afanasyev996b4872012-07-17 17:07:56 -070086 propagatedCount++;
87 break; // propagate only one interest
88 }
89
90 return propagatedCount > 0;
91}
92
93void
Alexander Afanasyev31cb4692012-08-17 13:08:20 -070094GreenYellowRed::WillSatisfyPendingInterest (Ptr<Face> inFace,
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070095 Ptr<pit::Entry> pitEntry)
Alexander Afanasyev996b4872012-07-17 17:07:56 -070096{
Alexander Afanasyev31cb4692012-08-17 13:08:20 -070097 if (inFace != 0)
Alexander Afanasyev39e0f342012-07-20 15:33:23 -070098 {
99 // Update metric status for the incoming interface in the corresponding FIB entry
Alexander Afanasyev31cb4692012-08-17 13:08:20 -0700100 pitEntry->GetFibEntry ()->UpdateStatus (inFace, fib::FaceMetric::NDN_FIB_GREEN);
Alexander Afanasyev39e0f342012-07-20 15:33:23 -0700101 }
Alexander Afanasyev996b4872012-07-17 17:07:56 -0700102
Alexander Afanasyev31cb4692012-08-17 13:08:20 -0700103 super::WillSatisfyPendingInterest (inFace, pitEntry);
Alexander Afanasyev996b4872012-07-17 17:07:56 -0700104}
105
Alexander Afanasyev786936a2012-07-17 19:48:15 -0700106void
Alexander Afanasyevb14f0bc2013-03-03 20:30:00 -0800107GreenYellowRed::WillEraseTimedOutPendingInterest (Ptr<pit::Entry> pitEntry)
108{
109 NS_LOG_DEBUG ("WillEraseTimedOutPendingInterest for " << pitEntry->GetPrefix ());
Alexander Afanasyeve6dc0ac2013-02-21 14:00:48 -0800110
Alexander Afanasyevb14f0bc2013-03-03 20:30:00 -0800111 for (pit::Entry::out_container::iterator face = pitEntry->GetOutgoing ().begin ();
112 face != pitEntry->GetOutgoing ().end ();
113 face ++)
114 {
Alexander Afanasyev4e489602013-03-04 09:51:43 -0800115 // NS_LOG_DEBUG ("Face: " << face->m_face);
Alexander Afanasyevb14f0bc2013-03-03 20:30:00 -0800116 pitEntry->GetFibEntry ()->UpdateStatus (face->m_face, fib::FaceMetric::NDN_FIB_YELLOW);
117 }
118
119 super::WillEraseTimedOutPendingInterest (pitEntry);
120}
121
122void
Alexander Afanasyev31cb4692012-08-17 13:08:20 -0700123GreenYellowRed::DidReceiveValidNack (Ptr<Face> inFace,
Alexander Afanasyev786936a2012-07-17 19:48:15 -0700124 uint32_t nackCode,
Alexander Afanasyevb989b122013-07-10 17:15:46 -0700125 Ptr<const Interest> nack,
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -0700126 Ptr<pit::Entry> pitEntry)
Alexander Afanasyev786936a2012-07-17 19:48:15 -0700127{
Alexander Afanasyevb989b122013-07-10 17:15:46 -0700128 super::DidReceiveValidNack (inFace, nackCode, nack, pitEntry);
Alexander Afanasyev786936a2012-07-17 19:48:15 -0700129
Alexander Afanasyev31cb4692012-08-17 13:08:20 -0700130 if (inFace != 0 &&
Alexander Afanasyeveae83ee2013-03-15 15:01:10 -0700131 (nackCode == Interest::NACK_CONGESTION ||
132 nackCode == Interest::NACK_GIVEUP_PIT))
Alexander Afanasyev786936a2012-07-17 19:48:15 -0700133 {
Alexander Afanasyev31cb4692012-08-17 13:08:20 -0700134 pitEntry->GetFibEntry ()->UpdateStatus (inFace, fib::FaceMetric::NDN_FIB_YELLOW);
Alexander Afanasyev786936a2012-07-17 19:48:15 -0700135 }
136}
137
138
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -0700139} // namespace fw
140} // namespace ndn
Alexander Afanasyev996b4872012-07-17 17:07:56 -0700141} // namespace ns3