pscan: size optimization (Tito <farmatito@tiscali.it>)
[oweals/busybox.git] / networking / ping.c
index 4176ab390da6c9b6bdba1b9f9f3466d236a3c23c..c4a498cd8e925b154864978d94f5dfad488ac29e 100644 (file)
+/* vi: set sw=4 ts=4: */
 /*
- * $Id: ping.c,v 1.5 1999/12/09 06:11:36 andersen Exp $
  * Mini ping implementation for busybox
  *
  * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
+ * Adapted from the ping in netkit-base 0.10:
+ * Copyright (c) 1989 The Regents of the University of California.
+ * All rights reserved.
  *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
+ * This code is derived from software contributed to Berkeley by
+ * Mike Muuss.
  *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ */
+/* from ping6.c:
+ * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
  *
  * This version of ping is adapted from the ping in netkit-base 0.10,
  * which is:
  *
- * Copyright (c) 1989 The Regents of the University of California.
- * All rights reserved.
- *
- * This code is derived from software contributed to Berkeley by
- * Mike Muuss.
- * 
  * Original copyright notice is retained at the end of this file.
+ *
+ * This version is an adaptation of ping.c from busybox.
+ * The code was modified by Bart Visscher <magick@linux-fan.com>
  */
 
-#include "internal.h"
-#include <sys/param.h>
-#include <sys/socket.h>
-#include <sys/file.h>
-#include <sys/time.h>
-#include <sys/times.h>
-#include <sys/signal.h>
-
-#include <netinet/in.h>
-#include <netinet/ip.h>
+#include <net/if.h>
 #include <netinet/ip_icmp.h>
-#include <arpa/inet.h>
-#include <netdb.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <errno.h>
-
-#define DEFDATALEN      56
-#define        MAXIPLEN        60
-#define        MAXICMPLEN      76
-#define        MAXPACKET       65468
-#define        MAX_DUP_CHK     (8 * 128)
-#define MAXWAIT         10
-#define PINGINTERVAL    1 /* second */
-
-#define O_QUIET         (1 << 0)
+#include "libbb.h"
+
+#if ENABLE_PING6
+#include <netinet/icmp6.h>
+/* I see RENUMBERED constants in bits/in.h - !!?
+ * What a fuck is going on with libc? Is it a glibc joke? */
+#ifdef IPV6_2292HOPLIMIT
+#undef IPV6_HOPLIMIT
+#define IPV6_HOPLIMIT IPV6_2292HOPLIMIT
+#endif
+#endif
+
+enum {
+       DEFDATALEN = 56,
+       MAXIPLEN = 60,
+       MAXICMPLEN = 76,
+       MAXPACKET = 65468,
+       MAX_DUP_CHK = (8 * 128),
+       MAXWAIT = 10,
+       PINGINTERVAL = 1, /* 1 second */
+};
+
+/* common routines */
+
+static int in_cksum(unsigned short *buf, int sz)
+{
+       int nleft = sz;
+       int sum = 0;
+       unsigned short *w = buf;
+       unsigned short ans = 0;
+
+       while (nleft > 1) {
+               sum += *w++;
+               nleft -= 2;
+       }
+
+       if (nleft == 1) {
+               *(unsigned char *) (&ans) = *(unsigned char *) w;
+               sum += ans;
+       }
+
+       sum = (sum >> 16) + (sum & 0xFFFF);
+       sum += (sum >> 16);
+       ans = ~sum;
+       return ans;
+}
+
+#if !ENABLE_FEATURE_FANCY_PING
+
+/* simple version */
+
+static char *hostname;
+
+static void noresp(int ign ATTRIBUTE_UNUSED)
+{
+       printf("No response from %s\n", hostname);
+       exit(EXIT_FAILURE);
+}
+
+static void ping4(len_and_sockaddr *lsa)
+{
+       struct sockaddr_in pingaddr;
+       struct icmp *pkt;
+       int pingsock, c;
+       char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
+
+       pingsock = create_icmp_socket();
+       pingaddr = lsa->sin;
+
+       pkt = (struct icmp *) packet;
+       memset(pkt, 0, sizeof(packet));
+       pkt->icmp_type = ICMP_ECHO;
+       pkt->icmp_cksum = in_cksum((unsigned short *) pkt, sizeof(packet));
+
+       c = xsendto(pingsock, packet, DEFDATALEN + ICMP_MINLEN,
+                          (struct sockaddr *) &pingaddr, sizeof(pingaddr));
+
+       /* listen for replies */
+       while (1) {
+               struct sockaddr_in from;
+               socklen_t fromlen = sizeof(from);
+
+               c = recvfrom(pingsock, packet, sizeof(packet), 0,
+                               (struct sockaddr *) &from, &fromlen);
+               if (c < 0) {
+                       if (errno != EINTR)
+                               bb_perror_msg("recvfrom");
+                       continue;
+               }
+               if (c >= 76) {                  /* ip + icmp */
+                       struct iphdr *iphdr = (struct iphdr *) packet;
+
+                       pkt = (struct icmp *) (packet + (iphdr->ihl << 2));     /* skip ip hdr */
+                       if (pkt->icmp_type == ICMP_ECHOREPLY)
+                               break;
+               }
+       }
+       if (ENABLE_FEATURE_CLEAN_UP)
+               close(pingsock);
+}
+
+#if ENABLE_PING6
+static void ping6(len_and_sockaddr *lsa)
+{
+       struct sockaddr_in6 pingaddr;
+       struct icmp6_hdr *pkt;
+       int pingsock, c;
+       int sockopt;
+       char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
+
+       pingsock = create_icmp6_socket();
+       pingaddr = lsa->sin6;
+
+       pkt = (struct icmp6_hdr *) packet;
+       memset(pkt, 0, sizeof(packet));
+       pkt->icmp6_type = ICMP6_ECHO_REQUEST;
+
+       sockopt = offsetof(struct icmp6_hdr, icmp6_cksum);
+       setsockopt(pingsock, SOL_RAW, IPV6_CHECKSUM, &sockopt, sizeof(sockopt));
+
+       c = xsendto(pingsock, packet, DEFDATALEN + sizeof (struct icmp6_hdr),
+                          (struct sockaddr *) &pingaddr, sizeof(pingaddr));
+
+       /* listen for replies */
+       while (1) {
+               struct sockaddr_in6 from;
+               socklen_t fromlen = sizeof(from);
+
+               c = recvfrom(pingsock, packet, sizeof(packet), 0,
+                               (struct sockaddr *) &from, &fromlen);
+               if (c < 0) {
+                       if (errno != EINTR)
+                               bb_perror_msg("recvfrom");
+                       continue;
+               }
+               if (c >= 8) {                   /* icmp6_hdr */
+                       pkt = (struct icmp6_hdr *) packet;
+                       if (pkt->icmp6_type == ICMP6_ECHO_REPLY)
+                               break;
+               }
+       }
+       if (ENABLE_FEATURE_CLEAN_UP)
+               close(pingsock);
+}
+#endif
+
+int ping_main(int argc, char **argv);
+int ping_main(int argc, char **argv)
+{
+       len_and_sockaddr *lsa;
+#if ENABLE_PING6
+       sa_family_t af = AF_UNSPEC;
+       while (++argv[0][0] == '-') {
+               if (argv[0][1] == '4') {
+                       af = AF_INET;
+                       continue;
+               }
+               if (argv[0][1] == '6') {
+                       af = AF_INET6;
+                       continue;
+               }
+               bb_show_usage();
+       }
+#else
+       argv++;
+#endif
+
+       hostname = *argv;
+       if (!hostname)
+               bb_show_usage();
+
+#if ENABLE_PING6
+       lsa = xhost_and_af2sockaddr(hostname, 0, af);
+#else
+       lsa = xhost_and_af2sockaddr(hostname, 0, AF_INET);
+#endif
+       /* Set timer _after_ DNS resolution */
+       signal(SIGALRM, noresp);
+       alarm(5); /* give the host 5000ms to respond */
+
+#if ENABLE_PING6
+       if (lsa->sa.sa_family == AF_INET6)
+               ping6(lsa);
+       else
+#endif
+               ping4(lsa);
+       printf("%s is alive!\n", hostname);
+       return EXIT_SUCCESS;
+}
+
+
+#else /* FEATURE_FANCY_PING */
+
+
+/* full(er) version */
+
+#define OPT_STRING ("qvc:s:I:4" USE_PING6("6"))
+enum {
+       OPT_QUIET = 1 << 0,
+       OPT_VERBOSE = 1 << 1,
+       OPT_c = 1 << 2,
+       OPT_s = 1 << 3,
+       OPT_I = 1 << 4,
+       OPT_IPV4 = 1 << 5,
+       OPT_IPV6 = (1 << 6) * ENABLE_PING6,
+};
+
+
+struct globals {
+       int pingsock;
+       len_and_sockaddr *source_lsa;
+       unsigned datalen;
+       int if_index;
+       unsigned long ntransmitted, nreceived, nrepeats, pingcount;
+       uint16_t myid;
+       unsigned tmin, tmax; /* in us */
+       unsigned long long tsum; /* in us, sum of all times */
+       const char *hostname;
+       const char *dotted;
+       union {
+               struct sockaddr sa;
+               struct sockaddr_in sin;
+#if ENABLE_PING6
+               struct sockaddr_in6 sin6;
+#endif
+       } pingaddr;
+       char rcvd_tbl[MAX_DUP_CHK / 8];
+};
+#define G (*(struct globals*)&bb_common_bufsiz1)
+#define pingsock     (G.pingsock    )
+#define source_lsa   (G.source_lsa  )
+#define datalen      (G.datalen     )
+#define if_index     (G.if_index    )
+#define ntransmitted (G.ntransmitted)
+#define nreceived    (G.nreceived   )
+#define nrepeats     (G.nrepeats    )
+#define pingcount    (G.pingcount   )
+#define myid         (G.myid        )
+#define tmin         (G.tmin        )
+#define tmax         (G.tmax        )
+#define tsum         (G.tsum        )
+#define hostname     (G.hostname    )
+#define dotted       (G.dotted      )
+#define pingaddr     (G.pingaddr    )
+#define rcvd_tbl     (G.rcvd_tbl    )
+void BUG_ping_globals_too_big(void);
+#define INIT_G() do { \
+        if (sizeof(G) > COMMON_BUFSIZE) \
+                BUG_ping_globals_too_big(); \
+       pingsock = -1; \
+       tmin = UINT_MAX; \
+} while (0)
+
 
 #define        A(bit)          rcvd_tbl[(bit)>>3]      /* identify byte in array */
 #define        B(bit)          (1 << ((bit) & 0x07))   /* identify bit in byte */
 #define        CLR(bit)        (A(bit) &= (~B(bit)))
 #define        TST(bit)        (A(bit) & B(bit))
 
-static const char* ping_usage = "ping [OPTION]... host\n\n"
-"Send ICMP ECHO_REQUEST packets to network hosts.\n\n"
-"Options:\n"
-"\t-q\t\tQuiet mode, only displays output at start and when finished.\n"
-"\t-c COUNT\tSend only COUNT pings.\n";
-
-static char *hostname = NULL;
-static struct sockaddr_in pingaddr;
-static int pingsock = -1;
-static long ntransmitted = 0, nreceived = 0, nrepeats = 0, pingcount = 0;
-static int myid = 0, options = 0;
-static unsigned long tmin = ULONG_MAX, tmax = 0, tsum = 0;
-static char rcvd_tbl[MAX_DUP_CHK / 8];
-
-static void pingstats(int);
-static void sendping(int);
-static void unpack(char *, int, struct sockaddr_in *);
-static void ping(char *);
-static int in_cksum(unsigned short *, int);
-
 /**************************************************************************/
 
-static int in_cksum(unsigned short *buf, int sz)
+static void pingstats(int junk ATTRIBUTE_UNUSED)
+{
+       signal(SIGINT, SIG_IGN);
+
+       printf("\n--- %s ping statistics ---\n", hostname);
+       printf("%lu packets transmitted, ", ntransmitted);
+       printf("%lu packets received, ", nreceived);
+       if (nrepeats)
+               printf("%lu duplicates, ", nrepeats);
+       if (ntransmitted)
+               ntransmitted = (ntransmitted - nreceived) * 100 / ntransmitted;
+       printf("%lu%% packet loss\n", ntransmitted);
+       if (tmin != UINT_MAX) {
+               unsigned tavg = tsum / (nreceived + nrepeats);
+               printf("round-trip min/avg/max = %u.%03u/%u.%03u/%u.%03u ms\n",
+                       tmin / 1000, tmin % 1000,
+                       tavg / 1000, tavg % 1000,
+                       tmax / 1000, tmax % 1000);
+       }
+       exit(nreceived == 0); /* (nreceived == 0) is true (1) -- 'failure' */
+}
+
+static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt)
 {
-    int nleft = sz;
-    int sum = 0;
-    unsigned short *w = buf;
-    unsigned short ans = 0;
-
-    while (nleft > 1) {
-        sum += *w++;
-        nleft -= 2;
-    }
-
-    if (nleft == 1) { 
-        *(unsigned char *)(&ans) = *(unsigned char *)w;
-        sum += ans;
-    }
-
-    sum = (sum >> 16) + (sum & 0xFFFF);
-    sum += (sum >> 16);
-    ans = ~sum;
-    return(ans);
-}   
-
-static void pingstats(int ign) {
-    signal(SIGINT, SIG_IGN);
-    
-    printf("\n--- %s ping statistics ---\n", hostname);
-    printf("%ld packets transmitted, ", ntransmitted);
-    printf("%ld packets received, ", nreceived);
-    if (nrepeats)
-        printf("%ld duplicates, ", nrepeats);
-    if (ntransmitted)
-       printf("%ld%% packet loss\n", 
-              (ntransmitted - nreceived)*100/ntransmitted);
-    if (nreceived)
-        printf("round-trip min/avg/max = %lu.%lu/%lu.%lu/%lu.%lu ms\n",
-              tmin/10, tmin%10, 
-              (tsum/(nreceived+nrepeats))/10,
-              (tsum/(nreceived+nrepeats))%10,
-              tmax/10, tmax%10);
-    exit(0);
+       int sz;
+
+       CLR((uint16_t)ntransmitted % MAX_DUP_CHK);
+       ntransmitted++;
+
+       /* sizeof(pingaddr) can be larger than real sa size, but I think
+        * it doesn't matter */
+       sz = xsendto(pingsock, pkt, size_pkt, &pingaddr.sa, sizeof(pingaddr));
+       if (sz != size_pkt)
+               bb_error_msg_and_die(bb_msg_write_error);
+
+       signal(SIGALRM, sp);
+       if (pingcount == 0 || ntransmitted < pingcount) { /* schedule next in 1s */
+               alarm(PINGINTERVAL);
+       } else { /* done, wait for the last ping to come back */
+               /* todo, don't necessarily need to wait so long... */
+               signal(SIGALRM, pingstats);
+               alarm(MAXWAIT);
+       }
 }
 
-static void sendping(int ign)
+static void sendping4(int junk ATTRIBUTE_UNUSED)
 {
-    struct icmp *pkt;
-    int i;
-    char packet[DEFDATALEN + 8];
-
-    pkt = (struct icmp *)packet;
-    
-    pkt->icmp_type = ICMP_ECHO;
-    pkt->icmp_code = 0;
-    pkt->icmp_cksum = 0;
-    pkt->icmp_seq = ntransmitted++;
-    pkt->icmp_id = myid; 
-    CLR(pkt->icmp_seq % MAX_DUP_CHK);
-
-    gettimeofday((struct timeval *)&packet[8], NULL);
-    pkt->icmp_cksum = in_cksum((unsigned short *)pkt, sizeof(packet));
-    
-    i = sendto(pingsock, packet, sizeof(packet), 0, 
-              (struct sockaddr *)&pingaddr, sizeof(struct sockaddr_in));
-
-    if (i < 0 || i != sizeof(packet)) {
-        if (i < 0) perror("ping");
-        fprintf(stderr, "ping wrote %d chars; %d expected\n", i, sizeof(packet));
-       exit(1);
-    }
-
-    signal(SIGALRM, sendping);
-    if (pingcount == 0 || ntransmitted < pingcount) { /* schedule next in 1s */
-        alarm(PINGINTERVAL);
-    } else { /* done, wait for the last ping to come back */
-       /* todo, don't necessarily need to wait so long... */
-        signal(SIGALRM, pingstats);
-       alarm(MAXWAIT);
-    }
+       /* +4 reserves a place for timestamp, which may end up sitting
+        * *after* packet. Saves one if() */
+       struct icmp *pkt = alloca(datalen + ICMP_MINLEN + 4);
+
+       pkt->icmp_type = ICMP_ECHO;
+       pkt->icmp_code = 0;
+       pkt->icmp_cksum = 0;
+       pkt->icmp_seq = htons(ntransmitted); /* don't ++ here, it can be a macro */
+       pkt->icmp_id = myid;
+
+       /* We don't do hton, because we will read it back on the same machine */
+       /*if (datalen >= 4)*/
+               *(uint32_t*)&pkt->icmp_dun = monotonic_us();
+
+       pkt->icmp_cksum = in_cksum((unsigned short *) pkt, datalen + ICMP_MINLEN);
+
+       sendping_tail(sendping4, pkt, datalen + ICMP_MINLEN);
 }
-    
-static void unpack(char *buf, int sz, struct sockaddr_in *from)
+#if ENABLE_PING6
+static void sendping6(int junk ATTRIBUTE_UNUSED)
 {
-    struct icmp *icmppkt;
-    struct iphdr *iphdr;
-    struct timeval tv, *tp;
-    int hlen, dupflag;
-    unsigned long triptime;
-    
-    gettimeofday(&tv, NULL);
-    
-    /* check IP header */
-    iphdr = (struct iphdr *)buf;
-    hlen = iphdr->ihl << 2;
-    /* discard if too short */
-    if (sz < (DEFDATALEN + ICMP_MINLEN)) return;
-
-    sz -= hlen;
-    icmppkt = (struct icmp *)(buf + hlen);
-    
-    if (icmppkt->icmp_type == ICMP_ECHOREPLY) {
-        if (icmppkt->icmp_id != myid) return; /* not our ping */
+       struct icmp6_hdr *pkt = alloca(datalen + sizeof(struct icmp6_hdr) + 4);
+
+       pkt->icmp6_type = ICMP6_ECHO_REQUEST;
+       pkt->icmp6_code = 0;
+       pkt->icmp6_cksum = 0;
+       pkt->icmp6_seq = htons(ntransmitted); /* don't ++ here, it can be a macro */
+       pkt->icmp6_id = myid;
+
+       /*if (datalen >= 4)*/
+               *(uint32_t*)(&pkt->icmp6_data8[4]) = monotonic_us();
+
+       sendping_tail(sendping6, pkt, datalen + sizeof(struct icmp6_hdr));
+}
+#endif
+
+static const char *icmp_type_name(int id)
+{
+       switch (id) {
+       case ICMP_ECHOREPLY:      return "Echo Reply";
+       case ICMP_DEST_UNREACH:   return "Destination Unreachable";
+       case ICMP_SOURCE_QUENCH:  return "Source Quench";
+       case ICMP_REDIRECT:       return "Redirect (change route)";
+       case ICMP_ECHO:           return "Echo Request";
+       case ICMP_TIME_EXCEEDED:  return "Time Exceeded";
+       case ICMP_PARAMETERPROB:  return "Parameter Problem";
+       case ICMP_TIMESTAMP:      return "Timestamp Request";
+       case ICMP_TIMESTAMPREPLY: return "Timestamp Reply";
+       case ICMP_INFO_REQUEST:   return "Information Request";
+       case ICMP_INFO_REPLY:     return "Information Reply";
+       case ICMP_ADDRESS:        return "Address Mask Request";
+       case ICMP_ADDRESSREPLY:   return "Address Mask Reply";
+       default:                  return "unknown ICMP type";
+       }
+}
+#if ENABLE_PING6
+/* RFC3542 changed some definitions from RFC2292 for no good reason, whee!
+ * the newer 3542 uses a MLD_ prefix where as 2292 uses ICMP6_ prefix */
+#ifndef MLD_LISTENER_QUERY
+# define MLD_LISTENER_QUERY ICMP6_MEMBERSHIP_QUERY
+#endif
+#ifndef MLD_LISTENER_REPORT
+# define MLD_LISTENER_REPORT ICMP6_MEMBERSHIP_REPORT
+#endif
+#ifndef MLD_LISTENER_REDUCTION
+# define MLD_LISTENER_REDUCTION ICMP6_MEMBERSHIP_REDUCTION
+#endif
+static const char *icmp6_type_name(int id)
+{
+       switch (id) {
+       case ICMP6_DST_UNREACH:      return "Destination Unreachable";
+       case ICMP6_PACKET_TOO_BIG:   return "Packet too big";
+       case ICMP6_TIME_EXCEEDED:    return "Time Exceeded";
+       case ICMP6_PARAM_PROB:       return "Parameter Problem";
+       case ICMP6_ECHO_REPLY:       return "Echo Reply";
+       case ICMP6_ECHO_REQUEST:     return "Echo Request";
+       case MLD_LISTENER_QUERY:     return "Listener Query";
+       case MLD_LISTENER_REPORT:    return "Listener Report";
+       case MLD_LISTENER_REDUCTION: return "Listener Reduction";
+       default:                     return "unknown ICMP type";
+       }
+}
+#endif
+
+static void unpack_tail(int sz, uint32_t *tp,
+               const char *from_str,
+               uint16_t recv_seq, int ttl)
+{
+       const char *dupmsg = " (DUP!)";
+       unsigned triptime = triptime; /* for gcc */
+
        ++nreceived;
-       tp = (struct timeval *)icmppkt->icmp_data;
 
-       if ((tv.tv_usec -= tp->tv_usec) < 0) {
-           --tv.tv_sec;
-           tv.tv_usec += 1000000;
+       if (tp) {
+               /* (int32_t) cast is for hypothetical 64-bit unsigned */
+               /* (doesn't hurt 32-bit real-world anyway) */
+               triptime = (int32_t) ((uint32_t)monotonic_us() - *tp);
+               tsum += triptime;
+               if (triptime < tmin)
+                       tmin = triptime;
+               if (triptime > tmax)
+                       tmax = triptime;
        }
-       tv.tv_sec -= tp->tv_sec;
-       
-       triptime = tv.tv_sec * 10000 + (tv.tv_usec / 100);
-       tsum += triptime;
-       if (triptime < tmin) tmin = triptime;
-       if (triptime > tmax) tmax = triptime;
-
-       if (TST(icmppkt->icmp_seq % MAX_DUP_CHK)) {
-           ++nrepeats;
-           --nreceived;
-           dupflag = 1;
+
+       if (TST(recv_seq % MAX_DUP_CHK)) {
+               ++nrepeats;
+               --nreceived;
        } else {
-           SET(icmppkt->icmp_seq % MAX_DUP_CHK);
-           dupflag = 0;
+               SET(recv_seq % MAX_DUP_CHK);
+               dupmsg += 7;
        }
 
-       if (options & O_QUIET) return;
-
-       printf("%d bytes from %s: icmp_seq=%u", sz, 
-              inet_ntoa(*(struct in_addr *)&from->sin_addr.s_addr),
-              icmppkt->icmp_seq);
-       printf(" ttl=%d", iphdr->ttl);
-       printf(" time=%lu.%lu ms", triptime/10, triptime%10);
-       if (dupflag) printf(" (DUP!)");
-        printf("\n");
-    } else {
-        fprintf(stderr, "Warning: unknown ICMP packet received (not echo-reply)\n");
-    }  
+       if (option_mask32 & OPT_QUIET)
+               return;
+
+       printf("%d bytes from %s: seq=%u ttl=%d", sz,
+               from_str, recv_seq, ttl);
+       if (tp)
+               printf(" time=%u.%03u ms", triptime / 1000, triptime % 1000);
+       puts(dupmsg);
+       fflush(stdout);
 }
+static void unpack4(char *buf, int sz, struct sockaddr_in *from)
+{
+       struct icmp *icmppkt;
+       struct iphdr *iphdr;
+       int hlen;
+
+       /* discard if too short */
+       if (sz < (datalen + ICMP_MINLEN))
+               return;
+
+       /* check IP header */
+       iphdr = (struct iphdr *) buf;
+       hlen = iphdr->ihl << 2;
+       sz -= hlen;
+       icmppkt = (struct icmp *) (buf + hlen);
+       if (icmppkt->icmp_id != myid)
+               return;                         /* not our ping */
+
+       if (icmppkt->icmp_type == ICMP_ECHOREPLY) {
+               uint16_t recv_seq = ntohs(icmppkt->icmp_seq);
+               uint32_t *tp = NULL;
+
+               if (sz >= ICMP_MINLEN + sizeof(uint32_t))
+                       tp = (uint32_t *) icmppkt->icmp_data;
+               unpack_tail(sz, tp,
+                       inet_ntoa(*(struct in_addr *) &from->sin_addr.s_addr),
+                       recv_seq, iphdr->ttl);
+       } else if (icmppkt->icmp_type != ICMP_ECHO) {
+               bb_error_msg("warning: got ICMP %d (%s)",
+                               icmppkt->icmp_type,
+                               icmp_type_name(icmppkt->icmp_type));
+       }
+}
+#if ENABLE_PING6
+static void unpack6(char *packet, int sz, struct sockaddr_in6 *from, int hoplimit)
+{
+       struct icmp6_hdr *icmppkt;
+       char buf[INET6_ADDRSTRLEN];
+
+       /* discard if too short */
+       if (sz < (datalen + sizeof(struct icmp6_hdr)))
+               return;
+
+       icmppkt = (struct icmp6_hdr *) packet;
+       if (icmppkt->icmp6_id != myid)
+               return;                         /* not our ping */
+
+       if (icmppkt->icmp6_type == ICMP6_ECHO_REPLY) {
+               uint16_t recv_seq = ntohs(icmppkt->icmp6_seq);
+               uint32_t *tp = NULL;
+
+               if (sz >= sizeof(struct icmp6_hdr) + sizeof(uint32_t))
+                       tp = (uint32_t *) &icmppkt->icmp6_data8[4];
+               unpack_tail(sz, tp,
+                       inet_ntop(AF_INET6, &pingaddr.sin6.sin6_addr,
+                                       buf, sizeof(buf)),
+                       recv_seq, hoplimit);
+       } else if (icmppkt->icmp6_type != ICMP6_ECHO_REQUEST) {
+               bb_error_msg("warning: got ICMP %d (%s)",
+                               icmppkt->icmp6_type,
+                               icmp6_type_name(icmppkt->icmp6_type));
+       }
+}
+#endif
 
-static void ping(char *host)
+static void ping4(len_and_sockaddr *lsa)
 {
-    struct protoent *proto;
-    struct hostent *h;
-    char buf[MAXHOSTNAMELEN];
-    char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
-    int sockopt;
-    
-    if (!(proto = getprotobyname("icmp"))) {
-        fprintf(stderr, "ping: unknown protocol icmp\n");
-       exit(1);
-    }
-    if ((pingsock = socket(AF_INET, SOCK_RAW, proto->p_proto)) < 0) {
-       if (errno == EPERM) {
-           fprintf(stderr, "ping: permission denied. (are you root?)\n");
-       } else {
-           perror("ping");
+       char packet[datalen + MAXIPLEN + MAXICMPLEN];
+       int sockopt;
+
+       pingsock = create_icmp_socket();
+       pingaddr.sin = lsa->sin;
+       if (source_lsa)
+               xbind(pingsock, &lsa->sa, lsa->len);
+
+       /* enable broadcast pings */
+       setsockopt_broadcast(pingsock);
+
+       /* set recv buf for broadcast pings */
+       sockopt = 48 * 1024; /* explain why 48k? */
+       setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
+
+       signal(SIGINT, pingstats);
+
+       /* start the ping's going ... */
+       sendping4(0);
+
+       /* listen for replies */
+       while (1) {
+               struct sockaddr_in from;
+               socklen_t fromlen = (socklen_t) sizeof(from);
+               int c;
+
+               c = recvfrom(pingsock, packet, sizeof(packet), 0,
+                               (struct sockaddr *) &from, &fromlen);
+               if (c < 0) {
+                       if (errno != EINTR)
+                               bb_perror_msg("recvfrom");
+                       continue;
+               }
+               unpack4(packet, c, &from);
+               if (pingcount > 0 && nreceived >= pingcount)
+                       break;
        }
-       exit(1);
-    }
-
-    /* drop root privs if running setuid */
-    setuid(getuid());
-    
-    memset(&pingaddr, 0, sizeof(struct sockaddr_in));
-    pingaddr.sin_family = AF_INET;
-    if (inet_aton(host, &pingaddr.sin_addr)) {
-        hostname = host;
-    } else {    
-        if (!(h = gethostbyname(host))) {
-            fprintf(stderr, "ping: unknown host %s\n", host);
-           exit(1);
-        }
-
-       if (h->h_addrtype != AF_INET) {
-           fprintf(stderr, "ping: unknown address type; only AF_INET is currently supported.\n");
-           exit(1);
+}
+#if ENABLE_PING6
+extern int BUG_bad_offsetof_icmp6_cksum(void);
+static void ping6(len_and_sockaddr *lsa)
+{
+       char packet[datalen + MAXIPLEN + MAXICMPLEN];
+       int sockopt;
+       struct msghdr msg;
+       struct sockaddr_in6 from;
+       struct iovec iov;
+       char control_buf[CMSG_SPACE(36)];
+
+       pingsock = create_icmp6_socket();
+       pingaddr.sin6 = lsa->sin6;
+       /* untested whether "-I addr" really works for IPv6: */
+       if (source_lsa)
+               xbind(pingsock, &lsa->sa, lsa->len);
+
+#ifdef ICMP6_FILTER
+       {
+               struct icmp6_filter filt;
+               if (!(option_mask32 & OPT_VERBOSE)) {
+                       ICMP6_FILTER_SETBLOCKALL(&filt);
+                       ICMP6_FILTER_SETPASS(ICMP6_ECHO_REPLY, &filt);
+               } else {
+                       ICMP6_FILTER_SETPASSALL(&filt);
+               }
+               if (setsockopt(pingsock, IPPROTO_ICMPV6, ICMP6_FILTER, &filt,
+                                          sizeof(filt)) < 0)
+                       bb_error_msg_and_die("setsockopt(ICMP6_FILTER)");
        }
-       
-       pingaddr.sin_family = AF_INET; /* h->h_addrtype */
-        memcpy(&pingaddr.sin_addr, h->h_addr, sizeof(pingaddr.sin_addr)); 
-        strncpy(buf, h->h_name, sizeof(buf)-1);
-       hostname = buf;
-    }
-
-    /* enable broadcast pings */
-    sockopt = 1;
-    setsockopt(pingsock, SOL_SOCKET, SO_BROADCAST, (char *)&sockopt, sizeof(sockopt));
-    
-    /* set recv buf for broadcast pings */
-    sockopt = 48 * 1024;
-    setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, (char *)&sockopt, sizeof(sockopt));
-
-    printf("PING %s (%s): %d data bytes\n", 
-           hostname, inet_ntoa(*(struct in_addr *)&pingaddr.sin_addr.s_addr),
-          DEFDATALEN);
-
-    signal(SIGINT, pingstats);
-    
-    /* start the ping's going ... */
-    sendping(0);
-       
-    /* listen for replies */
-    while (1) {
-        struct sockaddr_in from;
-       size_t fromlen = sizeof(from);
-        int c;
-
-        if ((c = recvfrom(pingsock, packet, sizeof(packet), 0,
-                          (struct sockaddr *)&from, &fromlen)) < 0) {
-            if (errno == EINTR) continue;
-           perror("ping");
-           continue;
+#endif /*ICMP6_FILTER*/
+
+       /* enable broadcast pings */
+       setsockopt_broadcast(pingsock);
+
+       /* set recv buf for broadcast pings */
+       sockopt = 48 * 1024; /* explain why 48k? */
+       setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
+
+       sockopt = offsetof(struct icmp6_hdr, icmp6_cksum);
+       if (offsetof(struct icmp6_hdr, icmp6_cksum) != 2)
+               BUG_bad_offsetof_icmp6_cksum();
+       setsockopt(pingsock, SOL_RAW, IPV6_CHECKSUM, &sockopt, sizeof(sockopt));
+
+       /* request ttl info to be returned in ancillary data */
+       setsockopt(pingsock, SOL_IPV6, IPV6_HOPLIMIT, &const_int_1, sizeof(const_int_1));
+
+       if (if_index)
+               pingaddr.sin6.sin6_scope_id = if_index;
+
+       signal(SIGINT, pingstats);
+
+       /* start the ping's going ... */
+       sendping6(0);
+
+       /* listen for replies */
+       msg.msg_name = &from;
+       msg.msg_namelen = sizeof(from);
+       msg.msg_iov = &iov;
+       msg.msg_iovlen = 1;
+       msg.msg_control = control_buf;
+       iov.iov_base = packet;
+       iov.iov_len = sizeof(packet);
+       while (1) {
+               int c;
+               struct cmsghdr *mp;
+               int hoplimit = -1;
+               msg.msg_controllen = sizeof(control_buf);
+
+               c = recvmsg(pingsock, &msg, 0);
+               if (c < 0) {
+                       if (errno != EINTR)
+                               bb_perror_msg("recvfrom");
+                       continue;
+               }
+               for (mp = CMSG_FIRSTHDR(&msg); mp; mp = CMSG_NXTHDR(&msg, mp)) {
+                       if (mp->cmsg_level == SOL_IPV6
+                        && mp->cmsg_type == IPV6_HOPLIMIT
+                        /* don't check len - we trust the kernel: */
+                        /* && mp->cmsg_len >= CMSG_LEN(sizeof(int)) */
+                       ) {
+                               hoplimit = *(int*)CMSG_DATA(mp);
+                       }
+               }
+               unpack6(packet, c, &from, hoplimit);
+               if (pingcount > 0 && nreceived >= pingcount)
+                       break;
        }
-       unpack(packet, c, &from);
-       if (pingcount > 0 && nreceived >= pingcount) break;     
-    }
-    pingstats(0);
 }
+#endif
 
-extern int ping_main(int argc, char **argv)
+static void ping(len_and_sockaddr *lsa)
 {
-    char *thisarg;
-    
-    argc--;
-    argv++;
-    options = 0;
-    /* Parse any options */
-    while (argc > 1) {
-        if (**argv != '-') usage(ping_usage);
-        thisarg = *argv; thisarg++;
-        switch (*thisarg) {
-           case 'q': options |= O_QUIET; break;
-           case 'c':
-               argc--; argv++;
-               pingcount = atoi(*argv);                
-               break;
-           default:
-               usage(ping_usage);
-        }
-        argc--; argv++;
-    }
-    if (argc < 1) usage(ping_usage);       
-
-    myid = getpid() & 0xFFFF;
-    ping(*argv);
-    exit(TRUE);
+       printf("PING %s (%s)", hostname, dotted);
+       if (source_lsa) {
+               printf(" from %s",
+                       xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len));
+       }
+       printf(": %d data bytes\n", datalen);
+
+#if ENABLE_PING6
+       if (lsa->sa.sa_family == AF_INET6)
+               ping6(lsa);
+       else
+#endif
+               ping4(lsa);
 }
 
-/*
+int ping_main(int argc, char **argv);
+int ping_main(int argc, char **argv)
+{
+       len_and_sockaddr *lsa;
+       char *opt_c, *opt_s, *opt_I;
+       USE_PING6(sa_family_t af = AF_UNSPEC;)
+
+       INIT_G();
+
+       datalen = DEFDATALEN; /* initialized here rather than in global scope to work around gcc bug */
+
+       /* exactly one argument needed, -v and -q don't mix */
+       opt_complementary = "=1:q--v:v--q";
+       getopt32(argc, argv, OPT_STRING, &opt_c, &opt_s, &opt_I);
+       if (option_mask32 & OPT_c) pingcount = xatoul(opt_c); // -c
+       if (option_mask32 & OPT_s) datalen = xatou16(opt_s); // -s
+       if (option_mask32 & OPT_I) { // -I
+               if_index = if_nametoindex(opt_I);
+               if (!if_index) {
+                       /* TODO: I'm not sure it takes IPv6 unless in [XX:XX..] format */
+                       /* (ping doesn't support source IPv6 addresses yet anyway) */
+                       source_lsa = xdotted2sockaddr(opt_I, 0);
+               }
+       }
+       myid = (uint16_t) getpid();
+       hostname = argv[optind];
+#if ENABLE_PING6
+       if (option_mask32 & OPT_IPV4)
+               af = AF_INET;
+       if (option_mask32 & OPT_IPV6)
+               af = AF_INET6;
+       lsa = xhost_and_af2sockaddr(hostname, 0, af);
+#else
+       lsa = xhost_and_af2sockaddr(hostname, 0, AF_INET);
+#endif
+
+       if (source_lsa && source_lsa->sa.sa_family != lsa->sa.sa_family)
+               /* leaking it here... */
+               source_lsa = NULL;
+
+       dotted = xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len);
+       ping(lsa);
+       pingstats(0);
+       return EXIT_SUCCESS;
+}
+#endif /* FEATURE_FANCY_PING */
+
+
+#if ENABLE_PING6
+int ping6_main(int argc, char **argv);
+int ping6_main(int argc, char **argv)
+{
+       argv[0] = (char*)"-6";
+       return ping_main(argc + 1, argv - 1);
+}
+#endif
+
+/* from ping6.c:
  * Copyright (c) 1989 The Regents of the University of California.
  * All rights reserved.
  *
@@ -347,10 +743,10 @@ extern int ping_main(int argc, char **argv)
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
- * 3. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *     This product includes software developed by the University of
- *     California, Berkeley and its contributors.
+ *
+ * 3. <BSD Advertising Clause omitted per the July 22, 1999 licensing change
+ *             ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change>
+ *
  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
@@ -367,5 +763,3 @@ extern int ping_main(int argc, char **argv)
  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  * SUCH DAMAGE.
  */
-
-