xref: /freebsd/usr.sbin/ifmcstat/ifmcstat.c (revision db33c6f3ae9d1231087710068ee4ea5398aacca7)
1 /*	$KAME: ifmcstat.c,v 1.48 2006/11/15 05:13:59 itojun Exp $	*/
2 
3 /*-
4  * SPDX-License-Identifier: BSD-3-Clause
5  *
6  * Copyright (c) 2007-2009 Bruce Simpson.
7  * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
8  * All rights reserved.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. Neither the name of the project nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  */
34 
35 #include <sys/types.h>
36 #include <sys/param.h>
37 #include <sys/sysctl.h>
38 #include <sys/socket.h>
39 #include <sys/queue.h>
40 #include <sys/tree.h>
41 
42 #include <net/if.h>
43 #include <net/if_types.h>
44 #include <net/if_dl.h>
45 #include <net/route.h>
46 
47 #include <netinet/in.h>
48 #include <netinet/in_var.h>
49 #include <netinet/in_systm.h>
50 #include <netinet/ip.h>
51 #include <netinet/igmp.h>
52 #include <netinet/if_ether.h>
53 #include <netinet/igmp_var.h>
54 
55 #ifdef INET6
56 #include <netinet/icmp6.h>
57 #include <netinet6/mld6_var.h>
58 #endif /* INET6 */
59 
60 #include <arpa/inet.h>
61 #include <netdb.h>
62 
63 #include <stddef.h>
64 #include <stdarg.h>
65 #include <stdint.h>
66 #include <stdio.h>
67 #include <stdlib.h>
68 #include <string.h>
69 
70 #include <ctype.h>
71 #include <err.h>
72 #include <errno.h>
73 #include <fcntl.h>
74 #include <limits.h>
75 #include <ifaddrs.h>
76 #include <sysexits.h>
77 #include <unistd.h>
78 
79 #ifdef KVM
80 /*
81  * Currently the KVM build is broken. To be fixed it requires uncovering
82  * large amount of _KERNEL code in include files, and it is also very
83  * tentative to internal kernel ABI changes. If anyone wishes to restore
84  * it, please move it out of src/usr.sbin to src/tools/tools.
85  */
86 #include <kvm.h>
87 #include <nlist.h>
88 #endif
89 
90 /* XXX: This file currently assumes INET support in the base system. */
91 #ifndef INET
92 #define INET
93 #endif
94 
95 extern void	printb(const char *, unsigned int, const char *);
96 
97 union sockunion {
98 	struct sockaddr_storage	ss;
99 	struct sockaddr		sa;
100 	struct sockaddr_dl	sdl;
101 #ifdef INET
102 	struct sockaddr_in	sin;
103 #endif
104 #ifdef INET6
105 	struct sockaddr_in6	sin6;
106 #endif
107 };
108 typedef union sockunion sockunion_t;
109 
110 uint32_t	ifindex = 0;
111 int		af = AF_UNSPEC;
112 #ifdef WITH_KVM
113 int		Kflag = 0;
114 #endif
115 int		vflag = 0;
116 
117 #define	sa_dl_equal(a1, a2)	\
118 	((((struct sockaddr_dl *)(a1))->sdl_len ==			\
119 	 ((struct sockaddr_dl *)(a2))->sdl_len) &&			\
120 	 (bcmp(LLADDR((struct sockaddr_dl *)(a1)),			\
121 	       LLADDR((struct sockaddr_dl *)(a2)),			\
122 	       ((struct sockaddr_dl *)(a1))->sdl_alen) == 0))
123 
124 /*
125  * Most of the code in this utility is to support the use of KVM for
126  * post-mortem debugging of the multicast code.
127  */
128 #ifdef WITH_KVM
129 
130 #ifdef INET
131 static void		if_addrlist(struct ifaddr *);
132 static struct in_multi *
133 			in_multientry(struct in_multi *);
134 #endif /* INET */
135 
136 #ifdef INET6
137 static void		if6_addrlist(struct ifaddr *);
138 static struct in6_multi *
139 			in6_multientry(struct in6_multi *);
140 #endif /* INET6 */
141 
142 static void		kread(u_long, void *, int);
143 static void		ll_addrlist(struct ifaddr *);
144 
145 static int		ifmcstat_kvm(const char *kernel, const char *core);
146 
147 #define	KREAD(addr, buf, type) \
148 	kread((u_long)addr, (void *)buf, sizeof(type))
149 
150 kvm_t	*kvmd;
151 struct	nlist nl[] = {
152 	{ "_ifnet", 0, 0, 0, 0, },
153 	{ "", 0, 0, 0, 0, },
154 };
155 #define	N_IFNET	0
156 
157 #endif /* WITH_KVM */
158 
159 static int		ifmcstat_getifmaddrs(void);
160 #ifdef INET
161 static void		in_ifinfo(struct igmp_ifinfo *);
162 static const char *	inm_mode(u_int mode);
163 #endif
164 #ifdef INET6
165 static void		in6_ifinfo(struct mld_ifinfo *);
166 static const char *	inet6_n2a(struct in6_addr *, uint32_t);
167 #endif
168 int			main(int, char **);
169 
170 static void
171 usage()
172 {
173 
174 	fprintf(stderr,
175 	    "usage: ifmcstat [-i interface] [-f address family]"
176 	    " [-v]"
177 #ifdef WITH_KVM
178 	    " [-K] [-M core] [-N system]"
179 #endif
180 	    "\n");
181 	exit(EX_USAGE);
182 }
183 
184 static const char *options = "i:f:vM:N:"
185 #ifdef WITH_KVM
186 	"K"
187 #endif
188 	;
189 
190 int
191 main(int argc, char **argv)
192 {
193 	int c, error;
194 #ifdef WITH_KVM
195 	const char *kernel = NULL;
196 	const char *core = NULL;
197 #endif
198 
199 	while ((c = getopt(argc, argv, options)) != -1) {
200 		switch (c) {
201 		case 'i':
202 			if ((ifindex = if_nametoindex(optarg)) == 0) {
203 				fprintf(stderr, "%s: unknown interface\n",
204 				    optarg);
205 				exit(EX_NOHOST);
206 			}
207 			break;
208 
209 		case 'f':
210 #ifdef INET
211 			if (strcmp(optarg, "inet") == 0) {
212 				af = AF_INET;
213 				break;
214 			}
215 #endif
216 #ifdef INET6
217 			if (strcmp(optarg, "inet6") == 0) {
218 				af = AF_INET6;
219 				break;
220 			}
221 #endif
222 			if (strcmp(optarg, "link") == 0) {
223 				af = AF_LINK;
224 				break;
225 			}
226 			fprintf(stderr, "%s: unknown address family\n", optarg);
227 			exit(EX_USAGE);
228 			/*NOTREACHED*/
229 			break;
230 
231 #ifdef WITH_KVM
232 		case 'K':
233 			++Kflag;
234 			break;
235 #endif
236 
237 		case 'v':
238 			++vflag;
239 			break;
240 
241 #ifdef WITH_KVM
242 		case 'M':
243 			core = strdup(optarg);
244 			break;
245 
246 		case 'N':
247 			kernel = strdup(optarg);
248 			break;
249 #endif
250 
251 		default:
252 			usage();
253 			break;
254 			/*NOTREACHED*/
255 		}
256 	}
257 
258 	if (af == AF_LINK && vflag)
259 		usage();
260 
261 #ifdef WITH_KVM
262 	if (Kflag)
263 		error = ifmcstat_kvm(kernel, core);
264 	/*
265 	 * If KVM failed, and user did not explicitly specify a core file,
266 	 * or force KVM backend to be disabled, try the sysctl backend.
267 	 */
268 	if (!Kflag || (error != 0 && (core == NULL && kernel == NULL)))
269 #endif
270 	error = ifmcstat_getifmaddrs();
271 	if (error != 0)
272 		exit(EX_OSERR);
273 
274 	exit(EX_OK);
275 	/*NOTREACHED*/
276 }
277 
278 #ifdef INET
279 
280 static void
281 in_ifinfo(struct igmp_ifinfo *igi)
282 {
283 
284 	printf("\t");
285 	switch (igi->igi_version) {
286 	case IGMP_VERSION_1:
287 	case IGMP_VERSION_2:
288 	case IGMP_VERSION_3:
289 		printf("igmpv%d", igi->igi_version);
290 		break;
291 	default:
292 		printf("igmpv?(%d)", igi->igi_version);
293 		break;
294 	}
295 	if (igi->igi_flags)
296 		printb(" flags", igi->igi_flags, "\020\1SILENT\2LOOPBACK");
297 	if (igi->igi_version == IGMP_VERSION_3) {
298 		printf(" rv %u qi %u qri %u uri %u",
299 		    igi->igi_rv, igi->igi_qi, igi->igi_qri, igi->igi_uri);
300 	}
301 	if (vflag >= 2) {
302 		printf(" v1timer %u v2timer %u v3timer %u",
303 		    igi->igi_v1_timer, igi->igi_v2_timer, igi->igi_v3_timer);
304 	}
305 	printf("\n");
306 }
307 
308 static const char *inm_modes[] = {
309 	"undefined",
310 	"include",
311 	"exclude",
312 };
313 
314 static const char *
315 inm_mode(u_int mode)
316 {
317 
318 	if (mode >= MCAST_UNDEFINED && mode <= MCAST_EXCLUDE)
319 		return (inm_modes[mode]);
320 	return (NULL);
321 }
322 
323 #endif /* INET */
324 
325 #ifdef WITH_KVM
326 
327 static int
328 ifmcstat_kvm(const char *kernel, const char *core)
329 {
330 	char	buf[_POSIX2_LINE_MAX], ifname[IFNAMSIZ];
331 	struct	ifnet	*ifp, *nifp, ifnet;
332 
333 	if ((kvmd = kvm_openfiles(kernel, core, NULL, O_RDONLY, buf)) ==
334 	    NULL) {
335 		perror("kvm_openfiles");
336 		return (-1);
337 	}
338 	if (kvm_nlist(kvmd, nl) < 0) {
339 		perror("kvm_nlist");
340 		return (-1);
341 	}
342 	if (nl[N_IFNET].n_value == 0) {
343 		printf("symbol %s not found\n", nl[N_IFNET].n_name);
344 		return (-1);
345 	}
346 	KREAD(nl[N_IFNET].n_value, &ifp, struct ifnet *);
347 	while (ifp) {
348 		KREAD(ifp, &ifnet, struct ifnet);
349 		nifp = ifnet.if_link.tqe_next;
350 		if (ifindex && ifindex != ifnet.if_index)
351 			goto next;
352 
353 		printf("%s:\n", if_indextoname(ifnet.if_index, ifname));
354 #ifdef INET
355 		if_addrlist(TAILQ_FIRST(&ifnet.if_addrhead));
356 #endif
357 #ifdef INET6
358 		if6_addrlist(TAILQ_FIRST(&ifnet.if_addrhead));
359 #endif
360 		if (vflag)
361 			ll_addrlist(TAILQ_FIRST(&ifnet.if_addrhead));
362 	next:
363 		ifp = nifp;
364 	}
365 
366 	return (0);
367 }
368 
369 static void
370 kread(u_long addr, void *buf, int len)
371 {
372 
373 	if (kvm_read(kvmd, addr, buf, len) != len) {
374 		perror("kvm_read");
375 		exit(EX_OSERR);
376 	}
377 }
378 
379 static void
380 ll_addrlist(struct ifaddr *ifap)
381 {
382 	char addrbuf[NI_MAXHOST];
383 	struct ifaddr ifa;
384 	struct sockaddr sa;
385 	struct sockaddr_dl sdl;
386 	struct ifaddr *ifap0;
387 
388 	if (af && af != AF_LINK)
389 		return;
390 
391 	ifap0 = ifap;
392 	while (ifap) {
393 		KREAD(ifap, &ifa, struct ifaddr);
394 		if (ifa.ifa_addr == NULL)
395 			goto nextifap;
396 		KREAD(ifa.ifa_addr, &sa, struct sockaddr);
397 		if (sa.sa_family != PF_LINK)
398 			goto nextifap;
399 		KREAD(ifa.ifa_addr, &sdl, struct sockaddr_dl);
400 		if (sdl.sdl_alen == 0)
401 			goto nextifap;
402 		addrbuf[0] = '\0';
403 		getnameinfo((struct sockaddr *)&sdl, sdl.sdl_len,
404 		    addrbuf, sizeof(addrbuf), NULL, 0, NI_NUMERICHOST);
405 		printf("\tlink %s\n", addrbuf);
406 	nextifap:
407 		ifap = ifa.ifa_link.tqe_next;
408 	}
409 	if (ifap0) {
410 		struct ifnet ifnet;
411 		struct ifmultiaddr ifm, *ifmp = 0;
412 
413 		KREAD(ifap0, &ifa, struct ifaddr);
414 		KREAD(ifa.ifa_ifp, &ifnet, struct ifnet);
415 		if (TAILQ_FIRST(&ifnet.if_multiaddrs))
416 			ifmp = TAILQ_FIRST(&ifnet.if_multiaddrs);
417 		while (ifmp) {
418 			KREAD(ifmp, &ifm, struct ifmultiaddr);
419 			if (ifm.ifma_addr == NULL)
420 				goto nextmulti;
421 			KREAD(ifm.ifma_addr, &sa, struct sockaddr);
422 			if (sa.sa_family != AF_LINK)
423 				goto nextmulti;
424 			KREAD(ifm.ifma_addr, &sdl, struct sockaddr_dl);
425 			addrbuf[0] = '\0';
426 			getnameinfo((struct sockaddr *)&sdl,
427 			    sdl.sdl_len, addrbuf, sizeof(addrbuf),
428 			    NULL, 0, NI_NUMERICHOST);
429 			printf("\t\tgroup %s refcnt %d\n",
430 			    addrbuf, ifm.ifma_refcount);
431 		nextmulti:
432 			ifmp = TAILQ_NEXT(&ifm, ifma_link);
433 		}
434 	}
435 }
436 
437 #ifdef INET6
438 
439 static void
440 if6_addrlist(struct ifaddr *ifap)
441 {
442 	struct ifnet ifnet;
443 	struct ifaddr ifa;
444 	struct sockaddr sa;
445 	struct in6_ifaddr if6a;
446 	struct ifaddr *ifap0;
447 
448 	if (af && af != AF_INET6)
449 		return;
450 	ifap0 = ifap;
451 	while (ifap) {
452 		KREAD(ifap, &ifa, struct ifaddr);
453 		if (ifa.ifa_addr == NULL)
454 			goto nextifap;
455 		KREAD(ifa.ifa_addr, &sa, struct sockaddr);
456 		if (sa.sa_family != PF_INET6)
457 			goto nextifap;
458 		KREAD(ifap, &if6a, struct in6_ifaddr);
459 		printf("\tinet6 %s\n", inet6_n2a(&if6a.ia_addr.sin6_addr,
460 		    if6a.ia_addr.sin6_scope_id));
461 		/*
462 		 * Print per-link MLD information, if available.
463 		 */
464 		if (ifa.ifa_ifp != NULL) {
465 			struct in6_ifextra ie;
466 			struct mld_ifinfo mli;
467 
468 			KREAD(ifa.ifa_ifp, &ifnet, struct ifnet);
469 			KREAD(ifnet.if_afdata[AF_INET6], &ie,
470 			    struct in6_ifextra);
471 			if (ie.mld_ifinfo != NULL) {
472 				KREAD(ie.mld_ifinfo, &mli, struct mld_ifinfo);
473 				in6_ifinfo(&mli);
474 			}
475 		}
476 	nextifap:
477 		ifap = ifa.ifa_link.tqe_next;
478 	}
479 	if (ifap0) {
480 		struct ifnet ifnet;
481 		struct ifmultiaddr ifm, *ifmp = 0;
482 		struct sockaddr_dl sdl;
483 
484 		KREAD(ifap0, &ifa, struct ifaddr);
485 		KREAD(ifa.ifa_ifp, &ifnet, struct ifnet);
486 		if (TAILQ_FIRST(&ifnet.if_multiaddrs))
487 			ifmp = TAILQ_FIRST(&ifnet.if_multiaddrs);
488 		while (ifmp) {
489 			KREAD(ifmp, &ifm, struct ifmultiaddr);
490 			if (ifm.ifma_addr == NULL)
491 				goto nextmulti;
492 			KREAD(ifm.ifma_addr, &sa, struct sockaddr);
493 			if (sa.sa_family != AF_INET6)
494 				goto nextmulti;
495 			(void)in6_multientry((struct in6_multi *)
496 					     ifm.ifma_protospec);
497 			if (ifm.ifma_lladdr == 0)
498 				goto nextmulti;
499 			KREAD(ifm.ifma_lladdr, &sdl, struct sockaddr_dl);
500 			printf("\t\t\tmcast-macaddr %s refcnt %d\n",
501 			       ether_ntoa((struct ether_addr *)LLADDR(&sdl)),
502 			       ifm.ifma_refcount);
503 		    nextmulti:
504 			ifmp = TAILQ_NEXT(&ifm, ifma_link);
505 		}
506 	}
507 }
508 
509 static struct in6_multi *
510 in6_multientry(struct in6_multi *mc)
511 {
512 	struct in6_multi multi;
513 
514 	KREAD(mc, &multi, struct in6_multi);
515 	printf("\t\tgroup %s", inet6_n2a(&multi.in6m_addr, 0));
516 	printf(" refcnt %u\n", multi.in6m_refcount);
517 
518 	return (multi.in6m_entry.le_next);
519 }
520 
521 #endif /* INET6 */
522 
523 #ifdef INET
524 
525 static void
526 if_addrlist(struct ifaddr *ifap)
527 {
528 	struct ifaddr ifa;
529 	struct ifnet ifnet;
530 	struct sockaddr sa;
531 	struct in_ifaddr ia;
532 	struct ifaddr *ifap0;
533 
534 	if (af && af != AF_INET)
535 		return;
536 	ifap0 = ifap;
537 	while (ifap) {
538 		KREAD(ifap, &ifa, struct ifaddr);
539 		if (ifa.ifa_addr == NULL)
540 			goto nextifap;
541 		KREAD(ifa.ifa_addr, &sa, struct sockaddr);
542 		if (sa.sa_family != PF_INET)
543 			goto nextifap;
544 		KREAD(ifap, &ia, struct in_ifaddr);
545 		printf("\tinet %s\n", inet_ntoa(ia.ia_addr.sin_addr));
546 		/*
547 		 * Print per-link IGMP information, if available.
548 		 */
549 		if (ifa.ifa_ifp != NULL) {
550 			struct in_ifinfo ii;
551 			struct igmp_ifinfo igi;
552 
553 			KREAD(ifa.ifa_ifp, &ifnet, struct ifnet);
554 			KREAD(ifnet.if_afdata[AF_INET], &ii, struct in_ifinfo);
555 			if (ii.ii_igmp != NULL) {
556 				KREAD(ii.ii_igmp, &igi, struct igmp_ifinfo);
557 				in_ifinfo(&igi);
558 			}
559 		}
560 	nextifap:
561 		ifap = ifa.ifa_link.tqe_next;
562 	}
563 	if (ifap0) {
564 		struct ifmultiaddr ifm, *ifmp = 0;
565 		struct sockaddr_dl sdl;
566 
567 		KREAD(ifap0, &ifa, struct ifaddr);
568 		KREAD(ifa.ifa_ifp, &ifnet, struct ifnet);
569 		if (TAILQ_FIRST(&ifnet.if_multiaddrs))
570 			ifmp = TAILQ_FIRST(&ifnet.if_multiaddrs);
571 		while (ifmp) {
572 			KREAD(ifmp, &ifm, struct ifmultiaddr);
573 			if (ifm.ifma_addr == NULL)
574 				goto nextmulti;
575 			KREAD(ifm.ifma_addr, &sa, struct sockaddr);
576 			if (sa.sa_family != AF_INET)
577 				goto nextmulti;
578 			(void)in_multientry((struct in_multi *)
579 					    ifm.ifma_protospec);
580 			if (ifm.ifma_lladdr == 0)
581 				goto nextmulti;
582 			KREAD(ifm.ifma_lladdr, &sdl, struct sockaddr_dl);
583 			printf("\t\t\tmcast-macaddr %s refcnt %d\n",
584 			       ether_ntoa((struct ether_addr *)LLADDR(&sdl)),
585 			       ifm.ifma_refcount);
586 		    nextmulti:
587 			ifmp = TAILQ_NEXT(&ifm, ifma_link);
588 		}
589 	}
590 }
591 
592 static const char *inm_states[] = {
593 	"not-member",
594 	"silent",
595 	"idle",
596 	"lazy",
597 	"sleeping",
598 	"awakening",
599 	"query-pending",
600 	"sg-query-pending",
601 	"leaving"
602 };
603 
604 static const char *
605 inm_state(u_int state)
606 {
607 
608 	if (state >= IGMP_NOT_MEMBER && state <= IGMP_LEAVING_MEMBER)
609 		return (inm_states[state]);
610 	return (NULL);
611 }
612 
613 #if 0
614 static struct ip_msource *
615 ims_min_kvm(struct in_multi *pinm)
616 {
617 	struct ip_msource ims0;
618 	struct ip_msource *tmp, *parent;
619 
620 	parent = NULL;
621 	tmp = RB_ROOT(&pinm->inm_srcs);
622 	while (tmp) {
623 		parent = tmp;
624 		KREAD(tmp, &ims0, struct ip_msource);
625 		tmp = RB_LEFT(&ims0, ims_link);
626 	}
627 	return (parent); /* kva */
628 }
629 
630 /* XXX This routine is buggy. See RB_NEXT in sys/tree.h. */
631 static struct ip_msource *
632 ims_next_kvm(struct ip_msource *ims)
633 {
634 	struct ip_msource ims0, ims1;
635 	struct ip_msource *tmp;
636 
637 	KREAD(ims, &ims0, struct ip_msource);
638 	if (RB_RIGHT(&ims0, ims_link)) {
639 		ims = RB_RIGHT(&ims0, ims_link);
640 		KREAD(ims, &ims1, struct ip_msource);
641 		while ((tmp = RB_LEFT(&ims1, ims_link))) {
642 			KREAD(tmp, &ims0, struct ip_msource);
643 			ims = RB_LEFT(&ims0, ims_link);
644 		}
645 	} else {
646 		tmp = RB_PARENT(&ims0, ims_link);
647 		if (tmp) {
648 			KREAD(tmp, &ims1, struct ip_msource);
649 			if (ims == RB_LEFT(&ims1, ims_link))
650 				ims = tmp;
651 		} else {
652 			while ((tmp = RB_PARENT(&ims0, ims_link))) {
653 				KREAD(tmp, &ims1, struct ip_msource);
654 				if (ims == RB_RIGHT(&ims1, ims_link)) {
655 					ims = tmp;
656 					KREAD(ims, &ims0, struct ip_msource);
657 				} else
658 					break;
659 			}
660 			ims = RB_PARENT(&ims0, ims_link);
661 		}
662 	}
663 	return (ims); /* kva */
664 }
665 
666 static void
667 inm_print_sources_kvm(struct in_multi *pinm)
668 {
669 	struct ip_msource ims0;
670 	struct ip_msource *ims;
671 	struct in_addr src;
672 	int cnt;
673 	uint8_t fmode;
674 
675 	cnt = 0;
676 	fmode = pinm->inm_st[1].iss_fmode;
677 	if (fmode == MCAST_UNDEFINED)
678 		return;
679 	for (ims = ims_min_kvm(pinm); ims != NULL; ims = ims_next_kvm(ims)) {
680 		if (cnt == 0)
681 			printf(" srcs ");
682 		KREAD(ims, &ims0, struct ip_msource);
683 		/* Only print sources in-mode at t1. */
684 		if (fmode != ims_get_mode(pinm, ims, 1))
685 			continue;
686 		src.s_addr = htonl(ims0.ims_haddr);
687 		printf("%s%s", (cnt++ == 0 ? "" : ","), inet_ntoa(src));
688 	}
689 }
690 #endif
691 
692 static struct in_multi *
693 in_multientry(struct in_multi *pinm)
694 {
695 	struct in_multi inm;
696 	const char *state, *mode;
697 
698 	KREAD(pinm, &inm, struct in_multi);
699 	printf("\t\tgroup %s", inet_ntoa(inm.inm_addr));
700 	printf(" refcnt %u", inm.inm_refcount);
701 
702 	state = inm_state(inm.inm_state);
703 	if (state)
704 		printf(" state %s", state);
705 	else
706 		printf(" state (%d)", inm.inm_state);
707 
708 	mode = inm_mode(inm.inm_st[1].iss_fmode);
709 	if (mode)
710 		printf(" mode %s", mode);
711 	else
712 		printf(" mode (%d)", inm.inm_st[1].iss_fmode);
713 
714 	if (vflag >= 2) {
715 		printf(" asm %u ex %u in %u rec %u",
716 		    (u_int)inm.inm_st[1].iss_asm,
717 		    (u_int)inm.inm_st[1].iss_ex,
718 		    (u_int)inm.inm_st[1].iss_in,
719 		    (u_int)inm.inm_st[1].iss_rec);
720 	}
721 
722 #if 0
723 	/* Buggy. */
724 	if (vflag)
725 		inm_print_sources_kvm(&inm);
726 #endif
727 
728 	printf("\n");
729 	return (NULL);
730 }
731 
732 #endif /* INET */
733 
734 #endif /* WITH_KVM */
735 
736 #ifdef INET6
737 
738 static void
739 in6_ifinfo(struct mld_ifinfo *mli)
740 {
741 
742 	printf("\t");
743 	switch (mli->mli_version) {
744 	case MLD_VERSION_1:
745 	case MLD_VERSION_2:
746 		printf("mldv%d", mli->mli_version);
747 		break;
748 	default:
749 		printf("mldv?(%d)", mli->mli_version);
750 		break;
751 	}
752 	if (mli->mli_flags)
753 		printb(" flags", mli->mli_flags, "\020\1SILENT\2USEALLOW");
754 	if (mli->mli_version == MLD_VERSION_2) {
755 		printf(" rv %u qi %u qri %u uri %u",
756 		    mli->mli_rv, mli->mli_qi, mli->mli_qri, mli->mli_uri);
757 	}
758 	if (vflag >= 2) {
759 		printf(" v1timer %u v2timer %u", mli->mli_v1_timer,
760 		   mli->mli_v2_timer);
761 	}
762 	printf("\n");
763 }
764 
765 static const char *
766 inet6_n2a(struct in6_addr *p, uint32_t scope_id)
767 {
768 	static char buf[NI_MAXHOST];
769 	struct sockaddr_in6 sin6;
770 	const int niflags = NI_NUMERICHOST;
771 
772 	memset(&sin6, 0, sizeof(sin6));
773 	sin6.sin6_family = AF_INET6;
774 	sin6.sin6_len = sizeof(struct sockaddr_in6);
775 	sin6.sin6_addr = *p;
776 	sin6.sin6_scope_id = scope_id;
777 	if (getnameinfo((struct sockaddr *)&sin6, sin6.sin6_len,
778 	    buf, sizeof(buf), NULL, 0, niflags) == 0) {
779 		return (buf);
780 	} else {
781 		return ("(invalid)");
782 	}
783 }
784 #endif /* INET6 */
785 
786 #ifdef INET
787 /*
788  * Retrieve per-group source filter mode and lists via sysctl.
789  */
790 static void
791 inm_print_sources_sysctl(uint32_t ifindex, struct in_addr gina)
792 {
793 #define	MAX_SYSCTL_TRY	5
794 	int mib[7];
795 	int ntry = 0;
796 	size_t mibsize;
797 	size_t len;
798 	size_t needed;
799 	size_t cnt;
800 	int i;
801 	char *buf;
802 	struct in_addr *pina;
803 	uint32_t *p;
804 	uint32_t fmode;
805 	const char *modestr;
806 
807 	mibsize = nitems(mib);
808 	if (sysctlnametomib("net.inet.ip.mcast.filters", mib, &mibsize) == -1) {
809 		perror("sysctlnametomib");
810 		return;
811 	}
812 
813 	needed = 0;
814 	mib[5] = ifindex;
815 	mib[6] = gina.s_addr;	/* 32 bits wide */
816 	mibsize = nitems(mib);
817 	do {
818 		if (sysctl(mib, mibsize, NULL, &needed, NULL, 0) == -1) {
819 			perror("sysctl net.inet.ip.mcast.filters");
820 			return;
821 		}
822 		if ((buf = malloc(needed)) == NULL) {
823 			perror("malloc");
824 			return;
825 		}
826 		if (sysctl(mib, mibsize, buf, &needed, NULL, 0) == -1) {
827 			if (errno != ENOMEM || ++ntry >= MAX_SYSCTL_TRY) {
828 				perror("sysctl");
829 				goto out_free;
830 			}
831 			free(buf);
832 			buf = NULL;
833 		}
834 	} while (buf == NULL);
835 
836 	len = needed;
837 	if (len < sizeof(uint32_t)) {
838 		perror("sysctl");
839 		goto out_free;
840 	}
841 
842 	p = (uint32_t *)buf;
843 	fmode = *p++;
844 	len -= sizeof(uint32_t);
845 
846 	modestr = inm_mode(fmode);
847 	if (modestr)
848 		printf(" mode %s", modestr);
849 	else
850 		printf(" mode (%u)", fmode);
851 
852 	if (vflag == 0)
853 		goto out_free;
854 
855 	cnt = len / sizeof(struct in_addr);
856 	pina = (struct in_addr *)p;
857 
858 	for (i = 0; i < cnt; i++) {
859 		if (i == 0)
860 			printf(" srcs ");
861 		fprintf(stdout, "%s%s", (i == 0 ? "" : ","),
862 		    inet_ntoa(*pina++));
863 		len -= sizeof(struct in_addr);
864 	}
865 	if (len > 0) {
866 		fprintf(stderr, "warning: %u trailing bytes from %s\n",
867 		    (unsigned int)len, "net.inet.ip.mcast.filters");
868 	}
869 
870 out_free:
871 	free(buf);
872 #undef	MAX_SYSCTL_TRY
873 }
874 
875 #endif /* INET */
876 
877 #ifdef INET6
878 /*
879  * Retrieve MLD per-group source filter mode and lists via sysctl.
880  *
881  * Note: The 128-bit IPv6 group address needs to be segmented into
882  * 32-bit pieces for marshaling to sysctl. So the MIB name ends
883  * up looking like this:
884  *  a.b.c.d.e.ifindex.g[0].g[1].g[2].g[3]
885  * Assumes that pgroup originated from the kernel, so its components
886  * are already in network-byte order.
887  */
888 static void
889 in6m_print_sources_sysctl(uint32_t ifindex, struct in6_addr *pgroup)
890 {
891 #define	MAX_SYSCTL_TRY	5
892 	char addrbuf[INET6_ADDRSTRLEN];
893 	int mib[10];
894 	int ntry = 0;
895 	int *pi;
896 	size_t mibsize;
897 	size_t len;
898 	size_t needed;
899 	size_t cnt;
900 	int i;
901 	char *buf;
902 	struct in6_addr *pina;
903 	uint32_t *p;
904 	uint32_t fmode;
905 	const char *modestr;
906 
907 	mibsize = nitems(mib);
908 	if (sysctlnametomib("net.inet6.ip6.mcast.filters", mib,
909 	    &mibsize) == -1) {
910 		perror("sysctlnametomib");
911 		return;
912 	}
913 
914 	needed = 0;
915 	mib[5] = ifindex;
916 	pi = (int *)pgroup;
917 	for (i = 0; i < 4; i++)
918 		mib[6 + i] = *pi++;
919 
920 	mibsize = nitems(mib);
921 	do {
922 		if (sysctl(mib, mibsize, NULL, &needed, NULL, 0) == -1) {
923 			perror("sysctl net.inet6.ip6.mcast.filters");
924 			return;
925 		}
926 		if ((buf = malloc(needed)) == NULL) {
927 			perror("malloc");
928 			return;
929 		}
930 		if (sysctl(mib, mibsize, buf, &needed, NULL, 0) == -1) {
931 			if (errno != ENOMEM || ++ntry >= MAX_SYSCTL_TRY) {
932 				perror("sysctl");
933 				goto out_free;
934 			}
935 			free(buf);
936 			buf = NULL;
937 		}
938 	} while (buf == NULL);
939 
940 	len = needed;
941 	if (len < sizeof(uint32_t)) {
942 		perror("sysctl");
943 		goto out_free;
944 	}
945 
946 	p = (uint32_t *)buf;
947 	fmode = *p++;
948 	len -= sizeof(uint32_t);
949 
950 	modestr = inm_mode(fmode);
951 	if (modestr)
952 		printf(" mode %s", modestr);
953 	else
954 		printf(" mode (%u)", fmode);
955 
956 	if (vflag == 0)
957 		goto out_free;
958 
959 	cnt = len / sizeof(struct in6_addr);
960 	pina = (struct in6_addr *)p;
961 
962 	for (i = 0; i < cnt; i++) {
963 		if (i == 0)
964 			printf(" srcs ");
965 		inet_ntop(AF_INET6, (const char *)pina++, addrbuf,
966 		    INET6_ADDRSTRLEN);
967 		fprintf(stdout, "%s%s", (i == 0 ? "" : ","), addrbuf);
968 		len -= sizeof(struct in6_addr);
969 	}
970 	if (len > 0) {
971 		fprintf(stderr, "warning: %u trailing bytes from %s\n",
972 		    (unsigned int)len, "net.inet6.ip6.mcast.filters");
973 	}
974 
975 out_free:
976 	free(buf);
977 #undef	MAX_SYSCTL_TRY
978 }
979 #endif /* INET6 */
980 
981 static int
982 ifmcstat_getifmaddrs(void)
983 {
984 	char			 thisifname[IFNAMSIZ];
985 	char			 addrbuf[NI_MAXHOST];
986 	struct ifaddrs		*ifap, *ifa;
987 	struct ifmaddrs		*ifmap, *ifma;
988 	sockunion_t		 lastifasa;
989 	sockunion_t		*psa, *pgsa, *pllsa, *pifasa;
990 	char			*pcolon;
991 	char			*pafname;
992 	uint32_t		 lastifindex, thisifindex;
993 	int			 error;
994 
995 	error = 0;
996 	ifap = NULL;
997 	ifmap = NULL;
998 	lastifindex = 0;
999 	thisifindex = 0;
1000 	lastifasa.ss.ss_family = AF_UNSPEC;
1001 
1002 	if (getifaddrs(&ifap) != 0) {
1003 		warn("getifmaddrs");
1004 		return (-1);
1005 	}
1006 
1007 	if (getifmaddrs(&ifmap) != 0) {
1008 		warn("getifmaddrs");
1009 		error = -1;
1010 		goto out;
1011 	}
1012 
1013 	for (ifma = ifmap; ifma; ifma = ifma->ifma_next) {
1014 		error = 0;
1015 		if (ifma->ifma_name == NULL || ifma->ifma_addr == NULL)
1016 			continue;
1017 
1018 		psa = (sockunion_t *)ifma->ifma_name;
1019 		if (psa->sa.sa_family != AF_LINK) {
1020 			fprintf(stderr,
1021 			    "WARNING: Kernel returned invalid data.\n");
1022 			error = -1;
1023 			break;
1024 		}
1025 
1026 		/* Filter on interface name. */
1027 		thisifindex = psa->sdl.sdl_index;
1028 		if (ifindex != 0 && thisifindex != ifindex)
1029 			continue;
1030 
1031 		/* Filter on address family. */
1032 		pgsa = (sockunion_t *)ifma->ifma_addr;
1033 		if (af != 0 && pgsa->sa.sa_family != af)
1034 			continue;
1035 
1036 		strlcpy(thisifname, link_ntoa(&psa->sdl), IFNAMSIZ);
1037 		pcolon = strchr(thisifname, ':');
1038 		if (pcolon)
1039 			*pcolon = '\0';
1040 
1041 		/* Only print the banner for the first ifmaddrs entry. */
1042 		if (lastifindex == 0 || lastifindex != thisifindex) {
1043 			lastifindex = thisifindex;
1044 			fprintf(stdout, "%s:\n", thisifname);
1045 		}
1046 
1047 		/*
1048 		 * Currently, multicast joins only take place on the
1049 		 * primary IPv4 address, and only on the link-local IPv6
1050 		 * address, as per IGMPv2/3 and MLDv1/2 semantics.
1051 		 * Therefore, we only look up the primary address on
1052 		 * the first pass.
1053 		 */
1054 		pifasa = NULL;
1055 		for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
1056 			if ((strcmp(ifa->ifa_name, thisifname) != 0) ||
1057 			    (ifa->ifa_addr == NULL) ||
1058 			    (ifa->ifa_addr->sa_family != pgsa->sa.sa_family))
1059 				continue;
1060 			/*
1061 			 * For AF_INET6 only the link-local address should
1062 			 * be returned. If built without IPv6 support,
1063 			 * skip this address entirely.
1064 			 */
1065 			pifasa = (sockunion_t *)ifa->ifa_addr;
1066 			if (pifasa->sa.sa_family == AF_INET6
1067 #ifdef INET6
1068 			    && !IN6_IS_ADDR_LINKLOCAL(&pifasa->sin6.sin6_addr)
1069 #endif
1070 			) {
1071 				pifasa = NULL;
1072 				continue;
1073 			}
1074 			break;
1075 		}
1076 		if (pifasa == NULL)
1077 			continue;	/* primary address not found */
1078 
1079 		if (!vflag && pifasa->sa.sa_family == AF_LINK)
1080 			continue;
1081 
1082 		/* Parse and print primary address, if not already printed. */
1083 		if (lastifasa.ss.ss_family == AF_UNSPEC ||
1084 		    ((lastifasa.ss.ss_family == AF_LINK &&
1085 		      !sa_dl_equal(&lastifasa.sa, &pifasa->sa)) ||
1086 		     !sa_equal(&lastifasa.sa, &pifasa->sa))) {
1087 
1088 			switch (pifasa->sa.sa_family) {
1089 			case AF_INET:
1090 				pafname = "inet";
1091 				break;
1092 			case AF_INET6:
1093 				pafname = "inet6";
1094 				break;
1095 			case AF_LINK:
1096 				pafname = "link";
1097 				break;
1098 			default:
1099 				pafname = "unknown";
1100 				break;
1101 			}
1102 
1103 			switch (pifasa->sa.sa_family) {
1104 			case AF_INET6:
1105 #ifdef INET6
1106 			{
1107 				const char *p =
1108 				    inet6_n2a(&pifasa->sin6.sin6_addr,
1109 					pifasa->sin6.sin6_scope_id);
1110 				strlcpy(addrbuf, p, sizeof(addrbuf));
1111 				break;
1112 			}
1113 #else
1114 			/* FALLTHROUGH */
1115 #endif
1116 			case AF_INET:
1117 			case AF_LINK:
1118 				error = getnameinfo(&pifasa->sa,
1119 				    pifasa->sa.sa_len,
1120 				    addrbuf, sizeof(addrbuf), NULL, 0,
1121 				    NI_NUMERICHOST);
1122 				if (error)
1123 					perror("getnameinfo");
1124 				break;
1125 			default:
1126 				addrbuf[0] = '\0';
1127 				break;
1128 			}
1129 
1130 			fprintf(stdout, "\t%s %s", pafname, addrbuf);
1131 #ifdef INET6
1132 			if (pifasa->sa.sa_family == AF_INET6 &&
1133 			    pifasa->sin6.sin6_scope_id)
1134 				fprintf(stdout, " scopeid 0x%x",
1135 				    pifasa->sin6.sin6_scope_id);
1136 #endif
1137 			fprintf(stdout, "\n");
1138 #ifdef INET
1139 			/*
1140 			 * Print per-link IGMP information, if available.
1141 			 */
1142 			if (pifasa->sa.sa_family == AF_INET) {
1143 				struct igmp_ifinfo igi;
1144 				size_t mibsize, len;
1145 				int mib[5];
1146 
1147 				mibsize = nitems(mib);
1148 				if (sysctlnametomib("net.inet.igmp.ifinfo",
1149 				    mib, &mibsize) == -1) {
1150 					perror("sysctlnametomib");
1151 					goto next_ifnet;
1152 				}
1153 				mib[mibsize] = thisifindex;
1154 				len = sizeof(struct igmp_ifinfo);
1155 				if (sysctl(mib, mibsize + 1, &igi, &len, NULL,
1156 				    0) == -1) {
1157 					perror("sysctl net.inet.igmp.ifinfo");
1158 					goto next_ifnet;
1159 				}
1160 				in_ifinfo(&igi);
1161 			}
1162 #endif /* INET */
1163 #ifdef INET6
1164 			/*
1165 			 * Print per-link MLD information, if available.
1166 			 */
1167 			if (pifasa->sa.sa_family == AF_INET6) {
1168 				struct mld_ifinfo mli;
1169 				size_t mibsize, len;
1170 				int mib[5];
1171 
1172 				mibsize = nitems(mib);
1173 				if (sysctlnametomib("net.inet6.mld.ifinfo",
1174 				    mib, &mibsize) == -1) {
1175 					perror("sysctlnametomib");
1176 					goto next_ifnet;
1177 				}
1178 				mib[mibsize] = thisifindex;
1179 				len = sizeof(struct mld_ifinfo);
1180 				if (sysctl(mib, mibsize + 1, &mli, &len, NULL,
1181 				    0) == -1) {
1182 					perror("sysctl net.inet6.mld.ifinfo");
1183 					goto next_ifnet;
1184 				}
1185 				in6_ifinfo(&mli);
1186 			}
1187 #endif /* INET6 */
1188 #if defined(INET) || defined(INET6)
1189 next_ifnet:
1190 #endif
1191 			lastifasa = *pifasa;
1192 		}
1193 
1194 		/* Print this group address. */
1195 #ifdef INET6
1196 		if (pgsa->sa.sa_family == AF_INET6) {
1197 			const char *p = inet6_n2a(&pgsa->sin6.sin6_addr,
1198 			    pgsa->sin6.sin6_scope_id);
1199 			strlcpy(addrbuf, p, sizeof(addrbuf));
1200 		} else
1201 #endif
1202 		{
1203 			error = getnameinfo(&pgsa->sa, pgsa->sa.sa_len,
1204 			    addrbuf, sizeof(addrbuf), NULL, 0, NI_NUMERICHOST);
1205 			if (error)
1206 				perror("getnameinfo");
1207 		}
1208 
1209 		fprintf(stdout, "\t\tgroup %s", addrbuf);
1210 #ifdef INET6
1211 		if (pgsa->sa.sa_family == AF_INET6 &&
1212 		    pgsa->sin6.sin6_scope_id)
1213 			fprintf(stdout, " scopeid 0x%x",
1214 			    pgsa->sin6.sin6_scope_id);
1215 #endif
1216 #ifdef INET
1217 		if (pgsa->sa.sa_family == AF_INET) {
1218 			inm_print_sources_sysctl(thisifindex,
1219 			    pgsa->sin.sin_addr);
1220 		}
1221 #endif
1222 #ifdef INET6
1223 		if (pgsa->sa.sa_family == AF_INET6) {
1224 			in6m_print_sources_sysctl(thisifindex,
1225 			    &pgsa->sin6.sin6_addr);
1226 		}
1227 #endif
1228 		fprintf(stdout, "\n");
1229 
1230 		/* Link-layer mapping, if present. */
1231 		pllsa = (sockunion_t *)ifma->ifma_lladdr;
1232 		if (pllsa != NULL) {
1233 			error = getnameinfo(&pllsa->sa, pllsa->sa.sa_len,
1234 			    addrbuf, sizeof(addrbuf), NULL, 0, NI_NUMERICHOST);
1235 			fprintf(stdout, "\t\t\tmcast-macaddr %s\n", addrbuf);
1236 		}
1237 	}
1238 out:
1239 	if (ifmap != NULL)
1240 		freeifmaddrs(ifmap);
1241 	if (ifap != NULL)
1242 		freeifaddrs(ifap);
1243 
1244 	return (error);
1245 }
1246