1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
229a23f9dSHeiko Schocher /*
329a23f9dSHeiko Schocher * (C) Copyright 2014
429a23f9dSHeiko Schocher * DENX Software Engineering
529a23f9dSHeiko Schocher * Heiko Schocher <hs@denx.de>
629a23f9dSHeiko Schocher *
729a23f9dSHeiko Schocher * Based on:
829a23f9dSHeiko Schocher * (C) Copyright 2008 Semihalf
929a23f9dSHeiko Schocher *
1029a23f9dSHeiko Schocher * (C) Copyright 2000-2004
1129a23f9dSHeiko Schocher * DENX Software Engineering
1229a23f9dSHeiko Schocher * Wolfgang Denk, wd@denx.de
1329a23f9dSHeiko Schocher *
1429a23f9dSHeiko Schocher * Updated-by: Prafulla Wadaskar <prafulla@marvell.com>
1529a23f9dSHeiko Schocher * FIT image specific code abstracted from mkimage.c
1629a23f9dSHeiko Schocher * some functions added to address abstraction
1729a23f9dSHeiko Schocher *
1829a23f9dSHeiko Schocher * All rights reserved.
1929a23f9dSHeiko Schocher */
2029a23f9dSHeiko Schocher
2129a23f9dSHeiko Schocher #include "mkimage.h"
2229a23f9dSHeiko Schocher #include "fit_common.h"
2329a23f9dSHeiko Schocher #include <image.h>
2429a23f9dSHeiko Schocher #include <u-boot/crc.h>
2529a23f9dSHeiko Schocher
usage(char * cmdname)2629a23f9dSHeiko Schocher void usage(char *cmdname)
2729a23f9dSHeiko Schocher {
2829a23f9dSHeiko Schocher fprintf(stderr, "Usage: %s -f fit file -k key file\n"
2929a23f9dSHeiko Schocher " -f ==> set fit file which should be checked'\n"
3029a23f9dSHeiko Schocher " -k ==> set key file which contains the key'\n",
3129a23f9dSHeiko Schocher cmdname);
3229a23f9dSHeiko Schocher exit(EXIT_FAILURE);
3329a23f9dSHeiko Schocher }
3429a23f9dSHeiko Schocher
main(int argc,char ** argv)3529a23f9dSHeiko Schocher int main(int argc, char **argv)
3629a23f9dSHeiko Schocher {
3729a23f9dSHeiko Schocher int ffd = -1;
3829a23f9dSHeiko Schocher int kfd = -1;
3929a23f9dSHeiko Schocher struct stat fsbuf;
4029a23f9dSHeiko Schocher struct stat ksbuf;
4129a23f9dSHeiko Schocher void *fit_blob;
4229a23f9dSHeiko Schocher char *fdtfile = NULL;
4329a23f9dSHeiko Schocher char *keyfile = NULL;
4464375014SMichael van der Westhuizen char cmdname[256];
4529a23f9dSHeiko Schocher int ret;
4629a23f9dSHeiko Schocher void *key_blob;
4729a23f9dSHeiko Schocher int c;
4829a23f9dSHeiko Schocher
4964375014SMichael van der Westhuizen strncpy(cmdname, *argv, sizeof(cmdname) - 1);
5064375014SMichael van der Westhuizen cmdname[sizeof(cmdname) - 1] = '\0';
5129a23f9dSHeiko Schocher while ((c = getopt(argc, argv, "f:k:")) != -1)
5229a23f9dSHeiko Schocher switch (c) {
5329a23f9dSHeiko Schocher case 'f':
5429a23f9dSHeiko Schocher fdtfile = optarg;
5529a23f9dSHeiko Schocher break;
5629a23f9dSHeiko Schocher case 'k':
5729a23f9dSHeiko Schocher keyfile = optarg;
5829a23f9dSHeiko Schocher break;
5929a23f9dSHeiko Schocher default:
6029a23f9dSHeiko Schocher usage(cmdname);
6129a23f9dSHeiko Schocher break;
6229a23f9dSHeiko Schocher }
6329a23f9dSHeiko Schocher
64ba923cabSSimon Glass if (!fdtfile) {
65ba923cabSSimon Glass fprintf(stderr, "%s: Missing fdt file\n", *argv);
66ba923cabSSimon Glass usage(*argv);
67ba923cabSSimon Glass }
68ba923cabSSimon Glass if (!keyfile) {
69ba923cabSSimon Glass fprintf(stderr, "%s: Missing key file\n", *argv);
70ba923cabSSimon Glass usage(*argv);
71ba923cabSSimon Glass }
72ba923cabSSimon Glass
73a9468115SSimon Glass ffd = mmap_fdt(cmdname, fdtfile, 0, &fit_blob, &fsbuf, false);
7429a23f9dSHeiko Schocher if (ffd < 0)
7529a23f9dSHeiko Schocher return EXIT_FAILURE;
76a9468115SSimon Glass kfd = mmap_fdt(cmdname, keyfile, 0, &key_blob, &ksbuf, false);
77310ae37eSThomas Huth if (kfd < 0)
7829a23f9dSHeiko Schocher return EXIT_FAILURE;
7929a23f9dSHeiko Schocher
8029a23f9dSHeiko Schocher image_set_host_blob(key_blob);
8129a23f9dSHeiko Schocher ret = fit_check_sign(fit_blob, key_blob);
82ce1400f6SSimon Glass if (!ret) {
8329a23f9dSHeiko Schocher ret = EXIT_SUCCESS;
84ce1400f6SSimon Glass fprintf(stderr, "Signature check OK\n");
85ce1400f6SSimon Glass } else {
8629a23f9dSHeiko Schocher ret = EXIT_FAILURE;
87ce1400f6SSimon Glass fprintf(stderr, "Signature check Bad (error %d)\n", ret);
88ce1400f6SSimon Glass }
8929a23f9dSHeiko Schocher
9029a23f9dSHeiko Schocher (void) munmap((void *)fit_blob, fsbuf.st_size);
9129a23f9dSHeiko Schocher (void) munmap((void *)key_blob, ksbuf.st_size);
9229a23f9dSHeiko Schocher
9329a23f9dSHeiko Schocher close(ffd);
9429a23f9dSHeiko Schocher close(kfd);
9529a23f9dSHeiko Schocher exit(ret);
9629a23f9dSHeiko Schocher }
97