03_make_img_f2fs_separated.sh

#!/bin/bash

#
# Produce system image with root on F2FS filesystem
#

set -e

PWD="$(pwd)"
WORK_DIR="${PWD}/build"
L4T_DIR="${WORK_DIR}/l4t"
ROOTFS_DIR="${L4T_DIR}/rootfs"

L4T_BOOTLOADER_DIR="${L4T_DIR}/bootloader"
L4T_SIGNED_IMAGE_DIR="${L4T_BOOTLOADER_DIR}/signed"
L4T_TOOLS_DIR="${L4T_DIR}/tools"

JETSON_BOOT_IMAGE_FILENAME=jetson_boot.img
JETSON_ROOT_IMAGE_FILENAME=jetson_root.img
JETSON_BOOT_IMAGE_PATH="${WORK_DIR}/${JETSON_BOOT_IMAGE_FILENAME}"
JETSON_ROOT_IMAGE_PATH="${WORK_DIR}/${JETSON_ROOT_IMAGE_FILENAME}"

BOOT_PARTITION_SIZE=64 # Space available for kernel and DTB files

#
# Create FS images
#

ROOTFS_DIR_SIZE="$(du -bms "${ROOTFS_DIR}" | awk '{print $1}')"
ROOTFS_DIR_SIZE="$(("${ROOTFS_DIR_SIZE}" * 3 / 2))"

echo "RootFS size: ${ROOTFS_DIR_SIZE} MB"

pushd "${L4T_DIR}"
BOARDID=3448 FAB=200 BUILD_SD_IMAGE=1 bash "${L4T_DIR}/flash.sh" --no-root-check --no-flash --sign -S "${ROOTFS_DIR_SIZE}MiB" jetson-nano-qspi-sd mmcblk0p1
popd

#
# Allocate image
#

JETSON_IMAGE_SIZE=$("${L4T_TOOLS_DIR}/nvptparser.py" "${L4T_SIGNED_IMAGE_DIR}/flash.xml" sdcard | awk -F'[=;]' '{sum += (int($6 / (2048 * 512)) + 1)} END {printf "%d\n", sum + 2}')

# Allocate SD card and secondary storage image files
dd if=/dev/zero of="${JETSON_BOOT_IMAGE_PATH}" bs=1048576 count=$(( ${BOOT_PARTITION_SIZE} + 16 ))
dd if=/dev/zero of="${JETSON_ROOT_IMAGE_PATH}" bs=1048576 count="${JETSON_IMAGE_SIZE}"

#
# Create partitions
#

# Create boot storage GPT table
sgdisk -og "${JETSON_BOOT_IMAGE_PATH}"

# Create partitions (except APP)
"${L4T_TOOLS_DIR}/nvptparser.py" "${L4T_SIGNED_IMAGE_DIR}/flash.xml" sdcard | grep -v 'part_name=APP' | while IFS= read -r PROPERTIES; do
	eval "${PROPERTIES}"
	sgdisk -n "${part_num}:0:+$(( ${part_size} / 512 ))" \
		   -c "${part_num}:${part_name}" \
		   -t "${part_num}:8300" \
		   "${JETSON_BOOT_IMAGE_PATH}"
done

# Get BOOT (APP) partition properties
eval $("${L4T_TOOLS_DIR}/nvptparser.py" "${L4T_SIGNED_IMAGE_DIR}/flash.xml" sdcard | grep 'part_name=APP')

# Define BOOT (APP) partition size
BOOT_PART_SIZE=$(( ${BOOT_PARTITION_SIZE} * 1048576 ))

# Create root storage GPT table
sgdisk -og "${JETSON_ROOT_IMAGE_PATH}"

# Create BOOT (APP) partition
sgdisk -n "1:0:+$(( ${BOOT_PART_SIZE} / 512 ))" \
	   -c "1:APP" \
	   -t "1:8300" \
       "${JETSON_BOOT_IMAGE_PATH}"

# Create ROOT partition
sgdisk -n "1:0:+$(( ( ${part_size} - ${BOOT_PART_SIZE} ) / 512 ))" \
	   -c "1:ROOT" \
	   -t "1:8300" \
       "${JETSON_ROOT_IMAGE_PATH}"

#
# Write partitions
#

JETSON_BOOT_IMAGE_LOOP_DEV="$(losetup --show -f -P "${JETSON_BOOT_IMAGE_PATH}")"
JETSON_ROOT_IMAGE_LOOP_DEV="$(losetup --show -f -P "${JETSON_ROOT_IMAGE_PATH}")"

# Flash partition images (except APP)
"${L4T_TOOLS_DIR}/nvptparser.py" "${L4T_SIGNED_IMAGE_DIR}/flash.xml" sdcard | grep -v 'part_name=APP' | while IFS= read -r PROPERTIES; do
	eval "${PROPERTIES}"
	if [ -f "${L4T_SIGNED_IMAGE_DIR}/${part_file}" ]; then
		PARTITION_IMAGE_PATH="${L4T_SIGNED_IMAGE_DIR}/${part_file}"
	elif [ -f "${L4T_BOOTLOADER_DIR}/${part_file}" ]; then
		PARTITION_IMAGE_PATH="${L4T_BOOTLOADER_DIR}/${part_file}"
	fi
	dd if="${PARTITION_IMAGE_PATH}" of="${JETSON_BOOT_IMAGE_LOOP_DEV}p${part_num}"
done

# Mount temporary system partition
MOUNT_DIR="${WORK_DIR}/mount"
mkdir -p "${MOUNT_DIR}"/{app,root,tmp}
mount "${L4T_BOOTLOADER_DIR}/system.img.raw" "${MOUNT_DIR}/tmp"

# Flash BOOT (APP) partition
mkfs.ext4 -O ^has_journal -F "${JETSON_BOOT_IMAGE_LOOP_DEV}p1"
mount "${JETSON_BOOT_IMAGE_LOOP_DEV}p1" "${MOUNT_DIR}/app"
rsync -aAHx --info=progress2 --delete "${MOUNT_DIR}/tmp/boot" "${MOUNT_DIR}/app"

# Fix boot arguments
nano "${MOUNT_DIR}/app/boot/extlinux/extlinux.conf"
# APPEND ${cbootargs} quiet root=/dev/sda1 rw rootwait rootfstype=f2fs console=ttyS0,115200n8 console=tty0 fbcon=map:0 net.ifnames=0
umount "${MOUNT_DIR}/app"

# Flash ROOT partition
mkfs.f2fs -f "${JETSON_ROOT_IMAGE_LOOP_DEV}p1"
mount "${JETSON_ROOT_IMAGE_LOOP_DEV}p1" "${MOUNT_DIR}/root"
rsync -aAHx --info=progress2 --delete --exclude boot "${MOUNT_DIR}/tmp/" "${MOUNT_DIR}/root/"
echo '/dev/root / f2fs defaults,noatime 0 1' > "${MOUNT_DIR}/root/etc/fstab"
umount "${MOUNT_DIR}/root"

umount "${MOUNT_DIR}/tmp"
rm -rf "${MOUNT_DIR}"

# Unmount image
losetup -d "${JETSON_ROOT_IMAGE_LOOP_DEV}"
losetup -d "${JETSON_BOOT_IMAGE_LOOP_DEV}"

# Compress image using zstd
zstd -f "${JETSON_BOOT_IMAGE_PATH}"
zstd -f "${JETSON_ROOT_IMAGE_PATH}"