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