blob: aff4670be35e4ab118d652ca69ae7ec74dcba616 [file] [log] [blame]
Tom Rini10e47792018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Simon Glassda0af362013-05-07 06:11:53 +00002/*
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.
Simon Glassda0af362013-05-07 06:11:53 +00009 */
10
Simon Glassd563c252021-02-15 17:08:09 -070011#define LOG_CATEGORY LOGC_BOOT
12
Simon Glassda0af362013-05-07 06:11:53 +000013#ifdef USE_HOSTCC
14#include "mkimage.h"
Simon Glassda0af362013-05-07 06:11:53 +000015#include <time.h>
Simon Glass2dc9c342020-05-10 11:40:01 -060016#include <linux/libfdt.h>
Simon Glass48b6c6b2019-11-14 12:57:16 -070017#include <u-boot/crc.h>
Simon Glassda0af362013-05-07 06:11:53 +000018#else
Andreas Dannenberg67aaa6d2016-07-27 12:12:39 -050019#include <linux/compiler.h>
Marek Vasutf7d24612021-06-11 04:09:56 +020020#include <linux/sizes.h>
Simon Glassda0af362013-05-07 06:11:53 +000021#include <common.h>
Simon Glass384d86d2013-05-16 13:53:21 +000022#include <errno.h>
Simon Glass0f2af882020-05-10 11:40:05 -060023#include <log.h>
Joe Hershberger65b905b2015-03-22 17:08:59 -050024#include <mapmem.h>
Simon Glass384d86d2013-05-16 13:53:21 +000025#include <asm/io.h>
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +030026#include <malloc.h>
Simon Glass3ba929a2020-10-30 21:38:53 -060027#include <asm/global_data.h>
Simon Glass384d86d2013-05-16 13:53:21 +000028DECLARE_GLOBAL_DATA_PTR;
Simon Glassda0af362013-05-07 06:11:53 +000029#endif /* !USE_HOSTCC*/
30
Julius Werner97b09cd2019-07-24 19:37:55 -070031#include <bootm.h>
Andreas Dannenberg67aaa6d2016-07-27 12:12:39 -050032#include <image.h>
Simon Glassda0af362013-05-07 06:11:53 +000033#include <bootstage.h>
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +010034#include <linux/kconfig.h>
Simon Glassda0af362013-05-07 06:11:53 +000035#include <u-boot/crc.h>
36#include <u-boot/md5.h>
Jeroen Hofsteebfe88fe2014-06-12 22:27:12 +020037#include <u-boot/sha1.h>
38#include <u-boot/sha256.h>
Reuben Dowle1908fd92020-04-16 17:36:52 +120039#include <u-boot/sha512.h>
Simon Glassda0af362013-05-07 06:11:53 +000040
41/*****************************************************************************/
42/* New uImage format routines */
43/*****************************************************************************/
44#ifndef USE_HOSTCC
45static int fit_parse_spec(const char *spec, char sepc, ulong addr_curr,
46 ulong *addr, const char **name)
47{
48 const char *sep;
49
50 *addr = addr_curr;
51 *name = NULL;
52
53 sep = strchr(spec, sepc);
54 if (sep) {
55 if (sep - spec > 0)
Simon Glass3ff49ec2021-07-24 09:03:29 -060056 *addr = hextoul(spec, NULL);
Simon Glassda0af362013-05-07 06:11:53 +000057
58 *name = sep + 1;
59 return 1;
60 }
61
62 return 0;
63}
64
65/**
66 * fit_parse_conf - parse FIT configuration spec
67 * @spec: input string, containing configuration spec
68 * @add_curr: current image address (to be used as a possible default)
69 * @addr: pointer to a ulong variable, will hold FIT image address of a given
70 * configuration
71 * @conf_name double pointer to a char, will hold pointer to a configuration
72 * unit name
73 *
Masahiro Yamada4ac739a2013-09-18 09:36:38 +090074 * fit_parse_conf() expects configuration spec in the form of [<addr>]#<conf>,
Simon Glassda0af362013-05-07 06:11:53 +000075 * where <addr> is a FIT image address that contains configuration
76 * with a <conf> unit name.
77 *
78 * Address part is optional, and if omitted default add_curr will
79 * be used instead.
80 *
81 * returns:
82 * 1 if spec is a valid configuration string,
83 * addr and conf_name are set accordingly
84 * 0 otherwise
85 */
86int fit_parse_conf(const char *spec, ulong addr_curr,
87 ulong *addr, const char **conf_name)
88{
89 return fit_parse_spec(spec, '#', addr_curr, addr, conf_name);
90}
91
92/**
93 * fit_parse_subimage - parse FIT subimage spec
94 * @spec: input string, containing subimage spec
95 * @add_curr: current image address (to be used as a possible default)
96 * @addr: pointer to a ulong variable, will hold FIT image address of a given
97 * subimage
98 * @image_name: double pointer to a char, will hold pointer to a subimage name
99 *
Masahiro Yamada4ac739a2013-09-18 09:36:38 +0900100 * fit_parse_subimage() expects subimage spec in the form of
Simon Glassda0af362013-05-07 06:11:53 +0000101 * [<addr>]:<subimage>, where <addr> is a FIT image address that contains
102 * subimage with a <subimg> unit name.
103 *
104 * Address part is optional, and if omitted default add_curr will
105 * be used instead.
106 *
107 * returns:
108 * 1 if spec is a valid subimage string,
109 * addr and image_name are set accordingly
110 * 0 otherwise
111 */
112int fit_parse_subimage(const char *spec, ulong addr_curr,
113 ulong *addr, const char **image_name)
114{
115 return fit_parse_spec(spec, ':', addr_curr, addr, image_name);
116}
117#endif /* !USE_HOSTCC */
118
Joel Stanleyc3baacd2020-12-08 14:42:14 +1030119#ifdef USE_HOSTCC
120/* Host tools use these implementations for Cipher and Signature support */
121static void *host_blob;
122
123void image_set_host_blob(void *blob)
124{
125 host_blob = blob;
126}
127
128void *image_get_host_blob(void)
129{
130 return host_blob;
131}
132#endif /* USE_HOSTCC */
133
Simon Glassda0af362013-05-07 06:11:53 +0000134static void fit_get_debug(const void *fit, int noffset,
135 char *prop_name, int err)
136{
137 debug("Can't get '%s' property from FIT 0x%08lx, node: offset %d, name %s (%s)\n",
138 prop_name, (ulong)fit, noffset, fit_get_name(fit, noffset, NULL),
139 fdt_strerror(err));
140}
141
Guilherme Maciel Ferreira3c46bcd2015-01-15 02:54:42 -0200142/**
143 * fit_get_subimage_count - get component (sub-image) count
144 * @fit: pointer to the FIT format image header
145 * @images_noffset: offset of images node
146 *
147 * returns:
148 * number of image components
149 */
150int fit_get_subimage_count(const void *fit, int images_noffset)
151{
152 int noffset;
153 int ndepth;
154 int count = 0;
155
156 /* Process its subnodes, print out component images details */
157 for (ndepth = 0, count = 0,
158 noffset = fdt_next_node(fit, images_noffset, &ndepth);
159 (noffset >= 0) && (ndepth > 0);
160 noffset = fdt_next_node(fit, noffset, &ndepth)) {
161 if (ndepth == 1) {
162 count++;
163 }
164 }
165
166 return count;
167}
168
Ravik Hasija11dac412021-01-27 14:01:48 -0800169#if CONFIG_IS_ENABLED(FIT_PRINT) || CONFIG_IS_ENABLED(SPL_FIT_PRINT)
Simon Glassda0af362013-05-07 06:11:53 +0000170/**
Tom Rini7d77b662018-05-08 14:34:05 -0400171 * fit_image_print_data() - prints out the hash node details
172 * @fit: pointer to the FIT format image header
173 * @noffset: offset of the hash node
174 * @p: pointer to prefix string
175 * @type: Type of information to print ("hash" or "sign")
176 *
177 * fit_image_print_data() lists properties for the processed hash node
178 *
179 * This function avoid using puts() since it prints a newline on the host
180 * but does not in U-Boot.
181 *
182 * returns:
183 * no returned results
184 */
185static void fit_image_print_data(const void *fit, int noffset, const char *p,
186 const char *type)
187{
188 const char *keyname;
189 uint8_t *value;
190 int value_len;
191 char *algo;
Philippe Reynes12468352018-11-14 13:51:00 +0100192 const char *padding;
Simon Glassd7aabcc2020-03-18 11:44:06 -0600193 bool required;
Tom Rini7d77b662018-05-08 14:34:05 -0400194 int ret, i;
195
196 debug("%s %s node: '%s'\n", p, type,
197 fit_get_name(fit, noffset, NULL));
198 printf("%s %s algo: ", p, type);
199 if (fit_image_hash_get_algo(fit, noffset, &algo)) {
200 printf("invalid/unsupported\n");
201 return;
202 }
203 printf("%s", algo);
Simon Glassd7aabcc2020-03-18 11:44:06 -0600204 keyname = fdt_getprop(fit, noffset, FIT_KEY_HINT, NULL);
205 required = fdt_getprop(fit, noffset, FIT_KEY_REQUIRED, NULL) != NULL;
Tom Rini7d77b662018-05-08 14:34:05 -0400206 if (keyname)
207 printf(":%s", keyname);
208 if (required)
209 printf(" (required)");
210 printf("\n");
211
Philippe Reynes12468352018-11-14 13:51:00 +0100212 padding = fdt_getprop(fit, noffset, "padding", NULL);
213 if (padding)
214 printf("%s %s padding: %s\n", p, type, padding);
215
Tom Rini7d77b662018-05-08 14:34:05 -0400216 ret = fit_image_hash_get_value(fit, noffset, &value,
217 &value_len);
218 printf("%s %s value: ", p, type);
219 if (ret) {
220 printf("unavailable\n");
221 } else {
222 for (i = 0; i < value_len; i++)
223 printf("%02x", value[i]);
224 printf("\n");
225 }
226
227 debug("%s %s len: %d\n", p, type, value_len);
228
229 /* Signatures have a time stamp */
230 if (IMAGE_ENABLE_TIMESTAMP && keyname) {
231 time_t timestamp;
232
233 printf("%s Timestamp: ", p);
234 if (fit_get_timestamp(fit, noffset, &timestamp))
235 printf("unavailable\n");
236 else
237 genimg_print_time(timestamp);
238 }
239}
240
241/**
242 * fit_image_print_verification_data() - prints out the hash/signature details
243 * @fit: pointer to the FIT format image header
244 * @noffset: offset of the hash or signature node
245 * @p: pointer to prefix string
246 *
247 * This lists properties for the processed hash node
248 *
249 * returns:
250 * no returned results
251 */
252static void fit_image_print_verification_data(const void *fit, int noffset,
253 const char *p)
254{
255 const char *name;
256
257 /*
258 * Check subnode name, must be equal to "hash" or "signature".
259 * Multiple hash/signature nodes require unique unit node
260 * names, e.g. hash-1, hash-2, signature-1, signature-2, etc.
261 */
262 name = fit_get_name(fit, noffset, NULL);
263 if (!strncmp(name, FIT_HASH_NODENAME, strlen(FIT_HASH_NODENAME))) {
264 fit_image_print_data(fit, noffset, p, "Hash");
265 } else if (!strncmp(name, FIT_SIG_NODENAME,
266 strlen(FIT_SIG_NODENAME))) {
267 fit_image_print_data(fit, noffset, p, "Sign");
268 }
269}
270
271/**
272 * fit_conf_print - prints out the FIT configuration details
273 * @fit: pointer to the FIT format image header
274 * @noffset: offset of the configuration node
275 * @p: pointer to prefix string
276 *
277 * fit_conf_print() lists all mandatory properties for the processed
278 * configuration node.
279 *
280 * returns:
281 * no returned results
282 */
283static void fit_conf_print(const void *fit, int noffset, const char *p)
284{
285 char *desc;
286 const char *uname;
287 int ret;
288 int fdt_index, loadables_index;
289 int ndepth;
290
291 /* Mandatory properties */
292 ret = fit_get_desc(fit, noffset, &desc);
293 printf("%s Description: ", p);
294 if (ret)
295 printf("unavailable\n");
296 else
297 printf("%s\n", desc);
298
299 uname = fdt_getprop(fit, noffset, FIT_KERNEL_PROP, NULL);
300 printf("%s Kernel: ", p);
301 if (!uname)
302 printf("unavailable\n");
303 else
304 printf("%s\n", uname);
305
306 /* Optional properties */
307 uname = fdt_getprop(fit, noffset, FIT_RAMDISK_PROP, NULL);
308 if (uname)
309 printf("%s Init Ramdisk: %s\n", p, uname);
310
311 uname = fdt_getprop(fit, noffset, FIT_FIRMWARE_PROP, NULL);
312 if (uname)
313 printf("%s Firmware: %s\n", p, uname);
314
315 for (fdt_index = 0;
316 uname = fdt_stringlist_get(fit, noffset, FIT_FDT_PROP,
317 fdt_index, NULL), uname;
318 fdt_index++) {
319 if (fdt_index == 0)
320 printf("%s FDT: ", p);
321 else
322 printf("%s ", p);
323 printf("%s\n", uname);
324 }
325
326 uname = fdt_getprop(fit, noffset, FIT_FPGA_PROP, NULL);
327 if (uname)
328 printf("%s FPGA: %s\n", p, uname);
329
330 /* Print out all of the specified loadables */
331 for (loadables_index = 0;
332 uname = fdt_stringlist_get(fit, noffset, FIT_LOADABLE_PROP,
333 loadables_index, NULL), uname;
334 loadables_index++) {
335 if (loadables_index == 0) {
336 printf("%s Loadables: ", p);
337 } else {
338 printf("%s ", p);
339 }
340 printf("%s\n", uname);
341 }
342
343 /* Process all hash subnodes of the component configuration node */
344 for (ndepth = 0, noffset = fdt_next_node(fit, noffset, &ndepth);
345 (noffset >= 0) && (ndepth > 0);
346 noffset = fdt_next_node(fit, noffset, &ndepth)) {
347 if (ndepth == 1) {
348 /* Direct child node of the component configuration node */
349 fit_image_print_verification_data(fit, noffset, p);
350 }
351 }
352}
353
354/**
Simon Glassda0af362013-05-07 06:11:53 +0000355 * fit_print_contents - prints out the contents of the FIT format image
356 * @fit: pointer to the FIT format image header
357 * @p: pointer to prefix string
358 *
359 * fit_print_contents() formats a multi line FIT image contents description.
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -0500360 * The routine prints out FIT image properties (root node level) followed by
Simon Glassda0af362013-05-07 06:11:53 +0000361 * the details of each component image.
362 *
363 * returns:
364 * no returned results
365 */
366void fit_print_contents(const void *fit)
367{
368 char *desc;
369 char *uname;
370 int images_noffset;
371 int confs_noffset;
372 int noffset;
373 int ndepth;
374 int count = 0;
375 int ret;
376 const char *p;
377 time_t timestamp;
378
Simon Glass1030f162013-05-08 08:05:58 +0000379 /* Indent string is defined in header image.h */
380 p = IMAGE_INDENT_STRING;
Simon Glassda0af362013-05-07 06:11:53 +0000381
382 /* Root node properties */
383 ret = fit_get_desc(fit, 0, &desc);
384 printf("%sFIT description: ", p);
385 if (ret)
386 printf("unavailable\n");
387 else
388 printf("%s\n", desc);
389
390 if (IMAGE_ENABLE_TIMESTAMP) {
391 ret = fit_get_timestamp(fit, 0, &timestamp);
392 printf("%sCreated: ", p);
393 if (ret)
394 printf("unavailable\n");
395 else
396 genimg_print_time(timestamp);
397 }
398
399 /* Find images parent node offset */
400 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
401 if (images_noffset < 0) {
402 printf("Can't find images parent node '%s' (%s)\n",
403 FIT_IMAGES_PATH, fdt_strerror(images_noffset));
404 return;
405 }
406
407 /* Process its subnodes, print out component images details */
408 for (ndepth = 0, count = 0,
409 noffset = fdt_next_node(fit, images_noffset, &ndepth);
410 (noffset >= 0) && (ndepth > 0);
411 noffset = fdt_next_node(fit, noffset, &ndepth)) {
412 if (ndepth == 1) {
413 /*
414 * Direct child node of the images parent node,
415 * i.e. component image node.
416 */
417 printf("%s Image %u (%s)\n", p, count++,
418 fit_get_name(fit, noffset, NULL));
419
420 fit_image_print(fit, noffset, p);
421 }
422 }
423
424 /* Find configurations parent node offset */
425 confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH);
426 if (confs_noffset < 0) {
427 debug("Can't get configurations parent node '%s' (%s)\n",
428 FIT_CONFS_PATH, fdt_strerror(confs_noffset));
429 return;
430 }
431
432 /* get default configuration unit name from default property */
433 uname = (char *)fdt_getprop(fit, noffset, FIT_DEFAULT_PROP, NULL);
434 if (uname)
435 printf("%s Default Configuration: '%s'\n", p, uname);
436
437 /* Process its subnodes, print out configurations details */
438 for (ndepth = 0, count = 0,
439 noffset = fdt_next_node(fit, confs_noffset, &ndepth);
440 (noffset >= 0) && (ndepth > 0);
441 noffset = fdt_next_node(fit, noffset, &ndepth)) {
442 if (ndepth == 1) {
443 /*
444 * Direct child node of the configurations parent node,
445 * i.e. configuration node.
446 */
447 printf("%s Configuration %u (%s)\n", p, count++,
448 fit_get_name(fit, noffset, NULL));
449
450 fit_conf_print(fit, noffset, p);
451 }
452 }
453}
454
455/**
456 * fit_image_print - prints out the FIT component image details
457 * @fit: pointer to the FIT format image header
458 * @image_noffset: offset of the component image node
459 * @p: pointer to prefix string
460 *
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -0500461 * fit_image_print() lists all mandatory properties for the processed component
Simon Glassda0af362013-05-07 06:11:53 +0000462 * image. If present, hash nodes are printed out as well. Load
463 * address for images of type firmware is also printed out. Since the load
464 * address is not mandatory for firmware images, it will be output as
465 * "unavailable" when not present.
466 *
467 * returns:
468 * no returned results
469 */
470void fit_image_print(const void *fit, int image_noffset, const char *p)
471{
472 char *desc;
473 uint8_t type, arch, os, comp;
474 size_t size;
475 ulong load, entry;
476 const void *data;
477 int noffset;
478 int ndepth;
479 int ret;
480
481 /* Mandatory properties */
482 ret = fit_get_desc(fit, image_noffset, &desc);
483 printf("%s Description: ", p);
484 if (ret)
485 printf("unavailable\n");
486 else
487 printf("%s\n", desc);
488
Simon Glass749a6e72013-07-16 20:10:01 -0700489 if (IMAGE_ENABLE_TIMESTAMP) {
490 time_t timestamp;
491
492 ret = fit_get_timestamp(fit, 0, &timestamp);
493 printf("%s Created: ", p);
494 if (ret)
495 printf("unavailable\n");
496 else
497 genimg_print_time(timestamp);
498 }
499
Simon Glassda0af362013-05-07 06:11:53 +0000500 fit_image_get_type(fit, image_noffset, &type);
501 printf("%s Type: %s\n", p, genimg_get_type_name(type));
502
503 fit_image_get_comp(fit, image_noffset, &comp);
504 printf("%s Compression: %s\n", p, genimg_get_comp_name(comp));
505
Kelvin Cheung186cc992018-05-19 18:21:37 +0800506 ret = fit_image_get_data_and_size(fit, image_noffset, &data, &size);
Simon Glassda0af362013-05-07 06:11:53 +0000507
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +0100508 if (!host_build()) {
509 printf("%s Data Start: ", p);
510 if (ret) {
511 printf("unavailable\n");
512 } else {
513 void *vdata = (void *)data;
Simon Glasse8d6a5b2013-05-16 13:53:26 +0000514
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +0100515 printf("0x%08lx\n", (ulong)map_to_sysmem(vdata));
516 }
Simon Glasse8d6a5b2013-05-16 13:53:26 +0000517 }
Simon Glassda0af362013-05-07 06:11:53 +0000518
519 printf("%s Data Size: ", p);
520 if (ret)
521 printf("unavailable\n");
522 else
523 genimg_print_size(size);
524
525 /* Remaining, type dependent properties */
526 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) ||
527 (type == IH_TYPE_RAMDISK) || (type == IH_TYPE_FIRMWARE) ||
528 (type == IH_TYPE_FLATDT)) {
529 fit_image_get_arch(fit, image_noffset, &arch);
530 printf("%s Architecture: %s\n", p, genimg_get_arch_name(arch));
531 }
532
Michal Simek55f698f2018-03-26 16:31:27 +0200533 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_RAMDISK) ||
534 (type == IH_TYPE_FIRMWARE)) {
Simon Glassda0af362013-05-07 06:11:53 +0000535 fit_image_get_os(fit, image_noffset, &os);
536 printf("%s OS: %s\n", p, genimg_get_os_name(os));
537 }
538
539 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) ||
Michal Simekebae78b2016-05-17 14:03:50 +0200540 (type == IH_TYPE_FIRMWARE) || (type == IH_TYPE_RAMDISK) ||
541 (type == IH_TYPE_FPGA)) {
Simon Glassda0af362013-05-07 06:11:53 +0000542 ret = fit_image_get_load(fit, image_noffset, &load);
543 printf("%s Load Address: ", p);
544 if (ret)
545 printf("unavailable\n");
546 else
547 printf("0x%08lx\n", load);
548 }
549
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +0300550 /* optional load address for FDT */
551 if (type == IH_TYPE_FLATDT && !fit_image_get_load(fit, image_noffset, &load))
552 printf("%s Load Address: 0x%08lx\n", p, load);
553
Simon Glassda0af362013-05-07 06:11:53 +0000554 if ((type == IH_TYPE_KERNEL) || (type == IH_TYPE_STANDALONE) ||
555 (type == IH_TYPE_RAMDISK)) {
York Sun2b0464e2016-02-29 15:48:40 -0800556 ret = fit_image_get_entry(fit, image_noffset, &entry);
Simon Glassda0af362013-05-07 06:11:53 +0000557 printf("%s Entry Point: ", p);
558 if (ret)
559 printf("unavailable\n");
560 else
561 printf("0x%08lx\n", entry);
562 }
563
564 /* Process all hash subnodes of the component image node */
565 for (ndepth = 0, noffset = fdt_next_node(fit, image_noffset, &ndepth);
566 (noffset >= 0) && (ndepth > 0);
567 noffset = fdt_next_node(fit, noffset, &ndepth)) {
568 if (ndepth == 1) {
569 /* Direct child node of the component image node */
Simon Glasse61aa442013-05-07 06:12:02 +0000570 fit_image_print_verification_data(fit, noffset, p);
Simon Glassda0af362013-05-07 06:11:53 +0000571 }
572 }
Simon Glassda0af362013-05-07 06:11:53 +0000573}
Marek Vasut25114a62018-05-13 00:22:53 +0200574#else
575void fit_print_contents(const void *fit) { }
576void fit_image_print(const void *fit, int image_noffset, const char *p) { }
Ravik Hasija11dac412021-01-27 14:01:48 -0800577#endif /* CONFIG_IS_ENABLED(FIR_PRINT) || CONFIG_IS_ENABLED(SPL_FIT_PRINT) */
Simon Glassda0af362013-05-07 06:11:53 +0000578
579/**
580 * fit_get_desc - get node description property
581 * @fit: pointer to the FIT format image header
582 * @noffset: node offset
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -0500583 * @desc: double pointer to the char, will hold pointer to the description
Simon Glassda0af362013-05-07 06:11:53 +0000584 *
585 * fit_get_desc() reads description property from a given node, if
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -0500586 * description is found pointer to it is returned in third call argument.
Simon Glassda0af362013-05-07 06:11:53 +0000587 *
588 * returns:
589 * 0, on success
590 * -1, on failure
591 */
592int fit_get_desc(const void *fit, int noffset, char **desc)
593{
594 int len;
595
596 *desc = (char *)fdt_getprop(fit, noffset, FIT_DESC_PROP, &len);
597 if (*desc == NULL) {
598 fit_get_debug(fit, noffset, FIT_DESC_PROP, len);
599 return -1;
600 }
601
602 return 0;
603}
604
605/**
606 * fit_get_timestamp - get node timestamp property
607 * @fit: pointer to the FIT format image header
608 * @noffset: node offset
609 * @timestamp: pointer to the time_t, will hold read timestamp
610 *
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -0500611 * fit_get_timestamp() reads timestamp property from given node, if timestamp
612 * is found and has a correct size its value is returned in third call
Simon Glassda0af362013-05-07 06:11:53 +0000613 * argument.
614 *
615 * returns:
616 * 0, on success
617 * -1, on property read failure
618 * -2, on wrong timestamp size
619 */
620int fit_get_timestamp(const void *fit, int noffset, time_t *timestamp)
621{
622 int len;
623 const void *data;
624
625 data = fdt_getprop(fit, noffset, FIT_TIMESTAMP_PROP, &len);
626 if (data == NULL) {
627 fit_get_debug(fit, noffset, FIT_TIMESTAMP_PROP, len);
628 return -1;
629 }
630 if (len != sizeof(uint32_t)) {
631 debug("FIT timestamp with incorrect size of (%u)\n", len);
632 return -2;
633 }
634
635 *timestamp = uimage_to_cpu(*((uint32_t *)data));
636 return 0;
637}
638
639/**
640 * fit_image_get_node - get node offset for component image of a given unit name
641 * @fit: pointer to the FIT format image header
642 * @image_uname: component image node unit name
643 *
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -0500644 * fit_image_get_node() finds a component image (within the '/images'
Simon Glassda0af362013-05-07 06:11:53 +0000645 * node) of a provided unit name. If image is found its node offset is
646 * returned to the caller.
647 *
648 * returns:
649 * image node offset when found (>=0)
650 * negative number on failure (FDT_ERR_* code)
651 */
652int fit_image_get_node(const void *fit, const char *image_uname)
653{
654 int noffset, images_noffset;
655
656 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
657 if (images_noffset < 0) {
658 debug("Can't find images parent node '%s' (%s)\n",
659 FIT_IMAGES_PATH, fdt_strerror(images_noffset));
660 return images_noffset;
661 }
662
663 noffset = fdt_subnode_offset(fit, images_noffset, image_uname);
664 if (noffset < 0) {
665 debug("Can't get node offset for image unit name: '%s' (%s)\n",
666 image_uname, fdt_strerror(noffset));
667 }
668
669 return noffset;
670}
671
672/**
673 * fit_image_get_os - get os id for a given component image node
674 * @fit: pointer to the FIT format image header
675 * @noffset: component image node offset
676 * @os: pointer to the uint8_t, will hold os numeric id
677 *
678 * fit_image_get_os() finds os property in a given component image node.
679 * If the property is found, its (string) value is translated to the numeric
680 * id which is returned to the caller.
681 *
682 * returns:
683 * 0, on success
684 * -1, on failure
685 */
686int fit_image_get_os(const void *fit, int noffset, uint8_t *os)
687{
688 int len;
689 const void *data;
690
691 /* Get OS name from property data */
692 data = fdt_getprop(fit, noffset, FIT_OS_PROP, &len);
693 if (data == NULL) {
694 fit_get_debug(fit, noffset, FIT_OS_PROP, len);
695 *os = -1;
696 return -1;
697 }
698
699 /* Translate OS name to id */
700 *os = genimg_get_os_id(data);
701 return 0;
702}
703
704/**
705 * fit_image_get_arch - get arch id for a given component image node
706 * @fit: pointer to the FIT format image header
707 * @noffset: component image node offset
708 * @arch: pointer to the uint8_t, will hold arch numeric id
709 *
710 * fit_image_get_arch() finds arch property in a given component image node.
711 * If the property is found, its (string) value is translated to the numeric
712 * id which is returned to the caller.
713 *
714 * returns:
715 * 0, on success
716 * -1, on failure
717 */
718int fit_image_get_arch(const void *fit, int noffset, uint8_t *arch)
719{
720 int len;
721 const void *data;
722
723 /* Get architecture name from property data */
724 data = fdt_getprop(fit, noffset, FIT_ARCH_PROP, &len);
725 if (data == NULL) {
726 fit_get_debug(fit, noffset, FIT_ARCH_PROP, len);
727 *arch = -1;
728 return -1;
729 }
730
731 /* Translate architecture name to id */
732 *arch = genimg_get_arch_id(data);
733 return 0;
734}
735
736/**
737 * fit_image_get_type - get type id for a given component image node
738 * @fit: pointer to the FIT format image header
739 * @noffset: component image node offset
740 * @type: pointer to the uint8_t, will hold type numeric id
741 *
742 * fit_image_get_type() finds type property in a given component image node.
743 * If the property is found, its (string) value is translated to the numeric
744 * id which is returned to the caller.
745 *
746 * returns:
747 * 0, on success
748 * -1, on failure
749 */
750int fit_image_get_type(const void *fit, int noffset, uint8_t *type)
751{
752 int len;
753 const void *data;
754
755 /* Get image type name from property data */
756 data = fdt_getprop(fit, noffset, FIT_TYPE_PROP, &len);
757 if (data == NULL) {
758 fit_get_debug(fit, noffset, FIT_TYPE_PROP, len);
759 *type = -1;
760 return -1;
761 }
762
763 /* Translate image type name to id */
764 *type = genimg_get_type_id(data);
765 return 0;
766}
767
768/**
769 * fit_image_get_comp - get comp id for a given component image node
770 * @fit: pointer to the FIT format image header
771 * @noffset: component image node offset
772 * @comp: pointer to the uint8_t, will hold comp numeric id
773 *
774 * fit_image_get_comp() finds comp property in a given component image node.
775 * If the property is found, its (string) value is translated to the numeric
776 * id which is returned to the caller.
777 *
778 * returns:
779 * 0, on success
780 * -1, on failure
781 */
782int fit_image_get_comp(const void *fit, int noffset, uint8_t *comp)
783{
784 int len;
785 const void *data;
786
787 /* Get compression name from property data */
788 data = fdt_getprop(fit, noffset, FIT_COMP_PROP, &len);
789 if (data == NULL) {
790 fit_get_debug(fit, noffset, FIT_COMP_PROP, len);
791 *comp = -1;
792 return -1;
793 }
794
795 /* Translate compression name to id */
796 *comp = genimg_get_comp_id(data);
797 return 0;
798}
799
York Sun2b0464e2016-02-29 15:48:40 -0800800static int fit_image_get_address(const void *fit, int noffset, char *name,
801 ulong *load)
802{
York Sun31e5add2016-02-29 15:48:41 -0800803 int len, cell_len;
804 const fdt32_t *cell;
805 uint64_t load64 = 0;
York Sun2b0464e2016-02-29 15:48:40 -0800806
York Sun31e5add2016-02-29 15:48:41 -0800807 cell = fdt_getprop(fit, noffset, name, &len);
808 if (cell == NULL) {
York Sun2b0464e2016-02-29 15:48:40 -0800809 fit_get_debug(fit, noffset, name, len);
810 return -1;
811 }
812
York Sun31e5add2016-02-29 15:48:41 -0800813 cell_len = len >> 2;
814 /* Use load64 to avoid compiling warning for 32-bit target */
815 while (cell_len--) {
816 load64 = (load64 << 32) | uimage_to_cpu(*cell);
817 cell++;
818 }
Michal Simekd31b40f2020-09-03 12:44:51 +0200819
820 if (len > sizeof(ulong) && (uint32_t)(load64 >> 32)) {
821 printf("Unsupported %s address size\n", name);
822 return -1;
823 }
824
York Sun31e5add2016-02-29 15:48:41 -0800825 *load = (ulong)load64;
York Sun2b0464e2016-02-29 15:48:40 -0800826
827 return 0;
828}
Simon Glassda0af362013-05-07 06:11:53 +0000829/**
830 * fit_image_get_load() - get load addr property for given component image node
831 * @fit: pointer to the FIT format image header
832 * @noffset: component image node offset
833 * @load: pointer to the uint32_t, will hold load address
834 *
835 * fit_image_get_load() finds load address property in a given component
836 * image node. If the property is found, its value is returned to the caller.
837 *
838 * returns:
839 * 0, on success
840 * -1, on failure
841 */
842int fit_image_get_load(const void *fit, int noffset, ulong *load)
843{
York Sun2b0464e2016-02-29 15:48:40 -0800844 return fit_image_get_address(fit, noffset, FIT_LOAD_PROP, load);
Simon Glassda0af362013-05-07 06:11:53 +0000845}
846
847/**
848 * fit_image_get_entry() - get entry point address property
849 * @fit: pointer to the FIT format image header
850 * @noffset: component image node offset
851 * @entry: pointer to the uint32_t, will hold entry point address
852 *
853 * This gets the entry point address property for a given component image
854 * node.
855 *
856 * fit_image_get_entry() finds entry point address property in a given
857 * component image node. If the property is found, its value is returned
858 * to the caller.
859 *
860 * returns:
861 * 0, on success
862 * -1, on failure
863 */
864int fit_image_get_entry(const void *fit, int noffset, ulong *entry)
865{
York Sun2b0464e2016-02-29 15:48:40 -0800866 return fit_image_get_address(fit, noffset, FIT_ENTRY_PROP, entry);
Simon Glassda0af362013-05-07 06:11:53 +0000867}
868
869/**
870 * fit_image_get_data - get data property and its size for a given component image node
871 * @fit: pointer to the FIT format image header
872 * @noffset: component image node offset
873 * @data: double pointer to void, will hold data property's data address
874 * @size: pointer to size_t, will hold data property's data size
875 *
876 * fit_image_get_data() finds data property in a given component image node.
877 * If the property is found its data start address and size are returned to
878 * the caller.
879 *
880 * returns:
881 * 0, on success
882 * -1, on failure
883 */
884int fit_image_get_data(const void *fit, int noffset,
885 const void **data, size_t *size)
886{
887 int len;
888
889 *data = fdt_getprop(fit, noffset, FIT_DATA_PROP, &len);
890 if (*data == NULL) {
891 fit_get_debug(fit, noffset, FIT_DATA_PROP, len);
892 *size = 0;
893 return -1;
894 }
895
896 *size = len;
897 return 0;
898}
899
900/**
tomas.melin@vaisala.com45ee7902017-01-13 13:20:14 +0200901 * Get 'data-offset' property from a given image node.
902 *
903 * @fit: pointer to the FIT image header
904 * @noffset: component image node offset
905 * @data_offset: holds the data-offset property
906 *
907 * returns:
908 * 0, on success
909 * -ENOENT if the property could not be found
910 */
911int fit_image_get_data_offset(const void *fit, int noffset, int *data_offset)
912{
913 const fdt32_t *val;
914
915 val = fdt_getprop(fit, noffset, FIT_DATA_OFFSET_PROP, NULL);
916 if (!val)
917 return -ENOENT;
918
919 *data_offset = fdt32_to_cpu(*val);
920
921 return 0;
922}
923
924/**
Peng Fan8876c7e2017-12-05 13:20:59 +0800925 * Get 'data-position' property from a given image node.
926 *
927 * @fit: pointer to the FIT image header
928 * @noffset: component image node offset
929 * @data_position: holds the data-position property
930 *
931 * returns:
932 * 0, on success
933 * -ENOENT if the property could not be found
934 */
935int fit_image_get_data_position(const void *fit, int noffset,
936 int *data_position)
937{
938 const fdt32_t *val;
939
940 val = fdt_getprop(fit, noffset, FIT_DATA_POSITION_PROP, NULL);
941 if (!val)
942 return -ENOENT;
943
944 *data_position = fdt32_to_cpu(*val);
945
946 return 0;
947}
948
949/**
tomas.melin@vaisala.com45ee7902017-01-13 13:20:14 +0200950 * Get 'data-size' property from a given image node.
951 *
952 * @fit: pointer to the FIT image header
953 * @noffset: component image node offset
954 * @data_size: holds the data-size property
955 *
956 * returns:
957 * 0, on success
958 * -ENOENT if the property could not be found
959 */
960int fit_image_get_data_size(const void *fit, int noffset, int *data_size)
961{
962 const fdt32_t *val;
963
964 val = fdt_getprop(fit, noffset, FIT_DATA_SIZE_PROP, NULL);
965 if (!val)
966 return -ENOENT;
967
968 *data_size = fdt32_to_cpu(*val);
969
970 return 0;
971}
972
973/**
Philippe Reynes3d964702019-12-18 18:25:42 +0100974 * Get 'data-size-unciphered' property from a given image node.
975 *
976 * @fit: pointer to the FIT image header
977 * @noffset: component image node offset
978 * @data_size: holds the data-size property
979 *
980 * returns:
981 * 0, on success
982 * -ENOENT if the property could not be found
983 */
984int fit_image_get_data_size_unciphered(const void *fit, int noffset,
985 size_t *data_size)
986{
987 const fdt32_t *val;
988
989 val = fdt_getprop(fit, noffset, "data-size-unciphered", NULL);
990 if (!val)
991 return -ENOENT;
992
993 *data_size = (size_t)fdt32_to_cpu(*val);
994
995 return 0;
996}
997
998/**
Kelvin Cheung186cc992018-05-19 18:21:37 +0800999 * fit_image_get_data_and_size - get data and its size including
1000 * both embedded and external data
1001 * @fit: pointer to the FIT format image header
1002 * @noffset: component image node offset
1003 * @data: double pointer to void, will hold data property's data address
1004 * @size: pointer to size_t, will hold data property's data size
1005 *
1006 * fit_image_get_data_and_size() finds data and its size including
1007 * both embedded and external data. If the property is found
1008 * its data start address and size are returned to the caller.
1009 *
1010 * returns:
1011 * 0, on success
1012 * otherwise, on failure
1013 */
1014int fit_image_get_data_and_size(const void *fit, int noffset,
1015 const void **data, size_t *size)
1016{
1017 bool external_data = false;
1018 int offset;
1019 int len;
1020 int ret;
1021
1022 if (!fit_image_get_data_position(fit, noffset, &offset)) {
1023 external_data = true;
1024 } else if (!fit_image_get_data_offset(fit, noffset, &offset)) {
1025 external_data = true;
1026 /*
1027 * For FIT with external data, figure out where
1028 * the external images start. This is the base
1029 * for the data-offset properties in each image.
1030 */
1031 offset += ((fdt_totalsize(fit) + 3) & ~3);
1032 }
1033
1034 if (external_data) {
1035 debug("External Data\n");
1036 ret = fit_image_get_data_size(fit, noffset, &len);
Heinrich Schuchardt0be986f2020-03-11 21:51:08 +01001037 if (!ret) {
1038 *data = fit + offset;
1039 *size = len;
1040 }
Kelvin Cheung186cc992018-05-19 18:21:37 +08001041 } else {
1042 ret = fit_image_get_data(fit, noffset, data, size);
1043 }
1044
1045 return ret;
1046}
1047
1048/**
Simon Glassda0af362013-05-07 06:11:53 +00001049 * fit_image_hash_get_algo - get hash algorithm name
1050 * @fit: pointer to the FIT format image header
1051 * @noffset: hash node offset
1052 * @algo: double pointer to char, will hold pointer to the algorithm name
1053 *
1054 * fit_image_hash_get_algo() finds hash algorithm property in a given hash node.
1055 * If the property is found its data start address is returned to the caller.
1056 *
1057 * returns:
1058 * 0, on success
1059 * -1, on failure
1060 */
1061int fit_image_hash_get_algo(const void *fit, int noffset, char **algo)
1062{
1063 int len;
1064
1065 *algo = (char *)fdt_getprop(fit, noffset, FIT_ALGO_PROP, &len);
1066 if (*algo == NULL) {
1067 fit_get_debug(fit, noffset, FIT_ALGO_PROP, len);
1068 return -1;
1069 }
1070
1071 return 0;
1072}
1073
1074/**
1075 * fit_image_hash_get_value - get hash value and length
1076 * @fit: pointer to the FIT format image header
1077 * @noffset: hash node offset
1078 * @value: double pointer to uint8_t, will hold address of a hash value data
1079 * @value_len: pointer to an int, will hold hash data length
1080 *
1081 * fit_image_hash_get_value() finds hash value property in a given hash node.
1082 * If the property is found its data start address and size are returned to
1083 * the caller.
1084 *
1085 * returns:
1086 * 0, on success
1087 * -1, on failure
1088 */
1089int fit_image_hash_get_value(const void *fit, int noffset, uint8_t **value,
1090 int *value_len)
1091{
1092 int len;
1093
1094 *value = (uint8_t *)fdt_getprop(fit, noffset, FIT_VALUE_PROP, &len);
1095 if (*value == NULL) {
1096 fit_get_debug(fit, noffset, FIT_VALUE_PROP, len);
1097 *value_len = 0;
1098 return -1;
1099 }
1100
1101 *value_len = len;
1102 return 0;
1103}
1104
Simon Glassda0af362013-05-07 06:11:53 +00001105/**
1106 * fit_image_hash_get_ignore - get hash ignore flag
1107 * @fit: pointer to the FIT format image header
1108 * @noffset: hash node offset
1109 * @ignore: pointer to an int, will hold hash ignore flag
1110 *
1111 * fit_image_hash_get_ignore() finds hash ignore property in a given hash node.
1112 * If the property is found and non-zero, the hash algorithm is not verified by
1113 * u-boot automatically.
1114 *
1115 * returns:
1116 * 0, on ignore not found
1117 * value, on ignore found
1118 */
Simon Glassf0fd5112013-05-07 06:11:58 +00001119static int fit_image_hash_get_ignore(const void *fit, int noffset, int *ignore)
Simon Glassda0af362013-05-07 06:11:53 +00001120{
1121 int len;
1122 int *value;
1123
1124 value = (int *)fdt_getprop(fit, noffset, FIT_IGNORE_PROP, &len);
1125 if (value == NULL || len != sizeof(int))
1126 *ignore = 0;
1127 else
1128 *ignore = *value;
1129
1130 return 0;
1131}
Simon Glassda0af362013-05-07 06:11:53 +00001132
Philippe Reynes3148e422019-12-18 18:25:41 +01001133/**
1134 * fit_image_cipher_get_algo - get cipher algorithm name
1135 * @fit: pointer to the FIT format image header
1136 * @noffset: cipher node offset
1137 * @algo: double pointer to char, will hold pointer to the algorithm name
1138 *
1139 * fit_image_cipher_get_algo() finds cipher algorithm property in a given
1140 * cipher node. If the property is found its data start address is returned
1141 * to the caller.
1142 *
1143 * returns:
1144 * 0, on success
1145 * -1, on failure
1146 */
1147int fit_image_cipher_get_algo(const void *fit, int noffset, char **algo)
1148{
1149 int len;
1150
1151 *algo = (char *)fdt_getprop(fit, noffset, FIT_ALGO_PROP, &len);
1152 if (!*algo) {
1153 fit_get_debug(fit, noffset, FIT_ALGO_PROP, len);
1154 return -1;
1155 }
1156
1157 return 0;
1158}
1159
Simon Glass5b539a02016-02-24 09:14:42 -07001160ulong fit_get_end(const void *fit)
1161{
1162 return map_to_sysmem((void *)(fit + fdt_totalsize(fit)));
1163}
1164
Simon Glassda0af362013-05-07 06:11:53 +00001165/**
1166 * fit_set_timestamp - set node timestamp property
1167 * @fit: pointer to the FIT format image header
1168 * @noffset: node offset
1169 * @timestamp: timestamp value to be set
1170 *
1171 * fit_set_timestamp() attempts to set timestamp property in the requested
1172 * node and returns operation status to the caller.
1173 *
1174 * returns:
1175 * 0, on success
Simon Glassaf2f9d52014-06-02 22:04:51 -06001176 * -ENOSPC if no space in device tree, -1 for other error
Simon Glassda0af362013-05-07 06:11:53 +00001177 */
1178int fit_set_timestamp(void *fit, int noffset, time_t timestamp)
1179{
1180 uint32_t t;
1181 int ret;
1182
1183 t = cpu_to_uimage(timestamp);
1184 ret = fdt_setprop(fit, noffset, FIT_TIMESTAMP_PROP, &t,
1185 sizeof(uint32_t));
1186 if (ret) {
Simon Glasse4016fa2016-05-01 13:55:37 -06001187 debug("Can't set '%s' property for '%s' node (%s)\n",
1188 FIT_TIMESTAMP_PROP, fit_get_name(fit, noffset, NULL),
1189 fdt_strerror(ret));
Simon Glassaf2f9d52014-06-02 22:04:51 -06001190 return ret == -FDT_ERR_NOSPACE ? -ENOSPC : -1;
Simon Glassda0af362013-05-07 06:11:53 +00001191 }
1192
1193 return 0;
1194}
1195
1196/**
1197 * calculate_hash - calculate and return hash for provided input data
1198 * @data: pointer to the input data
1199 * @data_len: data length
1200 * @algo: requested hash algorithm
1201 * @value: pointer to the char, will hold hash value data (caller must
1202 * allocate enough free space)
1203 * value_len: length of the calculated hash
1204 *
1205 * calculate_hash() computes input data hash according to the requested
1206 * algorithm.
1207 * Resulting hash value is placed in caller provided 'value' buffer, length
1208 * of the calculated hash is returned via value_len pointer argument.
1209 *
1210 * returns:
1211 * 0, on success
1212 * -1, when algo is unsupported
1213 */
Simon Glass10a1eca2013-05-07 06:11:54 +00001214int calculate_hash(const void *data, int data_len, const char *algo,
Simon Glassda0af362013-05-07 06:11:53 +00001215 uint8_t *value, int *value_len)
1216{
Simon Glass82d94532013-05-08 08:05:59 +00001217 if (IMAGE_ENABLE_CRC32 && strcmp(algo, "crc32") == 0) {
Simon Glassda0af362013-05-07 06:11:53 +00001218 *((uint32_t *)value) = crc32_wd(0, data, data_len,
1219 CHUNKSZ_CRC32);
1220 *((uint32_t *)value) = cpu_to_uimage(*((uint32_t *)value));
1221 *value_len = 4;
Simon Glassc2da20f2021-07-14 17:05:34 -05001222 } else if (CONFIG_IS_ENABLED(SHA1) && strcmp(algo, "sha1") == 0) {
Simon Glassda0af362013-05-07 06:11:53 +00001223 sha1_csum_wd((unsigned char *)data, data_len,
1224 (unsigned char *)value, CHUNKSZ_SHA1);
1225 *value_len = 20;
Simon Glass64a46242021-07-14 17:05:35 -05001226 } else if (CONFIG_IS_ENABLED(SHA256) && strcmp(algo, "sha256") == 0) {
Heiko Schocher8ae33802014-03-03 12:19:25 +01001227 sha256_csum_wd((unsigned char *)data, data_len,
1228 (unsigned char *)value, CHUNKSZ_SHA256);
1229 *value_len = SHA256_SUM_LEN;
Simon Glass64a46242021-07-14 17:05:35 -05001230 } else if (CONFIG_IS_ENABLED(SHA384) && strcmp(algo, "sha384") == 0) {
Reuben Dowle1908fd92020-04-16 17:36:52 +12001231 sha384_csum_wd((unsigned char *)data, data_len,
1232 (unsigned char *)value, CHUNKSZ_SHA384);
1233 *value_len = SHA384_SUM_LEN;
Simon Glass64a46242021-07-14 17:05:35 -05001234 } else if (CONFIG_IS_ENABLED(SHA512) && strcmp(algo, "sha512") == 0) {
Reuben Dowle1908fd92020-04-16 17:36:52 +12001235 sha512_csum_wd((unsigned char *)data, data_len,
1236 (unsigned char *)value, CHUNKSZ_SHA512);
1237 *value_len = SHA512_SUM_LEN;
Simon Glass82d94532013-05-08 08:05:59 +00001238 } else if (IMAGE_ENABLE_MD5 && strcmp(algo, "md5") == 0) {
Simon Glassda0af362013-05-07 06:11:53 +00001239 md5_wd((unsigned char *)data, data_len, value, CHUNKSZ_MD5);
1240 *value_len = 16;
1241 } else {
1242 debug("Unsupported hash alogrithm\n");
1243 return -1;
1244 }
1245 return 0;
1246}
1247
Simon Glassf0fd5112013-05-07 06:11:58 +00001248static int fit_image_check_hash(const void *fit, int noffset, const void *data,
1249 size_t size, char **err_msgp)
1250{
1251 uint8_t value[FIT_MAX_HASH_LEN];
1252 int value_len;
1253 char *algo;
1254 uint8_t *fit_value;
1255 int fit_value_len;
1256 int ignore;
1257
1258 *err_msgp = NULL;
1259
1260 if (fit_image_hash_get_algo(fit, noffset, &algo)) {
Simon Glass8f3aa462013-05-07 06:11:59 +00001261 *err_msgp = "Can't get hash algo property";
Simon Glassf0fd5112013-05-07 06:11:58 +00001262 return -1;
1263 }
1264 printf("%s", algo);
1265
1266 if (IMAGE_ENABLE_IGNORE) {
1267 fit_image_hash_get_ignore(fit, noffset, &ignore);
1268 if (ignore) {
1269 printf("-skipped ");
1270 return 0;
1271 }
1272 }
1273
1274 if (fit_image_hash_get_value(fit, noffset, &fit_value,
1275 &fit_value_len)) {
Simon Glass8f3aa462013-05-07 06:11:59 +00001276 *err_msgp = "Can't get hash value property";
Simon Glassf0fd5112013-05-07 06:11:58 +00001277 return -1;
1278 }
1279
1280 if (calculate_hash(data, size, algo, value, &value_len)) {
Simon Glass8f3aa462013-05-07 06:11:59 +00001281 *err_msgp = "Unsupported hash algorithm";
Simon Glassf0fd5112013-05-07 06:11:58 +00001282 return -1;
1283 }
1284
1285 if (value_len != fit_value_len) {
Simon Glass8f3aa462013-05-07 06:11:59 +00001286 *err_msgp = "Bad hash value len";
Simon Glassf0fd5112013-05-07 06:11:58 +00001287 return -1;
1288 } else if (memcmp(value, fit_value, value_len) != 0) {
Simon Glass8f3aa462013-05-07 06:11:59 +00001289 *err_msgp = "Bad hash value";
Simon Glassf0fd5112013-05-07 06:11:58 +00001290 return -1;
1291 }
1292
1293 return 0;
1294}
1295
Jun Nieadf5d1c2018-02-27 16:55:58 +08001296int fit_image_verify_with_data(const void *fit, int image_noffset,
1297 const void *data, size_t size)
Simon Glassda0af362013-05-07 06:11:53 +00001298{
Simon Glassfbabc0f2013-06-13 15:10:01 -07001299 int noffset = 0;
Simon Glassda0af362013-05-07 06:11:53 +00001300 char *err_msg = "";
Simon Glassfbabc0f2013-06-13 15:10:01 -07001301 int verify_all = 1;
1302 int ret;
Simon Glassda0af362013-05-07 06:11:53 +00001303
Simon Glassfbabc0f2013-06-13 15:10:01 -07001304 /* Verify all required signatures */
AKASHI Takahiro2223c7d2020-02-21 15:12:55 +09001305 if (FIT_IMAGE_ENABLE_VERIFY &&
Simon Glassfbabc0f2013-06-13 15:10:01 -07001306 fit_image_verify_required_sigs(fit, image_noffset, data, size,
1307 gd_fdt_blob(), &verify_all)) {
1308 err_msg = "Unable to verify required signature";
1309 goto error;
1310 }
1311
Simon Glassda0af362013-05-07 06:11:53 +00001312 /* Process all hash subnodes of the component image node */
Simon Glass499c29e2016-10-02 17:59:29 -06001313 fdt_for_each_subnode(noffset, fit, image_noffset) {
Simon Glassf0fd5112013-05-07 06:11:58 +00001314 const char *name = fit_get_name(fit, noffset, NULL);
Simon Glassda0af362013-05-07 06:11:53 +00001315
Simon Glassf0fd5112013-05-07 06:11:58 +00001316 /*
1317 * Check subnode name, must be equal to "hash".
1318 * Multiple hash nodes require unique unit node
Andre Przywara3234ecd2017-12-04 02:05:10 +00001319 * names, e.g. hash-1, hash-2, etc.
Simon Glassf0fd5112013-05-07 06:11:58 +00001320 */
1321 if (!strncmp(name, FIT_HASH_NODENAME,
1322 strlen(FIT_HASH_NODENAME))) {
1323 if (fit_image_check_hash(fit, noffset, data, size,
1324 &err_msg))
Simon Glassda0af362013-05-07 06:11:53 +00001325 goto error;
Simon Glassf0fd5112013-05-07 06:11:58 +00001326 puts("+ ");
AKASHI Takahiro2223c7d2020-02-21 15:12:55 +09001327 } else if (FIT_IMAGE_ENABLE_VERIFY && verify_all &&
Simon Glassfbabc0f2013-06-13 15:10:01 -07001328 !strncmp(name, FIT_SIG_NODENAME,
1329 strlen(FIT_SIG_NODENAME))) {
1330 ret = fit_image_check_sig(fit, noffset, data,
1331 size, -1, &err_msg);
Simon Glass51026aa2016-02-24 09:14:43 -07001332
1333 /*
1334 * Show an indication on failure, but do not return
1335 * an error. Only keys marked 'required' can cause
1336 * an image validation failure. See the call to
1337 * fit_image_verify_required_sigs() above.
1338 */
1339 if (ret)
Simon Glassfbabc0f2013-06-13 15:10:01 -07001340 puts("- ");
1341 else
1342 puts("+ ");
Simon Glassda0af362013-05-07 06:11:53 +00001343 }
1344 }
1345
1346 if (noffset == -FDT_ERR_TRUNCATED || noffset == -FDT_ERR_BADSTRUCTURE) {
Simon Glass8f3aa462013-05-07 06:11:59 +00001347 err_msg = "Corrupted or truncated tree";
Simon Glassda0af362013-05-07 06:11:53 +00001348 goto error;
1349 }
1350
1351 return 1;
1352
1353error:
Simon Glass8f3aa462013-05-07 06:11:59 +00001354 printf(" error!\n%s for '%s' hash node in '%s' image node\n",
Simon Glassda0af362013-05-07 06:11:53 +00001355 err_msg, fit_get_name(fit, noffset, NULL),
1356 fit_get_name(fit, image_noffset, NULL));
1357 return 0;
1358}
1359
1360/**
Jun Nieadf5d1c2018-02-27 16:55:58 +08001361 * fit_image_verify - verify data integrity
1362 * @fit: pointer to the FIT format image header
1363 * @image_noffset: component image node offset
1364 *
1365 * fit_image_verify() goes over component image hash nodes,
1366 * re-calculates each data hash and compares with the value stored in hash
1367 * node.
1368 *
1369 * returns:
1370 * 1, if all hashes are valid
1371 * 0, otherwise (or on error)
1372 */
1373int fit_image_verify(const void *fit, int image_noffset)
1374{
Simon Glass44338902021-02-15 17:08:06 -07001375 const char *name = fit_get_name(fit, image_noffset, NULL);
Jun Nieadf5d1c2018-02-27 16:55:58 +08001376 const void *data;
1377 size_t size;
Jun Nieadf5d1c2018-02-27 16:55:58 +08001378 char *err_msg = "";
1379
Simon Glass3d702182021-07-05 16:32:56 -06001380 if (IS_ENABLED(CONFIG_FIT_SIGNATURE) && strchr(name, '@')) {
Simon Glass44338902021-02-15 17:08:06 -07001381 /*
1382 * We don't support this since libfdt considers names with the
1383 * name root but different @ suffix to be equal
1384 */
1385 err_msg = "Node name contains @";
1386 goto err;
1387 }
Jun Nieadf5d1c2018-02-27 16:55:58 +08001388 /* Get image data and data length */
Kelvin Cheung186cc992018-05-19 18:21:37 +08001389 if (fit_image_get_data_and_size(fit, image_noffset, &data, &size)) {
Jun Nieadf5d1c2018-02-27 16:55:58 +08001390 err_msg = "Can't get image data/size";
Simon Glass44338902021-02-15 17:08:06 -07001391 goto err;
Jun Nieadf5d1c2018-02-27 16:55:58 +08001392 }
1393
1394 return fit_image_verify_with_data(fit, image_noffset, data, size);
Simon Glass44338902021-02-15 17:08:06 -07001395
1396err:
1397 printf("error!\n%s in '%s' image node\n", err_msg,
1398 fit_get_name(fit, image_noffset, NULL));
1399 return 0;
Jun Nieadf5d1c2018-02-27 16:55:58 +08001400}
1401
1402/**
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -05001403 * fit_all_image_verify - verify data integrity for all images
Simon Glassda0af362013-05-07 06:11:53 +00001404 * @fit: pointer to the FIT format image header
1405 *
Simon Glass7428ad12013-05-07 06:11:57 +00001406 * fit_all_image_verify() goes over all images in the FIT and
Simon Glassda0af362013-05-07 06:11:53 +00001407 * for every images checks if all it's hashes are valid.
1408 *
1409 * returns:
1410 * 1, if all hashes of all images are valid
1411 * 0, otherwise (or on error)
1412 */
Simon Glass7428ad12013-05-07 06:11:57 +00001413int fit_all_image_verify(const void *fit)
Simon Glassda0af362013-05-07 06:11:53 +00001414{
1415 int images_noffset;
1416 int noffset;
1417 int ndepth;
1418 int count;
1419
1420 /* Find images parent node offset */
1421 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
1422 if (images_noffset < 0) {
1423 printf("Can't find images parent node '%s' (%s)\n",
1424 FIT_IMAGES_PATH, fdt_strerror(images_noffset));
1425 return 0;
1426 }
1427
1428 /* Process all image subnodes, check hashes for each */
1429 printf("## Checking hash(es) for FIT Image at %08lx ...\n",
1430 (ulong)fit);
1431 for (ndepth = 0, count = 0,
1432 noffset = fdt_next_node(fit, images_noffset, &ndepth);
1433 (noffset >= 0) && (ndepth > 0);
1434 noffset = fdt_next_node(fit, noffset, &ndepth)) {
1435 if (ndepth == 1) {
1436 /*
1437 * Direct child node of the images parent node,
1438 * i.e. component image node.
1439 */
Simon Glasse3ee2fb2016-02-22 22:55:43 -07001440 printf(" Hash(es) for Image %u (%s): ", count,
Simon Glassda0af362013-05-07 06:11:53 +00001441 fit_get_name(fit, noffset, NULL));
Simon Glasse3ee2fb2016-02-22 22:55:43 -07001442 count++;
Simon Glassda0af362013-05-07 06:11:53 +00001443
Simon Glass7428ad12013-05-07 06:11:57 +00001444 if (!fit_image_verify(fit, noffset))
Simon Glassda0af362013-05-07 06:11:53 +00001445 return 0;
1446 printf("\n");
1447 }
1448 }
1449 return 1;
1450}
1451
Philippe Reynes3d964702019-12-18 18:25:42 +01001452static int fit_image_uncipher(const void *fit, int image_noffset,
1453 void **data, size_t *size)
1454{
1455 int cipher_noffset, ret;
1456 void *dst;
1457 size_t size_dst;
1458
1459 cipher_noffset = fdt_subnode_offset(fit, image_noffset,
1460 FIT_CIPHER_NODENAME);
1461 if (cipher_noffset < 0)
1462 return 0;
1463
1464 ret = fit_image_decrypt_data(fit, image_noffset, cipher_noffset,
1465 *data, *size, &dst, &size_dst);
1466 if (ret)
1467 goto out;
1468
1469 *data = dst;
1470 *size = size_dst;
1471
1472 out:
1473 return ret;
1474}
Philippe Reynes3d964702019-12-18 18:25:42 +01001475
Simon Glassda0af362013-05-07 06:11:53 +00001476/**
1477 * fit_image_check_os - check whether image node is of a given os type
1478 * @fit: pointer to the FIT format image header
1479 * @noffset: component image node offset
1480 * @os: requested image os
1481 *
1482 * fit_image_check_os() reads image os property and compares its numeric
1483 * id with the requested os. Comparison result is returned to the caller.
1484 *
1485 * returns:
1486 * 1 if image is of given os type
1487 * 0 otherwise (or on error)
1488 */
1489int fit_image_check_os(const void *fit, int noffset, uint8_t os)
1490{
1491 uint8_t image_os;
1492
1493 if (fit_image_get_os(fit, noffset, &image_os))
1494 return 0;
1495 return (os == image_os);
1496}
1497
1498/**
1499 * fit_image_check_arch - check whether image node is of a given arch
1500 * @fit: pointer to the FIT format image header
1501 * @noffset: component image node offset
1502 * @arch: requested imagearch
1503 *
1504 * fit_image_check_arch() reads image arch property and compares its numeric
1505 * id with the requested arch. Comparison result is returned to the caller.
1506 *
1507 * returns:
1508 * 1 if image is of given arch
1509 * 0 otherwise (or on error)
1510 */
1511int fit_image_check_arch(const void *fit, int noffset, uint8_t arch)
1512{
1513 uint8_t image_arch;
Alison Wang73818d52016-11-10 10:49:03 +08001514 int aarch32_support = 0;
1515
Simon Glass062e79f2021-03-15 18:11:12 +13001516 /* Let's assume that sandbox can load any architecture */
1517 if (IS_ENABLED(CONFIG_SANDBOX))
1518 return true;
1519
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +01001520 if (IS_ENABLED(CONFIG_ARM64_SUPPORT_AARCH32))
1521 aarch32_support = 1;
Simon Glassda0af362013-05-07 06:11:53 +00001522
1523 if (fit_image_get_arch(fit, noffset, &image_arch))
1524 return 0;
Simon Glass9d428302014-10-10 08:21:57 -06001525 return (arch == image_arch) ||
Alison Wang73818d52016-11-10 10:49:03 +08001526 (arch == IH_ARCH_I386 && image_arch == IH_ARCH_X86_64) ||
1527 (arch == IH_ARCH_ARM64 && image_arch == IH_ARCH_ARM &&
1528 aarch32_support);
Simon Glassda0af362013-05-07 06:11:53 +00001529}
1530
1531/**
1532 * fit_image_check_type - check whether image node is of a given type
1533 * @fit: pointer to the FIT format image header
1534 * @noffset: component image node offset
1535 * @type: requested image type
1536 *
1537 * fit_image_check_type() reads image type property and compares its numeric
1538 * id with the requested type. Comparison result is returned to the caller.
1539 *
1540 * returns:
1541 * 1 if image is of given type
1542 * 0 otherwise (or on error)
1543 */
1544int fit_image_check_type(const void *fit, int noffset, uint8_t type)
1545{
1546 uint8_t image_type;
1547
1548 if (fit_image_get_type(fit, noffset, &image_type))
1549 return 0;
1550 return (type == image_type);
1551}
1552
1553/**
1554 * fit_image_check_comp - check whether image node uses given compression
1555 * @fit: pointer to the FIT format image header
1556 * @noffset: component image node offset
1557 * @comp: requested image compression type
1558 *
1559 * fit_image_check_comp() reads image compression property and compares its
1560 * numeric id with the requested compression type. Comparison result is
1561 * returned to the caller.
1562 *
1563 * returns:
1564 * 1 if image uses requested compression
1565 * 0 otherwise (or on error)
1566 */
1567int fit_image_check_comp(const void *fit, int noffset, uint8_t comp)
1568{
1569 uint8_t image_comp;
1570
1571 if (fit_image_get_comp(fit, noffset, &image_comp))
1572 return 0;
1573 return (comp == image_comp);
1574}
1575
Simon Glassb823daa2021-02-15 17:08:12 -07001576/**
1577 * fdt_check_no_at() - Check for nodes whose names contain '@'
1578 *
1579 * This checks the parent node and all subnodes recursively
1580 *
1581 * @fit: FIT to check
1582 * @parent: Parent node to check
1583 * @return 0 if OK, -EADDRNOTAVAIL is a node has a name containing '@'
1584 */
1585static int fdt_check_no_at(const void *fit, int parent)
1586{
1587 const char *name;
1588 int node;
1589 int ret;
1590
1591 name = fdt_get_name(fit, parent, NULL);
1592 if (!name || strchr(name, '@'))
1593 return -EADDRNOTAVAIL;
1594
1595 fdt_for_each_subnode(node, fit, parent) {
1596 ret = fdt_check_no_at(fit, node);
1597 if (ret)
1598 return ret;
1599 }
1600
1601 return 0;
1602}
1603
Simon Glassd563c252021-02-15 17:08:09 -07001604int fit_check_format(const void *fit, ulong size)
Simon Glassda0af362013-05-07 06:11:53 +00001605{
Simon Glassd563c252021-02-15 17:08:09 -07001606 int ret;
1607
Heinrich Schuchardtef98b212021-01-13 02:09:12 +01001608 /* A FIT image must be a valid FDT */
Simon Glassd563c252021-02-15 17:08:09 -07001609 ret = fdt_check_header(fit);
1610 if (ret) {
1611 log_debug("Wrong FIT format: not a flattened device tree (err=%d)\n",
1612 ret);
1613 return -ENOEXEC;
Heinrich Schuchardtef98b212021-01-13 02:09:12 +01001614 }
1615
Simon Glass244705b2021-02-15 17:08:10 -07001616 if (CONFIG_IS_ENABLED(FIT_FULL_CHECK)) {
1617 /*
1618 * If we are not given the size, make do wtih calculating it.
1619 * This is not as secure, so we should consider a flag to
1620 * control this.
1621 */
1622 if (size == IMAGE_SIZE_INVAL)
1623 size = fdt_totalsize(fit);
1624 ret = fdt_check_full(fit, size);
Simon Glassb823daa2021-02-15 17:08:12 -07001625 if (ret)
1626 ret = -EINVAL;
1627
1628 /*
1629 * U-Boot stopped using unit addressed in 2017. Since libfdt
1630 * can match nodes ignoring any unit address, signature
1631 * verification can see the wrong node if one is inserted with
1632 * the same name as a valid node but with a unit address
1633 * attached. Protect against this by disallowing unit addresses.
1634 */
1635 if (!ret && CONFIG_IS_ENABLED(FIT_SIGNATURE)) {
1636 ret = fdt_check_no_at(fit, 0);
Simon Glass244705b2021-02-15 17:08:10 -07001637
Simon Glassb823daa2021-02-15 17:08:12 -07001638 if (ret) {
1639 log_debug("FIT check error %d\n", ret);
1640 return ret;
1641 }
1642 }
Simon Glass244705b2021-02-15 17:08:10 -07001643 if (ret) {
1644 log_debug("FIT check error %d\n", ret);
Simon Glassb823daa2021-02-15 17:08:12 -07001645 return ret;
Simon Glass244705b2021-02-15 17:08:10 -07001646 }
1647 }
1648
Simon Glassda0af362013-05-07 06:11:53 +00001649 /* mandatory / node 'description' property */
Simon Glassd563c252021-02-15 17:08:09 -07001650 if (!fdt_getprop(fit, 0, FIT_DESC_PROP, NULL)) {
1651 log_debug("Wrong FIT format: no description\n");
1652 return -ENOMSG;
Simon Glassda0af362013-05-07 06:11:53 +00001653 }
1654
1655 if (IMAGE_ENABLE_TIMESTAMP) {
1656 /* mandatory / node 'timestamp' property */
Simon Glassd563c252021-02-15 17:08:09 -07001657 if (!fdt_getprop(fit, 0, FIT_TIMESTAMP_PROP, NULL)) {
1658 log_debug("Wrong FIT format: no timestamp\n");
Simon Glassb641de82021-02-24 08:50:32 -05001659 return -EBADMSG;
Simon Glassda0af362013-05-07 06:11:53 +00001660 }
1661 }
1662
1663 /* mandatory subimages parent '/images' node */
1664 if (fdt_path_offset(fit, FIT_IMAGES_PATH) < 0) {
Simon Glassd563c252021-02-15 17:08:09 -07001665 log_debug("Wrong FIT format: no images parent node\n");
1666 return -ENOENT;
Simon Glassda0af362013-05-07 06:11:53 +00001667 }
1668
Simon Glassd563c252021-02-15 17:08:09 -07001669 return 0;
Simon Glassda0af362013-05-07 06:11:53 +00001670}
1671
Simon Glassda0af362013-05-07 06:11:53 +00001672/**
1673 * fit_conf_find_compat
1674 * @fit: pointer to the FIT format image header
1675 * @fdt: pointer to the device tree to compare against
1676 *
1677 * fit_conf_find_compat() attempts to find the configuration whose fdt is the
1678 * most compatible with the passed in device tree.
1679 *
1680 * Example:
1681 *
1682 * / o image-tree
1683 * |-o images
Andre Przywara3234ecd2017-12-04 02:05:10 +00001684 * | |-o fdt-1
1685 * | |-o fdt-2
Simon Glassda0af362013-05-07 06:11:53 +00001686 * |
1687 * |-o configurations
Andre Przywara3234ecd2017-12-04 02:05:10 +00001688 * |-o config-1
1689 * | |-fdt = fdt-1
Simon Glassda0af362013-05-07 06:11:53 +00001690 * |
Andre Przywara3234ecd2017-12-04 02:05:10 +00001691 * |-o config-2
1692 * |-fdt = fdt-2
Simon Glassda0af362013-05-07 06:11:53 +00001693 *
1694 * / o U-Boot fdt
1695 * |-compatible = "foo,bar", "bim,bam"
1696 *
1697 * / o kernel fdt1
1698 * |-compatible = "foo,bar",
1699 *
1700 * / o kernel fdt2
1701 * |-compatible = "bim,bam", "baz,biz"
1702 *
1703 * Configuration 1 would be picked because the first string in U-Boot's
1704 * compatible list, "foo,bar", matches a compatible string in the root of fdt1.
1705 * "bim,bam" in fdt2 matches the second string which isn't as good as fdt1.
1706 *
Julius Werner4e823522019-07-24 19:37:56 -07001707 * As an optimization, the compatible property from the FDT's root node can be
1708 * copied into the configuration node in the FIT image. This is required to
1709 * match configurations with compressed FDTs.
1710 *
Simon Glassda0af362013-05-07 06:11:53 +00001711 * returns:
1712 * offset to the configuration to use if one was found
1713 * -1 otherwise
1714 */
1715int fit_conf_find_compat(const void *fit, const void *fdt)
1716{
1717 int ndepth = 0;
1718 int noffset, confs_noffset, images_noffset;
1719 const void *fdt_compat;
1720 int fdt_compat_len;
1721 int best_match_offset = 0;
1722 int best_match_pos = 0;
1723
1724 confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH);
1725 images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
1726 if (confs_noffset < 0 || images_noffset < 0) {
1727 debug("Can't find configurations or images nodes.\n");
1728 return -1;
1729 }
1730
1731 fdt_compat = fdt_getprop(fdt, 0, "compatible", &fdt_compat_len);
1732 if (!fdt_compat) {
1733 debug("Fdt for comparison has no \"compatible\" property.\n");
1734 return -1;
1735 }
1736
1737 /*
1738 * Loop over the configurations in the FIT image.
1739 */
1740 for (noffset = fdt_next_node(fit, confs_noffset, &ndepth);
1741 (noffset >= 0) && (ndepth > 0);
1742 noffset = fdt_next_node(fit, noffset, &ndepth)) {
Julius Werner4e823522019-07-24 19:37:56 -07001743 const void *fdt;
Simon Glassda0af362013-05-07 06:11:53 +00001744 const char *kfdt_name;
Julius Werner4e823522019-07-24 19:37:56 -07001745 int kfdt_noffset, compat_noffset;
Simon Glassda0af362013-05-07 06:11:53 +00001746 const char *cur_fdt_compat;
1747 int len;
Julius Werner4e823522019-07-24 19:37:56 -07001748 size_t sz;
Simon Glassda0af362013-05-07 06:11:53 +00001749 int i;
1750
1751 if (ndepth > 1)
1752 continue;
1753
Julius Werner4e823522019-07-24 19:37:56 -07001754 /* If there's a compat property in the config node, use that. */
1755 if (fdt_getprop(fit, noffset, "compatible", NULL)) {
1756 fdt = fit; /* search in FIT image */
1757 compat_noffset = noffset; /* search under config node */
1758 } else { /* Otherwise extract it from the kernel FDT. */
1759 kfdt_name = fdt_getprop(fit, noffset, "fdt", &len);
1760 if (!kfdt_name) {
1761 debug("No fdt property found.\n");
1762 continue;
1763 }
1764 kfdt_noffset = fdt_subnode_offset(fit, images_noffset,
1765 kfdt_name);
1766 if (kfdt_noffset < 0) {
1767 debug("No image node named \"%s\" found.\n",
1768 kfdt_name);
1769 continue;
1770 }
Julius Werner97b09cd2019-07-24 19:37:55 -07001771
Julius Werner4e823522019-07-24 19:37:56 -07001772 if (!fit_image_check_comp(fit, kfdt_noffset,
1773 IH_COMP_NONE)) {
1774 debug("Can't extract compat from \"%s\" "
1775 "(compressed)\n", kfdt_name);
1776 continue;
1777 }
Julius Werner97b09cd2019-07-24 19:37:55 -07001778
Julius Werner4e823522019-07-24 19:37:56 -07001779 /* search in this config's kernel FDT */
John Keeping645a3122021-06-25 17:58:04 +01001780 if (fit_image_get_data_and_size(fit, kfdt_noffset,
1781 &fdt, &sz)) {
Julius Werner4e823522019-07-24 19:37:56 -07001782 debug("Failed to get fdt \"%s\".\n", kfdt_name);
1783 continue;
1784 }
1785
1786 compat_noffset = 0; /* search kFDT under root node */
Simon Glassda0af362013-05-07 06:11:53 +00001787 }
1788
1789 len = fdt_compat_len;
1790 cur_fdt_compat = fdt_compat;
1791 /*
1792 * Look for a match for each U-Boot compatibility string in
Julius Werner4e823522019-07-24 19:37:56 -07001793 * turn in the compat string property.
Simon Glassda0af362013-05-07 06:11:53 +00001794 */
1795 for (i = 0; len > 0 &&
1796 (!best_match_offset || best_match_pos > i); i++) {
1797 int cur_len = strlen(cur_fdt_compat) + 1;
1798
Julius Werner4e823522019-07-24 19:37:56 -07001799 if (!fdt_node_check_compatible(fdt, compat_noffset,
Simon Glassda0af362013-05-07 06:11:53 +00001800 cur_fdt_compat)) {
1801 best_match_offset = noffset;
1802 best_match_pos = i;
1803 break;
1804 }
1805 len -= cur_len;
1806 cur_fdt_compat += cur_len;
1807 }
1808 }
1809 if (!best_match_offset) {
1810 debug("No match found.\n");
1811 return -1;
1812 }
1813
1814 return best_match_offset;
1815}
1816
Simon Glassda0af362013-05-07 06:11:53 +00001817int fit_conf_get_node(const void *fit, const char *conf_uname)
1818{
1819 int noffset, confs_noffset;
1820 int len;
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03001821 const char *s;
1822 char *conf_uname_copy = NULL;
Simon Glassda0af362013-05-07 06:11:53 +00001823
1824 confs_noffset = fdt_path_offset(fit, FIT_CONFS_PATH);
1825 if (confs_noffset < 0) {
1826 debug("Can't find configurations parent node '%s' (%s)\n",
1827 FIT_CONFS_PATH, fdt_strerror(confs_noffset));
1828 return confs_noffset;
1829 }
1830
1831 if (conf_uname == NULL) {
1832 /* get configuration unit name from the default property */
1833 debug("No configuration specified, trying default...\n");
Sebastian Reichel0890f942021-01-04 20:48:04 +01001834 if (!host_build() && IS_ENABLED(CONFIG_MULTI_DTB_FIT)) {
1835 noffset = fit_find_config_node(fit);
1836 if (noffset < 0)
1837 return noffset;
1838 conf_uname = fdt_get_name(fit, noffset, NULL);
1839 } else {
1840 conf_uname = (char *)fdt_getprop(fit, confs_noffset,
1841 FIT_DEFAULT_PROP, &len);
1842 if (conf_uname == NULL) {
1843 fit_get_debug(fit, confs_noffset, FIT_DEFAULT_PROP,
1844 len);
1845 return len;
1846 }
Simon Glassda0af362013-05-07 06:11:53 +00001847 }
1848 debug("Found default configuration: '%s'\n", conf_uname);
1849 }
1850
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03001851 s = strchr(conf_uname, '#');
1852 if (s) {
1853 len = s - conf_uname;
1854 conf_uname_copy = malloc(len + 1);
1855 if (!conf_uname_copy) {
1856 debug("Can't allocate uname copy: '%s'\n",
1857 conf_uname);
1858 return -ENOMEM;
1859 }
1860 memcpy(conf_uname_copy, conf_uname, len);
1861 conf_uname_copy[len] = '\0';
1862 conf_uname = conf_uname_copy;
1863 }
1864
Simon Glassda0af362013-05-07 06:11:53 +00001865 noffset = fdt_subnode_offset(fit, confs_noffset, conf_uname);
1866 if (noffset < 0) {
1867 debug("Can't get node offset for configuration unit name: '%s' (%s)\n",
1868 conf_uname, fdt_strerror(noffset));
1869 }
1870
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03001871 if (conf_uname_copy)
1872 free(conf_uname_copy);
1873
Simon Glassda0af362013-05-07 06:11:53 +00001874 return noffset;
1875}
1876
Pantelis Antonioua4880742017-09-04 23:12:14 +03001877int fit_conf_get_prop_node_count(const void *fit, int noffset,
Simon Glassda0af362013-05-07 06:11:53 +00001878 const char *prop_name)
1879{
Pantelis Antonioua4880742017-09-04 23:12:14 +03001880 return fdt_stringlist_count(fit, noffset, prop_name);
1881}
1882
1883int fit_conf_get_prop_node_index(const void *fit, int noffset,
1884 const char *prop_name, int index)
1885{
1886 const char *uname;
Simon Glassda0af362013-05-07 06:11:53 +00001887 int len;
1888
1889 /* get kernel image unit name from configuration kernel property */
Pantelis Antonioua4880742017-09-04 23:12:14 +03001890 uname = fdt_stringlist_get(fit, noffset, prop_name, index, &len);
Simon Glassda0af362013-05-07 06:11:53 +00001891 if (uname == NULL)
1892 return len;
1893
1894 return fit_image_get_node(fit, uname);
1895}
1896
Pantelis Antonioua4880742017-09-04 23:12:14 +03001897int fit_conf_get_prop_node(const void *fit, int noffset,
1898 const char *prop_name)
1899{
1900 return fit_conf_get_prop_node_index(fit, noffset, prop_name, 0);
1901}
Simon Glassda0af362013-05-07 06:11:53 +00001902
Jeroen Hofsteeffa60da2014-10-08 22:57:38 +02001903static int fit_image_select(const void *fit, int rd_noffset, int verify)
Simon Glass384d86d2013-05-16 13:53:21 +00001904{
1905 fit_image_print(fit, rd_noffset, " ");
1906
1907 if (verify) {
1908 puts(" Verifying Hash Integrity ... ");
1909 if (!fit_image_verify(fit, rd_noffset)) {
1910 puts("Bad Data Hash\n");
1911 return -EACCES;
1912 }
1913 puts("OK\n");
1914 }
1915
1916 return 0;
1917}
1918
Simon Glass384d86d2013-05-16 13:53:21 +00001919int fit_get_node_from_config(bootm_headers_t *images, const char *prop_name,
1920 ulong addr)
1921{
1922 int cfg_noffset;
1923 void *fit_hdr;
1924 int noffset;
1925
1926 debug("* %s: using config '%s' from image at 0x%08lx\n",
1927 prop_name, images->fit_uname_cfg, addr);
1928
1929 /* Check whether configuration has this property defined */
1930 fit_hdr = map_sysmem(addr, 0);
1931 cfg_noffset = fit_conf_get_node(fit_hdr, images->fit_uname_cfg);
1932 if (cfg_noffset < 0) {
1933 debug("* %s: no such config\n", prop_name);
Paul Burton14171b12016-09-20 18:17:12 +01001934 return -EINVAL;
Simon Glass384d86d2013-05-16 13:53:21 +00001935 }
1936
1937 noffset = fit_conf_get_prop_node(fit_hdr, cfg_noffset, prop_name);
1938 if (noffset < 0) {
1939 debug("* %s: no '%s' in config\n", prop_name, prop_name);
Jonathan Grayae6a02f2016-09-03 08:30:14 +10001940 return -ENOENT;
Simon Glass384d86d2013-05-16 13:53:21 +00001941 }
1942
1943 return noffset;
1944}
1945
Simon Glassa0c0b632014-06-12 07:24:47 -06001946/**
1947 * fit_get_image_type_property() - get property name for IH_TYPE_...
1948 *
1949 * @return the properly name where we expect to find the image in the
1950 * config node
1951 */
1952static const char *fit_get_image_type_property(int type)
1953{
1954 /*
1955 * This is sort-of available in the uimage_type[] table in image.c
Andreas Dannenberg8ca5a902016-06-15 17:00:19 -05001956 * but we don't have access to the short name, and "fdt" is different
Simon Glassa0c0b632014-06-12 07:24:47 -06001957 * anyway. So let's just keep it here.
1958 */
1959 switch (type) {
1960 case IH_TYPE_FLATDT:
1961 return FIT_FDT_PROP;
1962 case IH_TYPE_KERNEL:
1963 return FIT_KERNEL_PROP;
Alexandru Gagniucd1f78262021-04-01 13:25:30 -05001964 case IH_TYPE_FIRMWARE:
1965 return FIT_FIRMWARE_PROP;
Simon Glassa0c0b632014-06-12 07:24:47 -06001966 case IH_TYPE_RAMDISK:
1967 return FIT_RAMDISK_PROP;
Simon Glass0129b522014-10-19 21:11:24 -06001968 case IH_TYPE_X86_SETUP:
1969 return FIT_SETUP_PROP;
Karl Apsite1b21c282015-05-21 09:52:48 -04001970 case IH_TYPE_LOADABLE:
1971 return FIT_LOADABLE_PROP;
Michal Simekebae78b2016-05-17 14:03:50 +02001972 case IH_TYPE_FPGA:
1973 return FIT_FPGA_PROP;
Marek Vasut93e95752018-05-13 00:22:54 +02001974 case IH_TYPE_STANDALONE:
1975 return FIT_STANDALONE_PROP;
Simon Glassa0c0b632014-06-12 07:24:47 -06001976 }
1977
1978 return "unknown";
1979}
1980
1981int fit_image_load(bootm_headers_t *images, ulong addr,
Simon Glassad68fc32013-07-10 23:08:10 -07001982 const char **fit_unamep, const char **fit_uname_configp,
Simon Glass384d86d2013-05-16 13:53:21 +00001983 int arch, int image_type, int bootstage_id,
1984 enum fit_load_op load_op, ulong *datap, ulong *lenp)
1985{
1986 int cfg_noffset, noffset;
1987 const char *fit_uname;
Simon Glassad68fc32013-07-10 23:08:10 -07001988 const char *fit_uname_config;
Pantelis Antoniou5e671d62017-09-04 23:12:15 +03001989 const char *fit_base_uname_config;
Simon Glass384d86d2013-05-16 13:53:21 +00001990 const void *fit;
Julius Werner97b09cd2019-07-24 19:37:55 -07001991 void *buf;
1992 void *loadbuf;
Simon Glass384d86d2013-05-16 13:53:21 +00001993 size_t size;
1994 int type_ok, os_ok;
Julius Werner97b09cd2019-07-24 19:37:55 -07001995 ulong load, load_end, data, len;
1996 uint8_t os, comp;
Alison Wang73818d52016-11-10 10:49:03 +08001997#ifndef USE_HOSTCC
1998 uint8_t os_arch;
1999#endif
Simon Glassa0c0b632014-06-12 07:24:47 -06002000 const char *prop_name;
Simon Glass384d86d2013-05-16 13:53:21 +00002001 int ret;
2002
2003 fit = map_sysmem(addr, 0);
2004 fit_uname = fit_unamep ? *fit_unamep : NULL;
Simon Glassad68fc32013-07-10 23:08:10 -07002005 fit_uname_config = fit_uname_configp ? *fit_uname_configp : NULL;
Pantelis Antoniou5e671d62017-09-04 23:12:15 +03002006 fit_base_uname_config = NULL;
Simon Glassa0c0b632014-06-12 07:24:47 -06002007 prop_name = fit_get_image_type_property(image_type);
Simon Glass384d86d2013-05-16 13:53:21 +00002008 printf("## Loading %s from FIT Image at %08lx ...\n", prop_name, addr);
2009
2010 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_FORMAT);
Simon Glassb823daa2021-02-15 17:08:12 -07002011 ret = fit_check_format(fit, IMAGE_SIZE_INVAL);
2012 if (ret) {
2013 printf("Bad FIT %s image format! (err=%d)\n", prop_name, ret);
2014 if (CONFIG_IS_ENABLED(FIT_SIGNATURE) && ret == -EADDRNOTAVAIL)
2015 printf("Signature checking prevents use of unit addresses (@) in nodes\n");
Simon Glass384d86d2013-05-16 13:53:21 +00002016 bootstage_error(bootstage_id + BOOTSTAGE_SUB_FORMAT);
Simon Glassb823daa2021-02-15 17:08:12 -07002017 return ret;
Simon Glass384d86d2013-05-16 13:53:21 +00002018 }
2019 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_FORMAT_OK);
2020 if (fit_uname) {
Masahiro Yamadaaa58eec2014-02-18 15:39:21 +09002021 /* get FIT component image node offset */
Simon Glass384d86d2013-05-16 13:53:21 +00002022 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_UNIT_NAME);
2023 noffset = fit_image_get_node(fit, fit_uname);
2024 } else {
2025 /*
2026 * no image node unit name, try to get config
2027 * node first. If config unit node name is NULL
2028 * fit_conf_get_node() will try to find default config node
2029 */
2030 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_NO_UNIT_NAME);
Simon Glass0434ea12021-07-14 17:05:36 -05002031 if (IS_ENABLED(CONFIG_FIT_BEST_MATCH) && !fit_uname_config) {
Simon Glass384d86d2013-05-16 13:53:21 +00002032 cfg_noffset = fit_conf_find_compat(fit, gd_fdt_blob());
2033 } else {
2034 cfg_noffset = fit_conf_get_node(fit,
2035 fit_uname_config);
2036 }
2037 if (cfg_noffset < 0) {
2038 puts("Could not find configuration node\n");
2039 bootstage_error(bootstage_id +
2040 BOOTSTAGE_SUB_NO_UNIT_NAME);
2041 return -ENOENT;
2042 }
Marek Vasut5dee1342018-05-31 17:59:07 +02002043
Pantelis Antoniou5e671d62017-09-04 23:12:15 +03002044 fit_base_uname_config = fdt_get_name(fit, cfg_noffset, NULL);
2045 printf(" Using '%s' configuration\n", fit_base_uname_config);
Marek Vasut5dee1342018-05-31 17:59:07 +02002046 /* Remember this config */
2047 if (image_type == IH_TYPE_KERNEL)
Pantelis Antoniou5e671d62017-09-04 23:12:15 +03002048 images->fit_uname_cfg = fit_base_uname_config;
Marek Vasut5dee1342018-05-31 17:59:07 +02002049
AKASHI Takahiro2223c7d2020-02-21 15:12:55 +09002050 if (FIT_IMAGE_ENABLE_VERIFY && images->verify) {
Marek Vasut5dee1342018-05-31 17:59:07 +02002051 puts(" Verifying Hash Integrity ... ");
2052 if (fit_config_verify(fit, cfg_noffset)) {
2053 puts("Bad Data Hash\n");
2054 bootstage_error(bootstage_id +
2055 BOOTSTAGE_SUB_HASH);
2056 return -EACCES;
Simon Glass384d86d2013-05-16 13:53:21 +00002057 }
Marek Vasut5dee1342018-05-31 17:59:07 +02002058 puts("OK\n");
Simon Glass384d86d2013-05-16 13:53:21 +00002059 }
2060
Marek Vasut5dee1342018-05-31 17:59:07 +02002061 bootstage_mark(BOOTSTAGE_ID_FIT_CONFIG);
2062
Simon Glass384d86d2013-05-16 13:53:21 +00002063 noffset = fit_conf_get_prop_node(fit, cfg_noffset,
2064 prop_name);
2065 fit_uname = fit_get_name(fit, noffset, NULL);
2066 }
2067 if (noffset < 0) {
Simon Glassa559bb22020-03-18 11:43:56 -06002068 printf("Could not find subimage node type '%s'\n", prop_name);
Simon Glass384d86d2013-05-16 13:53:21 +00002069 bootstage_error(bootstage_id + BOOTSTAGE_SUB_SUBNODE);
2070 return -ENOENT;
2071 }
2072
2073 printf(" Trying '%s' %s subimage\n", fit_uname, prop_name);
2074
2075 ret = fit_image_select(fit, noffset, images->verify);
2076 if (ret) {
2077 bootstage_error(bootstage_id + BOOTSTAGE_SUB_HASH);
2078 return ret;
2079 }
2080
2081 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_CHECK_ARCH);
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +01002082 if (!host_build() && IS_ENABLED(CONFIG_SANDBOX)) {
2083 if (!fit_image_check_target_arch(fit, noffset)) {
2084 puts("Unsupported Architecture\n");
2085 bootstage_error(bootstage_id + BOOTSTAGE_SUB_CHECK_ARCH);
2086 return -ENOEXEC;
2087 }
Simon Glass384d86d2013-05-16 13:53:21 +00002088 }
Alison Wang73818d52016-11-10 10:49:03 +08002089
2090#ifndef USE_HOSTCC
2091 fit_image_get_arch(fit, noffset, &os_arch);
2092 images->os.arch = os_arch;
2093#endif
2094
Simon Glass384d86d2013-05-16 13:53:21 +00002095 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_CHECK_ALL);
2096 type_ok = fit_image_check_type(fit, noffset, image_type) ||
mario.six@gdsys.cca82eeaf2016-07-20 08:32:50 +02002097 fit_image_check_type(fit, noffset, IH_TYPE_FIRMWARE) ||
Alexandru Gagniuce14c9b42021-04-01 13:25:31 -05002098 fit_image_check_type(fit, noffset, IH_TYPE_TEE) ||
mario.six@gdsys.cca82eeaf2016-07-20 08:32:50 +02002099 (image_type == IH_TYPE_KERNEL &&
2100 fit_image_check_type(fit, noffset, IH_TYPE_KERNEL_NOLOAD));
Marek Vasutcaee41f2014-12-16 14:07:22 +01002101
Andreas Bießmannb4cc60d2016-08-14 20:31:24 +02002102 os_ok = image_type == IH_TYPE_FLATDT ||
2103 image_type == IH_TYPE_FPGA ||
Marek Vasutcaee41f2014-12-16 14:07:22 +01002104 fit_image_check_os(fit, noffset, IH_OS_LINUX) ||
mario.six@gdsys.cca82eeaf2016-07-20 08:32:50 +02002105 fit_image_check_os(fit, noffset, IH_OS_U_BOOT) ||
Alexandru Gagniuce14c9b42021-04-01 13:25:31 -05002106 fit_image_check_os(fit, noffset, IH_OS_TEE) ||
Cristian Ciocaltea217f9642019-12-24 18:05:38 +02002107 fit_image_check_os(fit, noffset, IH_OS_OPENRTOS) ||
Lihua Zhao03b0d1e2020-03-18 07:32:07 -07002108 fit_image_check_os(fit, noffset, IH_OS_EFI) ||
2109 fit_image_check_os(fit, noffset, IH_OS_VXWORKS);
Karl Apsite1b21c282015-05-21 09:52:48 -04002110
2111 /*
2112 * If either of the checks fail, we should report an error, but
2113 * if the image type is coming from the "loadables" field, we
2114 * don't care what it is
2115 */
2116 if ((!type_ok || !os_ok) && image_type != IH_TYPE_LOADABLE) {
Marek Vasutcaee41f2014-12-16 14:07:22 +01002117 fit_image_get_os(fit, noffset, &os);
2118 printf("No %s %s %s Image\n",
2119 genimg_get_os_name(os),
2120 genimg_get_arch_name(arch),
Simon Glass384d86d2013-05-16 13:53:21 +00002121 genimg_get_type_name(image_type));
2122 bootstage_error(bootstage_id + BOOTSTAGE_SUB_CHECK_ALL);
2123 return -EIO;
2124 }
2125
2126 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_CHECK_ALL_OK);
2127
2128 /* get image data address and length */
Julius Werner97b09cd2019-07-24 19:37:55 -07002129 if (fit_image_get_data_and_size(fit, noffset,
2130 (const void **)&buf, &size)) {
Simon Glass384d86d2013-05-16 13:53:21 +00002131 printf("Could not find %s subimage data!\n", prop_name);
2132 bootstage_error(bootstage_id + BOOTSTAGE_SUB_GET_DATA);
Simon Glass9d0da5b2013-06-16 07:46:49 -07002133 return -ENOENT;
Simon Glass384d86d2013-05-16 13:53:21 +00002134 }
Andrew F. Davisc6d8cc02016-11-21 14:37:09 -06002135
Philippe Reynes3d964702019-12-18 18:25:42 +01002136 /* Decrypt data before uncompress/move */
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +01002137 if (IS_ENABLED(CONFIG_FIT_CIPHER) && IMAGE_ENABLE_DECRYPT) {
Philippe Reynes3d964702019-12-18 18:25:42 +01002138 puts(" Decrypting Data ... ");
2139 if (fit_image_uncipher(fit, noffset, &buf, &size)) {
2140 puts("Error\n");
2141 return -EACCES;
2142 }
2143 puts("OK\n");
2144 }
Philippe Reynes3d964702019-12-18 18:25:42 +01002145
Andrew F. Davisc6d8cc02016-11-21 14:37:09 -06002146 /* perform any post-processing on the image data */
Sebastian Reichelbfe8ddf2021-01-04 20:48:03 +01002147 if (!host_build() && IS_ENABLED(CONFIG_FIT_IMAGE_POST_PROCESS))
Lokesh Vutlab36dd3e2021-06-11 11:45:05 +03002148 board_fit_image_post_process(fit, noffset, &buf, &size);
Andrew F. Davisc6d8cc02016-11-21 14:37:09 -06002149
Simon Glass384d86d2013-05-16 13:53:21 +00002150 len = (ulong)size;
2151
Simon Glass384d86d2013-05-16 13:53:21 +00002152 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_GET_DATA_OK);
2153
Julius Werner97b09cd2019-07-24 19:37:55 -07002154 data = map_to_sysmem(buf);
2155 load = data;
Simon Glass384d86d2013-05-16 13:53:21 +00002156 if (load_op == FIT_LOAD_IGNORED) {
2157 /* Don't load */
2158 } else if (fit_image_get_load(fit, noffset, &load)) {
2159 if (load_op == FIT_LOAD_REQUIRED) {
2160 printf("Can't get %s subimage load address!\n",
2161 prop_name);
2162 bootstage_error(bootstage_id + BOOTSTAGE_SUB_LOAD);
2163 return -EBADF;
2164 }
Simon Glass05a9ad72014-08-22 14:26:43 -06002165 } else if (load_op != FIT_LOAD_OPTIONAL_NON_ZERO || load) {
Simon Glass384d86d2013-05-16 13:53:21 +00002166 ulong image_start, image_end;
Simon Glass384d86d2013-05-16 13:53:21 +00002167
2168 /*
2169 * move image data to the load address,
2170 * make sure we don't overwrite initial image
2171 */
2172 image_start = addr;
2173 image_end = addr + fit_get_size(fit);
2174
2175 load_end = load + len;
2176 if (image_type != IH_TYPE_KERNEL &&
2177 load < image_end && load_end > image_start) {
2178 printf("Error: %s overwritten\n", prop_name);
2179 return -EXDEV;
2180 }
2181
2182 printf(" Loading %s from 0x%08lx to 0x%08lx\n",
2183 prop_name, data, load);
Julius Werner97b09cd2019-07-24 19:37:55 -07002184 } else {
2185 load = data; /* No load address specified */
2186 }
2187
2188 comp = IH_COMP_NONE;
2189 loadbuf = buf;
2190 /* Kernel images get decompressed later in bootm_load_os(). */
Julius Werner88040022019-08-02 15:52:28 -07002191 if (!fit_image_get_comp(fit, noffset, &comp) &&
2192 comp != IH_COMP_NONE &&
2193 !(image_type == IH_TYPE_KERNEL ||
2194 image_type == IH_TYPE_KERNEL_NOLOAD ||
2195 image_type == IH_TYPE_RAMDISK)) {
Julius Werner97b09cd2019-07-24 19:37:55 -07002196 ulong max_decomp_len = len * 20;
2197 if (load == data) {
2198 loadbuf = malloc(max_decomp_len);
2199 load = map_to_sysmem(loadbuf);
2200 } else {
2201 loadbuf = map_sysmem(load, max_decomp_len);
2202 }
2203 if (image_decomp(comp, load, data, image_type,
2204 loadbuf, buf, len, max_decomp_len, &load_end)) {
2205 printf("Error decompressing %s\n", prop_name);
Simon Glass384d86d2013-05-16 13:53:21 +00002206
Julius Werner97b09cd2019-07-24 19:37:55 -07002207 return -ENOEXEC;
2208 }
2209 len = load_end - load;
2210 } else if (load != data) {
2211 loadbuf = map_sysmem(load, len);
2212 memcpy(loadbuf, buf, len);
Simon Glass384d86d2013-05-16 13:53:21 +00002213 }
Julius Werner97b09cd2019-07-24 19:37:55 -07002214
Julius Werner88040022019-08-02 15:52:28 -07002215 if (image_type == IH_TYPE_RAMDISK && comp != IH_COMP_NONE)
2216 puts("WARNING: 'compression' nodes for ramdisks are deprecated,"
2217 " please fix your .its file!\n");
2218
Julius Werner97b09cd2019-07-24 19:37:55 -07002219 /* verify that image data is a proper FDT blob */
2220 if (image_type == IH_TYPE_FLATDT && fdt_check_header(loadbuf)) {
2221 puts("Subimage data is not a FDT");
2222 return -ENOEXEC;
2223 }
2224
Simon Glass384d86d2013-05-16 13:53:21 +00002225 bootstage_mark(bootstage_id + BOOTSTAGE_SUB_LOAD);
2226
Julius Werner97b09cd2019-07-24 19:37:55 -07002227 *datap = load;
Simon Glass384d86d2013-05-16 13:53:21 +00002228 *lenp = len;
2229 if (fit_unamep)
2230 *fit_unamep = (char *)fit_uname;
Simon Glassad68fc32013-07-10 23:08:10 -07002231 if (fit_uname_configp)
Pantelis Antoniou5e671d62017-09-04 23:12:15 +03002232 *fit_uname_configp = (char *)(fit_uname_config ? :
2233 fit_base_uname_config);
Simon Glass384d86d2013-05-16 13:53:21 +00002234
2235 return noffset;
2236}
Simon Glass0129b522014-10-19 21:11:24 -06002237
2238int boot_get_setup_fit(bootm_headers_t *images, uint8_t arch,
2239 ulong *setup_start, ulong *setup_len)
2240{
2241 int noffset;
2242 ulong addr;
2243 ulong len;
2244 int ret;
2245
2246 addr = map_to_sysmem(images->fit_hdr_os);
2247 noffset = fit_get_node_from_config(images, FIT_SETUP_PROP, addr);
2248 if (noffset < 0)
2249 return noffset;
2250
2251 ret = fit_image_load(images, addr, NULL, NULL, arch,
2252 IH_TYPE_X86_SETUP, BOOTSTAGE_ID_FIT_SETUP_START,
2253 FIT_LOAD_REQUIRED, setup_start, &len);
2254
2255 return ret;
2256}
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002257
2258#ifndef USE_HOSTCC
2259int boot_get_fdt_fit(bootm_headers_t *images, ulong addr,
2260 const char **fit_unamep, const char **fit_uname_configp,
2261 int arch, ulong *datap, ulong *lenp)
2262{
2263 int fdt_noffset, cfg_noffset, count;
2264 const void *fit;
2265 const char *fit_uname = NULL;
2266 const char *fit_uname_config = NULL;
2267 char *fit_uname_config_copy = NULL;
2268 char *next_config = NULL;
2269 ulong load, len;
2270#ifdef CONFIG_OF_LIBFDT_OVERLAY
2271 ulong image_start, image_end;
Marek Vasutf7d24612021-06-11 04:09:56 +02002272 ulong ovload, ovlen, ovcopylen;
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002273 const char *uconfig;
2274 const char *uname;
Marek Vasutf7d24612021-06-11 04:09:56 +02002275 void *base, *ov, *ovcopy = NULL;
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002276 int i, err, noffset, ov_noffset;
2277#endif
2278
2279 fit_uname = fit_unamep ? *fit_unamep : NULL;
2280
2281 if (fit_uname_configp && *fit_uname_configp) {
2282 fit_uname_config_copy = strdup(*fit_uname_configp);
2283 if (!fit_uname_config_copy)
2284 return -ENOMEM;
2285
2286 next_config = strchr(fit_uname_config_copy, '#');
2287 if (next_config)
2288 *next_config++ = '\0';
2289 if (next_config - 1 > fit_uname_config_copy)
2290 fit_uname_config = fit_uname_config_copy;
2291 }
2292
2293 fdt_noffset = fit_image_load(images,
2294 addr, &fit_uname, &fit_uname_config,
2295 arch, IH_TYPE_FLATDT,
2296 BOOTSTAGE_ID_FIT_FDT_START,
2297 FIT_LOAD_OPTIONAL, &load, &len);
2298
2299 if (fdt_noffset < 0)
2300 goto out;
2301
2302 debug("fit_uname=%s, fit_uname_config=%s\n",
2303 fit_uname ? fit_uname : "<NULL>",
2304 fit_uname_config ? fit_uname_config : "<NULL>");
2305
2306 fit = map_sysmem(addr, 0);
2307
2308 cfg_noffset = fit_conf_get_node(fit, fit_uname_config);
2309
2310 /* single blob, or error just return as well */
2311 count = fit_conf_get_prop_node_count(fit, cfg_noffset, FIT_FDT_PROP);
2312 if (count <= 1 && !next_config)
2313 goto out;
2314
2315 /* we need to apply overlays */
2316
2317#ifdef CONFIG_OF_LIBFDT_OVERLAY
2318 image_start = addr;
2319 image_end = addr + fit_get_size(fit);
2320 /* verify that relocation took place by load address not being in fit */
2321 if (load >= image_start && load < image_end) {
2322 /* check is simplified; fit load checks for overlaps */
2323 printf("Overlayed FDT requires relocation\n");
2324 fdt_noffset = -EBADF;
2325 goto out;
2326 }
2327
2328 base = map_sysmem(load, len);
2329
2330 /* apply extra configs in FIT first, followed by args */
2331 for (i = 1; ; i++) {
2332 if (i < count) {
2333 noffset = fit_conf_get_prop_node_index(fit, cfg_noffset,
2334 FIT_FDT_PROP, i);
2335 uname = fit_get_name(fit, noffset, NULL);
2336 uconfig = NULL;
2337 } else {
2338 if (!next_config)
2339 break;
2340 uconfig = next_config;
2341 next_config = strchr(next_config, '#');
2342 if (next_config)
2343 *next_config++ = '\0';
2344 uname = NULL;
Peter Ujfalusiaf15dc42019-03-06 15:52:27 +02002345
2346 /*
2347 * fit_image_load() would load the first FDT from the
2348 * extra config only when uconfig is specified.
2349 * Check if the extra config contains multiple FDTs and
2350 * if so, load them.
2351 */
2352 cfg_noffset = fit_conf_get_node(fit, uconfig);
2353
2354 i = 0;
2355 count = fit_conf_get_prop_node_count(fit, cfg_noffset,
2356 FIT_FDT_PROP);
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002357 }
2358
2359 debug("%d: using uname=%s uconfig=%s\n", i, uname, uconfig);
2360
2361 ov_noffset = fit_image_load(images,
2362 addr, &uname, &uconfig,
2363 arch, IH_TYPE_FLATDT,
2364 BOOTSTAGE_ID_FIT_FDT_START,
Marek Vasutf7d24612021-06-11 04:09:56 +02002365 FIT_LOAD_IGNORED, &ovload, &ovlen);
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002366 if (ov_noffset < 0) {
2367 printf("load of %s failed\n", uname);
2368 continue;
2369 }
2370 debug("%s loaded at 0x%08lx len=0x%08lx\n",
2371 uname, ovload, ovlen);
2372 ov = map_sysmem(ovload, ovlen);
2373
Marek Vasutf7d24612021-06-11 04:09:56 +02002374 ovcopylen = ALIGN(fdt_totalsize(ov), SZ_4K);
2375 ovcopy = malloc(ovcopylen);
2376 if (!ovcopy) {
2377 printf("failed to duplicate DTO before application\n");
2378 fdt_noffset = -ENOMEM;
2379 goto out;
2380 }
2381
2382 err = fdt_open_into(ov, ovcopy, ovcopylen);
2383 if (err < 0) {
2384 printf("failed on fdt_open_into for DTO\n");
2385 fdt_noffset = err;
2386 goto out;
2387 }
2388
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002389 base = map_sysmem(load, len + ovlen);
2390 err = fdt_open_into(base, base, len + ovlen);
2391 if (err < 0) {
2392 printf("failed on fdt_open_into\n");
2393 fdt_noffset = err;
2394 goto out;
2395 }
Marek Vasutf7d24612021-06-11 04:09:56 +02002396
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002397 /* the verbose method prints out messages on error */
Marek Vasutf7d24612021-06-11 04:09:56 +02002398 err = fdt_overlay_apply_verbose(base, ovcopy);
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002399 if (err < 0) {
2400 fdt_noffset = err;
2401 goto out;
2402 }
2403 fdt_pack(base);
2404 len = fdt_totalsize(base);
Marek Vasutf7d24612021-06-11 04:09:56 +02002405
2406 free(ovcopy);
2407 ovcopy = NULL;
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002408 }
2409#else
2410 printf("config with overlays but CONFIG_OF_LIBFDT_OVERLAY not set\n");
2411 fdt_noffset = -EBADF;
2412#endif
2413
2414out:
2415 if (datap)
2416 *datap = load;
2417 if (lenp)
2418 *lenp = len;
2419 if (fit_unamep)
2420 *fit_unamep = fit_uname;
2421 if (fit_uname_configp)
2422 *fit_uname_configp = fit_uname_config;
2423
Marek Vasutf7d24612021-06-11 04:09:56 +02002424#ifdef CONFIG_OF_LIBFDT_OVERLAY
2425 if (ovcopy)
2426 free(ovcopy);
2427#endif
Pantelis Antoniou6e51fb22017-09-04 23:12:16 +03002428 if (fit_uname_config_copy)
2429 free(fit_uname_config_copy);
2430 return fdt_noffset;
2431}
2432#endif