b4d60a7ccb
Signed-off-by: Aaron Griffin <aaronmgriffin@gmail.com>
258 lines
7.7 KiB
Bash
Executable File
258 lines
7.7 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
CPIOCONFIG="$(pwd)/archiso-mkinitcpio.conf"
|
|
PKGLIST=""
|
|
QUIET="y"
|
|
FORCE="n"
|
|
MOUNTFILE="$(pwd)/mounts"
|
|
|
|
APPNAME=$(basename "${0}")
|
|
|
|
# usage: usage <exitvalue>
|
|
usage ()
|
|
{
|
|
echo "usage ${APPNAME} [options] command <command options>"
|
|
echo " general options:"
|
|
echo " -f Force overwrite of working files/squashfs image/bootable image"
|
|
echo " -i CPIO_CONFIG Use CONFIG file for mkinitcpio. default: ${CPIOCONFIG}"
|
|
echo " -p PACKAGE(S) Additional package(s) to install, can be used multiple times"
|
|
echo " -t <iso,disk> Type of image to create. Defaults to iso."
|
|
echo " -v Enable verbose output."
|
|
echo " -h This message."
|
|
echo " commands:"
|
|
echo " install <working dir> : install packages to the working dir"
|
|
echo " squash <working dir> <sqfs name> : generate a squashfs image of the working dir"
|
|
echo " img <working dir> <image name> : build an image from the working dir"
|
|
exit $1
|
|
}
|
|
|
|
while getopts 'i:P:p:a:t:fvh' arg; do
|
|
case "${arg}" in
|
|
i) CPIOCONFIG="${OPTARG}" ;;
|
|
p) PKGLIST="${PKGLIST} ${OPTARG}" ;;
|
|
t) IMG_TYPE="${OPTARG}" ;;
|
|
f) FORCE="y" ;;
|
|
v) QUIET="n" ;;
|
|
h|?) usage 0 ;;
|
|
*) echo "invalid argument '${arg}'"; usage 1 ;;
|
|
esac
|
|
done
|
|
|
|
shift $(($OPTIND - 1))
|
|
echo "ARGS: $@"
|
|
|
|
# do UID checking here so someone can at least get usage instructions
|
|
if [ "$EUID" != "0" ]; then
|
|
echo "error: This script must be run as root."
|
|
exit 1
|
|
fi
|
|
|
|
command_name="${1}"
|
|
work_dir=""
|
|
imgname=""
|
|
|
|
case "${command_name}" in
|
|
install) work_dir="${2}"; imgname="none" ;;
|
|
squash) work_dir="${2}"; imgname="${3}" ;;
|
|
img) work_dir="${2}"; imgname="${3}" ;;
|
|
*) echo "invalid command name '${command_name}'"; usage 1 ;;
|
|
esac
|
|
|
|
[ "x${imgname}" = "x" ] && (echo "Image name must be specified" && usage 1)
|
|
[ "x${work_dir}" = "x" ] && (echo "Please specify a working directory" && usage 1)
|
|
|
|
|
|
echo "${APPNAME} : Configuration Settings"
|
|
echo " mkinitcpio config file: ${CPIOCONFIG}"
|
|
echo " mount description file: ${MOUNTFILE}"
|
|
echo " working directory: ${work_dir}"
|
|
echo " image name: ${imgname}"
|
|
echo " image type: ${IMG_TYPE}"
|
|
|
|
|
|
_kversion ()
|
|
{
|
|
# Man this is gross... we need a better way to get the kernel version
|
|
source ${work_dir}/etc/mkinitcpio.d/kernel26.kver
|
|
echo ${ALL_kver}
|
|
}
|
|
|
|
# usage: _pacman <packages>...
|
|
_pacman ()
|
|
{
|
|
local ret
|
|
if [ "${QUIET}" = "y" ]; then
|
|
mkarchroot -f ${work_dir} $* 2>&1 >/dev/null
|
|
ret=$?
|
|
else
|
|
mkarchroot -f ${work_dir} $*
|
|
ret=$?
|
|
fi
|
|
|
|
# Cleanup
|
|
find "${work_dir}" -name *.pacnew -name *.pacsave -name *.pacorig -delete
|
|
|
|
if [ $ret -ne 0 ]; then
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
command_install () {
|
|
echo "====> Installing packages to '${work_dir}'"
|
|
if [ -e "${work_dir}" -a "${FORCE}" = "n" ]; then
|
|
echo "error: Working dir '${work_dir}' already exists, aborting."
|
|
exit 1
|
|
fi
|
|
|
|
mkdir -p "${work_dir}"
|
|
|
|
if [ "${PKGLIST}" = "" ]; then
|
|
echo "error: no packages to install"
|
|
exit 1
|
|
fi
|
|
|
|
echo "Installing packages..."
|
|
_pacman "${PKGLIST}"
|
|
|
|
if [ -d "${work_dir}/lib/modules/" ]; then
|
|
echo "Updating kernel module dependencies"
|
|
kernelver=$(_kversion)
|
|
depmod -a -b "${work_dir}" "${kernelver}"
|
|
fi
|
|
|
|
echo "Cleaning up what we can"
|
|
if [ -d "${work_dir}/boot/" ]; then
|
|
# remove the initcpio images that were generated for the host system
|
|
find "${work_dir}/boot" -name *.img -delete
|
|
fi
|
|
|
|
#TODO is this needed? do it at the Makefile level?
|
|
if [ -d "${work_dir}/home/" ]; then
|
|
echo "Creating default home directory"
|
|
install -d -o1000 -g100 -m0755 "${work_dir}/home/arch"
|
|
fi
|
|
|
|
# delete a lot of unnecessary cache/log files
|
|
kill_dirs="var/abs var/cache/man var/cache/pacman var/log/* var/mail tmp/* initrd"
|
|
for x in ${kill_dirs}; do
|
|
if [ -e "${work_dir}/${x}" ]; then
|
|
rm -rf "${work_dir}/${x}"
|
|
fi
|
|
done
|
|
|
|
# pacman DBs are big, delete all sync dbs
|
|
rm -rf "${work_dir}/var/lib/pacman/sync"
|
|
|
|
#TODO test for existance of images in work_dir
|
|
cp "${MOUNTFILE}" "${work_dir}/mounts"
|
|
}
|
|
|
|
# command_squash path image
|
|
command_squash () {
|
|
echo "====> Generating SquashFS image ${imgname}"
|
|
if [ -e "${imgname}" ]; then
|
|
if [ "${FORCE}" = "y" ]; then
|
|
echo -n "Removing old SquashFS image..."
|
|
rm "${imgname}"
|
|
echo "done."
|
|
else
|
|
echo "error: SquashFS image '${imgname}' already exists, aborting."
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
echo "Creating SquashFS image. This may take some time..."
|
|
start=$(date +%s)
|
|
if [ "${QUIET}" = "y" ]; then
|
|
mksquashfs "${work_dir}" "${imgname}" -noappend >/dev/null
|
|
else
|
|
mksquashfs "${work_dir}" "${imgname}" -noappend
|
|
fi
|
|
minutes=$(echo $start $(date +%s) | awk '{ printf "%0.2f",($2-$1)/60 }')
|
|
echo "Image creation done in $minutes minutes."
|
|
}
|
|
|
|
command_img () {
|
|
echo "====> Making bootable image"
|
|
if [ -e "${imgname}" ]; then
|
|
if [ "${FORCE}" = "y" ]; then
|
|
echo "Removing existing bootable image..."
|
|
rm -rf "${imgname}"
|
|
else
|
|
echo "error: Image '${imgname}' already exists, aborting."
|
|
exit 1
|
|
fi
|
|
fi
|
|
if [ ! -e "${CPIOCONFIG}" ]; then
|
|
echo "error: mkinitcpio config '${CPIOCONFIG}' does not exist, aborting."
|
|
exit 1
|
|
fi
|
|
|
|
kernelver=$(_kversion)
|
|
basedir=${work_dir}
|
|
[ "${work_dir:0:1}" != "/" ] && basedir="$(pwd)/${work_dir}"
|
|
echo "Generating initcpio for image..."
|
|
if [ "${QUIET}" = "y" ]; then
|
|
mkinitcpio -c "${CPIOCONFIG}" -b "${basedir}" -k "${kernelver}" -g "${work_dir}/boot/archiso.img" >/dev/null
|
|
ret=$?
|
|
else
|
|
mkinitcpio -c "${CPIOCONFIG}" -b "${basedir}" -k "${kernelver}" -g "${work_dir}/boot/archiso.img"
|
|
ret=$?
|
|
fi
|
|
if [ $ret -ne 0 ]; then
|
|
echo "error: initcpio image creation failed..."
|
|
exit 1
|
|
fi
|
|
|
|
USE_GRUB=1
|
|
bootflags=""
|
|
if [ "$USE_GRUB" = "1" ]; then
|
|
_pacman grub #grub-gfx ??
|
|
mkdir -p "${work_dir}/boot/grub"
|
|
cp "${work_dir}/usr/lib/grub/i386-pc/*" "${work_dir}/boot/grub"
|
|
|
|
# copy over kernel and grub configs for boot
|
|
if [ -d "${work_dir}/boot" -a -e "${DEF_CONFIG_DIR}/boot" ]; then
|
|
rm -rf "${work_dir}/boot"
|
|
cp -r "${work_dir}/boot" "${work_dir}"
|
|
cp -rf "${DEF_CONFIG_DIR}/boot" "${work_dir}"
|
|
fi
|
|
bootflags="boot/grub/stage2_eltorito"
|
|
else
|
|
_pacman isolinux
|
|
bootflags="boot/whatever/isolinux"
|
|
fi
|
|
|
|
if [ "x$IMG_TYPE" == "xdisk" ]; then
|
|
echo "Creating DISK image..."
|
|
mkusbimg "${work_dir}" "${imgname}"
|
|
else
|
|
echo "Creating ISO image..."
|
|
if [ -z "$bootflags" ]; then
|
|
echo "Eeek, no boot flags found. This probably won't be bootable"
|
|
fi
|
|
qflag=""
|
|
[ "${QUIET}" = "y" ] && qflag="-q"
|
|
mkisofs ${qflag} -r -l $bootflags -uid 0 -gid 0 \
|
|
-no-emul-boot -boot-load-size 4 -boot-info-table \
|
|
-publisher "Arch Linux <archlinux.org>" \
|
|
-input-charset=UTF-8 -p "prepared by mkarchiso" \
|
|
-A "Arch Linux Live/Rescue CD" \
|
|
-o "${imgname}" "${work_dir}"
|
|
fi
|
|
}
|
|
|
|
# Go through the main commands in order. If 'all' was specified, then we want
|
|
# to do everything. Start with 'install'.
|
|
if [ "${command_name}" = "install" -o "${command_name}" = "all" ]; then
|
|
command_install
|
|
fi
|
|
if [ "${command_name}" = "squash" -o "${command_name}" = "all" ]; then
|
|
command_squash
|
|
fi
|
|
if [ "${command_name}" = "img" -o "${command_name}" = "all" ]; then
|
|
command_img
|
|
fi
|
|
|
|
# vim:ts=4:sw=4:et:
|