1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * (C) Copyright David Gibson <dwg@au1.ibm.com>, IBM Corporation.  2005.
4  */
5 
6 #include "dtc.h"
7 #include "srcpos.h"
8 
9 #define FTF_FULLPATH	0x1
10 #define FTF_VARALIGN	0x2
11 #define FTF_NAMEPROPS	0x4
12 #define FTF_BOOTCPUID	0x8
13 #define FTF_STRTABSIZE	0x10
14 #define FTF_STRUCTSIZE	0x20
15 #define FTF_NOPS	0x40
16 
17 static struct version_info {
18 	int version;
19 	int last_comp_version;
20 	int hdr_size;
21 	int flags;
22 } version_table[] = {
23 	{1, 1, FDT_V1_SIZE,
24 	 FTF_FULLPATH|FTF_VARALIGN|FTF_NAMEPROPS},
25 	{2, 1, FDT_V2_SIZE,
26 	 FTF_FULLPATH|FTF_VARALIGN|FTF_NAMEPROPS|FTF_BOOTCPUID},
27 	{3, 1, FDT_V3_SIZE,
28 	 FTF_FULLPATH|FTF_VARALIGN|FTF_NAMEPROPS|FTF_BOOTCPUID|FTF_STRTABSIZE},
29 	{16, 16, FDT_V3_SIZE,
30 	 FTF_BOOTCPUID|FTF_STRTABSIZE|FTF_NOPS},
31 	{17, 16, FDT_V17_SIZE,
32 	 FTF_BOOTCPUID|FTF_STRTABSIZE|FTF_STRUCTSIZE|FTF_NOPS},
33 };
34 
35 struct emitter {
36 	void (*cell)(void *, cell_t);
37 	void (*string)(void *, const char *, int);
38 	void (*align)(void *, int);
39 	void (*data)(void *, struct data);
40 	void (*beginnode)(void *, struct label *labels);
41 	void (*endnode)(void *, struct label *labels);
42 	void (*property)(void *, struct label *labels);
43 };
44 
bin_emit_cell(void * e,cell_t val)45 static void bin_emit_cell(void *e, cell_t val)
46 {
47 	struct data *dtbuf = e;
48 
49 	*dtbuf = data_append_cell(*dtbuf, val);
50 }
51 
bin_emit_string(void * e,const char * str,int len)52 static void bin_emit_string(void *e, const char *str, int len)
53 {
54 	struct data *dtbuf = e;
55 
56 	if (len == 0)
57 		len = strlen(str);
58 
59 	*dtbuf = data_append_data(*dtbuf, str, len);
60 	*dtbuf = data_append_byte(*dtbuf, '\0');
61 }
62 
bin_emit_align(void * e,int a)63 static void bin_emit_align(void *e, int a)
64 {
65 	struct data *dtbuf = e;
66 
67 	*dtbuf = data_append_align(*dtbuf, a);
68 }
69 
bin_emit_data(void * e,struct data d)70 static void bin_emit_data(void *e, struct data d)
71 {
72 	struct data *dtbuf = e;
73 
74 	*dtbuf = data_append_data(*dtbuf, d.val, d.len);
75 }
76 
bin_emit_beginnode(void * e,struct label * labels)77 static void bin_emit_beginnode(void *e, struct label *labels)
78 {
79 	bin_emit_cell(e, FDT_BEGIN_NODE);
80 }
81 
bin_emit_endnode(void * e,struct label * labels)82 static void bin_emit_endnode(void *e, struct label *labels)
83 {
84 	bin_emit_cell(e, FDT_END_NODE);
85 }
86 
bin_emit_property(void * e,struct label * labels)87 static void bin_emit_property(void *e, struct label *labels)
88 {
89 	bin_emit_cell(e, FDT_PROP);
90 }
91 
92 static struct emitter bin_emitter = {
93 	.cell = bin_emit_cell,
94 	.string = bin_emit_string,
95 	.align = bin_emit_align,
96 	.data = bin_emit_data,
97 	.beginnode = bin_emit_beginnode,
98 	.endnode = bin_emit_endnode,
99 	.property = bin_emit_property,
100 };
101 
emit_label(FILE * f,const char * prefix,const char * label)102 static void emit_label(FILE *f, const char *prefix, const char *label)
103 {
104 	fprintf(f, "\t.globl\t%s_%s\n", prefix, label);
105 	fprintf(f, "%s_%s:\n", prefix, label);
106 	fprintf(f, "_%s_%s:\n", prefix, label);
107 }
108 
emit_offset_label(FILE * f,const char * label,int offset)109 static void emit_offset_label(FILE *f, const char *label, int offset)
110 {
111 	fprintf(f, "\t.globl\t%s\n", label);
112 	fprintf(f, "%s\t= . + %d\n", label, offset);
113 }
114 
115 #define ASM_EMIT_BELONG(f, fmt, ...) \
116 	{ \
117 		fprintf((f), "\t.byte\t((" fmt ") >> 24) & 0xff\n", __VA_ARGS__); \
118 		fprintf((f), "\t.byte\t((" fmt ") >> 16) & 0xff\n", __VA_ARGS__); \
119 		fprintf((f), "\t.byte\t((" fmt ") >> 8) & 0xff\n", __VA_ARGS__); \
120 		fprintf((f), "\t.byte\t(" fmt ") & 0xff\n", __VA_ARGS__); \
121 	}
122 
asm_emit_cell(void * e,cell_t val)123 static void asm_emit_cell(void *e, cell_t val)
124 {
125 	FILE *f = e;
126 
127 	fprintf(f, "\t.byte\t0x%02x\n" "\t.byte\t0x%02x\n"
128 		"\t.byte\t0x%02x\n" "\t.byte\t0x%02x\n",
129 		(val >> 24) & 0xff, (val >> 16) & 0xff,
130 		(val >> 8) & 0xff, val & 0xff);
131 }
132 
asm_emit_string(void * e,const char * str,int len)133 static void asm_emit_string(void *e, const char *str, int len)
134 {
135 	FILE *f = e;
136 
137 	if (len != 0)
138 		fprintf(f, "\t.asciz\t\"%.*s\"\n", len, str);
139 	else
140 		fprintf(f, "\t.asciz\t\"%s\"\n", str);
141 }
142 
asm_emit_align(void * e,int a)143 static void asm_emit_align(void *e, int a)
144 {
145 	FILE *f = e;
146 
147 	fprintf(f, "\t.balign\t%d, 0\n", a);
148 }
149 
asm_emit_data(void * e,struct data d)150 static void asm_emit_data(void *e, struct data d)
151 {
152 	FILE *f = e;
153 	unsigned int off = 0;
154 	struct marker *m = d.markers;
155 
156 	for_each_marker_of_type(m, LABEL)
157 		emit_offset_label(f, m->ref, m->offset);
158 
159 	while ((d.len - off) >= sizeof(uint32_t)) {
160 		asm_emit_cell(e, dtb_ld32(d.val + off));
161 		off += sizeof(uint32_t);
162 	}
163 
164 	while ((d.len - off) >= 1) {
165 		fprintf(f, "\t.byte\t0x%hhx\n", d.val[off]);
166 		off += 1;
167 	}
168 
169 	assert(off == d.len);
170 }
171 
asm_emit_beginnode(void * e,struct label * labels)172 static void asm_emit_beginnode(void *e, struct label *labels)
173 {
174 	FILE *f = e;
175 	struct label *l;
176 
177 	for_each_label(labels, l) {
178 		fprintf(f, "\t.globl\t%s\n", l->label);
179 		fprintf(f, "%s:\n", l->label);
180 	}
181 	fprintf(f, "\t/* FDT_BEGIN_NODE */\n");
182 	asm_emit_cell(e, FDT_BEGIN_NODE);
183 }
184 
asm_emit_endnode(void * e,struct label * labels)185 static void asm_emit_endnode(void *e, struct label *labels)
186 {
187 	FILE *f = e;
188 	struct label *l;
189 
190 	fprintf(f, "\t/* FDT_END_NODE */\n");
191 	asm_emit_cell(e, FDT_END_NODE);
192 	for_each_label(labels, l) {
193 		fprintf(f, "\t.globl\t%s_end\n", l->label);
194 		fprintf(f, "%s_end:\n", l->label);
195 	}
196 }
197 
asm_emit_property(void * e,struct label * labels)198 static void asm_emit_property(void *e, struct label *labels)
199 {
200 	FILE *f = e;
201 	struct label *l;
202 
203 	for_each_label(labels, l) {
204 		fprintf(f, "\t.globl\t%s\n", l->label);
205 		fprintf(f, "%s:\n", l->label);
206 	}
207 	fprintf(f, "\t/* FDT_PROP */\n");
208 	asm_emit_cell(e, FDT_PROP);
209 }
210 
211 static struct emitter asm_emitter = {
212 	.cell = asm_emit_cell,
213 	.string = asm_emit_string,
214 	.align = asm_emit_align,
215 	.data = asm_emit_data,
216 	.beginnode = asm_emit_beginnode,
217 	.endnode = asm_emit_endnode,
218 	.property = asm_emit_property,
219 };
220 
stringtable_insert(struct data * d,const char * str)221 static int stringtable_insert(struct data *d, const char *str)
222 {
223 	unsigned int i;
224 
225 	/* FIXME: do this more efficiently? */
226 
227 	for (i = 0; i < d->len; i++) {
228 		if (streq(str, d->val + i))
229 			return i;
230 	}
231 
232 	*d = data_append_data(*d, str, strlen(str)+1);
233 	return i;
234 }
235 
flatten_tree(struct node * tree,struct emitter * emit,void * etarget,struct data * strbuf,struct version_info * vi)236 static void flatten_tree(struct node *tree, struct emitter *emit,
237 			 void *etarget, struct data *strbuf,
238 			 struct version_info *vi)
239 {
240 	struct property *prop;
241 	struct node *child;
242 	bool seen_name_prop = false;
243 
244 	if (tree->deleted)
245 		return;
246 
247 	emit->beginnode(etarget, tree->labels);
248 
249 	if (vi->flags & FTF_FULLPATH)
250 		emit->string(etarget, tree->fullpath, 0);
251 	else
252 		emit->string(etarget, tree->name, 0);
253 
254 	emit->align(etarget, sizeof(cell_t));
255 
256 	for_each_property(tree, prop) {
257 		int nameoff;
258 
259 		if (streq(prop->name, "name"))
260 			seen_name_prop = true;
261 
262 		nameoff = stringtable_insert(strbuf, prop->name);
263 
264 		emit->property(etarget, prop->labels);
265 		emit->cell(etarget, prop->val.len);
266 		emit->cell(etarget, nameoff);
267 
268 		if ((vi->flags & FTF_VARALIGN) && (prop->val.len >= 8))
269 			emit->align(etarget, 8);
270 
271 		emit->data(etarget, prop->val);
272 		emit->align(etarget, sizeof(cell_t));
273 	}
274 
275 	if ((vi->flags & FTF_NAMEPROPS) && !seen_name_prop) {
276 		emit->property(etarget, NULL);
277 		emit->cell(etarget, tree->basenamelen+1);
278 		emit->cell(etarget, stringtable_insert(strbuf, "name"));
279 
280 		if ((vi->flags & FTF_VARALIGN) && ((tree->basenamelen+1) >= 8))
281 			emit->align(etarget, 8);
282 
283 		emit->string(etarget, tree->name, tree->basenamelen);
284 		emit->align(etarget, sizeof(cell_t));
285 	}
286 
287 	for_each_child(tree, child) {
288 		flatten_tree(child, emit, etarget, strbuf, vi);
289 	}
290 
291 	emit->endnode(etarget, tree->labels);
292 }
293 
flatten_reserve_list(struct reserve_info * reservelist,struct version_info * vi)294 static struct data flatten_reserve_list(struct reserve_info *reservelist,
295 				 struct version_info *vi)
296 {
297 	struct reserve_info *re;
298 	struct data d = empty_data;
299 	unsigned int j;
300 
301 	for (re = reservelist; re; re = re->next) {
302 		d = data_append_re(d, re->address, re->size);
303 	}
304 	/*
305 	 * Add additional reserved slots if the user asked for them.
306 	 */
307 	for (j = 0; j < reservenum; j++) {
308 		d = data_append_re(d, 0, 0);
309 	}
310 
311 	return d;
312 }
313 
make_fdt_header(struct fdt_header * fdt,struct version_info * vi,int reservesize,int dtsize,int strsize,int boot_cpuid_phys)314 static void make_fdt_header(struct fdt_header *fdt,
315 			    struct version_info *vi,
316 			    int reservesize, int dtsize, int strsize,
317 			    int boot_cpuid_phys)
318 {
319 	int reserve_off;
320 
321 	reservesize += sizeof(struct fdt_reserve_entry);
322 
323 	memset(fdt, 0xff, sizeof(*fdt));
324 
325 	fdt->magic = cpu_to_fdt32(FDT_MAGIC);
326 	fdt->version = cpu_to_fdt32(vi->version);
327 	fdt->last_comp_version = cpu_to_fdt32(vi->last_comp_version);
328 
329 	/* Reserve map should be doubleword aligned */
330 	reserve_off = ALIGN(vi->hdr_size, 8);
331 
332 	fdt->off_mem_rsvmap = cpu_to_fdt32(reserve_off);
333 	fdt->off_dt_struct = cpu_to_fdt32(reserve_off + reservesize);
334 	fdt->off_dt_strings = cpu_to_fdt32(reserve_off + reservesize
335 					  + dtsize);
336 	fdt->totalsize = cpu_to_fdt32(reserve_off + reservesize + dtsize + strsize);
337 
338 	if (vi->flags & FTF_BOOTCPUID)
339 		fdt->boot_cpuid_phys = cpu_to_fdt32(boot_cpuid_phys);
340 	if (vi->flags & FTF_STRTABSIZE)
341 		fdt->size_dt_strings = cpu_to_fdt32(strsize);
342 	if (vi->flags & FTF_STRUCTSIZE)
343 		fdt->size_dt_struct = cpu_to_fdt32(dtsize);
344 }
345 
dt_to_blob(FILE * f,struct dt_info * dti,int version)346 void dt_to_blob(FILE *f, struct dt_info *dti, int version)
347 {
348 	struct version_info *vi = NULL;
349 	unsigned int i;
350 	struct data blob       = empty_data;
351 	struct data reservebuf = empty_data;
352 	struct data dtbuf      = empty_data;
353 	struct data strbuf     = empty_data;
354 	struct fdt_header fdt;
355 	int padlen = 0;
356 
357 	for (i = 0; i < ARRAY_SIZE(version_table); i++) {
358 		if (version_table[i].version == version)
359 			vi = &version_table[i];
360 	}
361 	if (!vi)
362 		die("Unknown device tree blob version %d\n", version);
363 
364 	flatten_tree(dti->dt, &bin_emitter, &dtbuf, &strbuf, vi);
365 	bin_emit_cell(&dtbuf, FDT_END);
366 
367 	reservebuf = flatten_reserve_list(dti->reservelist, vi);
368 
369 	/* Make header */
370 	make_fdt_header(&fdt, vi, reservebuf.len, dtbuf.len, strbuf.len,
371 			dti->boot_cpuid_phys);
372 
373 	/*
374 	 * If the user asked for more space than is used, adjust the totalsize.
375 	 */
376 	if (minsize > 0) {
377 		padlen = minsize - fdt32_to_cpu(fdt.totalsize);
378 		if (padlen < 0) {
379 			padlen = 0;
380 			if (quiet < 1)
381 				fprintf(stderr,
382 					"Warning: blob size %"PRIu32" >= minimum size %d\n",
383 					fdt32_to_cpu(fdt.totalsize), minsize);
384 		}
385 	}
386 
387 	if (padsize > 0)
388 		padlen = padsize;
389 
390 	if (alignsize > 0)
391 		padlen = ALIGN(fdt32_to_cpu(fdt.totalsize) + padlen, alignsize)
392 			- fdt32_to_cpu(fdt.totalsize);
393 
394 	if (padlen > 0) {
395 		int tsize = fdt32_to_cpu(fdt.totalsize);
396 		tsize += padlen;
397 		fdt.totalsize = cpu_to_fdt32(tsize);
398 	}
399 
400 	/*
401 	 * Assemble the blob: start with the header, add with alignment
402 	 * the reserve buffer, add the reserve map terminating zeroes,
403 	 * the device tree itself, and finally the strings.
404 	 */
405 	blob = data_append_data(blob, &fdt, vi->hdr_size);
406 	blob = data_append_align(blob, 8);
407 	blob = data_merge(blob, reservebuf);
408 	blob = data_append_zeroes(blob, sizeof(struct fdt_reserve_entry));
409 	blob = data_merge(blob, dtbuf);
410 	blob = data_merge(blob, strbuf);
411 
412 	/*
413 	 * If the user asked for more space than is used, pad out the blob.
414 	 */
415 	if (padlen > 0)
416 		blob = data_append_zeroes(blob, padlen);
417 
418 	if (fwrite(blob.val, blob.len, 1, f) != 1) {
419 		if (ferror(f))
420 			die("Error writing device tree blob: %s\n",
421 			    strerror(errno));
422 		else
423 			die("Short write on device tree blob\n");
424 	}
425 
426 	/*
427 	 * data_merge() frees the right-hand element so only the blob
428 	 * remains to be freed.
429 	 */
430 	data_free(blob);
431 }
432 
dump_stringtable_asm(FILE * f,struct data strbuf)433 static void dump_stringtable_asm(FILE *f, struct data strbuf)
434 {
435 	const char *p;
436 	int len;
437 
438 	p = strbuf.val;
439 
440 	while (p < (strbuf.val + strbuf.len)) {
441 		len = strlen(p);
442 		fprintf(f, "\t.asciz \"%s\"\n", p);
443 		p += len+1;
444 	}
445 }
446 
dt_to_asm(FILE * f,struct dt_info * dti,int version)447 void dt_to_asm(FILE *f, struct dt_info *dti, int version)
448 {
449 	struct version_info *vi = NULL;
450 	unsigned int i;
451 	struct data strbuf = empty_data;
452 	struct reserve_info *re;
453 	const char *symprefix = "dt";
454 
455 	for (i = 0; i < ARRAY_SIZE(version_table); i++) {
456 		if (version_table[i].version == version)
457 			vi = &version_table[i];
458 	}
459 	if (!vi)
460 		die("Unknown device tree blob version %d\n", version);
461 
462 	fprintf(f, "/* autogenerated by dtc, do not edit */\n\n");
463 
464 	emit_label(f, symprefix, "blob_start");
465 	emit_label(f, symprefix, "header");
466 	fprintf(f, "\t/* magic */\n");
467 	asm_emit_cell(f, FDT_MAGIC);
468 	fprintf(f, "\t/* totalsize */\n");
469 	ASM_EMIT_BELONG(f, "_%s_blob_abs_end - _%s_blob_start",
470 			symprefix, symprefix);
471 	fprintf(f, "\t/* off_dt_struct */\n");
472 	ASM_EMIT_BELONG(f, "_%s_struct_start - _%s_blob_start",
473 		symprefix, symprefix);
474 	fprintf(f, "\t/* off_dt_strings */\n");
475 	ASM_EMIT_BELONG(f, "_%s_strings_start - _%s_blob_start",
476 		symprefix, symprefix);
477 	fprintf(f, "\t/* off_mem_rsvmap */\n");
478 	ASM_EMIT_BELONG(f, "_%s_reserve_map - _%s_blob_start",
479 		symprefix, symprefix);
480 	fprintf(f, "\t/* version */\n");
481 	asm_emit_cell(f, vi->version);
482 	fprintf(f, "\t/* last_comp_version */\n");
483 	asm_emit_cell(f, vi->last_comp_version);
484 
485 	if (vi->flags & FTF_BOOTCPUID) {
486 		fprintf(f, "\t/* boot_cpuid_phys */\n");
487 		asm_emit_cell(f, dti->boot_cpuid_phys);
488 	}
489 
490 	if (vi->flags & FTF_STRTABSIZE) {
491 		fprintf(f, "\t/* size_dt_strings */\n");
492 		ASM_EMIT_BELONG(f, "_%s_strings_end - _%s_strings_start",
493 				symprefix, symprefix);
494 	}
495 
496 	if (vi->flags & FTF_STRUCTSIZE) {
497 		fprintf(f, "\t/* size_dt_struct */\n");
498 		ASM_EMIT_BELONG(f, "_%s_struct_end - _%s_struct_start",
499 			symprefix, symprefix);
500 	}
501 
502 	/*
503 	 * Reserve map entries.
504 	 * Align the reserve map to a doubleword boundary.
505 	 * Each entry is an (address, size) pair of u64 values.
506 	 * Always supply a zero-sized temination entry.
507 	 */
508 	asm_emit_align(f, 8);
509 	emit_label(f, symprefix, "reserve_map");
510 
511 	fprintf(f, "/* Memory reserve map from source file */\n");
512 
513 	/*
514 	 * Use .long on high and low halves of u64s to avoid .quad
515 	 * as it appears .quad isn't available in some assemblers.
516 	 */
517 	for (re = dti->reservelist; re; re = re->next) {
518 		struct label *l;
519 
520 		for_each_label(re->labels, l) {
521 			fprintf(f, "\t.globl\t%s\n", l->label);
522 			fprintf(f, "%s:\n", l->label);
523 		}
524 		ASM_EMIT_BELONG(f, "0x%08x", (unsigned int)(re->address >> 32));
525 		ASM_EMIT_BELONG(f, "0x%08x",
526 				(unsigned int)(re->address & 0xffffffff));
527 		ASM_EMIT_BELONG(f, "0x%08x", (unsigned int)(re->size >> 32));
528 		ASM_EMIT_BELONG(f, "0x%08x", (unsigned int)(re->size & 0xffffffff));
529 	}
530 	for (i = 0; i < reservenum; i++) {
531 		fprintf(f, "\t.long\t0, 0\n\t.long\t0, 0\n");
532 	}
533 
534 	fprintf(f, "\t.long\t0, 0\n\t.long\t0, 0\n");
535 
536 	emit_label(f, symprefix, "struct_start");
537 	flatten_tree(dti->dt, &asm_emitter, f, &strbuf, vi);
538 
539 	fprintf(f, "\t/* FDT_END */\n");
540 	asm_emit_cell(f, FDT_END);
541 	emit_label(f, symprefix, "struct_end");
542 
543 	emit_label(f, symprefix, "strings_start");
544 	dump_stringtable_asm(f, strbuf);
545 	emit_label(f, symprefix, "strings_end");
546 
547 	emit_label(f, symprefix, "blob_end");
548 
549 	/*
550 	 * If the user asked for more space than is used, pad it out.
551 	 */
552 	if (minsize > 0) {
553 		fprintf(f, "\t.space\t%d - (_%s_blob_end - _%s_blob_start), 0\n",
554 			minsize, symprefix, symprefix);
555 	}
556 	if (padsize > 0) {
557 		fprintf(f, "\t.space\t%d, 0\n", padsize);
558 	}
559 	if (alignsize > 0)
560 		asm_emit_align(f, alignsize);
561 	emit_label(f, symprefix, "blob_abs_end");
562 
563 	data_free(strbuf);
564 }
565 
566 struct inbuf {
567 	char *base, *limit, *ptr;
568 };
569 
inbuf_init(struct inbuf * inb,void * base,void * limit)570 static void inbuf_init(struct inbuf *inb, void *base, void *limit)
571 {
572 	inb->base = base;
573 	inb->limit = limit;
574 	inb->ptr = inb->base;
575 }
576 
flat_read_chunk(struct inbuf * inb,void * p,int len)577 static void flat_read_chunk(struct inbuf *inb, void *p, int len)
578 {
579 	if ((inb->ptr + len) > inb->limit)
580 		die("Premature end of data parsing flat device tree\n");
581 
582 	memcpy(p, inb->ptr, len);
583 
584 	inb->ptr += len;
585 }
586 
flat_read_word(struct inbuf * inb)587 static uint32_t flat_read_word(struct inbuf *inb)
588 {
589 	fdt32_t val;
590 
591 	assert(((inb->ptr - inb->base) % sizeof(val)) == 0);
592 
593 	flat_read_chunk(inb, &val, sizeof(val));
594 
595 	return fdt32_to_cpu(val);
596 }
597 
flat_realign(struct inbuf * inb,int align)598 static void flat_realign(struct inbuf *inb, int align)
599 {
600 	int off = inb->ptr - inb->base;
601 
602 	inb->ptr = inb->base + ALIGN(off, align);
603 	if (inb->ptr > inb->limit)
604 		die("Premature end of data parsing flat device tree\n");
605 }
606 
flat_read_string(struct inbuf * inb)607 static char *flat_read_string(struct inbuf *inb)
608 {
609 	int len = 0;
610 	const char *p = inb->ptr;
611 	char *str;
612 
613 	do {
614 		if (p >= inb->limit)
615 			die("Premature end of data parsing flat device tree\n");
616 		len++;
617 	} while ((*p++) != '\0');
618 
619 	str = xstrdup(inb->ptr);
620 
621 	inb->ptr += len;
622 
623 	flat_realign(inb, sizeof(uint32_t));
624 
625 	return str;
626 }
627 
flat_read_data(struct inbuf * inb,int len)628 static struct data flat_read_data(struct inbuf *inb, int len)
629 {
630 	struct data d = empty_data;
631 
632 	if (len == 0)
633 		return empty_data;
634 
635 	d = data_grow_for(d, len);
636 	d.len = len;
637 
638 	flat_read_chunk(inb, d.val, len);
639 
640 	flat_realign(inb, sizeof(uint32_t));
641 
642 	return d;
643 }
644 
flat_read_stringtable(struct inbuf * inb,int offset)645 static char *flat_read_stringtable(struct inbuf *inb, int offset)
646 {
647 	const char *p;
648 
649 	p = inb->base + offset;
650 	while (1) {
651 		if (p >= inb->limit || p < inb->base)
652 			die("String offset %d overruns string table\n",
653 			    offset);
654 
655 		if (*p == '\0')
656 			break;
657 
658 		p++;
659 	}
660 
661 	return xstrdup(inb->base + offset);
662 }
663 
flat_read_property(struct inbuf * dtbuf,struct inbuf * strbuf,int flags)664 static struct property *flat_read_property(struct inbuf *dtbuf,
665 					   struct inbuf *strbuf, int flags)
666 {
667 	uint32_t proplen, stroff;
668 	char *name;
669 	struct data val;
670 
671 	proplen = flat_read_word(dtbuf);
672 	stroff = flat_read_word(dtbuf);
673 
674 	name = flat_read_stringtable(strbuf, stroff);
675 
676 	if ((flags & FTF_VARALIGN) && (proplen >= 8))
677 		flat_realign(dtbuf, 8);
678 
679 	val = flat_read_data(dtbuf, proplen);
680 
681 	return build_property(name, val, NULL);
682 }
683 
684 
flat_read_mem_reserve(struct inbuf * inb)685 static struct reserve_info *flat_read_mem_reserve(struct inbuf *inb)
686 {
687 	struct reserve_info *reservelist = NULL;
688 	struct reserve_info *new;
689 	struct fdt_reserve_entry re;
690 
691 	/*
692 	 * Each entry is a pair of u64 (addr, size) values for 4 cell_t's.
693 	 * List terminates at an entry with size equal to zero.
694 	 *
695 	 * First pass, count entries.
696 	 */
697 	while (1) {
698 		uint64_t address, size;
699 
700 		flat_read_chunk(inb, &re, sizeof(re));
701 		address  = fdt64_to_cpu(re.address);
702 		size = fdt64_to_cpu(re.size);
703 		if (size == 0)
704 			break;
705 
706 		new = build_reserve_entry(address, size);
707 		reservelist = add_reserve_entry(reservelist, new);
708 	}
709 
710 	return reservelist;
711 }
712 
713 
nodename_from_path(const char * ppath,const char * cpath)714 static char *nodename_from_path(const char *ppath, const char *cpath)
715 {
716 	int plen;
717 
718 	plen = strlen(ppath);
719 
720 	if (!strstarts(cpath, ppath))
721 		die("Path \"%s\" is not valid as a child of \"%s\"\n",
722 		    cpath, ppath);
723 
724 	/* root node is a special case */
725 	if (!streq(ppath, "/"))
726 		plen++;
727 
728 	return xstrdup(cpath + plen);
729 }
730 
unflatten_tree(struct inbuf * dtbuf,struct inbuf * strbuf,const char * parent_flatname,int flags)731 static struct node *unflatten_tree(struct inbuf *dtbuf,
732 				   struct inbuf *strbuf,
733 				   const char *parent_flatname, int flags)
734 {
735 	struct node *node;
736 	char *flatname;
737 	uint32_t val;
738 
739 	node = build_node(NULL, NULL, NULL);
740 
741 	flatname = flat_read_string(dtbuf);
742 
743 	if (flags & FTF_FULLPATH)
744 		node->name = nodename_from_path(parent_flatname, flatname);
745 	else
746 		node->name = flatname;
747 
748 	do {
749 		struct property *prop;
750 		struct node *child;
751 
752 		val = flat_read_word(dtbuf);
753 		switch (val) {
754 		case FDT_PROP:
755 			if (node->children)
756 				fprintf(stderr, "Warning: Flat tree input has "
757 					"subnodes preceding a property.\n");
758 			prop = flat_read_property(dtbuf, strbuf, flags);
759 			add_property(node, prop);
760 			break;
761 
762 		case FDT_BEGIN_NODE:
763 			child = unflatten_tree(dtbuf,strbuf, flatname, flags);
764 			add_child(node, child);
765 			break;
766 
767 		case FDT_END_NODE:
768 			break;
769 
770 		case FDT_END:
771 			die("Premature FDT_END in device tree blob\n");
772 			break;
773 
774 		case FDT_NOP:
775 			if (!(flags & FTF_NOPS))
776 				fprintf(stderr, "Warning: NOP tag found in flat tree"
777 					" version <16\n");
778 
779 			/* Ignore */
780 			break;
781 
782 		default:
783 			die("Invalid opcode word %08x in device tree blob\n",
784 			    val);
785 		}
786 	} while (val != FDT_END_NODE);
787 
788 	if (node->name != flatname) {
789 		free(flatname);
790 	}
791 
792 	return node;
793 }
794 
795 
dt_from_blob(const char * fname)796 struct dt_info *dt_from_blob(const char *fname)
797 {
798 	FILE *f;
799 	fdt32_t magic_buf, totalsize_buf;
800 	uint32_t magic, totalsize, version, size_dt, boot_cpuid_phys;
801 	uint32_t off_dt, off_str, off_mem_rsvmap;
802 	int rc;
803 	char *blob;
804 	struct fdt_header *fdt;
805 	char *p;
806 	struct inbuf dtbuf, strbuf;
807 	struct inbuf memresvbuf;
808 	int sizeleft;
809 	struct reserve_info *reservelist;
810 	struct node *tree;
811 	uint32_t val;
812 	int flags = 0;
813 
814 	f = srcfile_relative_open(fname, NULL);
815 
816 	rc = fread(&magic_buf, sizeof(magic_buf), 1, f);
817 	if (ferror(f))
818 		die("Error reading DT blob magic number: %s\n",
819 		    strerror(errno));
820 	if (rc < 1) {
821 		if (feof(f))
822 			die("EOF reading DT blob magic number\n");
823 		else
824 			die("Mysterious short read reading magic number\n");
825 	}
826 
827 	magic = fdt32_to_cpu(magic_buf);
828 	if (magic != FDT_MAGIC)
829 		die("Blob has incorrect magic number\n");
830 
831 	rc = fread(&totalsize_buf, sizeof(totalsize_buf), 1, f);
832 	if (ferror(f))
833 		die("Error reading DT blob size: %s\n", strerror(errno));
834 	if (rc < 1) {
835 		if (feof(f))
836 			die("EOF reading DT blob size\n");
837 		else
838 			die("Mysterious short read reading blob size\n");
839 	}
840 
841 	totalsize = fdt32_to_cpu(totalsize_buf);
842 	if (totalsize < FDT_V1_SIZE)
843 		die("DT blob size (%d) is too small\n", totalsize);
844 
845 	blob = xmalloc(totalsize);
846 
847 	fdt = (struct fdt_header *)blob;
848 	fdt->magic = cpu_to_fdt32(magic);
849 	fdt->totalsize = cpu_to_fdt32(totalsize);
850 
851 	sizeleft = totalsize - sizeof(magic) - sizeof(totalsize);
852 	p = blob + sizeof(magic)  + sizeof(totalsize);
853 
854 	while (sizeleft) {
855 		if (feof(f))
856 			die("EOF before reading %d bytes of DT blob\n",
857 			    totalsize);
858 
859 		rc = fread(p, 1, sizeleft, f);
860 		if (ferror(f))
861 			die("Error reading DT blob: %s\n",
862 			    strerror(errno));
863 
864 		sizeleft -= rc;
865 		p += rc;
866 	}
867 
868 	off_dt = fdt32_to_cpu(fdt->off_dt_struct);
869 	off_str = fdt32_to_cpu(fdt->off_dt_strings);
870 	off_mem_rsvmap = fdt32_to_cpu(fdt->off_mem_rsvmap);
871 	version = fdt32_to_cpu(fdt->version);
872 	boot_cpuid_phys = fdt32_to_cpu(fdt->boot_cpuid_phys);
873 
874 	if (off_mem_rsvmap >= totalsize)
875 		die("Mem Reserve structure offset exceeds total size\n");
876 
877 	if (off_dt >= totalsize)
878 		die("DT structure offset exceeds total size\n");
879 
880 	if (off_str > totalsize)
881 		die("String table offset exceeds total size\n");
882 
883 	if (version >= 3) {
884 		uint32_t size_str = fdt32_to_cpu(fdt->size_dt_strings);
885 		if ((off_str+size_str < off_str) || (off_str+size_str > totalsize))
886 			die("String table extends past total size\n");
887 		inbuf_init(&strbuf, blob + off_str, blob + off_str + size_str);
888 	} else {
889 		inbuf_init(&strbuf, blob + off_str, blob + totalsize);
890 	}
891 
892 	if (version >= 17) {
893 		size_dt = fdt32_to_cpu(fdt->size_dt_struct);
894 		if ((off_dt+size_dt < off_dt) || (off_dt+size_dt > totalsize))
895 			die("Structure block extends past total size\n");
896 	}
897 
898 	if (version < 16) {
899 		flags |= FTF_FULLPATH | FTF_NAMEPROPS | FTF_VARALIGN;
900 	} else {
901 		flags |= FTF_NOPS;
902 	}
903 
904 	inbuf_init(&memresvbuf,
905 		   blob + off_mem_rsvmap, blob + totalsize);
906 	inbuf_init(&dtbuf, blob + off_dt, blob + totalsize);
907 
908 	reservelist = flat_read_mem_reserve(&memresvbuf);
909 
910 	val = flat_read_word(&dtbuf);
911 
912 	if (val != FDT_BEGIN_NODE)
913 		die("Device tree blob doesn't begin with FDT_BEGIN_NODE (begins with 0x%08x)\n", val);
914 
915 	tree = unflatten_tree(&dtbuf, &strbuf, "", flags);
916 
917 	val = flat_read_word(&dtbuf);
918 	if (val != FDT_END)
919 		die("Device tree blob doesn't end with FDT_END\n");
920 
921 	free(blob);
922 
923 	fclose(f);
924 
925 	return build_dt_info(DTSF_V1, reservelist, tree, boot_cpuid_phys);
926 }
927