145 lines
2.8 KiB
Bash
Executable File
145 lines
2.8 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Copyright (c) 2020 Fuzhou Rockchip Electronics Co., Ltd
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0
|
|
#
|
|
set -e
|
|
|
|
IMAGE_ITS="image.its"
|
|
|
|
function usage()
|
|
{
|
|
echo
|
|
echo "usage:"
|
|
echo " $0 -f [fit/itb] -o [output]"
|
|
echo
|
|
}
|
|
|
|
function args_process()
|
|
{
|
|
if [ $# -ne 4 -a $# -ne 2 ]; then
|
|
usage
|
|
exit 1
|
|
fi
|
|
|
|
while [ $# -gt 0 ]; do
|
|
case $1 in
|
|
-f)
|
|
IMAGE=$2
|
|
shift 2
|
|
;;
|
|
-o)
|
|
IMAGE_DIR=$2
|
|
shift 2
|
|
;;
|
|
*)
|
|
usage
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [ ! -f $IMAGE ]; then
|
|
echo "ERROR: No $IMAGE"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z $IMAGE_DIR ]; then
|
|
IMAGE_DIR="out"
|
|
fi
|
|
|
|
mkdir -p $IMAGE_DIR
|
|
}
|
|
|
|
function gen_images()
|
|
{
|
|
printf "\n# Unpack $IMAGE to directory $IMAGE_DIR/\n"
|
|
fdtget -l $IMAGE /images > $IMAGE_DIR/unpack.txt
|
|
cat $IMAGE_DIR/unpack.txt | while read line
|
|
do
|
|
# generate image
|
|
NODE="/images/${line}"
|
|
NAME=`fdtget -ts $IMAGE $NODE image`
|
|
OFFS=`fdtget -ti $IMAGE $NODE data-position`
|
|
SIZE=`fdtget -ti $IMAGE $NODE data-size`
|
|
if [ -z $OFFS ]; then
|
|
continue;
|
|
fi
|
|
|
|
if [ $SIZE -ne 0 ]; then
|
|
dd if=$IMAGE of=$IMAGE_DIR/dd.tmp bs=$OFFS skip=1 >/dev/null 2>&1
|
|
dd if=$IMAGE_DIR/dd.tmp of=$IMAGE_DIR/$NAME bs=$SIZE count=1 >/dev/null 2>&1
|
|
rm $IMAGE_DIR/dd.tmp
|
|
else
|
|
touch $IMAGE_DIR/$NAME
|
|
fi
|
|
|
|
# hash verify
|
|
algo=`fdtget -ts $IMAGE $NODE/hash@1 algo`
|
|
if [ -z $algo ]; then
|
|
printf " %-20s: %d bytes" $NAME $SIZE
|
|
continue;
|
|
fi
|
|
|
|
data=`fdtget -tx $IMAGE $NODE/hash@1 value`
|
|
data=`echo " "$data | sed "s/ / 0x/g"`
|
|
csum=`"$algo"sum $IMAGE_DIR/$NAME | awk '{ print $1}'`
|
|
|
|
hash=""
|
|
for((i=1;;i++));
|
|
do
|
|
hex=`echo $data | awk -v idx=$i '{ print $idx }'`
|
|
if [ -z $hex ]; then
|
|
break;
|
|
fi
|
|
|
|
hex=`printf "%08x" $hex` # align !!
|
|
hash="$hash$hex"
|
|
done
|
|
|
|
printf " %-20s: %d bytes... %s" $NAME $SIZE $algo
|
|
if [ "$csum" = "$hash" -o $SIZE -eq 0 ]; then
|
|
echo "+"
|
|
else
|
|
echo "-"
|
|
fi
|
|
done
|
|
|
|
echo
|
|
}
|
|
|
|
function gen_its()
|
|
{
|
|
./scripts/dtc/dtc -I dtb -O dts $IMAGE -o $IMAGE_DIR/$IMAGE_ITS >/dev/null 2>&1
|
|
|
|
FIT_IMAGE_ITS=$IMAGE_DIR/$IMAGE_ITS
|
|
|
|
# remove
|
|
sed -i "/memreserve/d" $FIT_IMAGE_ITS
|
|
sed -i "/data-size/d" $FIT_IMAGE_ITS
|
|
sed -i "/data-position/d" $FIT_IMAGE_ITS
|
|
sed -i "/value/d" $FIT_IMAGE_ITS
|
|
sed -i "/hashed-strings/d" $FIT_IMAGE_ITS
|
|
sed -i "/hashed-nodes/d" $FIT_IMAGE_ITS
|
|
sed -i "/signer-version/d" $FIT_IMAGE_ITS
|
|
sed -i "/signer-name/d" $FIT_IMAGE_ITS
|
|
sed -i "/timestamp/d" $FIT_IMAGE_ITS
|
|
|
|
# add placeholder
|
|
sed -i '/image = /a\ \ \ data = /incbin/("IMAGE_PATH");' $FIT_IMAGE_ITS
|
|
|
|
# fixup placeholder: "data = /incbin/("...");"
|
|
num=`grep 'image =' $FIT_IMAGE_ITS | wc -l`
|
|
for ((i = 1; i <= $num; i++));
|
|
do
|
|
NAME=`grep 'image =' $FIT_IMAGE_ITS | sed -n ''${i}p'' | awk '{ printf $3 }' | tr -d '";'`
|
|
sed -i ''$i',/IMAGE_PATH/{s/IMAGE_PATH/.\/'$NAME'/}' $FIT_IMAGE_ITS
|
|
done
|
|
}
|
|
|
|
args_process $*
|
|
gen_images
|
|
gen_its
|
|
|