trafficselector.h

/***************************************************************************
*   Copyright (C) 2005 by                                                 *
*   Pedro J. Fernandez Ruiz    pedroj.fernandez@dif.um.es                 *
*   Alejandro Perez Mendez     alejandro_perez@dif.um.es                  *
*                                                                         *
*   This library is free software; you can redistribute it and/or         *
*   modify it under the terms of the GNU Lesser General Public            *
*   License as published by the Free Software Foundation; either          *
*   version 2.1 of the License, or (at your option) any later version.    *
*                                                                         *
*   This library 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     *
*   Lesser General Public License for more details.                       *
*                                                                         *
*   You should have received a copy of the GNU Lesser General Public      *
*   License along with this library; if not, write to the Free Software   *
*   Foundation, Inc., 51 Franklin St, Fifth Floor,                        *
*   Boston, MA  02110-1301  USA                                           *
***************************************************************************/
#ifndef TRAFFICSELECTOR_H
#define TRAFFICSELECTOR_H

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "printable.h"
#include "ipaddress.h"
#include "bytebuffer.h"

namespace openikev2 {

    class TrafficSelector : public Printable {

            /****************************** ENUMS ******************************/
        public:
            enum TS_TYPE{
                TS_IPV4_ADDR_RANGE = 7,           
                TS_IPV6_ADDR_RANGE = 8,           
        };

            /****************************** ATTRIBUTES ******************************/
        public:
            TS_TYPE ts_type;                
            auto_ptr<ByteArray> start_addr;
            auto_ptr<ByteArray> end_addr;   
            uint16_t start_port;            
            uint16_t end_port;              
            uint8_t ip_protocol_id;         
            /****************************** METHODS ******************************/
        protected:

            static uint16_t getExpectedAddressSize( TS_TYPE ts_type);
            
            void construct(TS_TYPE ts_type, auto_ptr<ByteArray> start_addr, auto_ptr<ByteArray> end_addr, uint16_t start_port, uint16_t end_port, uint8_t ip_protocol_id );

            void construct( IpAddress& address, uint8_t prefixlen, uint16_t port , uint8_t ip_protocol_id );

        public:
            TrafficSelector(TS_TYPE ts_type, auto_ptr<ByteArray> start_addr, auto_ptr<ByteArray> end_addr, uint16_t start_port, uint16_t end_port, uint8_t ip_protocol_id );

            TrafficSelector(TS_TYPE ts_type, auto_ptr<ByteArray> start_addr, auto_ptr<ByteArray> end_addr, uint8_t start_icmp_type, uint8_t start_icmp_code, uint8_t end_icmp_type, uint8_t end_icmp_code, uint8_t ip_protocol_id );

            TrafficSelector( IpAddress& address, uint8_t prefixlen, uint16_t port, uint8_t ip_protocol_id );

            TrafficSelector( IpAddress& address, uint8_t prefixlen, uint8_t icmp_type, uint8_t icmp_code, uint8_t ip_protocol_id );

            TrafficSelector( const TrafficSelector& other );

            TrafficSelector( ByteBuffer& byte_buffer );

            static string TS_TYPE_STR( TS_TYPE ts_type );

            static auto_ptr<TrafficSelector> intersection( const TrafficSelector& ts1, const TrafficSelector& ts2 );

            virtual bool operator>=( const TrafficSelector& other ) const;
            
            virtual bool operator<=( const TrafficSelector& other ) const;
            
            virtual bool operator==( const TrafficSelector& other ) const;

            virtual uint16_t getStartPort() const ;

            virtual uint16_t getEndPort() const ;

            virtual uint8_t getStartIcmpType() const ;

            virtual uint8_t getStartIcmpCode() const ;

            virtual uint8_t getEndIcmpType() const;

            virtual uint8_t getEndIcmpCode() const;

            virtual auto_ptr<TrafficSelector> clone() const;

            virtual void getBinaryRepresentation( ByteBuffer& byte_buffer ) const;

            virtual string toStringTab( uint8_t tabs ) const;

            virtual ~TrafficSelector();
    };
}
#endif

---