Home | History | Annotate | Download | only in tools
      1 // SPDX-License-Identifier: GPL-2.0+
      2 /*
      3  * (C) Copyright 2014
      4  * DENX Software Engineering
      5  * Heiko Schocher <hs (at) denx.de>
      6  *
      7  * Based on:
      8  * (C) Copyright 2008 Semihalf
      9  *
     10  * (C) Copyright 2000-2004
     11  * DENX Software Engineering
     12  * Wolfgang Denk, wd (at) denx.de
     13  *
     14  * Updated-by: Prafulla Wadaskar <prafulla (at) marvell.com>
     15  *		FIT image specific code abstracted from mkimage.c
     16  *		some functions added to address abstraction
     17  *
     18  * All rights reserved.
     19  */
     20 
     21 #include "mkimage.h"
     22 #include "fit_common.h"
     23 #include <image.h>
     24 #include <u-boot/crc.h>
     25 
     26 void usage(char *cmdname)
     27 {
     28 	fprintf(stderr, "Usage: %s -f fit file -k key file\n"
     29 			 "          -f ==> set fit file which should be checked'\n"
     30 			 "          -k ==> set key file which contains the key'\n",
     31 		cmdname);
     32 	exit(EXIT_FAILURE);
     33 }
     34 
     35 int main(int argc, char **argv)
     36 {
     37 	int ffd = -1;
     38 	int kfd = -1;
     39 	struct stat fsbuf;
     40 	struct stat ksbuf;
     41 	void *fit_blob;
     42 	char *fdtfile = NULL;
     43 	char *keyfile = NULL;
     44 	char cmdname[256];
     45 	int ret;
     46 	void *key_blob;
     47 	int c;
     48 
     49 	strncpy(cmdname, *argv, sizeof(cmdname) - 1);
     50 	cmdname[sizeof(cmdname) - 1] = '\0';
     51 	while ((c = getopt(argc, argv, "f:k:")) != -1)
     52 		switch (c) {
     53 		case 'f':
     54 			fdtfile = optarg;
     55 			break;
     56 		case 'k':
     57 			keyfile = optarg;
     58 			break;
     59 		default:
     60 			usage(cmdname);
     61 			break;
     62 	}
     63 
     64 	if (!fdtfile) {
     65 		fprintf(stderr, "%s: Missing fdt file\n", *argv);
     66 		usage(*argv);
     67 	}
     68 	if (!keyfile) {
     69 		fprintf(stderr, "%s: Missing key file\n", *argv);
     70 		usage(*argv);
     71 	}
     72 
     73 	ffd = mmap_fdt(cmdname, fdtfile, 0, &fit_blob, &fsbuf, false);
     74 	if (ffd < 0)
     75 		return EXIT_FAILURE;
     76 	kfd = mmap_fdt(cmdname, keyfile, 0, &key_blob, &ksbuf, false);
     77 	if (kfd < 0)
     78 		return EXIT_FAILURE;
     79 
     80 	image_set_host_blob(key_blob);
     81 	ret = fit_check_sign(fit_blob, key_blob);
     82 	if (!ret) {
     83 		ret = EXIT_SUCCESS;
     84 		fprintf(stderr, "Signature check OK\n");
     85 	} else {
     86 		ret = EXIT_FAILURE;
     87 		fprintf(stderr, "Signature check Bad (error %d)\n", ret);
     88 	}
     89 
     90 	(void) munmap((void *)fit_blob, fsbuf.st_size);
     91 	(void) munmap((void *)key_blob, ksbuf.st_size);
     92 
     93 	close(ffd);
     94 	close(kfd);
     95 	exit(ret);
     96 }
     97