Remove code for protocols we don't properly support. (Most of this could
[oweals/busybox.git] / networking / ping.c
index c9cf5ffb4e553cde3e53065557fab340c46034e9..7fe52ab39e005408bdcf8afcf46ebfe329554a0a 100644 (file)
@@ -1,43 +1,23 @@
 /* vi: set sw=4 ts=4: */
 /*
- * $Id: ping.c,v 1.11 2000/02/08 19:58:47 erik Exp $
+ * $Id: ping.c,v 1.56 2004/03/15 08:28:48 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.
- *
- * 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.
- *
- * 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
- *
- * This version of ping is adapted from the ping in netkit-base 0.10,
- * which is:
- *
+ * Adapted from the ping in netkit-base 0.10:
  * Copyright (c) 1989 The Regents of the University of California.
- * All rights reserved.
+ * Derived from software contributed to Berkeley by Mike Muuss.
  *
- * This code is derived from software contributed to Berkeley by
- * Mike Muuss.
- * 
- * Original copyright notice is retained at the end of this file.
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
-#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 <signal.h>
 
 #include <netinet/in.h>
 #include <netinet/ip.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <errno.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdlib.h>
+#include "busybox.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 */
+enum {
+       DEFDATALEN = 56,
+       MAXIPLEN = 60,
+       MAXICMPLEN = 76,
+       MAXPACKET = 65468,
+       MAX_DUP_CHK = (8 * 128),
+       MAXWAIT = 10,
+       PINGINTERVAL = 1                /* second */
+};
 
 #define O_QUIET         (1 << 0)
 
@@ -64,6 +51,8 @@
 #define        CLR(bit)        (A(bit) &= (~B(bit)))
 #define        TST(bit)        (A(bit) & B(bit))
 
+static void ping(const char *host);
+
 /* common routines */
 static int in_cksum(unsigned short *buf, int sz)
 {
@@ -89,18 +78,15 @@ static int in_cksum(unsigned short *buf, int sz)
 }
 
 /* simple version */
-#ifdef BB_SIMPLE_PING
-static const char *ping_usage = "ping host\n\n";
-
+#ifndef CONFIG_FEATURE_FANCY_PING
 static char *hostname = NULL;
-
 static void noresp(int ign)
 {
        printf("No response from %s\n", hostname);
-       exit(0);
+       exit(EXIT_FAILURE);
 }
 
-static int ping(const char *host)
+static void ping(const char *host)
 {
        struct hostent *h;
        struct sockaddr_in pingaddr;
@@ -108,21 +94,12 @@ static int ping(const char *host)
        int pingsock, c;
        char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
 
-       if ((pingsock = socket(AF_INET, SOCK_RAW, 1)) < 0) {    /* 1 == ICMP */
-               perror("ping");
-               exit(1);
-       }
-
-       /* drop root privs if running setuid */
-       setuid(getuid());
+       pingsock = create_icmp_socket();
 
        memset(&pingaddr, 0, sizeof(struct sockaddr_in));
 
        pingaddr.sin_family = AF_INET;
-       if (!(h = gethostbyname(host))) {
-               fprintf(stderr, "ping: unknown host %s\n", host);
-               exit(1);
-       }
+       h = xgethostbyname(host);
        memcpy(&pingaddr.sin_addr, h->h_addr, sizeof(pingaddr.sin_addr));
        hostname = h->h_name;
 
@@ -131,28 +108,24 @@ static int ping(const char *host)
        pkt->icmp_type = ICMP_ECHO;
        pkt->icmp_cksum = in_cksum((unsigned short *) pkt, sizeof(packet));
 
-       c = sendto(pingsock, packet, sizeof(packet), 0,
+       c = sendto(pingsock, packet, DEFDATALEN + ICMP_MINLEN, 0,
                           (struct sockaddr *) &pingaddr, sizeof(struct sockaddr_in));
 
-       if (c < 0 || c != sizeof(packet)) {
-               if (c < 0)
-                       perror("ping");
-               fprintf(stderr, "ping: write incomplete\n");
-               exit(1);
-       }
+       if (c < 0 || c != sizeof(packet))
+               bb_perror_msg_and_die("sendto");
 
        signal(SIGALRM, noresp);
        alarm(5);                                       /* give the host 5000ms to respond */
        /* listen for replies */
        while (1) {
                struct sockaddr_in from;
-               size_t fromlen = sizeof(from);
+               socklen_t fromlen = sizeof(from);
 
                if ((c = recvfrom(pingsock, packet, sizeof(packet), 0,
                                                  (struct sockaddr *) &from, &fromlen)) < 0) {
                        if (errno == EINTR)
                                continue;
-                       perror("ping");
+                       bb_perror_msg("recvfrom");
                        continue;
                }
                if (c >= 76) {                  /* ip + icmp */
@@ -164,50 +137,48 @@ static int ping(const char *host)
                }
        }
        printf("%s is alive!\n", hostname);
-       return (TRUE);
+       return;
 }
 
-extern int ping_main(int argc, char **argv)
+int ping_main(int argc, char **argv)
 {
        argc--;
        argv++;
        if (argc < 1)
-               usage(ping_usage);
+               bb_show_usage();
        ping(*argv);
-       exit(TRUE);
+       return EXIT_SUCCESS;
 }
 
-#else
+#else /* ! CONFIG_FEATURE_FANCY_PING */
 /* full(er) version */
-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"
-
-       "\t\t\tand 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 int datalen; /* intentionally uninitialized to work around gcc bug */
 
-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 long ntransmitted, nreceived, nrepeats, pingcount;
+static int myid, options;
+static unsigned long tmin = ULONG_MAX, tmax, tsum;
 static char rcvd_tbl[MAX_DUP_CHK / 8];
 
+#ifndef CONFIG_FEATURE_FANCY_PING6
+static
+#endif
+       struct hostent *hostent;
+
 static void sendping(int);
 static void pingstats(int);
 static void unpack(char *, int, struct sockaddr_in *);
 
-static void ping(char *);
-
 /**************************************************************************/
 
-static void pingstats(int ign)
+static void pingstats(int junk)
 {
+       int status;
+
        signal(SIGINT, SIG_IGN);
 
-       printf("\n--- %s ping statistics ---\n", hostname);
+       printf("\n--- %s ping statistics ---\n", hostent->h_name);
        printf("%ld packets transmitted, ", ntransmitted);
        printf("%ld packets received, ", nreceived);
        if (nrepeats)
@@ -220,37 +191,39 @@ static void pingstats(int ign)
                           tmin / 10, tmin % 10,
                           (tsum / (nreceived + nrepeats)) / 10,
                           (tsum / (nreceived + nrepeats)) % 10, tmax / 10, tmax % 10);
-       exit(0);
+       if (nreceived != 0)
+               status = EXIT_SUCCESS;
+       else
+               status = EXIT_FAILURE;
+       exit(status);
 }
 
-static void sendping(int ign)
+static void sendping(int junk)
 {
        struct icmp *pkt;
        int i;
-       char packet[DEFDATALEN + 8];
+       char packet[datalen + ICMP_MINLEN];
 
        pkt = (struct icmp *) packet;
 
        pkt->icmp_type = ICMP_ECHO;
        pkt->icmp_code = 0;
        pkt->icmp_cksum = 0;
-       pkt->icmp_seq = ntransmitted++;
+       pkt->icmp_seq = htons(ntransmitted++);
        pkt->icmp_id = myid;
-       CLR(pkt->icmp_seq % MAX_DUP_CHK);
+       CLR(ntohs(pkt->icmp_seq) % MAX_DUP_CHK);
 
-       gettimeofday((struct timeval *) &packet[8], NULL);
+       gettimeofday((struct timeval *) &pkt->icmp_dun, 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);
-       }
+       if (i < 0)
+               bb_perror_msg_and_die("sendto");
+       else if ((size_t)i != sizeof(packet))
+               bb_error_msg_and_die("ping wrote %d chars; %d expected", i,
+                          (int)sizeof(packet));
 
        signal(SIGALRM, sendping);
        if (pingcount == 0 || ntransmitted < pingcount) {       /* schedule next in 1s */
@@ -262,6 +235,26 @@ static void sendping(int ign)
        }
 }
 
+static 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";
+       }
+}
+
 static void unpack(char *buf, int sz, struct sockaddr_in *from)
 {
        struct icmp *icmppkt;
@@ -276,16 +269,18 @@ static void unpack(char *buf, int sz, struct sockaddr_in *from)
        iphdr = (struct iphdr *) buf;
        hlen = iphdr->ihl << 2;
        /* discard if too short */
-       if (sz < (DEFDATALEN + ICMP_MINLEN))
+       if (sz < (datalen + ICMP_MINLEN))
                return;
 
        sz -= hlen;
        icmppkt = (struct icmp *) (buf + hlen);
 
+       if (icmppkt->icmp_id != myid)
+           return;                             /* not our ping */
+
        if (icmppkt->icmp_type == ICMP_ECHOREPLY) {
-               if (icmppkt->icmp_id != myid)
-                       return;                         /* not our ping */
-               ++nreceived;
+               u_int16_t recv_seq = ntohs(icmppkt->icmp_seq);
+           ++nreceived;
                tp = (struct timeval *) icmppkt->icmp_data;
 
                if ((tv.tv_usec -= tp->tv_usec) < 0) {
@@ -301,12 +296,12 @@ static void unpack(char *buf, int sz, struct sockaddr_in *from)
                if (triptime > tmax)
                        tmax = triptime;
 
-               if (TST(icmppkt->icmp_seq % MAX_DUP_CHK)) {
+               if (TST(recv_seq % MAX_DUP_CHK)) {
                        ++nrepeats;
                        --nreceived;
                        dupflag = 1;
                } else {
-                       SET(icmppkt->icmp_seq % MAX_DUP_CHK);
+                       SET(recv_seq % MAX_DUP_CHK);
                        dupflag = 0;
                }
 
@@ -315,60 +310,34 @@ static void unpack(char *buf, int sz, struct sockaddr_in *from)
 
                printf("%d bytes from %s: icmp_seq=%u", sz,
                           inet_ntoa(*(struct in_addr *) &from->sin_addr.s_addr),
-                          icmppkt->icmp_seq);
+                          recv_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");
-       }
+       } else
+               if (icmppkt->icmp_type != ICMP_ECHO)
+                       bb_error_msg("Warning: Got ICMP %d (%s)",
+                                       icmppkt->icmp_type, icmp_type_name (icmppkt->icmp_type));
+       fflush(stdout);
 }
 
-static void ping(char *host)
+static void ping(const char *host)
 {
-       struct protoent *proto;
-       struct hostent *h;
-       char buf[MAXHOSTNAMELEN];
-       char packet[DEFDATALEN + MAXIPLEN + MAXICMPLEN];
+       char packet[datalen + MAXIPLEN + MAXICMPLEN];
        int sockopt;
 
-       proto = getprotobyname("icmp");
-       /* if getprotobyname failed, just silently force 
-        * proto->p_proto to have the correct value for "icmp" */
-       if ((pingsock = socket(AF_INET, SOCK_RAW,
-                                                  (proto ? proto->p_proto : 1))) < 0) {        /* 1 == ICMP */
-               if (errno == EPERM) {
-                       fprintf(stderr, "ping: permission denied. (are you root?)\n");
-               } else {
-                       perror("ping");
-               }
-               exit(1);
-       }
-
-       /* drop root privs if running setuid */
-       setuid(getuid());
+       pingsock = create_icmp_socket();
 
        memset(&pingaddr, 0, sizeof(struct sockaddr_in));
 
        pingaddr.sin_family = AF_INET;
-       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);
-       }
+       hostent = xgethostbyname(host);
+       if (hostent->h_addrtype != AF_INET)
+               bb_error_msg_and_die("unknown address type; only AF_INET is currently supported.");
 
-       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;
+       memcpy(&pingaddr.sin_addr, hostent->h_addr, sizeof(pingaddr.sin_addr));
 
        /* enable broadcast pings */
        sockopt = 1;
@@ -381,9 +350,9 @@ static void ping(char *host)
                           sizeof(sockopt));
 
        printf("PING %s (%s): %d data bytes\n",
-                  hostname,
+                  hostent->h_name,
                   inet_ntoa(*(struct in_addr *) &pingaddr.sin_addr.s_addr),
-                  DEFDATALEN);
+                  datalen);
 
        signal(SIGINT, pingstats);
 
@@ -393,14 +362,14 @@ static void ping(char *host)
        /* listen for replies */
        while (1) {
                struct sockaddr_in from;
-               size_t fromlen = sizeof(from);
+               socklen_t fromlen = (socklen_t) sizeof(from);
                int c;
 
                if ((c = recvfrom(pingsock, packet, sizeof(packet), 0,
                                                  (struct sockaddr *) &from, &fromlen)) < 0) {
                        if (errno == EINTR)
                                continue;
-                       perror("ping");
+                       bb_perror_msg("recvfrom");
                        continue;
                }
                unpack(packet, c, &from);
@@ -410,17 +379,17 @@ static void ping(char *host)
        pingstats(0);
 }
 
-extern int ping_main(int argc, char **argv)
+int ping_main(int argc, char **argv)
 {
        char *thisarg;
 
+       datalen = DEFDATALEN; /* initialized here rather than in global scope to work around gcc bug */
+
        argc--;
        argv++;
        options = 0;
        /* Parse any options */
-       while (argc > 1) {
-               if (**argv != '-')
-                       usage(ping_usage);
+       while (argc >= 1 && **argv == '-') {
                thisarg = *argv;
                thisarg++;
                switch (*thisarg) {
@@ -428,57 +397,28 @@ extern int ping_main(int argc, char **argv)
                        options |= O_QUIET;
                        break;
                case 'c':
-                       argc--;
+                       if (--argc <= 0)
+                               bb_show_usage();
                        argv++;
                        pingcount = atoi(*argv);
                        break;
+               case 's':
+                       if (--argc <= 0)
+                               bb_show_usage();
+                       argv++;
+                       datalen = atoi(*argv);
+                       break;
                default:
-                       usage(ping_usage);
+                       bb_show_usage();
                }
                argc--;
                argv++;
        }
        if (argc < 1)
-               usage(ping_usage);
+               bb_show_usage();
 
        myid = getpid() & 0xFFFF;
        ping(*argv);
-       exit(TRUE);
+       return EXIT_SUCCESS;
 }
-#endif
-
-/*
- * 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.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- *    notice, this list of conditions and the following disclaimer.
- * 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.
- * 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.
- *
- * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- */
+#endif /* ! CONFIG_FEATURE_FANCY_PING */