Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 1 | /* -*- 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 | */ |
| 20 | |
Alexander Afanasyev | 0c39537 | 2014-12-20 15:54:02 -0800 | [diff] [blame] | 21 | #include "ndn-app.hpp" |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 22 | #include "ns3/log.h" |
| 23 | #include "ns3/assert.h" |
| 24 | #include "ns3/packet.h" |
| 25 | |
Alexander Afanasyev | 0c39537 | 2014-12-20 15:54:02 -0800 | [diff] [blame] | 26 | #include "ns3/ndn-interest.hpp" |
| 27 | #include "ns3/ndn-data.hpp" |
| 28 | #include "ns3/ndn-l3-protocol.hpp" |
| 29 | #include "ns3/ndn-fib.hpp" |
| 30 | #include "ns3/ndn-app-face.hpp" |
| 31 | #include "ns3/ndn-forwarding-strategy.hpp" |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 32 | |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 33 | NS_LOG_COMPONENT_DEFINE("ndn.App"); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 34 | |
Alexander Afanasyev | 2b4c947 | 2012-08-09 15:00:38 -0700 | [diff] [blame] | 35 | namespace ns3 { |
| 36 | namespace ndn { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 37 | |
| 38 | NS_OBJECT_ENSURE_REGISTERED(App); |
| 39 | |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 40 | TypeId |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 41 | App::GetTypeId(void) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 42 | { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 43 | static TypeId tid = TypeId("ns3::ndn::App") |
| 44 | .SetGroupName("Ndn") |
| 45 | .SetParent<Application>() |
| 46 | .AddConstructor<App>() |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 47 | |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 48 | .AddTraceSource("ReceivedInterests", "ReceivedInterests", |
| 49 | MakeTraceSourceAccessor(&App::m_receivedInterests)) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 50 | |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 51 | .AddTraceSource("ReceivedNacks", "ReceivedNacks", |
| 52 | MakeTraceSourceAccessor(&App::m_receivedNacks)) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 53 | |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 54 | .AddTraceSource("ReceivedDatas", "ReceivedDatas", |
| 55 | MakeTraceSourceAccessor(&App::m_receivedDatas)) |
| 56 | |
| 57 | .AddTraceSource("TransmittedInterests", "TransmittedInterests", |
| 58 | MakeTraceSourceAccessor(&App::m_transmittedInterests)) |
| 59 | |
| 60 | .AddTraceSource("TransmittedDatas", "TransmittedDatas", |
| 61 | MakeTraceSourceAccessor(&App::m_transmittedDatas)); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 62 | return tid; |
| 63 | } |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 64 | |
| 65 | App::App() |
| 66 | : m_active(false) |
| 67 | , m_face(0) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 68 | { |
| 69 | } |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 70 | |
| 71 | App::~App() |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 72 | { |
| 73 | } |
| 74 | |
| 75 | void |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 76 | App::DoDispose(void) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 77 | { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 78 | NS_LOG_FUNCTION_NOARGS(); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 79 | |
| 80 | // Unfortunately, this causes SEGFAULT |
| 81 | // The best reason I see is that apps are freed after ndn stack is removed |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 82 | // StopApplication (); |
| 83 | Application::DoDispose(); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 84 | } |
| 85 | |
Alexander Afanasyev | db64ff1 | 2013-01-18 16:37:31 -0800 | [diff] [blame] | 86 | uint32_t |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 87 | App::GetId() const |
Alexander Afanasyev | db64ff1 | 2013-01-18 16:37:31 -0800 | [diff] [blame] | 88 | { |
| 89 | if (m_face == 0) |
| 90 | return (uint32_t)-1; |
| 91 | else |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 92 | return m_face->GetId(); |
Alexander Afanasyev | db64ff1 | 2013-01-18 16:37:31 -0800 | [diff] [blame] | 93 | } |
| 94 | |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 95 | void |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 96 | App::OnInterest(Ptr<const Interest> interest) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 97 | { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 98 | NS_LOG_FUNCTION(this << interest); |
| 99 | m_receivedInterests(interest, this, m_face); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 100 | } |
| 101 | |
| 102 | void |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 103 | App::OnNack(Ptr<const Interest> interest) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 104 | { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 105 | NS_LOG_FUNCTION(this << interest); |
| 106 | m_receivedNacks(interest, this, m_face); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 107 | } |
| 108 | |
| 109 | void |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 110 | App::OnData(Ptr<const Data> contentObject) |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 111 | { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 112 | NS_LOG_FUNCTION(this << contentObject); |
| 113 | m_receivedDatas(contentObject, this, m_face); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 114 | } |
| 115 | |
| 116 | // Application Methods |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 117 | void |
| 118 | App::StartApplication() // Called at time specified by Start |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 119 | { |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 120 | NS_LOG_FUNCTION_NOARGS(); |
| 121 | |
| 122 | NS_ASSERT(m_active != true); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 123 | m_active = true; |
| 124 | |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 125 | NS_ASSERT_MSG(GetNode()->GetObject<L3Protocol>() != 0, |
| 126 | "Ndn stack should be installed on the node " << GetNode()); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 127 | |
| 128 | // step 1. Create a face |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 129 | m_face = CreateObject<AppFace>(/*Ptr<App> (this)*/ this); |
| 130 | |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 131 | // step 2. Add face to the Ndn stack |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 132 | GetNode()->GetObject<L3Protocol>()->AddFace(m_face); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 133 | |
| 134 | // step 3. Enable face |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 135 | m_face->SetUp(true); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 136 | } |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 137 | |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 138 | void |
| 139 | App::StopApplication() // Called at time specified by Stop |
| 140 | { |
| 141 | NS_LOG_FUNCTION_NOARGS(); |
| 142 | |
| 143 | if (!m_active) |
| 144 | return; // don't assert here, just return |
| 145 | |
| 146 | NS_ASSERT(GetNode()->GetObject<L3Protocol>() != 0); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 147 | |
| 148 | m_active = false; |
| 149 | |
| 150 | // step 1. Disable face |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 151 | m_face->SetUp(false); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 152 | |
| 153 | // step 2. Remove face from Ndn stack |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 154 | GetNode()->GetObject<L3Protocol>()->RemoveFace(m_face); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 155 | |
| 156 | // step 3. Destroy face |
Alexander Afanasyev | be55cf6 | 2014-12-20 17:51:09 -0800 | [diff] [blame] | 157 | if (m_face->GetReferenceCount() != 1) { |
| 158 | NS_LOG_ERROR("Please a bug report on https://github.com/NDN-Routing/ndnSIM/issues"); |
| 159 | NS_LOG_ERROR("At this point, nobody else should have referenced this face, but we have " |
| 160 | << m_face->GetReferenceCount() << " references"); |
| 161 | } |
Alexander Afanasyev | f383047 | 2012-10-09 18:23:21 -0700 | [diff] [blame] | 162 | // NS_ASSERT_MSG (m_face->GetReferenceCount ()==2, |
| 163 | // "At this point, nobody else should have referenced this face, but we have " |
| 164 | // << m_face->GetReferenceCount () << " references"); |
Alexander Afanasyev | 4aac557 | 2012-08-09 10:49:55 -0700 | [diff] [blame] | 165 | m_face = 0; |
| 166 | } |
| 167 | |
Alexander Afanasyev | 2b4c947 | 2012-08-09 15:00:38 -0700 | [diff] [blame] | 168 | } // namespace ndn |
| 169 | } // namespace ns3 |