Annotation of /trunk/mkinitrd-magellan/busybox/networking/ping.c
Parent Directory | 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)
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 | */ |