#!/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: