From 84aa702ff8b66f883e9ac5792b26d4e9d0760573 Mon Sep 17 00:00:00 2001
From: Michael Brown
Date: Mon, 22 Nov 2010 20:20:33 +0000
Subject: [script] Allow "exit" to exit a script

Signed-off-by: Michael Brown <mcb30@ipxe.org>
---
 src/core/exec.c | 61 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 57 insertions(+), 4 deletions(-)

(limited to 'src/core')

diff --git a/src/core/exec.c b/src/core/exec.c
index 20b97a6d8..60d8cb975 100644
--- a/src/core/exec.c
+++ b/src/core/exec.c
@@ -29,6 +29,7 @@ FILE_LICENCE ( GPL2_OR_LATER );
 #include <assert.h>
 #include <ipxe/tables.h>
 #include <ipxe/command.h>
+#include <ipxe/parseopt.h>
 #include <ipxe/settings.h>
 
 /** @file
@@ -37,9 +38,8 @@ FILE_LICENCE ( GPL2_OR_LATER );
  *
  */
 
-/* Avoid dragging in getopt.o unless a command really uses it */
-int optind;
-int nextchar;
+/** Shell exit flag */
+int shell_exit;
 
 /**
  * Execute command
@@ -270,6 +270,9 @@ int system ( const char *command ) {
 	int count;
 	int rc = 0;
 
+	/* Reset exit flag */
+	shell_exit = 0;
+
 	/* Perform variable expansion */
 	expcmd = expand_command ( command );
 	if ( ! expcmd )
@@ -294,6 +297,10 @@ int system ( const char *command ) {
 			argv[argc] = NULL;
 			rc = execv ( argv[0], argv );
 
+			/* Check exit flag */
+			if ( shell_exit )
+				break;
+
 			/* Handle terminator */
 			if ( terminator ( rc ) )
 				break;
@@ -307,7 +314,7 @@ int system ( const char *command ) {
 }
 
 /**
- * The "echo" command
+ * "echo" command
  *
  * @v argc		Argument count
  * @v argv		Argument list
@@ -328,3 +335,49 @@ struct command echo_command __command = {
 	.name = "echo",
 	.exec = echo_exec,
 };
+
+/** "exit" options */
+struct exit_options {};
+
+/** "exit" option list */
+static struct option_descriptor exit_opts[] = {};
+
+/** "exit" command descriptor */
+static struct command_descriptor exit_cmd =
+	COMMAND_DESC ( struct exit_options, exit_opts, 0, 1,
+		       "[<status>]", "" );
+
+/**
+ * "exit" command
+ *
+ * @v argc		Argument count
+ * @v argv		Argument list
+ * @ret rc		Return status code
+ */
+static int exit_exec ( int argc, char **argv ) {
+	struct exit_options opts;
+	unsigned int exit_code = 0;
+	int rc;
+
+	/* Parse options */
+	if ( ( rc = parse_options ( argc, argv, &exit_cmd, &opts ) ) != 0 )
+		return rc;
+
+	/* Parse exit status, if present */
+	if ( optind != argc ) {
+		if ( ( rc = parse_integer ( argv[optind], &exit_code ) ) != 0 )
+			return rc;
+	}
+
+	/* Set exit flag */
+	shell_exit = 1;
+
+	return exit_code;
+}
+
+/** "exit" command */
+struct command exit_command __command = {
+	.name = "exit",
+	.exec = exit_exec,
+};
+
-- 
cgit v1.2.3-55-g7522