ipv4-l3-protocol.h

Go to the documentation of this file.

// -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*-
//
// Copyright (c) 2006 Georgia Tech Research Corporation
//
// 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
//
// Author: George F. Riley<riley@ece.gatech.edu>
//
 
#ifndef IPV4_L3_PROTOCOL_H
#define IPV4_L3_PROTOCOL_H
 
#include <list>
#include <map>
#include <vector>
#include <stdint.h>
#include "ns3/ipv4-address.h"
#include "ns3/ptr.h"
#include "ns3/net-device.h"
#include "ns3/ipv4.h"
#include "ns3/traced-callback.h"
#include "ns3/ipv4-header.h"
#include "ns3/ipv4-routing-protocol.h"
#include "ns3/nstime.h"
#include "ns3/simulator.h"
 
class Ipv4L3ProtocolTestCase;
 
namespace ns3 {
 
class Packet;
class NetDevice;
class Ipv4Interface;
class Ipv4Address;
class Ipv4Header;
class Ipv4RoutingTableEntry;
class Ipv4Route;
class Node;
class Socket;
class Ipv4RawSocketImpl;
class IpL4Protocol;
class Icmpv4L4Protocol;
 
class Ipv4L3Protocol : public Ipv4
{
public:
  static TypeId GetTypeId (void);
  static const uint16_t PROT_NUMBER; 
 
  Ipv4L3Protocol();
  virtual ~Ipv4L3Protocol ();
 
  enum DropReason 
  {
    DROP_TTL_EXPIRED = 1,   
    DROP_NO_ROUTE,   
    DROP_BAD_CHECKSUM,   
    DROP_INTERFACE_DOWN,   
    DROP_ROUTE_ERROR,   
    DROP_FRAGMENT_TIMEOUT, 
    DROP_DUPLICATE  
  };
 
  void SetNode (Ptr<Node> node);
 
  // functions defined in base class Ipv4
 
  void SetRoutingProtocol (Ptr<Ipv4RoutingProtocol> routingProtocol);
  Ptr<Ipv4RoutingProtocol> GetRoutingProtocol (void) const;
 
  Ptr<Socket> CreateRawSocket (void);
  void DeleteRawSocket (Ptr<Socket> socket);
 
  virtual void Insert (Ptr<IpL4Protocol> protocol);
  virtual void Insert (Ptr<IpL4Protocol> protocol, uint32_t interfaceIndex);
 
  virtual void Remove (Ptr<IpL4Protocol> protocol);
  virtual void Remove (Ptr<IpL4Protocol> protocol, uint32_t interfaceIndex);
 
  virtual Ptr<IpL4Protocol> GetProtocol (int protocolNumber) const;
  virtual Ptr<IpL4Protocol> GetProtocol (int protocolNumber, int32_t interfaceIndex) const;
 
  virtual Ipv4Address SourceAddressSelection (uint32_t interface, Ipv4Address dest);
 
  void SetDefaultTtl (uint8_t ttl);
 
  void Receive ( Ptr<NetDevice> device, Ptr<const Packet> p, uint16_t protocol, const Address &from,
                 const Address &to, NetDevice::PacketType packetType);
 
  void Send (Ptr<Packet> packet, Ipv4Address source, 
             Ipv4Address destination, uint8_t protocol, Ptr<Ipv4Route> route);
  void SendWithHeader (Ptr<Packet> packet, Ipv4Header ipHeader, Ptr<Ipv4Route> route);
 
  uint32_t AddInterface (Ptr<NetDevice> device);
  Ptr<Ipv4Interface> GetInterface (uint32_t i) const;
  uint32_t GetNInterfaces (void) const;
 
  int32_t GetInterfaceForAddress (Ipv4Address addr) const;
  int32_t GetInterfaceForPrefix (Ipv4Address addr, Ipv4Mask mask) const;
  int32_t GetInterfaceForDevice (Ptr<const NetDevice> device) const;
  bool IsDestinationAddress (Ipv4Address address, uint32_t iif) const;
 
  bool AddAddress (uint32_t i, Ipv4InterfaceAddress address);
  Ipv4InterfaceAddress GetAddress (uint32_t interfaceIndex, uint32_t addressIndex) const;
  uint32_t GetNAddresses (uint32_t interface) const;
  bool RemoveAddress (uint32_t interfaceIndex, uint32_t addressIndex);
  bool RemoveAddress (uint32_t interface, Ipv4Address address);
  Ipv4Address SelectSourceAddress (Ptr<const NetDevice> device,
                                   Ipv4Address dst, Ipv4InterfaceAddress::InterfaceAddressScope_e scope);
 
 
  void SetMetric (uint32_t i, uint16_t metric);
  uint16_t GetMetric (uint32_t i) const;
  uint16_t GetMtu (uint32_t i) const;
  bool IsUp (uint32_t i) const;
  void SetUp (uint32_t i);
  void SetDown (uint32_t i);
  bool IsForwarding (uint32_t i) const;
  void SetForwarding (uint32_t i, bool val);
 
  Ptr<NetDevice> GetNetDevice (uint32_t i);
 
  bool IsUnicast (Ipv4Address ad) const;
 
  typedef void (* SentTracedCallback)
      (const Ipv4Header & header, Ptr<const Packet> packet, uint32_t interface);
 
  typedef void (* TxRxTracedCallback)
      (Ptr<const Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface);
 
  typedef void (* DropTracedCallback)
      (const Ipv4Header & header, Ptr<const Packet> packet,
          DropReason reason, Ptr<Ipv4> ipv4,
          uint32_t interface);
 
protected:
 
  virtual void DoDispose (void);
  virtual void NotifyNewAggregate ();
private:
  friend class ::Ipv4L3ProtocolTestCase;
 
  Ipv4L3Protocol(const Ipv4L3Protocol &);
 
  Ipv4L3Protocol &operator = (const Ipv4L3Protocol &);
 
  // class Ipv4 attributes
  virtual void SetIpForward (bool forward);
  virtual bool GetIpForward (void) const;
  virtual void SetWeakEsModel (bool model);
  virtual bool GetWeakEsModel (void) const;
 
  void DecreaseIdentification (Ipv4Address source,
                               Ipv4Address destination,
                               uint8_t protocol);
 
  Ipv4Header BuildHeader (
      Ipv4Address source,
      Ipv4Address destination,
      uint8_t protocol,
      uint16_t payloadSize,
      uint8_t ttl,
      uint8_t tos,
      bool mayFragment);
 
  void
  SendRealOut (Ptr<Ipv4Route> route,
               Ptr<Packet> packet,
               Ipv4Header const &ipHeader);
 
  void 
  IpForward (Ptr<Ipv4Route> rtentry, 
             Ptr<const Packet> p, 
             const Ipv4Header &header);
 
  void
  IpMulticastForward (Ptr<Ipv4MulticastRoute> mrtentry, 
                      Ptr<const Packet> p, 
                      const Ipv4Header &header);
 
  void LocalDeliver (Ptr<const Packet> p, Ipv4Header const&ip, uint32_t iif);
 
  void RouteInputError (Ptr<const Packet> p, const Ipv4Header & ipHeader, Socket::SocketErrno sockErrno);
 
  uint32_t AddIpv4Interface (Ptr<Ipv4Interface> interface);
 
  void SetupLoopback (void);
 
  Ptr<Icmpv4L4Protocol> GetIcmp (void) const;
 
  bool IsUnicast (Ipv4Address ad, Ipv4Mask interfaceMask) const;
 
  typedef std::pair<Ptr<Packet>, Ipv4Header> Ipv4PayloadHeaderPair;
 
  void DoFragmentation (Ptr<Packet> packet, const Ipv4Header& ipv4Header, uint32_t outIfaceMtu, std::list<Ipv4PayloadHeaderPair>& listFragments);
 
  bool ProcessFragment (Ptr<Packet>& packet, Ipv4Header & ipHeader, uint32_t iif);
 
  void CallTxTrace (const Ipv4Header & ipHeader, Ptr<Packet> packet, Ptr<Ipv4> ipv4, uint32_t interface);
 
  typedef std::vector<Ptr<Ipv4Interface> > Ipv4InterfaceList;
  typedef std::map<Ptr<const NetDevice>, uint32_t > Ipv4InterfaceReverseContainer;
  typedef std::list<Ptr<Ipv4RawSocketImpl> > SocketList;
 
  typedef std::pair<int, int32_t> L4ListKey_t;
 
  typedef std::map<L4ListKey_t, Ptr<IpL4Protocol> > L4List_t;
 
  bool m_ipForward;      
  bool m_weakEsModel;    
  L4List_t m_protocols;  
  Ipv4InterfaceList m_interfaces; 
  Ipv4InterfaceReverseContainer m_reverseInterfacesContainer; 
  uint8_t m_defaultTtl;  
  std::map<std::pair<uint64_t, uint8_t>, uint16_t> m_identification; 
  Ptr<Node> m_node; 
 
  TracedCallback<const Ipv4Header &, Ptr<const Packet>, uint32_t> m_sendOutgoingTrace;
  TracedCallback<const Ipv4Header &, Ptr<const Packet>, uint32_t> m_unicastForwardTrace;
  TracedCallback<const Ipv4Header &, Ptr<const Packet>, uint32_t> m_multicastForwardTrace;
  TracedCallback<const Ipv4Header &, Ptr<const Packet>, uint32_t> m_localDeliverTrace;
 
  // The following two traces pass a packet with an IP header
  TracedCallback<Ptr<const Packet>, Ptr<Ipv4>,  uint32_t> m_txTrace;
  TracedCallback<Ptr<const Packet>, Ptr<Ipv4>, uint32_t> m_rxTrace;
  // <ip-header, payload, reason, ifindex> (ifindex not valid if reason is DROP_NO_ROUTE)
  TracedCallback<const Ipv4Header &, Ptr<const Packet>, DropReason, Ptr<Ipv4>, uint32_t> m_dropTrace;
 
  Ptr<Ipv4RoutingProtocol> m_routingProtocol; 
 
  SocketList m_sockets; 
 
  typedef std::pair<uint64_t, uint32_t> FragmentKey_t;
 
  typedef std::list< std::tuple <Time, FragmentKey_t, Ipv4Header, uint32_t > > FragmentsTimeoutsList_t;
  typedef std::list< std::tuple <Time, FragmentKey_t, Ipv4Header, uint32_t > >::iterator FragmentsTimeoutsListI_t;
 
  void HandleFragmentsTimeout (FragmentKey_t key, Ipv4Header & ipHeader, uint32_t iif);
 
  FragmentsTimeoutsListI_t SetTimeout (FragmentKey_t key, Ipv4Header ipHeader, uint32_t iif);
 
  void HandleTimeout (void);
 
  FragmentsTimeoutsList_t m_timeoutEventList;  
 
  EventId m_timeoutEvent;  
 
  class Fragments : public SimpleRefCount<Fragments>
  {
  public:
    Fragments ();
 
    ~Fragments ();
 
    void AddFragment (Ptr<Packet> fragment, uint16_t fragmentOffset, bool moreFragment);
 
    bool IsEntire () const;
 
    Ptr<Packet> GetPacket () const;
 
    Ptr<Packet> GetPartialPacket () const;
 
    void SetTimeoutIter (FragmentsTimeoutsListI_t iter);
 
    FragmentsTimeoutsListI_t GetTimeoutIter ();
 
  private:
    bool m_moreFragment;
 
    std::list<std::pair<Ptr<Packet>, uint16_t> > m_fragments;
 
    FragmentsTimeoutsListI_t m_timeoutIter;
  };
 
  typedef std::map< FragmentKey_t, Ptr<Fragments> > MapFragments_t;
 
  MapFragments_t       m_fragments; 
  Time                 m_fragmentExpirationTimeout; 
 
  typedef std::tuple <uint64_t, uint8_t, Ipv4Address, Ipv4Address> DupTuple_t;
  typedef std::map<DupTuple_t, Time> DupMap_t;
 
  bool UpdateDuplicate (Ptr<const Packet> p, const Ipv4Header &header);
  void RemoveDuplicates (void);
 
  bool                m_enableDpd;    
  DupMap_t            m_dups;         
  Time                m_expire;       
  Time                m_purge;        
  EventId             m_cleanDpd;     
};
 
} // Namespace ns3
 
#endif /* IPV4_L3_PROTOCOL_H */