blob: cc5e9b27364abbac47bc63f7aaa805a420b7179d [file] [log] [blame]
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -08001/* -*- 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
21#include "ccnx-app.h"
22#include "ns3/log.h"
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080023#include "ns3/assert.h"
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080024#include "ns3/packet.h"
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080025
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080026#include "ns3/ccnx-interest-header.h"
27#include "ns3/ccnx-content-object-header.h"
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080028#include "ns3/ccnx.h"
29#include "ns3/ccnx-fib.h"
Alexander Afanasyevf9f4eb02011-12-16 01:51:14 -080030#include "../model/ccnx-local-face.h"
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080031
32NS_LOG_COMPONENT_DEFINE ("CcnxApp");
33
34namespace ns3
35{
36
37NS_OBJECT_ENSURE_REGISTERED (CcnxApp);
38
39TypeId
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080040CcnxApp::GetTypeId (void)
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080041{
42 static TypeId tid = TypeId ("ns3::CcnxApp")
43 .SetParent<Application> ()
44 .AddConstructor<CcnxApp> ()
Alexander Afanasyevbdc0d982011-12-16 01:15:26 -080045
46 .AddTraceSource ("ReceivedInterests", "ReceivedInterests",
47 MakeTraceSourceAccessor (&CcnxApp::m_receivedInterests))
48
49 .AddTraceSource ("ReceivedNacks", "ReceivedNacks",
50 MakeTraceSourceAccessor (&CcnxApp::m_receivedNacks))
51
52 .AddTraceSource ("ReceivedContentObjects", "ReceivedContentObjects",
53 MakeTraceSourceAccessor (&CcnxApp::m_receivedContentObjects))
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080054 ;
55 return tid;
56}
57
58CcnxApp::CcnxApp ()
59 : m_protocolHandler (0)
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080060 , m_active (false)
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080061 , m_face (0)
62{
63}
64
65CcnxApp::~CcnxApp ()
66{
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080067}
68
69void
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080070CcnxApp::DoDispose (void)
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080071{
72 NS_LOG_FUNCTION_NOARGS ();
73
Alexander Afanasyev4a5c2c12011-12-12 18:50:57 -080074 // Unfortunately, this causes SEGFAULT
75 // The best reason I see is that apps are freed after ccnx stack is removed
76 // StopApplication ();
Alexander Afanasyev19426ef2011-11-23 20:55:28 -080077 Application::DoDispose ();
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080078}
79
80void
81CcnxApp::RegisterProtocolHandler (ProtocolHandler handler)
82{
83 m_protocolHandler = handler;
84}
85
86void
87CcnxApp::OnInterest (const Ptr<const CcnxInterestHeader> &interest)
88{
89 NS_LOG_FUNCTION (this << interest);
Alexander Afanasyevbdc0d982011-12-16 01:15:26 -080090 m_receivedInterests (interest, this, m_face);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080091}
92
93void
94CcnxApp::OnNack (const Ptr<const CcnxInterestHeader> &interest)
95{
96 NS_LOG_FUNCTION (this << interest);
Alexander Afanasyevbdc0d982011-12-16 01:15:26 -080097 m_receivedNacks (interest, this, m_face);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -080098}
99
100void
101CcnxApp::OnContentObject (const Ptr<const CcnxContentObjectHeader> &contentObject,
102 const Ptr<const Packet> &payload)
103{
104 NS_LOG_FUNCTION (this << contentObject << payload);
Alexander Afanasyevbdc0d982011-12-16 01:15:26 -0800105 m_receivedContentObjects (contentObject, payload, this, m_face);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800106}
107
108// Application Methods
109void
110CcnxApp::StartApplication () // Called at time specified by Start
111{
112 NS_LOG_FUNCTION_NOARGS ();
113
114 NS_ASSERT (m_active != true);
115 m_active = true;
116
Alexander Afanasyev19426ef2011-11-23 20:55:28 -0800117 NS_ASSERT_MSG (GetNode ()->GetObject<Ccnx> () != 0,
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800118 "Ccnx stack should be installed on the node " << GetNode ());
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800119
120 // step 1. Create a face
Alexander Afanasyevcbe92ae2011-12-16 13:06:18 -0800121 m_face = CreateObject<CcnxLocalFace> (/*Ptr<CcnxApp> (this)*/this);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800122
123 // step 2. Add face to the CCNx stack
Alexander Afanasyev19426ef2011-11-23 20:55:28 -0800124 GetNode ()->GetObject<Ccnx> ()->AddFace (m_face);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800125
126 // step 3. Enable face
127 m_face->SetUp (true);
128}
129
130void
131CcnxApp::StopApplication () // Called at time specified by Stop
132{
133 NS_LOG_FUNCTION_NOARGS ();
134
135 if (!m_active) return; //don't assert here, just return
136
Alexander Afanasyev19426ef2011-11-23 20:55:28 -0800137 NS_ASSERT (GetNode ()->GetObject<Ccnx> () != 0);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800138
139 m_active = false;
140
141 // step 1. Disable face
142 m_face->SetUp (false);
143
144 // step 2. Remove face from CCNx stack
Alexander Afanasyev19426ef2011-11-23 20:55:28 -0800145 GetNode ()->GetObject<Ccnx> ()->RemoveFace (m_face);
146 GetNode ()->GetObject<CcnxFib> ()->RemoveFromAll (m_face);
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800147
148 // step 3. Destroy face
149 NS_ASSERT_MSG (m_face->GetReferenceCount ()==1,
Alexander Afanasyev19426ef2011-11-23 20:55:28 -0800150 "At this point, nobody else should have referenced this face, but we have "
151 << m_face->GetReferenceCount () << " references");
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800152 m_face = 0;
153}
154
Alexander Afanasyev09c7deb2011-11-23 14:50:10 -0800155}