/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2011 Bucknell University

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Authors: L. Felipe Perrone (perrone@bucknell.edu)

 *          Tiago G. Rodrigues (tgr002@bucknell.edu)

 *

 * Modified by: Mitch Watrous (watrous@u.washington.edu)

 */

#include "ns3/ipv4-packet-probe.h"

#include "ns3/object.h"

#include "ns3/log.h"

#include "ns3/names.h"

#include "ns3/config.h"

#include "ns3/trace-source-accessor.h"

namespace ns3 {

NS_LOG_COMPONENT_DEFINE ("Ipv4PacketProbe");

NS_OBJECT_ENSURE_REGISTERED (Ipv4PacketProbe);

TypeId

Ipv4PacketProbe::GetTypeId ()

{

  static TypeId tid = TypeId ("ns3::Ipv4PacketProbe")

    .SetParent<Probe> ()

    .AddConstructor<Ipv4PacketProbe> ()

    .AddTraceSource ( "Output",

                      "The packet plus its IPv4 object and interface that serve as the output for this probe",

                      MakeTraceSourceAccessor (&Ipv4PacketProbe::m_output))

    .AddTraceSource ( "OutputBytes",

                      "The number of bytes in the packet",

                      MakeTraceSourceAccessor (&Ipv4PacketProbe::m_outputBytes))

  ;

  return tid;

}

Ipv4PacketProbe::Ipv4PacketProbe ()

{

  NS_LOG_FUNCTION (this);

  m_packet    = 0;

  m_packetSizeOld = 0;

  m_ipv4      = 0;

  m_interface = 0;

}

Ipv4PacketProbe::~Ipv4PacketProbe ()

{

  NS_LOG_FUNCTION (this);

}

void

Ipv4PacketProbe::SetValue (Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface)

{

  NS_LOG_FUNCTION (this << packet << ipv4 << interface);

  m_packet    = packet;

  m_ipv4      = ipv4;

  m_interface = interface;

  m_output (packet, ipv4, interface);

  uint32_t packetSizeNew = packet->GetSize ();

  m_outputBytes (m_packetSizeOld, packetSizeNew);

  m_packetSizeOld = packetSizeNew;

}

void

Ipv4PacketProbe::SetValueByPath (std::string path, Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface)

{

  NS_LOG_FUNCTION (path << packet << ipv4 << interface);

  Ptr<Ipv4PacketProbe> probe = Names::Find<Ipv4PacketProbe> (path);

  NS_ASSERT_MSG (probe, "Error:  Can't find probe for path " << path);

  probe->SetValue (packet, ipv4, interface);

}

bool

Ipv4PacketProbe::ConnectByObject (std::string traceSource, Ptr<Object> obj)

{

  NS_LOG_FUNCTION (this << traceSource << obj);

  NS_LOG_DEBUG ("Name of probe (if any) in names database: " << Names::FindPath (obj));

  bool connected = obj->TraceConnectWithoutContext (traceSource, MakeCallback (&ns3::Ipv4PacketProbe::TraceSink, this));

  return connected;

}

void

Ipv4PacketProbe::ConnectByPath (std::string path)

{

  NS_LOG_FUNCTION (this << path);

  NS_LOG_DEBUG ("Name of probe to search for in config database: " << path);

  Config::ConnectWithoutContext (path, MakeCallback (&ns3::Ipv4PacketProbe::TraceSink, this));

}

void

Ipv4PacketProbe::TraceSink (Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface)

{

  NS_LOG_FUNCTION (this << packet << ipv4 << interface);

  if (IsEnabled ())

    {

      m_packet    = packet;

      m_ipv4      = ipv4;

      m_interface = interface;

      m_output (packet, ipv4, interface);

      uint32_t packetSizeNew = packet->GetSize ();

      m_outputBytes (m_packetSizeOld, packetSizeNew);

      m_packetSizeOld = packetSizeNew;

    }

}

} // namespace ns3