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