From 6992225da8c43fc16f5de1a02d218f80e9e60bf1 Mon Sep 17 00:00:00 2001
From: "Matthias P. Braendli" <matthias.braendli@mpb.li>
Date: Mon, 1 Jul 2024 13:19:23 +0200
Subject: Common: 710f5e6, Socket multicast reception

---
 lib/Socket.cpp | 38 ++++++++++++++++++++++++++++++++------
 lib/Socket.h   |  1 +
 2 files changed, 33 insertions(+), 6 deletions(-)

(limited to 'lib')

diff --git a/lib/Socket.cpp b/lib/Socket.cpp
index b71c01e..1281066 100644
--- a/lib/Socket.cpp
+++ b/lib/Socket.cpp
@@ -193,6 +193,34 @@ void UDPSocket::reinit(int port, const std::string& name)
     }
 }
 
+void UDPSocket::init_receive_multicast(int port, const string& local_if_addr, const string& mcastaddr)
+{
+    if (m_sock != INVALID_SOCKET) {
+        ::close(m_sock);
+    }
+
+    m_port = port;
+    m_sock = ::socket(AF_INET, SOCK_DGRAM, 0);
+
+    int reuse_setting = 1;
+    if (setsockopt(m_sock, SOL_SOCKET, SO_REUSEADDR, &reuse_setting, sizeof(reuse_setting)) == -1) {
+        throw runtime_error("Can't reuse address");
+    }
+
+    struct sockaddr_in la;
+    memset((char *) &la, 0, sizeof(la));
+    la.sin_family = AF_INET;
+    la.sin_port = htons(port);
+    la.sin_addr.s_addr = INADDR_ANY;
+    if (::bind(m_sock, (struct sockaddr*)&la, sizeof(la))) {
+        throw runtime_error(string("Could not bind: ") + strerror(errno));
+    }
+
+    joinGroup(mcastaddr.c_str(), local_if_addr.c_str());
+
+}
+
+
 void UDPSocket::close()
 {
     if (m_sock != INVALID_SOCKET) {
@@ -276,7 +304,7 @@ void UDPSocket::joinGroup(const char* groupname, const char* if_addr)
         throw runtime_error("Cannot convert multicast group name");
     }
     if (!IN_MULTICAST(ntohl(group.imr_multiaddr.s_addr))) {
-        throw runtime_error("Group name is not a multicast address");
+        throw runtime_error(string("Group name '") + groupname + "' is not a multicast address");
     }
 
     if (if_addr) {
@@ -309,7 +337,7 @@ void UDPSocket::setMulticastTTL(int ttl)
 {
     if (setsockopt(m_sock, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl))
             == SOCKET_ERROR) {
-        throw runtime_error(string("Can't set multicast ttl") + strerror(errno));
+        throw runtime_error(string("Can't set multicast ttl: ") + strerror(errno));
     }
 }
 
@@ -327,15 +355,13 @@ void UDPReceiver::add_receive_port(int port, const string& bindto, const string&
     UDPSocket sock;
 
     if (IN_MULTICAST(ntohl(inet_addr(mcastaddr.c_str())))) {
-        sock.reinit(port, mcastaddr);
-        sock.setMulticastSource(bindto.c_str());
-        sock.joinGroup(mcastaddr.c_str(), bindto.c_str());
+        sock.init_receive_multicast(port, bindto, mcastaddr);
     }
     else {
         sock.reinit(port, bindto);
     }
 
-    m_sockets.push_back(move(sock));
+    m_sockets.push_back(std::move(sock));
 }
 
 vector<UDPReceiver::ReceivedPacket> UDPReceiver::receive(int timeout_ms)
diff --git a/lib/Socket.h b/lib/Socket.h
index d8242e2..44f93d0 100644
--- a/lib/Socket.h
+++ b/lib/Socket.h
@@ -111,6 +111,7 @@ class UDPSocket
         /** Close the already open socket, and create a new one. Throws a runtime_error on error.  */
         void reinit(int port);
         void reinit(int port, const std::string& name);
+        void init_receive_multicast(int port, const std::string& local_if_addr, const std::string& mcastaddr);
 
         void close(void);
         void send(UDPPacket& packet);
-- 
cgit v1.2.3