[U-Boot] [PATCH] rockchip: make_fit_atf: Use BL31 environ variable for file location

[Emmanuel Vadot]

Other make_fit script (like imx or sunxi) use the BL31 environment
variable to indicate the location of the file.
Also do that for rockchip so we don't need to copy the file in the source
directory.

Signed-off-by: Emmanuel Vadot <manu at freebsd.org>
---
 arch/arm/mach-rockchip/make_fit_atf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/arch/arm/mach-rockchip/make_fit_atf.py b/arch/arm/mach-rockchip/make_fit_atf.py
index d1faff1957..7bf58e3dd1 100755
--- a/arch/arm/mach-rockchip/make_fit_atf.py
+++ b/arch/arm/mach-rockchip/make_fit_atf.py
@@ -194,7 +194,7 @@ def get_bl31_segments_info(bl31_file_name):
 
 def main():
     uboot_elf="./u-boot"
-    bl31_elf="./bl31.elf"
+    bl31_elf=os.getenv("BL31", "./bl31.elf")
     FIT_ITS=sys.stdout
 
     opts, args = getopt.getopt(sys.argv[1:], "o:u:b:h")
-- 
2.20.1