LXR uboot/tools/fit_check

   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * (C) Copyright 2014
   4 * DENX Software Engineering
   5 * Heiko Schocher <hs@denx.de>
   6 *
   7 * Based on:
   8 * (C) Copyright 2008 Semihalf
   9 *
  10 * (C) Copyright 2000-2004
  11 * DENX Software Engineering
  12 * Wolfgang Denk, wd@denx.de
  13 *
  14 * Updated-by: Prafulla Wadaskar <prafulla@marvell.com>
  15 *              FIT image specific code abstracted from mkimage.c
  16 *              some functions added to address abstraction
  17 *
  18 * All rights reserved.
  19 */
  20
  21#include "mkimage.h"
  22#include "fit_common.h"
  23#include <image.h>
  24#include <u-boot/crc.h>
  25
  26void usage(char *cmdname)
  27{
  28        fprintf(stderr, "Usage: %s -f fit file -k key file -c config name\n"
  29                         "          -f ==> set fit file which should be checked'\n"
  30                         "          -k ==> set key .dtb file which contains the key'\n"
  31                         "          -c ==> set the configuration name'\n",
  32                cmdname);
  33        exit(EXIT_FAILURE);
  34}
  35
  36int 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 *config_name = NULL;
  46        char cmdname[256];
  47        int ret;
  48        void *key_blob;
  49        int c;
  50
  51        strncpy(cmdname, *argv, sizeof(cmdname) - 1);
  52        cmdname[sizeof(cmdname) - 1] = '\0';
  53        while ((c = getopt(argc, argv, "f:k:c:")) != -1)
  54                switch (c) {
  55                case 'f':
  56                        fdtfile = optarg;
  57                        break;
  58                case 'k':
  59                        keyfile = optarg;
  60                        break;
  61                case 'c':
  62                        config_name = optarg;
  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, true);
  79        if (ffd < 0)
  80                return EXIT_FAILURE;
  81        kfd = mmap_fdt(cmdname, keyfile, 0, &key_blob, &ksbuf, false, true);
  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, config_name);
  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