Merge pull request #12798 from donaldsharp/rib_match_multicast

[mirror_frr.git] / pimd / pim_msg.h

diff --git a/pimd/pim_msg.h b/pimd/pim_msg.h
index 2d69a4b03af6008c57dc219a67a315c4637e89b8..56923b7ec18ff2d55e3902fba947f1040fe03eef 100644 (file)
--- a/pimd/pim_msg.h
+++ b/pimd/pim_msg.h
@@ -1,26 +1,16 @@
+// 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
  */
 
 #ifndef PIM_MSG_H
 #define PIM_MSG_H
 
 #include <netinet/in.h>
+#if PIM_IPV == 6
+#include <netinet/ip6.h>
+#endif
 
 #include "pim_jp_agg.h"
 
@@ -75,6 +65,12 @@ struct pim_encoded_ipv4_unicast {
        struct in_addr addr;
 } __attribute__((packed));
 
+struct pim_encoded_ipv6_unicast {
+       uint8_t family;
+       uint8_t reserved;
+       struct in6_addr addr;
+} __attribute__((packed));
+
 /*
  *  Encoded Group format. RFC 4601 Sec 4.9.1
  *   0                   1                   2                   3
@@ -103,6 +99,23 @@ struct pim_encoded_group_ipv4 {
        struct in_addr addr;
 } __attribute__((packed));
 
+struct pim_encoded_group_ipv6 {
+       uint8_t family;
+       uint8_t ne;
+#if (BYTE_ORDER == LITTLE_ENDIAN)
+       uint8_t sz : 1;       /* scope zone bit */
+       uint8_t reserved : 6; /* Reserved */
+       uint8_t bidir : 1;    /* Bidir bit */
+#elif (BYTE_ORDER == BIG_ENDIAN)
+       uint8_t bidir : 1;      /* Bidir bit */
+       uint8_t reserved : 6;   /* Reserved */
+       uint8_t sz : 1;         /* scope zone bit */
+#else
+#error "Please set byte order"
+#endif
+       uint8_t mask;
+       struct in6_addr addr;
+} __attribute__((packed));
 
 /*
  *  Encoded Source format. RFC 4601 Sec 4.9.1
@@ -122,24 +135,77 @@ struct pim_encoded_source_ipv4 {
        struct in_addr addr;
 } __attribute__((packed));
 
+struct pim_encoded_source_ipv6 {
+       uint8_t family;
+       uint8_t ne;
+       uint8_t bits;
+       uint8_t mask;
+       struct in6_addr addr;
+} __attribute__((packed));
+
+/* clang-format off */
+#if PIM_IPV == 4
+typedef struct pim_encoded_ipv4_unicast pim_encoded_unicast;
+typedef struct pim_encoded_group_ipv4   pim_encoded_group;
+typedef struct pim_encoded_source_ipv4  pim_encoded_source;
+typedef struct ip                      ipv_hdr;
+#define IPV_SRC(ip_hdr)                        ((ip_hdr))->ip_src
+#define IPV_DST(ip_hdr)                        ((ip_hdr))->ip_dst
+#define IPV_LEN(ip_hdr)                        ((ip_hdr))->ip_len
+#else
+typedef struct pim_encoded_ipv6_unicast pim_encoded_unicast;
+typedef struct pim_encoded_group_ipv6   pim_encoded_group;
+typedef struct pim_encoded_source_ipv6  pim_encoded_source;
+typedef struct ip6_hdr                         ipv_hdr;
+#define IPV_SRC(ip_hdr)                        ((ip_hdr))->ip6_src
+#define IPV_DST(ip_hdr)                        ((ip_hdr))->ip6_dst
+#define IPV_LEN(ip_hdr)                        ((ip_hdr))->ip6_plen
+#endif
+/* clang-format on */
+
 struct pim_jp_groups {
-       struct pim_encoded_group_ipv4 g;
+       pim_encoded_group g;
        uint16_t joins;
        uint16_t prunes;
-       struct pim_encoded_source_ipv4 s[1];
+       pim_encoded_source s[1];
 } __attribute__((packed));
 
 struct pim_jp {
        struct pim_msg_header header;
-       struct pim_encoded_ipv4_unicast addr;
+       pim_encoded_unicast addr;
        uint8_t reserved;
        uint8_t num_groups;
        uint16_t holdtime;
        struct pim_jp_groups groups[1];
 } __attribute__((packed));
 
-void pim_msg_build_header(uint8_t *pim_msg, size_t pim_msg_size,
-                         uint8_t pim_msg_type, bool no_fwd);
+#if PIM_IPV == 4
+static inline pim_sgaddr pim_sgaddr_from_iphdr(const void *iphdr)
+{
+       const struct ip *ipv4_hdr = iphdr;
+       pim_sgaddr sg;
+
+       sg.src = ipv4_hdr->ip_src;
+       sg.grp = ipv4_hdr->ip_dst;
+
+       return sg;
+}
+#else
+static inline pim_sgaddr pim_sgaddr_from_iphdr(const void *iphdr)
+{
+       const struct ip6_hdr *ipv6_hdr = iphdr;
+       pim_sgaddr sg;
+
+       sg.src = ipv6_hdr->ip6_src;
+       sg.grp = ipv6_hdr->ip6_dst;
+
+       return sg;
+}
+#endif
+
+void pim_msg_build_header(pim_addr src, pim_addr dst, uint8_t *pim_msg,
+                         size_t pim_msg_size, uint8_t pim_msg_type,
+                         bool no_fwd);
 uint8_t *pim_msg_addr_encode_ipv4_ucast(uint8_t *buf, struct in_addr addr);
 uint8_t *pim_msg_addr_encode_ipv4_group(uint8_t *buf, struct in_addr addr);
 
@@ -149,6 +215,14 @@ uint8_t *pim_msg_addr_encode_ipv4_group(uint8_t *buf, struct in_addr addr);
 uint8_t *pim_msg_addr_encode_ipv4_source(uint8_t *buf, struct in_addr addr,
                                         uint8_t bits);
 
+uint8_t *pim_msg_addr_encode_ipv6_ucast(uint8_t *buf, struct in6_addr addr);
+uint8_t *pim_msg_addr_encode_ipv6_group(uint8_t *buf, struct in6_addr addr);
+uint8_t *pim_msg_addr_encode_ipv6_source(uint8_t *buf, struct in6_addr addr,
+                                        uint8_t bits);
+
+uint8_t *pim_msg_addr_encode_ucast(uint8_t *buf, pim_addr addr);
+uint8_t *pim_msg_addr_encode_group(uint8_t *buf, pim_addr addr);
+uint8_t *pim_msg_addr_encode_source(uint8_t *buf, pim_addr addr, uint8_t bits);
 
 size_t pim_msg_get_jp_group_size(struct list *sources);
 size_t pim_msg_build_jp_groups(struct pim_jp_groups *grp,