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
usage(char * cmdname)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
main(int argc,char ** argv)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 int is_spl = 0;
50
51 strncpy(cmdname, *argv, sizeof(cmdname) - 1);
52 cmdname[sizeof(cmdname) - 1] = '\0';
53 while ((c = getopt(argc, argv, "f:k:s")) != -1)
54 switch (c) {
55 case 'f':
56 fdtfile = optarg;
57 break;
58 case 'k':
59 keyfile = optarg;
60 break;
61 case 's':
62 is_spl = 1;
63 break;
64 default:
65 usage(cmdname);
66 break;
67 }
68
69 if (!fdtfile) {
70 fprintf(stderr, "%s: Missing fdt file\n", *argv);
71 usage(*argv);
72 }
73 if (!keyfile) {
74 fprintf(stderr, "%s: Missing key file\n", *argv);
75 usage(*argv);
76 }
77
78 ffd = mmap_fdt(cmdname, fdtfile, 0, &fit_blob, &fsbuf, false);
79 if (ffd < 0)
80 return EXIT_FAILURE;
81 kfd = mmap_fdt(cmdname, keyfile, 0, &key_blob, &ksbuf, false);
82 if (kfd < 0)
83 return EXIT_FAILURE;
84
85 image_set_host_blob(key_blob);
86 ret = fit_check_sign(fit_blob, key_blob, is_spl);
87 if (!ret) {
88 ret = EXIT_SUCCESS;
89 fprintf(stderr, "Signature check OK\n");
90 } else {
91 ret = EXIT_FAILURE;
92 fprintf(stderr, "Signature check Bad (error %d)\n", ret);
93 }
94
95 (void) munmap((void *)fit_blob, fsbuf.st_size);
96 (void) munmap((void *)key_blob, ksbuf.st_size);
97
98 close(ffd);
99 close(kfd);
100 exit(ret);
101 }
102