~ubuntu-installer/kickseed/master

#! /bin/sh -e

. /usr/share/debconf/confmodule

. /lib/kickseed/kickseed.sh

. /lib/kickseed/cmdline.sh

. /lib/preseed/preseed.sh

ks_log () {

	logger -t kickseed "$@"

}

ks_preseed () {

	echo "$@" >> "$SPOOL/parse/preseed.cfg"

}

# No support for chrooting here, as /target isn't available yet and %pre

# scripts don't need it.

ks_run_script () {

	TYPE="$1"

	INTERPRETER="$2"

	SCRIPT="$4"

	logger -t kickseed "Running $TYPE script $SCRIPT using interpreter $INTERPRETER:"

	log-output -t kickseed --pass-stdout "$INTERPRETER" "$SCRIPT"

}

ks_write_script () {

	if [ "${1%/*}" != "$1" ]; then

		mkdir -p "${1%/*}"

	fi

	cat > "$1"

	chmod +x "$1"

}

preseed_fetch () {

	ln -sf "$1" "$2"

}

preseed_relative () {

	if [ -z "${1##/*}" ]; then

		return 1

	else

		return 0

	fi

}

FETCH_ERROR=

fetch_url () {

	local url="$1"

	local file="$2"

	iters=0

	while [ $iters -lt 3 ]; do

		# TODO proxy support? Would it be useful?

		# TODO add progress bar

		if FETCH_ERROR="$(wget -q "$url" -O "$file" 2>&1)"; then

			return 0

		fi

		echo "$FETCH_ERROR" | logger -t kickseed

		iters=$(($iters + 1))

	done

	return 1

}

KS="$(kickseed_cmdline /proc/cmdline ks)"

KSCFG="$(kickseed_file "$KS")"

case $KS in

	ftp://*/*|http://*/*|https://*/*)

		logger -t kickseed "Downloading kickstart file from $KS"

		if ! fetch_url "$KS" "$KSCFG"; then

			logger -t kickseed "... failed"

			db_subst initrd-kickseed/wget-failed URL "$KS"

			db_subst initrd-kickseed/wget-failed \

				ERROR "$FETCH_ERROR"

			db_input high initrd-kickseed/wget-failed || true

			db_go

			exit 1

		fi

		;;

	nfs:*:/*)

		logger -t kickseed "Retrieving kickstart file from $KS"

		file="${KS#nfs:}"

		server="${file%%:*}"

		file="${file#*:}"

		if ! mount -t nfs -o ro,intr,nolock \

				"$server:${file%/*}" "${KSCFG%/*}"; then

			logger -t kickseed "... failed"

			db_subst initrd-kickseed/nfs-mount-failed \

				DIR "${file%/*}"

			db_subst initrd-kickseed/nfs-mount-failed \

				SERVER "$server"

			db_input high initrd-kickseed/nfs-mount-failed || true

			db_go

			exit 1

		fi

		trap "umount '${KSCFG%/*}' || true" \

			EXIT HUP INT QUIT TERM

		;;

esac

case $KSCFG in

	/floppy/*)

		mountmedia floppy || true

		KSCFG="/media/${KSCFG#/floppy/}"

		trap 'umount /media || true' EXIT HUP INT QUIT TERM

		;;

	/media/*)

		device="${KSCFG#/media/}"

		device="${device%%/*}"

		# TODO: relies on having non-devfs compatibility paths

		if log-output -t kickseed \

				mount "/dev/$device" "/media/$device"; then

			trap "umount '/media/$device' || true" \

				EXIT HUP INT QUIT TERM

		fi

		;;

esac

case $KSCFG in

	'')

		# not handled in initrd (yet?)

		: ;;

	*)

		logger -t kickseed "Reading kickstart file from $KSCFG"

		kickseed "$KSCFG"

		if [ -s "$SPOOL/parse/preseed.cfg" ]; then

			preseed_location "$SPOOL/parse/preseed.cfg"

		fi

		;;

esac