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