ibrcommon/ibrcommon/net/MulticastSocket.cpp

Go to the documentation of this file.

/*
 * MulticastSocket.cpp
 *
 *  Created on: 28.06.2010
 *      Author: morgenro
 */

#include "ibrcommon/config.h"
#include "ibrcommon/net/MulticastSocket.h"
#include <string.h>

#ifndef HAVE_BZERO
#define bzero(s,n) (memset((s), '\0', (n)), (void) 0)
#endif

namespace ibrcommon
{
        MulticastSocket::MulticastSocket()
         : udpsocket(0)
        {
                int val = 1;
                if ( ::setsockopt(_socket, IPPROTO_IP, IP_MULTICAST_LOOP, (const char *)&val, sizeof(val)) < 0 )
                {
                        throw SocketException("MulticastSocket: setsockopt(IP_MULTICAST_LOOP)");
                }

                u_char ttl = 255; // Multicast TTL
                if ( ::setsockopt(_socket, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl)) < 0 )
                {
                        throw SocketException("MulticastSocket: setsockopt(IP_MULTICAST_TTL)");
                }

                u_char ittl = 255; // IP TTL
                if ( ::setsockopt(_socket, IPPROTO_IP, IP_TTL, &ittl, sizeof(ittl)) < 0 ) {
                        throw SocketException("MulticastSocket: setsockopt(IP_TTL)");
                }
        }

        MulticastSocket::~MulticastSocket()
        {
        }

        bool MulticastSocket::isMulticast(std::string address)
        {
                struct sockaddr_in destination;
                bzero(&destination, sizeof(destination));

                // convert given address
                if (inet_pton(AF_INET, address.c_str(), &destination.sin_addr) <= 0)
                {
                        throw SocketException("MulticastSocket: can not parse address " + address);
                }

                // check if address is in a multicast range
                uint32_t haddr = ntohl((uint32_t&)destination.sin_addr);

                return (IN_MULTICAST(haddr));
        }

        void MulticastSocket::setInterface(const NetInterface &iface)
        {
                struct ip_mreq imr;
                ::memset(&imr, 0, sizeof(ip_mreq));

                // set interface address
                iface.getAddress(&imr.imr_interface);

                if ( ::setsockopt(_socket, IPPROTO_IP, IP_MULTICAST_IF, (char *)&imr.imr_interface.s_addr, sizeof(imr.imr_interface.s_addr)) < 0 )
                {
                        throw SocketException("MulticastSocket: cannot set default interface");
                }
        }

        void MulticastSocket::bind(int port)
        {
                ::memset(&_sockaddr, 0, sizeof(_sockaddr));
                _sockaddr.sin_family = AF_INET;
                _sockaddr.sin_port = htons(port);

                if ( ::bind(_socket, (struct sockaddr *) &_sockaddr, sizeof(_sockaddr)) < 0 )
                {
                        throw SocketException("MulticastSocket: cannot bind socket");
                }
        }

        void MulticastSocket::joinGroup(std::string group)
        {
                struct ip_mreq imr;
                ::memset(&imr, 0, sizeof(ip_mreq));

                if (inet_pton(AF_INET, group.c_str(), &imr.imr_multiaddr) <= 0)
                {
                        throw SocketException("MulticastSocket: can not parse address " + group);
                }

                if ( ::setsockopt(_socket, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const char *)&imr, sizeof(struct ip_mreq)) < 0)
                {
                        throw SocketException("MulticastSocket: setsockopt(IP_ADD_MEMBERSHIP)");
                }
        }

        void MulticastSocket::joinGroup(std::string group, const NetInterface &iface)
        {
                struct ip_mreq imr;
                ::memset(&imr, 0, sizeof(ip_mreq));

                if (inet_pton(AF_INET, group.c_str(), &imr.imr_multiaddr) <= 0)
                {
                        throw SocketException("MulticastSocket: can not parse address " + group);
                }

                // set interface address
                iface.getAddress(&imr.imr_interface);

                if ( ::setsockopt(_socket, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const char *)&imr, sizeof(struct ip_mreq)) < 0)
                {
                        throw SocketException("MulticastSocket: setsockopt(IP_ADD_MEMBERSHIP)");
                }
        }

        void MulticastSocket::leaveGroup(std::string group)
        {
                struct ip_mreq imr;
                ::memset(&imr, 0, sizeof(ip_mreq));

                if (inet_pton(AF_INET, group.c_str(), &imr.imr_multiaddr) <= 0)
                {
                        throw SocketException("MulticastSocket: can not parse address " + group);
                }

                if ( ::setsockopt(_socket, IPPROTO_IP, IP_DROP_MEMBERSHIP, (const char *)&imr, sizeof(struct ip_mreq)) < 0)
                {
                        throw SocketException("MulticastSocket: setsockopt(IP_DROP_MEMBERSHIP)");
                }
        }

        void MulticastSocket::leaveGroup(std::string group, const NetInterface &iface)
        {
                struct ip_mreq imr;
                ::memset(&imr, 0, sizeof(ip_mreq));

                if (inet_pton(AF_INET, group.c_str(), &imr.imr_multiaddr) <= 0)
                {
                        throw SocketException("MulticastSocket: can not parse address " + group);
                }

                // set interface address
                iface.getAddress(&imr.imr_interface);

                if ( ::setsockopt(_socket, IPPROTO_IP, IP_DROP_MEMBERSHIP, (const char *)&imr, sizeof(struct ip_mreq)) < 0)
                {
                        throw SocketException("MulticastSocket: setsockopt(IP_ADD_MEMBERSHIP)");
                }
        }
}