[zbin] Fix check for existence of most recent output byte
[ipxe.git] / src / usr / pingmgmt.c
1 /*
2 * Copyright (C) 2013 Michael Brown <mbrown@fensystems.co.uk>.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License as
6 * published by the Free Software Foundation; either version 2 of the
7 * License, or any later version.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17 * 02110-1301, USA.
18 */
19
20 FILE_LICENCE ( GPL2_OR_LATER );
21
22 #include <stdint.h>
23 #include <stdio.h>
24 #include <string.h>
25 #include <ipxe/pinger.h>
26 #include <ipxe/monojob.h>
27 #include <ipxe/timer.h>
28 #include <usr/pingmgmt.h>
29
30 /** @file
31 *
32 * ICMP ping management
33 *
34 */
35
36 /**
37 * Display ping result
38 *
39 * @v src Source socket address, or NULL
40 * @v sequence Sequence number
41 * @v len Payload length
42 * @v rc Status code
43 */
44 static void ping_callback ( struct sockaddr *peer, unsigned int sequence,
45 size_t len, int rc ) {
46
47 /* Display ping response */
48 printf ( "%zd bytes from %s: seq=%d",
49 len, ( peer ? sock_ntoa ( peer ) : "<none>" ), sequence );
50 if ( rc != 0 )
51 printf ( ": %s", strerror ( rc ) );
52 printf ( "\n" );
53 }
54
55 /**
56 * Ping a host
57 *
58 * @v hostname Hostname
59 * @v timeout Timeout between pings, in ticks
60 * @v len Payload length
61 * @v count Number of packets to send (or zero for no limit)
62 * @v quiet Inhibit output
63 * @ret rc Return status code
64 */
65 int ping ( const char *hostname, unsigned long timeout, size_t len,
66 unsigned int count, int quiet ) {
67 int rc;
68
69 /* Create pinger */
70 if ( ( rc = create_pinger ( &monojob, hostname, timeout, len, count,
71 ( quiet ? NULL : ping_callback ) ) ) != 0 ){
72 printf ( "Could not start ping: %s\n", strerror ( rc ) );
73 return rc;
74 }
75
76 /* Wait for ping to complete */
77 if ( ( rc = monojob_wait ( NULL, 0 ) ) != 0 ) {
78 if ( ! quiet )
79 printf ( "Finished: %s\n", strerror ( rc ) );
80 return rc;
81 }
82
83 return 0;
84 }