ping: add -w, -W support (James Simmons <jsimmons AT infradead.org>)

function                                             old     new   delta
print_stats_and_exit                                   -     282    +282
sendping_tail                                        151     231     +80
packed_usage                                       23976   24054     +78
ping_main                                            401     412     +11
arm_ioctl                                             13      20      +7
pingstats                                            259       -    -259
------------------------------------------------------------------------------
(add/remove: 1/1 grow/shrink: 4/0 up/down: 458/-259)          Total: 199 bytes
This commit is contained in:
Denis Vlasenko 2008-04-07 00:46:29 +00:00
parent 278a1c2264
commit 90c31b3d4b
2 changed files with 54 additions and 23 deletions

View File

@ -2935,6 +2935,10 @@
"\n -c CNT Send only CNT pings" \ "\n -c CNT Send only CNT pings" \
"\n -s SIZE Send SIZE data bytes in packets (default=56)" \ "\n -s SIZE Send SIZE data bytes in packets (default=56)" \
"\n -I iface/IP Use interface or IP address as source" \ "\n -I iface/IP Use interface or IP address as source" \
"\n -W timeout Seconds to wait for the first response (default:10)" \
"\n (after all -c CNT packets are sent)" \
"\n -w deadline Seconds until ping exits (default:infinite)" \
"\n (can exit earlier with -c CNT)" \
"\n -q Quiet, only displays output at start" \ "\n -q Quiet, only displays output at start" \
"\n and when finished" \ "\n and when finished" \

View File

@ -224,15 +224,17 @@ int ping_main(int argc ATTRIBUTE_UNUSED, char **argv)
/* full(er) version */ /* full(er) version */
#define OPT_STRING ("qvc:s:I:4" USE_PING6("6")) #define OPT_STRING ("qvc:s:w:W:I:4" USE_PING6("6"))
enum { enum {
OPT_QUIET = 1 << 0, OPT_QUIET = 1 << 0,
OPT_VERBOSE = 1 << 1, OPT_VERBOSE = 1 << 1,
OPT_c = 1 << 2, OPT_c = 1 << 2,
OPT_s = 1 << 3, OPT_s = 1 << 3,
OPT_I = 1 << 4, OPT_w = 1 << 4,
OPT_IPV4 = 1 << 5, OPT_W = 1 << 5,
OPT_IPV6 = (1 << 6) * ENABLE_PING6, OPT_I = 1 << 6,
OPT_IPV4 = 1 << 7,
OPT_IPV6 = (1 << 8) * ENABLE_PING6,
}; };
@ -246,6 +248,9 @@ struct globals {
uint16_t myid; uint16_t myid;
unsigned tmin, tmax; /* in us */ unsigned tmin, tmax; /* in us */
unsigned long long tsum; /* in us, sum of all times */ unsigned long long tsum; /* in us, sum of all times */
unsigned deadline;
unsigned timeout;
unsigned total_secs;
const char *hostname; const char *hostname;
const char *dotted; const char *dotted;
union { union {
@ -271,6 +276,9 @@ struct globals {
#define tmin (G.tmin ) #define tmin (G.tmin )
#define tmax (G.tmax ) #define tmax (G.tmax )
#define tsum (G.tsum ) #define tsum (G.tsum )
#define deadline (G.deadline )
#define timeout (G.timeout )
#define total_secs (G.total_secs )
#define hostname (G.hostname ) #define hostname (G.hostname )
#define dotted (G.dotted ) #define dotted (G.dotted )
#define pingaddr (G.pingaddr ) #define pingaddr (G.pingaddr )
@ -280,6 +288,8 @@ void BUG_ping_globals_too_big(void);
if (sizeof(G) > COMMON_BUFSIZE) \ if (sizeof(G) > COMMON_BUFSIZE) \
BUG_ping_globals_too_big(); \ BUG_ping_globals_too_big(); \
pingsock = -1; \ pingsock = -1; \
datalen = DEFDATALEN; \
timeout = MAXWAIT; \
tmin = UINT_MAX; \ tmin = UINT_MAX; \
} while (0) } while (0)
@ -292,7 +302,8 @@ void BUG_ping_globals_too_big(void);
/**************************************************************************/ /**************************************************************************/
static void pingstats(int junk ATTRIBUTE_UNUSED) static void print_stats_and_exit(int junk) ATTRIBUTE_NORETURN;
static void print_stats_and_exit(int junk ATTRIBUTE_UNUSED)
{ {
signal(SIGINT, SIG_IGN); signal(SIGINT, SIG_IGN);
@ -311,7 +322,8 @@ static void pingstats(int junk ATTRIBUTE_UNUSED)
tavg / 1000, tavg % 1000, tavg / 1000, tavg % 1000,
tmax / 1000, tmax % 1000); tmax / 1000, tmax % 1000);
} }
exit(nreceived == 0); /* (nreceived == 0) is true (1) -- 'failure' */ /* if condition is true, exit with 1 -- 'failure' */
exit(nreceived == 0 || (deadline && nreceived < pingcount));
} }
static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt) static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt)
@ -327,13 +339,30 @@ static void sendping_tail(void (*sp)(int), const void *pkt, int size_pkt)
if (sz != size_pkt) if (sz != size_pkt)
bb_error_msg_and_die(bb_msg_write_error); bb_error_msg_and_die(bb_msg_write_error);
signal(SIGALRM, sp); if (pingcount == 0 || deadline || ntransmitted < pingcount) {
if (pingcount == 0 || ntransmitted < pingcount) { /* schedule next in 1s */ /* Didn't send all pings yet - schedule next in 1s */
signal(SIGALRM, sp);
if (deadline) {
total_secs += PINGINTERVAL;
if (total_secs >= deadline)
signal(SIGALRM, print_stats_and_exit);
}
alarm(PINGINTERVAL); alarm(PINGINTERVAL);
} else { /* done, wait for the last ping to come back */ } else { /* -c NN, and all NN are sent (and no deadline) */
/* todo, don't necessarily need to wait so long... */ /* Wait for the last ping to come back.
signal(SIGALRM, pingstats); * -W timeout: wait for a response in seconds.
alarm(MAXWAIT); * Affects only timeout in absense of any responses,
* otherwise ping waits for two RTTs. */
unsigned expire = timeout;
if (nreceived) {
/* approx. 2*tmax, in seconds (2 RTT) */
expire = tmax / (512*1024);
if (expire == 0)
expire = 1;
}
signal(SIGALRM, print_stats_and_exit);
alarm(expire);
} }
} }
@ -549,7 +578,7 @@ static void ping4(len_and_sockaddr *lsa)
sockopt = 48 * 1024; /* explain why 48k? */ sockopt = 48 * 1024; /* explain why 48k? */
setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt)); setsockopt(pingsock, SOL_SOCKET, SO_RCVBUF, &sockopt, sizeof(sockopt));
signal(SIGINT, pingstats); signal(SIGINT, print_stats_and_exit);
/* start the ping's going ... */ /* start the ping's going ... */
sendping4(0); sendping4(0);
@ -568,7 +597,7 @@ static void ping4(len_and_sockaddr *lsa)
continue; continue;
} }
unpack4(packet, c, &from); unpack4(packet, c, &from);
if (pingcount > 0 && nreceived >= pingcount) if (pingcount && nreceived >= pingcount)
break; break;
} }
} }
@ -624,7 +653,7 @@ static void ping6(len_and_sockaddr *lsa)
if (if_index) if (if_index)
pingaddr.sin6.sin6_scope_id = if_index; pingaddr.sin6.sin6_scope_id = if_index;
signal(SIGINT, pingstats); signal(SIGINT, print_stats_and_exit);
/* start the ping's going ... */ /* start the ping's going ... */
sendping6(0); sendping6(0);
@ -659,7 +688,7 @@ static void ping6(len_and_sockaddr *lsa)
} }
} }
unpack6(packet, c, /*&from,*/ hoplimit); unpack6(packet, c, /*&from,*/ hoplimit);
if (pingcount > 0 && nreceived >= pingcount) if (pingcount && nreceived >= pingcount)
break; break;
} }
} }
@ -691,11 +720,9 @@ int ping_main(int argc ATTRIBUTE_UNUSED, char **argv)
INIT_G(); INIT_G();
datalen = DEFDATALEN; /* exactly one argument needed; -v and -q don't mix; -w NUM, -W NUM */
opt_complementary = "=1:q--v:v--q:w+:W+";
/* exactly one argument needed, -v and -q don't mix */ getopt32(argv, OPT_STRING, &opt_c, &opt_s, &deadline, &timeout, &opt_I);
opt_complementary = "=1:q--v:v--q";
getopt32(argv, OPT_STRING, &opt_c, &opt_s, &opt_I);
if (option_mask32 & OPT_c) if (option_mask32 & OPT_c)
pingcount = xatoul(opt_c); // -c pingcount = xatoul(opt_c); // -c
if (option_mask32 & OPT_s) if (option_mask32 & OPT_s)
@ -726,8 +753,8 @@ int ping_main(int argc ATTRIBUTE_UNUSED, char **argv)
dotted = xmalloc_sockaddr2dotted_noport(&lsa->u.sa); dotted = xmalloc_sockaddr2dotted_noport(&lsa->u.sa);
ping(lsa); ping(lsa);
pingstats(0); print_stats_and_exit(0);
return EXIT_SUCCESS; /*return EXIT_SUCCESS;*/
} }
#endif /* FEATURE_FANCY_PING */ #endif /* FEATURE_FANCY_PING */