101 lines
3.1 KiB
Bash
Executable File
101 lines
3.1 KiB
Bash
Executable File
#!/bin/bash -e
|
|
|
|
ISO_FILE="${STAGE_WORK_DIR}/${IMG_FILENAME}${IMG_SUFFIX}.iso"
|
|
|
|
unmount_image "${ISO_FILE}"
|
|
|
|
rm -f "${ISO_FILE}"
|
|
rm -rf "${ROOTFS_DIR}"
|
|
mkdir -p "${ROOTFS_DIR}"
|
|
|
|
BOOT_SIZE=$((512 * 1024 * 1024))
|
|
RECOVERY_SIZE=$((256 * 1024 * 1024)) # 256MB recovery partition
|
|
ROOT_SIZE=$(du -x --apparent-size -s "${EXPORT_ROOTFS_DIR}" --exclude var/cache/apt/archives --exclude boot/firmware --block-size=1 | cut -f 1)
|
|
|
|
ALIGN=$((4 * 1024 * 1024))
|
|
ROOT_MARGIN="$(echo "($ROOT_SIZE * 0.2 + 200 * 1024 * 1024) / 1" | bc)"
|
|
|
|
BOOT_PART_START=$((ALIGN))
|
|
BOOT_PART_SIZE=$(((BOOT_SIZE + ALIGN - 1) / ALIGN * ALIGN))
|
|
|
|
ROOT_PART_START=$((BOOT_PART_START + BOOT_PART_SIZE))
|
|
ROOT_PART_SIZE=$(((ROOT_SIZE + ROOT_MARGIN + ALIGN - 1) / ALIGN * ALIGN))
|
|
|
|
RECOVERY_PART_START=$((ROOT_PART_START + ROOT_PART_SIZE))
|
|
RECOVERY_PART_SIZE=$(((RECOVERY_SIZE + ALIGN - 1) / ALIGN * ALIGN))
|
|
|
|
IMG_SIZE=$((RECOVERY_PART_START + RECOVERY_PART_SIZE))
|
|
|
|
truncate -s "${IMG_SIZE}" "${ISO_FILE}"
|
|
|
|
parted --script "${ISO_FILE}" mklabel msdos
|
|
parted --script "${ISO_FILE}" unit B mkpart primary fat32 "${BOOT_PART_START}" "$((BOOT_PART_START + BOOT_PART_SIZE - 1))"
|
|
parted --script "${ISO_FILE}" unit B mkpart primary ext4 "${ROOT_PART_START}" "$((ROOT_PART_START + ROOT_PART_SIZE - 1))"
|
|
parted --script "${ISO_FILE}" unit B mkpart primary ext4 "${RECOVERY_PART_START}" "$((RECOVERY_PART_START + RECOVERY_PART_SIZE - 1))"
|
|
|
|
echo "Creating loop device..."
|
|
cnt=0
|
|
until ensure_next_loopdev && LOOP_DEV="$(losetup --show --find --partscan "$ISO_FILE")"; do
|
|
if [ $cnt -lt 5 ]; then
|
|
cnt=$((cnt + 1))
|
|
echo "Error in losetup. Retrying..."
|
|
sleep 5
|
|
else
|
|
echo "ERROR: losetup failed; exiting"
|
|
exit 1
|
|
fi
|
|
done
|
|
|
|
ensure_loopdev_partitions "$LOOP_DEV"
|
|
BOOT_DEV="${LOOP_DEV}p1"
|
|
ROOT_DEV="${LOOP_DEV}p2"
|
|
RECOVERY_DEV="${LOOP_DEV}p3"
|
|
|
|
ROOT_FEATURES="^huge_file"
|
|
for FEATURE in 64bit; do
|
|
if grep -q "$FEATURE" /etc/mke2fs.conf; then
|
|
ROOT_FEATURES="^$FEATURE,$ROOT_FEATURES"
|
|
fi
|
|
done
|
|
|
|
if [ "$BOOT_SIZE" -lt 134742016 ]; then
|
|
FAT_SIZE=16
|
|
else
|
|
FAT_SIZE=32
|
|
fi
|
|
|
|
mkdosfs -n bootfs -F "$FAT_SIZE" -s 4 -v "$BOOT_DEV" > /dev/null
|
|
mkfs.ext4 -L rootfs -O "$ROOT_FEATURES" "$ROOT_DEV" > /dev/null
|
|
mkfs.ext4 -L recovery "$RECOVERY_DEV" > /dev/null
|
|
|
|
mount -v "$ROOT_DEV" "${ROOTFS_DIR}" -t ext4
|
|
mkdir -p "${ROOTFS_DIR}/boot"
|
|
mount -v "$BOOT_DEV" "${ROOTFS_DIR}/boot" -t vfat
|
|
|
|
rsync -aHAXx --exclude /var/cache/apt/archives --exclude /boot/firmware "${EXPORT_ROOTFS_DIR}/" "${ROOTFS_DIR}/"
|
|
#rsync -rtx "${EXPORT_ROOTFS_DIR}/boot/firmware/" "${ROOTFS_DIR}/boot/firmware/"
|
|
|
|
# Optionally mount and populate the recovery partition
|
|
mkdir -p "${ROOTFS_DIR}/recovery"
|
|
mount -v "$RECOVERY_DEV" "${ROOTFS_DIR}/recovery" -t ext4
|
|
|
|
# (Optional) Example: copy recovery script
|
|
# cp recovery-tools/recovery.sh "${ROOTFS_DIR}/recovery/"
|
|
|
|
# Create bootable ISO with GRUB
|
|
echo "Creating bootable ISO..."
|
|
mkdir -p "${ROOTFS_DIR}/boot/grub"
|
|
|
|
cat > "${ROOTFS_DIR}/boot/grub/grub.cfg" <<EOF
|
|
set default=0
|
|
set timeout=5
|
|
|
|
menuentry "Linux" {
|
|
linux /boot/vmlinuz root=/dev/sda2
|
|
initrd /boot/initrd.img
|
|
}
|
|
EOF
|
|
|
|
grub-mkrescue -o "${ISO_FILE}" "${ROOTFS_DIR}"
|
|
|
|
echo "Bootable ISO with recovery partition created at ${ISO_FILE}" |