mld6.c revision 78064
1/* $KAME: mld6.c,v 1.11 2001/05/13 15:45:07 suz Exp $ */ 2/* $FreeBSD: head/usr.sbin/mld6query/mld6.c 78064 2001-06-11 12:39:29Z ume $ */ 3 4/* 5 * Copyright (C) 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#include <sys/param.h> 33#include <sys/uio.h> 34#include <sys/socket.h> 35#include <sys/types.h> 36#include <sys/time.h> 37#include <unistd.h> 38#include <signal.h> 39 40#include <net/if.h> 41#include <net/if_var.h> 42 43#include <netinet/in.h> 44#include <netinet/ip6.h> 45#include <netinet/icmp6.h> 46 47#include <arpa/inet.h> 48 49#include <stdlib.h> 50#include <stdio.h> 51#include <string.h> 52#include <err.h> 53 54struct msghdr m; 55struct sockaddr_in6 dst; 56struct mld6_hdr mldh; 57struct in6_addr maddr = IN6ADDR_ANY_INIT, any = IN6ADDR_ANY_INIT; 58struct ipv6_mreq mreq; 59u_short ifindex; 60int s; 61 62#define QUERY_RESPONSE_INTERVAL 10000 63 64void make_msg(int index, struct in6_addr *addr, u_int type); 65void usage(void); 66void dump(int); 67void quit(int); 68 69int 70main(int argc, char *argv[]) 71{ 72 int i; 73 struct icmp6_filter filt; 74 u_int hlim = 1; 75 fd_set fdset; 76 struct itimerval itimer; 77 u_int type; 78 int ch; 79 80 type = MLD6_LISTENER_QUERY; 81 while ((ch = getopt(argc, argv, "dr")) != -1) { 82 switch (ch) { 83 case 'd': 84 type = MLD6_LISTENER_DONE; 85 break; 86 case 'r': 87 type = MLD6_LISTENER_REPORT; 88 break; 89 default: 90 usage(); 91 /*NOTREACHED*/ 92 } 93 } 94 95 argv += optind; 96 argc -= optind; 97 98 if (argc != 1 && argc != 2) 99 usage(); 100 101 ifindex = (u_short)if_nametoindex(argv[0]); 102 if (ifindex == 0) 103 usage(); 104 if (argc == 2 && inet_pton(AF_INET6, argv[1], &maddr) != 1) 105 usage(); 106 107 if ((s = socket(AF_INET6, SOCK_RAW, IPPROTO_ICMPV6)) < 0) 108 err(1, "socket"); 109 110 if (setsockopt(s, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &hlim, 111 sizeof(hlim)) == -1) 112 err(1, "setsockopt(IPV6_MULTICAST_HOPS)"); 113 114 mreq.ipv6mr_multiaddr = any; 115 mreq.ipv6mr_interface = ifindex; 116 if (setsockopt(s, IPPROTO_IPV6, IPV6_JOIN_GROUP, &mreq, 117 sizeof(mreq)) == -1) 118 err(1, "setsockopt(IPV6_JOIN_GROUP)"); 119 120 ICMP6_FILTER_SETBLOCKALL(&filt); 121 ICMP6_FILTER_SETPASS(ICMP6_MEMBERSHIP_QUERY, &filt); 122 ICMP6_FILTER_SETPASS(ICMP6_MEMBERSHIP_REPORT, &filt); 123 ICMP6_FILTER_SETPASS(ICMP6_MEMBERSHIP_REDUCTION, &filt); 124 if (setsockopt(s, IPPROTO_ICMPV6, ICMP6_FILTER, &filt, 125 sizeof(filt)) < 0) 126 err(1, "setsockopt(ICMP6_FILTER)"); 127 128 make_msg(ifindex, &maddr, type); 129 130 if (sendmsg(s, &m, 0) < 0) 131 err(1, "sendmsg"); 132 133 itimer.it_value.tv_sec = QUERY_RESPONSE_INTERVAL / 1000; 134 itimer.it_interval.tv_sec = 0; 135 itimer.it_interval.tv_usec = 0; 136 itimer.it_value.tv_usec = 0; 137 138 (void)signal(SIGALRM, quit); 139 (void)setitimer(ITIMER_REAL, &itimer, NULL); 140 141 FD_ZERO(&fdset); 142 for (;;) { 143 FD_SET(s, &fdset); 144 if ((i = select(s + 1, &fdset, NULL, NULL, NULL)) < 0) 145 perror("select"); 146 if (i == 0) 147 continue; 148 else 149 dump(s); 150 } 151} 152 153void 154make_msg(int index, struct in6_addr *addr, u_int type) 155{ 156 static struct iovec iov[2]; 157 static u_char *cmsgbuf; 158 int cmsglen, hbhlen = 0; 159 u_int8_t raopt[IP6OPT_RTALERT_LEN]; 160 struct in6_pktinfo *pi; 161 struct cmsghdr *cmsgp; 162 u_short rtalert_code = htons(IP6OPT_RTALERT_MLD); 163 164 dst.sin6_len = sizeof(dst); 165 dst.sin6_family = AF_INET6; 166 if (IN6_IS_ADDR_UNSPECIFIED(addr)) { 167 if (inet_pton(AF_INET6, "ff02::1", &dst.sin6_addr) != 1) 168 errx(1, "inet_pton failed"); 169 } 170 else 171 dst.sin6_addr = *addr; 172 m.msg_name = (caddr_t)&dst; 173 m.msg_namelen = dst.sin6_len; 174 iov[0].iov_base = (caddr_t)&mldh; 175 iov[0].iov_len = sizeof(mldh); 176 m.msg_iov = iov; 177 m.msg_iovlen = 1; 178 179 bzero(&mldh, sizeof(mldh)); 180 mldh.mld6_type = type & 0xff; 181 mldh.mld6_maxdelay = htons(QUERY_RESPONSE_INTERVAL); 182 mldh.mld6_addr = *addr; 183 184 hbhlen = sizeof(raopt); 185 cmsglen = CMSG_SPACE(sizeof(struct in6_pktinfo)) + 186 inet6_option_space(hbhlen); 187 188 if ((cmsgbuf = malloc(cmsglen)) == NULL) 189 errx(1, "can't allocate enough memory for cmsg"); 190 cmsgp = (struct cmsghdr *)cmsgbuf; 191 m.msg_control = (caddr_t)cmsgbuf; 192 m.msg_controllen = cmsglen; 193 /* specify the outgoing interface */ 194 cmsgp->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo)); 195 cmsgp->cmsg_level = IPPROTO_IPV6; 196 cmsgp->cmsg_type = IPV6_PKTINFO; 197 pi = (struct in6_pktinfo *)CMSG_DATA(cmsgp); 198 pi->ipi6_ifindex = index; 199 memset(&pi->ipi6_addr, 0, sizeof(pi->ipi6_addr)); 200 /* specifiy to insert router alert option in a hop-by-hop opt hdr. */ 201 cmsgp = CMSG_NXTHDR(&m, cmsgp); 202 if (inet6_option_init((void *)cmsgp, &cmsgp, IPV6_HOPOPTS)) 203 errx(1, "inet6_option_init failed\n"); 204 raopt[0] = IP6OPT_RTALERT; 205 raopt[1] = IP6OPT_RTALERT_LEN - 2; 206 memcpy(&raopt[2], (caddr_t)&rtalert_code, sizeof(u_short)); 207 if (inet6_option_append(cmsgp, raopt, 4, 0)) 208 errx(1, "inet6_option_append failed\n"); 209} 210 211void 212dump(int s) 213{ 214 int i; 215 struct mld6_hdr *mld; 216 u_char buf[1024]; 217 struct sockaddr_in6 from; 218 int from_len = sizeof(from); 219 char ntop_buf[256]; 220 221 if ((i = recvfrom(s, buf, sizeof(buf), 0, 222 (struct sockaddr *)&from, 223 &from_len)) < 0) 224 return; 225 226 if (i < sizeof(struct mld6_hdr)) { 227 printf("too short!\n"); 228 return; 229 } 230 231 mld = (struct mld6_hdr *)buf; 232 233 printf("from %s, ", inet_ntop(AF_INET6, &from.sin6_addr, 234 ntop_buf, sizeof(ntop_buf))); 235 236 switch (mld->mld6_type) { 237 case ICMP6_MEMBERSHIP_QUERY: 238 printf("type=Multicast Listener Query, "); 239 break; 240 case ICMP6_MEMBERSHIP_REPORT: 241 printf("type=Multicast Listener Report, "); 242 break; 243 case ICMP6_MEMBERSHIP_REDUCTION: 244 printf("type=Multicast Listener Done, "); 245 break; 246 } 247 printf("addr=%s\n", inet_ntop(AF_INET6, &mld->mld6_addr, 248 ntop_buf, sizeof(ntop_buf))); 249 250 fflush(stdout); 251} 252 253/* ARGSUSED */ 254void 255quit(int signum) { 256 mreq.ipv6mr_multiaddr = any; 257 mreq.ipv6mr_interface = ifindex; 258 if (setsockopt(s, IPPROTO_IPV6, IPV6_LEAVE_GROUP, &mreq, 259 sizeof(mreq)) == -1) 260 err(1, "setsockopt(IPV6_LEAVE_GROUP)"); 261 262 exit(0); 263} 264 265void 266usage() 267{ 268 (void)fprintf(stderr, "usage: mld6query ifname [addr]\n"); 269 exit(1); 270} 271