forked from rrcarlosr/Jetpack
98 lines
2.0 KiB
C
98 lines
2.0 KiB
C
|
/*
|
||
|
* (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;
|
||
|
|
||
|
strncpy(cmdname, *argv, sizeof(cmdname) - 1);
|
||
|
cmdname[sizeof(cmdname) - 1] = '\0';
|
||
|
while ((c = getopt(argc, argv, "f:k:")) != -1)
|
||
|
switch (c) {
|
||
|
case 'f':
|
||
|
fdtfile = optarg;
|
||
|
break;
|
||
|
case 'k':
|
||
|
keyfile = optarg;
|
||
|
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);
|
||
|
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);
|
||
|
}
|