#include "icmpv4.h"
#include "ns3/packet.h"

namespace ns3 {

/********************************************************
 *        Icmpv4Header
 ********************************************************/

TypeId 
Icmpv4Header::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::Icmpv4Header")
    .SetParent<Header> ()
    .AddConstructor<Icmpv4Header> ()
    ;
  return tid;
}
Icmpv4Header::Icmpv4Header ()
  : m_type (0),
    m_code (0),
    m_calcChecksum (false)
{}
Icmpv4Header::~Icmpv4Header ()
{}
void 
Icmpv4Header::EnableChecksum (void)
{
  m_calcChecksum = true;
}
TypeId 
Icmpv4Header::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
uint32_t 
Icmpv4Header::GetSerializedSize (void) const
{
  return 4;
}
void 
Icmpv4Header::Serialize (Buffer::Iterator start) const
{
  Buffer::Iterator i = start;
  i.WriteU8 (m_type);
  i.WriteU8 (m_code);
  i.WriteHtonU16 (0);
  if (m_calcChecksum)
    {
      i = start;
      uint16_t checksum = i.CalculateIpChecksum (i.GetSize ());
      i = start;
      i.Next (2);
      i.WriteU16 (checksum);
    }
  
}
uint32_t 
Icmpv4Header::Deserialize (Buffer::Iterator start)
{
  m_type = start.ReadU8 ();
  m_code = start.ReadU8 ();
  uint16_t checksum;
  checksum = start.ReadNtohU16 ();
  return 4;
}
void 
Icmpv4Header::Print (std::ostream &os) const
{
  os << "type=" << (uint32_t)m_type << ", code=" << (uint32_t)m_code;
}

void 
Icmpv4Header::SetType (uint8_t type)
{
  m_type = type;
}
void 
Icmpv4Header::SetCode (uint8_t code)
{
  m_code = code;
}
uint8_t 
Icmpv4Header::GetType (void) const
{
  return m_type;
}
uint8_t 
Icmpv4Header::GetCode (void) const
{
  return m_code;
}

/********************************************************
 *        Icmpv4Echo
 ********************************************************/

void 
Icmpv4Echo::SetIdentifier (uint16_t id)
{
  m_identifier = id;
}
void 
Icmpv4Echo::SetSequenceNumber (uint16_t seq)
{
  m_sequence = seq;
}
void 
Icmpv4Echo::SetData (Ptr<const Packet> data)
{
  m_data = data->Copy ();
}
uint16_t 
Icmpv4Echo::GetIdentifier (void) const
{
  return m_identifier;
}
uint16_t 
Icmpv4Echo::GetSequenceNumber (void) const
{
  return m_sequence;
}
Ptr<const Packet> 
Icmpv4Echo::GetData (void) const
{
  return m_data->Copy ();
}


TypeId 
Icmpv4Echo::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::Icmpv4Echo")
    .SetParent<Header> ()
    .AddConstructor<Icmpv4Echo> ()
    ;
  return tid;
}
Icmpv4Echo::Icmpv4Echo ()
  : m_identifier (0),
    m_sequence (0),
    m_data (0)
{}
Icmpv4Echo::~Icmpv4Echo ()
{}
TypeId 
Icmpv4Echo::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
uint32_t 
Icmpv4Echo::GetSerializedSize (void) const
{
  return 4 + m_data->GetSize ();
}
void 
Icmpv4Echo::Serialize (Buffer::Iterator start) const
{
  start.WriteHtonU16 (m_identifier);
  start.WriteHtonU16 (m_sequence);
  start.Write (m_data->PeekData (), m_data->GetSize ());
}
uint32_t 
Icmpv4Echo::Deserialize (Buffer::Iterator start)
{
  m_identifier = start.ReadNtohU16 ();
  m_sequence = start.ReadNtohU16 ();
  NS_ASSERT (start.GetSize () >= 4);
  uint32_t size = start.GetSize () - 4;
  uint8_t *buffer = new uint8_t[size] ();
  start.Read (buffer, size);
  m_data = Create<Packet> (buffer, size);
  delete[] buffer;
  return start.GetSize ();
}
void 
Icmpv4Echo::Print (std::ostream &os) const
{
  os << "identifier=" << m_identifier << ", sequence="  << m_sequence;
}


/********************************************************
 *        Icmpv4DestinationUnreachable
 ********************************************************/

TypeId 
Icmpv4DestinationUnreachable::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::Icmpv4DestinationUnreachable")
    .SetParent<Header> ()
    .AddConstructor<Icmpv4DestinationUnreachable> ()
    ;
  return tid;
}
Icmpv4DestinationUnreachable::Icmpv4DestinationUnreachable ()
{
  // make sure that thing is initialized to get initialized bytes
  // when the ip payload's size is smaller than 8 bytes.
  for (uint8_t j = 0; j < 8; j++)
    {
      m_data[j] = 0;
    }
}

void 
Icmpv4DestinationUnreachable::SetNextHopMtu (uint16_t mtu)
{
  m_nextHopMtu = mtu;
}
uint16_t 
Icmpv4DestinationUnreachable::GetNextHopMtu (void) const
{
  return m_nextHopMtu;
}

void 
Icmpv4DestinationUnreachable::SetData (Ptr<const Packet> data)
{
  data->CopyData (m_data, 8);
}
void 
Icmpv4DestinationUnreachable::SetHeader (Ipv4Header header)
{
  m_header = header;
}
void 
Icmpv4DestinationUnreachable::GetData (uint8_t payload[8]) const
{
  memcpy (payload, m_data, 8);
}
Ipv4Header 
Icmpv4DestinationUnreachable::GetHeader (void) const
{
  return m_header;
}


Icmpv4DestinationUnreachable::~Icmpv4DestinationUnreachable ()
{}
TypeId 
Icmpv4DestinationUnreachable::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
uint32_t 
Icmpv4DestinationUnreachable::GetSerializedSize (void) const
{
  return 4 + m_header.GetSerializedSize () + 8;
}
void 
Icmpv4DestinationUnreachable::Serialize (Buffer::Iterator start) const
{
  start.WriteU16 (0);
  start.WriteHtonU16 (m_nextHopMtu);
  uint32_t size = m_header.GetSerializedSize ();
  m_header.Serialize (start);
  start.Next (size);
  start.Write (m_data, 8);
}
  
uint32_t 
Icmpv4DestinationUnreachable::Deserialize (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
  i.Next (2);
  m_nextHopMtu = i.ReadNtohU16 ();
  uint32_t read = m_header.Deserialize (i);
  i.Next (read);
  for (uint8_t j = 0; j < 8; j++)
    {
      m_data[j] = i.ReadU8 ();
    }
  return i.GetDistanceFrom (start);
}
void 
Icmpv4DestinationUnreachable::Print (std::ostream &os) const
{
  m_header.Print (os);
  os << " org data=";
  for (uint8_t i = 0; i < 8; i++)
    {
      os << (uint32_t) m_data[i];
      if (i != 8)
	{
	  os << " ";
	}
    }
}

/********************************************************
 *        Icmpv4TimeExceeded
 ********************************************************/

TypeId 
Icmpv4TimeExceeded::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::Icmpv4TimeExceeded")
    .SetParent<Header> ()
    .AddConstructor<Icmpv4TimeExceeded> ()
    ;
  return tid;
}
Icmpv4TimeExceeded::Icmpv4TimeExceeded ()
{
  // make sure that thing is initialized to get initialized bytes
  // when the ip payload's size is smaller than 8 bytes.
  for (uint8_t j = 0; j < 8; j++)
    {
      m_data[j] = 0;
    }
}


void 
Icmpv4TimeExceeded::SetData (Ptr<const Packet> data)
{
  data->CopyData (m_data, 8);
}
void 
Icmpv4TimeExceeded::SetHeader (Ipv4Header header)
{
  m_header = header;
}
void 
Icmpv4TimeExceeded::GetData (uint8_t payload[8]) const
{
  memcpy (payload, m_data, 8);
}
Ipv4Header 
Icmpv4TimeExceeded::GetHeader (void) const
{
  return m_header;
}


Icmpv4TimeExceeded::~Icmpv4TimeExceeded ()
{}
TypeId 
Icmpv4TimeExceeded::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
uint32_t 
Icmpv4TimeExceeded::GetSerializedSize (void) const
{
  return 4 + m_header.GetSerializedSize () + 8;
}
void 
Icmpv4TimeExceeded::Serialize (Buffer::Iterator start) const
{
  start.WriteU32 (0);
  uint32_t size = m_header.GetSerializedSize ();
  m_header.Serialize (start);
  start.Next (size);
  start.Write (m_data, 8);
}
  
uint32_t 
Icmpv4TimeExceeded::Deserialize (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
  i.Next (4);
  uint32_t read = m_header.Deserialize (i);
  i.Next (read);
  for (uint8_t j = 0; j < 8; j++)
    {
      m_data[j] = i.ReadU8 ();
    }
  return i.GetDistanceFrom (start);
}
void 
Icmpv4TimeExceeded::Print (std::ostream &os) const
{
  m_header.Print (os);
  os << " org data=";
  for (uint8_t i = 0; i < 8; i++)
    {
      os << (uint32_t) m_data[i];
      if (i != 8)
	{
	  os << " ";
	}
    }
}

} // namespace ns3
