#!/bin/bash
|
|
if [ ! -d "$TARGET_OUTPUT_DIR" ]; then
|
echo "Source buildroot/build/envsetup.sh firstly!!!"
|
exit 1
|
fi
|
|
# Prefer using buildroot host tools for compatible.
|
HOST_DIR=$TARGET_OUTPUT_DIR/host
|
BUILDROOT_IMAGE_DIR=$HOST_DIR/../images/
|
export PATH=$HOST_DIR/usr/sbin:$HOST_DIR/usr/bin:$HOST_DIR/sbin:$HOST_DIR/bin:$PATH
|
|
fatal()
|
{
|
echo -e "FATAL: " $@
|
exit 1
|
}
|
|
usage()
|
{
|
echo $@
|
fatal "Usage: $0 <src_dir> <target_image> <fs_type> <size(M|K)|auto(0)> [label]"
|
}
|
|
[ ! $# -lt 4 ] || usage "Not enough args${@+: $0 $@}"
|
|
export SRC_DIR=$1
|
export TARGET=$2
|
FS_TYPE=$3
|
SIZE=$4
|
LABEL=$5
|
|
case $SIZE in
|
auto)
|
SIZE_KB=0
|
;;
|
*K)
|
SIZE_KB=$(( ${SIZE%K} ))
|
;;
|
*)
|
SIZE_KB=$(( ${SIZE%M} * 1024 )) # default is MB
|
;;
|
esac
|
|
echo $SIZE_KB | grep -vq [^0-9] || usage "Invalid size: $SIZE_KB"
|
|
if [ "$FS_TYPE" = "ubi" ]; then
|
UBI_VOL_NAME=${LABEL:-test}
|
# default page size 2KB
|
DEFAULT_UBI_PAGE_SIZE=${RK_UBI_PAGE_SIZE:-2048}
|
# default block size 128KB
|
DEFAULT_UBI_BLOCK_SIZE=${RK_UBI_BLOCK_SIZE:-0x20000}
|
fi
|
|
TEMP=$(mktemp -u)
|
|
[ -d "$SRC_DIR" ] || usage "No such src dir: $SRC_DIR"
|
|
copy_to_ntfs()
|
{
|
DEPTH=1
|
while true;do
|
find $SRC_DIR -maxdepth $DEPTH -mindepth $DEPTH -type d|grep -q "" \
|
|| break
|
find $SRC_DIR -maxdepth $DEPTH -mindepth $DEPTH -type d \
|
-exec sh -c 'ntfscp $TARGET "$1" "${1#$SRC_DIR}"' sh {} \; || \
|
fatal "Please update buildroot to: \n83c061e7c9 rockchip: Select host-ntfs-3g"
|
DEPTH=$(($DEPTH + 1))
|
done
|
|
find $SRC_DIR -type f \
|
-exec sh -c 'ntfscp $TARGET "$1" "${1#$SRC_DIR}"' sh {} \; || \
|
fatal "Failed to do ntfscp!"
|
}
|
|
copy_to_image()
|
{
|
ls $SRC_DIR/* &>/dev/null || return 0
|
|
echo "Copying $SRC_DIR into $TARGET (root permission required)"
|
mkdir -p $TEMP || return -1
|
sudo mount $TARGET $TEMP || return -1
|
|
cp -rp $SRC_DIR/* $TEMP
|
RET=$?
|
|
sudo umount $TEMP
|
rm -rf $TEMP
|
|
return $RET
|
}
|
|
check_host_tool()
|
{
|
which $1|grep -q "^$TARGET_OUTPUT_DIR"
|
}
|
|
mkimage()
|
{
|
echo "Making $TARGET from $SRC_DIR with size(${SIZE_KB}KB)"
|
rm -rf $TARGET
|
dd of=$TARGET bs=1K seek=$SIZE_KB count=0 &>/dev/null || \
|
fatal "Failed to dd image!"
|
case $FS_TYPE in
|
ext[234])
|
if mke2fs -h 2>&1 | grep -wq "\-d"; then
|
mke2fs -t $FS_TYPE $TARGET -d $SRC_DIR \
|
|| return -1
|
else
|
echo "Detected old mke2fs(doesn't support '-d' option)!"
|
mke2fs -t $FS_TYPE $TARGET || return -1
|
copy_to_image || return -1
|
fi
|
# Set max-mount-counts to 0, and disable the time-dependent checking.
|
tune2fs -c 0 -i 0 $TARGET ${LABEL:+-L $LABEL}
|
;;
|
msdos|fat|vfat)
|
# Use fat32 by default
|
mkfs.vfat -F 32 $TARGET ${LABEL:+-n $LABEL} && \
|
MTOOLS_SKIP_CHECK=1 \
|
mcopy -bspmn -D s -i $TARGET $SRC_DIR/* ::/
|
;;
|
ntfs)
|
# Enable compression
|
mkntfs -FCQ $TARGET ${LABEL:+-L $LABEL}
|
if check_host_tool ntfscp; then
|
copy_to_ntfs
|
else
|
copy_to_image
|
fi
|
;;
|
esac
|
}
|
|
mkimage_auto_sized()
|
{
|
tar cf $TEMP $SRC_DIR &>/dev/null
|
SIZE_KB=$(du -k $TEMP|grep -o "^[0-9]*")
|
rm -rf $TEMP
|
echo "Making $TARGET from $SRC_DIR (auto sized)"
|
|
MAX_RETRY=10
|
RETRY=0
|
|
while true;do
|
EXTRA_SIZE=$(($SIZE_KB / 50))
|
SIZE_KB=$(($SIZE_KB + ($EXTRA_SIZE > 4096 ? $EXTRA_SIZE : 4096)))
|
mkimage && break
|
|
RETRY=$[RETRY+1]
|
[ $RETRY -gt $MAX_RETRY ] && fatal "Failed to make image!"
|
echo "Retring with increased size....($RETRY/$MAX_RETRY)"
|
done
|
}
|
|
mk_ubi_image()
|
{
|
temp_dir="`dirname $TARGET`"
|
|
if [ $(( $UBI_BLOCK_SIZE )) -eq $(( 0x20000 )) ]; then
|
ubi_block_size_str="128KB"
|
elif [ $(( $UBI_BLOCK_SIZE )) -eq $(( 0x40000 )) ]; then
|
ubi_block_size_str="256KB"
|
else
|
echo "Error: Please add ubi block size [$UBI_BLOCK_SIZE] to $PWD/$0"
|
exit 1
|
fi
|
|
if [ $(( $UBI_PAGE_SIZE )) -eq 2048 ]; then
|
ubi_page_size_str="2KB"
|
elif [ $(( $UBI_PAGE_SIZE )) -eq 4096 ]; then
|
ubi_page_size_str="4KB"
|
else
|
echo "Error: Please add ubi block size [$UBI_PAGE_SIZE] to $PWD/$0"
|
exit 1
|
fi
|
|
if [ -z "$UBI_VOL_NAME" ]; then
|
echo "Error: Please config ubifs partition volume name"
|
exit 1
|
fi
|
|
ubifs_lebsize=$(( $UBI_BLOCK_SIZE - 2 * $UBI_PAGE_SIZE ))
|
ubifs_miniosize=$UBI_PAGE_SIZE
|
|
partition_size_str="$(( $SIZE_KB / 1024 ))MB"
|
output_image=${temp_dir}/${UBI_VOL_NAME}_${ubi_page_size_str}_${ubi_block_size_str}_${partition_size_str}.ubi
|
temp_ubifs_image=$BUILDROOT_IMAGE_DIR/${UBI_VOL_NAME}_${ubi_page_size_str}_${ubi_block_size_str}_${partition_size_str}.ubifs
|
temp_ubinize_file=$BUILDROOT_IMAGE_DIR/${UBI_VOL_NAME}_${ubi_page_size_str}_${ubi_block_size_str}_${partition_size_str}_ubinize.cfg
|
|
ubifs_maxlebcnt=$(( $SIZE_KB * 1024 / $ubifs_lebsize ))
|
|
echo "ubifs_lebsize=$UBI_BLOCK_SIZE"
|
echo "ubifs_miniosize=$UBI_PAGE_SIZE"
|
echo "ubifs_maxlebcnt=$ubifs_maxlebcnt"
|
mkfs.ubifs -x lzo -e $ubifs_lebsize -m $ubifs_miniosize -c $ubifs_maxlebcnt -d $SRC_DIR -F -v -o $temp_ubifs_image
|
|
echo "[ubifs]" > $temp_ubinize_file
|
echo "mode=ubi" >> $temp_ubinize_file
|
echo "vol_id=0" >> $temp_ubinize_file
|
echo "vol_type=dynamic" >> $temp_ubinize_file
|
echo "vol_name=$UBI_VOL_NAME" >> $temp_ubinize_file
|
echo "vol_alignment=1" >> $temp_ubinize_file
|
echo "vol_flags=autoresize" >> $temp_ubinize_file
|
echo "image=$temp_ubifs_image" >> $temp_ubinize_file
|
ubinize -o $output_image -m $ubifs_miniosize -p $UBI_BLOCK_SIZE -v $temp_ubinize_file
|
|
# Pick a default ubi image
|
if [ $(( $DEFAULT_UBI_PAGE_SIZE )) -eq $(( $UBI_PAGE_SIZE )) \
|
-a $(( $DEFAULT_UBI_BLOCK_SIZE )) -eq $(( $UBI_BLOCK_SIZE )) ]; then
|
ln -rfs $output_image $TARGET
|
fi
|
}
|
|
rm -rf $TARGET
|
case $FS_TYPE in
|
ext[234]|msdos|fat|vfat|ntfs)
|
if [ $SIZE_KB -eq 0 ]; then
|
mkimage_auto_sized
|
else
|
mkimage && echo "Generated $TARGET"
|
fi
|
;;
|
ubi)
|
[ $SIZE_KB -eq 0 ] && fatal "$FS_TYPE: auto size not supported."
|
|
UBI_PAGE_SIZE=2048
|
UBI_BLOCK_SIZE=0x20000
|
mk_ubi_image
|
|
UBI_PAGE_SIZE=2048
|
UBI_BLOCK_SIZE=0x40000
|
mk_ubi_image
|
|
UBI_PAGE_SIZE=4096
|
UBI_BLOCK_SIZE=0x40000
|
mk_ubi_image
|
;;
|
squashfs)
|
[ $SIZE_KB -eq 0 ] || fatal "$FS_TYPE: fixed size not supported."
|
mksquashfs $SRC_DIR $TARGET -noappend -comp lz4
|
;;
|
jffs2)
|
[ $SIZE_KB -eq 0 ] || fatal "$FS_TYPE: fixed size not supported."
|
mkfs.jffs2 -r $SRC_DIR -o $TARGET 0x10000 --pad=0x400000 -s 0x1000 -n
|
;;
|
*)
|
usage "File system: $FS_TYPE not supported."
|
;;
|
esac
|
