scripts: support Sercomm load tags

[oweals/openwrt.git] / scripts / ubinize-image.sh

diff --git a/scripts/ubinize-image.sh b/scripts/ubinize-image.sh
index 6ffedc5f32620deec6c9de494001fa59a32638be..c6f8bcefe5e04bc3e3093873f593aa8ab0ada9f2 100755 (executable)
--- a/scripts/ubinize-image.sh
+++ b/scripts/ubinize-image.sh
@@ -1,28 +1,22 @@
 #!/bin/sh
 
+. $TOPDIR/scripts/functions.sh
+
+part=""
 ubootenv=""
-nokernel=""
 ubinize_param=""
 kernel=""
 rootfs=""
 outfile=""
 err=""
-
-get_magic_word() {
-       dd if=$1 bs=2 count=1 2>/dev/null | hexdump -v -n 2 -e '1/1 "%02x"'
-}
-
-is_ubifs() {
-       if [ "$( get_magic_word $1 )" = "3118" ]; then
-               echo "1"
-       fi
-}
+ubinize_seq=""
 
 ubivol() {
        volid=$1
        name=$2
        image=$3
        autoresize=$4
+       size="$5"
        echo "[$name]"
        echo "mode=ubi"
        echo "vol_id=$volid"
@@ -30,6 +24,7 @@ ubivol() {
        echo "vol_name=$name"
        if [ "$image" ]; then
                echo "image=$image"
+               [ -n "$size" ] && echo "vol_size=${size}"
        else
                echo "vol_size=1MiB"
        fi
@@ -40,20 +35,57 @@ ubivol() {
 
 ubilayout() {
        local vol_id=0
-       local root_is_ubifs="$( is_ubifs "$2" )"
+       local rootsize=
+       local autoresize=
+       local rootfs_type="$( get_fs_type "$2" )"
+
        if [ "$1" = "ubootenv" ]; then
                ubivol $vol_id ubootenv
                vol_id=$(( $vol_id + 1 ))
                ubivol $vol_id ubootenv2
                vol_id=$(( $vol_id + 1 ))
        fi
+       for part in $parts; do
+               name="${part%%=*}"
+               prev="$part"
+               part="${part#*=}"
+               [ "$prev" = "$part" ] && part=
+
+               image="${part%%=*}"
+               prev="$part"
+               part="${part#*=}"
+               [ "$prev" = "$part" ] && part=
+
+               size="$part"
+
+               ubivol $vol_id "$name" "$image" "" "${size}MiB"
+               vol_id=$(( $vol_id + 1 ))
+       done
        if [ "$3" ]; then
                ubivol $vol_id kernel "$3"
                vol_id=$(( $vol_id + 1 ))
        fi
-       ubivol $vol_id rootfs "$2" $root_is_ubifs
+
+       case "$rootfs_type" in
+       "ubifs")
+               autoresize=1
+               ;;
+       "squashfs")
+               # squashfs uses 1k block size, ensure we do not
+               # violate that
+               rootsize="$( round_up "$( stat -c%s "$2" )" 1024 )"
+               ;;
+       esac
+       ubivol $vol_id rootfs "$2" "$autoresize" "$rootsize"
+
        vol_id=$(( $vol_id + 1 ))
-       [ "$root_is_ubifs" ] || ubivol $vol_id rootfs_data "" 1
+       [ "$rootfs_type" = "ubifs" ] || ubivol $vol_id rootfs_data "" 1
+}
+
+set_ubinize_seq() {
+       if [ -n "$SOURCE_DATE_EPOCH" ] ; then
+               ubinize_seq="-Q $SOURCE_DATE_EPOCH"
+       fi
 }
 
 while [ "$1" ]; do
@@ -63,8 +95,15 @@ while [ "$1" ]; do
                shift
                continue
                ;;
-       "--no-kernel")
-               nokernel="nokernel"
+       "--kernel")
+               kernel="$2"
+               shift
+               shift
+               continue
+               ;;
+       "--part")
+               parts="$parts $2"
+               shift
                shift
                continue
                ;;
@@ -73,11 +112,6 @@ while [ "$1" ]; do
                break
                ;;
        *)
-               if [ ! "$kernel" -a ! "$nokernel" ]; then
-                       kernel=$1
-                       shift
-                       continue
-               fi
                if [ ! "$rootfs" ]; then
                        rootfs=$1
                        shift
@@ -92,8 +126,8 @@ while [ "$1" ]; do
        esac
 done
 
-if [ ! -r "$rootfs" -o ! -r "$kernel" -a ! "$nokernel" -o ! "$outfile" ]; then
-       echo "syntax: $0 [--no-kernel] [--uboot-env] rootfs [kernel] out [ubinize opts]"
+if [ ! -r "$rootfs" -o ! -r "$kernel" -a ! "$outfile" ]; then
+       echo "syntax: $0 [--uboot-env] [--part <name>=<file>] [--kernel kernelimage] rootfs out [ubinize opts]"
        exit 1
 fi
 
@@ -103,14 +137,18 @@ if [ ! -x "$ubinize" ]; then
        exit 1
 fi
 
-ubinizecfg="$( mktemp )"
+ubinizecfg="$( mktemp 2> /dev/null )"
+if [ -z "$ubinizecfg" ]; then
+       # try OSX signature
+       ubinizecfg="$( mktemp -t 'ubitmp' )"
+fi
 ubilayout "$ubootenv" "$rootfs" "$kernel" > "$ubinizecfg"
 
+set_ubinize_seq
 cat "$ubinizecfg"
-ubinize -o "$outfile" $ubinize_param "$ubinizecfg"
+ubinize $ubinize_seq -o "$outfile" $ubinize_param "$ubinizecfg"
 err="$?"
 [ ! -e "$outfile" ] && err=2
 rm "$ubinizecfg"
 
 exit $err
-