imx: imx8m: introduce script to generate fit image

Introduce script to generate fit image for i.MX8M

Signed-off-by: Peng Fan <peng.fan@nxp.com>

diff --git a/arch/arm/mach-imx/mkimage_fit_atf.sh b/arch/arm/mach-imx/mkimage_fit_atf.sh
new file mode 100755 (executable)
index 0000000..77f7143
--- /dev/null
+++ b/arch/arm/mach-imx/mkimage_fit_atf.sh
@@ -0,0 +1,137 @@
+#!/bin/sh
+# SPDX-License-Identifier: GPL-2.0+
+#
+# script to generate FIT image source for i.MX8MQ boards with
+# ARM Trusted Firmware and multiple device trees (given on the command line)
+#
+# usage: $0 <dt_name> [<dt_name> [<dt_name] ...]
+
+[ -z "$BL31" ] && BL31="bl31.bin"
+[ -z "$TEE_LOAD_ADDR" ] && TEE_LOAD_ADDR="0xfe000000"
+[ -z "$ATF_LOAD_ADDR" ] && ATF_LOAD_ADDR="0x00910000"
+
+if [ ! -f $BL31 ]; then
+       echo "ERROR: BL31 file $BL31 NOT found" >&2
+       exit 0
+else
+       echo "$BL31 size: " >&2
+       ls -lct $BL31 | awk '{print $5}' >&2
+fi
+
+BL32="tee.bin"
+
+if [ ! -f $BL32 ]; then
+       BL32=/dev/null
+else
+       echo "Building with TEE support, make sure your $BL31 is compiled with spd. If you do not want tee, please delete $BL31" >&2
+       echo "$BL32 size: " >&2
+       ls -lct $BL32 | awk '{print $5}' >&2
+fi
+
+BL33="u-boot-nodtb.bin"
+
+if [ ! -f $BL33 ]; then
+       echo "ERROR: $BL33 file NOT found" >&2
+       exit 0
+else
+       echo "u-boot-nodtb.bin size: " >&2
+       ls -lct u-boot-nodtb.bin | awk '{print $5}' >&2
+fi
+
+for dtname in $*
+do
+       echo "$dtname size: " >&2
+       ls -lct $dtname | awk '{print $5}' >&2
+done
+
+
+cat << __HEADER_EOF
+/dts-v1/;
+
+/ {
+       description = "Configuration to load ATF before U-Boot";
+
+       images {
+               uboot@1 {
+                       description = "U-Boot (64-bit)";
+                       data = /incbin/("$BL33");
+                       type = "standalone";
+                       arch = "arm64";
+                       compression = "none";
+                       load = <0x40200000>;
+               };
+               atf@1 {
+                       description = "ARM Trusted Firmware";
+                       data = /incbin/("$BL31");
+                       type = "firmware";
+                       arch = "arm64";
+                       compression = "none";
+                       load = <$ATF_LOAD_ADDR>;
+                       entry = <$ATF_LOAD_ADDR>;
+               };
+__HEADER_EOF
+
+if [ -f $BL32 ]; then
+cat << __HEADER_EOF
+               tee@1 {
+                       description = "TEE firmware";
+                       data = /incbin/("$BL32");
+                       type = "firmware";
+                       arch = "arm64";
+                       compression = "none";
+                       load = <$TEE_LOAD_ADDR>;
+                       entry = <$TEE_LOAD_ADDR>;
+               };
+__HEADER_EOF
+fi
+
+cnt=1
+for dtname in $*
+do
+       cat << __FDT_IMAGE_EOF
+               fdt@$cnt {
+                       description = "$(basename $dtname .dtb)";
+                       data = /incbin/("$dtname");
+                       type = "flat_dt";
+                       compression = "none";
+               };
+__FDT_IMAGE_EOF
+cnt=$((cnt+1))
+done
+
+cat << __CONF_HEADER_EOF
+       };
+       configurations {
+               default = "config@1";
+
+__CONF_HEADER_EOF
+
+cnt=1
+for dtname in $*
+do
+if [ -f $BL32 ]; then
+cat << __CONF_SECTION_EOF
+               config@$cnt {
+                       description = "$(basename $dtname .dtb)";
+                       firmware = "uboot@1";
+                       loadables = "atf@1", "tee@1";
+                       fdt = "fdt@$cnt";
+               };
+__CONF_SECTION_EOF
+else
+cat << __CONF_SECTION1_EOF
+               config@$cnt {
+                       description = "$(basename $dtname .dtb)";
+                       firmware = "uboot@1";
+                       loadables = "atf@1";
+                       fdt = "fdt@$cnt";
+               };
+__CONF_SECTION1_EOF
+fi
+cnt=$((cnt+1))
+done
+
+cat << __ITS_EOF
+       };
+};
+__ITS_EOF