aodv-packet.h

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2009 IITP RAS
 *
 * 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
 *
 * Based on
 *      NS-2 AODV model developed by the CMU/MONARCH group and optimized and
 *      tuned by Samir Das and Mahesh Marina, University of Cincinnati;
 *
 *      AODV-UU implementation by Erik Nordström of Uppsala University
 *      http://core.it.uu.se/core/index.php/AODV-UU
 *
 * Authors: Elena Buchatskaia <borovkovaes@iitp.ru>
 *          Pavel Boyko <boyko@iitp.ru>
 */
#ifndef AODVPACKET_H
#define AODVPACKET_H

#include <iostream>
#include "ns3/header.h"
#include "ns3/enum.h"
#include "ns3/ipv4-address.h"
#include <map>
#include "ns3/nstime.h"

namespace ns3 {
namespace aodv {

enum MessageType
{
  AODVTYPE_RREQ  = 1,   
  AODVTYPE_RREP  = 2,   
  AODVTYPE_RERR  = 3,   
  AODVTYPE_RREP_ACK = 4 
};

class TypeHeader : public Header
{
public:
  TypeHeader (MessageType t = AODVTYPE_RREQ);

  static TypeId GetTypeId ();
  TypeId GetInstanceTypeId () const;
  uint32_t GetSerializedSize () const;
  void Serialize (Buffer::Iterator start) const;
  uint32_t Deserialize (Buffer::Iterator start);
  void Print (std::ostream &os) const;

  MessageType Get () const
  {
    return m_type;
  }
  bool IsValid () const
  {
    return m_valid;
  }
  bool operator== (TypeHeader const & o) const;
private:
  MessageType m_type; 
  bool m_valid; 
};

std::ostream & operator<< (std::ostream & os, TypeHeader const & h);

class RreqHeader : public Header
{
public:
   RreqHeader (uint8_t flags = 0, uint8_t reserved = 0, uint8_t hopCount = 0,
              uint32_t requestID = 0, Ipv4Address dst = Ipv4Address (),
              uint32_t dstSeqNo = 0, Ipv4Address origin = Ipv4Address (),
              uint32_t originSeqNo = 0);

  static TypeId GetTypeId ();
  TypeId GetInstanceTypeId () const;
  uint32_t GetSerializedSize () const;
  void Serialize (Buffer::Iterator start) const;
  uint32_t Deserialize (Buffer::Iterator start);
  void Print (std::ostream &os) const;

  // Fields
  void SetHopCount (uint8_t count)
  {
    m_hopCount = count;
  }
  uint8_t GetHopCount () const
  {
    return m_hopCount;
  }
  void SetId (uint32_t id)
  {
    m_requestID = id;
  }
  uint32_t GetId () const
  {
    return m_requestID;
  }
  void SetDst (Ipv4Address a)
  {
    m_dst = a;
  }
  Ipv4Address GetDst () const
  {
    return m_dst;
  }
  void SetDstSeqno (uint32_t s)
  {
    m_dstSeqNo = s;
  }
  uint32_t GetDstSeqno () const
  {
    return m_dstSeqNo;
  }
  void SetOrigin (Ipv4Address a)
  {
    m_origin = a;
  }
  Ipv4Address GetOrigin () const
  {
    return m_origin;
  }
  void SetOriginSeqno (uint32_t s)
  {
    m_originSeqNo = s;
  }
  uint32_t GetOriginSeqno () const
  {
    return m_originSeqNo;
  }

  // Flags
  void SetGratuitousRrep (bool f);
  bool GetGratuitousRrep () const;
  void SetDestinationOnly (bool f);
  bool GetDestinationOnly () const;
  void SetUnknownSeqno (bool f);
  bool GetUnknownSeqno () const;

  bool operator== (RreqHeader const & o) const;
private:
  uint8_t        m_flags;          
  uint8_t        m_reserved;       
  uint8_t        m_hopCount;       
  uint32_t       m_requestID;      
  Ipv4Address    m_dst;            
  uint32_t       m_dstSeqNo;       
  Ipv4Address    m_origin;         
  uint32_t       m_originSeqNo;    
};

std::ostream & operator<< (std::ostream & os, RreqHeader const &);

class RrepHeader : public Header
{
public:
  RrepHeader (uint8_t prefixSize = 0, uint8_t hopCount = 0, Ipv4Address dst =
                Ipv4Address (), uint32_t dstSeqNo = 0, Ipv4Address origin =
                Ipv4Address (), Time lifetime = MilliSeconds (0));
  static TypeId GetTypeId ();
  TypeId GetInstanceTypeId () const;
  uint32_t GetSerializedSize () const;
  void Serialize (Buffer::Iterator start) const;
  uint32_t Deserialize (Buffer::Iterator start);
  void Print (std::ostream &os) const;

  // Fields
  void SetHopCount (uint8_t count)
  {
    m_hopCount = count;
  }
  uint8_t GetHopCount () const
  {
    return m_hopCount;
  }
  void SetDst (Ipv4Address a)
  {
    m_dst = a;
  }
  Ipv4Address GetDst () const
  {
    return m_dst;
  }
  void SetDstSeqno (uint32_t s)
  {
    m_dstSeqNo = s;
  }
  uint32_t GetDstSeqno () const
  {
    return m_dstSeqNo;
  }
  void SetOrigin (Ipv4Address a)
  {
    m_origin = a;
  }
  Ipv4Address GetOrigin () const
  {
    return m_origin;
  }
  void SetLifeTime (Time t);
  Time GetLifeTime () const;

  // Flags
  void SetAckRequired (bool f);
  bool GetAckRequired () const;
  void SetPrefixSize (uint8_t sz);
  uint8_t GetPrefixSize () const;

  void SetHello (Ipv4Address src, uint32_t srcSeqNo, Time lifetime);

  bool operator== (RrepHeader const & o) const;
private:
  uint8_t       m_flags;                  
  uint8_t       m_prefixSize;         
  uint8_t             m_hopCount;         
  Ipv4Address   m_dst;              
  uint32_t      m_dstSeqNo;         
  Ipv4Address     m_origin;           
  uint32_t      m_lifeTime;         
};

std::ostream & operator<< (std::ostream & os, RrepHeader const &);

class RrepAckHeader : public Header
{
public:
  RrepAckHeader ();

  static TypeId GetTypeId ();
  TypeId GetInstanceTypeId () const;
  uint32_t GetSerializedSize () const;
  void Serialize (Buffer::Iterator start) const;
  uint32_t Deserialize (Buffer::Iterator start);
  void Print (std::ostream &os) const;

  bool operator== (RrepAckHeader const & o) const;
private:
  uint8_t       m_reserved; 
};

std::ostream & operator<< (std::ostream & os, RrepAckHeader const &);


class RerrHeader : public Header
{
public:
  RerrHeader ();

  static TypeId GetTypeId ();
  TypeId GetInstanceTypeId () const;
  uint32_t GetSerializedSize () const;
  void Serialize (Buffer::Iterator i) const;
  uint32_t Deserialize (Buffer::Iterator start);
  void Print (std::ostream &os) const;

  // No delete flag
  void SetNoDelete (bool f);
  bool GetNoDelete () const;

  bool AddUnDestination (Ipv4Address dst, uint32_t seqNo);
  bool RemoveUnDestination (std::pair<Ipv4Address, uint32_t> & un);
  void Clear ();
  uint8_t GetDestCount () const
  {
    return (uint8_t)m_unreachableDstSeqNo.size ();
  }

  bool operator== (RerrHeader const & o) const;
private:
  uint8_t m_flag;            
  uint8_t m_reserved;        

  std::map<Ipv4Address, uint32_t> m_unreachableDstSeqNo;
};

std::ostream & operator<< (std::ostream & os, RerrHeader const &);

}  // namespace aodv
}  // namespace ns3

#endif /* AODVPACKET_H */