Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
[autoboot] Split main control flow out of main() into a new function …
…ipxe()

Signed-off-by: Michael Brown <mcb30@ipxe.org>
  • Loading branch information
mcb30 committed Mar 13, 2013
1 parent 1025835 commit 033f4c9
Show file tree
Hide file tree
Showing 3 changed files with 98 additions and 82 deletions.
84 changes: 2 additions & 82 deletions src/core/main.c
Expand Up @@ -14,106 +14,26 @@ Literature dealing with the network protocols:

FILE_LICENCE ( GPL2_OR_LATER );

#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <ipxe/init.h>
#include <ipxe/features.h>
#include <ipxe/shell.h>
#include <ipxe/image.h>
#include <ipxe/keys.h>
#include <ipxe/version.h>
#include <usr/prompt.h>
#include <usr/autoboot.h>
#include <config/general.h>

#define NORMAL "\033[0m"
#define BOLD "\033[1m"
#define CYAN "\033[36m"

/** The "scriptlet" setting */
struct setting scriptlet_setting __setting ( SETTING_MISC ) = {
.name = "scriptlet",
.description = "Boot scriptlet",
.tag = DHCP_EB_SCRIPTLET,
.type = &setting_type_string,
};

/**
* Prompt for shell entry
*
* @ret enter_shell User wants to enter shell
*/
static int shell_banner ( void ) {

/* Skip prompt if timeout is zero */
if ( BANNER_TIMEOUT <= 0 )
return 0;

/* Prompt user */
printf ( "\n" );
return ( prompt ( "Press Ctrl-B for the iPXE command line...",
( BANNER_TIMEOUT * 100 ), CTRL_B ) == 0 );
}

/**
* Main entry point
*
* @ret rc Return status code
*/
__asmcall int main ( void ) {
struct feature *feature;
struct image *image;
char *scriptlet;

/* Some devices take an unreasonably long time to initialise */
printf ( PRODUCT_SHORT_NAME " initialising devices..." );
initialise();
startup();
printf ( "ok\n" );

/*
* Print welcome banner
*
*
* If you wish to brand this build of iPXE, please do so by
* defining the string PRODUCT_NAME in config/general.h.
*
* While nothing in the GPL prevents you from removing all
* references to iPXE or http://ipxe.org, we prefer you not to
* do so.
*
*/
printf ( NORMAL "\n\n" PRODUCT_NAME "\n" BOLD "iPXE %s"
NORMAL " -- Open Source Network Boot Firmware -- "
CYAN "http://ipxe.org" NORMAL "\n"
"Features:", product_version );
for_each_table_entry ( feature, FEATURES )
printf ( " %s", feature->name );
printf ( "\n" );

/* Boot system */
if ( ( image = first_image() ) != NULL ) {
/* We have an embedded image; execute it */
image_exec ( image );
} else if ( shell_banner() ) {
/* User wants shell; just give them a shell */
shell();
} else {
fetch_string_setting_copy ( NULL, &scriptlet_setting,
&scriptlet );
if ( scriptlet ) {
/* User has defined a scriptlet; execute it */
system ( scriptlet );
free ( scriptlet );
} else {
/* Try booting. If booting fails, offer the
* user another chance to enter the shell.
*/
autoboot();
if ( shell_banner() )
shell();
}
}
ipxe ( NULL );

shutdown_exit();

Expand Down
1 change: 1 addition & 0 deletions src/include/usr/autoboot.h
Expand Up @@ -31,6 +31,7 @@ extern struct uri *
fetch_next_server_and_filename ( struct settings *settings );
extern int netboot ( struct net_device *netdev );
extern int autoboot ( void );
extern void ipxe ( struct net_device *netdev );

extern int pxe_menu_boot ( struct net_device *netdev );

Expand Down
95 changes: 95 additions & 0 deletions src/usr/autoboot.c
Expand Up @@ -30,11 +30,18 @@ FILE_LICENCE ( GPL2_OR_LATER );
#include <ipxe/uri.h>
#include <ipxe/open.h>
#include <ipxe/init.h>
#include <ipxe/keys.h>
#include <ipxe/version.h>
#include <ipxe/shell.h>
#include <ipxe/features.h>
#include <ipxe/image.h>
#include <usr/ifmgmt.h>
#include <usr/route.h>
#include <usr/dhcpmgmt.h>
#include <usr/imgmgmt.h>
#include <usr/prompt.h>
#include <usr/autoboot.h>
#include <config/general.h>

/** @file
*
Expand All @@ -47,6 +54,18 @@ FILE_LICENCE ( GPL2_OR_LATER );
#define EINFO_ENOENT_BOOT \
__einfo_uniqify ( EINFO_ENOENT, 0x01, "Nothing to boot" )

#define NORMAL "\033[0m"
#define BOLD "\033[1m"
#define CYAN "\033[36m"

/** The "scriptlet" setting */
struct setting scriptlet_setting __setting ( SETTING_MISC ) = {
.name = "scriptlet",
.description = "Boot scriptlet",
.tag = DHCP_EB_SCRIPTLET,
.type = &setting_type_string,
};

/**
* Perform PXE menu boot when PXE stack is not available
*/
Expand Down Expand Up @@ -423,3 +442,79 @@ int autoboot ( void ) {
printf ( "No more network devices\n" );
return rc;
}

/**
* Prompt for shell entry
*
* @ret enter_shell User wants to enter shell
*/
static int shell_banner ( void ) {

/* Skip prompt if timeout is zero */
if ( BANNER_TIMEOUT <= 0 )
return 0;

/* Prompt user */
printf ( "\n" );
return ( prompt ( "Press Ctrl-B for the iPXE command line...",
( BANNER_TIMEOUT * 100 ), CTRL_B ) == 0 );
}

/**
* Main iPXE flow of execution
*
* @v netdev Network device, or NULL
*/
void ipxe ( struct net_device *netdev ) {
struct feature *feature;
struct image *image;
char *scriptlet;

/*
* Print welcome banner
*
*
* If you wish to brand this build of iPXE, please do so by
* defining the string PRODUCT_NAME in config/general.h.
*
* While nothing in the GPL prevents you from removing all
* references to iPXE or http://ipxe.org, we prefer you not to
* do so.
*
*/
printf ( NORMAL "\n\n" PRODUCT_NAME "\n" BOLD "iPXE %s"
NORMAL " -- Open Source Network Boot Firmware -- "
CYAN "http://ipxe.org" NORMAL "\n"
"Features:", product_version );
for_each_table_entry ( feature, FEATURES )
printf ( " %s", feature->name );
printf ( "\n" );

/* Boot system */
if ( ( image = first_image() ) != NULL ) {
/* We have an embedded image; execute it */
image_exec ( image );
} else if ( shell_banner() ) {
/* User wants shell; just give them a shell */
shell();
} else {
fetch_string_setting_copy ( NULL, &scriptlet_setting,
&scriptlet );
if ( scriptlet ) {
/* User has defined a scriptlet; execute it */
system ( scriptlet );
free ( scriptlet );
} else {
/* Try booting. If booting fails, offer the
* user another chance to enter the shell.
*/
if ( netdev ) {
netboot ( netdev );
} else {
autoboot();
}
if ( shell_banner() )
shell();
}
}
}

0 comments on commit 033f4c9

Please sign in to comment.