X-Git-Url: https://git.ozlabs.org/?a=blobdiff_plain;f=utils%2Fpb-console;h=5ba98cc4692564b435d1f7d271bf07355e838c06;hb=5c726bd40a4a745e115a22e336131be8f6666b55;hp=636c5ad36dff65d449d310309989d8c6967e6e65;hpb=d92dcc900005d21911748cc85cd21df04ebad76d;p=petitboot

diff --git a/utils/pb-console b/utils/pb-console
index 636c5ad..5ba98cc 100644
--- a/utils/pb-console
+++ b/utils/pb-console
@@ -1,49 +1,145 @@
 #!/bin/sh
+#
+# Petitboot utility script for running a petitboot UI program
+# on a console tty.
+#
 
 ui=petitboot-nc
 shell=sh
-
+getty=/sbin/getty
+use_getty=0
 detach=0
+pb_config=pb-config
+
+usage() {
+	cat >&2 <<EOF
+pb-console [OPTIONS] -- [ARGS]
+OPTIONS
+     -d, --detach
+             Start in a detached (background) state.
+
+     -g, --getty[=PATH]
+             Start a getty (specified by PATH, otherwise $getty),
+             passing additional ARGS to the getty process
+
+     -s, --shell=PATH
+             Use PATH as the exit-to-shell shell
+
+     -u, --ui=PATH
+             Use PATH as the petitboot UI
+
+     -h, --help
+             Print a help message.
+EOF
+	exit 1
+}
 
-while [ -n "$1" ]
+opts=$(getopt --options 'hdg::s:u:' \
+              --long 'help,detach,getty::,shell:,ui:' \
+              -- "$@")
+[ $? = 0 ] || exit 1
+
+eval set -- "$opts"
+
+while :
 do
 	case "$1" in
-	-d)
+	-d | --detach)
 		detach=1
+		shift
+		;;
+	-g | --getty)
+		use_getty=1
+		getty_arg="$2"
+		shift 2
+		;;
+	-s | --shell)
+		shell="$2"
+		shift 2
+		;;
+	-u | --ui)
+		ui="$2"
+		shift 2
+		;;
+	--help | -h)
+		usage
 		;;
 	--)
+		shift
+		break
 		;;
 	*)
-		console=$1
-		;;
+		echo "getopt error"
+		exit 1
 	esac
-	shift
 done
 
-if [ -n "$console" -a "${console#/dev/}" = "$console" ]
+if [ "$use_getty" = 1 ]
 then
-	console="/dev/$console"
-fi
+	if [ -n "$getty_arg" ]
+	then
+		getty="$getty_arg"
+	fi
 
-pb_loop() {
-	while :
+	login_arg="-l$0"
+	for ttyarg in "$@"
 	do
-		$ui
-		echo "Exiting petitboot. Type 'exit' to return"
-		$shell
+		# If the getty args include autologin don't override with -l
+		# and leave calling petitboot-nc to the user's init
+		if [ "$ttyarg" == "-a" ]
+		then
+			login_arg=""
+		fi
 	done
-}
 
-if [ -n "$console" ]
+	if [ "$detach" = 1 ]
+	then
+		setsid -c $getty $login_arg "$@" &
+		exit
+	else
+		exec $getty $login_arg "$@"
+	fi
+fi
+
+for f in /etc/environment /etc/locale
+do
+	if [ -e "$f" ]
+	then
+		export $(cat "$f")
+	fi
+done
+
+# we force local terminals to use the linux termcap definition
+case "$(tty)" in
+/dev/tty[0-9]*)
+	export TERM=linux
+	;;
+esac
+
+# we may have been run from udev - ensure we have a sensible PATH
+if [ -z "$PATH" ]
 then
-	exec <$console >$console 2>&1
+	PATH=/usr/bin:/usr/sbin:/bin:/sbin
 fi
+PATH=/var/lib/pb-plugins/bin:$PATH
+export PATH
 
-if [ "$detach" = 1 ]
+verbose_opt=
+if $pb_config debug | grep -q enabled
 then
-	(
-		pb_loop
-	) &
-else
-	pb_loop
+	verbose_opt=--verbose
 fi
+
+# kernel messages may write over the ncurses ui - change log level to only
+# show particularly important messages
+dmesg -n 1
+
+trap '' SIGINT
+trap 'reset; echo "SIGTERM received, booting..."; sleep 2' SIGTERM
+
+while :
+do
+	$ui $verbose_opt
+	reset
+	$shell
+done