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 ffd = mmap_fdt(cmdname, fdtfile, &fit_blob, &fsbuf, 0); 66 if (ffd < 0) 67 return EXIT_FAILURE; 68 kfd = mmap_fdt(cmdname, keyfile, &key_blob, &ksbuf, 0); 69 if (ffd < 0) 70 return EXIT_FAILURE; 71 72 image_set_host_blob(key_blob); 73 ret = fit_check_sign(fit_blob, key_blob); 74 75 if (ret) 76 ret = EXIT_SUCCESS; 77 else 78 ret = EXIT_FAILURE; 79 80 (void) munmap((void *)fit_blob, fsbuf.st_size); 81 (void) munmap((void *)key_blob, ksbuf.st_size); 82 83 close(ffd); 84 close(kfd); 85 exit(ret); 86 } 87