/* * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>. * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License as * published by the Free Software Foundation; either version 2 of the * License, or any later version. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ FILE_LICENCE ( GPL2_OR_LATER ); #include <stdio.h> #include <stdint.h> #include <stdlib.h> #include <stdio.h> #include <errno.h> #include <stddef.h> #include <string.h> #include <assert.h> #include <getopt.h> #include <gpxe/netdevice.h> #include <gpxe/in.h> #include <gpxe/command.h> #include <usr/dhcpmgmt.h> /** @file * * DHCP management commands * */ /** * "dhcp" command syntax message * * @v argv Argument list */ static void dhcp_syntax ( char **argv ) { printf ( "Usage:\n" " %s <interface>\n" "\n" "Configure a network interface using DHCP\n", argv[0] ); } /** * The "dhcp" command * * @v argc Argument count * @v argv Argument list * @ret rc Exit code */ static int dhcp_exec ( int argc, char **argv ) { static struct option longopts[] = { { "help", 0, NULL, 'h' }, { NULL, 0, NULL, 0 }, }; const char *netdev_txt; struct net_device *netdev; int c; int rc; /* Parse options */ while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){ switch ( c ) { case 'h': /* Display help text */ default: /* Unrecognised/invalid option */ dhcp_syntax ( argv ); return 1; } } /* Need exactly one interface name remaining after the options */ if ( optind != ( argc - 1 ) ) { dhcp_syntax ( argv ); return 1; } netdev_txt = argv[optind]; /* Parse arguments */ netdev = find_netdev ( netdev_txt ); if ( ! netdev ) { printf ( "No such interface: %s\n", netdev_txt ); return 1; } /* Perform DHCP */ if ( ( rc = dhcp ( netdev ) ) != 0 ) { printf ( "Could not configure %s: %s\n", netdev->name, strerror ( rc ) ); return 1; } return 0; } /** * "pxebs" command syntax message * * @v argv Argument list */ static void pxebs_syntax ( char **argv ) { printf ( "Usage:\n" " %s <interface> <server_type>\n" "\n" "Perform PXE Boot Server discovery\n", argv[0] ); } /** * The "pxebs" command * * @v argc Argument count * @v argv Argument list * @ret rc Exit code */ static int pxebs_exec ( int argc, char **argv ) { static struct option longopts[] = { { "help", 0, NULL, 'h' }, { NULL, 0, NULL, 0 }, }; const char *netdev_txt; const char *pxe_type_txt; struct net_device *netdev; unsigned int pxe_type; char *end; int c; int rc; /* Parse options */ while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){ switch ( c ) { case 'h': /* Display help text */ default: /* Unrecognised/invalid option */ pxebs_syntax ( argv ); return 1; } } if ( optind != ( argc - 2 ) ) { pxebs_syntax ( argv ); return 1; } netdev_txt = argv[optind]; pxe_type_txt = argv[ optind + 1 ]; /* Parse arguments */ netdev = find_netdev ( netdev_txt ); if ( ! netdev ) { printf ( "No such interface: %s\n", netdev_txt ); return 1; } pxe_type = strtoul ( pxe_type_txt, &end, 0 ); if ( *end ) { printf ( "Bad server type: %s\n", pxe_type_txt ); return 1; } /* Perform Boot Server Discovery */ if ( ( rc = pxebs ( netdev, pxe_type ) ) != 0 ) { printf ( "Could not discover boot server on %s: %s\n", netdev->name, strerror ( rc ) ); return 1; } return 0; } /** DHCP management commands */ struct command dhcp_commands[] __command = { { .name = "dhcp", .exec = dhcp_exec, }, { .name = "pxebs", .exec = pxebs_exec, }, };