qemu/roms/ipxe/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 }