88 lines
2.9 KiB
Bash
Executable File
88 lines
2.9 KiB
Bash
Executable File
#!/bin/bash -e
|
|
|
|
IMG_FILE="${STAGE_WORK_DIR}/${IMG_FILENAME}${IMG_SUFFIX}.img"
|
|
|
|
unmount_image "${IMG_FILE}"
|
|
|
|
rm -f "${IMG_FILE}"
|
|
|
|
rm -rf "${ROOTFS_DIR}"
|
|
mkdir -p "${ROOTFS_DIR}"
|
|
|
|
BOOT_SIZE="$((512 * 1024 * 1024))"
|
|
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 + 400 * 1024 * 1024) / 1" | bc)"
|
|
|
|
RECOVERY_SIZE="$((256 * 1024 * 1024))"
|
|
|
|
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}" "${IMG_FILE}"
|
|
|
|
parted --script "${IMG_FILE}" mklabel msdos
|
|
parted --script "${IMG_FILE}" unit B mkpart primary fat32 "${BOOT_PART_START}" "$((BOOT_PART_START + BOOT_PART_SIZE - 1))"
|
|
parted --script "${IMG_FILE}" unit B mkpart primary fat32 "${RECOVERY_PART_START}" "$((RECOVERY_PART_START + RECOVERY_PART_SIZE - 1))"
|
|
parted --script "${IMG_FILE}" unit B mkpart primary ext4 "${ROOT_PART_START}" "$((ROOT_PART_START + ROOT_PART_SIZE - 1))"
|
|
|
|
echo "Creating loop device..."
|
|
cnt=0
|
|
until ensure_next_loopdev && LOOP_DEV="$(losetup --show --find --partscan "$IMG_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"
|
|
RECOVERY_DEV="${LOOP_DEV}p2"
|
|
ROOT_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
|
|
mkdosfs -n recovery -F "16" -s 4 -v "$RECOVERY_DEV" > /dev/null
|
|
mkfs.ext4 -L rootfs -O "$ROOT_FEATURES" "$ROOT_DEV" > /dev/null
|
|
|
|
# Create new layout
|
|
mkdir -p "${ROOTFS_DIR}/part1"
|
|
mkdir -p "${ROOTFS_DIR}/part2"
|
|
|
|
mount -v "$BOOT_DEV" "${ROOTFS_DIR}/part1" -t vfat
|
|
mount -v "$RECOVERY_DEV" "${ROOTFS_DIR}/part2" -t ext4
|
|
|
|
mkdir -p "${ROOTFS_DIR}/part1/bootfs"
|
|
mkdir -p "${ROOTFS_DIR}/part1/rootfs"
|
|
mkdir -p "${ROOTFS_DIR}/part2/recovery"
|
|
|
|
mount -v "$ROOT_DEV" "${ROOTFS_DIR}/part1/rootfs" -t ext4
|
|
|
|
# Copy files
|
|
rsync -aHAXx --exclude /var/cache/apt/archives --exclude /boot/firmware "${EXPORT_ROOTFS_DIR}/" "${ROOTFS_DIR}/part1/rootfs/"
|
|
rsync -rtx "${EXPORT_ROOTFS_DIR}/boot/firmware/" "${ROOTFS_DIR}/part1/bootfs/"
|
|
|
|
# If you have recovery files to copy, uncomment:
|
|
# rsync -aHAXx "${EXPORT_ROOTFS_DIR}/recovery/" "${ROOTFS_DIR}/part2/recovery/" |