DatagramPacket.h

Go to the documentation of this file.

// -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 4; tab-width: 4; -*-
// vim: set shiftwidth=4 softtabstop=4 expandtab:
/*
 ********************************************************************
 ** NIDAS: NCAR In-situ Data Acquistion Software
 **
 ** 2006, Copyright University Corporation for Atmospheric Research
 **
 ** This program is free software; you can redistribute it and/or modify
 ** it under the terms of the GNU 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 General Public License for more details.
 **
 ** The LICENSE.txt file accompanying this software contains
 ** a copy of the GNU General Public License. If it is not found,
 ** write to the Free Software Foundation, Inc.,
 ** 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 **
 ********************************************************************
*/

#ifndef NIDAS_UTIL_DATAGRAMPACKET_H
#define NIDAS_UTIL_DATAGRAMPACKET_H

#include "Inet4SocketAddress.h"

#include <sys/param.h>  // MIN

namespace nidas { namespace util {

class DatagramPacketBase {
public:

    DatagramPacketBase(int l) :
        maxlength(l),length(l),addr(new Inet4SocketAddress()) {}

    DatagramPacketBase(int l, const Inet4Address& add, int port):
        maxlength(l),length(l),
        addr(new Inet4SocketAddress(add,port)) {}

    DatagramPacketBase(int l, const SocketAddress& add):
        maxlength(l),length(l),addr(add.clone()) {}

    DatagramPacketBase(const DatagramPacketBase& x) :
        maxlength(x.maxlength),length(x.length),addr(x.addr->clone())
    {
    }

    DatagramPacketBase& operator=(const DatagramPacketBase& rhs)
    {
        if (this != &rhs) {
            maxlength = rhs.maxlength;
            length = rhs.length;
            delete addr;
            addr = rhs.addr->clone();
        }
    return *this;
    }

    virtual ~DatagramPacketBase()
    {
        delete addr;
    }

    SocketAddress& getSocketAddress() const { return *addr; }

    void setSocketAddress(const SocketAddress& val)
    {
        SocketAddress* oldaddr = addr;
        addr = val.clone();
    delete oldaddr;
    }

    struct sockaddr* getSockAddrPtr() { return addr->getSockAddrPtr(); }

    const struct sockaddr* getConstSockAddrPtr() const {
        return addr->getConstSockAddrPtr();
    }

    int getSockAddrLen() const { return addr->getSockAddrLen(); }

    virtual void* getDataVoidPtr() = 0;

    virtual const void* getConstDataVoidPtr() const = 0;

    virtual void setLength(int val) { length = val; }

    virtual int getLength() const { return length; }

    void setMaxLength(int val) { maxlength = val; }

    int getMaxLength() const { return maxlength; }

protected:
    int maxlength;

    int length;

    SocketAddress* addr;

};

template <class DataT>
class DatagramPacketT: public DatagramPacketBase {
public:

    DatagramPacketT(DataT* buf, int n) :
        DatagramPacketBase(n * sizeof(DataT)),data(buf) {}

    DatagramPacketT(DataT* buf, int n, const Inet4Address& add, int port):
        DatagramPacketBase(n * sizeof(DataT),add,port),data(buf) {}

    DatagramPacketT(DataT* buf, int n, const SocketAddress& add):
        DatagramPacketBase(n * sizeof(DataT),add),data(buf) {}

    DatagramPacketT(const DatagramPacketT& x):
        DatagramPacketBase(x), data(x.data)
    {
    }

    DatagramPacketT& operator=(const DatagramPacketT& rhs)
    {
        if (this != &rhs) {
            (*(DatagramPacketBase*)this) = rhs;
            data = rhs.data;
        }
        return *this;
    }

    virtual void* getDataVoidPtr() { return data; }

    virtual const void* getConstDataVoidPtr() const { return data; }

    DataT* getData() { return data; }

    void setData(DataT *val) { data = val; }

protected:
    DataT* data;

private:
};

class DatagramPacket: public DatagramPacketT<char>
{
public:
    DatagramPacket(char* buf, int length) :
        DatagramPacketT<char>(buf,length) {}

    DatagramPacket(char* buf, int length, const Inet4Address& add, int port):
        DatagramPacketT<char>(buf,length,add,port) {}

    DatagramPacket(char* buf, int length, const SocketAddress& add):
        DatagramPacketT<char>(buf,length,add) {}
};

} } // namespace nidas namespace util

#endif