1 files changed, 92 insertions, 0 deletions
diff --git a/pimd/pim_igmp_mtrace.h b/pimd/pim_igmp_mtrace.h
new file mode 100644
index 0000000..bba9c10
--- /dev/null
+++ b/pimd/pim_igmp_mtrace.h
@@ -0,0 +1,92 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Multicast traceroute for FRRouting
+ * Copyright (C) 2017  Mladen Sablic
+ */
+
+#ifndef PIM_IGMP_MTRACE_H
+#define PIM_IGMP_MTRACE_H
+
+#include <zebra.h>
+
+#include "pim_igmp.h"
+
+#define MTRACE_MAX_HOPS (255)
+#define MTRACE_UNKNOWN_COUNT (0xffffffff)
+#define MTRACE_SRC_MASK_GROUP (0x3f) /* forwarding on group state (*,G) */
+#define MTRACE_SRC_MASK_SOURCE (0x20) /* i.e. 32 forwarding on (S,G) */
+
+enum mtrace_fwd_code {
+	MTRACE_FWD_CODE_NO_ERROR = 0x00,
+	MTRACE_FWD_CODE_WRONG_IF = 0x01,
+	MTRACE_FWD_CODE_PRUNE_SENT = 0x02,
+	MTRACE_FWD_CODE_PRUNE_RCVD = 0x03,
+	MTRACE_FWD_CODE_SCOPED = 0x04,
+	MTRACE_FWD_CODE_NO_ROUTE = 0x05,
+	MTRACE_FWD_CODE_WRONG_LAST_HOP = 0x06,
+	MTRACE_FWD_CODE_NOT_FORWARDING = 0x07,
+	MTRACE_FWD_CODE_REACHED_RP = 0x08,
+	MTRACE_FWD_CODE_RPF_IF = 0x09,
+	MTRACE_FWD_CODE_NO_MULTICAST = 0x0A,
+	MTRACE_FWD_CODE_INFO_HIDDEN = 0x0B,
+	MTRACE_FWD_CODE_NO_SPACE = 0x81,
+	MTRACE_FWD_CODE_OLD_ROUTER = 0x82,
+	MTRACE_FWD_CODE_ADMIN_PROHIB = 0x83
+};
+
+enum mtrace_rtg_proto {
+	MTRACE_RTG_PROTO_DVMRP = 1,
+	MTRACE_RTG_PROTO_MOSPF = 2,
+	MTRACE_RTG_PROTO_PIM = 3,
+	MTRACE_RTG_PROTO_CBT = 4,
+	MTRACE_RTG_PROTO_PIM_SPECIAL = 5,
+	MTRACE_RTG_PROTO_PIM_STATIC = 6,
+	MTRACE_RTG_PROTO_DVMRP_STATIC = 7,
+	MTRACE_RTG_PROTO_PIM_MBGP = 8,
+	MTRACE_RTG_PROTO_CBT_SPECIAL = 9,
+	MTRACE_RTG_PROTO_CBT_STATIC = 10,
+	MTRACE_RTG_PROTO_PIM_ASSERT = 11,
+};
+
+struct igmp_mtrace_rsp {
+	uint32_t arrival;
+	struct in_addr incoming;
+	struct in_addr outgoing;
+	struct in_addr prev_hop;
+	uint32_t in_count;
+	uint32_t out_count;
+	uint32_t total;
+	uint32_t rtg_proto : 8;
+	uint32_t fwd_ttl : 8;
+	/* little endian order for next three fields */
+	uint32_t src_mask : 6;
+	uint32_t s : 1;
+	uint32_t mbz : 1;
+	uint32_t fwd_code : 8;
+} __attribute__((packed));
+
+struct igmp_mtrace {
+	uint8_t type;
+	uint8_t hops;
+	uint16_t checksum;
+	struct in_addr grp_addr;
+	struct in_addr src_addr;
+	struct in_addr dst_addr;
+	struct in_addr rsp_addr;
+	uint32_t rsp_ttl : 8;
+	uint32_t qry_id : 24;
+	struct igmp_mtrace_rsp rsp[0];
+} __attribute__((packed));
+
+#define MTRACE_HDR_SIZE (sizeof(struct igmp_mtrace))
+#define MTRACE_RSP_SIZE (sizeof(struct igmp_mtrace_rsp))
+
+int igmp_mtrace_recv_qry_req(struct gm_sock *igmp, struct ip *ip_hdr,
+			     struct in_addr from, const char *from_str,
+			     char *igmp_msg, int igmp_msg_len);
+
+int igmp_mtrace_recv_response(struct gm_sock *igmp, struct ip *ip_hdr,
+			      struct in_addr from, const char *from_str,
+			      char *igmp_msg, int igmp_msg_len);
+
+#endif /* PIM_IGMP_MTRACE_H */