1 files changed, 374 insertions, 0 deletions

diff --git a/ping.c b/ping.c
new file mode 100644
index 000000000..9f8a103bd
--- /dev/null
+++ b/ping.c
@@ -0,0 +1,374 @@
+/*
+ * $Id: ping.c,v 1.1 1999/12/07 23:14:59 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:
+ *
+ * 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.
+ */
+
+#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 <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 O_QUIET         (1 << 0)
+
+#define A(bit)          rcvd_tbl[(bit)>>3]      /* identify byte in array */
+#define B(bit)          (1 << ((bit) & 0x07))   /* identify bit in byte */
+#define SET(bit)        (A(bit) |= B(bit))
+#define CLR(bit)        (A(bit) &= (~B(bit)))
+#define TST(bit)        (A(bit) & B(bit))
+
+static const char* ping_usage = "ping [OPTION]... host\n\n"
+"Options:\n"
+"\t-q\t\tquiet\n"
+"\t-c\t\tping count\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)
+{
+    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);
+}
+
+static void sendping(int ign)
+{
+    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 */
+        alarm(1);
+    } 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 unpack(char *buf, int sz, struct sockaddr_in *from)
+{
+    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 */
+        ++nreceived;
+        tp = (struct timeval *)icmppkt->icmp_data;
+
+        if ((tv.tv_usec -= tp->tv_usec) < 0) {
+            --tv.tv_sec;
+            tv.tv_usec += 1000000;
+        }
+        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;
+        } else {
+            SET(icmppkt->icmp_seq % MAX_DUP_CHK);
+            dupflag = 0;
+        }
+
+        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");
+    }   
+}
+
+static void ping(char *host)
+{
+    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");
+        }
+        exit(1);
+    }
+
+#ifdef SUID_BUSYBOX
+    setuid(getuid());
+#endif
+    
+    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);
+        }
+        
+        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;
+        }
+        unpack(packet, c, &from);
+        if (pingcount > 0 && nreceived >= pingcount) break;     
+    }
+    pingstats(0);
+}
+
+extern int ping_main(int argc, char **argv)
+{
+    argc--;
+    argv++;
+    options = 0;
+    /* Parse any options */
+    if (argc < 1) usage(ping_usage);
+
+    while (**argv == '-') {
+        while (*++(*argv))
+            switch (**argv) {
+            case 'c':
+                argc--; argv++;
+                if (argc < 1) usage(ping_usage);
+                pingcount = atoi(*argv);
+                break;
+            case 'q':
+                options |= O_QUIET;
+                break;
+            default:
+                usage(ping_usage);
+            }
+        argc--;
+        argv++;
+    }
+
+    if (argc < 1) usage(ping_usage);
+
+    myid = getpid() & 0xFFFF;
+    ping(*(argv++));
+    exit( TRUE);
+}
+
+/*
+ * 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.
+ */
+
+