tools: Check arguments in fit_check_sign/fit_info

[oweals/u-boot.git] / tools / fit_check_sign.c

diff --git a/tools/fit_check_sign.c b/tools/fit_check_sign.c
index af257cc8a2a2bf0bce3b51efc78f9aa58d1db4a0..e1198bca4498983593b31b9afd16b35c7713c202 100644 (file)
--- a/tools/fit_check_sign.c
+++ b/tools/fit_check_sign.c
@@ -62,6 +62,15 @@ int main(int argc, char **argv)
                        break;
        }
 
+       if (!fdtfile) {
+               fprintf(stderr, "%s: Missing fdt file\n", *argv);
+               usage(*argv);
+       }
+       if (!keyfile) {
+               fprintf(stderr, "%s: Missing key file\n", *argv);
+               usage(*argv);
+       }
+
        ffd = mmap_fdt(cmdname, fdtfile, 0, &fit_blob, &fsbuf, false);
        if (ffd < 0)
                return EXIT_FAILURE;