1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * (C) Copyright 2017
3*4882a593Smuzhiyun * Texas Instruments, <www.ti.com>
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * Franklin S Cooper Jr. <fcooper@ti.com>
6*4882a593Smuzhiyun *
7*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
8*4882a593Smuzhiyun */
9*4882a593Smuzhiyun
10*4882a593Smuzhiyun #include <boot_fit.h>
11*4882a593Smuzhiyun #include <common.h>
12*4882a593Smuzhiyun #include <errno.h>
13*4882a593Smuzhiyun #include <image.h>
14*4882a593Smuzhiyun #include <linux/libfdt.h>
15*4882a593Smuzhiyun
fdt_offset(const void * fit)16*4882a593Smuzhiyun static int fdt_offset(const void *fit)
17*4882a593Smuzhiyun {
18*4882a593Smuzhiyun int images, node, fdt_len, fdt_node, fdt_offset;
19*4882a593Smuzhiyun const char *fdt_name;
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun node = fit_find_config_node(fit);
22*4882a593Smuzhiyun if (node < 0)
23*4882a593Smuzhiyun return node;
24*4882a593Smuzhiyun
25*4882a593Smuzhiyun images = fdt_path_offset(fit, FIT_IMAGES_PATH);
26*4882a593Smuzhiyun if (images < 0) {
27*4882a593Smuzhiyun debug("%s: Cannot find /images node: %d\n", __func__, images);
28*4882a593Smuzhiyun return -EINVAL;
29*4882a593Smuzhiyun }
30*4882a593Smuzhiyun
31*4882a593Smuzhiyun fdt_name = fdt_getprop(fit, node, FIT_FDT_PROP, &fdt_len);
32*4882a593Smuzhiyun if (!fdt_name) {
33*4882a593Smuzhiyun debug("%s: Cannot find fdt name property: %d\n",
34*4882a593Smuzhiyun __func__, fdt_len);
35*4882a593Smuzhiyun return -EINVAL;
36*4882a593Smuzhiyun }
37*4882a593Smuzhiyun
38*4882a593Smuzhiyun fdt_node = fdt_subnode_offset(fit, images, fdt_name);
39*4882a593Smuzhiyun if (fdt_node < 0) {
40*4882a593Smuzhiyun debug("%s: Cannot find fdt node '%s': %d\n",
41*4882a593Smuzhiyun __func__, fdt_name, fdt_node);
42*4882a593Smuzhiyun return -EINVAL;
43*4882a593Smuzhiyun }
44*4882a593Smuzhiyun
45*4882a593Smuzhiyun fdt_offset = fdt_getprop_u32(fit, fdt_node, "data-offset");
46*4882a593Smuzhiyun
47*4882a593Smuzhiyun if (fdt_offset == FDT_ERROR)
48*4882a593Smuzhiyun return -ENOENT;
49*4882a593Smuzhiyun
50*4882a593Smuzhiyun fdt_len = fdt_getprop_u32(fit, fdt_node, "data-size");
51*4882a593Smuzhiyun
52*4882a593Smuzhiyun if (fdt_len < 0)
53*4882a593Smuzhiyun return fdt_len;
54*4882a593Smuzhiyun
55*4882a593Smuzhiyun return fdt_offset;
56*4882a593Smuzhiyun }
57*4882a593Smuzhiyun
locate_dtb_in_fit(const void * fit)58*4882a593Smuzhiyun void *locate_dtb_in_fit(const void *fit)
59*4882a593Smuzhiyun {
60*4882a593Smuzhiyun struct image_header *header;
61*4882a593Smuzhiyun int size;
62*4882a593Smuzhiyun int ret;
63*4882a593Smuzhiyun
64*4882a593Smuzhiyun size = fdt_totalsize(fit);
65*4882a593Smuzhiyun size = FIT_ALIGN(size);
66*4882a593Smuzhiyun
67*4882a593Smuzhiyun header = (struct image_header *)fit;
68*4882a593Smuzhiyun
69*4882a593Smuzhiyun if (image_get_magic(header) != FDT_MAGIC) {
70*4882a593Smuzhiyun debug("No FIT image appended to U-boot\n");
71*4882a593Smuzhiyun return NULL;
72*4882a593Smuzhiyun }
73*4882a593Smuzhiyun
74*4882a593Smuzhiyun ret = fdt_offset(fit);
75*4882a593Smuzhiyun
76*4882a593Smuzhiyun if (ret < 0)
77*4882a593Smuzhiyun return NULL;
78*4882a593Smuzhiyun else
79*4882a593Smuzhiyun return (void *)fit+size+ret;
80*4882a593Smuzhiyun }
81