ubuntu-iso-customization.sh

#!/bin/bash

# Mensaje de ayuda en el uso de la aplicación.
function myhelp(){
  echo "Modo de empleo: $0 <opciones> IMAGEN.ISO 

Donde:
  IMAGEN.ISO es la ruta al archivo ISO original

Opciones:

  -d modo desarrollo, crea un zip apartir de la carpeta actual
  -z archivo.zip el archivo zip como repositorio
  -c directorio_cache Ruta absoluta al directorio donde se encuentra el cache de APT a utilizar
  -h muestra esta ayuda

Toma una imagen de Ubuntu, la personaliza de acuerdo al script de configuración y genera el archivo ISO personalizado para ser distribuido.";
}

# Mensajes de error y salida del script
error_exit(){
	echo "${1:-"Error desconocido"}" 1>&2
	exit 1
}

# Captando parámetros
# Is in development environment ?
DEVELOPMENT=false
ZIP=""
APT_CACHE=""
APT_CACHE_CHROOT=""

while getopts zc:hd option
do
 case "${option}"
 in
 z) ZIP=${OPTARG};;
 d) DEVELOPMENT=true;;
 c) APT_CACHE=${OPTARG};;
 h) myhelp
    exit 0 ;;
 esac
done

shift $((OPTIND -1))

if [ -z $1 ]; then
  myhelp
  exit 1
fi

## VARIABLES

# ruta absoluta al archivo ISO original
ISOPATH=$(cd "$(dirname "$1")"; pwd)/$(basename "$1")

# directorio del script
SCRIPTPATH=$(readlink -f $0)
SCRIPTDIR=$(dirname "$SCRIPTPATH")

# directorio para perzonalizacion
CUSTOMIZATIONDIR=$(pwd)/ubuntu-iso-customization
mkdir -p $CUSTOMIZATIONDIR

# nombre del archivo ISO original
ISONAME=$(basename "$ISOPATH")
# nombre del archivo ISO personalizado,
# de la forma nombre-original-ucr-yyyymmdd.iso
CUSTOMISONAME="${ISONAME%.*}-ucr-`date +%Y%m%d`.iso"

# directorio donde extraer iso
EXTRACT=${ISONAME%.*}-extract
# directorio donde editar sistema de archivos
EDIT=${ISONAME%.*}-squashfs

## PERSONALIZACION

if [ -z $ZIP ]; then
    if $DEVELOPMENT ; then
        echo "Modo Local (Desarrollo) activado"
        mkdir $CUSTOMIZATIONDIR/ubuntu-ucr-master/
        cp -ar $SCRIPTDIR/plymouth/ $SCRIPTDIR/gschema/ $SCRIPTDIR/*.list ubuntu-16.04-ucr-* $CUSTOMIZATIONDIR/ubuntu-ucr-master/
        ( cd $CUSTOMIZATIONDIR; zip -r master.zip ubuntu-ucr-master; ) || error_exit "No pude generar master.zip"
        rm -rf $CUSTOMIZATIONDIR/ubuntu-ucr-master/
    else
        wget -O $CUSTOMIZATIONDIR/master.zip https://github.com/cslucr/ubuntu-ucr/archive/master.zip || error_exit "No pude descargar master.zip"
    fi
else
    cp $ZIP $CUSTOMIZATIONDIR/master.zip || error_exit "No pude copiar master.zip desde $ZIP "
fi
if [[ -d "$APT_CACHE" ]]; then
  echo "Usando cache APT: $APT_CACHE"
  APT_CACHE=$(readlink -f $APT_CACHE)
fi


echo "Se trabajará en el directorio $CUSTOMIZATIONDIR"
cd $CUSTOMIZATIONDIR
mkdir mnt
sudo mount -o loop $ISOPATH mnt || error_exit "Error al montar ISO $ISOPATH"
mkdir $EXTRACT
sudo rsync --exclude=/casper/filesystem.squashfs -a mnt/ $EXTRACT
sudo dd if=$ISOPATH bs=512 count=1 of=$EXTRACT/isolinux/isohdpfx.bin
sudo unsquashfs -d $EDIT mnt/casper/filesystem.squashfs || error_exit "Error al desempaquetar Squashfs"
sudo umount mnt
sudo mv $CUSTOMIZATIONDIR/master.zip $EDIT/root
sudo mv $EDIT/etc/resolv.conf $EDIT/etc/resolv.conf.bak
sudo cp /etc/resolv.conf /etc/hosts $EDIT/etc/
sudo mount --bind /dev/ $EDIT/dev/

# Usa cache de APT
if [[ -n "$APT_CACHE" ]]; then
  mkdir -p $EDIT/apt/
  rsync -a --link-dest="${APT_CACHE}/" "${APT_CACHE}/" "${EDIT}/apt/" || error_exit "Error al sincronizar cache APT desde ${APT_CACHE}"
  APT_CACHE_CHROOT=" -c '/apt/'"
fi


# Ejecuta ordenes dentro de directorio de edicion
cat << EOF | sudo chroot $EDIT || error_exit "Personalización fallida"
mount -t proc none /proc
mount -t sysfs none /sys
mount -t devpts none /dev/pts

export HOME=/root
export LC_ALL=C
dbus-uuidgen > /var/lib/dbus/machine-id
dpkg-divert --local --rename --add /sbin/initctl
ln -s /bin/true /sbin/initctl
cd ~

# Descarga y ejecuta script de personalizacion ubuntu-ucr.
# Puede omitir el script y en su lugar realizar una personalizacion manual
unzip master.zip && rm master.zip
bash ubuntu-ucr-master/ubuntu-16.04-ucr-config.sh -y $APT_CACHE_CHROOT || exit 1
rm -r ubuntu-ucr-master

rm -rf /tmp/* ~/.bash_history
rm /var/lib/dbus/machine-id
rm /sbin/initctl
dpkg-divert --rename --remove /sbin/initctl

umount /proc || umount -lf /proc
umount /sys
umount /dev/pts
# Sale del directorio de edicion
EOF

# Actualiza cache de APT
if [[ -d "$APT_CACHE" ]]; then
  echo "Salvando cache APT: $APT_CACHE"
  rsync -a --link-dest="${APT_CACHE}/" "${EDIT}/apt/" "${APT_CACHE}/" || error_exit "Error al salvar el cache APT hacia ${APT_CACHE}"
  rm -r "${EDIT}/apt/" || error_exit "Error al liberar cache de APT en ${EDIT}"
fi


sudo umount $EDIT/dev
sudo rm $EDIT/etc/resolv.conf $EDIT/etc/hosts
sudo mv $EDIT/etc/resolv.conf.bak $EDIT/etc/resolv.conf

# CREACION DE NUEVA IMAGEN ISO
# regenera manifest
sudo chmod +w $EXTRACT/casper/filesystem.manifest
sudo chroot $EDIT dpkg-query -W --showformat='${Package} ${Version}\n' | sudo tee $EXTRACT/casper/filesystem.manifest
sudo cp $EXTRACT/casper/filesystem.manifest $EXTRACT/casper/filesystem.manifest-desktop 
sudo sed -i '/ubiquity/d' $EXTRACT/casper/filesystem.manifest-desktop
sudo sed -i '/casper/d' $EXTRACT/casper/filesystem.manifest-desktop

# Comprime el sistema de archivos recien editado
sudo mksquashfs $EDIT $EXTRACT/casper/filesystem.squashfs -b 1048576 || error_exit "Error al generar Squashfs"
printf $(sudo du -sx --block-size=1 $EDIT | cut -f1) | sudo tee $EXTRACT/casper/filesystem.size
cd $EXTRACT
sudo rm md5sum.txt
find -type f -print0 | sudo xargs -0 md5sum | grep -v isolinux/boot.cat | sudo tee md5sum.txt

sudo xorriso -as mkisofs -isohybrid-mbr isolinux/isohdpfx.bin \
-c isolinux/boot.cat -b isolinux/isolinux.bin -no-emul-boot -boot-load-size 4 \
-boot-info-table -eltorito-alt-boot -e boot/grub/efi.img -no-emul-boot \
-isohybrid-gpt-basdat -o ../$CUSTOMISONAME .  || error_exit "Error al generar ISO en ${CUSTOMISONAME}"

echo "Generando sumas de verificación";
cd ..
md5sum $CUSTOMISONAME >> MD5SUMS
sha1sum $CUSTOMISONAME >> SHA1SUMS
sha256sum $CUSTOMISONAME >> SHA256SUMS

exit 0