fix #>&- syntax for closing fds
[oweals/busybox.git] / networking / ping.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini ping implementation for busybox
4  *
5  * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
6  *
7  * Adapted from the ping in netkit-base 0.10:
8  * Copyright (c) 1989 The Regents of the University of California.
9  * All rights reserved.
10  *
11  * This code is derived from software contributed to Berkeley by
12  * Mike Muuss.
13  *
14  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
15  */
16 /* from ping6.c:
17  * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
18  *
19  * This version of ping is adapted from the ping in netkit-base 0.10,
20  * which is:
21  *
22  * Original copyright notice is retained at the end of this file.
23  *
24  * This version is an adaptation of ping.c from busybox.
25  * The code was modified by Bart Visscher <magick@linux-fan.com>
26  */
27
28 #include <net/if.h>
29 #include <netinet/ip_icmp.h>
30 #include "libbb.h"
31
32 #if ENABLE_PING6
33 #include <netinet/icmp6.h>
34 /* I see RENUMBERED constants in bits/in.h - !!?
35  * What a fuck is going on with libc? Is it a glibc joke? */
36 #ifdef IPV6_2292HOPLIMIT
37 #undef IPV6_HOPLIMIT
38 #define IPV6_HOPLIMIT IPV6_2292HOPLIMIT
39 #endif
40 #endif
41
42 enum {
43         DEFDATALEN = 56,
44         MAXIPLEN = 60,
45         MAXICMPLEN = 76,
46         MAXPACKET = 65468,
47         MAX_DUP_CHK = (8 * 128),
48         MAXWAIT = 10,
49         PINGINTERVAL = 1, /* 1 second */
50 };
51
52 /* common routines */
53
54 static int in_cksum(unsigned short *buf, int sz)
55 {
56         int nleft = sz;
57         int sum = 0;
58         unsigned short *w = buf;
59         unsigned short ans = 0;
60
61         while (nleft > 1) {
62                 sum += *w++;
63                 nleft -= 2;
64         }
65
66         if (nleft == 1) {
67                 *(unsigned char *) (&ans) = *(unsigned char *) w;
68                 sum += ans;
69         }
70
71         sum = (sum >> 16) + (sum & 0xFFFF);
72         sum += (sum >> 16);
73         ans = ~sum;
74         return ans;
75 }
76
77 #if !ENABLE_FEATURE_FANCY_PING
78
79 /* simple version */
80
81 static char *hostname;
82
83 static void noresp(int ign UNUSED_PARAM)
84 {
85         printf("No response from %s\n", hostname);
86         exit(EXIT_FAILURE);
87 }
88
89 static void ping4(len_and_sockaddr *lsa)
90 {
91         struct sockaddr_in pingaddr;
92         struct icmp *pkt;
93         int pingsock, c;
94         char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
95
96         pingsock = create_icmp_socket();
97         pingaddr = lsa->u.sin;
98
99         pkt = (struct icmp *) packet;
100         memset(pkt, 0, sizeof(packet));
101         pkt->icmp_type = ICMP_ECHO;
102         pkt->icmp_cksum = in_cksum((unsigned short *) pkt, sizeof(packet));
103
104         c = xsendto(pingsock, packet, DEFDATALEN + ICMP_MINLEN,
105                            (struct sockaddr *) &pingaddr, sizeof(pingaddr));
106
107         /* listen for replies */
108         while (1) {
109                 struct sockaddr_in from;
110                 socklen_t fromlen = sizeof(from);
111
112                 c = recvfrom(pingsock, packet, sizeof(packet), 0,
113                                 (struct sockaddr *) &from, &fromlen);
114                 if (c < 0) {
115                         if (errno != EINTR)
116                                 bb_perror_msg("recvfrom");
117                         continue;
118                 }
119                 if (c >= 76) {                  /* ip + icmp */
120                         struct iphdr *iphdr = (struct iphdr *) packet;
121
122                         pkt = (struct icmp *) (packet + (iphdr->ihl << 2));     /* skip ip hdr */
123                         if (pkt->icmp_type == ICMP_ECHOREPLY)
124                                 break;
125                 }
126         }
127         if (ENABLE_FEATURE_CLEAN_UP)
128                 close(pingsock);
129 }
130
131 #if ENABLE_PING6
132 static void ping6(len_and_sockaddr *lsa)
133 {
134         struct sockaddr_in6 pingaddr;
135         struct icmp6_hdr *pkt;
136         int pingsock, c;
137         int sockopt;
138         char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
139
140         pingsock = create_icmp6_socket();
141         pingaddr = lsa->u.sin6;
142
143         pkt = (struct icmp6_hdr *) packet;
144         memset(pkt, 0, sizeof(packet));
145         pkt->icmp6_type = ICMP6_ECHO_REQUEST;
146
147         sockopt = offsetof(struct icmp6_hdr, icmp6_cksum);
148         setsockopt(pingsock, SOL_RAW, IPV6_CHECKSUM, &sockopt, sizeof(sockopt));
149
150         c = xsendto(pingsock, packet, DEFDATALEN + sizeof (struct icmp6_hdr),
151                            (struct sockaddr *) &pingaddr, sizeof(pingaddr));
152
153         /* listen for replies */
154         while (1) {
155                 struct sockaddr_in6 from;
156                 socklen_t fromlen = sizeof(from);
157
158                 c = recvfrom(pingsock, packet, sizeof(packet), 0,
159                                 (struct sockaddr *) &from, &fromlen);
160                 if (c < 0) {
161                         if (errno != EINTR)
162                                 bb_perror_msg("recvfrom");
163                         continue;
164                 }
165                 if (c >= 8) {                   /* icmp6_hdr */
166                         pkt = (struct icmp6_hdr *) packet;
167                         if (pkt->icmp6_type == ICMP6_ECHO_REPLY)
168                                 break;
169                 }
170         }
171         if (ENABLE_FEATURE_CLEAN_UP)
172                 close(pingsock);
173 }
174 #endif
175
176 int ping_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
177 int ping_main(int argc UNUSED_PARAM, char **argv)
178 {
179         len_and_sockaddr *lsa;
180 #if ENABLE_PING6
181         sa_family_t af = AF_UNSPEC;
182
183         while ((++argv)[0] && argv[0][0] == '-') {
184                 if (argv[0][1] == '4') {
185                         af = AF_INET;
186                         continue;
187                 }
188                 if (argv[0][1] == '6') {
189                         af = AF_INET6;
190                         continue;
191                 }
192                 bb_show_usage();
193         }
194 #else
195         argv++;
196 #endif
197
198         hostname = *argv;
199         if (!hostname)
200                 bb_show_usage();
201
202 #if ENABLE_PING6
203         lsa = xhost_and_af2sockaddr(hostname, 0, af);
204 #else
205         lsa = xhost_and_af2sockaddr(hostname, 0, AF_INET);
206 #endif
207         /* Set timer _after_ DNS resolution */
208         signal(SIGALRM, noresp);
209         alarm(5); /* give the host 5000ms to respond */
210
211 #if ENABLE_PING6
212         if (lsa->u.sa.sa_family == AF_INET6)
213                 ping6(lsa);
214         else
215 #endif
216                 ping4(lsa);
217         printf("%s is alive!\n", hostname);
218         return EXIT_SUCCESS;
219 }
220
221
222 #else /* FEATURE_FANCY_PING */
223
224
225 /* full(er) version */
226
227 #define OPT_STRING ("qvc:s:w:W:I:4" USE_PING6("6"))
228 enum {
229         OPT_QUIET = 1 << 0,
230         OPT_VERBOSE = 1 << 1,
231         OPT_c = 1 << 2,
232         OPT_s = 1 << 3,
233         OPT_w = 1 << 4,
234         OPT_W = 1 << 5,
235         OPT_I = 1 << 6,
236         OPT_IPV4 = 1 << 7,
237         OPT_IPV6 = (1 << 8) * ENABLE_PING6,
238 };
239
240
241 struct globals {
242         int pingsock;
243         int if_index;
244         char *str_I;
245         len_and_sockaddr *source_lsa;
246         unsigned datalen;
247         unsigned pingcount; /* must be int-sized */
248         unsigned long ntransmitted, nreceived, nrepeats;
249         uint16_t myid;
250         unsigned tmin, tmax; /* in us */
251         unsigned long long tsum; /* in us, sum of all times */
252         unsigned deadline;
253         unsigned timeout;
254         unsigned total_secs;
255         const char *hostname;
256         const char *dotted;
257         union {
258                 struct sockaddr sa;
259                 struct sockaddr_in sin;
260 #if ENABLE_PING6
261                 struct sockaddr_in6 sin6;
262 #endif
263         } pingaddr;
264         char rcvd_tbl[MAX_DUP_CHK / 8];
265 };
266 #define G (*(struct globals*)&bb_common_bufsiz1)
267 #define pingsock     (G.pingsock    )
268 #define if_index     (G.if_index    )
269 #define source_lsa   (G.source_lsa  )
270 #define str_I        (G.str_I       )
271 #define datalen      (G.datalen     )
272 #define ntransmitted (G.ntransmitted)
273 #define nreceived    (G.nreceived   )
274 #define nrepeats     (G.nrepeats    )
275 #define pingcount    (G.pingcount   )
276 #define myid         (G.myid        )
277 #define tmin         (G.tmin        )
278 #define tmax         (G.tmax        )
279 #define tsum         (G.tsum        )
280 #define deadline     (G.deadline    )
281 #define timeout      (G.timeout     )
282 #define total_secs   (G.total_secs  )
283 #define hostname     (G.hostname    )
284 #define dotted       (G.dotted      )
285 #define pingaddr     (G.pingaddr    )
286 #define rcvd_tbl     (G.rcvd_tbl    )
287 void BUG_ping_globals_too_big(void);
288 #define INIT_G() do { \
289         if (sizeof(G) > COMMON_BUFSIZE) \
290                 BUG_ping_globals_too_big(); \
291         pingsock = -1; \
292         datalen = DEFDATALEN; \
293         timeout = MAXWAIT; \
294         tmin = UINT_MAX; \
295 } while (0)
296
297
298 #define A(bit)          rcvd_tbl[(bit)>>3]      /* identify byte in array */
299 #define B(bit)          (1 << ((bit) & 0x07))   /* identify bit in byte */
300 #define SET(bit)        (A(bit) |= B(bit))
301 #define CLR(bit)        (A(bit) &= (~B(bit)))
302 #define TST(bit)        (A(bit) & B(bit))
303
304 /**************************************************************************/
305
306 static void print_stats_and_exit(int junk) NORETURN;
307 static void print_stats_and_exit(int junk UNUSED_PARAM)
308 {
309         signal(SIGINT, SIG_IGN);
310
311         printf("\n--- %s ping statistics ---\n", hostname);
312         printf("%lu packets transmitted, ", ntransmitted);
313         printf("%lu packets received, ", nreceived);
314         if (nrepeats)
315                 printf("%lu duplicates, ", nrepeats);
316         if (ntransmitted)
317                 ntransmitted = (ntransmitted - nreceived) * 100 / ntransmitted;
318         printf("%lu%% packet loss\n", ntransmitted);
319         if (tmin != UINT_MAX) {
320                 unsigned tavg = tsum / (nreceived + nrepeats);
321                 printf("round-trip min/avg/max = %u.%03u/%u.%03u/%u.%03u ms\n",
322                         tmin / 1000, tmin % 1000,
323                         tavg / 1000, tavg % 1000,
324                         tmax / 1000, tmax % 1000);
325         }
326         /* if condition is true, exit with 1 -- 'failure' */
327         exit(nreceived == 0 || (deadline && nreceived < pingcount));
328 }
329
330 static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt)
331 {
332         int sz;
333
334         CLR((uint16_t)ntransmitted % MAX_DUP_CHK);
335         ntransmitted++;
336
337         /* sizeof(pingaddr) can be larger than real sa size, but I think
338          * it doesn't matter */
339         sz = xsendto(pingsock, pkt, size_pkt, &pingaddr.sa, sizeof(pingaddr));
340         if (sz != size_pkt)
341                 bb_error_msg_and_die(bb_msg_write_error);
342
343         if (pingcount == 0 || deadline || ntransmitted < pingcount) {
344                 /* Didn't send all pings yet - schedule next in 1s */
345                 signal(SIGALRM, sp);
346                 if (deadline) {
347                         total_secs += PINGINTERVAL;
348                         if (total_secs >= deadline)
349                                 signal(SIGALRM, print_stats_and_exit);
350                 }
351                 alarm(PINGINTERVAL);
352         } else { /* -c NN, and all NN are sent (and no deadline) */
353                 /* Wait for the last ping to come back.
354                  * -W timeout: wait for a response in seconds.
355                  * Affects only timeout in absense of any responses,
356                  * otherwise ping waits for two RTTs. */
357                 unsigned expire = timeout;
358
359                 if (nreceived) {
360                         /* approx. 2*tmax, in seconds (2 RTT) */
361                         expire = tmax / (512*1024);
362                         if (expire == 0)
363                                 expire = 1;
364                 }
365                 signal(SIGALRM, print_stats_and_exit);
366                 alarm(expire);
367         }
368 }
369
370 static void sendping4(int junk UNUSED_PARAM)
371 {
372         /* +4 reserves a place for timestamp, which may end up sitting
373          * *after* packet. Saves one if() */
374         struct icmp *pkt = alloca(datalen + ICMP_MINLEN + 4);
375
376         memset(pkt, 0, datalen + ICMP_MINLEN + 4);
377         pkt->icmp_type = ICMP_ECHO;
378         /*pkt->icmp_code = 0;*/
379         /*pkt->icmp_cksum = 0;*/
380         pkt->icmp_seq = htons(ntransmitted); /* don't ++ here, it can be a macro */
381         pkt->icmp_id = myid;
382
383         /* We don't do hton, because we will read it back on the same machine */
384         /*if (datalen >= 4)*/
385                 *(uint32_t*)&pkt->icmp_dun = monotonic_us();
386
387         pkt->icmp_cksum = in_cksum((unsigned short *) pkt, datalen + ICMP_MINLEN);
388
389         sendping_tail(sendping4, pkt, datalen + ICMP_MINLEN);
390 }
391 #if ENABLE_PING6
392 static void sendping6(int junk UNUSED_PARAM)
393 {
394         struct icmp6_hdr *pkt = alloca(datalen + sizeof(struct icmp6_hdr) + 4);
395
396         memset(pkt, 0, datalen + sizeof(struct icmp6_hdr) + 4);
397         pkt->icmp6_type = ICMP6_ECHO_REQUEST;
398         /*pkt->icmp6_code = 0;*/
399         /*pkt->icmp6_cksum = 0;*/
400         pkt->icmp6_seq = htons(ntransmitted); /* don't ++ here, it can be a macro */
401         pkt->icmp6_id = myid;
402
403         /*if (datalen >= 4)*/
404                 *(uint32_t*)(&pkt->icmp6_data8[4]) = monotonic_us();
405
406         sendping_tail(sendping6, pkt, datalen + sizeof(struct icmp6_hdr));
407 }
408 #endif
409
410 static const char *icmp_type_name(int id)
411 {
412         switch (id) {
413         case ICMP_ECHOREPLY:      return "Echo Reply";
414         case ICMP_DEST_UNREACH:   return "Destination Unreachable";
415         case ICMP_SOURCE_QUENCH:  return "Source Quench";
416         case ICMP_REDIRECT:       return "Redirect (change route)";
417         case ICMP_ECHO:           return "Echo Request";
418         case ICMP_TIME_EXCEEDED:  return "Time Exceeded";
419         case ICMP_PARAMETERPROB:  return "Parameter Problem";
420         case ICMP_TIMESTAMP:      return "Timestamp Request";
421         case ICMP_TIMESTAMPREPLY: return "Timestamp Reply";
422         case ICMP_INFO_REQUEST:   return "Information Request";
423         case ICMP_INFO_REPLY:     return "Information Reply";
424         case ICMP_ADDRESS:        return "Address Mask Request";
425         case ICMP_ADDRESSREPLY:   return "Address Mask Reply";
426         default:                  return "unknown ICMP type";
427         }
428 }
429 #if ENABLE_PING6
430 /* RFC3542 changed some definitions from RFC2292 for no good reason, whee!
431  * the newer 3542 uses a MLD_ prefix where as 2292 uses ICMP6_ prefix */
432 #ifndef MLD_LISTENER_QUERY
433 # define MLD_LISTENER_QUERY ICMP6_MEMBERSHIP_QUERY
434 #endif
435 #ifndef MLD_LISTENER_REPORT
436 # define MLD_LISTENER_REPORT ICMP6_MEMBERSHIP_REPORT
437 #endif
438 #ifndef MLD_LISTENER_REDUCTION
439 # define MLD_LISTENER_REDUCTION ICMP6_MEMBERSHIP_REDUCTION
440 #endif
441 static const char *icmp6_type_name(int id)
442 {
443         switch (id) {
444         case ICMP6_DST_UNREACH:      return "Destination Unreachable";
445         case ICMP6_PACKET_TOO_BIG:   return "Packet too big";
446         case ICMP6_TIME_EXCEEDED:    return "Time Exceeded";
447         case ICMP6_PARAM_PROB:       return "Parameter Problem";
448         case ICMP6_ECHO_REPLY:       return "Echo Reply";
449         case ICMP6_ECHO_REQUEST:     return "Echo Request";
450         case MLD_LISTENER_QUERY:     return "Listener Query";
451         case MLD_LISTENER_REPORT:    return "Listener Report";
452         case MLD_LISTENER_REDUCTION: return "Listener Reduction";
453         default:                     return "unknown ICMP type";
454         }
455 }
456 #endif
457
458 static void unpack_tail(int sz, uint32_t *tp,
459                 const char *from_str,
460                 uint16_t recv_seq, int ttl)
461 {
462         const char *dupmsg = " (DUP!)";
463         unsigned triptime = triptime; /* for gcc */
464
465         ++nreceived;
466
467         if (tp) {
468                 /* (int32_t) cast is for hypothetical 64-bit unsigned */
469                 /* (doesn't hurt 32-bit real-world anyway) */
470                 triptime = (int32_t) ((uint32_t)monotonic_us() - *tp);
471                 tsum += triptime;
472                 if (triptime < tmin)
473                         tmin = triptime;
474                 if (triptime > tmax)
475                         tmax = triptime;
476         }
477
478         if (TST(recv_seq % MAX_DUP_CHK)) {
479                 ++nrepeats;
480                 --nreceived;
481         } else {
482                 SET(recv_seq % MAX_DUP_CHK);
483                 dupmsg += 7;
484         }
485
486         if (option_mask32 & OPT_QUIET)
487                 return;
488
489         printf("%d bytes from %s: seq=%u ttl=%d", sz,
490                 from_str, recv_seq, ttl);
491         if (tp)
492                 printf(" time=%u.%03u ms", triptime / 1000, triptime % 1000);
493         puts(dupmsg);
494         fflush(stdout);
495 }
496 static void unpack4(char *buf, int sz, struct sockaddr_in *from)
497 {
498         struct icmp *icmppkt;
499         struct iphdr *iphdr;
500         int hlen;
501
502         /* discard if too short */
503         if (sz < (datalen + ICMP_MINLEN))
504                 return;
505
506         /* check IP header */
507         iphdr = (struct iphdr *) buf;
508         hlen = iphdr->ihl << 2;
509         sz -= hlen;
510         icmppkt = (struct icmp *) (buf + hlen);
511         if (icmppkt->icmp_id != myid)
512                 return;                         /* not our ping */
513
514         if (icmppkt->icmp_type == ICMP_ECHOREPLY) {
515                 uint16_t recv_seq = ntohs(icmppkt->icmp_seq);
516                 uint32_t *tp = NULL;
517
518                 if (sz >= ICMP_MINLEN + sizeof(uint32_t))
519                         tp = (uint32_t *) icmppkt->icmp_data;
520                 unpack_tail(sz, tp,
521                         inet_ntoa(*(struct in_addr *) &from->sin_addr.s_addr),
522                         recv_seq, iphdr->ttl);
523         } else if (icmppkt->icmp_type != ICMP_ECHO) {
524                 bb_error_msg("warning: got ICMP %d (%s)",
525                                 icmppkt->icmp_type,
526                                 icmp_type_name(icmppkt->icmp_type));
527         }
528 }
529 #if ENABLE_PING6
530 static void unpack6(char *packet, int sz, /*struct sockaddr_in6 *from,*/ int hoplimit)
531 {
532         struct icmp6_hdr *icmppkt;
533         char buf[INET6_ADDRSTRLEN];
534
535         /* discard if too short */
536         if (sz < (datalen + sizeof(struct icmp6_hdr)))
537                 return;
538
539         icmppkt = (struct icmp6_hdr *) packet;
540         if (icmppkt->icmp6_id != myid)
541                 return;                         /* not our ping */
542
543         if (icmppkt->icmp6_type == ICMP6_ECHO_REPLY) {
544                 uint16_t recv_seq = ntohs(icmppkt->icmp6_seq);
545                 uint32_t *tp = NULL;
546
547                 if (sz >= sizeof(struct icmp6_hdr) + sizeof(uint32_t))
548                         tp = (uint32_t *) &icmppkt->icmp6_data8[4];
549                 unpack_tail(sz, tp,
550                         inet_ntop(AF_INET6, &pingaddr.sin6.sin6_addr,
551                                         buf, sizeof(buf)),
552                         recv_seq, hoplimit);
553         } else if (icmppkt->icmp6_type != ICMP6_ECHO_REQUEST) {
554                 bb_error_msg("warning: got ICMP %d (%s)",
555                                 icmppkt->icmp6_type,
556                                 icmp6_type_name(icmppkt->icmp6_type));
557         }
558 }
559 #endif
560
561 static void ping4(len_and_sockaddr *lsa)
562 {
563         char packet[datalen + MAXIPLEN + MAXICMPLEN];
564         int sockopt;
565
566         pingsock = create_icmp_socket();
567         pingaddr.sin = lsa->u.sin;
568         if (source_lsa) {
569                 if (setsockopt(pingsock, IPPROTO_IP, IP_MULTICAST_IF,
570                                 &source_lsa->u.sa, source_lsa->len))
571                         bb_error_msg_and_die("can't set multicast source interface");
572                 xbind(pingsock, &source_lsa->u.sa, source_lsa->len);
573         }
574         if (str_I)
575                 setsockopt_bindtodevice(pingsock, str_I);
576
577         /* enable broadcast pings */
578         setsockopt_broadcast(pingsock);
579
580         /* set recv buf (needed if we can get lots of responses: flood ping,
581          * broadcast ping etc) */
582         sockopt = (datalen * 2) + 7 * 1024; /* giving it a bit of extra room */
583         setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
584
585         signal(SIGINT, print_stats_and_exit);
586
587         /* start the ping's going ... */
588         sendping4(0);
589
590         /* listen for replies */
591         while (1) {
592                 struct sockaddr_in from;
593                 socklen_t fromlen = (socklen_t) sizeof(from);
594                 int c;
595
596                 c = recvfrom(pingsock, packet, sizeof(packet), 0,
597                                 (struct sockaddr *) &from, &fromlen);
598                 if (c < 0) {
599                         if (errno != EINTR)
600                                 bb_perror_msg("recvfrom");
601                         continue;
602                 }
603                 unpack4(packet, c, &from);
604                 if (pingcount && nreceived >= pingcount)
605                         break;
606         }
607 }
608 #if ENABLE_PING6
609 extern int BUG_bad_offsetof_icmp6_cksum(void);
610 static void ping6(len_and_sockaddr *lsa)
611 {
612         char packet[datalen + MAXIPLEN + MAXICMPLEN];
613         int sockopt;
614         struct msghdr msg;
615         struct sockaddr_in6 from;
616         struct iovec iov;
617         char control_buf[CMSG_SPACE(36)];
618
619         pingsock = create_icmp6_socket();
620         pingaddr.sin6 = lsa->u.sin6;
621         /* untested whether "-I addr" really works for IPv6: */
622         if (source_lsa)
623                 xbind(pingsock, &source_lsa->u.sa, source_lsa->len);
624         if (str_I)
625                 setsockopt_bindtodevice(pingsock, str_I);
626
627 #ifdef ICMP6_FILTER
628         {
629                 struct icmp6_filter filt;
630                 if (!(option_mask32 & OPT_VERBOSE)) {
631                         ICMP6_FILTER_SETBLOCKALL(&filt);
632                         ICMP6_FILTER_SETPASS(ICMP6_ECHO_REPLY, &filt);
633                 } else {
634                         ICMP6_FILTER_SETPASSALL(&filt);
635                 }
636                 if (setsockopt(pingsock, IPPROTO_ICMPV6, ICMP6_FILTER, &filt,
637                                            sizeof(filt)) < 0)
638                         bb_error_msg_and_die("setsockopt(ICMP6_FILTER)");
639         }
640 #endif /*ICMP6_FILTER*/
641
642         /* enable broadcast pings */
643         setsockopt_broadcast(pingsock);
644
645         /* set recv buf (needed if we can get lots of responses: flood ping,
646          * broadcast ping etc) */
647         sockopt = (datalen * 2) + 7 * 1024; /* giving it a bit of extra room */
648         setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
649
650         sockopt = offsetof(struct icmp6_hdr, icmp6_cksum);
651         if (offsetof(struct icmp6_hdr, icmp6_cksum) != 2)
652                 BUG_bad_offsetof_icmp6_cksum();
653         setsockopt(pingsock, SOL_RAW, IPV6_CHECKSUM, &sockopt, sizeof(sockopt));
654
655         /* request ttl info to be returned in ancillary data */
656         setsockopt(pingsock, SOL_IPV6, IPV6_HOPLIMIT, &const_int_1, sizeof(const_int_1));
657
658         if (if_index)
659                 pingaddr.sin6.sin6_scope_id = if_index;
660
661         signal(SIGINT, print_stats_and_exit);
662
663         /* start the ping's going ... */
664         sendping6(0);
665
666         /* listen for replies */
667         msg.msg_name = &from;
668         msg.msg_namelen = sizeof(from);
669         msg.msg_iov = &iov;
670         msg.msg_iovlen = 1;
671         msg.msg_control = control_buf;
672         iov.iov_base = packet;
673         iov.iov_len = sizeof(packet);
674         while (1) {
675                 int c;
676                 struct cmsghdr *mp;
677                 int hoplimit = -1;
678                 msg.msg_controllen = sizeof(control_buf);
679
680                 c = recvmsg(pingsock, &msg, 0);
681                 if (c < 0) {
682                         if (errno != EINTR)
683                                 bb_perror_msg("recvfrom");
684                         continue;
685                 }
686                 for (mp = CMSG_FIRSTHDR(&msg); mp; mp = CMSG_NXTHDR(&msg, mp)) {
687                         if (mp->cmsg_level == SOL_IPV6
688                          && mp->cmsg_type == IPV6_HOPLIMIT
689                          /* don't check len - we trust the kernel: */
690                          /* && mp->cmsg_len >= CMSG_LEN(sizeof(int)) */
691                         ) {
692                                 hoplimit = *(int*)CMSG_DATA(mp);
693                         }
694                 }
695                 unpack6(packet, c, /*&from,*/ hoplimit);
696                 if (pingcount && nreceived >= pingcount)
697                         break;
698         }
699 }
700 #endif
701
702 static void ping(len_and_sockaddr *lsa)
703 {
704         printf("PING %s (%s)", hostname, dotted);
705         if (source_lsa) {
706                 printf(" from %s",
707                         xmalloc_sockaddr2dotted_noport(&source_lsa->u.sa));
708         }
709         printf(": %d data bytes\n", datalen);
710
711 #if ENABLE_PING6
712         if (lsa->u.sa.sa_family == AF_INET6)
713                 ping6(lsa);
714         else
715 #endif
716                 ping4(lsa);
717 }
718
719 int ping_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
720 int ping_main(int argc UNUSED_PARAM, char **argv)
721 {
722         len_and_sockaddr *lsa;
723         char *str_s;
724         int opt;
725
726         INIT_G();
727
728         /* exactly one argument needed; -v and -q don't mix; -c NUM, -w NUM, -W NUM */
729         opt_complementary = "=1:q--v:v--q:c+:w+:W+";
730         opt = getopt32(argv, OPT_STRING, &pingcount, &str_s, &deadline, &timeout, &str_I);
731         if (opt & OPT_s)
732                 datalen = xatou16(str_s); // -s
733         if (opt & OPT_I) { // -I
734                 if_index = if_nametoindex(str_I);
735                 if (!if_index) {
736                         /* TODO: I'm not sure it takes IPv6 unless in [XX:XX..] format */
737                         source_lsa = xdotted2sockaddr(str_I, 0);
738                         str_I = NULL; /* don't try to bind to device later */
739                 }
740         }
741         myid = (uint16_t) getpid();
742         hostname = argv[optind];
743 #if ENABLE_PING6
744         {
745                 sa_family_t af = AF_UNSPEC;
746                 if (opt & OPT_IPV4)
747                         af = AF_INET;
748                 if (opt & OPT_IPV6)
749                         af = AF_INET6;
750                 lsa = xhost_and_af2sockaddr(hostname, 0, af);
751         }
752 #else
753         lsa = xhost_and_af2sockaddr(hostname, 0, AF_INET);
754 #endif
755
756         if (source_lsa && source_lsa->u.sa.sa_family != lsa->u.sa.sa_family)
757                 /* leaking it here... */
758                 source_lsa = NULL;
759
760         dotted = xmalloc_sockaddr2dotted_noport(&lsa->u.sa);
761         ping(lsa);
762         print_stats_and_exit(EXIT_SUCCESS);
763         /*return EXIT_SUCCESS;*/
764 }
765 #endif /* FEATURE_FANCY_PING */
766
767
768 #if ENABLE_PING6
769 int ping6_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
770 int ping6_main(int argc UNUSED_PARAM, char **argv)
771 {
772         argv[0] = (char*)"-6";
773         return ping_main(0 /* argc+1 - but it's unused anyway */,
774                         argv - 1);
775 }
776 #endif
777
778 /* from ping6.c:
779  * Copyright (c) 1989 The Regents of the University of California.
780  * All rights reserved.
781  *
782  * This code is derived from software contributed to Berkeley by
783  * Mike Muuss.
784  *
785  * Redistribution and use in source and binary forms, with or without
786  * modification, are permitted provided that the following conditions
787  * are met:
788  * 1. Redistributions of source code must retain the above copyright
789  *    notice, this list of conditions and the following disclaimer.
790  * 2. Redistributions in binary form must reproduce the above copyright
791  *    notice, this list of conditions and the following disclaimer in the
792  *    documentation and/or other materials provided with the distribution.
793  *
794  * 3. <BSD Advertising Clause omitted per the July 22, 1999 licensing change
795  *              ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change>
796  *
797  * 4. Neither the name of the University nor the names of its contributors
798  *    may be used to endorse or promote products derived from this software
799  *    without specific prior written permission.
800  *
801  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
802  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
803  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
804  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
805  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
806  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
807  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
808  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
809  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
810  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
811  * SUCH DAMAGE.
812  */