rockchip: misc: protect serial# from getting overwritten

[oweals/u-boot.git] / cmd / source.c

diff --git a/cmd/source.c b/cmd/source.c
index 6d98a1cfd32b802ac2b0ff80d39fb3ec12fe47de..3a51ebf0b665c5d58db508f5d6b453f1bd64463a 100644 (file)
--- a/cmd/source.c
+++ b/cmd/source.c
@@ -16,6 +16,7 @@
 
 #include <common.h>
 #include <command.h>
+#include <env.h>
 #include <image.h>
 #include <malloc.h>
 #include <mapmem.h>
@@ -44,7 +45,7 @@ int
 source (ulong addr, const char *fit_uname)
 {
        ulong           len;
-#if defined(CONFIG_IMAGE_FORMAT_LEGACY)
+#if defined(CONFIG_LEGACY_IMAGE_FORMAT)
        const image_header_t *hdr;
 #endif
        u32             *data;
@@ -61,7 +62,7 @@ source (ulong addr, const char *fit_uname)
 
        buf = map_sysmem(addr, 0);
        switch (genimg_get_format(buf)) {
-#if defined(CONFIG_IMAGE_FORMAT_LEGACY)
+#if defined(CONFIG_LEGACY_IMAGE_FORMAT)
        case IMAGE_FORMAT_LEGACY:
                hdr = buf;