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