From 7ba6d1048cfd6c72b1cbf847e90a055e66a640dc Mon Sep 17 00:00:00 2001
From: MilhouseVH <milhouseVH.github@nmacleod.com>
Date: Tue, 13 Aug 2019 00:21:55 +0100
Subject: [PATCH] image: validate UBOOT_SYSTEM

---
 scripts/image | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/scripts/image b/scripts/image
index 5c91ee1d08..91af47a093 100755
--- a/scripts/image
+++ b/scripts/image
@@ -10,6 +10,15 @@ unset _CACHE_PACKAGE_LOCAL _CACHE_PACKAGE_GLOBAL _DEBUG_DEPENDS_LIST _DEBUG_PACK
 . config/multithread
 . config/show_config
 
+# Validate UBOOT_SYSTEM if it is specified
+if [ "${BOOTLOADER}" = "u-boot" -a -n "${DEVICE}" ]; then
+  if [ -z "${UBOOT_SYSTEM}" ]; then
+    ${SCRIPTS}/uboot_helper ${PROJECT} ${DEVICE} >/dev/null
+  elif [ "${UBOOT_VERSION}" != "vendor" ]; then
+    ${SCRIPTS}/uboot_helper ${PROJECT} ${DEVICE} ${UBOOT_SYSTEM} dtb >/dev/null
+  fi
+fi
+
 show_config
 save_build_config
 
@@ -310,9 +319,12 @@ if [ "${1}" = "release" -o "${1}" = "mkimage" -o "${1}" = "noobs" ]; then
     UUID_SYSTEM="$(date '+%d%m')-$(date '+%M%S')"
     UUID_STORAGE="$(uuidgen)"
 
-    DEVICE_BOARDS=$(${SCRIPTS}/uboot_helper "${PROJECT}" "${DEVICE}")
+    DEVICE_BOARDS=
+    if [ "${BOOTLOADER}" = "u-boot" -a -z "${UBOOT_SYSTEM}" -a -n "${DEVICE}" ]; then
+      DEVICE_BOARDS=$(${SCRIPTS}/uboot_helper "${PROJECT}" "${DEVICE}")
+    fi
 
-    if [ "${BOOTLOADER}" = "u-boot" -a -z "${UBOOT_SYSTEM}" -a -n "${DEVICE}" -a -n "${DEVICE_BOARDS}" ]; then
+    if [ -n "${DEVICE_BOARDS}" ]; then
       for UBOOT_SYSTEM in ${DEVICE_BOARDS}; do
 
         # Re-install u-boot package