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 <sys/stat.h>
22
23#include "dtc.h"
24#include "srcpos.h"
25
26/*
27 * Command line options
28 */
29int quiet;		/* Level of quietness */
30int reservenum;		/* Number of memory reservation slots */
31int minsize;		/* Minimum blob size */
32int padsize;		/* Additional padding to blob */
33int alignsize;		/* Additional padding to blob accroding to the alignsize */
34int phandle_format = PHANDLE_EPAPR;	/* Use linux,phandle or phandle properties */
35int generate_symbols;	/* enable symbols & fixup support */
36int generate_fixups;		/* suppress generation of fixups on symbol support */
37int auto_label_aliases;		/* auto generate labels -> aliases */
38
39static int is_power_of_2(int x)
40{
41	return (x > 0) && ((x & (x - 1)) == 0);
42}
43
44static void fill_fullpaths(struct node *tree, const char *prefix)
45{
46	struct node *child;
47	const char *unit;
48
49	tree->fullpath = join_path(prefix, tree->name);
50
51	unit = strchr(tree->name, '@');
52	if (unit)
53		tree->basenamelen = unit - tree->name;
54	else
55		tree->basenamelen = strlen(tree->name);
56
57	for_each_child(tree, child)
58		fill_fullpaths(child, tree->fullpath);
59}
60
61/* Usage related data. */
62static const char usage_synopsis[] = "dtc [options] <input file>";
63static const char usage_short_opts[] = "qI:O:o:V:d:R:S:p:a:fb:i:H:sW:E:@Ahv";
64static struct option const usage_long_opts[] = {
65	{"quiet",            no_argument, NULL, 'q'},
66	{"in-format",         a_argument, NULL, 'I'},
67	{"out",               a_argument, NULL, 'o'},
68	{"out-format",        a_argument, NULL, 'O'},
69	{"out-version",       a_argument, NULL, 'V'},
70	{"out-dependency",    a_argument, NULL, 'd'},
71	{"reserve",           a_argument, NULL, 'R'},
72	{"space",             a_argument, NULL, 'S'},
73	{"pad",               a_argument, NULL, 'p'},
74	{"align",             a_argument, NULL, 'a'},
75	{"boot-cpu",          a_argument, NULL, 'b'},
76	{"force",            no_argument, NULL, 'f'},
77	{"include",           a_argument, NULL, 'i'},
78	{"sort",             no_argument, NULL, 's'},
79	{"phandle",           a_argument, NULL, 'H'},
80	{"warning",           a_argument, NULL, 'W'},
81	{"error",             a_argument, NULL, 'E'},
82	{"symbols",	     no_argument, NULL, '@'},
83	{"auto-alias",       no_argument, NULL, 'A'},
84	{"help",             no_argument, NULL, 'h'},
85	{"version",          no_argument, NULL, 'v'},
86	{NULL,               no_argument, NULL, 0x0},
87};
88static const char * const usage_opts_help[] = {
89	"\n\tQuiet: -q suppress warnings, -qq errors, -qqq all",
90	"\n\tInput formats are:\n"
91	 "\t\tdts - device tree source text\n"
92	 "\t\tdtb - device tree blob\n"
93	 "\t\tfs  - /proc/device-tree style directory",
94	"\n\tOutput file",
95	"\n\tOutput formats are:\n"
96	 "\t\tdts - device tree source text\n"
97	 "\t\tdtb - device tree blob\n"
98	 "\t\tasm - assembler source",
99	"\n\tBlob version to produce, defaults to "stringify(DEFAULT_FDT_VERSION)" (for dtb and asm output)",
100	"\n\tOutput dependency file",
101	"\n\tMake space for <number> reserve map entries (for dtb and asm output)",
102	"\n\tMake the blob at least <bytes> long (extra space)",
103	"\n\tAdd padding to the blob of <bytes> long (extra space)",
104	"\n\tMake the blob align to the <bytes> (extra space)",
105	"\n\tSet the physical boot cpu",
106	"\n\tTry to produce output even if the input tree has errors",
107	"\n\tAdd a path to search for include files",
108	"\n\tSort nodes and properties before outputting (useful for comparing trees)",
109	"\n\tValid phandle formats are:\n"
110	 "\t\tlegacy - \"linux,phandle\" properties only\n"
111	 "\t\tepapr  - \"phandle\" properties only\n"
112	 "\t\tboth   - Both \"linux,phandle\" and \"phandle\" properties",
113	"\n\tEnable/disable warnings (prefix with \"no-\")",
114	"\n\tEnable/disable errors (prefix with \"no-\")",
115	"\n\tEnable generation of symbols",
116	"\n\tEnable auto-alias of labels",
117	"\n\tPrint this help and exit",
118	"\n\tPrint version and exit",
119	NULL,
120};
121
122static const char *guess_type_by_name(const char *fname, const char *fallback)
123{
124	const char *s;
125
126	s = strrchr(fname, '.');
127	if (s == NULL)
128		return fallback;
129	if (!strcasecmp(s, ".dts"))
130		return "dts";
131	if (!strcasecmp(s, ".dtb"))
132		return "dtb";
133	return fallback;
134}
135
136static const char *guess_input_format(const char *fname, const char *fallback)
137{
138	struct stat statbuf;
139	fdt32_t magic;
140	FILE *f;
141
142	if (stat(fname, &statbuf) != 0)
143		return fallback;
144
145	if (S_ISDIR(statbuf.st_mode))
146		return "fs";
147
148	if (!S_ISREG(statbuf.st_mode))
149		return fallback;
150
151	f = fopen(fname, "r");
152	if (f == NULL)
153		return fallback;
154	if (fread(&magic, 4, 1, f) != 1) {
155		fclose(f);
156		return fallback;
157	}
158	fclose(f);
159
160	if (fdt32_to_cpu(magic) == FDT_MAGIC)
161		return "dtb";
162
163	return guess_type_by_name(fname, fallback);
164}
165
166int main(int argc, char *argv[])
167{
168	struct dt_info *dti;
169	const char *inform = NULL;
170	const char *outform = NULL;
171	const char *outname = "-";
172	const char *depname = NULL;
173	bool force = false, sort = false;
174	const char *arg;
175	int opt;
176	FILE *outf = NULL;
177	int outversion = DEFAULT_FDT_VERSION;
178	long long cmdline_boot_cpuid = -1;
179
180	quiet      = 0;
181	reservenum = 0;
182	minsize    = 0;
183	padsize    = 0;
184	alignsize  = 0;
185
186	while ((opt = util_getopt_long()) != EOF) {
187		switch (opt) {
188		case 'I':
189			inform = optarg;
190			break;
191		case 'O':
192			outform = optarg;
193			break;
194		case 'o':
195			outname = optarg;
196			break;
197		case 'V':
198			outversion = strtol(optarg, NULL, 0);
199			break;
200		case 'd':
201			depname = optarg;
202			break;
203		case 'R':
204			reservenum = strtol(optarg, NULL, 0);
205			break;
206		case 'S':
207			minsize = strtol(optarg, NULL, 0);
208			break;
209		case 'p':
210			padsize = strtol(optarg, NULL, 0);
211			break;
212		case 'a':
213			alignsize = strtol(optarg, NULL, 0);
214			if (!is_power_of_2(alignsize))
215				die("Invalid argument \"%d\" to -a option\n",
216				    alignsize);
217			break;
218		case 'f':
219			force = true;
220			break;
221		case 'q':
222			quiet++;
223			break;
224		case 'b':
225			cmdline_boot_cpuid = strtoll(optarg, NULL, 0);
226			break;
227		case 'i':
228			srcfile_add_search_path(optarg);
229			break;
230		case 'v':
231			util_version();
232		case 'H':
233			if (streq(optarg, "legacy"))
234				phandle_format = PHANDLE_LEGACY;
235			else if (streq(optarg, "epapr"))
236				phandle_format = PHANDLE_EPAPR;
237			else if (streq(optarg, "both"))
238				phandle_format = PHANDLE_BOTH;
239			else
240				die("Invalid argument \"%s\" to -H option\n",
241				    optarg);
242			break;
243
244		case 's':
245			sort = true;
246			break;
247
248		case 'W':
249			parse_checks_option(true, false, optarg);
250			break;
251
252		case 'E':
253			parse_checks_option(false, true, optarg);
254			break;
255
256		case '@':
257			generate_symbols = 1;
258			break;
259		case 'A':
260			auto_label_aliases = 1;
261			break;
262
263		case 'h':
264			usage(NULL);
265		default:
266			usage("unknown option");
267		}
268	}
269
270	if (argc > (optind+1))
271		usage("missing files");
272	else if (argc < (optind+1))
273		arg = "-";
274	else
275		arg = argv[optind];
276
277	/* minsize and padsize are mutually exclusive */
278	if (minsize && padsize)
279		die("Can't set both -p and -S\n");
280
281	if (depname) {
282		depfile = fopen(depname, "w");
283		if (!depfile)
284			die("Couldn't open dependency file %s: %s\n", depname,
285			    strerror(errno));
286		fprintf(depfile, "%s:", outname);
287	}
288
289	if (inform == NULL)
290		inform = guess_input_format(arg, "dts");
291	if (outform == NULL) {
292		outform = guess_type_by_name(outname, NULL);
293		if (outform == NULL) {
294			if (streq(inform, "dts"))
295				outform = "dtb";
296			else
297				outform = "dts";
298		}
299	}
300	if (streq(inform, "dts"))
301		dti = dt_from_source(arg);
302	else if (streq(inform, "fs"))
303		dti = dt_from_fs(arg);
304	else if(streq(inform, "dtb"))
305		dti = dt_from_blob(arg);
306	else
307		die("Unknown input format \"%s\"\n", inform);
308
309	dti->outname = outname;
310
311	if (depfile) {
312		fputc('\n', depfile);
313		fclose(depfile);
314	}
315
316	if (cmdline_boot_cpuid != -1)
317		dti->boot_cpuid_phys = cmdline_boot_cpuid;
318
319	fill_fullpaths(dti->dt, "");
320
321	/* on a plugin, generate by default */
322	if (dti->dtsflags & DTSF_PLUGIN) {
323		generate_fixups = 1;
324	}
325
326	process_checks(force, dti);
327
328	if (auto_label_aliases)
329		generate_label_tree(dti, "aliases", false);
330
331	if (generate_symbols)
332		generate_label_tree(dti, "__symbols__", true);
333
334	if (generate_fixups) {
335		generate_fixups_tree(dti, "__fixups__");
336		generate_local_fixups_tree(dti, "__local_fixups__");
337	}
338
339	if (sort)
340		sort_tree(dti);
341
342	if (streq(outname, "-")) {
343		outf = stdout;
344	} else {
345		outf = fopen(outname, "wb");
346		if (! outf)
347			die("Couldn't open output file %s: %s\n",
348			    outname, strerror(errno));
349	}
350
351	if (streq(outform, "dts")) {
352		dt_to_source(outf, dti);
353	} else if (streq(outform, "dtb")) {
354		dt_to_blob(outf, dti, outversion);
355	} else if (streq(outform, "asm")) {
356		dt_to_asm(outf, dti, outversion);
357	} else if (streq(outform, "null")) {
358		/* do nothing */
359	} else {
360		die("Unknown output format \"%s\"\n", outform);
361	}
362
363	exit(0);
364}
365