void-packages/srcpkgs/vmklive/files/mklive.sh.in

#!/bin/sh
#
# Public Domain, 2009-2012 - Juan RP <xtraeme@gmail.com>
#
trap "echo; error_out $?" INT QUIT

info_msg()
{
	printf "\033[1m$@\n\033[m"
}

mount_pseudofs()
{
	local fs

	if [ -n "$ROOTDIR" ]; then
		mount --bind "$ROOTDIR" "$ROOTFS" || error_out $?
	fi

	for fs in sys proc dev; do
		if [ ! -d "$ROOTFS/$fs" ]; then
			mkdir -p "$ROOTFS/$fs"
		fi
		mount --bind /$fs "$ROOTFS/$fs" || error_out $?
	done
}

umount_pseudofs()
{
	local fs

	for fs in sys proc dev; do
		umount -f "$ROOTFS/$fs" >/dev/null 2>&1
	done
	if [ -n "$1" -a -n "$ROOTDIR" ]; then
		umount -f "$ROOTFS" >/dev/null 2>&1
	fi
}

error_out()
{
	umount_pseudofs

	info_msg "There was an error! cleaning up $BUILDDIR, exiting..."

	[ -d "$BUILDDIR" ] && rm -rf "$BUILDDIR"
	[ -f "$LOGFILE" ] && rm -f "$LOGFILE"

	exit 1
}

write_etc_motd()
{
	cat >> "$ROOTFS/etc/motd" <<_EOF
###############################################################################
  Autogenerated by vmklive @@MKLIVE_VERSION@@.
-------------------------------------------------------------------------------

Welcome to the Void Linux Live system, you have been autologged in.
This user has full sudo(8) permissions without any password, be careful
executing commands through sudo(8).

To play with package management use the xbps-bin(8) and xbps-repo(8)
utilities. Please visit:

	http://code.google.com/p/xbps/

for more information and/or documentation about using the X Binary
Package System. If you think it is useful, please make a donation
to improve further development from the above URL, thanks.

To install Void Linux type 'sudo void-installer'.

###############################################################################
_EOF
}

write_default_isolinux_conf()
{
	local kver="$1"

	if [ -r "$SPLASH_IMAGE" ]; then
		BACKGROUND="MENU BACKGROUND $(basename $SPLASH_IMAGE)"
	fi

	cat >> "$ISOLINUX_CFG" << _EOF
UI vesamenu.c32
PROMPT 0
TIMEOUT 100
ONTIMEOUT linux

MENU TABMSG Press ENTER to boot or TAB to edit a menu entry
MENU AUTOBOOT BIOS default device boot in # second{,s}...
$BACKGROUND
MENU WIDTH 78
MENU MARGIN 1
MENU ROWS 4
MENU VSHIFT 2
MENU TIMEOUTROW 8
MENU TABMSGROW 2
MENU CMDLINEROW 11
MENU HELPMSGROW 16
MENU HELPMSGENDROW 29

MENU COLOR title        * #FF5255FF *
MENU COLOR border       * #00000000 #00000000 none
MENU COLOR sel          * #ffffffff #FF5255FF *

LABEL linux
MENU LABEL Boot Void GNU/Linux ${kver} ($(uname -m))
KERNEL /casper/vmlinuz
INITRD /casper/initrd.lz
APPEND boot=casper vga=current keymap=${KEYMAP:-us} locale=${LOCALE:-en_US.UTF-8} ro

LABEL linuxtoram
MENU LABEL Boot Void GNU/Linux ${kver} (RAM) ($(uname -m))
KERNEL /casper/vmlinuz
INITRD /casper/initrd.lz
APPEND boot=casper vga=current toram keymap=${KEYMAP:-us} locale=${LOCALE:-en_US.UTF-8} ro

LABEL c
MENU LABEL Boot first HD found by the BIOS
LOCALBOOT 0x80
_EOF
}

write_conf_file()
{
	cat > "$1" <<_EOF
# *-*- sh -*-*
# Default configuration file for vmklive-@VERSION@.
#
# List of packages to be installed into the live image.
# At least 'base-system' or 'base-system-live' is required.
PACKAGE_LIST="base-system-live"

# Syslinux splash image.
SPLASH_IMAGE=/usr/share/void-artwork/splash.png

# Default keymap to use.
KEYMAP=us

# Default locale to use.
LOCALE=en_US.UTF-8

# Path to XBPS utilities.
#XBPS_BIN_CMD=xbps-bin
#XBPS_REPO_CMD=xbps-repo
#XBPS_UHELPER_CMD=xbps-uhelper

# XBPS cache directory to install packages from.
#REPOSITORY_CACHE=/blah/foo

_EOF
	chmod 644 "$1"
}

usage()
{
	cat <<_EOF
Usage: $(basename $0) [options] <kernel-version>

Options:
 -C file		Path to configuration file (defaults to ~/.mklive.conf)
 -c (gzip|bzip2|xz) 	Compression type for the squashfs/initramfs image.
 -o outfile		Output file name for the ISO image.
 -r rootdir		Rootfs directory.
 -s splash		Splash image file for isolinux.
 -v volname		ISO Volume name.
_EOF
	exit 1
}

#
# main()
#
while getopts "C:c:o:r:s:v:" opt; do
	case $opt in
	C) CONFIG_FILE="$OPTARG";;
	c) COMPRESSTYPE="$OPTARG";;
	o) OUTPUT_FILE="$OPTARG";;
	r) ROOTDIR="$OPTARG";;
	s) SPLASH_IMAGE="$OPTARG";;
	v) ISO_VOLUME="$OPTARG";;
	esac
done
shift $(($OPTIND - 1))

KERNELVERSION="$1"
if [ -z "$KERNELVERSION" ]; then
	usage
fi

# Check that specific kernel version contains the unionfs module.
if [ ! -r "/boot/config-${KERNELVERSION}" ]; then
	echo "Missing /boot/config-${KERNELVERSION} file, exiting..."
	exit 1
elif ! grep -Eq "^CONFIG_UNION_FS=(y|m)$" /boot/config-${KERNELVERSION}; then
	echo "kernel ${KERNELVERSION} does not contain unionfs support, exiting..."
	exit 1
fi

# Set defaults
if [ -z "$CONFIG_FILE" ]; then
	CONFIG_FILE="$HOME/.mklive.conf"
fi
if [ -z "$OUTPUT_FILE" ]; then
	OUTPUT_FILE="$HOME/void-live-$(uname -m)-${KERNELVERSION}-$(date +%Y%m%d).iso"
fi
LOGFILE="$(mktemp -t vmklive-XXXXXXXXXX.log)"

if [ -z "$ISO_VOLUME" ]; then
	ISO_VOLUME="Void Linux Live $(uname -m) ${KERNELVERSION}"
fi
if [ -z "$SYSLINUX_DATADIR" ]; then
	SYSLINUX_DATADIR=/usr/share/syslinux
fi
if [ -z "$SPLASH_IMAGE" ]; then
	SPLASH_IMAGE=/usr/share/void-artwork/splash.png
fi
if [ -z "$XBPS_REPO_CMD" ]; then
	XBPS_REPO_CMD=xbps-repo
fi
if [ -z "$XBPS_BIN_CMD" ]; then
	XBPS_BIN_CMD=xbps-bin
fi
if [ -z "$XBPS_UHELPER_CMD" ]; then
	XBPS_UHELPER_CMD=xbps-uhelper
fi
if [ -z "$COMPRESSTYPE" ]; then
	COMPRESSTYPE=xz
fi

# Create or read configuration file.
if [ ! -r $CONFIG_FILE ]; then
	info_msg "Creating config file at $CONFIG_FILE."
	write_conf_file $CONFIG_FILE
fi

. $CONFIG_FILE

if [ -z "$PACKAGE_LIST" ]; then
	PACKAGE_LIST="base-system-live"
else
	PACKAGE_LIST="base-system-live $PACKAGE_LIST"
fi
BUILDDIR=$(mktemp --tmpdir=$HOME -d) || exit 1
BUILDDIR=$(readlink -f $BUILDDIR)
ROOTFS="$BUILDDIR/casper/rootfs"
ISOLINUX_DIR="$BUILDDIR/isolinux"
ISOLINUX_CFG="$ISOLINUX_DIR/isolinux.cfg"

if [ ! -f $SYSLINUX_DATADIR/isolinux.bin ]; then
	echo "Missing required isolinux files in $SYSLINUX_DATADIR, exiting..."
	error_out
fi

# Check for root permissions.
if [ "$(id -u)" -ne 0 ]; then
	echo "Must be run as root, exiting..."
	error_out
fi

#
# Mount pseudofs in the target rootfs.
#
mount_pseudofs
mkdir -p "$ROOTFS/tmp"

XBPS_ARGS="-r $ROOTFS -y"
if [ -n "$REPOSITORY_CACHE" ]; then
	XBPS_ARGS="$XBPS_ARGS -c $REPOSITORY_CACHE"
fi
XBPS_VERSION=$($XBPS_BIN_CMD -V|awk '{print $2}')
case $XBPS_VERSION in
	# XBPS >= 0.12
	[0-9].[1-9][2-9]*);;
	# XBPS < 0.12
	*) purge_flag="-p" ;;
esac

info_msg "Redirecting stdout/stderr to $LOGFILE..."
info_msg "[1/9] Installing packages into the rootfs..."
for f in ${PACKAGE_LIST}; do
	info_msg "  $f"
done
if [ -z "$ROOTDIR" ]; then
	${XBPS_BIN_CMD} ${XBPS_ARGS} install ${PACKAGE_LIST} \
		2>&1|cat >> $LOGFILE || error_out
	${XBPS_BIN_CMD} ${XBPS_ARGS} autoupdate \
		2>&1|cat >> $LOGFILE || error_out
	${XBPS_BIN_CMD} ${XBPS_ARGS} ${purge_flag} autoremove \
		2>&1|cat >> $LOGFILE || error_out
fi

${XBPS_BIN_CMD} -r "$ROOTFS" list > \
	"${OUTPUT_FILE%.iso}"-package-list.txt || error_out

#
# Prepare /etc/motd.
#
info_msg "[2/9] Creating /etc/motd..."
write_etc_motd

#
# Create the initramfs with XZ compression.
#
info_msg "[3/9] Creating initramfs image ($COMPRESSTYPE)..."
mkinitramfs -c ${COMPRESSTYPE} \
	-o "$BUILDDIR/casper/initrd.lz" ${KERNELVERSION} \
	2>/dev/null || error_out
mkdir -p "$ROOTFS"/cow

#
# Copy the linux image to the target directory.
#
info_msg "[4/9] Copying kernel image/modules..."
cp -f /boot/vmlinuz-${KERNELVERSION} "$BUILDDIR/casper/vmlinuz" || error_out $?
mkdir -p "$ROOTFS/lib/modules"
cp -a /lib/modules/${KERNELVERSION} "$ROOTFS/lib/modules" || error_out $?

# XXX: install GNU find, grep and xargs.
for f in find grep xargs; do
	install -Dm755 /usr/bin/${f} "$ROOTFS/usr/bin/$f" || error_out $?
done

# XXX: install lsblk and blkid from util-linux.
install -Dm755 /bin/lsblk "$ROOTFS/usr/bin/lsblk" || error_out $?
install -Dm755 /sbin/blkid "$ROOTFS/sbin/blkid" || error_out $?

rm -f "$ROOTFS/bin/grep" || error_out $?
#
# Copy required xbps-casper files for autologin.
#
install -Dm644 /etc/casper.conf "$ROOTFS/etc/casper.conf" || error_out $?
install -Dm755 /sbin/casper-getty "$ROOTFS/sbin/casper-getty" || error_out $?
install -Dm755 /sbin/casper-login "$ROOTFS/sbin/casper-login" || error_out $?

#
# The pseudofs aren't needed anymore in target rootfs.
#
umount_pseudofs

# Save a few MBs from /usr/share.
for d in info man examples doc; do
	[ -d "$ROOTFS/usr/share/${d}" ] && rm -rf "$ROOTFS/usr/share/${d}"
done

#
# Copy required isolinux files in the target rootfs.
#
info_msg "[5/9] Copying isolinux files..."
mkdir -p "$ISOLINUX_DIR"
cp -f $SYSLINUX_DATADIR/isolinux.bin "$ISOLINUX_DIR"
cp -f $SYSLINUX_DATADIR/vesamenu.c32 "$ISOLINUX_DIR"
write_default_isolinux_conf $(uname -r)
if [ -f "$SPLASH_IMAGE" ]; then
	cp -f $SPLASH_IMAGE "$ISOLINUX_DIR"
fi

#
# Prepare the squashed rootfs image.
#
info_msg "[6/9] Creating squashfs image ($COMPRESSTYPE) from rootfs..."
mksquashfs "$ROOTFS" "$BUILDDIR/casper/filesystem.squashfs" \
	-root-becomes / -comp ${COMPRESSTYPE} \
	2>&1 | cat >> $LOGFILE || error_out
chmod 444 "$BUILDDIR/casper/filesystem.squashfs" || error_out $?

if [ -n "$ROOTDIR" ]; then
	umount_pseudofs yes
else
	info_msg "[7/9] Removing rootfs directory..."
	rm -rf "$ROOTFS" || error_out $?
fi

#
# Prepare the ISO image.
#
info_msg "[8/9] Building ISO image..."
mkisofs -J -r -V "$ISO_VOLUME" -b isolinux/isolinux.bin \
	-c isolinux/boot.cat -no-emul-boot \
	-boot-load-size 4 -boot-info-table \
	-o "$OUTPUT_FILE" "$BUILDDIR" 2>&1 | cat >>$LOGFILE || error_out $?

info_msg "[9/9] Removing build directory..."
rm -rf "$BUILDDIR" || error_out $?

hsize=$(du -sh "$OUTPUT_FILE"|awk '{print $1}')
info_msg "Created $(readlink -f $OUTPUT_FILE) ($hsize) successfully."

exit 0