IPv6Datagram.h

Go to the documentation of this file.

//
// Copyright (C) 2005 Andras Varga
//
// This program 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
// of the License, or (at your option) any later version.
//
// 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program; if not, see <http://www.gnu.org/licenses/>.
//


#ifndef _IPv6DATAGRAM_H_
#define _IPv6DATAGRAM_H_

#include <list>
#include "INETDefs.h"
#include "IPv6Datagram_m.h"

class INET_API IPv6Datagram : public IPv6Datagram_Base
{
  protected:
    typedef std::vector<IPv6ExtensionHeader*> ExtensionHeaders;
    ExtensionHeaders extensionHeaders;

  public:
    IPv6Datagram(const char *name=NULL, int kind=0) : IPv6Datagram_Base(name,kind) {}
    IPv6Datagram(const IPv6Datagram& other) : IPv6Datagram_Base(other.getName()) {operator=(other);}
    IPv6Datagram& operator=(const IPv6Datagram& other);
    virtual IPv6Datagram *dup() const {return new IPv6Datagram(*this);}

    virtual void setExtensionHeaderArraySize(unsigned int size);
    virtual void setExtensionHeader(unsigned int k, const IPv6ExtensionHeaderPtr& extensionHeader_var);

    virtual unsigned int getExtensionHeaderArraySize() const;

    virtual IPv6ExtensionHeaderPtr& getExtensionHeader(unsigned int k);

    virtual void addExtensionHeader(IPv6ExtensionHeader *eh, int atPos=-1);

    virtual int calculateHeaderByteLength() const;
};

class INET_API IPv6ExtensionHeader : public IPv6ExtensionHeader_Base
{
  public:
    IPv6ExtensionHeader() : IPv6ExtensionHeader_Base() {}
    IPv6ExtensionHeader(const IPv6ExtensionHeader& other) : IPv6ExtensionHeader_Base() {operator=(other);}
    IPv6ExtensionHeader& operator=(const IPv6ExtensionHeader& other) {IPv6ExtensionHeader_Base::operator=(other); return *this;}
    virtual IPv6ExtensionHeader *dup() const {throw cRuntimeError(this, "dup() should be redefined");}
    virtual IPProtocolId getExtensionType() const;
    virtual int getByteLength() const;
};

#endif