diff --git a/src/core/main.c b/src/core/main.c index 3295feaf..ca62db25 100644 --- a/src/core/main.c +++ b/src/core/main.c @@ -14,26 +14,54 @@ Literature dealing with the network protocols: **************************************************************************/ +#include #include +#include #include #include #include +#define NORMAL "\033[0m" +#define BOLD "\033[1m" +#define CYAN "\033[36m" + +static struct feature features[0] __table_start ( struct feature, features ); +static struct feature features_end[0] __table_end ( struct feature, features ); + /** * Main entry point * * @ret rc Return status code */ __cdecl int main ( void ) { + struct feature *feature; initialise(); startup(); - if ( shell_banner() ) + /* Print welcome banner */ + printf ( NORMAL "\n\n\n" BOLD "gPXE " VERSION + NORMAL " -- Open Source Boot Firmware -- " + CYAN "http://etherboot.org" NORMAL "\n" + "Features:" ); + for ( feature = features ; feature < features_end ; feature++ ) + printf ( " %s", feature->name ); + printf ( "\n" ); + + /* Prompt for shell */ + if ( shell_banner() ) { + /* User wants shell; just give them a shell */ shell(); - else + } else { + /* User doesn't want shell; try booting. If booting + * fails, offer a second chance to enter the shell for + * diagnostics. + */ autoboot(); - + if ( shell_banner() ) + shell(); + } + shutdown(); return 0; diff --git a/src/hci/shell_banner.c b/src/hci/shell_banner.c index 62da487c..92cd17dd 100644 --- a/src/hci/shell_banner.c +++ b/src/hci/shell_banner.c @@ -18,7 +18,6 @@ #include #include -#include #include #include @@ -30,13 +29,6 @@ #define BANNER_TIMEOUT ( 2 * TICKS_PER_SEC ) -#define NORMAL "\033[0m" -#define BOLD "\033[1m" -#define CYAN "\033[36m" - -static struct feature features[0] __table_start ( struct feature, features ); -static struct feature features_end[0] __table_end ( struct feature, features ); - /** * Print shell banner and prompt for shell entry * @@ -44,18 +36,9 @@ static struct feature features_end[0] __table_end ( struct feature, features ); */ int shell_banner ( void ) { unsigned long timeout = ( currticks() + BANNER_TIMEOUT ); - struct feature *feature; - int key; int enter_shell = 0; + int key; - /* Print welcome banner */ - printf ( NORMAL "\n\n\n" BOLD "gPXE " VERSION - NORMAL " -- Open Source Boot Firmware -- " - CYAN "http://etherboot.org" NORMAL "\n" - "Features:" ); - for ( feature = features ; feature < features_end ; feature++ ) { - printf ( " %s", feature->name ); - } printf ( "\nPress Ctrl-B for the gPXE command line..." ); /* Wait for key */