usr.sbin/ifmcstat/ifmcstat.c

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