Skip to content

Commit

Permalink
restore liveiso builder
Browse files Browse the repository at this point in the history
  • Loading branch information
@ajanis
ajanis committed Aug 27, 2024
1 parent 2e827ab commit 9191cb1
Showing 1 changed file with 45 additions and 301 deletions.
346 changes: 45 additions & 301 deletions harmonic/scripts/iso-builders/ubuntu-create-liveiso.sh
100755 → 100644
View file
Original file line number Diff line number Diff line change
@@ -1,312 +1,56 @@
#!/bin/bash
##############################################################################
#
# rebuild-iso.sh
#
# This script is executed by a user to rebuild the Apollo ISO using
# a modified squashfs root filesystem.
#
##############################################################################
export buildRoot="/opt/harmonic-image-build"
export isoMount="${buildRoot}/iso.mount"
export chrootPath="${buildRoot}/filesystem.tmp"
export buildDirs=("build.tmp" "iso.mount" "filesystem.tmp")
export buildTemp="${buildRoot}/build.tmp"
export bindMounts=("run" "dev")
export chrootMounts=('chrootArray["proc"]="/proc"' 'chrootArray["sysfs"]="/sys"' 'chrootArray["devpts"]="/dev/pts"')
export buildLog="${buildRoot}/harmonic-iso.log"
export artifactoryURL="https://artifactory.charterlab.com"
export artifactoryPath="artifactory/upload/harmonic"
declare -A chrootArray
runPrint() {
cat << EOF
===========================================================
$@
===========================================================
EOF
}

# Script Help Function
showHelp() {
cat << EOH
Usage: $0 -i <source iso> -f <squashfs path in source iso | default('/rootfs.squashfs')> [-x -c -r -b -d] [-h]
Rebuild Apollo ISO
-i | [REQUIRED] <path to source iso> | ISO image to rebuild (will be used as source and destination filenames)
-f | [REQUIRED] <path to squashfs in source iso> | Path to SquashFS archive in source ISO (used for Chroot customizations)

-x | Extract squashfs filesystem from original ISO image
# Variables
# ISO_URL="https://releases.ubuntu.com/20.04/ubuntu-20.04-desktop-amd64.iso"
ISO_NAME="$1"
WORK_DIR="/opt/live-image-build/ubuntu-iso"
MOUNT_DIR="/opt/live-image-build/ubuntu-iso.mount"

-c | Customize filesystem in chroot environment
# Download Ubuntu ISO
# wget $ISO_URL -O $ISO_NAME

-r | Generate new squashfs from modified filesystem
# Install required tools
sudo apt update
sudo apt install -y xorriso squashfs-tools genisoimage

-b | Build new iso image from source image and new squashfs
-d | Deploy iso to Artifactory
-h | Display help
EOH

}
# Mount the ISO
mkdir -p $MOUNT_DIR
sudo mount -o loop $ISO_NAME $MOUNT_DIR

# Copy ISO contents
mkdir -p $WORK_DIR
rsync -a $MOUNT_DIR/ $WORK_DIR/
sudo umount $MOUNT_DIR

# Make filesystem writable
chmod -R u+w $WORK_DIR

function createWorkspace() {
for dir in "${buildDirs[@]}"; do
if [[ ! -d "${buildRoot}/${dir}" ]]; then
runPrint "Creating ${buildRoot}/${dir}"
mkdir -p "${buildRoot}/${dir}"
fi
done
return
cat << 'EOG' >> "${WORK_DIR}/boot/grub/grub.cfg"
menuentry "Ubuntu 22.04 Live-Only" {
rmmod tpm
linux /casper/vmlinuz boot=casper iso-scan/filename=${iso_path} noprompt noeject nopersistent maybe-ubiquity nomodeset quiet splash fsck.mode=skip toram
initrd /casper/initrd
}

function extractRootfs() {
if mountpoint "${isoMount}"; then
runPrint "Unmounting ${isoMount}"
umount -lf "${isoMount}"
fi

runPrint "Mounting ${originalIsoPath} at ${isoMount}"
mount -o loop "${originalIsoPath}" "${isoMount}"

if [[ ! -f "${isoMount}/${squashfsIsoPath}" ]]; then
runPrint "No SquashFS Archive found at ${isoMount}/${squashfsIsoPath}!!"
runPrint "Leaving ${isoMount} mounted for debugging"
exit 1
fi

runPrint "Removing old ${chrootPath}"
rm -rf "${chrootPath}"

runPrint "Extract ${isoMount}/${squashfsIsoPath} to ${chrootPath}"
unsquashfs -d "${chrootPath}" "${isoMount}/${squashfsIsoPath}"

umount -f "${isoMount}"

}

function setupChroot() {

for bmount in "${bindMounts[@]}"; do
runPrint "Bind mounting /${bmount} at ${chrootPath}/${bmount}"
mount --bind "/${bmount}" "${chrootPath}/${bmount}"
done

for arraymount in "${chrootMounts[@]}"; do
eval "${arraymount}"
done

runPrint "${!chrootArray[@]}"
for cmount in "${!chrootArray[@]}"; do
runPrint "Mounting ${cmount} at ${chrootPath}${chrootArray[${cmount}]}"
mount -t "${cmount}" none "${chrootPath}${chrootArray[${cmount}]}"
done

return
}

function cleanupChroot() {

for cmount in "${!chrootArray[@]}"; do
runPrint "Unounting ${cmount} at ${chrootArray[${cmount}]}"
umount "${chrootPath}${chrootArray[${cmount}]}"
done

for bmount in "${bindMounts[@]}"; do
runPrint "Unmounting bindmount ${bmount} at ${chrootPath}/${bmount}"
umount "${chrootPath}/${bmount}"
done

return
}

function resquashRootfs() {

if [[ ! -d ${chrootPath} ]]; then
runPrint "No root source found!!"
exit 1
fi

rm -f "${newSquashfs}"

runPrint "Creating new squashfs at ${newSquashfs} from ${chrootPath} ..."

mksquashfs "${chrootPath}" "${newSquashfs}" -noappend

return
}

function buildIso() {

if [[ ! -f ${newSquashfs} ]]; then
runPrint "No squashFS found!!"
return 1
fi
export newIso="${buildTemp}/${isoFile}"
rm -f "${newIso}"
cat <<EOG
=====================================================================================
Would you like to add Ubuntu Live Image files? (live iso ONLY!)
======================================================================================
EOG
read -r -p "Add Live Image Files? [y/n] : " yesno || return 1

if [[ ${yesno} =~ (y|Y|es?) ]]; then
xorriso -overwrite on -indev "${originalIsoPath}" -outdev "${newIso}" -pathspecs on -add "${squashfsIsoPath}=${newSquashfs}" "boot/grub/grub.cfg=${buildTemp}/grub.cfg" "isolinux/txt.cfg=${buildTemp}/txt.cfg"
return
else
xorriso -overwrite on -indev "${originalIsoPath}" -outdev "${newIso}" -pathspecs on -add "${squashfsIsoPath}=${newSquashfs}"
return
fi

}

function deployIso() {

export newIso="${buildTemp}/${isoFile}"

if [[ ! -f ${newIso} ]]; then
runPrint "No ISO found at ${newIso} to deploy !!"
return 1
fi

cat <<EOD
=====================================================================================
Would you like to deploy ${newIso} to Artifactory? [y/n]
(You will be prompted for your JFrog username and password)
======================================================================================
EOD

read -rp "Press [Enter/Return] to deploy new ISO : ";echo || return 1
read -r -p "Enter Jfrog Username : " artifactUser || return 1
# shellcheck disable=SC2162
read -s -p "Enter JFrog Password : " artifactPassword || return 1

curl -u "${artifactUser}:${artifactPassword}" -T "${newIso}" "${artifactoryURL}/${artifactoryPath}/apollo/${isoFile}" || return 1

return
}

while getopts ":hxcrbdi:f:" o; do
case "${o}" in
h)
showHelp
exit 0
;;
i)
originalIsoPath=${OPTARG}
if [[ -z "${originalIsoPath}" ]]; then
echo "Error: Must define -i <original iso path>"
exit 1
fi
if [[ ! -f ${originalIsoPath} ]]; then
runPrint "No source iso found at ${originalIsoPath} !!"
fi
isoFile=$(basename "${originalIsoPath}")

;;
f)
squashfsIsoPath=${OPTARG}

if [[ -z "${squashfsIsoPath}" ]]; then
echo "Error: Must define -f <squashfs path>"
exit 1
fi
squashfsFilename=$(basename "${squashfsIsoPath}")
newSquashfs="${buildTemp}/${squashfsFilename}"
;;
x)
export doSetup=1
;;
c)
export doChroot=1
;;
r)
export doBuildRootfs=1
;;
b)
export doBuildIso=1
;;
d)
export doDeployIso=1
;;

:)
runPrint "Invalid option: -${OPTARG} requires an argument" 1>&2
showHelp
exit 1
;;
\?|*)
runPrint "Invalid option: -${OPTARG}" 1>&2
showHelp
exit 1
;;
esac
done
shift $((OPTIND-1))


# Main Runners

createWorkspace

if [[ ${doSetup} == 1 ]]; then
# shellcheck disable=SC2312
extractRootfs >(tee -a "${buildLog}" >&2) > >(tee -a "${buildLog}")
fi

if [[ ${doChroot} == 1 ]]; then
# shellcheck disable=SC2312
setupChroot >(tee -a "${buildLog}" >&2) > >(tee -a "${buildLog}")
trap cleanupChroot EXIT
chroot "${chrootPath}"
fi

if [[ "${doBuildRootfs}" == 1 ]]; then
# shellcheck disable=SC2312
resquashRootfs >(tee -a "${buildLog}" >&2) > >(tee -a "${buildLog}")
fi

if [[ "${doBuildIso}" == 1 ]]; then
# shellcheck disable=SC2312
buildIso >(tee -a "${buildLog}">&2) > >(tee -a "${buildLog}")

cat <<EOB
=====================================================================================
Build Complete!
Generated ${newIso} from:
iso: ${originalIsoPath}
squashfs: ${newSquashfs}
Activity log can be viewed at ${buildLog} ...
======================================================================================
EOB

fi

if [[ "${doDeployIso}" == 1 ]]; then
# shellcheck disable=SC2312
deployIso >(tee -a "${buildLog}">&2) > >(tee -a "${buildLog}")
# shellcheck disable=SC2181
if [[ $? == 0 ]]; then
echo -e "\n\nDone!\n\n"
else
echo -e "\n\nFailed - Check ${buildLog}\n\n"

fi
fi
cat << 'EOI' >> "${WORK_DIR}/isolinux/txt.cfg"
label ubuntu2204live
menu label ^Ubuntu 22.04 Live-Only
kernel /casper/vmlinuz
append initrd=/casper/initrd quiet noprompt noeject nopersistent maybe-ubiquity nomodeset quiet splash fsck.mode=skip toram ---
EOI

# Repack the ISO
cd "${WORK_DIR}" || exit
sudo xorriso -as mkisofs \
-r -V "Ubuntu Custom" \
-o /opt/live-image-build/custom-ubuntu.iso \
-J -l -b isolinux/isolinux.bin \
-c isolinux/boot.cat -no-emul-boot \
-boot-load-size 4 -boot-info-table \
-eltorito-alt-boot -e boot/grub/efi.img \
-no-emul-boot -isohybrid-gpt-basdat \
.

echo "Custom ISO created at /opt/live-image-build/custom-ubuntu.iso"

0 comments on commit 9191cb1

Please sign in to comment.