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

#include "ns3/packet.h"

#include "ns3/header.h"

#include <iostream>

using namespace ns3;

/* A sample Header implementation

 */

class MyHeader : public Header {

public:

  MyHeader ();

  virtual ~MyHeader ();

  void SetData (uint16_t data);

  uint16_t GetData (void) const;

private:

  virtual void PrintTo (std::ostream &os) const;

  virtual void SerializeTo (Buffer::Iterator start) const;

  virtual uint32_t DeserializeFrom (Buffer::Iterator start);

  virtual uint32_t GetSerializedSize (void) const;

  uint16_t m_data;

};

MyHeader::MyHeader ()

{}

MyHeader::~MyHeader ()

{}

void 

MyHeader::PrintTo (std::ostream &os) const

{

  os << "MyHeader data=" << m_data << std::endl;

}

uint32_t

MyHeader::GetSerializedSize (void) const

{

  return 2;

}

void

MyHeader::SerializeTo (Buffer::Iterator start) const

{

  // serialize in head of buffer

  start.WriteHtonU16 (m_data);

}

uint32_t

MyHeader::DeserializeFrom (Buffer::Iterator start)

{

  // deserialize from head of buffer

  m_data = start.ReadNtohU16 ();

  return GetSerializedSize ();

}

void 

MyHeader::SetData (uint16_t data)

{

  m_data = data;

}

uint16_t 

MyHeader::GetData (void) const

{

  return m_data;

}

/* A sample Tag implementation

 */

struct MyTag {

  uint16_t m_streamId;

};

static TagRegistration<struct MyTag> g_MyTagRegistration ("ns3::MyTag", 0);

static void

Receive (Packet p)

{

  MyHeader my;

  p.RemoveHeader (my);

  std::cout << "received data=" << my.GetData () << std::endl;

  struct MyTag myTag;

  p.PeekTag (myTag);

}

int main (int argc, char *argv[])

{

  Packet p;

  MyHeader my;

  my.SetData (2);

  std::cout << "send data=2" << std::endl;

  p.AddHeader (my);

  struct MyTag myTag;

  myTag.m_streamId = 5;

  p.AddTag (myTag);

  Receive (p);

  return 0;

}