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 ATTRIBUTE_UNUSED)
  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 ATTRIBUTE_UNUSED, 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 *opt_I;
 245         len_and_sockaddr *source_lsa;
 246         unsigned datalen;
 247         unsigned long ntransmitted, nreceived, nrepeats, pingcount;
 248         uint16_t myid;
 249         unsigned tmin, tmax; /* in us */
 250         unsigned long long tsum; /* in us, sum of all times */
 251         unsigned deadline;
 252         unsigned timeout;
 253         unsigned total_secs;
 254         const char *hostname;
 255         const char *dotted;
 256         union {
 257                 struct sockaddr sa;
 258                 struct sockaddr_in sin;
 259 #if ENABLE_PING6
 260                 struct sockaddr_in6 sin6;
 261 #endif
 262         } pingaddr;
 263         char rcvd_tbl[MAX_DUP_CHK / 8];
 264 };
 265 #define G (*(struct globals*)&bb_common_bufsiz1)
 266 #define pingsock     (G.pingsock    )
 267 #define if_index     (G.if_index    )
 268 #define source_lsa   (G.source_lsa  )
 269 #define opt_I        (G.opt_I       )
 270 #define datalen      (G.datalen     )
 271 #define ntransmitted (G.ntransmitted)
 272 #define nreceived    (G.nreceived   )
 273 #define nrepeats     (G.nrepeats    )
 274 #define pingcount    (G.pingcount   )
 275 #define myid         (G.myid        )
 276 #define tmin         (G.tmin        )
 277 #define tmax         (G.tmax        )
 278 #define tsum         (G.tsum        )
 279 #define deadline     (G.deadline    )
 280 #define timeout      (G.timeout     )
 281 #define total_secs   (G.total_secs  )
 282 #define hostname     (G.hostname    )
 283 #define dotted       (G.dotted      )
 284 #define pingaddr     (G.pingaddr    )
 285 #define rcvd_tbl     (G.rcvd_tbl    )
 286 void BUG_ping_globals_too_big(void);
 287 #define INIT_G() do { \
 288         if (sizeof(G) > COMMON_BUFSIZE) \
 289                 BUG_ping_globals_too_big(); \
 290         pingsock = -1; \
 291         datalen = DEFDATALEN; \
 292         timeout = MAXWAIT; \
 293         tmin = UINT_MAX; \
 294 } while (0)
 295
 296
 297 #define A(bit)          rcvd_tbl[(bit)>>3]      /* identify byte in array */
 298 #define B(bit)          (1 << ((bit) & 0x07))   /* identify bit in byte */
 299 #define SET(bit)        (A(bit) |= B(bit))
 300 #define CLR(bit)        (A(bit) &= (~B(bit)))
 301 #define TST(bit)        (A(bit) & B(bit))
 302
 303 /**************************************************************************/
 304
 305 static void print_stats_and_exit(int junk) ATTRIBUTE_NORETURN;
 306 static void print_stats_and_exit(int junk ATTRIBUTE_UNUSED)
 307 {
 308         signal(SIGINT, SIG_IGN);
 309
 310         printf("\n--- %s ping statistics ---\n", hostname);
 311         printf("%lu packets transmitted, ", ntransmitted);
 312         printf("%lu packets received, ", nreceived);
 313         if (nrepeats)
 314                 printf("%lu duplicates, ", nrepeats);
 315         if (ntransmitted)
 316                 ntransmitted = (ntransmitted - nreceived) * 100 / ntransmitted;
 317         printf("%lu%% packet loss\n", ntransmitted);
 318         if (tmin != UINT_MAX) {
 319                 unsigned tavg = tsum / (nreceived + nrepeats);
 320                 printf("round-trip min/avg/max = %u.%03u/%u.%03u/%u.%03u ms\n",
 321                         tmin / 1000, tmin % 1000,
 322                         tavg / 1000, tavg % 1000,
 323                         tmax / 1000, tmax % 1000);
 324         }
 325         /* if condition is true, exit with 1 -- 'failure' */
 326         exit(nreceived == 0 || (deadline && nreceived < pingcount));
 327 }
 328
 329 static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt)
 330 {
 331         int sz;
 332
 333         CLR((uint16_t)ntransmitted % MAX_DUP_CHK);
 334         ntransmitted++;
 335
 336         /* sizeof(pingaddr) can be larger than real sa size, but I think
 337          * it doesn't matter */
 338         sz = xsendto(pingsock, pkt, size_pkt, &pingaddr.sa, sizeof(pingaddr));
 339         if (sz != size_pkt)
 340                 bb_error_msg_and_die(bb_msg_write_error);
 341
 342         if (pingcount == 0 || deadline || ntransmitted < pingcount) {
 343                 /* Didn't send all pings yet - schedule next in 1s */
 344                 signal(SIGALRM, sp);
 345                 if (deadline) {
 346                         total_secs += PINGINTERVAL;
 347                         if (total_secs >= deadline)
 348                                 signal(SIGALRM, print_stats_and_exit);
 349                 }
 350                 alarm(PINGINTERVAL);
 351         } else { /* -c NN, and all NN are sent (and no deadline) */
 352                 /* Wait for the last ping to come back.
 353                  * -W timeout: wait for a response in seconds.
 354                  * Affects only timeout in absense of any responses,
 355                  * otherwise ping waits for two RTTs. */
 356                 unsigned expire = timeout;
 357
 358                 if (nreceived) {
 359                         /* approx. 2*tmax, in seconds (2 RTT) */
 360                         expire = tmax / (512*1024);
 361                         if (expire == 0)
 362                                 expire = 1;
 363                 }
 364                 signal(SIGALRM, print_stats_and_exit);
 365                 alarm(expire);
 366         }
 367 }
 368
 369 static void sendping4(int junk ATTRIBUTE_UNUSED)
 370 {
 371         /* +4 reserves a place for timestamp, which may end up sitting
 372          * *after* packet. Saves one if() */
 373         struct icmp *pkt = alloca(datalen + ICMP_MINLEN + 4);
 374
 375         memset(pkt, 0, datalen + ICMP_MINLEN + 4);
 376         pkt->icmp_type = ICMP_ECHO;
 377         /*pkt->icmp_code = 0;*/
 378         /*pkt->icmp_cksum = 0;*/
 379         pkt->icmp_seq = htons(ntransmitted); /* don't ++ here, it can be a macro */
 380         pkt->icmp_id = myid;
 381
 382         /* We don't do hton, because we will read it back on the same machine */
 383         /*if (datalen >= 4)*/
 384                 *(uint32_t*)&pkt->icmp_dun = monotonic_us();
 385
 386         pkt->icmp_cksum = in_cksum((unsigned short *) pkt, datalen + ICMP_MINLEN);
 387
 388         sendping_tail(sendping4, pkt, datalen + ICMP_MINLEN);
 389 }
 390 #if ENABLE_PING6
 391 static void sendping6(int junk ATTRIBUTE_UNUSED)
 392 {
 393         struct icmp6_hdr *pkt = alloca(datalen + sizeof(struct icmp6_hdr) + 4);
 394
 395         memset(pkt, 0, datalen + sizeof(struct icmp6_hdr) + 4);
 396         pkt->icmp6_type = ICMP6_ECHO_REQUEST;
 397         /*pkt->icmp6_code = 0;*/
 398         /*pkt->icmp6_cksum = 0;*/
 399         pkt->icmp6_seq = htons(ntransmitted); /* don't ++ here, it can be a macro */
 400         pkt->icmp6_id = myid;
 401
 402         /*if (datalen >= 4)*/
 403                 *(uint32_t*)(&pkt->icmp6_data8[4]) = monotonic_us();
 404
 405         sendping_tail(sendping6, pkt, datalen + sizeof(struct icmp6_hdr));
 406 }
 407 #endif
 408
 409 static const char *icmp_type_name(int id)
 410 {
 411         switch (id) {
 412         case ICMP_ECHOREPLY:      return "Echo Reply";
 413         case ICMP_DEST_UNREACH:   return "Destination Unreachable";
 414         case ICMP_SOURCE_QUENCH:  return "Source Quench";
 415         case ICMP_REDIRECT:       return "Redirect (change route)";
 416         case ICMP_ECHO:           return "Echo Request";
 417         case ICMP_TIME_EXCEEDED:  return "Time Exceeded";
 418         case ICMP_PARAMETERPROB:  return "Parameter Problem";
 419         case ICMP_TIMESTAMP:      return "Timestamp Request";
 420         case ICMP_TIMESTAMPREPLY: return "Timestamp Reply";
 421         case ICMP_INFO_REQUEST:   return "Information Request";
 422         case ICMP_INFO_REPLY:     return "Information Reply";
 423         case ICMP_ADDRESS:        return "Address Mask Request";
 424         case ICMP_ADDRESSREPLY:   return "Address Mask Reply";
 425         default:                  return "unknown ICMP type";
 426         }
 427 }
 428 #if ENABLE_PING6
 429 /* RFC3542 changed some definitions from RFC2292 for no good reason, whee!
 430  * the newer 3542 uses a MLD_ prefix where as 2292 uses ICMP6_ prefix */
 431 #ifndef MLD_LISTENER_QUERY
 432 # define MLD_LISTENER_QUERY ICMP6_MEMBERSHIP_QUERY
 433 #endif
 434 #ifndef MLD_LISTENER_REPORT
 435 # define MLD_LISTENER_REPORT ICMP6_MEMBERSHIP_REPORT
 436 #endif
 437 #ifndef MLD_LISTENER_REDUCTION
 438 # define MLD_LISTENER_REDUCTION ICMP6_MEMBERSHIP_REDUCTION
 439 #endif
 440 static const char *icmp6_type_name(int id)
 441 {
 442         switch (id) {
 443         case ICMP6_DST_UNREACH:      return "Destination Unreachable";
 444         case ICMP6_PACKET_TOO_BIG:   return "Packet too big";
 445         case ICMP6_TIME_EXCEEDED:    return "Time Exceeded";
 446         case ICMP6_PARAM_PROB:       return "Parameter Problem";
 447         case ICMP6_ECHO_REPLY:       return "Echo Reply";
 448         case ICMP6_ECHO_REQUEST:     return "Echo Request";
 449         case MLD_LISTENER_QUERY:     return "Listener Query";
 450         case MLD_LISTENER_REPORT:    return "Listener Report";
 451         case MLD_LISTENER_REDUCTION: return "Listener Reduction";
 452         default:                     return "unknown ICMP type";
 453         }
 454 }
 455 #endif
 456
 457 static void unpack_tail(int sz, uint32_t *tp,
 458                 const char *from_str,
 459                 uint16_t recv_seq, int ttl)
 460 {
 461         const char *dupmsg = " (DUP!)";
 462         unsigned triptime = triptime; /* for gcc */
 463
 464         ++nreceived;
 465
 466         if (tp) {
 467                 /* (int32_t) cast is for hypothetical 64-bit unsigned */
 468                 /* (doesn't hurt 32-bit real-world anyway) */
 469                 triptime = (int32_t) ((uint32_t)monotonic_us() - *tp);
 470                 tsum += triptime;
 471                 if (triptime < tmin)
 472                         tmin = triptime;
 473                 if (triptime > tmax)
 474                         tmax = triptime;
 475         }
 476
 477         if (TST(recv_seq % MAX_DUP_CHK)) {
 478                 ++nrepeats;
 479                 --nreceived;
 480         } else {
 481                 SET(recv_seq % MAX_DUP_CHK);
 482                 dupmsg += 7;
 483         }
 484
 485         if (option_mask32 & OPT_QUIET)
 486                 return;
 487
 488         printf("%d bytes from %s: seq=%u ttl=%d", sz,
 489                 from_str, recv_seq, ttl);
 490         if (tp)
 491                 printf(" time=%u.%03u ms", triptime / 1000, triptime % 1000);
 492         puts(dupmsg);
 493         fflush(stdout);
 494 }
 495 static void unpack4(char *buf, int sz, struct sockaddr_in *from)
 496 {
 497         struct icmp *icmppkt;
 498         struct iphdr *iphdr;
 499         int hlen;
 500
 501         /* discard if too short */
 502         if (sz < (datalen + ICMP_MINLEN))
 503                 return;
 504
 505         /* check IP header */
 506         iphdr = (struct iphdr *) buf;
 507         hlen = iphdr->ihl << 2;
 508         sz -= hlen;
 509         icmppkt = (struct icmp *) (buf + hlen);
 510         if (icmppkt->icmp_id != myid)
 511                 return;                         /* not our ping */
 512
 513         if (icmppkt->icmp_type == ICMP_ECHOREPLY) {
 514                 uint16_t recv_seq = ntohs(icmppkt->icmp_seq);
 515                 uint32_t *tp = NULL;
 516
 517                 if (sz >= ICMP_MINLEN + sizeof(uint32_t))
 518                         tp = (uint32_t *) icmppkt->icmp_data;
 519                 unpack_tail(sz, tp,
 520                         inet_ntoa(*(struct in_addr *) &from->sin_addr.s_addr),
 521                         recv_seq, iphdr->ttl);
 522         } else if (icmppkt->icmp_type != ICMP_ECHO) {
 523                 bb_error_msg("warning: got ICMP %d (%s)",
 524                                 icmppkt->icmp_type,
 525                                 icmp_type_name(icmppkt->icmp_type));
 526         }
 527 }
 528 #if ENABLE_PING6
 529 static void unpack6(char *packet, int sz, /*struct sockaddr_in6 *from,*/ int hoplimit)
 530 {
 531         struct icmp6_hdr *icmppkt;
 532         char buf[INET6_ADDRSTRLEN];
 533
 534         /* discard if too short */
 535         if (sz < (datalen + sizeof(struct icmp6_hdr)))
 536                 return;
 537
 538         icmppkt = (struct icmp6_hdr *) packet;
 539         if (icmppkt->icmp6_id != myid)
 540                 return;                         /* not our ping */
 541
 542         if (icmppkt->icmp6_type == ICMP6_ECHO_REPLY) {
 543                 uint16_t recv_seq = ntohs(icmppkt->icmp6_seq);
 544                 uint32_t *tp = NULL;
 545
 546                 if (sz >= sizeof(struct icmp6_hdr) + sizeof(uint32_t))
 547                         tp = (uint32_t *) &icmppkt->icmp6_data8[4];
 548                 unpack_tail(sz, tp,
 549                         inet_ntop(AF_INET6, &pingaddr.sin6.sin6_addr,
 550                                         buf, sizeof(buf)),
 551                         recv_seq, hoplimit);
 552         } else if (icmppkt->icmp6_type != ICMP6_ECHO_REQUEST) {
 553                 bb_error_msg("warning: got ICMP %d (%s)",
 554                                 icmppkt->icmp6_type,
 555                                 icmp6_type_name(icmppkt->icmp6_type));
 556         }
 557 }
 558 #endif
 559
 560 static void ping4(len_and_sockaddr *lsa)
 561 {
 562         char packet[datalen + MAXIPLEN + MAXICMPLEN];
 563         int sockopt;
 564
 565         pingsock = create_icmp_socket();
 566         pingaddr.sin = lsa->u.sin;
 567         if (source_lsa) {
 568                 if (setsockopt(pingsock, IPPROTO_IP, IP_MULTICAST_IF,
 569                                 &source_lsa->u.sa, source_lsa->len))
 570                         bb_error_msg_and_die("can't set multicast source interface");
 571                 xbind(pingsock, &source_lsa->u.sa, source_lsa->len);
 572         }
 573         if (opt_I)
 574                 setsockopt(pingsock, SOL_SOCKET, SO_BINDTODEVICE, opt_I, strlen(opt_I) + 1);
 575
 576         /* enable broadcast pings */
 577         setsockopt_broadcast(pingsock);
 578
 579         /* set recv buf for broadcast pings */
 580         sockopt = 48 * 1024; /* explain why 48k? */
 581         setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
 582
 583         signal(SIGINT, print_stats_and_exit);
 584
 585         /* start the ping's going ... */
 586         sendping4(0);
 587
 588         /* listen for replies */
 589         while (1) {
 590                 struct sockaddr_in from;
 591                 socklen_t fromlen = (socklen_t) sizeof(from);
 592                 int c;
 593
 594                 c = recvfrom(pingsock, packet, sizeof(packet), 0,
 595                                 (struct sockaddr *) &from, &fromlen);
 596                 if (c < 0) {
 597                         if (errno != EINTR)
 598                                 bb_perror_msg("recvfrom");
 599                         continue;
 600                 }
 601                 unpack4(packet, c, &from);
 602                 if (pingcount && nreceived >= pingcount)
 603                         break;
 604         }
 605 }
 606 #if ENABLE_PING6
 607 extern int BUG_bad_offsetof_icmp6_cksum(void);
 608 static void ping6(len_and_sockaddr *lsa)
 609 {
 610         char packet[datalen + MAXIPLEN + MAXICMPLEN];
 611         int sockopt;
 612         struct msghdr msg;
 613         struct sockaddr_in6 from;
 614         struct iovec iov;
 615         char control_buf[CMSG_SPACE(36)];
 616
 617         pingsock = create_icmp6_socket();
 618         pingaddr.sin6 = lsa->u.sin6;
 619         /* untested whether "-I addr" really works for IPv6: */
 620         if (source_lsa)
 621                 xbind(pingsock, &source_lsa->u.sa, source_lsa->len);
 622         if (opt_I)
 623                 setsockopt(pingsock, SOL_SOCKET, SO_BINDTODEVICE, opt_I, strlen(opt_I) + 1);
 624
 625 #ifdef ICMP6_FILTER
 626         {
 627                 struct icmp6_filter filt;
 628                 if (!(option_mask32 & OPT_VERBOSE)) {
 629                         ICMP6_FILTER_SETBLOCKALL(&filt);
 630                         ICMP6_FILTER_SETPASS(ICMP6_ECHO_REPLY, &filt);
 631                 } else {
 632                         ICMP6_FILTER_SETPASSALL(&filt);
 633                 }
 634                 if (setsockopt(pingsock, IPPROTO_ICMPV6, ICMP6_FILTER, &filt,
 635                                            sizeof(filt)) < 0)
 636                         bb_error_msg_and_die("setsockopt(ICMP6_FILTER)");
 637         }
 638 #endif /*ICMP6_FILTER*/
 639
 640         /* enable broadcast pings */
 641         setsockopt_broadcast(pingsock);
 642
 643         /* set recv buf for broadcast pings */
 644         sockopt = 48 * 1024; /* explain why 48k? */
 645         setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
 646
 647         sockopt = offsetof(struct icmp6_hdr, icmp6_cksum);
 648         if (offsetof(struct icmp6_hdr, icmp6_cksum) != 2)
 649                 BUG_bad_offsetof_icmp6_cksum();
 650         setsockopt(pingsock, SOL_RAW, IPV6_CHECKSUM, &sockopt, sizeof(sockopt));
 651
 652         /* request ttl info to be returned in ancillary data */
 653         setsockopt(pingsock, SOL_IPV6, IPV6_HOPLIMIT, &const_int_1, sizeof(const_int_1));
 654
 655         if (if_index)
 656                 pingaddr.sin6.sin6_scope_id = if_index;
 657
 658         signal(SIGINT, print_stats_and_exit);
 659
 660         /* start the ping's going ... */
 661         sendping6(0);
 662
 663         /* listen for replies */
 664         msg.msg_name = &from;
 665         msg.msg_namelen = sizeof(from);
 666         msg.msg_iov = &iov;
 667         msg.msg_iovlen = 1;
 668         msg.msg_control = control_buf;
 669         iov.iov_base = packet;
 670         iov.iov_len = sizeof(packet);
 671         while (1) {
 672                 int c;
 673                 struct cmsghdr *mp;
 674                 int hoplimit = -1;
 675                 msg.msg_controllen = sizeof(control_buf);
 676
 677                 c = recvmsg(pingsock, &msg, 0);
 678                 if (c < 0) {
 679                         if (errno != EINTR)
 680                                 bb_perror_msg("recvfrom");
 681                         continue;
 682                 }
 683                 for (mp = CMSG_FIRSTHDR(&msg); mp; mp = CMSG_NXTHDR(&msg, mp)) {
 684                         if (mp->cmsg_level == SOL_IPV6
 685                          && mp->cmsg_type == IPV6_HOPLIMIT
 686                          /* don't check len - we trust the kernel: */
 687                          /* && mp->cmsg_len >= CMSG_LEN(sizeof(int)) */
 688                         ) {
 689                                 hoplimit = *(int*)CMSG_DATA(mp);
 690                         }
 691                 }
 692                 unpack6(packet, c, /*&from,*/ hoplimit);
 693                 if (pingcount && nreceived >= pingcount)
 694                         break;
 695         }
 696 }
 697 #endif
 698
 699 static void ping(len_and_sockaddr *lsa)
 700 {
 701         printf("PING %s (%s)", hostname, dotted);
 702         if (source_lsa) {
 703                 printf(" from %s",
 704                         xmalloc_sockaddr2dotted_noport(&source_lsa->u.sa));
 705         }
 706         printf(": %d data bytes\n", datalen);
 707
 708 #if ENABLE_PING6
 709         if (lsa->u.sa.sa_family == AF_INET6)
 710                 ping6(lsa);
 711         else
 712 #endif
 713                 ping4(lsa);
 714 }
 715
 716 int ping_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 717 int ping_main(int argc ATTRIBUTE_UNUSED, char **argv)
 718 {
 719         len_and_sockaddr *lsa;
 720         char *opt_c, *opt_s;
 721         USE_PING6(sa_family_t af = AF_UNSPEC;)
 722
 723         INIT_G();
 724
 725         /* exactly one argument needed; -v and -q don't mix; -w NUM, -W NUM */
 726         opt_complementary = "=1:q--v:v--q:w+:W+";
 727         getopt32(argv, OPT_STRING, &opt_c, &opt_s, &deadline, &timeout, &opt_I);
 728         if (option_mask32 & OPT_c)
 729                 pingcount = xatoul(opt_c); // -c
 730         if (option_mask32 & OPT_s)
 731                 datalen = xatou16(opt_s); // -s
 732         if (option_mask32 & OPT_I) { // -I
 733                 if_index = if_nametoindex(opt_I);
 734                 if (!if_index) {
 735                         /* TODO: I'm not sure it takes IPv6 unless in [XX:XX..] format */
 736                         source_lsa = xdotted2sockaddr(opt_I, 0);
 737                         opt_I = NULL; /* don't try to bind to device later */
 738                 }
 739         }
 740         myid = (uint16_t) getpid();
 741         hostname = argv[optind];
 742 #if ENABLE_PING6
 743         if (option_mask32 & OPT_IPV4)
 744                 af = AF_INET;
 745         if (option_mask32 & OPT_IPV6)
 746                 af = AF_INET6;
 747         lsa = xhost_and_af2sockaddr(hostname, 0, af);
 748 #else
 749         lsa = xhost_and_af2sockaddr(hostname, 0, AF_INET);
 750 #endif
 751
 752         if (source_lsa && source_lsa->u.sa.sa_family != lsa->u.sa.sa_family)
 753                 /* leaking it here... */
 754                 source_lsa = NULL;
 755
 756         dotted = xmalloc_sockaddr2dotted_noport(&lsa->u.sa);
 757         ping(lsa);
 758         print_stats_and_exit(0);
 759         /*return EXIT_SUCCESS;*/
 760 }
 761 #endif /* FEATURE_FANCY_PING */
 762
 763
 764 #if ENABLE_PING6
 765 int ping6_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 766 int ping6_main(int argc, char **argv)
 767 {
 768         argv[0] = (char*)"-6";
 769         return ping_main(argc + 1, argv - 1);
 770 }
 771 #endif
 772
 773 /* from ping6.c:
 774  * Copyright (c) 1989 The Regents of the University of California.
 775  * All rights reserved.
 776  *
 777  * This code is derived from software contributed to Berkeley by
 778  * Mike Muuss.
 779  *
 780  * Redistribution and use in source and binary forms, with or without
 781  * modification, are permitted provided that the following conditions
 782  * are met:
 783  * 1. Redistributions of source code must retain the above copyright
 784  *    notice, this list of conditions and the following disclaimer.
 785  * 2. Redistributions in binary form must reproduce the above copyright
 786  *    notice, this list of conditions and the following disclaimer in the
 787  *    documentation and/or other materials provided with the distribution.
 788  *
 789  * 3. <BSD Advertising Clause omitted per the July 22, 1999 licensing change
 790  *              ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change>
 791  *
 792  * 4. Neither the name of the University nor the names of its contributors
 793  *    may be used to endorse or promote products derived from this software
 794  *    without specific prior written permission.
 795  *
 796  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 797  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 798  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 799  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 800  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 801  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 802  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 803  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 804  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 805  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 806  * SUCH DAMAGE.
 807  */