source: MondoRescue/branches/3.2/mindi-busybox/networking/ping.c

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