Merge pull request #12798 from donaldsharp/rib_match_multicast

[mirror_frr.git] / pimd / pim_igmp_join.h

diff --git a/pimd/pim_igmp_join.h b/pimd/pim_igmp_join.h
index 228d30c5c415b2d763e7877d61b9c7369b1af32f..d6d22be3e4c73565dcc11f1dfb3ec98751cd0970 100644 (file)
--- a/pimd/pim_igmp_join.h
+++ b/pimd/pim_igmp_join.h
@@ -1,22 +1,8 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
 /*
-  PIM for Quagga
-  Copyright (C) 2008  Everton da Silva Marques
-
-  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.
-  
-  You should have received a copy of the GNU General Public License
-  along with this program; see the file COPYING; if not, write to the
-  Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston,
-  MA 02110-1301 USA
-*/
+ * PIM for Quagga
+ * Copyright (C) 2008  Everton da Silva Marques
+ */
 
 #ifndef PIM_IGMP_JOIN_H
 #define PIM_IGMP_JOIN_H
@@ -27,43 +13,48 @@
 #define SOL_IP IPPROTO_IP
 #endif
 
+#ifndef MCAST_JOIN_GROUP
+#define MCAST_JOIN_GROUP 42
+#endif
+
 #ifndef MCAST_JOIN_SOURCE_GROUP
 #define MCAST_JOIN_SOURCE_GROUP 46
-struct group_source_req
-{
-  uint32_t gsr_interface;
-  struct sockaddr_storage gsr_group;
-  struct sockaddr_storage gsr_source;
+struct group_source_req {
+       uint32_t gsr_interface;
+       struct sockaddr_storage gsr_group;
+       struct sockaddr_storage gsr_source;
 };
 #endif
 
-static int pim_igmp_join_source(int fd, ifindex_t ifindex,
-                               struct in_addr group_addr,
-                               struct in_addr source_addr)
+static inline int pim_igmp_join_source(int fd, ifindex_t ifindex,
+                                      struct in_addr group_addr,
+                                      struct in_addr source_addr)
 {
-  struct group_source_req req;
-  struct sockaddr_in group;
-  struct sockaddr_in source;
-
-  memset(&req, 0, sizeof(req));
-  memset(&group, 0, sizeof(group));
-  group.sin_family = AF_INET;
-  group.sin_addr = group_addr;
-  group.sin_port = htons(0);
-  memcpy(&req.gsr_group, &group, sizeof(struct sockaddr_in));
-
-  memset(&source, 0, sizeof(source));
-  source.sin_family = AF_INET;
-  source.sin_addr = source_addr;
-  source.sin_port = htons(0);
-  memcpy(&req.gsr_source, &source, sizeof(struct sockaddr_in));
-
-  req.gsr_interface = ifindex;
-
-  return setsockopt(fd, SOL_IP, MCAST_JOIN_SOURCE_GROUP,
-                   &req, sizeof(req));
-
-  return 0;
+       struct group_source_req req;
+       struct sockaddr_in group;
+       struct sockaddr_in source;
+
+       memset(&req, 0, sizeof(req));
+       memset(&group, 0, sizeof(group));
+       group.sin_family = AF_INET;
+       group.sin_addr = group_addr;
+       group.sin_port = htons(0);
+       memcpy(&req.gsr_group, &group, sizeof(struct sockaddr_in));
+
+       memset(&source, 0, sizeof(source));
+       source.sin_family = AF_INET;
+       source.sin_addr = source_addr;
+       source.sin_port = htons(0);
+       memcpy(&req.gsr_source, &source, sizeof(struct sockaddr_in));
+
+       req.gsr_interface = ifindex;
+
+       if (source_addr.s_addr == INADDR_ANY)
+               return setsockopt(fd, SOL_IP, MCAST_JOIN_GROUP, &req,
+                                 sizeof(req));
+       else
+               return setsockopt(fd, SOL_IP, MCAST_JOIN_SOURCE_GROUP, &req,
+                                 sizeof(req));
 }
 
 #endif /* PIM_IGMP_JOIN_H */