common/image-sig.c

   1 /*
   2  * Copyright (c) 2013, Google Inc.
   3  *
   4  * SPDX-License-Identifier:     GPL-2.0+
   5  */
   6
   7 #ifdef USE_HOSTCC
   8 #include "mkimage.h"
   9 #include <time.h>
  10 #else
  11 #include <common.h>
  12 #include <malloc.h>
  13 DECLARE_GLOBAL_DATA_PTR;
  14 #endif /* !USE_HOSTCC*/
  15 #include <image.h>
  16 #include <u-boot/rsa.h>
  17 #include <u-boot/rsa-checksum.h>
  18
  19 #define IMAGE_MAX_HASHED_NODES          100
  20
  21 #ifdef USE_HOSTCC
  22 void *host_blob;
  23 void image_set_host_blob(void *blob)
  24 {
  25         host_blob = blob;
  26 }
  27 void *image_get_host_blob(void)
  28 {
  29         return host_blob;
  30 }
  31 #endif
  32
  33 struct checksum_algo checksum_algos[] = {
  34         {
  35                 "sha1",
  36                 SHA1_SUM_LEN,
  37                 RSA2048_BYTES,
  38 #if IMAGE_ENABLE_SIGN
  39                 EVP_sha1,
  40 #endif
  41                 sha1_calculate,
  42                 padding_sha1_rsa2048,
  43         },
  44         {
  45                 "sha256",
  46                 SHA256_SUM_LEN,
  47                 RSA2048_BYTES,
  48 #if IMAGE_ENABLE_SIGN
  49                 EVP_sha256,
  50 #endif
  51                 sha256_calculate,
  52                 padding_sha256_rsa2048,
  53         },
  54         {
  55                 "sha256",
  56                 SHA256_SUM_LEN,
  57                 RSA4096_BYTES,
  58 #if IMAGE_ENABLE_SIGN
  59                 EVP_sha256,
  60 #endif
  61                 sha256_calculate,
  62                 padding_sha256_rsa4096,
  63         }
  64
  65 };
  66
  67 struct image_sig_algo image_sig_algos[] = {
  68         {
  69                 "sha1,rsa2048",
  70                 rsa_sign,
  71                 rsa_add_verify_data,
  72                 rsa_verify,
  73                 &checksum_algos[0],
  74         },
  75         {
  76                 "sha256,rsa2048",
  77                 rsa_sign,
  78                 rsa_add_verify_data,
  79                 rsa_verify,
  80                 &checksum_algos[1],
  81         },
  82         {
  83                 "sha256,rsa4096",
  84                 rsa_sign,
  85                 rsa_add_verify_data,
  86                 rsa_verify,
  87                 &checksum_algos[2],
  88         }
  89
  90 };
  91
  92 struct image_sig_algo *image_get_sig_algo(const char *name)
  93 {
  94         int i;
  95
  96         for (i = 0; i < ARRAY_SIZE(image_sig_algos); i++) {
  97                 if (!strcmp(image_sig_algos[i].name, name))
  98                         return &image_sig_algos[i];
  99         }
 100
 101         return NULL;
 102 }
 103
 104 /**
 105  * fit_region_make_list() - Make a list of image regions
 106  *
 107  * Given a list of fdt_regions, create a list of image_regions. This is a
 108  * simple conversion routine since the FDT and image code use different
 109  * structures.
 110  *
 111  * @fit: FIT image
 112  * @fdt_regions: Pointer to FDT regions
 113  * @count: Number of FDT regions
 114  * @region: Pointer to image regions, which must hold @count records. If
 115  * region is NULL, then (except for an SPL build) the array will be
 116  * allocated.
 117  * @return: Pointer to image regions
 118  */
 119 struct image_region *fit_region_make_list(const void *fit,
 120                 struct fdt_region *fdt_regions, int count,
 121                 struct image_region *region)
 122 {
 123         int i;
 124
 125         debug("Hash regions:\n");
 126         debug("%10s %10s\n", "Offset", "Size");
 127
 128         /*
 129          * Use malloc() except in SPL (to save code size). In SPL the caller
 130          * must allocate the array.
 131          */
 132 #ifndef CONFIG_SPL_BUILD
 133         if (!region)
 134                 region = calloc(sizeof(*region), count);
 135 #endif
 136         if (!region)
 137                 return NULL;
 138         for (i = 0; i < count; i++) {
 139                 debug("%10x %10x\n", fdt_regions[i].offset,
 140                       fdt_regions[i].size);
 141                 region[i].data = fit + fdt_regions[i].offset;
 142                 region[i].size = fdt_regions[i].size;
 143         }
 144
 145         return region;
 146 }
 147
 148 static int fit_image_setup_verify(struct image_sign_info *info,
 149                 const void *fit, int noffset, int required_keynode,
 150                 char **err_msgp)
 151 {
 152         char *algo_name;
 153
 154         if (fit_image_hash_get_algo(fit, noffset, &algo_name)) {
 155                 *err_msgp = "Can't get hash algo property";
 156                 return -1;
 157         }
 158         memset(info, '\0', sizeof(*info));
 159         info->keyname = fdt_getprop(fit, noffset, "key-name-hint", NULL);
 160         info->fit = (void *)fit;
 161         info->node_offset = noffset;
 162         info->algo = image_get_sig_algo(algo_name);
 163         info->fdt_blob = gd_fdt_blob();
 164         info->required_keynode = required_keynode;
 165         printf("%s:%s", algo_name, info->keyname);
 166
 167         if (!info->algo) {
 168                 *err_msgp = "Unknown signature algorithm";
 169                 return -1;
 170         }
 171
 172         return 0;
 173 }
 174
 175 int fit_image_check_sig(const void *fit, int noffset, const void *data,
 176                 size_t size, int required_keynode, char **err_msgp)
 177 {
 178         struct image_sign_info info;
 179         struct image_region region;
 180         uint8_t *fit_value;
 181         int fit_value_len;
 182
 183         *err_msgp = NULL;
 184         if (fit_image_setup_verify(&info, fit, noffset, required_keynode,
 185                                    err_msgp))
 186                 return -1;
 187
 188         if (fit_image_hash_get_value(fit, noffset, &fit_value,
 189                                      &fit_value_len)) {
 190                 *err_msgp = "Can't get hash value property";
 191                 return -1;
 192         }
 193
 194         region.data = data;
 195         region.size = size;
 196
 197         if (info.algo->verify(&info, &region, 1, fit_value, fit_value_len)) {
 198                 *err_msgp = "Verification failed";
 199                 return -1;
 200         }
 201
 202         return 0;
 203 }
 204
 205 static int fit_image_verify_sig(const void *fit, int image_noffset,
 206                 const char *data, size_t size, const void *sig_blob,
 207                 int sig_offset)
 208 {
 209         int noffset;
 210         char *err_msg = "";
 211         int verified = 0;
 212         int ret;
 213
 214         /* Process all hash subnodes of the component image node */
 215         for (noffset = fdt_first_subnode(fit, image_noffset);
 216              noffset >= 0;
 217              noffset = fdt_next_subnode(fit, noffset)) {
 218                 const char *name = fit_get_name(fit, noffset, NULL);
 219
 220                 if (!strncmp(name, FIT_SIG_NODENAME,
 221                              strlen(FIT_SIG_NODENAME))) {
 222                         ret = fit_image_check_sig(fit, noffset, data,
 223                                                         size, -1, &err_msg);
 224                         if (ret) {
 225                                 puts("- ");
 226                         } else {
 227                                 puts("+ ");
 228                                 verified = 1;
 229                                 break;
 230                         }
 231                 }
 232         }
 233
 234         if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) {
 235                 err_msg = "Corrupted or truncated tree";
 236                 goto error;
 237         }
 238
 239         return verified ? 0 : -EPERM;
 240
 241 error:
 242         printf(" error!\n%s for '%s' hash node in '%s' image node\n",
 243                err_msg, fit_get_name(fit, noffset, NULL),
 244                fit_get_name(fit, image_noffset, NULL));
 245         return -1;
 246 }
 247
 248 int fit_image_verify_required_sigs(const void *fit, int image_noffset,
 249                 const char *data, size_t size, const void *sig_blob,
 250                 int *no_sigsp)
 251 {
 252         int verify_count = 0;
 253         int noffset;
 254         int sig_node;
 255
 256         /* Work out what we need to verify */
 257         *no_sigsp = 1;
 258         sig_node = fdt_subnode_offset(sig_blob, 0, FIT_SIG_NODENAME);
 259         if (sig_node < 0) {
 260                 debug("%s: No signature node found: %s\n", __func__,
 261                       fdt_strerror(sig_node));
 262                 return 0;
 263         }
 264
 265         for (noffset = fdt_first_subnode(sig_blob, sig_node);
 266              noffset >= 0;
 267              noffset = fdt_next_subnode(sig_blob, noffset)) {
 268                 const char *required;
 269                 int ret;
 270
 271                 required = fdt_getprop(sig_blob, noffset, "required", NULL);
 272                 if (!required || strcmp(required, "image"))
 273                         continue;
 274                 ret = fit_image_verify_sig(fit, image_noffset, data, size,
 275                                         sig_blob, noffset);
 276                 if (ret) {
 277                         printf("Failed to verify required signature '%s'\n",
 278                                fit_get_name(sig_blob, noffset, NULL));
 279                         return ret;
 280                 }
 281                 verify_count++;
 282         }
 283
 284         if (verify_count)
 285                 *no_sigsp = 0;
 286
 287         return 0;
 288 }
 289
 290 int fit_config_check_sig(const void *fit, int noffset, int required_keynode,
 291                          char **err_msgp)
 292 {
 293         char * const exc_prop[] = {"data"};
 294         const char *prop, *end, *name;
 295         struct image_sign_info info;
 296         const uint32_t *strings;
 297         uint8_t *fit_value;
 298         int fit_value_len;
 299         int max_regions;
 300         int i, prop_len;
 301         char path[200];
 302         int count;
 303
 304         debug("%s: fdt=%p, conf='%s', sig='%s'\n", __func__, gd_fdt_blob(),
 305               fit_get_name(fit, noffset, NULL),
 306               fit_get_name(gd_fdt_blob(), required_keynode, NULL));
 307         *err_msgp = NULL;
 308         if (fit_image_setup_verify(&info, fit, noffset, required_keynode,
 309                                    err_msgp))
 310                 return -1;
 311
 312         if (fit_image_hash_get_value(fit, noffset, &fit_value,
 313                                      &fit_value_len)) {
 314                 *err_msgp = "Can't get hash value property";
 315                 return -1;
 316         }
 317
 318         /* Count the number of strings in the property */
 319         prop = fdt_getprop(fit, noffset, "hashed-nodes", &prop_len);
 320         end = prop ? prop + prop_len : prop;
 321         for (name = prop, count = 0; name < end; name++)
 322                 if (!*name)
 323                         count++;
 324         if (!count) {
 325                 *err_msgp = "Can't get hashed-nodes property";
 326                 return -1;
 327         }
 328
 329         /* Add a sanity check here since we are using the stack */
 330         if (count > IMAGE_MAX_HASHED_NODES) {
 331                 *err_msgp = "Number of hashed nodes exceeds maximum";
 332                 return -1;
 333         }
 334
 335         /* Create a list of node names from those strings */
 336         char *node_inc[count];
 337
 338         debug("Hash nodes (%d):\n", count);
 339         for (name = prop, i = 0; name < end; name += strlen(name) + 1, i++) {
 340                 debug("   '%s'\n", name);
 341                 node_inc[i] = (char *)name;
 342         }
 343
 344         /*
 345          * Each node can generate one region for each sub-node. Allow for
 346          * 7 sub-nodes (hash@1, signature@1, etc.) and some extra.
 347          */
 348         max_regions = 20 + count * 7;
 349         struct fdt_region fdt_regions[max_regions];
 350
 351         /* Get a list of regions to hash */
 352         count = fdt_find_regions(fit, node_inc, count,
 353                         exc_prop, ARRAY_SIZE(exc_prop),
 354                         fdt_regions, max_regions - 1,
 355                         path, sizeof(path), 0);
 356         if (count < 0) {
 357                 *err_msgp = "Failed to hash configuration";
 358                 return -1;
 359         }
 360         if (count == 0) {
 361                 *err_msgp = "No data to hash";
 362                 return -1;
 363         }
 364         if (count >= max_regions - 1) {
 365                 *err_msgp = "Too many hash regions";
 366                 return -1;
 367         }
 368
 369         /* Add the strings */
 370         strings = fdt_getprop(fit, noffset, "hashed-strings", NULL);
 371         if (strings) {
 372                 fdt_regions[count].offset = fdt_off_dt_strings(fit) +
 373                                 fdt32_to_cpu(strings[0]);
 374                 fdt_regions[count].size = fdt32_to_cpu(strings[1]);
 375                 count++;
 376         }
 377
 378         /* Allocate the region list on the stack */
 379         struct image_region region[count];
 380
 381         fit_region_make_list(fit, fdt_regions, count, region);
 382         if (info.algo->verify(&info, region, count, fit_value,
 383                               fit_value_len)) {
 384                 *err_msgp = "Verification failed";
 385                 return -1;
 386         }
 387
 388         return 0;
 389 }
 390
 391 static int fit_config_verify_sig(const void *fit, int conf_noffset,
 392                 const void *sig_blob, int sig_offset)
 393 {
 394         int noffset;
 395         char *err_msg = "";
 396         int verified = 0;
 397         int ret;
 398
 399         /* Process all hash subnodes of the component conf node */
 400         for (noffset = fdt_first_subnode(fit, conf_noffset);
 401              noffset >= 0;
 402              noffset = fdt_next_subnode(fit, noffset)) {
 403                 const char *name = fit_get_name(fit, noffset, NULL);
 404
 405                 if (!strncmp(name, FIT_SIG_NODENAME,
 406                              strlen(FIT_SIG_NODENAME))) {
 407                         ret = fit_config_check_sig(fit, noffset, sig_offset,
 408                                                    &err_msg);
 409                         if (ret) {
 410                                 puts("- ");
 411                         } else {
 412                                 puts("+ ");
 413                                 verified = 1;
 414                                 break;
 415                         }
 416                 }
 417         }
 418
 419         if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) {
 420                 err_msg = "Corrupted or truncated tree";
 421                 goto error;
 422         }
 423
 424         return verified ? 0 : -EPERM;
 425
 426 error:
 427         printf(" error!\n%s for '%s' hash node in '%s' config node\n",
 428                err_msg, fit_get_name(fit, noffset, NULL),
 429                fit_get_name(fit, conf_noffset, NULL));
 430         return -1;
 431 }
 432
 433 int fit_config_verify_required_sigs(const void *fit, int conf_noffset,
 434                 const void *sig_blob)
 435 {
 436         int noffset;
 437         int sig_node;
 438
 439         /* Work out what we need to verify */
 440         sig_node = fdt_subnode_offset(sig_blob, 0, FIT_SIG_NODENAME);
 441         if (sig_node < 0) {
 442                 debug("%s: No signature node found: %s\n", __func__,
 443                       fdt_strerror(sig_node));
 444                 return 0;
 445         }
 446
 447         for (noffset = fdt_first_subnode(sig_blob, sig_node);
 448              noffset >= 0;
 449              noffset = fdt_next_subnode(sig_blob, noffset)) {
 450                 const char *required;
 451                 int ret;
 452
 453                 required = fdt_getprop(sig_blob, noffset, "required", NULL);
 454                 if (!required || strcmp(required, "conf"))
 455                         continue;
 456                 ret = fit_config_verify_sig(fit, conf_noffset, sig_blob,
 457                                             noffset);
 458                 if (ret) {
 459                         printf("Failed to verify required signature '%s'\n",
 460                                fit_get_name(sig_blob, noffset, NULL));
 461                         return ret;
 462                 }
 463         }
 464
 465         return 0;
 466 }
 467
 468 int fit_config_verify(const void *fit, int conf_noffset)
 469 {
 470         return fit_config_verify_required_sigs(fit, conf_noffset,
 471                                                gd_fdt_blob());
 472 }