ipv4-flow-classifier.h

Go to the documentation of this file.

//
// Copyright (c) 2009 INESC Porto
//
// 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: Gustavo J. A. M. Carneiro  <gjc@inescporto.pt> <gjcarneiro@gmail.com>
//
 
#ifndef IPV4_FLOW_CLASSIFIER_H
#define IPV4_FLOW_CLASSIFIER_H
 
#include "flow-classifier.h"
 
#include "ns3/ipv4-header.h"
 
#include <map>
#include <stdint.h>
 
namespace ns3
{
 
class Packet;
 
class Ipv4FlowClassifier : public FlowClassifier
{
  public:
    struct FiveTuple
    {
        Ipv4Address sourceAddress;      
        Ipv4Address destinationAddress; 
        uint8_t protocol;               
        uint16_t sourcePort;            
        uint16_t destinationPort;       
    };
 
    Ipv4FlowClassifier();
 
    bool Classify(const Ipv4Header& ipHeader,
                  Ptr<const Packet> ipPayload,
                  uint32_t* out_flowId,
                  uint32_t* out_packetId);
 
    FiveTuple FindFlow(FlowId flowId) const;
 
    class SortByCount
    {
      public:
        bool operator()(std::pair<Ipv4Header::DscpType, uint32_t> left,
                        std::pair<Ipv4Header::DscpType, uint32_t> right);
    };
 
    std::vector<std::pair<Ipv4Header::DscpType, uint32_t>> GetDscpCounts(FlowId flowId) const;
 
    void SerializeToXmlStream(std::ostream& os, uint16_t indent) const override;
 
  private:
    std::map<FiveTuple, FlowId> m_flowMap;
    std::map<FlowId, FlowPacketId> m_flowPktIdMap;
    std::map<FlowId, std::map<Ipv4Header::DscpType, uint32_t>> m_flowDscpMap;
};
 
bool operator<(const Ipv4FlowClassifier::FiveTuple& t1, const Ipv4FlowClassifier::FiveTuple& t2);
 
bool operator==(const Ipv4FlowClassifier::FiveTuple& t1, const Ipv4FlowClassifier::FiveTuple& t2);
 
} // namespace ns3
 
#endif /* IPV4_FLOW_CLASSIFIER_H */