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