xref: /aosp_15_r20/external/dtc/libfdt/fdt.c (revision cd60bc56d4bea3af4ec04523e4d71c2b272c8aff)
1 // SPDX-License-Identifier: (GPL-2.0-or-later OR BSD-2-Clause)
2 /*
3  * libfdt - Flat Device Tree manipulation
4  * Copyright (C) 2006 David Gibson, IBM Corporation.
5  */
6 #include "libfdt_env.h"
7 
8 #include <fdt.h>
9 #include <libfdt.h>
10 
11 #include "libfdt_internal.h"
12 
13 /*
14  * Minimal sanity check for a read-only tree. fdt_ro_probe_() checks
15  * that the given buffer contains what appears to be a flattened
16  * device tree with sane information in its header.
17  */
fdt_ro_probe_(const void * fdt)18 int32_t fdt_ro_probe_(const void *fdt)
19 {
20 	uint32_t totalsize = fdt_totalsize(fdt);
21 
22 	if (can_assume(VALID_DTB))
23 		return totalsize;
24 
25 	/* The device tree must be at an 8-byte aligned address */
26 	if ((uintptr_t)fdt & 7)
27 		return -FDT_ERR_ALIGNMENT;
28 
29 	if (fdt_magic(fdt) == FDT_MAGIC) {
30 		/* Complete tree */
31 		if (!can_assume(LATEST)) {
32 			if (fdt_version(fdt) < FDT_FIRST_SUPPORTED_VERSION)
33 				return -FDT_ERR_BADVERSION;
34 			if (fdt_last_comp_version(fdt) >
35 					FDT_LAST_SUPPORTED_VERSION)
36 				return -FDT_ERR_BADVERSION;
37 		}
38 	} else if (fdt_magic(fdt) == FDT_SW_MAGIC) {
39 		/* Unfinished sequential-write blob */
40 		if (!can_assume(VALID_INPUT) && fdt_size_dt_struct(fdt) == 0)
41 			return -FDT_ERR_BADSTATE;
42 	} else {
43 		return -FDT_ERR_BADMAGIC;
44 	}
45 
46 	if (totalsize < INT32_MAX)
47 		return totalsize;
48 	else
49 		return -FDT_ERR_TRUNCATED;
50 }
51 
check_off_(uint32_t hdrsize,uint32_t totalsize,uint32_t off)52 static int check_off_(uint32_t hdrsize, uint32_t totalsize, uint32_t off)
53 {
54 	return (off >= hdrsize) && (off <= totalsize);
55 }
56 
check_block_(uint32_t hdrsize,uint32_t totalsize,uint32_t base,uint32_t size)57 static int check_block_(uint32_t hdrsize, uint32_t totalsize,
58 			uint32_t base, uint32_t size)
59 {
60 	if (!check_off_(hdrsize, totalsize, base))
61 		return 0; /* block start out of bounds */
62 	if ((base + size) < base)
63 		return 0; /* overflow */
64 	if (!check_off_(hdrsize, totalsize, base + size))
65 		return 0; /* block end out of bounds */
66 	return 1;
67 }
68 
fdt_header_size_(uint32_t version)69 size_t fdt_header_size_(uint32_t version)
70 {
71 	if (version <= 1)
72 		return FDT_V1_SIZE;
73 	else if (version <= 2)
74 		return FDT_V2_SIZE;
75 	else if (version <= 3)
76 		return FDT_V3_SIZE;
77 	else if (version <= 16)
78 		return FDT_V16_SIZE;
79 	else
80 		return FDT_V17_SIZE;
81 }
82 
fdt_header_size(const void * fdt)83 size_t fdt_header_size(const void *fdt)
84 {
85 	return can_assume(LATEST) ? FDT_V17_SIZE :
86 		fdt_header_size_(fdt_version(fdt));
87 }
88 
fdt_check_header(const void * fdt)89 int fdt_check_header(const void *fdt)
90 {
91 	size_t hdrsize;
92 
93 	/* The device tree must be at an 8-byte aligned address */
94 	if ((uintptr_t)fdt & 7)
95 		return -FDT_ERR_ALIGNMENT;
96 
97 	if (fdt_magic(fdt) != FDT_MAGIC)
98 		return -FDT_ERR_BADMAGIC;
99 	if (!can_assume(LATEST)) {
100 		if ((fdt_version(fdt) < FDT_FIRST_SUPPORTED_VERSION)
101 		    || (fdt_last_comp_version(fdt) >
102 			FDT_LAST_SUPPORTED_VERSION))
103 			return -FDT_ERR_BADVERSION;
104 		if (fdt_version(fdt) < fdt_last_comp_version(fdt))
105 			return -FDT_ERR_BADVERSION;
106 	}
107 	hdrsize = fdt_header_size(fdt);
108 	if (!can_assume(VALID_DTB)) {
109 		if ((fdt_totalsize(fdt) < hdrsize)
110 		    || (fdt_totalsize(fdt) > INT_MAX))
111 			return -FDT_ERR_TRUNCATED;
112 
113 		/* Bounds check memrsv block */
114 		if (!check_off_(hdrsize, fdt_totalsize(fdt),
115 				fdt_off_mem_rsvmap(fdt)))
116 			return -FDT_ERR_TRUNCATED;
117 
118 		/* Bounds check structure block */
119 		if (!can_assume(LATEST) && fdt_version(fdt) < 17) {
120 			if (!check_off_(hdrsize, fdt_totalsize(fdt),
121 					fdt_off_dt_struct(fdt)))
122 				return -FDT_ERR_TRUNCATED;
123 		} else {
124 			if (!check_block_(hdrsize, fdt_totalsize(fdt),
125 					  fdt_off_dt_struct(fdt),
126 					  fdt_size_dt_struct(fdt)))
127 				return -FDT_ERR_TRUNCATED;
128 		}
129 
130 		/* Bounds check strings block */
131 		if (!check_block_(hdrsize, fdt_totalsize(fdt),
132 				  fdt_off_dt_strings(fdt),
133 				  fdt_size_dt_strings(fdt)))
134 			return -FDT_ERR_TRUNCATED;
135 	}
136 
137 	return 0;
138 }
139 
fdt_offset_ptr(const void * fdt,int offset,unsigned int len)140 const void *fdt_offset_ptr(const void *fdt, int offset, unsigned int len)
141 {
142 	unsigned int uoffset = offset;
143 	unsigned int absoffset = offset + fdt_off_dt_struct(fdt);
144 
145 	if (offset < 0)
146 		return NULL;
147 
148 	if (!can_assume(VALID_INPUT))
149 		if ((absoffset < uoffset)
150 		    || ((absoffset + len) < absoffset)
151 		    || (absoffset + len) > fdt_totalsize(fdt))
152 			return NULL;
153 
154 	if (can_assume(LATEST) || fdt_version(fdt) >= 0x11)
155 		if (((uoffset + len) < uoffset)
156 		    || ((offset + len) > fdt_size_dt_struct(fdt)))
157 			return NULL;
158 
159 	return fdt_offset_ptr_(fdt, offset);
160 }
161 
fdt_next_tag(const void * fdt,int startoffset,int * nextoffset)162 uint32_t fdt_next_tag(const void *fdt, int startoffset, int *nextoffset)
163 {
164 	const fdt32_t *tagp, *lenp;
165 	uint32_t tag, len, sum;
166 	int offset = startoffset;
167 	const char *p;
168 
169 	*nextoffset = -FDT_ERR_TRUNCATED;
170 	tagp = fdt_offset_ptr(fdt, offset, FDT_TAGSIZE);
171 	if (!can_assume(VALID_DTB) && !tagp)
172 		return FDT_END; /* premature end */
173 	tag = fdt32_to_cpu(*tagp);
174 	offset += FDT_TAGSIZE;
175 
176 	*nextoffset = -FDT_ERR_BADSTRUCTURE;
177 	switch (tag) {
178 	case FDT_BEGIN_NODE:
179 		/* skip name */
180 		do {
181 			p = fdt_offset_ptr(fdt, offset++, 1);
182 		} while (p && (*p != '\0'));
183 		if (!can_assume(VALID_DTB) && !p)
184 			return FDT_END; /* premature end */
185 		break;
186 
187 	case FDT_PROP:
188 		lenp = fdt_offset_ptr(fdt, offset, sizeof(*lenp));
189 		if (!can_assume(VALID_DTB) && !lenp)
190 			return FDT_END; /* premature end */
191 
192 		len = fdt32_to_cpu(*lenp);
193 		sum = len + offset;
194 		if (!can_assume(VALID_DTB) &&
195 		    (INT_MAX <= sum || sum < (uint32_t) offset))
196 			return FDT_END; /* premature end */
197 
198 		/* skip-name offset, length and value */
199 		offset += sizeof(struct fdt_property) - FDT_TAGSIZE + len;
200 
201 		if (!can_assume(LATEST) &&
202 		    fdt_version(fdt) < 0x10 && len >= 8 &&
203 		    ((offset - len) % 8) != 0)
204 			offset += 4;
205 		break;
206 
207 	case FDT_END:
208 	case FDT_END_NODE:
209 	case FDT_NOP:
210 		break;
211 
212 	default:
213 		return FDT_END;
214 	}
215 
216 	if (!can_assume(VALID_DTB) && (offset <= startoffset
217 	    || !fdt_offset_ptr(fdt, startoffset, offset - startoffset)))
218 		return FDT_END; /* premature end */
219 
220 	*nextoffset = FDT_TAGALIGN(offset);
221 	return tag;
222 }
223 
fdt_check_node_offset_(const void * fdt,int offset)224 int fdt_check_node_offset_(const void *fdt, int offset)
225 {
226 	if (!can_assume(VALID_INPUT)
227 	    && ((offset < 0) || (offset % FDT_TAGSIZE)))
228 		return -FDT_ERR_BADOFFSET;
229 
230 	if (fdt_next_tag(fdt, offset, &offset) != FDT_BEGIN_NODE)
231 		return -FDT_ERR_BADOFFSET;
232 
233 	return offset;
234 }
235 
fdt_check_prop_offset_(const void * fdt,int offset)236 int fdt_check_prop_offset_(const void *fdt, int offset)
237 {
238 	if (!can_assume(VALID_INPUT)
239 	    && ((offset < 0) || (offset % FDT_TAGSIZE)))
240 		return -FDT_ERR_BADOFFSET;
241 
242 	if (fdt_next_tag(fdt, offset, &offset) != FDT_PROP)
243 		return -FDT_ERR_BADOFFSET;
244 
245 	return offset;
246 }
247 
fdt_next_node(const void * fdt,int offset,int * depth)248 int fdt_next_node(const void *fdt, int offset, int *depth)
249 {
250 	int nextoffset = 0;
251 	uint32_t tag;
252 
253 	if (offset >= 0)
254 		if ((nextoffset = fdt_check_node_offset_(fdt, offset)) < 0)
255 			return nextoffset;
256 
257 	do {
258 		offset = nextoffset;
259 		tag = fdt_next_tag(fdt, offset, &nextoffset);
260 
261 		switch (tag) {
262 		case FDT_PROP:
263 		case FDT_NOP:
264 			break;
265 
266 		case FDT_BEGIN_NODE:
267 			if (depth)
268 				(*depth)++;
269 			break;
270 
271 		case FDT_END_NODE:
272 			if (depth && ((--(*depth)) < 0))
273 				return nextoffset;
274 			break;
275 
276 		case FDT_END:
277 			if ((nextoffset >= 0)
278 			    || ((nextoffset == -FDT_ERR_TRUNCATED) && !depth))
279 				return -FDT_ERR_NOTFOUND;
280 			else
281 				return nextoffset;
282 		}
283 	} while (tag != FDT_BEGIN_NODE);
284 
285 	return offset;
286 }
287 
fdt_first_subnode(const void * fdt,int offset)288 int fdt_first_subnode(const void *fdt, int offset)
289 {
290 	int depth = 0;
291 
292 	offset = fdt_next_node(fdt, offset, &depth);
293 	if (offset < 0 || depth != 1)
294 		return -FDT_ERR_NOTFOUND;
295 
296 	return offset;
297 }
298 
fdt_next_subnode(const void * fdt,int offset)299 int fdt_next_subnode(const void *fdt, int offset)
300 {
301 	int depth = 1;
302 
303 	/*
304 	 * With respect to the parent, the depth of the next subnode will be
305 	 * the same as the last.
306 	 */
307 	do {
308 		offset = fdt_next_node(fdt, offset, &depth);
309 		if (offset < 0 || depth < 1)
310 			return -FDT_ERR_NOTFOUND;
311 	} while (depth > 1);
312 
313 	return offset;
314 }
315 
fdt_find_string_(const char * strtab,int tabsize,const char * s)316 const char *fdt_find_string_(const char *strtab, int tabsize, const char *s)
317 {
318 	int len = strlen(s) + 1;
319 	const char *last = strtab + tabsize - len;
320 	const char *p;
321 
322 	for (p = strtab; p <= last; p++)
323 		if (memcmp(p, s, len) == 0)
324 			return p;
325 	return NULL;
326 }
327 
fdt_move(const void * fdt,void * buf,int bufsize)328 int fdt_move(const void *fdt, void *buf, int bufsize)
329 {
330 	if (!can_assume(VALID_INPUT) && bufsize < 0)
331 		return -FDT_ERR_NOSPACE;
332 
333 	FDT_RO_PROBE(fdt);
334 
335 	if (fdt_totalsize(fdt) > (unsigned int)bufsize)
336 		return -FDT_ERR_NOSPACE;
337 
338 	memmove(buf, fdt, fdt_totalsize(fdt));
339 	return 0;
340 }
341