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