tools/image-host.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (c) 2013, Google Inc.
   4  *
   5  * (C) Copyright 2008 Semihalf
   6  *
   7  * (C) Copyright 2000-2006
   8  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
   9  */
  10
  11 #include "mkimage.h"
  12 #include <bootm.h>
  13 #include <fdt_region.h>
  14 #include <image.h>
  15 #include <version.h>
  16
  17 /**
  18  * fit_set_hash_value - set hash value in requested has node
  19  * @fit: pointer to the FIT format image header
  20  * @noffset: hash node offset
  21  * @value: hash value to be set
  22  * @value_len: hash value length
  23  *
  24  * fit_set_hash_value() attempts to set hash value in a node at offset
  25  * given and returns operation status to the caller.
  26  *
  27  * returns
  28  *     0, on success
  29  *     -1, on failure
  30  */
  31 static int fit_set_hash_value(void *fit, int noffset, uint8_t *value,
  32                                 int value_len)
  33 {
  34         int ret;
  35
  36         ret = fdt_setprop(fit, noffset, FIT_VALUE_PROP, value, value_len);
  37         if (ret) {
  38                 printf("Can't set hash '%s' property for '%s' node(%s)\n",
  39                        FIT_VALUE_PROP, fit_get_name(fit, noffset, NULL),
  40                        fdt_strerror(ret));
  41                 return ret == -FDT_ERR_NOSPACE ? -ENOSPC : -EIO;
  42         }
  43
  44         return 0;
  45 }
  46
  47 /**
  48  * fit_image_process_hash - Process a single subnode of the images/ node
  49  *
  50  * Check each subnode and process accordingly. For hash nodes we generate
  51  * a hash of the supplied data and store it in the node.
  52  *
  53  * @fit:        pointer to the FIT format image header
  54  * @image_name: name of image being processed (used to display errors)
  55  * @noffset:    subnode offset
  56  * @data:       data to process
  57  * @size:       size of data in bytes
  58  * Return: 0 if ok, -1 on error
  59  */
  60 static int fit_image_process_hash(void *fit, const char *image_name,
  61                 int noffset, const void *data, size_t size)
  62 {
  63         uint8_t value[FIT_MAX_HASH_LEN];
  64         const char *node_name;
  65         int value_len;
  66         const char *algo;
  67         int ret;
  68
  69         node_name = fit_get_name(fit, noffset, NULL);
  70
  71         if (fit_image_hash_get_algo(fit, noffset, &algo)) {
  72                 printf("Can't get hash algo property for '%s' hash node in '%s' image node\n",
  73                        node_name, image_name);
  74                 return -ENOENT;
  75         }
  76
  77         if (calculate_hash(data, size, algo, value, &value_len)) {
  78                 printf("Unsupported hash algorithm (%s) for '%s' hash node in '%s' image node\n",
  79                        algo, node_name, image_name);
  80                 return -EPROTONOSUPPORT;
  81         }
  82
  83         ret = fit_set_hash_value(fit, noffset, value, value_len);
  84         if (ret) {
  85                 printf("Can't set hash value for '%s' hash node in '%s' image node\n",
  86                        node_name, image_name);
  87                 return ret;
  88         }
  89
  90         return 0;
  91 }
  92
  93 /**
  94  * fit_image_write_sig() - write the signature to a FIT
  95  *
  96  * This writes the signature and signer data to the FIT.
  97  *
  98  * @fit: pointer to the FIT format image header
  99  * @noffset: hash node offset
 100  * @value: signature value to be set
 101  * @value_len: signature value length
 102  * @comment: Text comment to write (NULL for none)
 103  *
 104  * returns
 105  *     0, on success
 106  *     -FDT_ERR_..., on failure
 107  */
 108 static int fit_image_write_sig(void *fit, int noffset, uint8_t *value,
 109                 int value_len, const char *comment, const char *region_prop,
 110                 int region_proplen, const char *cmdname, const char *algo_name)
 111 {
 112         int string_size;
 113         int ret;
 114
 115         /*
 116          * Get the current string size, before we update the FIT and add
 117          * more
 118          */
 119         string_size = fdt_size_dt_strings(fit);
 120
 121         ret = fdt_setprop(fit, noffset, FIT_VALUE_PROP, value, value_len);
 122         if (!ret) {
 123                 ret = fdt_setprop_string(fit, noffset, "signer-name",
 124                                          "mkimage");
 125         }
 126         if (!ret) {
 127                 ret = fdt_setprop_string(fit, noffset, "signer-version",
 128                                   PLAIN_VERSION);
 129         }
 130         if (comment && !ret)
 131                 ret = fdt_setprop_string(fit, noffset, "comment", comment);
 132         if (!ret) {
 133                 time_t timestamp = imagetool_get_source_date(cmdname,
 134                                                              time(NULL));
 135                 uint32_t t = cpu_to_uimage(timestamp);
 136
 137                 ret = fdt_setprop(fit, noffset, FIT_TIMESTAMP_PROP, &t,
 138                         sizeof(uint32_t));
 139         }
 140         if (region_prop && !ret) {
 141                 uint32_t strdata[2];
 142
 143                 ret = fdt_setprop(fit, noffset, "hashed-nodes",
 144                                    region_prop, region_proplen);
 145                 /* This is a legacy offset, it is unused, and must remain 0. */
 146                 strdata[0] = 0;
 147                 strdata[1] = cpu_to_fdt32(string_size);
 148                 if (!ret) {
 149                         ret = fdt_setprop(fit, noffset, "hashed-strings",
 150                                           strdata, sizeof(strdata));
 151                 }
 152         }
 153         if (algo_name && !ret)
 154                 ret = fdt_setprop_string(fit, noffset, "algo", algo_name);
 155
 156         return ret;
 157 }
 158
 159 static int fit_image_setup_sig(struct image_sign_info *info,
 160                 const char *keydir, const char *keyfile, void *fit,
 161                 const char *image_name, int noffset, const char *require_keys,
 162                 const char *engine_id, const char *algo_name)
 163 {
 164         const char *node_name;
 165         const char *padding_name;
 166
 167         node_name = fit_get_name(fit, noffset, NULL);
 168         if (!algo_name) {
 169                 if (fit_image_hash_get_algo(fit, noffset, &algo_name)) {
 170                         printf("Can't get algo property for '%s' signature node in '%s' image node\n",
 171                                node_name, image_name);
 172                         return -1;
 173                 }
 174         }
 175
 176         padding_name = fdt_getprop(fit, noffset, "padding", NULL);
 177
 178         memset(info, '\0', sizeof(*info));
 179         info->keydir = keydir;
 180         info->keyfile = keyfile;
 181         info->keyname = fdt_getprop(fit, noffset, FIT_KEY_HINT, NULL);
 182         info->fit = fit;
 183         info->node_offset = noffset;
 184         info->name = strdup(algo_name);
 185         info->checksum = image_get_checksum_algo(algo_name);
 186         info->crypto = image_get_crypto_algo(algo_name);
 187         info->padding = image_get_padding_algo(padding_name);
 188         info->require_keys = require_keys;
 189         info->engine_id = engine_id;
 190         if (!info->checksum || !info->crypto) {
 191                 printf("Unsupported signature algorithm (%s) for '%s' signature node in '%s' image node\n",
 192                        algo_name, node_name, image_name);
 193                 return -1;
 194         }
 195
 196         return 0;
 197 }
 198
 199 /**
 200  * fit_image_process_sig- Process a single subnode of the images/ node
 201  *
 202  * Check each subnode and process accordingly. For signature nodes we
 203  * generate a signed hash of the supplied data and store it in the node.
 204  *
 205  * @keydir:     Directory containing keys to use for signing
 206  * @keydest:    Destination FDT blob to write public keys into (NULL if none)
 207  * @fit:        pointer to the FIT format image header
 208  * @image_name: name of image being processed (used to display errors)
 209  * @noffset:    subnode offset
 210  * @data:       data to process
 211  * @size:       size of data in bytes
 212  * @comment:    Comment to add to signature nodes
 213  * @require_keys: Mark all keys as 'required'
 214  * @engine_id:  Engine to use for signing
 215  * Return: 0 if ok, -1 on error
 216  */
 217 static int fit_image_process_sig(const char *keydir, const char *keyfile,
 218                 void *keydest, void *fit, const char *image_name,
 219                 int noffset, const void *data, size_t size,
 220                 const char *comment, int require_keys, const char *engine_id,
 221                 const char *cmdname, const char *algo_name)
 222 {
 223         struct image_sign_info info;
 224         struct image_region region;
 225         const char *node_name;
 226         uint8_t *value;
 227         uint value_len;
 228         int ret;
 229
 230         if (fit_image_setup_sig(&info, keydir, keyfile, fit, image_name,
 231                                 noffset, require_keys ? "image" : NULL,
 232                                 engine_id, algo_name))
 233                 return -1;
 234
 235         node_name = fit_get_name(fit, noffset, NULL);
 236         region.data = data;
 237         region.size = size;
 238         ret = info.crypto->sign(&info, &region, 1, &value, &value_len);
 239         if (ret) {
 240                 printf("Failed to sign '%s' signature node in '%s' image node: %d\n",
 241                        node_name, image_name, ret);
 242
 243                 /* We allow keys to be missing */
 244                 if (ret == -ENOENT)
 245                         return 0;
 246                 return -1;
 247         }
 248
 249         ret = fit_image_write_sig(fit, noffset, value, value_len, comment,
 250                         NULL, 0, cmdname, algo_name);
 251         if (ret) {
 252                 if (ret == -FDT_ERR_NOSPACE)
 253                         return -ENOSPC;
 254                 printf("Can't write signature for '%s' signature node in '%s' conf node: %s\n",
 255                        node_name, image_name, fdt_strerror(ret));
 256                 return -1;
 257         }
 258         free(value);
 259
 260         /* Get keyname again, as FDT has changed and invalidated our pointer */
 261         info.keyname = fdt_getprop(fit, noffset, FIT_KEY_HINT, NULL);
 262
 263         /*
 264          * Write the public key into the supplied FDT file; this might fail
 265          * several times, since we try signing with successively increasing
 266          * size values
 267          */
 268         if (keydest) {
 269                 ret = info.crypto->add_verify_data(&info, keydest);
 270                 if (ret < 0) {
 271                         printf("Failed to add verification data for '%s' signature node in '%s' image node\n",
 272                                node_name, image_name);
 273                         return ret;
 274                 }
 275         }
 276
 277         return 0;
 278 }
 279
 280 static int fit_image_read_data(char *filename, unsigned char *data,
 281                                int expected_size)
 282 {
 283         struct stat sbuf;
 284         int fd, ret = -1;
 285         ssize_t n;
 286
 287         /* Open file */
 288         fd = open(filename, O_RDONLY | O_BINARY);
 289         if (fd < 0) {
 290                 printf("Can't open file %s (err=%d => %s)\n",
 291                        filename, errno, strerror(errno));
 292                 return -1;
 293         }
 294
 295         /* Compute file size */
 296         if (fstat(fd, &sbuf) < 0) {
 297                 printf("Can't fstat file %s (err=%d => %s)\n",
 298                        filename, errno, strerror(errno));
 299                 goto err;
 300         }
 301
 302         /* Check file size */
 303         if (sbuf.st_size != expected_size) {
 304                 printf("File %s don't have the expected size (size=%lld, expected=%d)\n",
 305                        filename, (long long)sbuf.st_size, expected_size);
 306                 goto err;
 307         }
 308
 309         /* Read data */
 310         n = read(fd, data, sbuf.st_size);
 311         if (n < 0) {
 312                 printf("Can't read file %s (err=%d => %s)\n",
 313                        filename, errno, strerror(errno));
 314                 goto err;
 315         }
 316
 317         /* Check that we have read all the file */
 318         if (n != sbuf.st_size) {
 319                 printf("Can't read all file %s (read %zd bytes, expected %lld)\n",
 320                        filename, n, (long long)sbuf.st_size);
 321                 goto err;
 322         }
 323
 324         ret = 0;
 325
 326 err:
 327         close(fd);
 328         return ret;
 329 }
 330
 331 static int get_random_data(void *data, int size)
 332 {
 333         unsigned char *tmp = data;
 334         struct timespec date;
 335         int i, ret;
 336
 337         if (!tmp) {
 338                 printf("%s: pointer data is NULL\n", __func__);
 339                 ret = -1;
 340                 goto out;
 341         }
 342
 343         ret = clock_gettime(CLOCK_MONOTONIC, &date);
 344         if (ret) {
 345                 printf("%s: clock_gettime has failed (%s)\n", __func__,
 346                        strerror(errno));
 347                 goto out;
 348         }
 349
 350         srandom(date.tv_nsec);
 351
 352         for (i = 0; i < size; i++) {
 353                 *tmp = random() & 0xff;
 354                 tmp++;
 355         }
 356
 357  out:
 358         return ret;
 359 }
 360
 361 static int fit_image_setup_cipher(struct image_cipher_info *info,
 362                                   const char *keydir, void *fit,
 363                                   const char *image_name, int image_noffset,
 364                                   int noffset)
 365 {
 366         char *algo_name;
 367         char filename[128];
 368         int ret = -1;
 369
 370         if (fit_image_cipher_get_algo(fit, noffset, &algo_name)) {
 371                 printf("Can't get algo name for cipher in image '%s'\n",
 372                        image_name);
 373                 goto out;
 374         }
 375
 376         info->keydir = keydir;
 377
 378         /* Read the key name */
 379         info->keyname = fdt_getprop(fit, noffset, FIT_KEY_HINT, NULL);
 380         if (!info->keyname) {
 381                 printf("Can't get key name for cipher in image '%s'\n",
 382                        image_name);
 383                 goto out;
 384         }
 385
 386         /*
 387          * Read the IV name
 388          *
 389          * If this property is not provided then mkimage will generate
 390          * a random IV and store it in the FIT image
 391          */
 392         info->ivname = fdt_getprop(fit, noffset, "iv-name-hint", NULL);
 393
 394         info->fit = fit;
 395         info->node_noffset = noffset;
 396         info->name = algo_name;
 397
 398         info->cipher = image_get_cipher_algo(algo_name);
 399         if (!info->cipher) {
 400                 printf("Can't get algo for cipher '%s'\n", image_name);
 401                 goto out;
 402         }
 403
 404         /* Read the key in the file */
 405         snprintf(filename, sizeof(filename), "%s/%s%s",
 406                  info->keydir, info->keyname, ".bin");
 407         info->key = malloc(info->cipher->key_len);
 408         if (!info->key) {
 409                 printf("Can't allocate memory for key\n");
 410                 ret = -1;
 411                 goto out;
 412         }
 413         ret = fit_image_read_data(filename, (unsigned char *)info->key,
 414                                   info->cipher->key_len);
 415         if (ret < 0)
 416                 goto out;
 417
 418         info->iv = malloc(info->cipher->iv_len);
 419         if (!info->iv) {
 420                 printf("Can't allocate memory for iv\n");
 421                 ret = -1;
 422                 goto out;
 423         }
 424
 425         if (info->ivname) {
 426                 /* Read the IV in the file */
 427                 snprintf(filename, sizeof(filename), "%s/%s%s",
 428                          info->keydir, info->ivname, ".bin");
 429                 ret = fit_image_read_data(filename, (unsigned char *)info->iv,
 430                                           info->cipher->iv_len);
 431         } else {
 432                 /* Generate an ramdom IV */
 433                 ret = get_random_data((void *)info->iv, info->cipher->iv_len);
 434         }
 435
 436  out:
 437         return ret;
 438 }
 439
 440 int fit_image_write_cipher(void *fit, int image_noffset, int noffset,
 441                            const void *data, size_t size,
 442                            unsigned char *data_ciphered, int data_ciphered_len)
 443 {
 444         int ret = -1;
 445
 446         /* Replace data with ciphered data */
 447         ret = fdt_setprop(fit, image_noffset, FIT_DATA_PROP,
 448                           data_ciphered, data_ciphered_len);
 449         if (ret == -FDT_ERR_NOSPACE) {
 450                 ret = -ENOSPC;
 451                 goto out;
 452         }
 453         if (ret) {
 454                 printf("Can't replace data with ciphered data (err = %d)\n", ret);
 455                 goto out;
 456         }
 457
 458         /* add non ciphered data size */
 459         ret = fdt_setprop_u32(fit, image_noffset, "data-size-unciphered", size);
 460         if (ret == -FDT_ERR_NOSPACE) {
 461                 ret = -ENOSPC;
 462                 goto out;
 463         }
 464         if (ret) {
 465                 printf("Can't add unciphered data size (err = %d)\n", ret);
 466                 goto out;
 467         }
 468
 469  out:
 470         return ret;
 471 }
 472
 473 static int
 474 fit_image_process_cipher(const char *keydir, void *keydest, void *fit,
 475                          const char *image_name, int image_noffset,
 476                          int node_noffset, const void *data, size_t size,
 477                          const char *cmdname)
 478 {
 479         struct image_cipher_info info;
 480         unsigned char *data_ciphered = NULL;
 481         int data_ciphered_len;
 482         int ret;
 483
 484         memset(&info, 0, sizeof(info));
 485
 486         ret = fit_image_setup_cipher(&info, keydir, fit, image_name,
 487                                      image_noffset, node_noffset);
 488         if (ret)
 489                 goto out;
 490
 491         ret = info.cipher->encrypt(&info, data, size,
 492                                     &data_ciphered, &data_ciphered_len);
 493         if (ret)
 494                 goto out;
 495
 496         /*
 497          * Write the public key into the supplied FDT file; this might fail
 498          * several times, since we try signing with successively increasing
 499          * size values
 500          * And, if needed, write the iv in the FIT file
 501          */
 502         if (keydest) {
 503                 ret = info.cipher->add_cipher_data(&info, keydest, fit, node_noffset);
 504                 if (ret) {
 505                         printf("Failed to add verification data for cipher '%s' in image '%s'\n",
 506                                info.keyname, image_name);
 507                         goto out;
 508                 }
 509         }
 510
 511         ret = fit_image_write_cipher(fit, image_noffset, node_noffset,
 512                                      data, size,
 513                                      data_ciphered, data_ciphered_len);
 514
 515  out:
 516         free(data_ciphered);
 517         free((void *)info.key);
 518         free((void *)info.iv);
 519         return ret;
 520 }
 521
 522 int fit_image_cipher_data(const char *keydir, void *keydest,
 523                           void *fit, int image_noffset, const char *comment,
 524                           int require_keys, const char *engine_id,
 525                           const char *cmdname)
 526 {
 527         const char *image_name;
 528         const void *data;
 529         size_t size;
 530         int cipher_node_offset, len;
 531
 532         /* Get image name */
 533         image_name = fit_get_name(fit, image_noffset, NULL);
 534         if (!image_name) {
 535                 printf("Can't get image name\n");
 536                 return -1;
 537         }
 538
 539         /* Get image data and data length */
 540         if (fit_image_get_data(fit, image_noffset, &data, &size)) {
 541                 printf("Can't get image data/size\n");
 542                 return -1;
 543         }
 544
 545         /*
 546          * Don't cipher ciphered data.
 547          *
 548          * If the data-size-unciphered property is present the data for this
 549          * image is already encrypted. This is important as 'mkimage -F' can be
 550          * run multiple times on a FIT image.
 551          */
 552         if (fdt_getprop(fit, image_noffset, "data-size-unciphered", &len))
 553                 return 0;
 554         if (len != -FDT_ERR_NOTFOUND) {
 555                 printf("Failure testing for data-size-unciphered\n");
 556                 return -1;
 557         }
 558
 559         /* Process cipher node if present */
 560         cipher_node_offset = fdt_subnode_offset(fit, image_noffset,
 561                                                 FIT_CIPHER_NODENAME);
 562         if (cipher_node_offset == -FDT_ERR_NOTFOUND)
 563                 return 0;
 564         if (cipher_node_offset < 0) {
 565                 printf("Failure getting cipher node\n");
 566                 return -1;
 567         }
 568         if (!IMAGE_ENABLE_ENCRYPT || !keydir)
 569                 return 0;
 570         return fit_image_process_cipher(keydir, keydest, fit, image_name,
 571                 image_noffset, cipher_node_offset, data, size, cmdname);
 572 }
 573
 574 /**
 575  * fit_image_add_verification_data() - calculate/set verig. data for image node
 576  *
 577  * This adds hash and signature values for an component image node.
 578  *
 579  * All existing hash subnodes are checked, if algorithm property is set to
 580  * one of the supported hash algorithms, hash value is computed and
 581  * corresponding hash node property is set, for example:
 582  *
 583  * Input component image node structure:
 584  *
 585  * o image-1 (at image_noffset)
 586  *   | - data = [binary data]
 587  *   o hash-1
 588  *     |- algo = "sha1"
 589  *
 590  * Output component image node structure:
 591  *
 592  * o image-1 (at image_noffset)
 593  *   | - data = [binary data]
 594  *   o hash-1
 595  *     |- algo = "sha1"
 596  *     |- value = sha1(data)
 597  *
 598  * For signature details, please see doc/uImage.FIT/signature.txt
 599  *
 600  * @keydir      Directory containing *.key and *.crt files (or NULL)
 601  * @keydest     FDT Blob to write public keys into (NULL if none)
 602  * @fit:        Pointer to the FIT format image header
 603  * @image_noffset: Requested component image node
 604  * @comment:    Comment to add to signature nodes
 605  * @require_keys: Mark all keys as 'required'
 606  * @engine_id:  Engine to use for signing
 607  * @return: 0 on success, <0 on failure
 608  */
 609 int fit_image_add_verification_data(const char *keydir, const char *keyfile,
 610                 void *keydest, void *fit, int image_noffset,
 611                 const char *comment, int require_keys, const char *engine_id,
 612                 const char *cmdname, const char* algo_name)
 613 {
 614         const char *image_name;
 615         const void *data;
 616         size_t size;
 617         int noffset;
 618
 619         /* Get image data and data length */
 620         if (fit_image_get_data(fit, image_noffset, &data, &size)) {
 621                 printf("Can't get image data/size\n");
 622                 return -1;
 623         }
 624
 625         image_name = fit_get_name(fit, image_noffset, NULL);
 626
 627         /* Process all hash subnodes of the component image node */
 628         for (noffset = fdt_first_subnode(fit, image_noffset);
 629              noffset >= 0;
 630              noffset = fdt_next_subnode(fit, noffset)) {
 631                 const char *node_name;
 632                 int ret = 0;
 633
 634                 /*
 635                  * Check subnode name, must be equal to "hash" or "signature".
 636                  * Multiple hash nodes require unique unit node
 637                  * names, e.g. hash-1, hash-2, signature-1, etc.
 638                  */
 639                 node_name = fit_get_name(fit, noffset, NULL);
 640                 if (!strncmp(node_name, FIT_HASH_NODENAME,
 641                              strlen(FIT_HASH_NODENAME))) {
 642                         ret = fit_image_process_hash(fit, image_name, noffset,
 643                                                 data, size);
 644                 } else if (IMAGE_ENABLE_SIGN && (keydir || keyfile) &&
 645                            !strncmp(node_name, FIT_SIG_NODENAME,
 646                                 strlen(FIT_SIG_NODENAME))) {
 647                         ret = fit_image_process_sig(keydir, keyfile, keydest,
 648                                 fit, image_name, noffset, data, size,
 649                                 comment, require_keys, engine_id, cmdname,
 650                                 algo_name);
 651                 }
 652                 if (ret)
 653                         return ret;
 654         }
 655
 656         return 0;
 657 }
 658
 659 struct strlist {
 660         int count;
 661         char **strings;
 662 };
 663
 664 static void strlist_init(struct strlist *list)
 665 {
 666         memset(list, '\0', sizeof(*list));
 667 }
 668
 669 static void strlist_free(struct strlist *list)
 670 {
 671         int i;
 672
 673         for (i = 0; i < list->count; i++)
 674                 free(list->strings[i]);
 675         free(list->strings);
 676 }
 677
 678 static int strlist_add(struct strlist *list, const char *str)
 679 {
 680         char *dup;
 681
 682         dup = strdup(str);
 683         list->strings = realloc(list->strings,
 684                                 (list->count + 1) * sizeof(char *));
 685         if (!list || !str)
 686                 return -1;
 687         list->strings[list->count++] = dup;
 688
 689         return 0;
 690 }
 691
 692 static const char *fit_config_get_image_list(const void *fit, int noffset,
 693                                              int *lenp, int *allow_missingp)
 694 {
 695         static const char default_list[] = FIT_KERNEL_PROP "\0"
 696                         FIT_FDT_PROP;
 697         const char *prop;
 698
 699         /* If there is an "sign-image" property, use that */
 700         prop = fdt_getprop(fit, noffset, "sign-images", lenp);
 701         if (prop) {
 702                 *allow_missingp = 0;
 703                 return *lenp ? prop : NULL;
 704         }
 705
 706         /* Default image list */
 707         *allow_missingp = 1;
 708         *lenp = sizeof(default_list);
 709
 710         return default_list;
 711 }
 712
 713 /**
 714  * fit_config_add_hash() - Add a list of nodes to hash for an image
 715  *
 716  * This adds a list of paths to image nodes (as referred to by a particular
 717  * offset) that need to be hashed, to protect a configuration
 718  *
 719  * @fit:        Pointer to the FIT format image header
 720  * @image_noffset: Offset of image to process (e.g. /images/kernel-1)
 721  * @node_inc:   List of nodes to add to
 722  * @conf_name   Configuration-node name, child of /configurations node (only
 723  *      used for error messages)
 724  * @sig_name    Signature-node name (only used for error messages)
 725  * @iname:      Name of image being processed (e.g. "kernel-1" (only used
 726  *      for error messages)
 727  */
 728 static int fit_config_add_hash(const void *fit, int image_noffset,
 729                                struct strlist *node_inc, const char *conf_name,
 730                                const char *sig_name, const char *iname)
 731 {
 732         char path[200];
 733         int noffset;
 734         int hash_count;
 735         int ret;
 736
 737         ret = fdt_get_path(fit, image_noffset, path, sizeof(path));
 738         if (ret < 0)
 739                 goto err_path;
 740         if (strlist_add(node_inc, path))
 741                 goto err_mem;
 742
 743         /* Add all this image's hashes */
 744         hash_count = 0;
 745         for (noffset = fdt_first_subnode(fit, image_noffset);
 746              noffset >= 0;
 747              noffset = fdt_next_subnode(fit, noffset)) {
 748                 const char *name = fit_get_name(fit, noffset, NULL);
 749
 750                 if (strncmp(name, FIT_HASH_NODENAME,
 751                             strlen(FIT_HASH_NODENAME)))
 752                         continue;
 753                 ret = fdt_get_path(fit, noffset, path, sizeof(path));
 754                 if (ret < 0)
 755                         goto err_path;
 756                 if (strlist_add(node_inc, path))
 757                         goto err_mem;
 758                 hash_count++;
 759         }
 760
 761         if (!hash_count) {
 762                 printf("Failed to find any hash nodes in configuration '%s/%s' image '%s' - without these it is not possible to verify this image\n",
 763                        conf_name, sig_name, iname);
 764                 return -ENOMSG;
 765         }
 766
 767         /* Add this image's cipher node if present */
 768         noffset = fdt_subnode_offset(fit, image_noffset,
 769                                      FIT_CIPHER_NODENAME);
 770         if (noffset != -FDT_ERR_NOTFOUND) {
 771                 if (noffset < 0) {
 772                         printf("Failed to get cipher node in configuration '%s/%s' image '%s': %s\n",
 773                                conf_name, sig_name, iname,
 774                                fdt_strerror(noffset));
 775                         return -EIO;
 776                 }
 777                 ret = fdt_get_path(fit, noffset, path, sizeof(path));
 778                 if (ret < 0)
 779                         goto err_path;
 780                 if (strlist_add(node_inc, path))
 781                         goto err_mem;
 782         }
 783
 784         return 0;
 785
 786 err_mem:
 787         printf("Out of memory processing configuration '%s/%s'\n", conf_name,
 788                sig_name);
 789         return -ENOMEM;
 790
 791 err_path:
 792         printf("Failed to get path for image '%s' in configuration '%s/%s': %s\n",
 793                iname, conf_name, sig_name, fdt_strerror(ret));
 794         return -ENOENT;
 795 }
 796
 797 /**
 798  * fit_config_get_hash_list() - Get the regions to sign
 799  *
 800  * This calculates a list of nodes to hash for this particular configuration,
 801  * returning it as a string list (struct strlist, not a devicetree string list)
 802  *
 803  * @fit:        Pointer to the FIT format image header
 804  * @conf_noffset: Offset of configuration node to sign (child of
 805  *      /configurations node)
 806  * @sig_offset: Offset of signature node containing info about how to sign it
 807  *      (child of 'signatures' node)
 808  * @return 0 if OK, -ENOENT if an image referred to by the configuration cannot
 809  *      be found, -ENOMSG if ther were no images in the configuration
 810  */
 811 static int fit_config_get_hash_list(const void *fit, int conf_noffset,
 812                                     int sig_offset, struct strlist *node_inc)
 813 {
 814         int allow_missing;
 815         const char *prop, *iname, *end;
 816         const char *conf_name, *sig_name;
 817         char name[200];
 818         int image_count;
 819         int ret, len;
 820
 821         conf_name = fit_get_name(fit, conf_noffset, NULL);
 822         sig_name = fit_get_name(fit, sig_offset, NULL);
 823
 824         /*
 825          * Build a list of nodes we need to hash. We always need the root
 826          * node and the configuration.
 827          */
 828         strlist_init(node_inc);
 829         snprintf(name, sizeof(name), "%s/%s", FIT_CONFS_PATH, conf_name);
 830         if (strlist_add(node_inc, "/") ||
 831             strlist_add(node_inc, name))
 832                 goto err_mem;
 833
 834         /* Get a list of images that we intend to sign */
 835         prop = fit_config_get_image_list(fit, sig_offset, &len,
 836                                         &allow_missing);
 837         if (!prop)
 838                 return 0;
 839
 840         /* Locate the images */
 841         end = prop + len;
 842         image_count = 0;
 843         for (iname = prop; iname < end; iname += strlen(iname) + 1) {
 844                 int image_noffset;
 845                 int index, max_index;
 846
 847                 max_index = fdt_stringlist_count(fit, conf_noffset, iname);
 848
 849                 for (index = 0; index < max_index; index++) {
 850                         image_noffset = fit_conf_get_prop_node_index(fit, conf_noffset,
 851                                                                      iname, index);
 852
 853                         if (image_noffset < 0) {
 854                                 printf("Failed to find image '%s' in  configuration '%s/%s'\n",
 855                                        iname, conf_name, sig_name);
 856                                 if (allow_missing)
 857                                         continue;
 858
 859                                 return -ENOENT;
 860                         }
 861
 862                         ret = fit_config_add_hash(fit, image_noffset, node_inc,
 863                                                   conf_name, sig_name, iname);
 864                         if (ret < 0)
 865                                 return ret;
 866
 867                         image_count++;
 868                 }
 869         }
 870
 871         if (!image_count) {
 872                 printf("Failed to find any images for configuration '%s/%s'\n",
 873                        conf_name, sig_name);
 874                 return -ENOMSG;
 875         }
 876
 877         return 0;
 878
 879 err_mem:
 880         printf("Out of memory processing configuration '%s/%s'\n", conf_name,
 881                sig_name);
 882         return -ENOMEM;
 883 }
 884
 885 /**
 886  * fit_config_get_regions() - Get the regions to sign
 887  *
 888  * This calculates a list of node to hash for this particular configuration,
 889  * then finds which regions of the devicetree they correspond to.
 890  *
 891  * @fit:        Pointer to the FIT format image header
 892  * @conf_noffset: Offset of configuration node to sign (child of
 893  *      /configurations node)
 894  * @sig_offset: Offset of signature node containing info about how to sign it
 895  *      (child of 'signatures' node)
 896  * @regionp: Returns list of regions that need to be hashed (allocated; must be
 897  *      freed by the caller)
 898  * @region_count: Returns number of regions
 899  * @region_propp: Returns string-list property containing the list of nodes
 900  *      that correspond to the regions. Each entry is a full path to the node.
 901  *      This is in devicetree format, i.e. a \0 between each string. This is
 902  *      allocated and must be freed by the caller.
 903  * @region_proplen: Returns length of *@@region_propp in bytes
 904  * @return 0 if OK, -ENOMEM if out of memory, -EIO if the regions to hash could
 905  * not be found, -EINVAL if no registers were found to hash
 906  */
 907 static int fit_config_get_regions(const void *fit, int conf_noffset,
 908                                   int sig_offset, struct image_region **regionp,
 909                                   int *region_countp, char **region_propp,
 910                                   int *region_proplen)
 911 {
 912         char * const exc_prop[] = {"data"};
 913         struct strlist node_inc;
 914         struct image_region *region;
 915         struct fdt_region fdt_regions[100];
 916         const char *conf_name, *sig_name;
 917         char path[200];
 918         int count, i;
 919         char *region_prop;
 920         int ret, len;
 921
 922         conf_name = fit_get_name(fit, conf_noffset, NULL);
 923         sig_name = fit_get_name(fit, sig_offset, NULL);
 924         debug("%s: conf='%s', sig='%s'\n", __func__, conf_name, sig_name);
 925
 926         /* Get a list of nodes we want to hash */
 927         ret = fit_config_get_hash_list(fit, conf_noffset, sig_offset,
 928                                        &node_inc);
 929         if (ret)
 930                 return ret;
 931
 932         /* Get a list of regions to hash */
 933         count = fdt_find_regions(fit, node_inc.strings, node_inc.count,
 934                         exc_prop, ARRAY_SIZE(exc_prop),
 935                         fdt_regions, ARRAY_SIZE(fdt_regions),
 936                         path, sizeof(path), 1);
 937         if (count < 0) {
 938                 printf("Failed to hash configuration '%s/%s': %s\n", conf_name,
 939                        sig_name, fdt_strerror(ret));
 940                 return -EIO;
 941         }
 942         if (count == 0) {
 943                 printf("No data to hash for configuration '%s/%s': %s\n",
 944                        conf_name, sig_name, fdt_strerror(ret));
 945                 return -EINVAL;
 946         }
 947
 948         /* Build our list of data blocks */
 949         region = fit_region_make_list(fit, fdt_regions, count, NULL);
 950         if (!region) {
 951                 printf("Out of memory hashing configuration '%s/%s'\n",
 952                        conf_name, sig_name);
 953                 return -ENOMEM;
 954         }
 955
 956         /* Create a list of all hashed properties */
 957         debug("Hash nodes:\n");
 958         for (i = len = 0; i < node_inc.count; i++) {
 959                 debug("   %s\n", node_inc.strings[i]);
 960                 len += strlen(node_inc.strings[i]) + 1;
 961         }
 962         region_prop = malloc(len);
 963         if (!region_prop) {
 964                 printf("Out of memory setting up regions for configuration '%s/%s'\n",
 965                        conf_name, sig_name);
 966                 return -ENOMEM;
 967         }
 968         for (i = len = 0; i < node_inc.count;
 969              len += strlen(node_inc.strings[i]) + 1, i++)
 970                 strcpy(region_prop + len, node_inc.strings[i]);
 971         strlist_free(&node_inc);
 972
 973         *region_countp = count;
 974         *regionp = region;
 975         *region_propp = region_prop;
 976         *region_proplen = len;
 977
 978         return 0;
 979 }
 980
 981 static int fit_config_process_sig(const char *keydir, const char *keyfile,
 982                 void *keydest, void *fit, const char *conf_name,
 983                 int conf_noffset, int noffset, const char *comment,
 984                 int require_keys, const char *engine_id, const char *cmdname,
 985                 const char *algo_name)
 986 {
 987         struct image_sign_info info;
 988         const char *node_name;
 989         struct image_region *region;
 990         char *region_prop;
 991         int region_proplen;
 992         int region_count;
 993         uint8_t *value;
 994         uint value_len;
 995         int ret;
 996
 997         node_name = fit_get_name(fit, noffset, NULL);
 998         if (fit_config_get_regions(fit, conf_noffset, noffset, &region,
 999                                    &region_count, &region_prop,
1000                                    &region_proplen))
1001                 return -1;
1002
1003         if (fit_image_setup_sig(&info, keydir, keyfile, fit, conf_name, noffset,
1004                                 require_keys ? "conf" : NULL, engine_id,
1005                                 algo_name))
1006                 return -1;
1007
1008         ret = info.crypto->sign(&info, region, region_count, &value,
1009                                 &value_len);
1010         free(region);
1011         if (ret) {
1012                 printf("Failed to sign '%s' signature node in '%s' conf node\n",
1013                        node_name, conf_name);
1014
1015                 /* We allow keys to be missing */
1016                 if (ret == -ENOENT)
1017                         return 0;
1018                 return -1;
1019         }
1020
1021         ret = fit_image_write_sig(fit, noffset, value, value_len, comment,
1022                                   region_prop, region_proplen, cmdname,
1023                                   algo_name);
1024         if (ret) {
1025                 if (ret == -FDT_ERR_NOSPACE)
1026                         return -ENOSPC;
1027                 printf("Can't write signature for '%s' signature node in '%s' conf node: %s\n",
1028                        node_name, conf_name, fdt_strerror(ret));
1029                 return -1;
1030         }
1031         free(value);
1032         free(region_prop);
1033
1034         /* Get keyname again, as FDT has changed and invalidated our pointer */
1035         info.keyname = fdt_getprop(fit, noffset, FIT_KEY_HINT, NULL);
1036
1037         /* Write the public key into the supplied FDT file */
1038         if (keydest) {
1039                 ret = info.crypto->add_verify_data(&info, keydest);
1040                 if (ret < 0) {
1041                         printf("Failed to add verification data for '%s' signature node in '%s' configuration node\n",
1042                                node_name, conf_name);
1043                 }
1044         }
1045
1046         return 0;
1047 }
1048
1049 static int fit_config_add_verification_data(const char *keydir,
1050                 const char *keyfile, void *keydest, void *fit, int conf_noffset,
1051                 const char *comment, int require_keys, const char *engine_id,
1052                 const char *cmdname, const char *algo_name)
1053 {
1054         const char *conf_name;
1055         int noffset;
1056
1057         conf_name = fit_get_name(fit, conf_noffset, NULL);
1058
1059         /* Process all hash subnodes of the configuration node */
1060         for (noffset = fdt_first_subnode(fit, conf_noffset);
1061              noffset >= 0;
1062              noffset = fdt_next_subnode(fit, noffset)) {
1063                 const char *node_name;
1064                 int ret = 0;
1065
1066                 node_name = fit_get_name(fit, noffset, NULL);
1067                 if (!strncmp(node_name, FIT_SIG_NODENAME,
1068                              strlen(FIT_SIG_NODENAME))) {
1069                         ret = fit_config_process_sig(keydir, keyfile, keydest,
1070                                 fit, conf_name, conf_noffset, noffset, comment,
1071                                 require_keys, engine_id, cmdname, algo_name);
1072                 }
1073                 if (ret)
1074                         return ret;
1075         }
1076
1077         return 0;
1078 }
1079
1080 int fit_cipher_data(const char *keydir, void *keydest, void *fit,
1081                     const char *comment, int require_keys,
1082                     const char *engine_id, const char *cmdname)
1083 {
1084         int images_noffset;
1085         int noffset;
1086         int ret;
1087
1088         /* Find images parent node offset */
1089         images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
1090         if (images_noffset < 0) {
1091                 printf("Can't find images parent node '%s' (%s)\n",
1092                        FIT_IMAGES_PATH, fdt_strerror(images_noffset));
1093                 return images_noffset;
1094         }
1095
1096         /* Process its subnodes, print out component images details */
1097         for (noffset = fdt_first_subnode(fit, images_noffset);
1098              noffset >= 0;
1099              noffset = fdt_next_subnode(fit, noffset)) {
1100                 /*
1101                  * Direct child node of the images parent node,
1102                  * i.e. component image node.
1103                  */
1104                 ret = fit_image_cipher_data(keydir, keydest,
1105                                             fit, noffset, comment,
1106                                             require_keys, engine_id,
1107                                             cmdname);
1108                 if (ret)
1109                         return ret;
1110         }
1111
1112         return 0;
1113 }
1114
1115 int fit_add_verification_data(const char *keydir, const char *keyfile,
1116                               void *keydest, void *fit, const char *comment,
1117                               int require_keys, const char *engine_id,
1118                               const char *cmdname, const char *algo_name)
1119 {
1120         int images_noffset, confs_noffset;
1121         int noffset;
1122         int ret;
1123
1124         /* Find images parent node offset */
1125         images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
1126         if (images_noffset < 0) {
1127                 printf("Can't find images parent node '%s' (%s)\n",
1128                        FIT_IMAGES_PATH, fdt_strerror(images_noffset));
1129                 return images_noffset;
1130         }
1131
1132         /* Process its subnodes, print out component images details */
1133         for (noffset = fdt_first_subnode(fit, images_noffset);
1134              noffset >= 0;
1135              noffset = fdt_next_subnode(fit, noffset)) {
1136                 /*
1137                  * Direct child node of the images parent node,
1138                  * i.e. component image node.
1139                  */
1140                 ret = fit_image_add_verification_data(keydir, keyfile, keydest,
1141                                 fit, noffset, comment, require_keys, engine_id,
1142                                 cmdname, algo_name);
1143                 if (ret)
1144                         return ret;
1145         }
1146
1147         /* If there are no keys, we can't sign configurations */
1148         if (!IMAGE_ENABLE_SIGN || !(keydir || keyfile))
1149                 return 0;
1150
1151         /* Find configurations parent node offset */
1152         confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH);
1153         if (confs_noffset < 0) {
1154                 printf("Can't find images parent node '%s' (%s)\n",
1155                        FIT_CONFS_PATH, fdt_strerror(confs_noffset));
1156                 return -ENOENT;
1157         }
1158
1159         /* Process its subnodes, print out component images details */
1160         for (noffset = fdt_first_subnode(fit, confs_noffset);
1161              noffset >= 0;
1162              noffset = fdt_next_subnode(fit, noffset)) {
1163                 ret = fit_config_add_verification_data(keydir, keyfile, keydest,
1164                                                        fit, noffset, comment,
1165                                                        require_keys,
1166                                                        engine_id, cmdname,
1167                                                        algo_name);
1168                 if (ret)
1169                         return ret;
1170         }
1171
1172         return 0;
1173 }
1174
1175 #ifdef CONFIG_FIT_SIGNATURE
1176 int fit_check_sign(const void *fit, const void *key,
1177                    const char *fit_uname_config)
1178 {
1179         int cfg_noffset;
1180         int ret;
1181
1182         cfg_noffset = fit_conf_get_node(fit, fit_uname_config);
1183         if (!cfg_noffset)
1184                 return -1;
1185
1186         printf("Verifying Hash Integrity for node '%s'... ",
1187                fdt_get_name(fit, cfg_noffset, NULL));
1188         ret = fit_config_verify(fit, cfg_noffset);
1189         if (ret)
1190                 return ret;
1191         printf("Verified OK, loading images\n");
1192         ret = bootm_host_load_images(fit, cfg_noffset);
1193
1194         return ret;
1195 }
1196 #endif