1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3 * Copyright (c) 2011 Bucknell University
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 * Authors: L. Felipe Perrone (perrone@bucknell.edu)
19 * Tiago G. Rodrigues (tgr002@bucknell.edu)
20 *
21 * Modified by: Mitch Watrous (watrous@u.washington.edu)
22 */
23
24 #include "ns3/ipv4-packet-probe.h"
25 #include "ns3/object.h"
26 #include "ns3/log.h"
27 #include "ns3/names.h"
28 #include "ns3/config.h"
29 #include "ns3/trace-source-accessor.h"
30
31 namespace ns3 {
32
33 NS_LOG_COMPONENT_DEFINE ("Ipv4PacketProbe");
34
35 NS_OBJECT_ENSURE_REGISTERED (Ipv4PacketProbe);
36
37 TypeId
GetTypeId()38 Ipv4PacketProbe::GetTypeId ()
39 {
40 static TypeId tid = TypeId ("ns3::Ipv4PacketProbe")
41 .SetParent<Probe> ()
42 .SetGroupName ("Internet")
43 .AddConstructor<Ipv4PacketProbe> ()
44 .AddTraceSource ( "Output",
45 "The packet plus its IPv4 object and interface "
46 "that serve as the output for this probe",
47 MakeTraceSourceAccessor (&Ipv4PacketProbe::m_output),
48 "ns3::Ipv4L3Protocol::TxRxTracedCallback")
49 .AddTraceSource ( "OutputBytes",
50 "The number of bytes in the packet",
51 MakeTraceSourceAccessor (&Ipv4PacketProbe::m_outputBytes),
52 "ns3::Packet::SizeTracedCallback")
53 ;
54 return tid;
55 }
56
Ipv4PacketProbe()57 Ipv4PacketProbe::Ipv4PacketProbe ()
58 {
59 NS_LOG_FUNCTION (this);
60 m_packet = 0;
61 m_packetSizeOld = 0;
62 m_ipv4 = 0;
63 m_interface = 0;
64 }
65
~Ipv4PacketProbe()66 Ipv4PacketProbe::~Ipv4PacketProbe ()
67 {
68 NS_LOG_FUNCTION (this);
69 }
70
71 void
SetValue(Ptr<const Packet> packet,Ptr<Ipv4> ipv4,uint32_t interface)72 Ipv4PacketProbe::SetValue (Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface)
73 {
74 NS_LOG_FUNCTION (this << packet << ipv4 << interface);
75 m_packet = packet;
76 m_ipv4 = ipv4;
77 m_interface = interface;
78 m_output (packet, ipv4, interface);
79
80 uint32_t packetSizeNew = packet->GetSize ();
81 m_outputBytes (m_packetSizeOld, packetSizeNew);
82 m_packetSizeOld = packetSizeNew;
83 }
84
85 void
SetValueByPath(std::string path,Ptr<const Packet> packet,Ptr<Ipv4> ipv4,uint32_t interface)86 Ipv4PacketProbe::SetValueByPath (std::string path, Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface)
87 {
88 NS_LOG_FUNCTION (path << packet << ipv4 << interface);
89 Ptr<Ipv4PacketProbe> probe = Names::Find<Ipv4PacketProbe> (path);
90 NS_ASSERT_MSG (probe, "Error: Can't find probe for path " << path);
91 probe->SetValue (packet, ipv4, interface);
92 }
93
94 bool
ConnectByObject(std::string traceSource,Ptr<Object> obj)95 Ipv4PacketProbe::ConnectByObject (std::string traceSource, Ptr<Object> obj)
96 {
97 NS_LOG_FUNCTION (this << traceSource << obj);
98 NS_LOG_DEBUG ("Name of probe (if any) in names database: " << Names::FindPath (obj));
99 bool connected = obj->TraceConnectWithoutContext (traceSource, MakeCallback (&ns3::Ipv4PacketProbe::TraceSink, this));
100 return connected;
101 }
102
103 void
ConnectByPath(std::string path)104 Ipv4PacketProbe::ConnectByPath (std::string path)
105 {
106 NS_LOG_FUNCTION (this << path);
107 NS_LOG_DEBUG ("Name of probe to search for in config database: " << path);
108 Config::ConnectWithoutContext (path, MakeCallback (&ns3::Ipv4PacketProbe::TraceSink, this));
109 }
110
111 void
TraceSink(Ptr<const Packet> packet,Ptr<Ipv4> ipv4,uint32_t interface)112 Ipv4PacketProbe::TraceSink (Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface)
113 {
114 NS_LOG_FUNCTION (this << packet << ipv4 << interface);
115 if (IsEnabled ())
116 {
117 m_packet = packet;
118 m_ipv4 = ipv4;
119 m_interface = interface;
120 m_output (packet, ipv4, interface);
121
122 uint32_t packetSizeNew = packet->GetSize ();
123 m_outputBytes (m_packetSizeOld, packetSizeNew);
124 m_packetSizeOld = packetSizeNew;
125 }
126 }
127
128 } // namespace ns3
129