- apply hunk that fixes an issue with (wrong?) dependencies. Initial notes were:
[oweals/busybox.git] / networking / ping.c
index 52e48867e3347388eedeebafd400439a8d87d1e6..e413bec701abbc677fb1a0d1e3fb1b22b66009eb 100644 (file)
@@ -101,15 +101,9 @@ static void ping4(len_and_sockaddr *lsa)
        pkt->icmp_type = ICMP_ECHO;
        pkt->icmp_cksum = in_cksum((unsigned short *) pkt, sizeof(packet));
 
-       c = sendto(pingsock, packet, DEFDATALEN + ICMP_MINLEN, 0,
+       c = xsendto(pingsock, packet, DEFDATALEN + ICMP_MINLEN,
                           (struct sockaddr *) &pingaddr, sizeof(pingaddr));
 
-       if (c < 0) {
-               if (ENABLE_FEATURE_CLEAN_UP)
-                       close(pingsock);
-               bb_perror_msg_and_die("sendto");
-       }
-
        /* listen for replies */
        while (1) {
                struct sockaddr_in from;
@@ -153,15 +147,9 @@ static void ping6(len_and_sockaddr *lsa)
        sockopt = offsetof(struct icmp6_hdr, icmp6_cksum);
        setsockopt(pingsock, SOL_RAW, IPV6_CHECKSUM, &sockopt, sizeof(sockopt));
 
-       c = sendto(pingsock, packet, DEFDATALEN + sizeof (struct icmp6_hdr), 0,
+       c = xsendto(pingsock, packet, DEFDATALEN + sizeof (struct icmp6_hdr),
                           (struct sockaddr *) &pingaddr, sizeof(pingaddr));
 
-       if (c < 0) {
-               if (ENABLE_FEATURE_CLEAN_UP)
-                       close(pingsock);
-               bb_perror_msg_and_die("sendto");
-       }
-
        /* listen for replies */
        while (1) {
                struct sockaddr_in6 from;
@@ -185,12 +173,13 @@ static void ping6(len_and_sockaddr *lsa)
 }
 #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]) == '-') {
+       while (++argv[0][0] == '-') {
                if (argv[0][1] == '4') {
                        af = AF_INET;
                        continue;
@@ -210,9 +199,9 @@ int ping_main(int argc, char **argv)
                bb_show_usage();
 
 #if ENABLE_PING6
-       lsa = host_and_af2sockaddr(hostname, 0, af);
+       lsa = xhost_and_af2sockaddr(hostname, 0, af);
 #else
-       lsa = host_and_af2sockaddr(hostname, 0, AF_INET);
+       lsa = xhost_and_af2sockaddr(hostname, 0, AF_INET);
 #endif
        /* Set timer _after_ DNS resolution */
        signal(SIGALRM, noresp);
@@ -253,7 +242,7 @@ static union {
        struct sockaddr_in6 sin6;
 #endif
 } pingaddr;
-static struct sockaddr_in sourceaddr;
+static len_and_sockaddr *source_lsa;
 static int pingsock = -1;
 static unsigned datalen; /* intentionally uninitialized to work around gcc bug */
 
@@ -261,7 +250,8 @@ static int if_index;
 
 static unsigned long ntransmitted, nreceived, nrepeats, pingcount;
 static int myid;
-static unsigned long tmin = ULONG_MAX, tmax, tsum;
+static unsigned tmin = UINT_MAX, tmax;
+static unsigned long tsum;
 static char rcvd_tbl[MAX_DUP_CHK / 8];
 
 static const char *hostname;
@@ -277,8 +267,6 @@ static const char *dotted;
 
 static void pingstats(int junk ATTRIBUTE_UNUSED)
 {
-       int status;
-
        signal(SIGINT, SIG_IGN);
 
        printf("\n--- %s ping statistics ---\n", hostname);
@@ -289,28 +277,29 @@ static void pingstats(int junk ATTRIBUTE_UNUSED)
        if (ntransmitted)
                ntransmitted = (ntransmitted - nreceived) * 100 / ntransmitted;
        printf("%lu%% packet loss\n", ntransmitted);
-       if (nreceived)
-               printf("round-trip min/avg/max = %lu.%lu/%lu.%lu/%lu.%lu ms\n",
+       if (tmin != UINT_MAX)
+               printf("round-trip min/avg/max = %u.%u/%lu.%lu/%u.%u ms\n",
                        tmin / 10, tmin % 10,
                        (tsum / (nreceived + nrepeats)) / 10,
                        (tsum / (nreceived + nrepeats)) % 10, tmax / 10, tmax % 10);
-       if (nreceived != 0)
-               status = EXIT_SUCCESS;
-       else
-               status = EXIT_FAILURE;
-       exit(status);
+       exit(nreceived == 0); /* (nreceived == 0) is true (1) -- 'failure' */
 }
 
-static void sendping_tail(void (*sp)(int), int sz, int sizeof_packet)
+static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt)
 {
-       if (sz < 0)
-               bb_perror_msg_and_die("sendto");
-       if (sz != sizeof_packet)
-               bb_error_msg_and_die("ping wrote %d chars; %d expected", sz,
-                       sizeof_packet);
+       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 */
+       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... */
@@ -321,55 +310,35 @@ static void sendping_tail(void (*sp)(int), int sz, int sizeof_packet)
 
 static void sendping4(int junk ATTRIBUTE_UNUSED)
 {
-       struct icmp *pkt;
-       int i;
-       char packet[datalen + ICMP_MINLEN];
-
-       pkt = (struct icmp *) packet;
+       struct icmp *pkt = alloca(datalen + ICMP_MINLEN);
 
        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;
-       CLR((uint16_t)ntransmitted % MAX_DUP_CHK);
-       ntransmitted++;
-
        gettimeofday((struct timeval *) &pkt->icmp_dun, NULL);
-       pkt->icmp_cksum = in_cksum((unsigned short *) pkt, sizeof(packet));
+       pkt->icmp_cksum = in_cksum((unsigned short *) pkt, datalen + ICMP_MINLEN);
 
-       i = sendto(pingsock, packet, sizeof(packet), 0,
-                       &pingaddr.sa, sizeof(pingaddr.sin));
-
-       sendping_tail(sendping4, i, sizeof(packet));
+       sendping_tail(sendping4, pkt, datalen + ICMP_MINLEN);
 }
 #if ENABLE_PING6
 static void sendping6(int junk ATTRIBUTE_UNUSED)
 {
-       struct icmp6_hdr *pkt;
-       int i;
-       char packet[datalen + sizeof (struct icmp6_hdr)];
-
-       pkt = (struct icmp6_hdr *) packet;
+       struct icmp6_hdr *pkt = alloca(datalen + sizeof(struct icmp6_hdr));
 
        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;
-       CLR((uint16_t)ntransmitted % MAX_DUP_CHK);
-       ntransmitted++;
-
        gettimeofday((struct timeval *) &pkt->icmp6_data8[4], NULL);
 
-       i = sendto(pingsock, packet, sizeof(packet), 0,
-                       &pingaddr.sa, sizeof(pingaddr.sin6));
-
-       sendping_tail(sendping6, i, sizeof(packet));
+       sendping_tail(sendping6, pkt, datalen + sizeof(struct icmp6_hdr));
 }
 #endif
 
-static char *icmp_type_name(int id)
+static const char *icmp_type_name(int id)
 {
        switch (id) {
        case ICMP_ECHOREPLY:      return "Echo Reply";
@@ -400,7 +369,7 @@ static char *icmp_type_name(int id)
 #ifndef MLD_LISTENER_REDUCTION
 # define MLD_LISTENER_REDUCTION ICMP6_MEMBERSHIP_REDUCTION
 #endif
-static char *icmp6_type_name(int id)
+static const char *icmp6_type_name(int id)
 {
        switch (id) {
        case ICMP6_DST_UNREACH:      return "Destination Unreachable";
@@ -417,15 +386,57 @@ static char *icmp6_type_name(int id)
 }
 #endif
 
+static void unpack_tail(int sz, struct timeval *tp,
+               const char *from_str,
+               uint16_t recv_seq, int ttl)
+{
+       const char *dupmsg = " (DUP!)";
+       unsigned triptime = triptime; /* for gcc */
+
+       ++nreceived;
+
+       if (tp) {
+               struct timeval tv;
+
+               gettimeofday(&tv, NULL);
+               tv.tv_usec -= tp->tv_usec;
+               if (tv.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(recv_seq % MAX_DUP_CHK)) {
+               ++nrepeats;
+               --nreceived;
+       } else {
+               SET(recv_seq % MAX_DUP_CHK);
+               dupmsg += 7;
+       }
+
+       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.%u ms", triptime / 10, triptime % 10);
+       puts(dupmsg);
+       fflush(stdout);
+}
 static void unpack4(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);
+       int hlen;
 
        /* discard if too short */
        if (sz < (datalen + ICMP_MINLEN))
@@ -441,61 +452,25 @@ static void unpack4(char *buf, int sz, struct sockaddr_in *from)
 
        if (icmppkt->icmp_type == ICMP_ECHOREPLY) {
                uint16_t recv_seq = ntohs(icmppkt->icmp_seq);
-               ++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(recv_seq % MAX_DUP_CHK)) {
-                       ++nrepeats;
-                       --nreceived;
-                       dupflag = 1;
-               } else {
-                       SET(recv_seq % MAX_DUP_CHK);
-                       dupflag = 0;
-               }
-
-               if (option_mask32 & OPT_QUIET)
-                       return;
-
-               printf("%d bytes from %s: icmp_seq=%u", sz,
-                          inet_ntoa(*(struct in_addr *) &from->sin_addr.s_addr),
-                          recv_seq);
-               printf(" ttl=%d", iphdr->ttl);
-               printf(" time=%lu.%lu ms", triptime / 10, triptime % 10);
-               if (dupflag)
-                       printf(" (DUP!)");
-               puts("");
-       } else {
-               if (icmppkt->icmp_type != ICMP_ECHO)
-                       bb_error_msg("warning: got ICMP %d (%s)",
-                                       icmppkt->icmp_type,
-                                       icmp_type_name(icmppkt->icmp_type));
+               struct timeval *tp = NULL;
+
+               if (sz >= ICMP_MINLEN + sizeof(struct timeval))
+                       tp = (struct timeval *) 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));
        }
-       fflush(stdout);
 }
 #if ENABLE_PING6
 static void unpack6(char *packet, int sz, struct sockaddr_in6 *from, int hoplimit)
 {
        struct icmp6_hdr *icmppkt;
-       struct timeval tv, *tp;
-       int dupflag;
-       unsigned long triptime;
        char buf[INET6_ADDRSTRLEN];
 
-       gettimeofday(&tv, NULL);
-
        /* discard if too short */
        if (sz < (datalen + sizeof(struct icmp6_hdr)))
                return;
@@ -506,50 +481,19 @@ static void unpack6(char *packet, int sz, struct sockaddr_in6 *from, int hoplimi
 
        if (icmppkt->icmp6_type == ICMP6_ECHO_REPLY) {
                uint16_t recv_seq = ntohs(icmppkt->icmp6_seq);
-               ++nreceived;
-               tp = (struct timeval *) &icmppkt->icmp6_data8[4];
-
-               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(recv_seq % MAX_DUP_CHK)) {
-                       ++nrepeats;
-                       --nreceived;
-                       dupflag = 1;
-               } else {
-                       SET(recv_seq % MAX_DUP_CHK);
-                       dupflag = 0;
-               }
-
-               if (option_mask32 & OPT_QUIET)
-                       return;
-
-               printf("%d bytes from %s: icmp6_seq=%u", sz,
-                          inet_ntop(AF_INET6, &pingaddr.sin6.sin6_addr,
-                                                buf, sizeof(buf)),
-                          recv_seq);
-               printf(" ttl=%d time=%lu.%lu ms", hoplimit,
-                          triptime / 10, triptime % 10);
-               if (dupflag)
-                       printf(" (DUP!)");
-               puts("");
-       } 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));
+               struct timeval *tp = NULL;
+
+               if (sz >= sizeof(struct icmp6_hdr) + sizeof(struct timeval))
+                       tp = (struct timeval *) &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));
        }
-       fflush(stdout);
 }
 #endif
 
@@ -560,9 +504,8 @@ static void ping4(len_and_sockaddr *lsa)
 
        pingsock = create_icmp_socket();
        pingaddr.sin = lsa->sin;
-       if (sourceaddr.sin_addr.s_addr) {
-               xbind(pingsock, (struct sockaddr*)&sourceaddr, sizeof(sourceaddr));
-       }
+       if (source_lsa)
+               xbind(pingsock, &lsa->sa, lsa->len);
 
        /* enable broadcast pings */
        setsockopt_broadcast(pingsock);
@@ -571,13 +514,6 @@ static void ping4(len_and_sockaddr *lsa)
        sockopt = 48 * 1024; /* explain why 48k? */
        setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
 
-       printf("PING %s (%s)", hostname, dotted);
-       if (sourceaddr.sin_addr.s_addr) {
-               printf(" from %s",
-                       inet_ntoa(*(struct in_addr *) &sourceaddr.sin_addr.s_addr));
-       }
-       printf(": %d data bytes\n", datalen);
-
        signal(SIGINT, pingstats);
 
        /* start the ping's going ... */
@@ -614,9 +550,9 @@ static void ping6(len_and_sockaddr *lsa)
 
        pingsock = create_icmp6_socket();
        pingaddr.sin6 = lsa->sin6;
-       //if (sourceaddr.sin_addr.s_addr) {
-       //      xbind(pingsock, (struct sockaddr*)&sourceaddr, sizeof(sourceaddr));
-       //}
+       /* untested whether "-I addr" really works for IPv6: */
+       if (source_lsa)
+               xbind(pingsock, &lsa->sa, lsa->len);
 
 #ifdef ICMP6_FILTER
        {
@@ -651,8 +587,6 @@ static void ping6(len_and_sockaddr *lsa)
        if (if_index)
                pingaddr.sin6.sin6_scope_id = if_index;
 
-       printf("PING %s (%s): %d data bytes\n", hostname, dotted, datalen);
-
        signal(SIGINT, pingstats);
 
        /* start the ping's going ... */
@@ -694,27 +628,24 @@ static void ping6(len_and_sockaddr *lsa)
 }
 #endif
 
-/* TODO: consolidate ether-wake.c, dnsd.c, ifupdown.c, nslookup.c
- * versions of below thing. BTW we have far too many "%u.%u.%u.%u" too...
-*/
-static int parse_nipquad(const char *str, struct sockaddr_in* addr)
+static void ping(len_and_sockaddr *lsa)
 {
-       char dummy;
-       unsigned i1, i2, i3, i4;
-       if (sscanf(str, "%u.%u.%u.%u%c",
-                          &i1, &i2, &i3, &i4, &dummy) == 4
-       && ( (i1|i2|i3|i4) <= 0xff )
-       ) {
-               uint8_t* ptr = (uint8_t*)&addr->sin_addr;
-               ptr[0] = i1;
-               ptr[1] = i2;
-               ptr[2] = i3;
-               ptr[3] = i4;
-               return 0;
+       printf("PING %s (%s)", hostname, dotted);
+       if (source_lsa) {
+               printf(" from %s",
+                       xmalloc_sockaddr2dotted_noport(&lsa->sa, lsa->len));
        }
-       return 1; /* error */
+       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;
@@ -730,9 +661,11 @@ int ping_main(int argc, char **argv)
        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)
-                       if (parse_nipquad(opt_I, &sourceaddr))
-                               bb_show_usage();
+               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 = (int16_t) getpid();
        hostname = argv[optind];
@@ -741,17 +674,17 @@ int ping_main(int argc, char **argv)
                af = AF_INET;
        if (option_mask32 & OPT_IPV6)
                af = AF_INET6;
-       lsa = host_and_af2sockaddr(hostname, 0, af);
+       lsa = xhost_and_af2sockaddr(hostname, 0, af);
 #else
-       lsa = host_and_af2sockaddr(hostname, 0, AF_INET);
+       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);
-#if ENABLE_PING6
-       if (lsa->sa.sa_family == AF_INET6)
-               ping6(lsa);
-       else
-#endif
-               ping4(lsa);
+       ping(lsa);
        pingstats(0);
        return EXIT_SUCCESS;
 }
@@ -759,9 +692,10 @@ int ping_main(int argc, char **argv)
 
 
 #if ENABLE_PING6
+int ping6_main(int argc, char **argv);
 int ping6_main(int argc, char **argv)
 {
-       argv[0] = "-6";
+       argv[0] = (char*)"-6";
        return ping_main(argc + 1, argv - 1);
 }
 #endif