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#ifdef USE_HOSTCC 12#include "mkimage.h" 13#include <time.h> 14#include <u-boot/crc.h> 15#else 16#include <linux/compiler.h> 17#include <linux/kconfig.h> 18#include <common.h> 19#include <errno.h> 20#include <mapmem.h> 21#include <asm/io.h> 22#include <malloc.h> 23DECLARE_GLOBAL_DATA_PTR; 24#endif /* !USE_HOSTCC*/ 25 26#include <bootm.h> 27#include <image.h> 28#include <bootstage.h> 29#include <u-boot/crc.h> 30#include <u-boot/md5.h> 31#include <u-boot/sha1.h> 32#include <u-boot/sha256.h> 33 34/*****************************************************************************/ 35/* New uImage format routines */ 36/*****************************************************************************/ 37#ifndef USE_HOSTCC 38static int fit_parse_spec(const char *spec, char sepc, ulong addr_curr, 39 ulong *addr, const char **name) 40{ 41 const char *sep; 42 43 *addr = addr_curr; 44 *name = NULL; 45 46 sep = strchr(spec, sepc); 47 if (sep) { 48 if (sep - spec > 0) 49 *addr = simple_strtoul(spec, NULL, 16); 50 51 *name = sep + 1; 52 return 1; 53 } 54 55 return 0; 56} 57 58/** 59 * fit_parse_conf - parse FIT configuration spec 60 * @spec: input string, containing configuration spec 61 * @add_curr: current image address (to be used as a possible default) 62 * @addr: pointer to a ulong variable, will hold FIT image address of a given 63 * configuration 64 * @conf_name double pointer to a char, will hold pointer to a configuration 65 * unit name 66 * 67 * fit_parse_conf() expects configuration spec in the form of [<addr>]#<conf>, 68 * where <addr> is a FIT image address that contains configuration 69 * with a <conf> unit name. 70 * 71 * Address part is optional, and if omitted default add_curr will 72 * be used instead. 73 * 74 * returns: 75 * 1 if spec is a valid configuration string, 76 * addr and conf_name are set accordingly 77 * 0 otherwise 78 */ 79int fit_parse_conf(const char *spec, ulong addr_curr, 80 ulong *addr, const char **conf_name) 81{ 82 return fit_parse_spec(spec, '#', addr_curr, addr, conf_name); 83} 84 85/** 86 * fit_parse_subimage - parse FIT subimage spec 87 * @spec: input string, containing subimage spec 88 * @add_curr: current image address (to be used as a possible default) 89 * @addr: pointer to a ulong variable, will hold FIT image address of a given 90 * subimage 91 * @image_name: double pointer to a char, will hold pointer to a subimage name 92 * 93 * fit_parse_subimage() expects subimage spec in the form of 94 * [<addr>]:<subimage>, where <addr> is a FIT image address that contains 95 * subimage with a <subimg> unit name. 96 * 97 * Address part is optional, and if omitted default add_curr will 98 * be used instead. 99 * 100 * returns: 101 * 1 if spec is a valid subimage string, 102 * addr and image_name are set accordingly 103 * 0 otherwise 104 */ 105int fit_parse_subimage(const char *spec, ulong addr_curr, 106 ulong *addr, const char **image_name) 107{ 108 return fit_parse_spec(spec, ':', addr_curr, addr, image_name); 109} 110#endif /* !USE_HOSTCC */ 111 112static void fit_get_debug(const void *fit, int noffset, 113 char *prop_name, int err) 114{ 115 debug("Can't get '%s' property from FIT 0x%08lx, node: offset %d, name %s (%s)\n", 116 prop_name, (ulong)fit, noffset, fit_get_name(fit, noffset, NULL), 117 fdt_strerror(err)); 118} 119 120/** 121 * fit_get_subimage_count - get component (sub-image) count 122 * @fit: pointer to the FIT format image header 123 * @images_noffset: offset of images node 124 * 125 * returns: 126 * number of image components 127 */ 128int fit_get_subimage_count(const void *fit, int images_noffset) 129{ 130 int noffset; 131 int ndepth; 132 int count = 0; 133 134 /* Process its subnodes, print out component images details */ 135 for (ndepth = 0, count = 0, 136 noffset = fdt_next_node(fit, images_noffset, &ndepth); 137 (noffset >= 0) && (ndepth > 0); 138 noffset = fdt_next_node(fit, noffset, &ndepth)) { 139 if (ndepth == 1) { 140 count++; 141 } 142 } 143 144 return count; 145} 146 147#if !defined(CONFIG_SPL_BUILD) || defined(CONFIG_SPL_FIT_PRINT) 148/** 149 * fit_image_print_data() - prints out the hash node details 150 * @fit: pointer to the FIT format image header 151 * @noffset: offset of the hash node 152 * @p: pointer to prefix string 153 * @type: Type of information to print ("hash" or "sign") 154 * 155 * fit_image_print_data() lists properties for the processed hash node 156 * 157 * This function avoid using puts() since it prints a newline on the host 158 * but does not in U-Boot. 159 * 160 * returns: 161 * no returned results 162 */ 163static void fit_image_print_data(const void *fit, int noffset, const char *p, 164 const char *type) 165{ 166 const char *keyname; 167 uint8_t *value; 168 int value_len; 169 char *algo; 170 const char *padding; 171 int required; 172 int ret, i; 173 174 debug("%s %s node: '%s'\n", p, type, 175 fit_get_name(fit, noffset, NULL)); 176 printf("%s %s algo: ", p, type); 177 if (fit_image_hash_get_algo(fit, noffset, &algo)) { 178 printf("invalid/unsupported\n"); 179 return; 180 } 181 printf("%s", algo); 182 keyname = fdt_getprop(fit, noffset, "key-name-hint", NULL); 183 required = fdt_getprop(fit, noffset, "required", NULL) != NULL; 184 if (keyname) 185 printf(":%s", keyname); 186 if (required) 187 printf(" (required)"); 188 printf("\n"); 189 190 padding = fdt_getprop(fit, noffset, "padding", NULL); 191 if (padding) 192 printf("%s %s padding: %s\n", p, type, padding); 193 194 ret = fit_image_hash_get_value(fit, noffset, &value, 195 &value_len); 196 printf("%s %s value: ", p, type); 197 if (ret) { 198 printf("unavailable\n"); 199 } else { 200 for (i = 0; i < value_len; i++) 201 printf("%02x", value[i]); 202 printf("\n"); 203 } 204 205 debug("%s %s len: %d\n", p, type, value_len); 206 207 /* Signatures have a time stamp */ 208 if (IMAGE_ENABLE_TIMESTAMP && keyname) { 209 time_t timestamp; 210 211 printf("%s Timestamp: ", p); 212 if (fit_get_timestamp(fit, noffset, ×tamp)) 213 printf("unavailable\n"); 214 else 215 genimg_print_time(timestamp); 216 } 217} 218 219/** 220 * fit_image_print_verification_data() - prints out the hash/signature details 221 * @fit: pointer to the FIT format image header 222 * @noffset: offset of the hash or signature node 223 * @p: pointer to prefix string 224 * 225 * This lists properties for the processed hash node 226 * 227 * returns: 228 * no returned results 229 */ 230static void fit_image_print_verification_data(const void *fit, int noffset, 231 const char *p) 232{ 233 const char *name; 234 235 /* 236 * Check subnode name, must be equal to "hash" or "signature". 237 * Multiple hash/signature nodes require unique unit node 238 * names, e.g. hash-1, hash-2, signature-1, signature-2, etc. 239 */ 240 name = fit_get_name(fit, noffset, NULL); 241 if (!strncmp(name, FIT_HASH_NODENAME, strlen(FIT_HASH_NODENAME))) { 242 fit_image_print_data(fit, noffset, p, "Hash"); 243 } else if (!strncmp(name, FIT_SIG_NODENAME, 244 strlen(FIT_SIG_NODENAME))) { 245 fit_image_print_data(fit, noffset, p, "Sign"); 246 } 247} 248 249/** 250 * fit_conf_print - prints out the FIT configuration details 251 * @fit: pointer to the FIT format image header 252 * @noffset: offset of the configuration node 253 * @p: pointer to prefix string 254 * 255 * fit_conf_print() lists all mandatory properties for the processed 256 * configuration node. 257 * 258 * returns: 259 * no returned results 260 */ 261static void fit_conf_print(const void *fit, int noffset, const char *p) 262{ 263 char *desc; 264 const char *uname; 265 int ret; 266 int fdt_index, loadables_index; 267 int ndepth; 268 269 /* Mandatory properties */ 270 ret = fit_get_desc(fit, noffset, &desc); 271 printf("%s Description: ", p); 272 if (ret) 273 printf("unavailable\n"); 274 else 275 printf("%s\n", desc); 276 277 uname = fdt_getprop(fit, noffset, FIT_KERNEL_PROP, NULL); 278 printf("%s Kernel: ", p); 279 if (!uname) 280 printf("unavailable\n"); 281 else 282 printf("%s\n", uname); 283 284 /* Optional properties */ 285 uname = fdt_getprop(fit, noffset, FIT_RAMDISK_PROP, NULL); 286 if (uname) 287 printf("%s Init Ramdisk: %s\n", p, uname); 288 289 uname = fdt_getprop(fit, noffset, FIT_FIRMWARE_PROP, NULL); 290 if (uname) 291 printf("%s Firmware: %s\n", p, uname); 292 293 for (fdt_index = 0; 294 uname = fdt_stringlist_get(fit, noffset, FIT_FDT_PROP, 295 fdt_index, NULL), uname; 296 fdt_index++) { 297 if (fdt_index == 0) 298 printf("%s FDT: ", p); 299 else 300 printf("%s ", p); 301 printf("%s\n", uname); 302 } 303 304 uname = fdt_getprop(fit, noffset, FIT_FPGA_PROP, NULL); 305 if (uname) 306 printf("%s FPGA: %s\n", p, uname); 307 308 /* Print out all of the specified loadables */ 309 for (loadables_index = 0; 310 uname = fdt_stringlist_get(fit, noffset, FIT_LOADABLE_PROP, 311 loadables_index, NULL), uname; 312 loadables_index++) { 313 if (loadables_index == 0) { 314 printf("%s Loadables: ", p); 315 } else { 316 printf("%s ", p); 317 } 318 printf("%s\n", uname); 319 } 320 321 /* Process all hash subnodes of the component configuration node */ 322 for (ndepth = 0, noffset = fdt_next_node(fit, noffset, &ndepth); 323 (noffset >= 0) && (ndepth > 0); 324 noffset = fdt_next_node(fit, noffset, &ndepth)) { 325 if (ndepth == 1) { 326 /* Direct child node of the component configuration node */ 327 fit_image_print_verification_data(fit, noffset, p); 328 } 329 } 330} 331 332/** 333 * fit_print_contents - prints out the contents of the FIT format image 334 * @fit: pointer to the FIT format image header 335 * @p: pointer to prefix string 336 * 337 * fit_print_contents() formats a multi line FIT image contents description. 338 * The routine prints out FIT image properties (root node level) followed by 339 * the details of each component image. 340 * 341 * returns: 342 * no returned results 343 */ 344void fit_print_contents(const void *fit) 345{ 346 char *desc; 347 char *uname; 348 int images_noffset; 349 int confs_noffset; 350 int noffset; 351 int ndepth; 352 int count = 0; 353 int ret; 354 const char *p; 355 time_t timestamp; 356 357 /* Indent string is defined in header image.h */ 358 p = IMAGE_INDENT_STRING; 359 360 /* Root node properties */ 361 ret = fit_get_desc(fit, 0, &desc); 362 printf("%sFIT description: ", p); 363 if (ret) 364 printf("unavailable\n"); 365 else 366 printf("%s\n", desc); 367 368 if (IMAGE_ENABLE_TIMESTAMP) { 369 ret = fit_get_timestamp(fit, 0, ×tamp); 370 printf("%sCreated: ", p); 371 if (ret) 372 printf("unavailable\n"); 373 else 374 genimg_print_time(timestamp); 375 } 376 377 /* Find images parent node offset */ 378 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 379 if (images_noffset < 0) { 380 printf("Can't find images parent node '%s' (%s)\n", 381 FIT_IMAGES_PATH, fdt_strerror(images_noffset)); 382 return; 383 } 384 385 /* Process its subnodes, print out component images details */ 386 for (ndepth = 0, count = 0, 387 noffset = fdt_next_node(fit, images_noffset, &ndepth); 388 (noffset >= 0) && (ndepth > 0); 389 noffset = fdt_next_node(fit, noffset, &ndepth)) { 390 if (ndepth == 1) { 391 /* 392 * Direct child node of the images parent node, 393 * i.e. component image node. 394 */ 395 printf("%s Image %u (%s)\n", p, count++, 396 fit_get_name(fit, noffset, NULL)); 397 398 fit_image_print(fit, noffset, p); 399 } 400 } 401 402 /* Find configurations parent node offset */ 403 confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH); 404 if (confs_noffset < 0) { 405 debug("Can't get configurations parent node '%s' (%s)\n", 406 FIT_CONFS_PATH, fdt_strerror(confs_noffset)); 407 return; 408 } 409 410 /* get default configuration unit name from default property */ 411 uname = (char *)fdt_getprop(fit, noffset, FIT_DEFAULT_PROP, NULL); 412 if (uname) 413 printf("%s Default Configuration: '%s'\n", p, uname); 414 415 /* Process its subnodes, print out configurations details */ 416 for (ndepth = 0, count = 0, 417 noffset = fdt_next_node(fit, confs_noffset, &ndepth); 418 (noffset >= 0) && (ndepth > 0); 419 noffset = fdt_next_node(fit, noffset, &ndepth)) { 420 if (ndepth == 1) { 421 /* 422 * Direct child node of the configurations parent node, 423 * i.e. configuration node. 424 */ 425 printf("%s Configuration %u (%s)\n", p, count++, 426 fit_get_name(fit, noffset, NULL)); 427 428 fit_conf_print(fit, noffset, p); 429 } 430 } 431} 432 433/** 434 * fit_image_print - prints out the FIT component image details 435 * @fit: pointer to the FIT format image header 436 * @image_noffset: offset of the component image node 437 * @p: pointer to prefix string 438 * 439 * fit_image_print() lists all mandatory properties for the processed component 440 * image. If present, hash nodes are printed out as well. Load 441 * address for images of type firmware is also printed out. Since the load 442 * address is not mandatory for firmware images, it will be output as 443 * "unavailable" when not present. 444 * 445 * returns: 446 * no returned results 447 */ 448void fit_image_print(const void *fit, int image_noffset, const char *p) 449{ 450 char *desc; 451 uint8_t type, arch, os, comp; 452 size_t size; 453 ulong load, entry; 454 const void *data; 455 int noffset; 456 int ndepth; 457 int ret; 458 459 /* Mandatory properties */ 460 ret = fit_get_desc(fit, image_noffset, &desc); 461 printf("%s Description: ", p); 462 if (ret) 463 printf("unavailable\n"); 464 else 465 printf("%s\n", desc); 466 467 if (IMAGE_ENABLE_TIMESTAMP) { 468 time_t timestamp; 469 470 ret = fit_get_timestamp(fit, 0, ×tamp); 471 printf("%s Created: ", p); 472 if (ret) 473 printf("unavailable\n"); 474 else 475 genimg_print_time(timestamp); 476 } 477 478 fit_image_get_type(fit, image_noffset, &type); 479 printf("%s Type: %s\n", p, genimg_get_type_name(type)); 480 481 fit_image_get_comp(fit, image_noffset, &comp); 482 printf("%s Compression: %s\n", p, genimg_get_comp_name(comp)); 483 484 ret = fit_image_get_data_and_size(fit, image_noffset, &data, &size); 485 486#ifndef USE_HOSTCC 487 printf("%s Data Start: ", p); 488 if (ret) { 489 printf("unavailable\n"); 490 } else { 491 void *vdata = (void *)data; 492 493 printf("0x%08lx\n", (ulong)map_to_sysmem(vdata)); 494 } 495#endif 496 497 printf("%s Data Size: ", p); 498 if (ret) 499 printf("unavailable\n"); 500 else 501 genimg_print_size(size); 502 503 /* Remaining, type dependent properties */ 504 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) || 505 (type == IH_TYPE_RAMDISK) || (type == IH_TYPE_FIRMWARE) || 506 (type == IH_TYPE_FLATDT)) { 507 fit_image_get_arch(fit, image_noffset, &arch); 508 printf("%s Architecture: %s\n", p, genimg_get_arch_name(arch)); 509 } 510 511 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_RAMDISK) || 512 (type == IH_TYPE_FIRMWARE)) { 513 fit_image_get_os(fit, image_noffset, &os); 514 printf("%s OS: %s\n", p, genimg_get_os_name(os)); 515 } 516 517 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) || 518 (type == IH_TYPE_FIRMWARE) || (type == IH_TYPE_RAMDISK) || 519 (type == IH_TYPE_FPGA)) { 520 ret = fit_image_get_load(fit, image_noffset, &load); 521 printf("%s Load Address: ", p); 522 if (ret) 523 printf("unavailable\n"); 524 else 525 printf("0x%08lx\n", load); 526 } 527 528 /* optional load address for FDT */ 529 if (type == IH_TYPE_FLATDT && !fit_image_get_load(fit, image_noffset, &load)) 530 printf("%s Load Address: 0x%08lx\n", p, load); 531 532 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) || 533 (type == IH_TYPE_RAMDISK)) { 534 ret = fit_image_get_entry(fit, image_noffset, &entry); 535 printf("%s Entry Point: ", p); 536 if (ret) 537 printf("unavailable\n"); 538 else 539 printf("0x%08lx\n", entry); 540 } 541 542 /* Process all hash subnodes of the component image node */ 543 for (ndepth = 0, noffset = fdt_next_node(fit, image_noffset, &ndepth); 544 (noffset >= 0) && (ndepth > 0); 545 noffset = fdt_next_node(fit, noffset, &ndepth)) { 546 if (ndepth == 1) { 547 /* Direct child node of the component image node */ 548 fit_image_print_verification_data(fit, noffset, p); 549 } 550 } 551} 552#else 553void fit_print_contents(const void *fit) { } 554void fit_image_print(const void *fit, int image_noffset, const char *p) { } 555#endif /* !defined(CONFIG_SPL_BUILD) || defined(CONFIG_SPL_FIT_PRINT) */ 556 557/** 558 * fit_get_desc - get node description property 559 * @fit: pointer to the FIT format image header 560 * @noffset: node offset 561 * @desc: double pointer to the char, will hold pointer to the description 562 * 563 * fit_get_desc() reads description property from a given node, if 564 * description is found pointer to it is returned in third call argument. 565 * 566 * returns: 567 * 0, on success 568 * -1, on failure 569 */ 570int fit_get_desc(const void *fit, int noffset, char **desc) 571{ 572 int len; 573 574 *desc = (char *)fdt_getprop(fit, noffset, FIT_DESC_PROP, &len); 575 if (*desc == NULL) { 576 fit_get_debug(fit, noffset, FIT_DESC_PROP, len); 577 return -1; 578 } 579 580 return 0; 581} 582 583/** 584 * fit_get_timestamp - get node timestamp property 585 * @fit: pointer to the FIT format image header 586 * @noffset: node offset 587 * @timestamp: pointer to the time_t, will hold read timestamp 588 * 589 * fit_get_timestamp() reads timestamp property from given node, if timestamp 590 * is found and has a correct size its value is returned in third call 591 * argument. 592 * 593 * returns: 594 * 0, on success 595 * -1, on property read failure 596 * -2, on wrong timestamp size 597 */ 598int fit_get_timestamp(const void *fit, int noffset, time_t *timestamp) 599{ 600 int len; 601 const void *data; 602 603 data = fdt_getprop(fit, noffset, FIT_TIMESTAMP_PROP, &len); 604 if (data == NULL) { 605 fit_get_debug(fit, noffset, FIT_TIMESTAMP_PROP, len); 606 return -1; 607 } 608 if (len != sizeof(uint32_t)) { 609 debug("FIT timestamp with incorrect size of (%u)\n", len); 610 return -2; 611 } 612 613 *timestamp = uimage_to_cpu(*((uint32_t *)data)); 614 return 0; 615} 616 617/** 618 * fit_image_get_node - get node offset for component image of a given unit name 619 * @fit: pointer to the FIT format image header 620 * @image_uname: component image node unit name 621 * 622 * fit_image_get_node() finds a component image (within the '/images' 623 * node) of a provided unit name. If image is found its node offset is 624 * returned to the caller. 625 * 626 * returns: 627 * image node offset when found (>=0) 628 * negative number on failure (FDT_ERR_* code) 629 */ 630int fit_image_get_node(const void *fit, const char *image_uname) 631{ 632 int noffset, images_noffset; 633 634 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 635 if (images_noffset < 0) { 636 debug("Can't find images parent node '%s' (%s)\n", 637 FIT_IMAGES_PATH, fdt_strerror(images_noffset)); 638 return images_noffset; 639 } 640 641 noffset = fdt_subnode_offset(fit, images_noffset, image_uname); 642 if (noffset < 0) { 643 debug("Can't get node offset for image unit name: '%s' (%s)\n", 644 image_uname, fdt_strerror(noffset)); 645 } 646 647 return noffset; 648} 649 650/** 651 * fit_image_get_os - get os id for a given component image node 652 * @fit: pointer to the FIT format image header 653 * @noffset: component image node offset 654 * @os: pointer to the uint8_t, will hold os numeric id 655 * 656 * fit_image_get_os() finds os property in a given component image node. 657 * If the property is found, its (string) value is translated to the numeric 658 * id which is returned to the caller. 659 * 660 * returns: 661 * 0, on success 662 * -1, on failure 663 */ 664int fit_image_get_os(const void *fit, int noffset, uint8_t *os) 665{ 666 int len; 667 const void *data; 668 669 /* Get OS name from property data */ 670 data = fdt_getprop(fit, noffset, FIT_OS_PROP, &len); 671 if (data == NULL) { 672 fit_get_debug(fit, noffset, FIT_OS_PROP, len); 673 *os = -1; 674 return -1; 675 } 676 677 /* Translate OS name to id */ 678 *os = genimg_get_os_id(data); 679 return 0; 680} 681 682/** 683 * fit_image_get_arch - get arch id for a given component image node 684 * @fit: pointer to the FIT format image header 685 * @noffset: component image node offset 686 * @arch: pointer to the uint8_t, will hold arch numeric id 687 * 688 * fit_image_get_arch() finds arch property in a given component image node. 689 * If the property is found, its (string) value is translated to the numeric 690 * id which is returned to the caller. 691 * 692 * returns: 693 * 0, on success 694 * -1, on failure 695 */ 696int fit_image_get_arch(const void *fit, int noffset, uint8_t *arch) 697{ 698 int len; 699 const void *data; 700 701 /* Get architecture name from property data */ 702 data = fdt_getprop(fit, noffset, FIT_ARCH_PROP, &len); 703 if (data == NULL) { 704 fit_get_debug(fit, noffset, FIT_ARCH_PROP, len); 705 *arch = -1; 706 return -1; 707 } 708 709 /* Translate architecture name to id */ 710 *arch = genimg_get_arch_id(data); 711 return 0; 712} 713 714/** 715 * fit_image_get_type - get type id for a given component image node 716 * @fit: pointer to the FIT format image header 717 * @noffset: component image node offset 718 * @type: pointer to the uint8_t, will hold type numeric id 719 * 720 * fit_image_get_type() finds type property in a given component image node. 721 * If the property is found, its (string) value is translated to the numeric 722 * id which is returned to the caller. 723 * 724 * returns: 725 * 0, on success 726 * -1, on failure 727 */ 728int fit_image_get_type(const void *fit, int noffset, uint8_t *type) 729{ 730 int len; 731 const void *data; 732 733 /* Get image type name from property data */ 734 data = fdt_getprop(fit, noffset, FIT_TYPE_PROP, &len); 735 if (data == NULL) { 736 fit_get_debug(fit, noffset, FIT_TYPE_PROP, len); 737 *type = -1; 738 return -1; 739 } 740 741 /* Translate image type name to id */ 742 *type = genimg_get_type_id(data); 743 return 0; 744} 745 746/** 747 * fit_image_get_comp - get comp id for a given component image node 748 * @fit: pointer to the FIT format image header 749 * @noffset: component image node offset 750 * @comp: pointer to the uint8_t, will hold comp numeric id 751 * 752 * fit_image_get_comp() finds comp property in a given component image node. 753 * If the property is found, its (string) value is translated to the numeric 754 * id which is returned to the caller. 755 * 756 * returns: 757 * 0, on success 758 * -1, on failure 759 */ 760int fit_image_get_comp(const void *fit, int noffset, uint8_t *comp) 761{ 762 int len; 763 const void *data; 764 765 /* Get compression name from property data */ 766 data = fdt_getprop(fit, noffset, FIT_COMP_PROP, &len); 767 if (data == NULL) { 768 fit_get_debug(fit, noffset, FIT_COMP_PROP, len); 769 *comp = -1; 770 return -1; 771 } 772 773 /* Translate compression name to id */ 774 *comp = genimg_get_comp_id(data); 775 return 0; 776} 777 778static int fit_image_get_address(const void *fit, int noffset, char *name, 779 ulong *load) 780{ 781 int len, cell_len; 782 const fdt32_t *cell; 783 uint64_t load64 = 0; 784 785 cell = fdt_getprop(fit, noffset, name, &len); 786 if (cell == NULL) { 787 fit_get_debug(fit, noffset, name, len); 788 return -1; 789 } 790 791 if (len > sizeof(ulong)) { 792 printf("Unsupported %s address size\n", name); 793 return -1; 794 } 795 796 cell_len = len >> 2; 797 /* Use load64 to avoid compiling warning for 32-bit target */ 798 while (cell_len--) { 799 load64 = (load64 << 32) | uimage_to_cpu(*cell); 800 cell++; 801 } 802 *load = (ulong)load64; 803 804 return 0; 805} 806/** 807 * fit_image_get_load() - get load addr property for given component image node 808 * @fit: pointer to the FIT format image header 809 * @noffset: component image node offset 810 * @load: pointer to the uint32_t, will hold load address 811 * 812 * fit_image_get_load() finds load address property in a given component 813 * image node. If the property is found, its value is returned to the caller. 814 * 815 * returns: 816 * 0, on success 817 * -1, on failure 818 */ 819int fit_image_get_load(const void *fit, int noffset, ulong *load) 820{ 821 return fit_image_get_address(fit, noffset, FIT_LOAD_PROP, load); 822} 823 824/** 825 * fit_image_get_entry() - get entry point address property 826 * @fit: pointer to the FIT format image header 827 * @noffset: component image node offset 828 * @entry: pointer to the uint32_t, will hold entry point address 829 * 830 * This gets the entry point address property for a given component image 831 * node. 832 * 833 * fit_image_get_entry() finds entry point address property in a given 834 * component image node. If the property is found, its value is returned 835 * to the caller. 836 * 837 * returns: 838 * 0, on success 839 * -1, on failure 840 */ 841int fit_image_get_entry(const void *fit, int noffset, ulong *entry) 842{ 843 return fit_image_get_address(fit, noffset, FIT_ENTRY_PROP, entry); 844} 845 846/** 847 * fit_image_get_data - get data property and its size for a given component image node 848 * @fit: pointer to the FIT format image header 849 * @noffset: component image node offset 850 * @data: double pointer to void, will hold data property's data address 851 * @size: pointer to size_t, will hold data property's data size 852 * 853 * fit_image_get_data() finds data property in a given component image node. 854 * If the property is found its data start address and size are returned to 855 * the caller. 856 * 857 * returns: 858 * 0, on success 859 * -1, on failure 860 */ 861int fit_image_get_data(const void *fit, int noffset, 862 const void **data, size_t *size) 863{ 864 int len; 865 866 *data = fdt_getprop(fit, noffset, FIT_DATA_PROP, &len); 867 if (*data == NULL) { 868 fit_get_debug(fit, noffset, FIT_DATA_PROP, len); 869 *size = 0; 870 return -1; 871 } 872 873 *size = len; 874 return 0; 875} 876 877/** 878 * Get 'data-offset' property from a given image node. 879 * 880 * @fit: pointer to the FIT image header 881 * @noffset: component image node offset 882 * @data_offset: holds the data-offset property 883 * 884 * returns: 885 * 0, on success 886 * -ENOENT if the property could not be found 887 */ 888int fit_image_get_data_offset(const void *fit, int noffset, int *data_offset) 889{ 890 const fdt32_t *val; 891 892 val = fdt_getprop(fit, noffset, FIT_DATA_OFFSET_PROP, NULL); 893 if (!val) 894 return -ENOENT; 895 896 *data_offset = fdt32_to_cpu(*val); 897 898 return 0; 899} 900 901/** 902 * Get 'data-position' property from a given image node. 903 * 904 * @fit: pointer to the FIT image header 905 * @noffset: component image node offset 906 * @data_position: holds the data-position property 907 * 908 * returns: 909 * 0, on success 910 * -ENOENT if the property could not be found 911 */ 912int fit_image_get_data_position(const void *fit, int noffset, 913 int *data_position) 914{ 915 const fdt32_t *val; 916 917 val = fdt_getprop(fit, noffset, FIT_DATA_POSITION_PROP, NULL); 918 if (!val) 919 return -ENOENT; 920 921 *data_position = fdt32_to_cpu(*val); 922 923 return 0; 924} 925 926/** 927 * Get 'data-size' property from a given image node. 928 * 929 * @fit: pointer to the FIT image header 930 * @noffset: component image node offset 931 * @data_size: holds the data-size property 932 * 933 * returns: 934 * 0, on success 935 * -ENOENT if the property could not be found 936 */ 937int fit_image_get_data_size(const void *fit, int noffset, int *data_size) 938{ 939 const fdt32_t *val; 940 941 val = fdt_getprop(fit, noffset, FIT_DATA_SIZE_PROP, NULL); 942 if (!val) 943 return -ENOENT; 944 945 *data_size = fdt32_to_cpu(*val); 946 947 return 0; 948} 949 950/** 951 * fit_image_get_data_and_size - get data and its size including 952 * both embedded and external data 953 * @fit: pointer to the FIT format image header 954 * @noffset: component image node offset 955 * @data: double pointer to void, will hold data property's data address 956 * @size: pointer to size_t, will hold data property's data size 957 * 958 * fit_image_get_data_and_size() finds data and its size including 959 * both embedded and external data. If the property is found 960 * its data start address and size are returned to the caller. 961 * 962 * returns: 963 * 0, on success 964 * otherwise, on failure 965 */ 966int fit_image_get_data_and_size(const void *fit, int noffset, 967 const void **data, size_t *size) 968{ 969 bool external_data = false; 970 int offset; 971 int len; 972 int ret; 973 974 if (!fit_image_get_data_position(fit, noffset, &offset)) { 975 external_data = true; 976 } else if (!fit_image_get_data_offset(fit, noffset, &offset)) { 977 external_data = true; 978 /* 979 * For FIT with external data, figure out where 980 * the external images start. This is the base 981 * for the data-offset properties in each image. 982 */ 983 offset += ((fdt_totalsize(fit) + 3) & ~3); 984 } 985 986 if (external_data) { 987 debug("External Data\n"); 988 ret = fit_image_get_data_size(fit, noffset, &len); 989 *data = fit + offset; 990 *size = len; 991 } else { 992 ret = fit_image_get_data(fit, noffset, data, size); 993 } 994 995 return ret; 996} 997 998/** 999 * fit_image_hash_get_algo - get hash algorithm name 1000 * @fit: pointer to the FIT format image header 1001 * @noffset: hash node offset 1002 * @algo: double pointer to char, will hold pointer to the algorithm name 1003 * 1004 * fit_image_hash_get_algo() finds hash algorithm property in a given hash node. 1005 * If the property is found its data start address is returned to the caller. 1006 * 1007 * returns: 1008 * 0, on success 1009 * -1, on failure 1010 */ 1011int fit_image_hash_get_algo(const void *fit, int noffset, char **algo) 1012{ 1013 int len; 1014 1015 *algo = (char *)fdt_getprop(fit, noffset, FIT_ALGO_PROP, &len); 1016 if (*algo == NULL) { 1017 fit_get_debug(fit, noffset, FIT_ALGO_PROP, len); 1018 return -1; 1019 } 1020 1021 return 0; 1022} 1023 1024/** 1025 * fit_image_hash_get_value - get hash value and length 1026 * @fit: pointer to the FIT format image header 1027 * @noffset: hash node offset 1028 * @value: double pointer to uint8_t, will hold address of a hash value data 1029 * @value_len: pointer to an int, will hold hash data length 1030 * 1031 * fit_image_hash_get_value() finds hash value property in a given hash node. 1032 * If the property is found its data start address and size are returned to 1033 * the caller. 1034 * 1035 * returns: 1036 * 0, on success 1037 * -1, on failure 1038 */ 1039int fit_image_hash_get_value(const void *fit, int noffset, uint8_t **value, 1040 int *value_len) 1041{ 1042 int len; 1043 1044 *value = (uint8_t *)fdt_getprop(fit, noffset, FIT_VALUE_PROP, &len); 1045 if (*value == NULL) { 1046 fit_get_debug(fit, noffset, FIT_VALUE_PROP, len); 1047 *value_len = 0; 1048 return -1; 1049 } 1050 1051 *value_len = len; 1052 return 0; 1053} 1054 1055/** 1056 * fit_image_hash_get_ignore - get hash ignore flag 1057 * @fit: pointer to the FIT format image header 1058 * @noffset: hash node offset 1059 * @ignore: pointer to an int, will hold hash ignore flag 1060 * 1061 * fit_image_hash_get_ignore() finds hash ignore property in a given hash node. 1062 * If the property is found and non-zero, the hash algorithm is not verified by 1063 * u-boot automatically. 1064 * 1065 * returns: 1066 * 0, on ignore not found 1067 * value, on ignore found 1068 */ 1069static int fit_image_hash_get_ignore(const void *fit, int noffset, int *ignore) 1070{ 1071 int len; 1072 int *value; 1073 1074 value = (int *)fdt_getprop(fit, noffset, FIT_IGNORE_PROP, &len); 1075 if (value == NULL || len != sizeof(int)) 1076 *ignore = 0; 1077 else 1078 *ignore = *value; 1079 1080 return 0; 1081} 1082 1083/** 1084 * fit_image_cipher_get_algo - get cipher algorithm name 1085 * @fit: pointer to the FIT format image header 1086 * @noffset: cipher node offset 1087 * @algo: double pointer to char, will hold pointer to the algorithm name 1088 * 1089 * fit_image_cipher_get_algo() finds cipher algorithm property in a given 1090 * cipher node. If the property is found its data start address is returned 1091 * to the caller. 1092 * 1093 * returns: 1094 * 0, on success 1095 * -1, on failure 1096 */ 1097int fit_image_cipher_get_algo(const void *fit, int noffset, char **algo) 1098{ 1099 int len; 1100 1101 *algo = (char *)fdt_getprop(fit, noffset, FIT_ALGO_PROP, &len); 1102 if (!*algo) { 1103 fit_get_debug(fit, noffset, FIT_ALGO_PROP, len); 1104 return -1; 1105 } 1106 1107 return 0; 1108} 1109 1110ulong fit_get_end(const void *fit) 1111{ 1112 return map_to_sysmem((void *)(fit + fdt_totalsize(fit))); 1113} 1114 1115/** 1116 * fit_set_timestamp - set node timestamp property 1117 * @fit: pointer to the FIT format image header 1118 * @noffset: node offset 1119 * @timestamp: timestamp value to be set 1120 * 1121 * fit_set_timestamp() attempts to set timestamp property in the requested 1122 * node and returns operation status to the caller. 1123 * 1124 * returns: 1125 * 0, on success 1126 * -ENOSPC if no space in device tree, -1 for other error 1127 */ 1128int fit_set_timestamp(void *fit, int noffset, time_t timestamp) 1129{ 1130 uint32_t t; 1131 int ret; 1132 1133 t = cpu_to_uimage(timestamp); 1134 ret = fdt_setprop(fit, noffset, FIT_TIMESTAMP_PROP, &t, 1135 sizeof(uint32_t)); 1136 if (ret) { 1137 debug("Can't set '%s' property for '%s' node (%s)\n", 1138 FIT_TIMESTAMP_PROP, fit_get_name(fit, noffset, NULL), 1139 fdt_strerror(ret)); 1140 return ret == -FDT_ERR_NOSPACE ? -ENOSPC : -1; 1141 } 1142 1143 return 0; 1144} 1145 1146/** 1147 * calculate_hash - calculate and return hash for provided input data 1148 * @data: pointer to the input data 1149 * @data_len: data length 1150 * @algo: requested hash algorithm 1151 * @value: pointer to the char, will hold hash value data (caller must 1152 * allocate enough free space) 1153 * value_len: length of the calculated hash 1154 * 1155 * calculate_hash() computes input data hash according to the requested 1156 * algorithm. 1157 * Resulting hash value is placed in caller provided 'value' buffer, length 1158 * of the calculated hash is returned via value_len pointer argument. 1159 * 1160 * returns: 1161 * 0, on success 1162 * -1, when algo is unsupported 1163 */ 1164int calculate_hash(const void *data, int data_len, const char *algo, 1165 uint8_t *value, int *value_len) 1166{ 1167 if (IMAGE_ENABLE_CRC32 && strcmp(algo, "crc32") == 0) { 1168 *((uint32_t *)value) = crc32_wd(0, data, data_len, 1169 CHUNKSZ_CRC32); 1170 *((uint32_t *)value) = cpu_to_uimage(*((uint32_t *)value)); 1171 *value_len = 4; 1172 } else if (IMAGE_ENABLE_SHA1 && strcmp(algo, "sha1") == 0) { 1173#ifdef CONFIG_SHA1 1174 sha1_csum_wd((unsigned char *)data, data_len, 1175 (unsigned char *)value, CHUNKSZ_SHA1); 1176 *value_len = 20; 1177#else 1178 debug("Unsupported hash sha1 alogrithm\n"); 1179 return -1; 1180#endif 1181 } else if (IMAGE_ENABLE_SHA256 && strcmp(algo, "sha256") == 0) { 1182#ifdef CONFIG_SHA256 1183 sha256_csum_wd((unsigned char *)data, data_len, 1184 (unsigned char *)value, CHUNKSZ_SHA256); 1185 *value_len = SHA256_SUM_LEN; 1186#else 1187 debug("Unsupported hash sha256 alogrithm\n"); 1188 return -1; 1189#endif 1190 } else if (IMAGE_ENABLE_MD5 && strcmp(algo, "md5") == 0) { 1191#ifdef CONFIG_MD5 1192 md5_wd((unsigned char *)data, data_len, value, CHUNKSZ_MD5); 1193 *value_len = 16; 1194#else 1195 debug("Unsupported hash mad5 alogrithm\n"); 1196 return -1; 1197#endif 1198 } else { 1199 debug("Unsupported hash alogrithm\n"); 1200 return -1; 1201 } 1202 return 0; 1203} 1204 1205static int fit_image_check_hash(const void *fit, int noffset, const void *data, 1206 size_t size, char **err_msgp) 1207{ 1208 uint8_t value[FIT_MAX_HASH_LEN]; 1209 int value_len; 1210 char *algo; 1211 uint8_t *fit_value; 1212 int fit_value_len; 1213 int ignore; 1214 1215 *err_msgp = NULL; 1216 1217 if (fit_image_hash_get_algo(fit, noffset, &algo)) { 1218 *err_msgp = "Can't get hash algo property"; 1219 return -1; 1220 } 1221 printf("%s", algo); 1222 1223 if (IMAGE_ENABLE_IGNORE) { 1224 fit_image_hash_get_ignore(fit, noffset, &ignore); 1225 if (ignore) { 1226 printf("-skipped "); 1227 return 0; 1228 } 1229 } 1230 1231 if (fit_image_hash_get_value(fit, noffset, &fit_value, 1232 &fit_value_len)) { 1233 *err_msgp = "Can't get hash value property"; 1234 return -1; 1235 } 1236 1237 if (calculate_hash(data, size, algo, value, &value_len)) { 1238 *err_msgp = "Unsupported hash algorithm"; 1239 return -1; 1240 } 1241 1242 if (value_len != fit_value_len) { 1243 *err_msgp = "Bad hash value len"; 1244 return -1; 1245 } else if (memcmp(value, fit_value, value_len) != 0) { 1246 *err_msgp = "Bad hash value"; 1247 return -1; 1248 } 1249 1250 return 0; 1251} 1252 1253int fit_image_verify_with_data(const void *fit, int image_noffset, 1254 const void *data, size_t size) 1255{ 1256 int noffset = 0; 1257 char *err_msg = ""; 1258 int verify_all = 1; 1259 int ret; 1260 1261 /* Verify all required signatures */ 1262 if (IMAGE_ENABLE_VERIFY && 1263 fit_image_verify_required_sigs(fit, image_noffset, data, size, 1264 gd_fdt_blob(), &verify_all)) { 1265 err_msg = "Unable to verify required signature"; 1266 goto error; 1267 } 1268 1269 /* Process all hash subnodes of the component image node */ 1270 fdt_for_each_subnode(noffset, fit, image_noffset) { 1271 const char *name = fit_get_name(fit, noffset, NULL); 1272 1273 /* 1274 * Check subnode name, must be equal to "hash". 1275 * Multiple hash nodes require unique unit node 1276 * names, e.g. hash-1, hash-2, etc. 1277 */ 1278 if (!strncmp(name, FIT_HASH_NODENAME, 1279 strlen(FIT_HASH_NODENAME))) { 1280 if (fit_image_check_hash(fit, noffset, data, size, 1281 &err_msg)) 1282 goto error; 1283 puts("+ "); 1284 } else if (IMAGE_ENABLE_VERIFY && verify_all && 1285 !strncmp(name, FIT_SIG_NODENAME, 1286 strlen(FIT_SIG_NODENAME))) { 1287 ret = fit_image_check_sig(fit, noffset, data, 1288 size, -1, &err_msg); 1289 1290 /* 1291 * Show an indication on failure, but do not return 1292 * an error. Only keys marked 'required' can cause 1293 * an image validation failure. See the call to 1294 * fit_image_verify_required_sigs() above. 1295 */ 1296 if (ret) 1297 puts("- "); 1298 else 1299 puts("+ "); 1300 } 1301 } 1302 1303 if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) { 1304 err_msg = "Corrupted or truncated tree"; 1305 goto error; 1306 } 1307 1308 return 1; 1309 1310error: 1311 printf(" error!\n%s for '%s' hash node in '%s' image node\n", 1312 err_msg, fit_get_name(fit, noffset, NULL), 1313 fit_get_name(fit, image_noffset, NULL)); 1314 return 0; 1315} 1316 1317/** 1318 * fit_image_verify - verify data integrity 1319 * @fit: pointer to the FIT format image header 1320 * @image_noffset: component image node offset 1321 * 1322 * fit_image_verify() goes over component image hash nodes, 1323 * re-calculates each data hash and compares with the value stored in hash 1324 * node. 1325 * 1326 * returns: 1327 * 1, if all hashes are valid 1328 * 0, otherwise (or on error) 1329 */ 1330int fit_image_verify(const void *fit, int image_noffset) 1331{ 1332 const void *data; 1333 size_t size; 1334 int noffset = 0; 1335 char *err_msg = ""; 1336 1337 /* Get image data and data length */ 1338 if (fit_image_get_data_and_size(fit, image_noffset, &data, &size)) { 1339 err_msg = "Can't get image data/size"; 1340 printf("error!\n%s for '%s' hash node in '%s' image node\n", 1341 err_msg, fit_get_name(fit, noffset, NULL), 1342 fit_get_name(fit, image_noffset, NULL)); 1343 return 0; 1344 } 1345 1346 return fit_image_verify_with_data(fit, image_noffset, data, size); 1347} 1348 1349/** 1350 * fit_all_image_verify - verify data integrity for all images 1351 * @fit: pointer to the FIT format image header 1352 * 1353 * fit_all_image_verify() goes over all images in the FIT and 1354 * for every images checks if all it's hashes are valid. 1355 * 1356 * returns: 1357 * 1, if all hashes of all images are valid 1358 * 0, otherwise (or on error) 1359 */ 1360int fit_all_image_verify(const void *fit) 1361{ 1362 int images_noffset; 1363 int noffset; 1364 int ndepth; 1365 int count; 1366 1367 /* Find images parent node offset */ 1368 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 1369 if (images_noffset < 0) { 1370 printf("Can't find images parent node '%s' (%s)\n", 1371 FIT_IMAGES_PATH, fdt_strerror(images_noffset)); 1372 return 0; 1373 } 1374 1375 /* Process all image subnodes, check hashes for each */ 1376 printf("## Checking hash(es) for FIT Image at %08lx ...\n", 1377 (ulong)fit); 1378 for (ndepth = 0, count = 0, 1379 noffset = fdt_next_node(fit, images_noffset, &ndepth); 1380 (noffset >= 0) && (ndepth > 0); 1381 noffset = fdt_next_node(fit, noffset, &ndepth)) { 1382 if (ndepth == 1) { 1383 /* 1384 * Direct child node of the images parent node, 1385 * i.e. component image node. 1386 */ 1387 printf(" Hash(es) for Image %u (%s): ", count, 1388 fit_get_name(fit, noffset, NULL)); 1389 count++; 1390 1391 if (!fit_image_verify(fit, noffset)) 1392 return 0; 1393 printf("\n"); 1394 } 1395 } 1396 return 1; 1397} 1398 1399/** 1400 * fit_image_check_os - check whether image node is of a given os type 1401 * @fit: pointer to the FIT format image header 1402 * @noffset: component image node offset 1403 * @os: requested image os 1404 * 1405 * fit_image_check_os() reads image os property and compares its numeric 1406 * id with the requested os. Comparison result is returned to the caller. 1407 * 1408 * returns: 1409 * 1 if image is of given os type 1410 * 0 otherwise (or on error) 1411 */ 1412int fit_image_check_os(const void *fit, int noffset, uint8_t os) 1413{ 1414 uint8_t image_os; 1415 1416 if (fit_image_get_os(fit, noffset, &image_os)) 1417 return 0; 1418 return (os == image_os); 1419} 1420 1421/** 1422 * fit_image_check_arch - check whether image node is of a given arch 1423 * @fit: pointer to the FIT format image header 1424 * @noffset: component image node offset 1425 * @arch: requested imagearch 1426 * 1427 * fit_image_check_arch() reads image arch property and compares its numeric 1428 * id with the requested arch. Comparison result is returned to the caller. 1429 * 1430 * returns: 1431 * 1 if image is of given arch 1432 * 0 otherwise (or on error) 1433 */ 1434int fit_image_check_arch(const void *fit, int noffset, uint8_t arch) 1435{ 1436 uint8_t image_arch; 1437 int aarch32_support = 0; 1438 1439#ifdef CONFIG_ARM64_SUPPORT_AARCH32 1440 aarch32_support = 1; 1441#endif 1442 1443 if (fit_image_get_arch(fit, noffset, &image_arch)) 1444 return 0; 1445 return (arch == image_arch) || 1446 (arch == IH_ARCH_I386 && image_arch == IH_ARCH_X86_64) || 1447 (arch == IH_ARCH_ARM64 && image_arch == IH_ARCH_ARM && 1448 aarch32_support); 1449} 1450 1451/** 1452 * fit_image_check_type - check whether image node is of a given type 1453 * @fit: pointer to the FIT format image header 1454 * @noffset: component image node offset 1455 * @type: requested image type 1456 * 1457 * fit_image_check_type() reads image type property and compares its numeric 1458 * id with the requested type. Comparison result is returned to the caller. 1459 * 1460 * returns: 1461 * 1 if image is of given type 1462 * 0 otherwise (or on error) 1463 */ 1464int fit_image_check_type(const void *fit, int noffset, uint8_t type) 1465{ 1466 uint8_t image_type; 1467 1468 if (fit_image_get_type(fit, noffset, &image_type)) 1469 return 0; 1470 return (type == image_type); 1471} 1472 1473/** 1474 * fit_image_check_comp - check whether image node uses given compression 1475 * @fit: pointer to the FIT format image header 1476 * @noffset: component image node offset 1477 * @comp: requested image compression type 1478 * 1479 * fit_image_check_comp() reads image compression property and compares its 1480 * numeric id with the requested compression type. Comparison result is 1481 * returned to the caller. 1482 * 1483 * returns: 1484 * 1 if image uses requested compression 1485 * 0 otherwise (or on error) 1486 */ 1487int fit_image_check_comp(const void *fit, int noffset, uint8_t comp) 1488{ 1489 uint8_t image_comp; 1490 1491 if (fit_image_get_comp(fit, noffset, &image_comp)) 1492 return 0; 1493 return (comp == image_comp); 1494} 1495 1496/** 1497 * fit_check_format - sanity check FIT image format 1498 * @fit: pointer to the FIT format image header 1499 * 1500 * fit_check_format() runs a basic sanity FIT image verification. 1501 * Routine checks for mandatory properties, nodes, etc. 1502 * 1503 * returns: 1504 * 1, on success 1505 * 0, on failure 1506 */ 1507int fit_check_format(const void *fit) 1508{ 1509 /* mandatory / node 'description' property */ 1510 if (fdt_getprop(fit, 0, FIT_DESC_PROP, NULL) == NULL) { 1511 debug("Wrong FIT format: no description\n"); 1512 return 0; 1513 } 1514 1515 if (IMAGE_ENABLE_TIMESTAMP) { 1516 /* mandatory / node 'timestamp' property */ 1517 if (fdt_getprop(fit, 0, FIT_TIMESTAMP_PROP, NULL) == NULL) { 1518 debug("Wrong FIT format: no timestamp\n"); 1519 return 0; 1520 } 1521 } 1522 1523 /* mandatory subimages parent '/images' node */ 1524 if (fdt_path_offset(fit, FIT_IMAGES_PATH) < 0) { 1525 debug("Wrong FIT format: no images parent node\n"); 1526 return 0; 1527 } 1528 1529 return 1; 1530} 1531 1532 1533/** 1534 * fit_conf_find_compat 1535 * @fit: pointer to the FIT format image header 1536 * @fdt: pointer to the device tree to compare against 1537 * 1538 * fit_conf_find_compat() attempts to find the configuration whose fdt is the 1539 * most compatible with the passed in device tree. 1540 * 1541 * Example: 1542 * 1543 * / o image-tree 1544 * |-o images 1545 * | |-o fdt-1 1546 * | |-o fdt-2 1547 * | 1548 * |-o configurations 1549 * |-o config-1 1550 * | |-fdt = fdt-1 1551 * | 1552 * |-o config-2 1553 * |-fdt = fdt-2 1554 * 1555 * / o U-Boot fdt 1556 * |-compatible = "foo,bar", "bim,bam" 1557 * 1558 * / o kernel fdt1 1559 * |-compatible = "foo,bar", 1560 * 1561 * / o kernel fdt2 1562 * |-compatible = "bim,bam", "baz,biz" 1563 * 1564 * Configuration 1 would be picked because the first string in U-Boot's 1565 * compatible list, "foo,bar", matches a compatible string in the root of fdt1. 1566 * "bim,bam" in fdt2 matches the second string which isn't as good as fdt1. 1567 * 1568 * As an optimization, the compatible property from the FDT's root node can be 1569 * copied into the configuration node in the FIT image. This is required to 1570 * match configurations with compressed FDTs. 1571 * 1572 * returns: 1573 * offset to the configuration to use if one was found 1574 * -1 otherwise 1575 */ 1576int fit_conf_find_compat(const void *fit, const void *fdt) 1577{ 1578 int ndepth = 0; 1579 int noffset, confs_noffset, images_noffset; 1580 const void *fdt_compat; 1581 int fdt_compat_len; 1582 int best_match_offset = 0; 1583 int best_match_pos = 0; 1584 1585 confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH); 1586 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 1587 if (confs_noffset < 0 || images_noffset < 0) { 1588 debug("Can't find configurations or images nodes.\n"); 1589 return -1; 1590 } 1591 1592 fdt_compat = fdt_getprop(fdt, 0, "compatible", &fdt_compat_len); 1593 if (!fdt_compat) { 1594 debug("Fdt for comparison has no \"compatible\" property.\n"); 1595 return -1; 1596 } 1597 1598 /* 1599 * Loop over the configurations in the FIT image. 1600 */ 1601 for (noffset = fdt_next_node(fit, confs_noffset, &ndepth); 1602 (noffset >= 0) && (ndepth > 0); 1603 noffset = fdt_next_node(fit, noffset, &ndepth)) { 1604 const void *fdt; 1605 const char *kfdt_name; 1606 int kfdt_noffset, compat_noffset; 1607 const char *cur_fdt_compat; 1608 int len; 1609 size_t sz; 1610 int i; 1611 1612 if (ndepth > 1) 1613 continue; 1614 1615 /* If there's a compat property in the config node, use that. */ 1616 if (fdt_getprop(fit, noffset, "compatible", NULL)) { 1617 fdt = fit; /* search in FIT image */ 1618 compat_noffset = noffset; /* search under config node */ 1619 } else { /* Otherwise extract it from the kernel FDT. */ 1620 kfdt_name = fdt_getprop(fit, noffset, "fdt", &len); 1621 if (!kfdt_name) { 1622 debug("No fdt property found.\n"); 1623 continue; 1624 } 1625 kfdt_noffset = fdt_subnode_offset(fit, images_noffset, 1626 kfdt_name); 1627 if (kfdt_noffset < 0) { 1628 debug("No image node named \"%s\" found.\n", 1629 kfdt_name); 1630 continue; 1631 } 1632 1633 if (!fit_image_check_comp(fit, kfdt_noffset, 1634 IH_COMP_NONE)) { 1635 debug("Can't extract compat from \"%s\" " 1636 "(compressed)\n", kfdt_name); 1637 continue; 1638 } 1639 1640 /* search in this config's kernel FDT */ 1641 if (fit_image_get_data(fit, kfdt_noffset, &fdt, &sz)) { 1642 debug("Failed to get fdt \"%s\".\n", kfdt_name); 1643 continue; 1644 } 1645 1646 compat_noffset = 0; /* search kFDT under root node */ 1647 } 1648 1649 len = fdt_compat_len; 1650 cur_fdt_compat = fdt_compat; 1651 /* 1652 * Look for a match for each U-Boot compatibility string in 1653 * turn in the compat string property. 1654 */ 1655 for (i = 0; len > 0 && 1656 (!best_match_offset || best_match_pos > i); i++) { 1657 int cur_len = strlen(cur_fdt_compat) + 1; 1658 1659 if (!fdt_node_check_compatible(fdt, compat_noffset, 1660 cur_fdt_compat)) { 1661 best_match_offset = noffset; 1662 best_match_pos = i; 1663 break; 1664 } 1665 len -= cur_len; 1666 cur_fdt_compat += cur_len; 1667 } 1668 } 1669 if (!best_match_offset) { 1670 debug("No match found.\n"); 1671 return -1; 1672 } 1673 1674 return best_match_offset; 1675} 1676 1677int fit_conf_get_node(const void *fit, const char *conf_uname) 1678{ 1679 int noffset, confs_noffset; 1680 int len; 1681 const char *s; 1682 char *conf_uname_copy = NULL; 1683 1684 confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH); 1685 if (confs_noffset < 0) { 1686 debug("Can't find configurations parent node '%s' (%s)\n", 1687 FIT_CONFS_PATH, fdt_strerror(confs_noffset)); 1688 return confs_noffset; 1689 } 1690 1691 if (conf_uname == NULL) { 1692 /* get configuration unit name from the default property */ 1693 debug("No configuration specified, trying default...\n"); 1694 conf_uname = (char *)fdt_getprop(fit, confs_noffset, 1695 FIT_DEFAULT_PROP, &len); 1696 if (conf_uname == NULL) { 1697 fit_get_debug(fit, confs_noffset, FIT_DEFAULT_PROP, 1698 len); 1699 return len; 1700 } 1701 debug("Found default configuration: '%s'\n", conf_uname); 1702 } 1703 1704 s = strchr(conf_uname, '#'); 1705 if (s) { 1706 len = s - conf_uname; 1707 conf_uname_copy = malloc(len + 1); 1708 if (!conf_uname_copy) { 1709 debug("Can't allocate uname copy: '%s'\n", 1710 conf_uname); 1711 return -ENOMEM; 1712 } 1713 memcpy(conf_uname_copy, conf_uname, len); 1714 conf_uname_copy[len] = '\0'; 1715 conf_uname = conf_uname_copy; 1716 } 1717 1718 noffset = fdt_subnode_offset(fit, confs_noffset, conf_uname); 1719 if (noffset < 0) { 1720 debug("Can't get node offset for configuration unit name: '%s' (%s)\n", 1721 conf_uname, fdt_strerror(noffset)); 1722 } 1723 1724 if (conf_uname_copy) 1725 free(conf_uname_copy); 1726 1727 return noffset; 1728} 1729 1730int fit_conf_get_prop_node_count(const void *fit, int noffset, 1731 const char *prop_name) 1732{ 1733 return fdt_stringlist_count(fit, noffset, prop_name); 1734} 1735 1736int fit_conf_get_prop_node_index(const void *fit, int noffset, 1737 const char *prop_name, int index) 1738{ 1739 const char *uname; 1740 int len; 1741 1742 /* get kernel image unit name from configuration kernel property */ 1743 uname = fdt_stringlist_get(fit, noffset, prop_name, index, &len); 1744 if (uname == NULL) 1745 return len; 1746 1747 return fit_image_get_node(fit, uname); 1748} 1749 1750int fit_conf_get_prop_node(const void *fit, int noffset, 1751 const char *prop_name) 1752{ 1753 return fit_conf_get_prop_node_index(fit, noffset, prop_name, 0); 1754} 1755 1756static int fit_image_select(const void *fit, int rd_noffset, int verify) 1757{ 1758 fit_image_print(fit, rd_noffset, " "); 1759 1760 if (verify) { 1761 puts(" Verifying Hash Integrity ... "); 1762 if (!fit_image_verify(fit, rd_noffset)) { 1763 puts("Bad Data Hash\n"); 1764 return -EACCES; 1765 } 1766 puts("OK\n"); 1767 } 1768 1769 return 0; 1770} 1771 1772int fit_get_node_from_config(bootm_headers_t *images, const char *prop_name, 1773 ulong addr) 1774{ 1775 int cfg_noffset; 1776 void *fit_hdr; 1777 int noffset; 1778 1779 debug("* %s: using config '%s' from image at 0x%08lx\n", 1780 prop_name, images->fit_uname_cfg, addr); 1781 1782 /* Check whether configuration has this property defined */ 1783 fit_hdr = map_sysmem(addr, 0); 1784 cfg_noffset = fit_conf_get_node(fit_hdr, images->fit_uname_cfg); 1785 if (cfg_noffset < 0) { 1786 debug("* %s: no such config\n", prop_name); 1787 return -EINVAL; 1788 } 1789 1790 noffset = fit_conf_get_prop_node(fit_hdr, cfg_noffset, prop_name); 1791 if (noffset < 0) { 1792 debug("* %s: no '%s' in config\n", prop_name, prop_name); 1793 return -ENOENT; 1794 } 1795 1796 return noffset; 1797} 1798 1799/** 1800 * fit_get_image_type_property() - get property name for IH_TYPE_... 1801 * 1802 * @return the properly name where we expect to find the image in the 1803 * config node 1804 */ 1805static const char *fit_get_image_type_property(int type) 1806{ 1807 /* 1808 * This is sort-of available in the uimage_type[] table in image.c 1809 * but we don't have access to the short name, and "fdt" is different 1810 * anyway. So let's just keep it here. 1811 */ 1812 switch (type) { 1813 case IH_TYPE_FLATDT: 1814 return FIT_FDT_PROP; 1815 case IH_TYPE_KERNEL: 1816 return FIT_KERNEL_PROP; 1817 case IH_TYPE_RAMDISK: 1818 return FIT_RAMDISK_PROP; 1819 case IH_TYPE_X86_SETUP: 1820 return FIT_SETUP_PROP; 1821 case IH_TYPE_LOADABLE: 1822 return FIT_LOADABLE_PROP; 1823 case IH_TYPE_FPGA: 1824 return FIT_FPGA_PROP; 1825 case IH_TYPE_STANDALONE: 1826 return FIT_STANDALONE_PROP; 1827 } 1828 1829 return "unknown"; 1830} 1831 1832int fit_image_load(bootm_headers_t *images, ulong addr, 1833 const char **fit_unamep, const char **fit_uname_configp, 1834 int arch, int image_type, int bootstage_id, 1835 enum fit_load_op load_op, ulong *datap, ulong *lenp) 1836{ 1837 int cfg_noffset, noffset; 1838 const char *fit_uname; 1839 const char *fit_uname_config; 1840 const char *fit_base_uname_config; 1841 const void *fit; 1842 void *buf; 1843 void *loadbuf; 1844 size_t size; 1845 int type_ok, os_ok; 1846 ulong load, load_end, data, len; 1847 uint8_t os, comp; 1848#ifndef USE_HOSTCC 1849 uint8_t os_arch; 1850#endif 1851 const char *prop_name; 1852 int ret; 1853 1854 fit = map_sysmem(addr, 0); 1855 fit_uname = fit_unamep ? *fit_unamep : NULL; 1856 fit_uname_config = fit_uname_configp ? *fit_uname_configp : NULL; 1857 fit_base_uname_config = NULL; 1858 prop_name = fit_get_image_type_property(image_type); 1859 printf("## Loading %s from FIT Image at %08lx ...\n", prop_name, addr); 1860 1861 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_FORMAT); 1862 if (!fit_check_format(fit)) { 1863 printf("Bad FIT %s image format!\n", prop_name); 1864 bootstage_error(bootstage_id + BOOTSTAGE_SUB_FORMAT); 1865 return -ENOEXEC; 1866 } 1867 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_FORMAT_OK); 1868 if (fit_uname) { 1869 /* get FIT component image node offset */ 1870 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_UNIT_NAME); 1871 noffset = fit_image_get_node(fit, fit_uname); 1872 } else { 1873 /* 1874 * no image node unit name, try to get config 1875 * node first. If config unit node name is NULL 1876 * fit_conf_get_node() will try to find default config node 1877 */ 1878 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_NO_UNIT_NAME); 1879 if (IMAGE_ENABLE_BEST_MATCH && !fit_uname_config) { 1880 cfg_noffset = fit_conf_find_compat(fit, gd_fdt_blob()); 1881 } else { 1882 cfg_noffset = fit_conf_get_node(fit, 1883 fit_uname_config); 1884 } 1885 if (cfg_noffset < 0) { 1886 puts("Could not find configuration node\n"); 1887 bootstage_error(bootstage_id + 1888 BOOTSTAGE_SUB_NO_UNIT_NAME); 1889 return -ENOENT; 1890 } 1891 1892 fit_base_uname_config = fdt_get_name(fit, cfg_noffset, NULL); 1893 printf(" Using '%s' configuration\n", fit_base_uname_config); 1894 /* Remember this config */ 1895 if (image_type == IH_TYPE_KERNEL) 1896 images->fit_uname_cfg = fit_base_uname_config; 1897 1898 if (IMAGE_ENABLE_VERIFY && images->verify) { 1899 puts(" Verifying Hash Integrity ... "); 1900 if (fit_config_verify(fit, cfg_noffset)) { 1901 puts("Bad Data Hash\n"); 1902 bootstage_error(bootstage_id + 1903 BOOTSTAGE_SUB_HASH); 1904 return -EACCES; 1905 } 1906 puts("OK\n"); 1907 } 1908 1909 bootstage_mark(BOOTSTAGE_ID_FIT_CONFIG); 1910 1911 noffset = fit_conf_get_prop_node(fit, cfg_noffset, 1912 prop_name); 1913 fit_uname = fit_get_name(fit, noffset, NULL); 1914 } 1915 if (noffset < 0) { 1916 printf("Could not find subimage node type '%s'\n", prop_name); 1917 bootstage_error(bootstage_id + BOOTSTAGE_SUB_SUBNODE); 1918 return -ENOENT; 1919 } 1920 1921 printf(" Trying '%s' %s subimage\n", fit_uname, prop_name); 1922 1923 ret = fit_image_select(fit, noffset, images->verify); 1924 if (ret) { 1925 bootstage_error(bootstage_id + BOOTSTAGE_SUB_HASH); 1926 return ret; 1927 } 1928 1929 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_CHECK_ARCH); 1930#if !defined(USE_HOSTCC) && !defined(CONFIG_SANDBOX) 1931 if (!fit_image_check_target_arch(fit, noffset)) { 1932 puts("Unsupported Architecture\n"); 1933 bootstage_error(bootstage_id + BOOTSTAGE_SUB_CHECK_ARCH); 1934 return -ENOEXEC; 1935 } 1936#endif 1937 1938#ifndef USE_HOSTCC 1939 fit_image_get_arch(fit, noffset, &os_arch); 1940 images->os.arch = os_arch; 1941#endif 1942 1943 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_CHECK_ALL); 1944 type_ok = fit_image_check_type(fit, noffset, image_type) || 1945 fit_image_check_type(fit, noffset, IH_TYPE_FIRMWARE) || 1946 (image_type == IH_TYPE_KERNEL && 1947 fit_image_check_type(fit, noffset, IH_TYPE_KERNEL_NOLOAD)); 1948 1949 os_ok = image_type == IH_TYPE_FLATDT || 1950 image_type == IH_TYPE_FPGA || 1951 fit_image_check_os(fit, noffset, IH_OS_LINUX) || 1952 fit_image_check_os(fit, noffset, IH_OS_U_BOOT) || 1953 fit_image_check_os(fit, noffset, IH_OS_OPENRTOS); 1954 1955 /* 1956 * If either of the checks fail, we should report an error, but 1957 * if the image type is coming from the "loadables" field, we 1958 * don't care what it is 1959 */ 1960 if ((!type_ok || !os_ok) && image_type != IH_TYPE_LOADABLE) { 1961 fit_image_get_os(fit, noffset, &os); 1962 printf("No %s %s %s Image\n", 1963 genimg_get_os_name(os), 1964 genimg_get_arch_name(arch), 1965 genimg_get_type_name(image_type)); 1966 bootstage_error(bootstage_id + BOOTSTAGE_SUB_CHECK_ALL); 1967 return -EIO; 1968 } 1969 1970 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_CHECK_ALL_OK); 1971 1972 /* get image data address and length */ 1973 if (fit_image_get_data_and_size(fit, noffset, 1974 (const void **)&buf, &size)) { 1975 printf("Could not find %s subimage data!\n", prop_name); 1976 bootstage_error(bootstage_id + BOOTSTAGE_SUB_GET_DATA); 1977 return -ENOENT; 1978 } 1979 1980#if !defined(USE_HOSTCC) && defined(CONFIG_FIT_IMAGE_POST_PROCESS) 1981 /* perform any post-processing on the image data */ 1982 board_fit_image_post_process(&buf, &size); 1983#endif 1984 1985 len = (ulong)size; 1986 1987 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_GET_DATA_OK); 1988 1989 data = map_to_sysmem(buf); 1990 load = data; 1991 if (load_op == FIT_LOAD_IGNORED) { 1992 /* Don't load */ 1993 } else if (fit_image_get_load(fit, noffset, &load)) { 1994 if (load_op == FIT_LOAD_REQUIRED) { 1995 printf("Can't get %s subimage load address!\n", 1996 prop_name); 1997 bootstage_error(bootstage_id + BOOTSTAGE_SUB_LOAD); 1998 return -EBADF; 1999 } 2000 } else if (load_op != FIT_LOAD_OPTIONAL_NON_ZERO || load) { 2001 ulong image_start, image_end; 2002 2003 /* 2004 * move image data to the load address, 2005 * make sure we don't overwrite initial image 2006 */ 2007 image_start = addr; 2008 image_end = addr + fit_get_size(fit); 2009 2010 load_end = load + len; 2011 if (image_type != IH_TYPE_KERNEL && 2012 load < image_end && load_end > image_start) { 2013 printf("Error: %s overwritten\n", prop_name); 2014 return -EXDEV; 2015 } 2016 2017 printf(" Loading %s from 0x%08lx to 0x%08lx\n", 2018 prop_name, data, load); 2019 } else { 2020 load = data; /* No load address specified */ 2021 } 2022 2023 comp = IH_COMP_NONE; 2024 loadbuf = buf; 2025 /* Kernel images get decompressed later in bootm_load_os(). */ 2026 if (!fit_image_get_comp(fit, noffset, &comp) && 2027 comp != IH_COMP_NONE && 2028 !(image_type == IH_TYPE_KERNEL || 2029 image_type == IH_TYPE_KERNEL_NOLOAD || 2030 image_type == IH_TYPE_RAMDISK)) { 2031 ulong max_decomp_len = len * 20; 2032 if (load == data) { 2033 loadbuf = malloc(max_decomp_len); 2034 load = map_to_sysmem(loadbuf); 2035 } else { 2036 loadbuf = map_sysmem(load, max_decomp_len); 2037 } 2038 if (image_decomp(comp, load, data, image_type, 2039 loadbuf, buf, len, max_decomp_len, &load_end)) { 2040 printf("Error decompressing %s\n", prop_name); 2041 2042 return -ENOEXEC; 2043 } 2044 len = load_end - load; 2045 } else if (load != data) { 2046 loadbuf = map_sysmem(load, len); 2047 memcpy(loadbuf, buf, len); 2048 } 2049 2050 if (image_type == IH_TYPE_RAMDISK && comp != IH_COMP_NONE) 2051 puts("WARNING: 'compression' nodes for ramdisks are deprecated," 2052 " please fix your .its file!\n"); 2053 2054 /* verify that image data is a proper FDT blob */ 2055 if (image_type == IH_TYPE_FLATDT && fdt_check_header(loadbuf)) { 2056 puts("Subimage data is not a FDT"); 2057 return -ENOEXEC; 2058 } 2059 2060 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_LOAD); 2061 2062 *datap = load; 2063 *lenp = len; 2064 if (fit_unamep) 2065 *fit_unamep = (char *)fit_uname; 2066 if (fit_uname_configp) 2067 *fit_uname_configp = (char *)(fit_uname_config ? : 2068 fit_base_uname_config); 2069 2070 return noffset; 2071} 2072 2073int boot_get_setup_fit(bootm_headers_t *images, uint8_t arch, 2074 ulong *setup_start, ulong *setup_len) 2075{ 2076 int noffset; 2077 ulong addr; 2078 ulong len; 2079 int ret; 2080 2081 addr = map_to_sysmem(images->fit_hdr_os); 2082 noffset = fit_get_node_from_config(images, FIT_SETUP_PROP, addr); 2083 if (noffset < 0) 2084 return noffset; 2085 2086 ret = fit_image_load(images, addr, NULL, NULL, arch, 2087 IH_TYPE_X86_SETUP, BOOTSTAGE_ID_FIT_SETUP_START, 2088 FIT_LOAD_REQUIRED, setup_start, &len); 2089 2090 return ret; 2091} 2092 2093#ifndef USE_HOSTCC 2094int boot_get_fdt_fit(bootm_headers_t *images, ulong addr, 2095 const char **fit_unamep, const char **fit_uname_configp, 2096 int arch, ulong *datap, ulong *lenp) 2097{ 2098 int fdt_noffset, cfg_noffset, count; 2099 const void *fit; 2100 const char *fit_uname = NULL; 2101 const char *fit_uname_config = NULL; 2102 char *fit_uname_config_copy = NULL; 2103 char *next_config = NULL; 2104 ulong load, len; 2105#ifdef CONFIG_OF_LIBFDT_OVERLAY 2106 ulong image_start, image_end; 2107 ulong ovload, ovlen; 2108 const char *uconfig; 2109 const char *uname; 2110 void *base, *ov; 2111 int i, err, noffset, ov_noffset; 2112#endif 2113 2114 fit_uname = fit_unamep ? *fit_unamep : NULL; 2115 2116 if (fit_uname_configp && *fit_uname_configp) { 2117 fit_uname_config_copy = strdup(*fit_uname_configp); 2118 if (!fit_uname_config_copy) 2119 return -ENOMEM; 2120 2121 next_config = strchr(fit_uname_config_copy, '#'); 2122 if (next_config) 2123 *next_config++ = '\0'; 2124 if (next_config - 1 > fit_uname_config_copy) 2125 fit_uname_config = fit_uname_config_copy; 2126 } 2127 2128 fdt_noffset = fit_image_load(images, 2129 addr, &fit_uname, &fit_uname_config, 2130 arch, IH_TYPE_FLATDT, 2131 BOOTSTAGE_ID_FIT_FDT_START, 2132 FIT_LOAD_OPTIONAL, &load, &len); 2133 2134 if (fdt_noffset < 0) 2135 goto out; 2136 2137 debug("fit_uname=%s, fit_uname_config=%s\n", 2138 fit_uname ? fit_uname : "<NULL>", 2139 fit_uname_config ? fit_uname_config : "<NULL>"); 2140 2141 fit = map_sysmem(addr, 0); 2142 2143 cfg_noffset = fit_conf_get_node(fit, fit_uname_config); 2144 2145 /* single blob, or error just return as well */ 2146 count = fit_conf_get_prop_node_count(fit, cfg_noffset, FIT_FDT_PROP); 2147 if (count <= 1 && !next_config) 2148 goto out; 2149 2150 /* we need to apply overlays */ 2151 2152#ifdef CONFIG_OF_LIBFDT_OVERLAY 2153 image_start = addr; 2154 image_end = addr + fit_get_size(fit); 2155 /* verify that relocation took place by load address not being in fit */ 2156 if (load >= image_start && load < image_end) { 2157 /* check is simplified; fit load checks for overlaps */ 2158 printf("Overlayed FDT requires relocation\n"); 2159 fdt_noffset = -EBADF; 2160 goto out; 2161 } 2162 2163 base = map_sysmem(load, len); 2164 2165 /* apply extra configs in FIT first, followed by args */ 2166 for (i = 1; ; i++) { 2167 if (i < count) { 2168 noffset = fit_conf_get_prop_node_index(fit, cfg_noffset, 2169 FIT_FDT_PROP, i); 2170 uname = fit_get_name(fit, noffset, NULL); 2171 uconfig = NULL; 2172 } else { 2173 if (!next_config) 2174 break; 2175 uconfig = next_config; 2176 next_config = strchr(next_config, '#'); 2177 if (next_config) 2178 *next_config++ = '\0'; 2179 uname = NULL; 2180 2181 /* 2182 * fit_image_load() would load the first FDT from the 2183 * extra config only when uconfig is specified. 2184 * Check if the extra config contains multiple FDTs and 2185 * if so, load them. 2186 */ 2187 cfg_noffset = fit_conf_get_node(fit, uconfig); 2188 2189 i = 0; 2190 count = fit_conf_get_prop_node_count(fit, cfg_noffset, 2191 FIT_FDT_PROP); 2192 } 2193 2194 debug("%d: using uname=%s uconfig=%s\n", i, uname, uconfig); 2195 2196 ov_noffset = fit_image_load(images, 2197 addr, &uname, &uconfig, 2198 arch, IH_TYPE_FLATDT, 2199 BOOTSTAGE_ID_FIT_FDT_START, 2200 FIT_LOAD_REQUIRED, &ovload, &ovlen); 2201 if (ov_noffset < 0) { 2202 printf("load of %s failed\n", uname); 2203 continue; 2204 } 2205 debug("%s loaded at 0x%08lx len=0x%08lx\n", 2206 uname, ovload, ovlen); 2207 ov = map_sysmem(ovload, ovlen); 2208 2209 base = map_sysmem(load, len + ovlen); 2210 err = fdt_open_into(base, base, len + ovlen); 2211 if (err < 0) { 2212 printf("failed on fdt_open_into\n"); 2213 fdt_noffset = err; 2214 goto out; 2215 } 2216 /* the verbose method prints out messages on error */ 2217 err = fdt_overlay_apply_verbose(base, ov); 2218 if (err < 0) { 2219 fdt_noffset = err; 2220 goto out; 2221 } 2222 fdt_pack(base); 2223 len = fdt_totalsize(base); 2224 } 2225#else 2226 printf("config with overlays but CONFIG_OF_LIBFDT_OVERLAY not set\n"); 2227 fdt_noffset = -EBADF; 2228#endif 2229 2230out: 2231 if (datap) 2232 *datap = load; 2233 if (lenp) 2234 *lenp = len; 2235 if (fit_unamep) 2236 *fit_unamep = fit_uname; 2237 if (fit_uname_configp) 2238 *fit_uname_configp = fit_uname_config; 2239 2240 if (fit_uname_config_copy) 2241 free(fit_uname_config_copy); 2242 return fdt_noffset; 2243} 2244#endif 2245