src/hci/commands/dhcp_cmd.c

   1 /*
   2  * Copyright (C) 2007 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., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 FILE_LICENCE ( GPL2_OR_LATER );
  20
  21 #include <stdio.h>
  22 #include <stdint.h>
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <errno.h>
  26 #include <stddef.h>
  27 #include <string.h>
  28 #include <assert.h>
  29 #include <getopt.h>
  30 #include <gpxe/netdevice.h>
  31 #include <gpxe/in.h>
  32 #include <gpxe/command.h>
  33 #include <usr/dhcpmgmt.h>
  34 #include <usr/ifmgmt.h>
  35
  36 /** @file
  37  *
  38  * DHCP management commands
  39  *
  40  */
  41
  42 /**
  43  * "dhcp" command syntax message
  44  *
  45  * @v argv              Argument list
  46  */
  47 static void dhcp_syntax ( char **argv ) {
  48         printf ( "Usage:\n"
  49                  "  %s <interface> [<interface> ...]\n"
  50                  "  %s any\n"
  51                  "\n"
  52                  "Configure a network interface using DHCP\n",
  53                  argv[0], argv[0] );
  54 }
  55
  56 /**
  57  * Attempt to configure a device with dhcp
  58  *
  59  * @v netdev            Device to configure
  60  * @ret rc              Exit code
  61  */
  62 static int dhcp_one_device ( struct net_device *netdev ) {
  63         int rc;
  64
  65         /* Perform DHCP */
  66         if ( ( rc = dhcp ( netdev ) ) != 0 ) {
  67                 /* Close the device on error to avoid out-of-memory */
  68                 netdev_close ( netdev );
  69
  70                 printf ( "Could not configure %s: %s\n", netdev->name,
  71                          strerror ( rc ) );
  72                 return 1;
  73         }
  74
  75         return 0;
  76 }
  77
  78 /**
  79  * Call dhcp_one_device() for each name in argv
  80  *
  81  * @v argc              Number of devices
  82  * @v argv              List of device names
  83  * @ret rc              Exit code
  84  */
  85 static int dhcp_each_device_name ( int argc, char **argv ) {
  86         int i;
  87         char *netdev_name;
  88         struct net_device *netdev;
  89
  90         for ( i = 0; i < argc; i++ ) {
  91                 netdev_name = argv[i];
  92                 netdev = find_netdev ( netdev_name );
  93
  94                 if ( ! netdev ) {
  95                         printf ( "No such interface: %s\n", netdev_name );
  96                         continue;
  97                 }
  98
  99                 if ( dhcp_one_device ( netdev ) == 0 )
 100                         return 0;
 101         }
 102
 103         printf ( "Could not configure any interface.\n" );
 104         return 1;
 105 }
 106
 107 /**
 108  * Call dhcp_one_device() for each device in net_devices
 109  *
 110  * @ret rc              Exit code
 111  */
 112 static int dhcp_each_device ( void ) {
 113         struct net_device *netdev;
 114
 115         for_each_netdev ( netdev ) {
 116                 if ( dhcp_one_device ( netdev ) == 0 )
 117                         return 0;
 118         }
 119
 120         printf ( "Could not configure any interface.\n" );
 121         return 1;
 122 }
 123
 124 /**
 125  * The "dhcp" command
 126  *
 127  * @v argc              Argument count
 128  * @v argv              Argument list
 129  * @ret rc              Exit code
 130  */
 131 static int dhcp_exec ( int argc, char **argv ) {
 132         static struct option longopts[] = {
 133                 { "help", 0, NULL, 'h' },
 134                 { NULL, 0, NULL, 0 },
 135         };
 136         int c;
 137
 138         /* Parse options */
 139         while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){
 140                 switch ( c ) {
 141                 case 'h':
 142                         /* Display help text */
 143                 default:
 144                         /* Unrecognised/invalid option */
 145                         dhcp_syntax ( argv );
 146                         return 1;
 147                 }
 148         }
 149
 150         /* Need one or more interface names remaining after the options */
 151         if ( ( argc - optind ) < 1 ) {
 152                 dhcp_syntax ( argv );
 153                 return 1;
 154         }
 155
 156         if ( strcmp ( argv[optind], "any" ) == 0 )
 157                 return dhcp_each_device();
 158
 159         return dhcp_each_device_name ( argc - optind, argv + optind );
 160 }
 161
 162 /**
 163  * "pxebs" command syntax message
 164  *
 165  * @v argv              Argument list
 166  */
 167 static void pxebs_syntax ( char **argv ) {
 168         printf ( "Usage:\n"
 169                  "  %s <interface> <server_type>\n"
 170                  "\n"
 171                  "Perform PXE Boot Server discovery\n",
 172                  argv[0] );
 173 }
 174
 175 /**
 176  * The "pxebs" command
 177  *
 178  * @v argc              Argument count
 179  * @v argv              Argument list
 180  * @ret rc              Exit code
 181  */
 182 static int pxebs_exec ( int argc, char **argv ) {
 183         static struct option longopts[] = {
 184                 { "help", 0, NULL, 'h' },
 185                 { NULL, 0, NULL, 0 },
 186         };
 187         const char *netdev_txt;
 188         const char *pxe_type_txt;
 189         struct net_device *netdev;
 190         unsigned int pxe_type;
 191         char *end;
 192         int c;
 193         int rc;
 194
 195         /* Parse options */
 196         while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){
 197                 switch ( c ) {
 198                 case 'h':
 199                         /* Display help text */
 200                 default:
 201                         /* Unrecognised/invalid option */
 202                         pxebs_syntax ( argv );
 203                         return 1;
 204                 }
 205         }
 206         if ( optind != ( argc - 2 ) ) {
 207                 pxebs_syntax ( argv );
 208                 return 1;
 209         }
 210         netdev_txt = argv[optind];
 211         pxe_type_txt = argv[ optind + 1 ];
 212
 213         /* Parse arguments */
 214         netdev = find_netdev ( netdev_txt );
 215         if ( ! netdev ) {
 216                 printf ( "No such interface: %s\n", netdev_txt );
 217                 return 1;
 218         }
 219         pxe_type = strtoul ( pxe_type_txt, &end, 0 );
 220         if ( *end ) {
 221                 printf ( "Bad server type: %s\n", pxe_type_txt );
 222                 return 1;
 223         }
 224
 225         /* Perform Boot Server Discovery */
 226         if ( ( rc = pxebs ( netdev, pxe_type ) ) != 0 ) {
 227                 printf ( "Could not discover boot server on %s: %s\n",
 228                          netdev->name, strerror ( rc ) );
 229                 return 1;
 230         }
 231
 232         return 0;
 233 }
 234
 235 /** DHCP management commands */
 236 struct command dhcp_commands[] __command = {
 237         {
 238                 .name = "dhcp",
 239                 .exec = dhcp_exec,
 240         },
 241         {
 242                 .name = "pxebs",
 243                 .exec = pxebs_exec,
 244         },
 245 };