kickstart-usb-creator

#!/bin/bash
#
# Creates usb stick for kickstart installations

script=$(readlink -f "$0")
script_path=$(dirname "$script")
script_name=$(basename "$script")

# shellcheck source=/dev/null
if ! source "$script_path/_functions"; then
  echo "Function inclusion failed."
  exit 1
fi

# 8GB usb min size limit
# todo determine min usb size based on iso size
device_size_limit=$((3 * 1000 * 1000 * 1000))

function help {
  echo
  echo "Usage: $scriptName --iso <isoFile> --device </dev/sdX> --ks <ksFile>"
  echo
  echo "Creates bootable USB stick for kickstart installations."
  echo "  --iso    Path to install .iso file"
  echo "  --device Use the specified unmounted device as target"
  echo "  --ks     Path to kickstart config"
  echo
  exit 1
}

function checkOptions {
  if [ -z "$ks" ]; then
    _error "Missing path to kickstart file."
    return 1
  fi
  if [ -z "$iso" ]; then
    _error "Missing path to .iso file."
    return 1
  fi
  if [ -z "$device" ]; then
    _error "Missing device."
    return 1
  fi
}



# process options
while [[ $# -ge 1 ]]; do
  key="$1"
  shift
  case "$key" in
    --device)
      device="$1"
      shift
      ;;
    --iso)
      iso="$1"
      shift
      ;;
    --ks)
      ks="$1"
      shift
      ;;
    --help)
      help
      ;;
    *)
      # unknown
    ;;
  esac
done

# required cmd line tools
_check_tools syslinux mkfs.vfat partprobe
if ! mbr_bin_path=$(_get_mbr_bin_path); then
  exit 1
fi

if ! checkOptions; then
  help
fi

echo -n "All data on ´$device´ will be deleted, proceed? [y/N] "
read -rn 1 confirm
echo ""
if [ "$confirm" != "y" ]; then
  echo ""
  exit 1
fi

cleanup() {
  _umount "$(_mnt_point "${device}1")"
  _umount "$(_mnt_point "${device}2")"
  _umount "$(_mnt_point "${iso}")"
}

execute() {
  if ! _exec "$*"; then
    handle_code 1
  fi
}

handle_code() {
  local code="$1"
  if [ "$code" -ne 0 ]; then
    cleanup
    exit 1
  fi
}

check_device() {
  _log "Check device ´$1´"
  local device="$1"
  local deviceSize
  deviceSize=$((512 * $(cat "/sys/block/$(basename "$device")/size")))
  if [ "${deviceSize}" -lt "${device_size_limit}" ]; then
    _error "Device is too small. Current: ${deviceSize} Expected min: ${device_size_limit}"
    handle_code 1
  fi
}

prepare_device() {
  _log "Prepare device ´$1´"
  local device="$1"
  _log "Create partitions..."
  execute "echo -e 'o\nn\np\n1\n\n+300M\na\nn\np\n2\n\n\nw' | fdisk ${device}"
  execute "mkfs.vfat -n BOOT ${device}1"
  execute "mkfs.ext4 -L DATA ${device}2"
  execute "dd conv=notrunc bs=440 count=1 if=\"$mbr_bin_path\" of=${device}"
  execute "syslinux -i ${device}1"
}

prepare_boot() {
  _log "Fill boot partition content..."
  local device="$1"
  local iso_mnt="$2"
  device_mnt=$(_mount_device "${device}1")
  handle_code $?
  ## BIOS-Boot
  execute "cp $iso_mnt/isolinux/* $device_mnt/"
  execute "mv \"$device_mnt/isolinux.cfg\" \"$device_mnt/syslinux.cfg\""
  execute "sed -i '0,/.*append.*/s//\ \ append\ biosdevname=0\ net.ifnames=0\ initrd=initrd.img\ method=hd:LABEL=DATA:\/\ ks=hd:LABEL=BOOT:\/biosks.cfg/' \"$device_mnt/syslinux.cfg\""
  execute "sed -i 's/^TIMEOUT.*/timeout\ 100/i' \"$device_mnt/syslinux.cfg\""
  execute "sed -i 's/menu\ default//i' \"$device_mnt/syslinux.cfg\""
  execute "sed -i '/label\ linux/a \ \ menu\ default' \"$device_mnt/syslinux.cfg\""
  execute "cp \"$ks\" \"$device_mnt/biosks.cfg\""
  execute "cp \"$ks\" \"$device_mnt/efiks.cfg\""
  # overwrite .c32
  local syslinuxDir=$(dirname "$mbr_bin_path")
  execute "cp -f \"$syslinuxDir/../modules/bios/ldlinux.c32\" $device_mnt/"
  execute "cp -f \"$syslinuxDir/../modules/bios/libcom32.c32\" $device_mnt/"
  execute "cp -f \"$syslinuxDir/../modules/bios/libutil.c32\" $device_mnt/"
  execute "cp -f \"$syslinuxDir/../modules/bios/menu.c32\" $device_mnt/"
  execute "cp -f \"$syslinuxDir/../modules/bios/vesamenu.c32\" $device_mnt/"
  ## UEFI-Boot
  execute "cp -rf $iso_mnt/EFI $device_mnt/"
  execute "cp -rf $iso_mnt/images $device_mnt/"
  execute "sed -i 's/set default=\"1\"/set default=\"0\"/g' \"$device_mnt/EFI/BOOT/grub.cfg\""
  execute "sed -i 's/set timeout=60/set timeout=5/g' \"$device_mnt/EFI/BOOT/grub.cfg\""
  execute "sed -i 's/inst.stage2=hd:LABEL=CentOS\\\x207\\\x20x86_64 quiet/biosdevname=0 net.ifnames=0 initrd=initrd.img method=hd:LABEL=DATA:\/ ks=hd:LABEL=BOOT:\/efiks.cfg/g' \"$device_mnt/EFI/BOOT/grub.cfg\""
  cat <<EOF > $device_mnt/startup.nsh
@echo -on
cls

for %m run (0 9)
  if exist FS%m:\EFI\BOOT\BOOTX64.EFI then
      FS%m:
      \EFI\BOOT\BOOTX64.EFI
  endif
endfor

for %m run (0 9)
  if exist FS%m:\BOOT\BOOTX64.EFI then
      FS%m:
      \BOOT\BOOTX64.EFI
  endif
endfor
EOF
}

prepare_data() {
  _log "Fill data partiton content..."
  local device="$1"
  local iso_mnt="$2"
  device_mnt=$(_mount_device "${device}2")
  handle_code $?
  execute "cp -r $iso_mnt/. \"$device_mnt\""
  # sometimes using path to iso in kickstart install works better
  execute "cp $iso \"$device_mnt/install.iso\""
}

check_iso_content() {
  _log "Check ISO content ´$1´"
  local mnt="$1"
  if [[ ! -d "$mnt/isolinux" ]]; then
    _error "Missing isolinux/ at .iso directory structure. Maybe wrong .iso?"
    handle_code 1
  fi
}

check_device "$device"
iso_mnt=$(_mount_iso "$iso")
handle_code $?
check_iso_content "$iso_mnt"
prepare_device "$device"
prepare_boot "$device" "$iso_mnt"
prepare_data "$device" "$iso_mnt"
cleanup

echo "Kickstart Usb Stick Ready."