MultipleUDPSockets.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
 **
 ** 2009, 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_CORE_MULTIPLESOCKETUDP_H
#define NIDAS_CORE_MULTIPLESOCKETUDP_H

#include "McSocketUDP.h"

#include <string>
#include <iostream>

namespace nidas { namespace core {

class MultipleUDPSockets: public McSocketUDP
{
public:
    MultipleUDPSockets();

    MultipleUDPSockets(const MultipleUDPSockets&x);

    MultipleUDPSockets* clone() const;

    IOChannel* connect() throw(nidas::util::IOException);

    void connected(nidas::util::DatagramSocket* sock,
        const nidas::util::Inet4PacketInfoX& pktinfo) throw();

    size_t getBufferSize() const throw();

    size_t read(void*, size_t) throw(nidas::util::IOException)
    {
        assert(false);
    return 0;
    }

    size_t write(const void* buf, size_t len)
            throw(nidas::util::IOException);

    size_t write(const struct iovec* iov, int iovcnt)
            throw(nidas::util::IOException);

    void close() throw(nidas::util::IOException);

    int getFd() const;

    void fromDOMElement(const xercesc::DOMElement*)
        throw(nidas::util::InvalidParameterException);

    void addClient(const ConnectionInfo& info);

    void removeClient(const nidas::util::Inet4SocketAddress& remoteSAddr);

    void removeClient(nidas::util::DatagramSocket*);

    void setDataPort(unsigned short val) 
    {
        _dataPortNumber = htons(val);
    }

    unsigned short getDataPort()  const
    {
        return ntohs(_dataPortNumber);
    }

private:

    void handleChangedSockets();

    mutable nidas::util::Mutex _socketMutex;

    std::list<std::pair<nidas::util::DatagramSocket*,nidas::util::Inet4SocketAddress> > _sockets;

    std::list<std::pair<nidas::util::DatagramSocket*,nidas::util::Inet4SocketAddress> > _pendingSockets;

    std::list<nidas::util::DatagramSocket*> _pendingRemoveSockets;

    std::map<nidas::util::Inet4SocketAddress,nidas::util::Inet4Address>
        _multicastInterfaces;

    std::map<nidas::util::Inet4Address,std::set<nidas::util::Inet4SocketAddress> >
        _multicastClients;

    std::map<nidas::util::Inet4Address,nidas::util::DatagramSocket*>
        _multicastSockets;

    std::map<nidas::util::Inet4SocketAddress,nidas::util::DatagramSocket*>
        _unicastSockets;

    bool _socketsChanged;

    unsigned short _dataPortNumber;
};


}}  // namespace nidas namespace core

#endif