Magellan Linux

Annotation of /trunk/mkinitrd-magellan/busybox/networking/ping.c

Parent Directory Parent Directory | Revision Log Revision Log


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