david/ipxe
Archived
1
0

[autoboot] Use generic option-parsing library

Total saving: 32 bytes.

Signed-off-by: Michael Brown <mcb30@ipxe.org>
This commit is contained in:
Michael Brown 2010-11-21 16:39:36 +00:00
parent 2877af3ff5
commit 17b337d4a8
3 changed files with 92 additions and 33 deletions

View File

@ -1,53 +1,110 @@
/*
* Copyright (C) 2010 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.
*/
#include <stdio.h> #include <stdio.h>
#include <getopt.h>
#include <ipxe/command.h> #include <ipxe/command.h>
#include <ipxe/parseopt.h>
#include <ipxe/netdevice.h> #include <ipxe/netdevice.h>
#include <usr/autoboot.h> #include <usr/autoboot.h>
FILE_LICENCE ( GPL2_OR_LATER ); FILE_LICENCE ( GPL2_OR_LATER );
/** @file
*
* Booting commands
*
*/
/** "autoboot" options */
struct autoboot_options {};
/** "autoboot" option list */
static struct option_descriptor autoboot_opts[] = {};
/** "autoboot" command descriptor */
static struct command_descriptor autoboot_cmd =
COMMAND_DESC ( struct autoboot_options, autoboot_opts, 0, 0,
"",
"Attempt to boot the system" );
/**
* "autoboot" command
*
* @v argc Argument count
* @v argv Argument list
* @ret rc Return status code
*/
static int autoboot_exec ( int argc, char **argv ) { static int autoboot_exec ( int argc, char **argv ) {
struct autoboot_options opts;
int rc;
if ( argc != 1 ) { /* Parse options */
printf ( "Usage:\n" if ( ( rc = parse_options ( argc, argv, &autoboot_cmd, &opts ) ) != 0 )
" %s\n" return rc;
"\n"
"Attempts to boot the system\n", /* Try to boot */
argv[0] ); if ( ( rc = autoboot() ) != 0 )
return 1; return rc;
return 0;
} }
autoboot(); /** "netboot" options */
struct netboot_options {};
/* Can never return success by definition */ /** "netboot" option list */
return 1; static struct option_descriptor netboot_opts[] = {};
}
/** "netboot" command descriptor */
static struct command_descriptor netboot_cmd =
COMMAND_DESC ( struct netboot_options, netboot_opts, 1, 1,
"<interface>",
"Attempt to boot the system from <interface>" );
/**
* "netboot" command
*
* @v argc Argument count
* @v argv Argument list
* @ret rc Return status code
*/
static int netboot_exec ( int argc, char **argv ) { static int netboot_exec ( int argc, char **argv ) {
const char *netdev_name; struct netboot_options opts;
struct net_device *netdev; struct net_device *netdev;
int rc;
if ( argc != 2 ) { /* Parse options */
printf ( "Usage:\n" if ( ( rc = parse_options ( argc, argv, &netboot_cmd, &opts ) ) != 0 )
" %s <interface>\n" return rc;
"\n"
"Attempts to boot the system from <interface>\n",
argv[0] );
return 1;
}
netdev_name = argv[1];
netdev = find_netdev ( netdev_name ); /* Parse interface */
if ( ! netdev ) { if ( ( rc = parse_netdev ( argv[optind], &netdev ) ) != 0 )
printf ( "%s: no such interface\n", netdev_name ); return rc;
return 1;
} /* Try to boot */
if ( ( rc = netboot ( netdev ) ) != 0 )
netboot ( netdev ); return rc;
/* Can never return success by definition */ return 0;
return 1;
} }
/** Booting commands */
struct command autoboot_commands[] __command = { struct command autoboot_commands[] __command = {
{ {
.name = "autoboot", .name = "autoboot",

View File

@ -15,7 +15,7 @@ struct net_device;
extern int shutdown_exit_flags; extern int shutdown_exit_flags;
extern int netboot ( struct net_device *netdev ); extern int netboot ( struct net_device *netdev );
extern void autoboot ( void ); extern int autoboot ( void );
extern int boot_next_server_and_filename ( struct in_addr next_server, extern int boot_next_server_and_filename ( struct in_addr next_server,
const char *filename ); const char *filename );
extern int boot_root_path ( const char *root_path ); extern int boot_root_path ( const char *root_path );

View File

@ -293,22 +293,24 @@ static void close_all_netdevs ( void ) {
/** /**
* Boot the system * Boot the system
*/ */
void autoboot ( void ) { int autoboot ( void ) {
struct net_device *boot_netdev; struct net_device *boot_netdev;
struct net_device *netdev; struct net_device *netdev;
int rc = -ENODEV;
/* If we have an identifable boot device, try that first */ /* If we have an identifable boot device, try that first */
close_all_netdevs(); close_all_netdevs();
if ( ( boot_netdev = find_boot_netdev() ) ) if ( ( boot_netdev = find_boot_netdev() ) )
netboot ( boot_netdev ); rc = netboot ( boot_netdev );
/* If that fails, try booting from any of the other devices */ /* If that fails, try booting from any of the other devices */
for_each_netdev ( netdev ) { for_each_netdev ( netdev ) {
if ( netdev == boot_netdev ) if ( netdev == boot_netdev )
continue; continue;
close_all_netdevs(); close_all_netdevs();
netboot ( netdev ); rc = netboot ( netdev );
} }
printf ( "No more network devices\n" ); printf ( "No more network devices\n" );
return rc;
} }