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

 /* 

  * Copyright (c) 2009 University of Texas at Dallas

  * 

  * 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: Qasim Javed <qasim@utdallas.edu>

  */

 #ifndef NETFILTER_CONNTRACK_TUPLE_H

 #define NETFILTER_CONNTRACK_TUPLE_H

 #include "ns3/log.h"

 #include "ns3/ipv4-address.h"

 #include "ns3/ref-count-base.h"

 #include "ns3/object.h"

 #include "ip-conntrack-info.h"

 //#include "jhash.h"

 namespace ns3 {

 class Object;

 class Ipv4Address;

 class NetfilterConntrackTuple : public RefCountBase {

   public:

     NetfilterConntrackTuple ();

     NetfilterConntrackTuple (Ipv4Address src, uint16_t srcPort, Ipv4Address dst, uint16_t dstPort);

     bool operator== (const NetfilterConntrackTuple t) const;

     bool SourceEqual (NetfilterConntrackTuple t1, NetfilterConntrackTuple t2);

     bool DestinationEqual (NetfilterConntrackTuple t1, NetfilterConntrackTuple t2);

     NetfilterConntrackTuple Invert ();

     Ipv4Address GetSource () const;

     Ipv4Address GetDestination () const;

     uint16_t GetSourcePort () const;

     uint16_t GetDestinationPort () const;

     char * ToString () const;

     uint16_t GetDestinationProtocol () const;

     uint8_t GetDirection () const;

     uint16_t GetProtocol ();

     void SetSource (Ipv4Address source);

     void SetSourcePort (uint16_t source);

     void SetDestination (Ipv4Address destination);

     void SetDestinationPort (uint16_t destination);

     void SetProtocol (uint16_t protocol);

     void SetDirection (ConntrackDirection_t direction);

     void Print (std::ostream &os) const;

     NetfilterConntrackTuple& operator= (const NetfilterConntrackTuple&  tuple);

     friend std::ostream& operator << (std::ostream& os, NetfilterConntrackTuple const& tuple);

   private:

     Ipv4Address m_l3Source;

     uint16_t m_l3Protocol;

     uint16_t m_l4Source;

     Ipv4Address m_l3Destination;

     uint16_t m_l4Destination;

     uint8_t m_protocolNumber;

     uint8_t m_direction;

 };

 class ConntrackTupleHash : public std::unary_function<NetfilterConntrackTuple, size_t> {

 public:

   size_t operator()(const NetfilterConntrackTuple &x) const;

 };

 }

 #endif /* NETFILTER_CONNTRACK_TUPLE_H */