xref: /freebsd/stand/fdt/fdt_loader_cmd.c (revision 0eb610e8e38bb3246cf902100aeae9d800bae53d)
1ca987d46SWarner Losh /*-
2ca987d46SWarner Losh  * Copyright (c) 2009-2010 The FreeBSD Foundation
3ca987d46SWarner Losh  * All rights reserved.
4ca987d46SWarner Losh  *
5ca987d46SWarner Losh  * This software was developed by Semihalf under sponsorship from
6ca987d46SWarner Losh  * the FreeBSD Foundation.
7ca987d46SWarner Losh  *
8ca987d46SWarner Losh  * Redistribution and use in source and binary forms, with or without
9ca987d46SWarner Losh  * modification, are permitted provided that the following conditions
10ca987d46SWarner Losh  * are met:
11ca987d46SWarner Losh  * 1. Redistributions of source code must retain the above copyright
12ca987d46SWarner Losh  *    notice, this list of conditions and the following disclaimer.
13ca987d46SWarner Losh  * 2. Redistributions in binary form must reproduce the above copyright
14ca987d46SWarner Losh  *    notice, this list of conditions and the following disclaimer in the
15ca987d46SWarner Losh  *    documentation and/or other materials provided with the distribution.
16ca987d46SWarner Losh  *
17ca987d46SWarner Losh  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18ca987d46SWarner Losh  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19ca987d46SWarner Losh  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20ca987d46SWarner Losh  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21ca987d46SWarner Losh  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22ca987d46SWarner Losh  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23ca987d46SWarner Losh  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24ca987d46SWarner Losh  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25ca987d46SWarner Losh  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26ca987d46SWarner Losh  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27ca987d46SWarner Losh  * SUCH DAMAGE.
28ca987d46SWarner Losh  */
29ca987d46SWarner Losh 
30ca987d46SWarner Losh #include <sys/cdefs.h>
31ca987d46SWarner Losh __FBSDID("$FreeBSD$");
32ca987d46SWarner Losh 
33ca987d46SWarner Losh #include <stand.h>
34ca987d46SWarner Losh #include <libfdt.h>
35d2084bf6SKyle Evans #include <fdt.h>
36ca987d46SWarner Losh #include <sys/param.h>
37ca987d46SWarner Losh #include <sys/linker.h>
38ca987d46SWarner Losh #include <machine/elf.h>
39ca987d46SWarner Losh 
40ca987d46SWarner Losh #include "bootstrap.h"
41ca987d46SWarner Losh #include "fdt_platform.h"
42ca987d46SWarner Losh 
43ca987d46SWarner Losh #ifdef DEBUG
44ca987d46SWarner Losh #define debugf(fmt, args...) do { printf("%s(): ", __func__);	\
45ca987d46SWarner Losh     printf(fmt,##args); } while (0)
46ca987d46SWarner Losh #else
47ca987d46SWarner Losh #define debugf(fmt, args...)
48ca987d46SWarner Losh #endif
49ca987d46SWarner Losh 
50ca987d46SWarner Losh #define FDT_CWD_LEN	256
51ca987d46SWarner Losh #define FDT_MAX_DEPTH	12
52ca987d46SWarner Losh 
53ca987d46SWarner Losh #define FDT_PROP_SEP	" = "
54ca987d46SWarner Losh 
55ca987d46SWarner Losh #define COPYOUT(s,d,l)	archsw.arch_copyout(s, d, l)
56ca987d46SWarner Losh #define COPYIN(s,d,l)	archsw.arch_copyin(s, d, l)
57ca987d46SWarner Losh 
58ca987d46SWarner Losh #define FDT_STATIC_DTB_SYMBOL	"fdt_static_dtb"
59ca987d46SWarner Losh 
60ca987d46SWarner Losh #define	CMD_REQUIRES_BLOB	0x01
61ca987d46SWarner Losh 
62ca987d46SWarner Losh /* Location of FDT yet to be loaded. */
63ca987d46SWarner Losh /* This may be in read-only memory, so can't be manipulated directly. */
64ca987d46SWarner Losh static struct fdt_header *fdt_to_load = NULL;
65ca987d46SWarner Losh /* Location of FDT on heap. */
66ca987d46SWarner Losh /* This is the copy we actually manipulate. */
67ca987d46SWarner Losh static struct fdt_header *fdtp = NULL;
68ca987d46SWarner Losh /* Size of FDT blob */
69ca987d46SWarner Losh static size_t fdtp_size = 0;
70ca987d46SWarner Losh 
71ca987d46SWarner Losh static int fdt_load_dtb(vm_offset_t va);
7213811ec8SKyle Evans static void fdt_print_overlay_load_error(int err, const char *filename);
731dff72ebSKyle Evans static int fdt_check_overlay_compatible(void *base_fdt, void *overlay_fdt);
74ca987d46SWarner Losh 
75ca987d46SWarner Losh static int fdt_cmd_nyi(int argc, char *argv[]);
768a613444SKyle Evans static int fdt_load_dtb_overlays_string(const char * filenames);
77ca987d46SWarner Losh 
78ca987d46SWarner Losh static int fdt_cmd_addr(int argc, char *argv[]);
79ca987d46SWarner Losh static int fdt_cmd_mkprop(int argc, char *argv[]);
80ca987d46SWarner Losh static int fdt_cmd_cd(int argc, char *argv[]);
81ca987d46SWarner Losh static int fdt_cmd_hdr(int argc, char *argv[]);
82ca987d46SWarner Losh static int fdt_cmd_ls(int argc, char *argv[]);
83ca987d46SWarner Losh static int fdt_cmd_prop(int argc, char *argv[]);
84ca987d46SWarner Losh static int fdt_cmd_pwd(int argc, char *argv[]);
85ca987d46SWarner Losh static int fdt_cmd_rm(int argc, char *argv[]);
86ca987d46SWarner Losh static int fdt_cmd_mknode(int argc, char *argv[]);
87ca987d46SWarner Losh static int fdt_cmd_mres(int argc, char *argv[]);
88ca987d46SWarner Losh 
89ca987d46SWarner Losh typedef int cmdf_t(int, char *[]);
90ca987d46SWarner Losh 
91ca987d46SWarner Losh struct cmdtab {
92ca987d46SWarner Losh 	const char	*name;
93ca987d46SWarner Losh 	cmdf_t		*handler;
94ca987d46SWarner Losh 	int		flags;
95ca987d46SWarner Losh };
96ca987d46SWarner Losh 
97ca987d46SWarner Losh static const struct cmdtab commands[] = {
98ca987d46SWarner Losh 	{ "addr", &fdt_cmd_addr,	0 },
99ca987d46SWarner Losh 	{ "alias", &fdt_cmd_nyi,	0 },
100ca987d46SWarner Losh 	{ "cd", &fdt_cmd_cd,		CMD_REQUIRES_BLOB },
101ca987d46SWarner Losh 	{ "header", &fdt_cmd_hdr,	CMD_REQUIRES_BLOB },
102ca987d46SWarner Losh 	{ "ls", &fdt_cmd_ls,		CMD_REQUIRES_BLOB },
103ca987d46SWarner Losh 	{ "mknode", &fdt_cmd_mknode,	CMD_REQUIRES_BLOB },
104ca987d46SWarner Losh 	{ "mkprop", &fdt_cmd_mkprop,	CMD_REQUIRES_BLOB },
105ca987d46SWarner Losh 	{ "mres", &fdt_cmd_mres,	CMD_REQUIRES_BLOB },
106ca987d46SWarner Losh 	{ "prop", &fdt_cmd_prop,	CMD_REQUIRES_BLOB },
107ca987d46SWarner Losh 	{ "pwd", &fdt_cmd_pwd,		CMD_REQUIRES_BLOB },
108ca987d46SWarner Losh 	{ "rm", &fdt_cmd_rm,		CMD_REQUIRES_BLOB },
109ca987d46SWarner Losh 	{ NULL, NULL }
110ca987d46SWarner Losh };
111ca987d46SWarner Losh 
112ca987d46SWarner Losh static char cwd[FDT_CWD_LEN] = "/";
113ca987d46SWarner Losh 
114ca987d46SWarner Losh static vm_offset_t
115ca987d46SWarner Losh fdt_find_static_dtb()
116ca987d46SWarner Losh {
117ca987d46SWarner Losh 	Elf_Ehdr *ehdr;
118ca987d46SWarner Losh 	Elf_Shdr *shdr;
119ca987d46SWarner Losh 	Elf_Sym sym;
120ca987d46SWarner Losh 	vm_offset_t strtab, symtab, fdt_start;
121ca987d46SWarner Losh 	uint64_t offs;
122ca987d46SWarner Losh 	struct preloaded_file *kfp;
123ca987d46SWarner Losh 	struct file_metadata *md;
124ca987d46SWarner Losh 	char *strp;
125ca987d46SWarner Losh 	int i, sym_count;
126ca987d46SWarner Losh 
127ca987d46SWarner Losh 	debugf("fdt_find_static_dtb()\n");
128ca987d46SWarner Losh 
129ca987d46SWarner Losh 	sym_count = symtab = strtab = 0;
130ca987d46SWarner Losh 	strp = NULL;
131ca987d46SWarner Losh 
132ca987d46SWarner Losh 	offs = __elfN(relocation_offset);
133ca987d46SWarner Losh 
134ca987d46SWarner Losh 	kfp = file_findfile(NULL, NULL);
135ca987d46SWarner Losh 	if (kfp == NULL)
136ca987d46SWarner Losh 		return (0);
137ca987d46SWarner Losh 
138ca987d46SWarner Losh 	/* Locate the dynamic symbols and strtab. */
139ca987d46SWarner Losh 	md = file_findmetadata(kfp, MODINFOMD_ELFHDR);
140ca987d46SWarner Losh 	if (md == NULL)
141ca987d46SWarner Losh 		return (0);
142ca987d46SWarner Losh 	ehdr = (Elf_Ehdr *)md->md_data;
143ca987d46SWarner Losh 
144ca987d46SWarner Losh 	md = file_findmetadata(kfp, MODINFOMD_SHDR);
145ca987d46SWarner Losh 	if (md == NULL)
146ca987d46SWarner Losh 		return (0);
147ca987d46SWarner Losh 	shdr = (Elf_Shdr *)md->md_data;
148ca987d46SWarner Losh 
149ca987d46SWarner Losh 	for (i = 0; i < ehdr->e_shnum; ++i) {
150ca987d46SWarner Losh 		if (shdr[i].sh_type == SHT_DYNSYM && symtab == 0) {
151ca987d46SWarner Losh 			symtab = shdr[i].sh_addr + offs;
152ca987d46SWarner Losh 			sym_count = shdr[i].sh_size / sizeof(Elf_Sym);
153ca987d46SWarner Losh 		} else if (shdr[i].sh_type == SHT_STRTAB && strtab == 0) {
154ca987d46SWarner Losh 			strtab = shdr[i].sh_addr + offs;
155ca987d46SWarner Losh 		}
156ca987d46SWarner Losh 	}
157ca987d46SWarner Losh 
158ca987d46SWarner Losh 	/*
159ca987d46SWarner Losh 	 * The most efficient way to find a symbol would be to calculate a
160ca987d46SWarner Losh 	 * hash, find proper bucket and chain, and thus find a symbol.
161ca987d46SWarner Losh 	 * However, that would involve code duplication (e.g. for hash
162ca987d46SWarner Losh 	 * function). So we're using simpler and a bit slower way: we're
163ca987d46SWarner Losh 	 * iterating through symbols, searching for the one which name is
164ca987d46SWarner Losh 	 * 'equal' to 'fdt_static_dtb'. To speed up the process a little bit,
165ca987d46SWarner Losh 	 * we are eliminating symbols type of which is not STT_NOTYPE, or(and)
166ca987d46SWarner Losh 	 * those which binding attribute is not STB_GLOBAL.
167ca987d46SWarner Losh 	 */
168ca987d46SWarner Losh 	fdt_start = 0;
169ca987d46SWarner Losh 	while (sym_count > 0 && fdt_start == 0) {
170ca987d46SWarner Losh 		COPYOUT(symtab, &sym, sizeof(sym));
171ca987d46SWarner Losh 		symtab += sizeof(sym);
172ca987d46SWarner Losh 		--sym_count;
173ca987d46SWarner Losh 		if (ELF_ST_BIND(sym.st_info) != STB_GLOBAL ||
174ca987d46SWarner Losh 		    ELF_ST_TYPE(sym.st_info) != STT_NOTYPE)
175ca987d46SWarner Losh 			continue;
176ca987d46SWarner Losh 		strp = strdupout(strtab + sym.st_name);
177ca987d46SWarner Losh 		if (strcmp(strp, FDT_STATIC_DTB_SYMBOL) == 0)
178ca987d46SWarner Losh 			fdt_start = (vm_offset_t)sym.st_value + offs;
179ca987d46SWarner Losh 		free(strp);
180ca987d46SWarner Losh 	}
181ca987d46SWarner Losh 	return (fdt_start);
182ca987d46SWarner Losh }
183ca987d46SWarner Losh 
184ca987d46SWarner Losh static int
185ca987d46SWarner Losh fdt_load_dtb(vm_offset_t va)
186ca987d46SWarner Losh {
187ca987d46SWarner Losh 	struct fdt_header header;
188ca987d46SWarner Losh 	int err;
189ca987d46SWarner Losh 
190ca987d46SWarner Losh 	debugf("fdt_load_dtb(0x%08jx)\n", (uintmax_t)va);
191ca987d46SWarner Losh 
192ca987d46SWarner Losh 	COPYOUT(va, &header, sizeof(header));
193ca987d46SWarner Losh 	err = fdt_check_header(&header);
194ca987d46SWarner Losh 	if (err < 0) {
195ca987d46SWarner Losh 		if (err == -FDT_ERR_BADVERSION) {
196ca987d46SWarner Losh 			snprintf(command_errbuf, sizeof(command_errbuf),
197ca987d46SWarner Losh 			    "incompatible blob version: %d, should be: %d",
198ca987d46SWarner Losh 			    fdt_version(fdtp), FDT_LAST_SUPPORTED_VERSION);
199ca987d46SWarner Losh 		} else {
200ca987d46SWarner Losh 			snprintf(command_errbuf, sizeof(command_errbuf),
201ca987d46SWarner Losh 			    "error validating blob: %s", fdt_strerror(err));
202ca987d46SWarner Losh 		}
203ca987d46SWarner Losh 		return (1);
204ca987d46SWarner Losh 	}
205ca987d46SWarner Losh 
206ca987d46SWarner Losh 	/*
207ca987d46SWarner Losh 	 * Release previous blob
208ca987d46SWarner Losh 	 */
209ca987d46SWarner Losh 	if (fdtp)
210ca987d46SWarner Losh 		free(fdtp);
211ca987d46SWarner Losh 
212ca987d46SWarner Losh 	fdtp_size = fdt_totalsize(&header);
213ca987d46SWarner Losh 	fdtp = malloc(fdtp_size);
214ca987d46SWarner Losh 
215ca987d46SWarner Losh 	if (fdtp == NULL) {
216ca987d46SWarner Losh 		command_errmsg = "can't allocate memory for device tree copy";
217ca987d46SWarner Losh 		return (1);
218ca987d46SWarner Losh 	}
219ca987d46SWarner Losh 
220ca987d46SWarner Losh 	COPYOUT(va, fdtp, fdtp_size);
221ca987d46SWarner Losh 	debugf("DTB blob found at 0x%jx, size: 0x%jx\n", (uintmax_t)va, (uintmax_t)fdtp_size);
222ca987d46SWarner Losh 
223ca987d46SWarner Losh 	return (0);
224ca987d46SWarner Losh }
225ca987d46SWarner Losh 
226ca987d46SWarner Losh int
227ca987d46SWarner Losh fdt_load_dtb_addr(struct fdt_header *header)
228ca987d46SWarner Losh {
229ca987d46SWarner Losh 	int err;
230ca987d46SWarner Losh 
231ca987d46SWarner Losh 	debugf("fdt_load_dtb_addr(%p)\n", header);
232ca987d46SWarner Losh 
233ca987d46SWarner Losh 	fdtp_size = fdt_totalsize(header);
234ca987d46SWarner Losh 	err = fdt_check_header(header);
235ca987d46SWarner Losh 	if (err < 0) {
236ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
237ca987d46SWarner Losh 		    "error validating blob: %s", fdt_strerror(err));
238ca987d46SWarner Losh 		return (err);
239ca987d46SWarner Losh 	}
240ca987d46SWarner Losh 	free(fdtp);
241ca987d46SWarner Losh 	if ((fdtp = malloc(fdtp_size)) == NULL) {
242ca987d46SWarner Losh 		command_errmsg = "can't allocate memory for device tree copy";
243ca987d46SWarner Losh 		return (1);
244ca987d46SWarner Losh 	}
245ca987d46SWarner Losh 
246ca987d46SWarner Losh 	bcopy(header, fdtp, fdtp_size);
247ca987d46SWarner Losh 	return (0);
248ca987d46SWarner Losh }
249ca987d46SWarner Losh 
250ca987d46SWarner Losh int
251ca987d46SWarner Losh fdt_load_dtb_file(const char * filename)
252ca987d46SWarner Losh {
253ca987d46SWarner Losh 	struct preloaded_file *bfp, *oldbfp;
254ca987d46SWarner Losh 	int err;
255ca987d46SWarner Losh 
256ca987d46SWarner Losh 	debugf("fdt_load_dtb_file(%s)\n", filename);
257ca987d46SWarner Losh 
258ca987d46SWarner Losh 	oldbfp = file_findfile(NULL, "dtb");
259ca987d46SWarner Losh 
260ca987d46SWarner Losh 	/* Attempt to load and validate a new dtb from a file. */
261ca987d46SWarner Losh 	if ((bfp = file_loadraw(filename, "dtb", 1)) == NULL) {
262ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
263ca987d46SWarner Losh 		    "failed to load file '%s'", filename);
264ca987d46SWarner Losh 		return (1);
265ca987d46SWarner Losh 	}
266ca987d46SWarner Losh 	if ((err = fdt_load_dtb(bfp->f_addr)) != 0) {
267ca987d46SWarner Losh 		file_discard(bfp);
268ca987d46SWarner Losh 		return (err);
269ca987d46SWarner Losh 	}
270ca987d46SWarner Losh 
271ca987d46SWarner Losh 	/* A new dtb was validated, discard any previous file. */
272ca987d46SWarner Losh 	if (oldbfp)
273ca987d46SWarner Losh 		file_discard(oldbfp);
274ca987d46SWarner Losh 	return (0);
275ca987d46SWarner Losh }
276ca987d46SWarner Losh 
277ca987d46SWarner Losh static int
278ca987d46SWarner Losh fdt_load_dtb_overlay(const char * filename)
279ca987d46SWarner Losh {
280a609d03bSKyle Evans 	struct preloaded_file *bfp;
281ca987d46SWarner Losh 	struct fdt_header header;
282ca987d46SWarner Losh 	int err;
283ca987d46SWarner Losh 
284ca987d46SWarner Losh 	debugf("fdt_load_dtb_overlay(%s)\n", filename);
285ca987d46SWarner Losh 
28613811ec8SKyle Evans 	/* Attempt to load and validate a new dtb from a file. FDT_ERR_NOTFOUND
28713811ec8SKyle Evans 	 * is normally a libfdt error code, but libfdt would actually return
28813811ec8SKyle Evans 	 * -FDT_ERR_NOTFOUND. We re-purpose the error code here to convey a
28913811ec8SKyle Evans 	 * similar meaning: the file itself was not found, which can still be
29013811ec8SKyle Evans 	 * considered an error dealing with FDT pieces.
29113811ec8SKyle Evans 	 */
29213811ec8SKyle Evans 	if ((bfp = file_loadraw(filename, "dtbo", 1)) == NULL)
29313811ec8SKyle Evans 		return (FDT_ERR_NOTFOUND);
294ca987d46SWarner Losh 
295ca987d46SWarner Losh 	COPYOUT(bfp->f_addr, &header, sizeof(header));
296ca987d46SWarner Losh 	err = fdt_check_header(&header);
297ca987d46SWarner Losh 
298ca987d46SWarner Losh 	if (err < 0) {
299ca987d46SWarner Losh 		file_discard(bfp);
30013811ec8SKyle Evans 		return (err);
301ca987d46SWarner Losh 	}
302ca987d46SWarner Losh 
303ca987d46SWarner Losh 	return (0);
304ca987d46SWarner Losh }
305ca987d46SWarner Losh 
30613811ec8SKyle Evans static void
30713811ec8SKyle Evans fdt_print_overlay_load_error(int err, const char *filename)
30813811ec8SKyle Evans {
30913811ec8SKyle Evans 
31013811ec8SKyle Evans 	switch (err) {
31113811ec8SKyle Evans 		case FDT_ERR_NOTFOUND:
31213811ec8SKyle Evans 			printf("%s: failed to load file\n", filename);
31313811ec8SKyle Evans 			break;
31413811ec8SKyle Evans 		case -FDT_ERR_BADVERSION:
31513811ec8SKyle Evans 			printf("%s: incompatible blob version: %d, should be: %d\n",
31613811ec8SKyle Evans 			    filename, fdt_version(fdtp),
31713811ec8SKyle Evans 			    FDT_LAST_SUPPORTED_VERSION);
31813811ec8SKyle Evans 			break;
31913811ec8SKyle Evans 		default:
32013811ec8SKyle Evans 			/* libfdt errs are negative */
32113811ec8SKyle Evans 			if (err < 0)
32213811ec8SKyle Evans 				printf("%s: error validating blob: %s\n",
32313811ec8SKyle Evans 				    filename, fdt_strerror(err));
32413811ec8SKyle Evans 			else
32513811ec8SKyle Evans 				printf("%s: unknown load error\n", filename);
32613811ec8SKyle Evans 			break;
32713811ec8SKyle Evans 	}
32813811ec8SKyle Evans }
32913811ec8SKyle Evans 
3308a613444SKyle Evans static int
3318a613444SKyle Evans fdt_load_dtb_overlays_string(const char * filenames)
332ca987d46SWarner Losh {
333ca987d46SWarner Losh 	char *names;
33413811ec8SKyle Evans 	char *name, *name_ext;
335ca987d46SWarner Losh 	char *comaptr;
33613811ec8SKyle Evans 	int err, namesz;
337ca987d46SWarner Losh 
3388a613444SKyle Evans 	debugf("fdt_load_dtb_overlays_string(%s)\n", filenames);
339ca987d46SWarner Losh 
340ca987d46SWarner Losh 	names = strdup(filenames);
341ca987d46SWarner Losh 	if (names == NULL)
342ca987d46SWarner Losh 		return (1);
343ca987d46SWarner Losh 	name = names;
344ca987d46SWarner Losh 	do {
345ca987d46SWarner Losh 		comaptr = strchr(name, ',');
346ca987d46SWarner Losh 		if (comaptr)
347ca987d46SWarner Losh 			*comaptr = '\0';
34813811ec8SKyle Evans 		err = fdt_load_dtb_overlay(name);
34913811ec8SKyle Evans 		if (err == FDT_ERR_NOTFOUND) {
35013811ec8SKyle Evans 			/* Allocate enough to append ".dtbo" */
35113811ec8SKyle Evans 			namesz = strlen(name) + 6;
35213811ec8SKyle Evans 			name_ext = malloc(namesz);
35313811ec8SKyle Evans 			if (name_ext == NULL) {
35413811ec8SKyle Evans 				fdt_print_overlay_load_error(err, name);
35513811ec8SKyle Evans 				name = comaptr + 1;
35613811ec8SKyle Evans 				continue;
35713811ec8SKyle Evans 			}
35813811ec8SKyle Evans 			snprintf(name_ext, namesz, "%s.dtbo", name);
35913811ec8SKyle Evans 			err = fdt_load_dtb_overlay(name_ext);
36013811ec8SKyle Evans 			free(name_ext);
36113811ec8SKyle Evans 		}
36213811ec8SKyle Evans 		/* Catch error with either initial load or fallback load */
36313811ec8SKyle Evans 		if (err != 0)
36413811ec8SKyle Evans 			fdt_print_overlay_load_error(err, name);
365ca987d46SWarner Losh 		name = comaptr + 1;
366ca987d46SWarner Losh 	} while(comaptr);
367ca987d46SWarner Losh 
368ca987d46SWarner Losh 	free(names);
369ca987d46SWarner Losh 	return (0);
370ca987d46SWarner Losh }
371ca987d46SWarner Losh 
3721dff72ebSKyle Evans /*
3731dff72ebSKyle Evans  * fdt_check_overlay_compatible - check that the overlay_fdt is compatible with
3741dff72ebSKyle Evans  * base_fdt before we attempt to apply it. It will need to re-calculate offsets
3751dff72ebSKyle Evans  * in the base every time, rather than trying to cache them earlier in the
3761dff72ebSKyle Evans  * process, because the overlay application process can/will invalidate a lot of
3771dff72ebSKyle Evans  * offsets.
3781dff72ebSKyle Evans  */
3791dff72ebSKyle Evans static int
3801dff72ebSKyle Evans fdt_check_overlay_compatible(void *base_fdt, void *overlay_fdt)
3811dff72ebSKyle Evans {
3821dff72ebSKyle Evans 	const char *compat;
3831dff72ebSKyle Evans 	int compat_len, ocompat_len;
3841dff72ebSKyle Evans 	int oroot_offset, root_offset;
3851dff72ebSKyle Evans 	int slidx, sllen;
3861dff72ebSKyle Evans 
3871dff72ebSKyle Evans 	oroot_offset = fdt_path_offset(overlay_fdt, "/");
3881dff72ebSKyle Evans 	if (oroot_offset < 0)
3891dff72ebSKyle Evans 		return (oroot_offset);
3901dff72ebSKyle Evans 	/*
3911dff72ebSKyle Evans 	 * If /compatible in the overlay does not exist or if it is empty, then
3921dff72ebSKyle Evans 	 * we're automatically compatible. We do this for the sake of rapid
3931dff72ebSKyle Evans 	 * overlay development for overlays that aren't intended to be deployed.
3941dff72ebSKyle Evans 	 * The user assumes the risk of using an overlay without /compatible.
3951dff72ebSKyle Evans 	 */
3961dff72ebSKyle Evans 	if (fdt_get_property(overlay_fdt, oroot_offset, "compatible",
3971dff72ebSKyle Evans 	    &ocompat_len) == NULL || ocompat_len == 0)
3981dff72ebSKyle Evans 		return (0);
3991dff72ebSKyle Evans 	root_offset = fdt_path_offset(base_fdt, "/");
4001dff72ebSKyle Evans 	if (root_offset < 0)
4011dff72ebSKyle Evans 		return (root_offset);
4021dff72ebSKyle Evans 	/*
4031dff72ebSKyle Evans 	 * However, an empty or missing /compatible on the base is an error,
4041dff72ebSKyle Evans 	 * because allowing this offers no advantages.
4051dff72ebSKyle Evans 	 */
4061dff72ebSKyle Evans 	if (fdt_get_property(base_fdt, root_offset, "compatible",
4071dff72ebSKyle Evans 	    &compat_len) == NULL)
4081dff72ebSKyle Evans 		return (compat_len);
4091dff72ebSKyle Evans 	else if(compat_len == 0)
4101dff72ebSKyle Evans 		return (1);
4111dff72ebSKyle Evans 
4121dff72ebSKyle Evans 	slidx = 0;
4131dff72ebSKyle Evans 	compat = fdt_stringlist_get(overlay_fdt, oroot_offset, "compatible",
4141dff72ebSKyle Evans 	    slidx, &sllen);
4151dff72ebSKyle Evans 	while (compat != NULL) {
4161dff72ebSKyle Evans 		if (fdt_stringlist_search(base_fdt, root_offset, "compatible",
4171dff72ebSKyle Evans 		    compat) >= 0)
4181dff72ebSKyle Evans 			return (0);
4191dff72ebSKyle Evans 		++slidx;
4201dff72ebSKyle Evans 		compat = fdt_stringlist_get(overlay_fdt, oroot_offset,
4211dff72ebSKyle Evans 		    "compatible", slidx, &sllen);
4221dff72ebSKyle Evans 	};
4231dff72ebSKyle Evans 
4241dff72ebSKyle Evans 	/* We've exhausted the overlay's /compatible property... no match */
4251dff72ebSKyle Evans 	return (1);
4261dff72ebSKyle Evans }
4271dff72ebSKyle Evans 
428ca987d46SWarner Losh void
429ca987d46SWarner Losh fdt_apply_overlays()
430ca987d46SWarner Losh {
431ca987d46SWarner Losh 	struct preloaded_file *fp;
432e9ca5fa5SKyle Evans 	size_t max_overlay_size, next_fdtp_size;
433e9ca5fa5SKyle Evans 	size_t current_fdtp_size;
434e9ca5fa5SKyle Evans 	void *current_fdtp;
435e9ca5fa5SKyle Evans 	void *next_fdtp;
436ca987d46SWarner Losh 	void *overlay;
437ca987d46SWarner Losh 	int rv;
438ca987d46SWarner Losh 
439ca987d46SWarner Losh 	if ((fdtp == NULL) || (fdtp_size == 0))
440ca987d46SWarner Losh 		return;
441ca987d46SWarner Losh 
442ca987d46SWarner Losh 	max_overlay_size = 0;
443ca987d46SWarner Losh 	for (fp = file_findfile(NULL, "dtbo"); fp != NULL; fp = fp->f_next) {
444ca987d46SWarner Losh 		if (max_overlay_size < fp->f_size)
445ca987d46SWarner Losh 			max_overlay_size = fp->f_size;
446ca987d46SWarner Losh 	}
447ca987d46SWarner Losh 
448ca987d46SWarner Losh 	/* Nothing to apply */
449e9ca5fa5SKyle Evans 	if (max_overlay_size == 0)
450ca987d46SWarner Losh 		return;
451ca987d46SWarner Losh 
452ca987d46SWarner Losh 	overlay = malloc(max_overlay_size);
453ca987d46SWarner Losh 	if (overlay == NULL) {
454ca987d46SWarner Losh 		printf("failed to allocate memory for DTB blob with overlays\n");
455ca987d46SWarner Losh 		return;
456ca987d46SWarner Losh 	}
457e9ca5fa5SKyle Evans 	current_fdtp = fdtp;
458e9ca5fa5SKyle Evans 	current_fdtp_size = fdtp_size;
459ca987d46SWarner Losh 	for (fp = file_findfile(NULL, "dtbo"); fp != NULL; fp = fp->f_next) {
4601dff72ebSKyle Evans 		COPYOUT(fp->f_addr, overlay, fp->f_size);
4611dff72ebSKyle Evans 		/* Check compatible first to avoid unnecessary allocation */
4621dff72ebSKyle Evans 		rv = fdt_check_overlay_compatible(current_fdtp, overlay);
4631dff72ebSKyle Evans 		if (rv != 0) {
4641dff72ebSKyle Evans 			printf("DTB overlay '%s' not compatible\n", fp->f_name);
4651dff72ebSKyle Evans 			continue;
4661dff72ebSKyle Evans 		}
467ca987d46SWarner Losh 		printf("applying DTB overlay '%s'\n", fp->f_name);
468e9ca5fa5SKyle Evans 		next_fdtp_size = current_fdtp_size + fp->f_size;
469e9ca5fa5SKyle Evans 		next_fdtp = malloc(next_fdtp_size);
470e9ca5fa5SKyle Evans 		if (next_fdtp == NULL) {
471e9ca5fa5SKyle Evans 			/*
472e9ca5fa5SKyle Evans 			 * Output warning, then move on to applying other
473e9ca5fa5SKyle Evans 			 * overlays in case this one is simply too large.
474e9ca5fa5SKyle Evans 			 */
475e9ca5fa5SKyle Evans 			printf("failed to allocate memory for overlay base\n");
476e9ca5fa5SKyle Evans 			continue;
477e9ca5fa5SKyle Evans 		}
478e9ca5fa5SKyle Evans 		rv = fdt_open_into(current_fdtp, next_fdtp, next_fdtp_size);
479e9ca5fa5SKyle Evans 		if (rv != 0) {
48067a3bb4eSKyle Evans 			free(next_fdtp);
481e9ca5fa5SKyle Evans 			printf("failed to open base dtb into overlay base\n");
482e9ca5fa5SKyle Evans 			continue;
483e9ca5fa5SKyle Evans 		}
484e7ca72dbSKyle Evans 		/* Both overlay and next_fdtp may be modified in place */
485e9ca5fa5SKyle Evans 		rv = fdt_overlay_apply(next_fdtp, overlay);
486e9ca5fa5SKyle Evans 		if (rv == 0) {
487e9ca5fa5SKyle Evans 			/* Rotate next -> current */
488e9ca5fa5SKyle Evans 			if (current_fdtp != fdtp)
489e9ca5fa5SKyle Evans 				free(current_fdtp);
490e9ca5fa5SKyle Evans 			current_fdtp = next_fdtp;
491e9ca5fa5SKyle Evans 			current_fdtp_size = next_fdtp_size;
492e9ca5fa5SKyle Evans 		} else {
493e9ca5fa5SKyle Evans 			/*
494e9ca5fa5SKyle Evans 			 * Assume here that the base we tried to apply on is
495e9ca5fa5SKyle Evans 			 * either trashed or in an inconsistent state. Trying to
496e9ca5fa5SKyle Evans 			 * load it might work, but it's better to discard it and
497e9ca5fa5SKyle Evans 			 * play it safe. */
498e9ca5fa5SKyle Evans 			free(next_fdtp);
499e9ca5fa5SKyle Evans 			printf("failed to apply overlay: %s\n",
500e9ca5fa5SKyle Evans 			    fdt_strerror(rv));
501ca987d46SWarner Losh 		}
502e9ca5fa5SKyle Evans 	}
503e9ca5fa5SKyle Evans 	/* We could have failed to apply all overlays; then we do nothing */
504e9ca5fa5SKyle Evans 	if (current_fdtp != fdtp) {
505ca987d46SWarner Losh 		free(fdtp);
506e9ca5fa5SKyle Evans 		fdtp = current_fdtp;
507e9ca5fa5SKyle Evans 		fdtp_size = current_fdtp_size;
508e9ca5fa5SKyle Evans 	}
509ca987d46SWarner Losh 	free(overlay);
510ca987d46SWarner Losh }
511ca987d46SWarner Losh 
512ca987d46SWarner Losh int
513*0eb610e8SEmmanuel Vadot fdt_is_setup(void)
514*0eb610e8SEmmanuel Vadot {
515*0eb610e8SEmmanuel Vadot 
516*0eb610e8SEmmanuel Vadot 	if (fdtp != NULL)
517*0eb610e8SEmmanuel Vadot 		return (1);
518*0eb610e8SEmmanuel Vadot 
519*0eb610e8SEmmanuel Vadot 	return (0);
520*0eb610e8SEmmanuel Vadot }
521*0eb610e8SEmmanuel Vadot 
522*0eb610e8SEmmanuel Vadot int
523ca987d46SWarner Losh fdt_setup_fdtp()
524ca987d46SWarner Losh {
525ca987d46SWarner Losh 	struct preloaded_file *bfp;
526ca987d46SWarner Losh 	vm_offset_t va;
527ca987d46SWarner Losh 
528ca987d46SWarner Losh 	debugf("fdt_setup_fdtp()\n");
529ca987d46SWarner Losh 
530ca987d46SWarner Losh 	/* If we already loaded a file, use it. */
531ca987d46SWarner Losh 	if ((bfp = file_findfile(NULL, "dtb")) != NULL) {
532ca987d46SWarner Losh 		if (fdt_load_dtb(bfp->f_addr) == 0) {
533ca987d46SWarner Losh 			printf("Using DTB from loaded file '%s'.\n",
534ca987d46SWarner Losh 			    bfp->f_name);
5352a1e52f3SKyle Evans 			fdt_platform_load_overlays();
536ca987d46SWarner Losh 			return (0);
537ca987d46SWarner Losh 		}
538ca987d46SWarner Losh 	}
539ca987d46SWarner Losh 
540ca987d46SWarner Losh 	/* If we were given the address of a valid blob in memory, use it. */
541ca987d46SWarner Losh 	if (fdt_to_load != NULL) {
542ca987d46SWarner Losh 		if (fdt_load_dtb_addr(fdt_to_load) == 0) {
543ca987d46SWarner Losh 			printf("Using DTB from memory address %p.\n",
544ca987d46SWarner Losh 			    fdt_to_load);
5452a1e52f3SKyle Evans 			fdt_platform_load_overlays();
546ca987d46SWarner Losh 			return (0);
547ca987d46SWarner Losh 		}
548ca987d46SWarner Losh 	}
549ca987d46SWarner Losh 
5502a1e52f3SKyle Evans 	if (fdt_platform_load_dtb() == 0) {
5512a1e52f3SKyle Evans 		fdt_platform_load_overlays();
552ca987d46SWarner Losh 		return (0);
5532a1e52f3SKyle Evans 	}
554ca987d46SWarner Losh 
555ca987d46SWarner Losh 	/* If there is a dtb compiled into the kernel, use it. */
556ca987d46SWarner Losh 	if ((va = fdt_find_static_dtb()) != 0) {
557ca987d46SWarner Losh 		if (fdt_load_dtb(va) == 0) {
558ca987d46SWarner Losh 			printf("Using DTB compiled into kernel.\n");
559ca987d46SWarner Losh 			return (0);
560ca987d46SWarner Losh 		}
561ca987d46SWarner Losh 	}
562ca987d46SWarner Losh 
563ca987d46SWarner Losh 	command_errmsg = "No device tree blob found!\n";
564ca987d46SWarner Losh 	return (1);
565ca987d46SWarner Losh }
566ca987d46SWarner Losh 
567ca987d46SWarner Losh #define fdt_strtovect(str, cellbuf, lim, cellsize) _fdt_strtovect((str), \
568ca987d46SWarner Losh     (cellbuf), (lim), (cellsize), 0);
569ca987d46SWarner Losh 
570ca987d46SWarner Losh /* Force using base 16 */
571ca987d46SWarner Losh #define fdt_strtovectx(str, cellbuf, lim, cellsize) _fdt_strtovect((str), \
572ca987d46SWarner Losh     (cellbuf), (lim), (cellsize), 16);
573ca987d46SWarner Losh 
574ca987d46SWarner Losh static int
575ca987d46SWarner Losh _fdt_strtovect(const char *str, void *cellbuf, int lim, unsigned char cellsize,
576ca987d46SWarner Losh     uint8_t base)
577ca987d46SWarner Losh {
578ca987d46SWarner Losh 	const char *buf = str;
579ca987d46SWarner Losh 	const char *end = str + strlen(str) - 2;
580ca987d46SWarner Losh 	uint32_t *u32buf = NULL;
581ca987d46SWarner Losh 	uint8_t *u8buf = NULL;
582ca987d46SWarner Losh 	int cnt = 0;
583ca987d46SWarner Losh 
584ca987d46SWarner Losh 	if (cellsize == sizeof(uint32_t))
585ca987d46SWarner Losh 		u32buf = (uint32_t *)cellbuf;
586ca987d46SWarner Losh 	else
587ca987d46SWarner Losh 		u8buf = (uint8_t *)cellbuf;
588ca987d46SWarner Losh 
589ca987d46SWarner Losh 	if (lim == 0)
590ca987d46SWarner Losh 		return (0);
591ca987d46SWarner Losh 
592ca987d46SWarner Losh 	while (buf < end) {
593ca987d46SWarner Losh 
594ca987d46SWarner Losh 		/* Skip white whitespace(s)/separators */
595ca987d46SWarner Losh 		while (!isxdigit(*buf) && buf < end)
596ca987d46SWarner Losh 			buf++;
597ca987d46SWarner Losh 
598ca987d46SWarner Losh 		if (u32buf != NULL)
599ca987d46SWarner Losh 			u32buf[cnt] =
600ca987d46SWarner Losh 			    cpu_to_fdt32((uint32_t)strtol(buf, NULL, base));
601ca987d46SWarner Losh 
602ca987d46SWarner Losh 		else
603ca987d46SWarner Losh 			u8buf[cnt] = (uint8_t)strtol(buf, NULL, base);
604ca987d46SWarner Losh 
605ca987d46SWarner Losh 		if (cnt + 1 <= lim - 1)
606ca987d46SWarner Losh 			cnt++;
607ca987d46SWarner Losh 		else
608ca987d46SWarner Losh 			break;
609ca987d46SWarner Losh 		buf++;
610ca987d46SWarner Losh 		/* Find another number */
611ca987d46SWarner Losh 		while ((isxdigit(*buf) || *buf == 'x') && buf < end)
612ca987d46SWarner Losh 			buf++;
613ca987d46SWarner Losh 	}
614ca987d46SWarner Losh 	return (cnt);
615ca987d46SWarner Losh }
616ca987d46SWarner Losh 
617ca987d46SWarner Losh void
618ca987d46SWarner Losh fdt_fixup_ethernet(const char *str, char *ethstr, int len)
619ca987d46SWarner Losh {
620ca987d46SWarner Losh 	uint8_t tmp_addr[6];
621ca987d46SWarner Losh 
622ca987d46SWarner Losh 	/* Convert macaddr string into a vector of uints */
623ca987d46SWarner Losh 	fdt_strtovectx(str, &tmp_addr, 6, sizeof(uint8_t));
624ca987d46SWarner Losh 	/* Set actual property to a value from vect */
625ca987d46SWarner Losh 	fdt_setprop(fdtp, fdt_path_offset(fdtp, ethstr),
626ca987d46SWarner Losh 	    "local-mac-address", &tmp_addr, 6 * sizeof(uint8_t));
627ca987d46SWarner Losh }
628ca987d46SWarner Losh 
629ca987d46SWarner Losh void
630ca987d46SWarner Losh fdt_fixup_cpubusfreqs(unsigned long cpufreq, unsigned long busfreq)
631ca987d46SWarner Losh {
632ca987d46SWarner Losh 	int lo, o = 0, o2, maxo = 0, depth;
633ca987d46SWarner Losh 	const uint32_t zero = 0;
634ca987d46SWarner Losh 
635ca987d46SWarner Losh 	/* We want to modify every subnode of /cpus */
636ca987d46SWarner Losh 	o = fdt_path_offset(fdtp, "/cpus");
637ca987d46SWarner Losh 	if (o < 0)
638ca987d46SWarner Losh 		return;
639ca987d46SWarner Losh 
640ca987d46SWarner Losh 	/* maxo should contain offset of node next to /cpus */
641ca987d46SWarner Losh 	depth = 0;
642ca987d46SWarner Losh 	maxo = o;
643ca987d46SWarner Losh 	while (depth != -1)
644ca987d46SWarner Losh 		maxo = fdt_next_node(fdtp, maxo, &depth);
645ca987d46SWarner Losh 
646ca987d46SWarner Losh 	/* Find CPU frequency properties */
647ca987d46SWarner Losh 	o = fdt_node_offset_by_prop_value(fdtp, o, "clock-frequency",
648ca987d46SWarner Losh 	    &zero, sizeof(uint32_t));
649ca987d46SWarner Losh 
650ca987d46SWarner Losh 	o2 = fdt_node_offset_by_prop_value(fdtp, o, "bus-frequency", &zero,
651ca987d46SWarner Losh 	    sizeof(uint32_t));
652ca987d46SWarner Losh 
653ca987d46SWarner Losh 	lo = MIN(o, o2);
654ca987d46SWarner Losh 
655ca987d46SWarner Losh 	while (o != -FDT_ERR_NOTFOUND && o2 != -FDT_ERR_NOTFOUND) {
656ca987d46SWarner Losh 
657ca987d46SWarner Losh 		o = fdt_node_offset_by_prop_value(fdtp, lo,
658ca987d46SWarner Losh 		    "clock-frequency", &zero, sizeof(uint32_t));
659ca987d46SWarner Losh 
660ca987d46SWarner Losh 		o2 = fdt_node_offset_by_prop_value(fdtp, lo, "bus-frequency",
661ca987d46SWarner Losh 		    &zero, sizeof(uint32_t));
662ca987d46SWarner Losh 
663ca987d46SWarner Losh 		/* We're only interested in /cpus subnode(s) */
664ca987d46SWarner Losh 		if (lo > maxo)
665ca987d46SWarner Losh 			break;
666ca987d46SWarner Losh 
667ca987d46SWarner Losh 		fdt_setprop_inplace_cell(fdtp, lo, "clock-frequency",
668ca987d46SWarner Losh 		    (uint32_t)cpufreq);
669ca987d46SWarner Losh 
670ca987d46SWarner Losh 		fdt_setprop_inplace_cell(fdtp, lo, "bus-frequency",
671ca987d46SWarner Losh 		    (uint32_t)busfreq);
672ca987d46SWarner Losh 
673ca987d46SWarner Losh 		lo = MIN(o, o2);
674ca987d46SWarner Losh 	}
675ca987d46SWarner Losh }
676ca987d46SWarner Losh 
677ca987d46SWarner Losh #ifdef notyet
678ca987d46SWarner Losh static int
679ca987d46SWarner Losh fdt_reg_valid(uint32_t *reg, int len, int addr_cells, int size_cells)
680ca987d46SWarner Losh {
681ca987d46SWarner Losh 	int cells_in_tuple, i, tuples, tuple_size;
682ca987d46SWarner Losh 	uint32_t cur_start, cur_size;
683ca987d46SWarner Losh 
684ca987d46SWarner Losh 	cells_in_tuple = (addr_cells + size_cells);
685ca987d46SWarner Losh 	tuple_size = cells_in_tuple * sizeof(uint32_t);
686ca987d46SWarner Losh 	tuples = len / tuple_size;
687ca987d46SWarner Losh 	if (tuples == 0)
688ca987d46SWarner Losh 		return (EINVAL);
689ca987d46SWarner Losh 
690ca987d46SWarner Losh 	for (i = 0; i < tuples; i++) {
691ca987d46SWarner Losh 		if (addr_cells == 2)
692ca987d46SWarner Losh 			cur_start = fdt64_to_cpu(reg[i * cells_in_tuple]);
693ca987d46SWarner Losh 		else
694ca987d46SWarner Losh 			cur_start = fdt32_to_cpu(reg[i * cells_in_tuple]);
695ca987d46SWarner Losh 
696ca987d46SWarner Losh 		if (size_cells == 2)
697ca987d46SWarner Losh 			cur_size = fdt64_to_cpu(reg[i * cells_in_tuple + 2]);
698ca987d46SWarner Losh 		else
699ca987d46SWarner Losh 			cur_size = fdt32_to_cpu(reg[i * cells_in_tuple + 1]);
700ca987d46SWarner Losh 
701ca987d46SWarner Losh 		if (cur_size == 0)
702ca987d46SWarner Losh 			return (EINVAL);
703ca987d46SWarner Losh 
704ca987d46SWarner Losh 		debugf(" reg#%d (start: 0x%0x size: 0x%0x) valid!\n",
705ca987d46SWarner Losh 		    i, cur_start, cur_size);
706ca987d46SWarner Losh 	}
707ca987d46SWarner Losh 	return (0);
708ca987d46SWarner Losh }
709ca987d46SWarner Losh #endif
710ca987d46SWarner Losh 
711ca987d46SWarner Losh void
712ca987d46SWarner Losh fdt_fixup_memory(struct fdt_mem_region *region, size_t num)
713ca987d46SWarner Losh {
714ca987d46SWarner Losh 	struct fdt_mem_region *curmr;
715ca987d46SWarner Losh 	uint32_t addr_cells, size_cells;
716ca987d46SWarner Losh 	uint32_t *addr_cellsp, *size_cellsp;
717ca987d46SWarner Losh 	int err, i, len, memory, root;
718ca987d46SWarner Losh 	size_t realmrno;
719ca987d46SWarner Losh 	uint8_t *buf, *sb;
720ca987d46SWarner Losh 	uint64_t rstart, rsize;
721ca987d46SWarner Losh 	int reserved;
722ca987d46SWarner Losh 
723ca987d46SWarner Losh 	root = fdt_path_offset(fdtp, "/");
724ca987d46SWarner Losh 	if (root < 0) {
725ca987d46SWarner Losh 		sprintf(command_errbuf, "Could not find root node !");
726ca987d46SWarner Losh 		return;
727ca987d46SWarner Losh 	}
728ca987d46SWarner Losh 
729ca987d46SWarner Losh 	memory = fdt_path_offset(fdtp, "/memory");
730ca987d46SWarner Losh 	if (memory <= 0) {
731ca987d46SWarner Losh 		/* Create proper '/memory' node. */
732ca987d46SWarner Losh 		memory = fdt_add_subnode(fdtp, root, "memory");
733ca987d46SWarner Losh 		if (memory <= 0) {
734ca987d46SWarner Losh 			snprintf(command_errbuf, sizeof(command_errbuf),
735ca987d46SWarner Losh 			    "Could not fixup '/memory' "
736ca987d46SWarner Losh 			    "node, error code : %d!\n", memory);
737ca987d46SWarner Losh 			return;
738ca987d46SWarner Losh 		}
739ca987d46SWarner Losh 
740ca987d46SWarner Losh 		err = fdt_setprop(fdtp, memory, "device_type", "memory",
741ca987d46SWarner Losh 		    sizeof("memory"));
742ca987d46SWarner Losh 
743ca987d46SWarner Losh 		if (err < 0)
744ca987d46SWarner Losh 			return;
745ca987d46SWarner Losh 	}
746ca987d46SWarner Losh 
747ca987d46SWarner Losh 	addr_cellsp = (uint32_t *)fdt_getprop(fdtp, root, "#address-cells",
748ca987d46SWarner Losh 	    NULL);
749ca987d46SWarner Losh 	size_cellsp = (uint32_t *)fdt_getprop(fdtp, root, "#size-cells", NULL);
750ca987d46SWarner Losh 
751ca987d46SWarner Losh 	if (addr_cellsp == NULL || size_cellsp == NULL) {
752ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
753ca987d46SWarner Losh 		    "Could not fixup '/memory' node : "
754ca987d46SWarner Losh 		    "%s %s property not found in root node!\n",
755ca987d46SWarner Losh 		    (!addr_cellsp) ? "#address-cells" : "",
756ca987d46SWarner Losh 		    (!size_cellsp) ? "#size-cells" : "");
757ca987d46SWarner Losh 		return;
758ca987d46SWarner Losh 	}
759ca987d46SWarner Losh 
760ca987d46SWarner Losh 	addr_cells = fdt32_to_cpu(*addr_cellsp);
761ca987d46SWarner Losh 	size_cells = fdt32_to_cpu(*size_cellsp);
762ca987d46SWarner Losh 
763ca987d46SWarner Losh 	/*
764ca987d46SWarner Losh 	 * Convert memreserve data to memreserve property
765ca987d46SWarner Losh 	 * Check if property already exists
766ca987d46SWarner Losh 	 */
767ca987d46SWarner Losh 	reserved = fdt_num_mem_rsv(fdtp);
768ca987d46SWarner Losh 	if (reserved &&
769ca987d46SWarner Losh 	    (fdt_getprop(fdtp, root, "memreserve", NULL) == NULL)) {
770ca987d46SWarner Losh 		len = (addr_cells + size_cells) * reserved * sizeof(uint32_t);
771ca987d46SWarner Losh 		sb = buf = (uint8_t *)malloc(len);
772ca987d46SWarner Losh 		if (!buf)
773ca987d46SWarner Losh 			return;
774ca987d46SWarner Losh 
775ca987d46SWarner Losh 		bzero(buf, len);
776ca987d46SWarner Losh 
777ca987d46SWarner Losh 		for (i = 0; i < reserved; i++) {
778ca987d46SWarner Losh 			if (fdt_get_mem_rsv(fdtp, i, &rstart, &rsize))
779ca987d46SWarner Losh 				break;
780ca987d46SWarner Losh 			if (rsize) {
781ca987d46SWarner Losh 				/* Ensure endianness, and put cells into a buffer */
782ca987d46SWarner Losh 				if (addr_cells == 2)
783ca987d46SWarner Losh 					*(uint64_t *)buf =
784ca987d46SWarner Losh 					    cpu_to_fdt64(rstart);
785ca987d46SWarner Losh 				else
786ca987d46SWarner Losh 					*(uint32_t *)buf =
787ca987d46SWarner Losh 					    cpu_to_fdt32(rstart);
788ca987d46SWarner Losh 
789ca987d46SWarner Losh 				buf += sizeof(uint32_t) * addr_cells;
790ca987d46SWarner Losh 				if (size_cells == 2)
791ca987d46SWarner Losh 					*(uint64_t *)buf =
792ca987d46SWarner Losh 					    cpu_to_fdt64(rsize);
793ca987d46SWarner Losh 				else
794ca987d46SWarner Losh 					*(uint32_t *)buf =
795ca987d46SWarner Losh 					    cpu_to_fdt32(rsize);
796ca987d46SWarner Losh 
797ca987d46SWarner Losh 				buf += sizeof(uint32_t) * size_cells;
798ca987d46SWarner Losh 			}
799ca987d46SWarner Losh 		}
800ca987d46SWarner Losh 
801ca987d46SWarner Losh 		/* Set property */
802ca987d46SWarner Losh 		if ((err = fdt_setprop(fdtp, root, "memreserve", sb, len)) < 0)
803ca987d46SWarner Losh 			printf("Could not fixup 'memreserve' property.\n");
804ca987d46SWarner Losh 
805ca987d46SWarner Losh 		free(sb);
806ca987d46SWarner Losh 	}
807ca987d46SWarner Losh 
808ca987d46SWarner Losh 	/* Count valid memory regions entries in sysinfo. */
809ca987d46SWarner Losh 	realmrno = num;
810ca987d46SWarner Losh 	for (i = 0; i < num; i++)
811ca987d46SWarner Losh 		if (region[i].start == 0 && region[i].size == 0)
812ca987d46SWarner Losh 			realmrno--;
813ca987d46SWarner Losh 
814ca987d46SWarner Losh 	if (realmrno == 0) {
815ca987d46SWarner Losh 		sprintf(command_errbuf, "Could not fixup '/memory' node : "
816ca987d46SWarner Losh 		    "sysinfo doesn't contain valid memory regions info!\n");
817ca987d46SWarner Losh 		return;
818ca987d46SWarner Losh 	}
819ca987d46SWarner Losh 
820ca987d46SWarner Losh 	len = (addr_cells + size_cells) * realmrno * sizeof(uint32_t);
821ca987d46SWarner Losh 	sb = buf = (uint8_t *)malloc(len);
822ca987d46SWarner Losh 	if (!buf)
823ca987d46SWarner Losh 		return;
824ca987d46SWarner Losh 
825ca987d46SWarner Losh 	bzero(buf, len);
826ca987d46SWarner Losh 
827ca987d46SWarner Losh 	for (i = 0; i < num; i++) {
828ca987d46SWarner Losh 		curmr = &region[i];
829ca987d46SWarner Losh 		if (curmr->size != 0) {
830ca987d46SWarner Losh 			/* Ensure endianness, and put cells into a buffer */
831ca987d46SWarner Losh 			if (addr_cells == 2)
832ca987d46SWarner Losh 				*(uint64_t *)buf =
833ca987d46SWarner Losh 				    cpu_to_fdt64(curmr->start);
834ca987d46SWarner Losh 			else
835ca987d46SWarner Losh 				*(uint32_t *)buf =
836ca987d46SWarner Losh 				    cpu_to_fdt32(curmr->start);
837ca987d46SWarner Losh 
838ca987d46SWarner Losh 			buf += sizeof(uint32_t) * addr_cells;
839ca987d46SWarner Losh 			if (size_cells == 2)
840ca987d46SWarner Losh 				*(uint64_t *)buf =
841ca987d46SWarner Losh 				    cpu_to_fdt64(curmr->size);
842ca987d46SWarner Losh 			else
843ca987d46SWarner Losh 				*(uint32_t *)buf =
844ca987d46SWarner Losh 				    cpu_to_fdt32(curmr->size);
845ca987d46SWarner Losh 
846ca987d46SWarner Losh 			buf += sizeof(uint32_t) * size_cells;
847ca987d46SWarner Losh 		}
848ca987d46SWarner Losh 	}
849ca987d46SWarner Losh 
850ca987d46SWarner Losh 	/* Set property */
851ca987d46SWarner Losh 	if ((err = fdt_setprop(fdtp, memory, "reg", sb, len)) < 0)
852ca987d46SWarner Losh 		sprintf(command_errbuf, "Could not fixup '/memory' node.\n");
853ca987d46SWarner Losh 
854ca987d46SWarner Losh 	free(sb);
855ca987d46SWarner Losh }
856ca987d46SWarner Losh 
857ca987d46SWarner Losh void
858ca987d46SWarner Losh fdt_fixup_stdout(const char *str)
859ca987d46SWarner Losh {
860ca987d46SWarner Losh 	char *ptr;
861ca987d46SWarner Losh 	int len, no, sero;
862ca987d46SWarner Losh 	const struct fdt_property *prop;
863ca987d46SWarner Losh 	char *tmp[10];
864ca987d46SWarner Losh 
865ca987d46SWarner Losh 	ptr = (char *)str + strlen(str) - 1;
866ca987d46SWarner Losh 	while (ptr > str && isdigit(*(str - 1)))
867ca987d46SWarner Losh 		str--;
868ca987d46SWarner Losh 
869ca987d46SWarner Losh 	if (ptr == str)
870ca987d46SWarner Losh 		return;
871ca987d46SWarner Losh 
872ca987d46SWarner Losh 	no = fdt_path_offset(fdtp, "/chosen");
873ca987d46SWarner Losh 	if (no < 0)
874ca987d46SWarner Losh 		return;
875ca987d46SWarner Losh 
876ca987d46SWarner Losh 	prop = fdt_get_property(fdtp, no, "stdout", &len);
877ca987d46SWarner Losh 
878ca987d46SWarner Losh 	/* If /chosen/stdout does not extist, create it */
879ca987d46SWarner Losh 	if (prop == NULL || (prop != NULL && len == 0)) {
880ca987d46SWarner Losh 
881ca987d46SWarner Losh 		bzero(tmp, 10 * sizeof(char));
882ca987d46SWarner Losh 		strcpy((char *)&tmp, "serial");
883ca987d46SWarner Losh 		if (strlen(ptr) > 3)
884ca987d46SWarner Losh 			/* Serial number too long */
885ca987d46SWarner Losh 			return;
886ca987d46SWarner Losh 
887ca987d46SWarner Losh 		strncpy((char *)tmp + 6, ptr, 3);
888ca987d46SWarner Losh 		sero = fdt_path_offset(fdtp, (const char *)tmp);
889ca987d46SWarner Losh 		if (sero < 0)
890ca987d46SWarner Losh 			/*
891ca987d46SWarner Losh 			 * If serial device we're trying to assign
892ca987d46SWarner Losh 			 * stdout to doesn't exist in DT -- return.
893ca987d46SWarner Losh 			 */
894ca987d46SWarner Losh 			return;
895ca987d46SWarner Losh 
896ca987d46SWarner Losh 		fdt_setprop(fdtp, no, "stdout", &tmp,
897ca987d46SWarner Losh 		    strlen((char *)&tmp) + 1);
898ca987d46SWarner Losh 		fdt_setprop(fdtp, no, "stdin", &tmp,
899ca987d46SWarner Losh 		    strlen((char *)&tmp) + 1);
900ca987d46SWarner Losh 	}
901ca987d46SWarner Losh }
902ca987d46SWarner Losh 
9038a613444SKyle Evans void
9048a613444SKyle Evans fdt_load_dtb_overlays(const char *extras)
9058a613444SKyle Evans {
9068a613444SKyle Evans 	const char *s;
9078a613444SKyle Evans 
9088a613444SKyle Evans 	/* Any extra overlays supplied by pre-loader environment */
9098a613444SKyle Evans 	if (extras != NULL && *extras != '\0') {
9108a613444SKyle Evans 		printf("Loading DTB overlays: '%s'\n", extras);
9118a613444SKyle Evans 		fdt_load_dtb_overlays_string(extras);
9128a613444SKyle Evans 	}
9138a613444SKyle Evans 
9148a613444SKyle Evans 	/* Any overlays supplied by loader environment */
9158a613444SKyle Evans 	s = getenv("fdt_overlays");
9168a613444SKyle Evans 	if (s != NULL && *s != '\0') {
9178a613444SKyle Evans 		printf("Loading DTB overlays: '%s'\n", s);
9188a613444SKyle Evans 		fdt_load_dtb_overlays_string(s);
9198a613444SKyle Evans 	}
9208a613444SKyle Evans }
9218a613444SKyle Evans 
922ca987d46SWarner Losh /*
923ca987d46SWarner Losh  * Locate the blob, fix it up and return its location.
924ca987d46SWarner Losh  */
925ca987d46SWarner Losh static int
926ca987d46SWarner Losh fdt_fixup(void)
927ca987d46SWarner Losh {
92813eade28SKyle Evans 	int chosen;
929ca987d46SWarner Losh 
930ca987d46SWarner Losh 	debugf("fdt_fixup()\n");
931ca987d46SWarner Losh 
932ca987d46SWarner Losh 	if (fdtp == NULL && fdt_setup_fdtp() != 0)
933ca987d46SWarner Losh 		return (0);
934ca987d46SWarner Losh 
935ca987d46SWarner Losh 	/* Create /chosen node (if not exists) */
936ca987d46SWarner Losh 	if ((chosen = fdt_subnode_offset(fdtp, 0, "chosen")) ==
937ca987d46SWarner Losh 	    -FDT_ERR_NOTFOUND)
938ca987d46SWarner Losh 		chosen = fdt_add_subnode(fdtp, 0, "chosen");
939ca987d46SWarner Losh 
940ca987d46SWarner Losh 	/* Value assigned to fixup-applied does not matter. */
941ca987d46SWarner Losh 	if (fdt_getprop(fdtp, chosen, "fixup-applied", NULL))
942ca987d46SWarner Losh 		return (1);
943ca987d46SWarner Losh 
944ca987d46SWarner Losh 	fdt_platform_fixups();
945ca987d46SWarner Losh 
94698d03353SKyle Evans 	/*
94798d03353SKyle Evans 	 * Re-fetch the /chosen subnode; our fixups may apply overlays or add
94898d03353SKyle Evans 	 * nodes/properties that invalidate the offset we grabbed or created
94998d03353SKyle Evans 	 * above, so we can no longer trust it.
95098d03353SKyle Evans 	 */
95198d03353SKyle Evans 	chosen = fdt_subnode_offset(fdtp, 0, "chosen");
952ca987d46SWarner Losh 	fdt_setprop(fdtp, chosen, "fixup-applied", NULL, 0);
953ca987d46SWarner Losh 	return (1);
954ca987d46SWarner Losh }
955ca987d46SWarner Losh 
956ca987d46SWarner Losh /*
957ca987d46SWarner Losh  * Copy DTB blob to specified location and return size
958ca987d46SWarner Losh  */
959ca987d46SWarner Losh int
960ca987d46SWarner Losh fdt_copy(vm_offset_t va)
961ca987d46SWarner Losh {
962ca987d46SWarner Losh 	int err;
963ca987d46SWarner Losh 	debugf("fdt_copy va 0x%08x\n", va);
964ca987d46SWarner Losh 	if (fdtp == NULL) {
965ca987d46SWarner Losh 		err = fdt_setup_fdtp();
966ca987d46SWarner Losh 		if (err) {
967ca987d46SWarner Losh 			printf("No valid device tree blob found!\n");
968ca987d46SWarner Losh 			return (0);
969ca987d46SWarner Losh 		}
970ca987d46SWarner Losh 	}
971ca987d46SWarner Losh 
972ca987d46SWarner Losh 	if (fdt_fixup() == 0)
973ca987d46SWarner Losh 		return (0);
974ca987d46SWarner Losh 
975ca987d46SWarner Losh 	COPYIN(fdtp, va, fdtp_size);
976ca987d46SWarner Losh 	return (fdtp_size);
977ca987d46SWarner Losh }
978ca987d46SWarner Losh 
979ca987d46SWarner Losh 
980ca987d46SWarner Losh 
981ca987d46SWarner Losh int
982ca987d46SWarner Losh command_fdt_internal(int argc, char *argv[])
983ca987d46SWarner Losh {
984ca987d46SWarner Losh 	cmdf_t *cmdh;
985ca987d46SWarner Losh 	int flags;
986ca987d46SWarner Losh 	int i, err;
987ca987d46SWarner Losh 
988ca987d46SWarner Losh 	if (argc < 2) {
989ca987d46SWarner Losh 		command_errmsg = "usage is 'fdt <command> [<args>]";
990ca987d46SWarner Losh 		return (CMD_ERROR);
991ca987d46SWarner Losh 	}
992ca987d46SWarner Losh 
993ca987d46SWarner Losh 	/*
994ca987d46SWarner Losh 	 * Validate fdt <command>.
995ca987d46SWarner Losh 	 */
996ca987d46SWarner Losh 	i = 0;
997ca987d46SWarner Losh 	cmdh = NULL;
998ca987d46SWarner Losh 	while (!(commands[i].name == NULL)) {
99913eade28SKyle Evans 		if (strcmp(argv[1], commands[i].name) == 0) {
1000ca987d46SWarner Losh 			/* found it */
1001ca987d46SWarner Losh 			cmdh = commands[i].handler;
1002ca987d46SWarner Losh 			flags = commands[i].flags;
1003ca987d46SWarner Losh 			break;
1004ca987d46SWarner Losh 		}
1005ca987d46SWarner Losh 		i++;
1006ca987d46SWarner Losh 	}
1007ca987d46SWarner Losh 	if (cmdh == NULL) {
1008ca987d46SWarner Losh 		command_errmsg = "unknown command";
1009ca987d46SWarner Losh 		return (CMD_ERROR);
1010ca987d46SWarner Losh 	}
1011ca987d46SWarner Losh 
1012ca987d46SWarner Losh 	if (flags & CMD_REQUIRES_BLOB) {
1013ca987d46SWarner Losh 		/*
1014ca987d46SWarner Losh 		 * Check if uboot env vars were parsed already. If not, do it now.
1015ca987d46SWarner Losh 		 */
1016ca987d46SWarner Losh 		if (fdt_fixup() == 0)
1017ca987d46SWarner Losh 			return (CMD_ERROR);
1018ca987d46SWarner Losh 	}
1019ca987d46SWarner Losh 
1020ca987d46SWarner Losh 	/*
1021ca987d46SWarner Losh 	 * Call command handler.
1022ca987d46SWarner Losh 	 */
1023ca987d46SWarner Losh 	err = (*cmdh)(argc, argv);
1024ca987d46SWarner Losh 
1025ca987d46SWarner Losh 	return (err);
1026ca987d46SWarner Losh }
1027ca987d46SWarner Losh 
1028ca987d46SWarner Losh static int
1029ca987d46SWarner Losh fdt_cmd_addr(int argc, char *argv[])
1030ca987d46SWarner Losh {
1031ca987d46SWarner Losh 	struct preloaded_file *fp;
1032ca987d46SWarner Losh 	struct fdt_header *hdr;
1033ca987d46SWarner Losh 	const char *addr;
1034ca987d46SWarner Losh 	char *cp;
1035ca987d46SWarner Losh 
1036ca987d46SWarner Losh 	fdt_to_load = NULL;
1037ca987d46SWarner Losh 
1038ca987d46SWarner Losh 	if (argc > 2)
1039ca987d46SWarner Losh 		addr = argv[2];
1040ca987d46SWarner Losh 	else {
1041ca987d46SWarner Losh 		sprintf(command_errbuf, "no address specified");
1042ca987d46SWarner Losh 		return (CMD_ERROR);
1043ca987d46SWarner Losh 	}
1044ca987d46SWarner Losh 
1045ca987d46SWarner Losh 	hdr = (struct fdt_header *)strtoul(addr, &cp, 16);
1046ca987d46SWarner Losh 	if (cp == addr) {
1047ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
1048ca987d46SWarner Losh 		    "Invalid address: %s", addr);
1049ca987d46SWarner Losh 		return (CMD_ERROR);
1050ca987d46SWarner Losh 	}
1051ca987d46SWarner Losh 
1052ca987d46SWarner Losh 	while ((fp = file_findfile(NULL, "dtb")) != NULL) {
1053ca987d46SWarner Losh 		file_discard(fp);
1054ca987d46SWarner Losh 	}
1055ca987d46SWarner Losh 
1056ca987d46SWarner Losh 	fdt_to_load = hdr;
1057ca987d46SWarner Losh 	return (CMD_OK);
1058ca987d46SWarner Losh }
1059ca987d46SWarner Losh 
1060ca987d46SWarner Losh static int
1061ca987d46SWarner Losh fdt_cmd_cd(int argc, char *argv[])
1062ca987d46SWarner Losh {
1063ca987d46SWarner Losh 	char *path;
1064ca987d46SWarner Losh 	char tmp[FDT_CWD_LEN];
1065ca987d46SWarner Losh 	int len, o;
1066ca987d46SWarner Losh 
1067ca987d46SWarner Losh 	path = (argc > 2) ? argv[2] : "/";
1068ca987d46SWarner Losh 
1069ca987d46SWarner Losh 	if (path[0] == '/') {
1070ca987d46SWarner Losh 		len = strlen(path);
1071ca987d46SWarner Losh 		if (len >= FDT_CWD_LEN)
1072ca987d46SWarner Losh 			goto fail;
1073ca987d46SWarner Losh 	} else {
1074ca987d46SWarner Losh 		/* Handle path specification relative to cwd */
1075ca987d46SWarner Losh 		len = strlen(cwd) + strlen(path) + 1;
1076ca987d46SWarner Losh 		if (len >= FDT_CWD_LEN)
1077ca987d46SWarner Losh 			goto fail;
1078ca987d46SWarner Losh 
1079ca987d46SWarner Losh 		strcpy(tmp, cwd);
1080ca987d46SWarner Losh 		strcat(tmp, "/");
1081ca987d46SWarner Losh 		strcat(tmp, path);
1082ca987d46SWarner Losh 		path = tmp;
1083ca987d46SWarner Losh 	}
1084ca987d46SWarner Losh 
1085ca987d46SWarner Losh 	o = fdt_path_offset(fdtp, path);
1086ca987d46SWarner Losh 	if (o < 0) {
1087ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
1088ca987d46SWarner Losh 		    "could not find node: '%s'", path);
1089ca987d46SWarner Losh 		return (CMD_ERROR);
1090ca987d46SWarner Losh 	}
1091ca987d46SWarner Losh 
1092ca987d46SWarner Losh 	strcpy(cwd, path);
1093ca987d46SWarner Losh 	return (CMD_OK);
1094ca987d46SWarner Losh 
1095ca987d46SWarner Losh fail:
1096ca987d46SWarner Losh 	snprintf(command_errbuf, sizeof(command_errbuf),
1097ca987d46SWarner Losh 	    "path too long: %d, max allowed: %d", len, FDT_CWD_LEN - 1);
1098ca987d46SWarner Losh 	return (CMD_ERROR);
1099ca987d46SWarner Losh }
1100ca987d46SWarner Losh 
1101ca987d46SWarner Losh static int
1102ca987d46SWarner Losh fdt_cmd_hdr(int argc __unused, char *argv[] __unused)
1103ca987d46SWarner Losh {
1104ca987d46SWarner Losh 	char line[80];
1105ca987d46SWarner Losh 	int ver;
1106ca987d46SWarner Losh 
1107ca987d46SWarner Losh 	if (fdtp == NULL) {
1108ca987d46SWarner Losh 		command_errmsg = "no device tree blob pointer?!";
1109ca987d46SWarner Losh 		return (CMD_ERROR);
1110ca987d46SWarner Losh 	}
1111ca987d46SWarner Losh 
1112ca987d46SWarner Losh 	ver = fdt_version(fdtp);
1113ca987d46SWarner Losh 	pager_open();
1114ca987d46SWarner Losh 	sprintf(line, "\nFlattened device tree header (%p):\n", fdtp);
1115ca987d46SWarner Losh 	if (pager_output(line))
1116ca987d46SWarner Losh 		goto out;
1117ca987d46SWarner Losh 	sprintf(line, " magic                   = 0x%08x\n", fdt_magic(fdtp));
1118ca987d46SWarner Losh 	if (pager_output(line))
1119ca987d46SWarner Losh 		goto out;
1120ca987d46SWarner Losh 	sprintf(line, " size                    = %d\n", fdt_totalsize(fdtp));
1121ca987d46SWarner Losh 	if (pager_output(line))
1122ca987d46SWarner Losh 		goto out;
1123ca987d46SWarner Losh 	sprintf(line, " off_dt_struct           = 0x%08x\n",
1124ca987d46SWarner Losh 	    fdt_off_dt_struct(fdtp));
1125ca987d46SWarner Losh 	if (pager_output(line))
1126ca987d46SWarner Losh 		goto out;
1127ca987d46SWarner Losh 	sprintf(line, " off_dt_strings          = 0x%08x\n",
1128ca987d46SWarner Losh 	    fdt_off_dt_strings(fdtp));
1129ca987d46SWarner Losh 	if (pager_output(line))
1130ca987d46SWarner Losh 		goto out;
1131ca987d46SWarner Losh 	sprintf(line, " off_mem_rsvmap          = 0x%08x\n",
1132ca987d46SWarner Losh 	    fdt_off_mem_rsvmap(fdtp));
1133ca987d46SWarner Losh 	if (pager_output(line))
1134ca987d46SWarner Losh 		goto out;
1135ca987d46SWarner Losh 	sprintf(line, " version                 = %d\n", ver);
1136ca987d46SWarner Losh 	if (pager_output(line))
1137ca987d46SWarner Losh 		goto out;
1138ca987d46SWarner Losh 	sprintf(line, " last compatible version = %d\n",
1139ca987d46SWarner Losh 	    fdt_last_comp_version(fdtp));
1140ca987d46SWarner Losh 	if (pager_output(line))
1141ca987d46SWarner Losh 		goto out;
1142ca987d46SWarner Losh 	if (ver >= 2) {
1143ca987d46SWarner Losh 		sprintf(line, " boot_cpuid              = %d\n",
1144ca987d46SWarner Losh 		    fdt_boot_cpuid_phys(fdtp));
1145ca987d46SWarner Losh 		if (pager_output(line))
1146ca987d46SWarner Losh 			goto out;
1147ca987d46SWarner Losh 	}
1148ca987d46SWarner Losh 	if (ver >= 3) {
1149ca987d46SWarner Losh 		sprintf(line, " size_dt_strings         = %d\n",
1150ca987d46SWarner Losh 		    fdt_size_dt_strings(fdtp));
1151ca987d46SWarner Losh 		if (pager_output(line))
1152ca987d46SWarner Losh 			goto out;
1153ca987d46SWarner Losh 	}
1154ca987d46SWarner Losh 	if (ver >= 17) {
1155ca987d46SWarner Losh 		sprintf(line, " size_dt_struct          = %d\n",
1156ca987d46SWarner Losh 		    fdt_size_dt_struct(fdtp));
1157ca987d46SWarner Losh 		if (pager_output(line))
1158ca987d46SWarner Losh 			goto out;
1159ca987d46SWarner Losh 	}
1160ca987d46SWarner Losh out:
1161ca987d46SWarner Losh 	pager_close();
1162ca987d46SWarner Losh 
1163ca987d46SWarner Losh 	return (CMD_OK);
1164ca987d46SWarner Losh }
1165ca987d46SWarner Losh 
1166ca987d46SWarner Losh static int
1167ca987d46SWarner Losh fdt_cmd_ls(int argc, char *argv[])
1168ca987d46SWarner Losh {
1169ca987d46SWarner Losh 	const char *prevname[FDT_MAX_DEPTH] = { NULL };
1170ca987d46SWarner Losh 	const char *name;
1171ca987d46SWarner Losh 	char *path;
1172ca987d46SWarner Losh 	int i, o, depth;
1173ca987d46SWarner Losh 
1174ca987d46SWarner Losh 	path = (argc > 2) ? argv[2] : NULL;
1175ca987d46SWarner Losh 	if (path == NULL)
1176ca987d46SWarner Losh 		path = cwd;
1177ca987d46SWarner Losh 
1178ca987d46SWarner Losh 	o = fdt_path_offset(fdtp, path);
1179ca987d46SWarner Losh 	if (o < 0) {
1180ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
1181ca987d46SWarner Losh 		    "could not find node: '%s'", path);
1182ca987d46SWarner Losh 		return (CMD_ERROR);
1183ca987d46SWarner Losh 	}
1184ca987d46SWarner Losh 
1185ca987d46SWarner Losh 	for (depth = 0;
1186ca987d46SWarner Losh 	    (o >= 0) && (depth >= 0);
1187ca987d46SWarner Losh 	    o = fdt_next_node(fdtp, o, &depth)) {
1188ca987d46SWarner Losh 
1189ca987d46SWarner Losh 		name = fdt_get_name(fdtp, o, NULL);
1190ca987d46SWarner Losh 
1191ca987d46SWarner Losh 		if (depth > FDT_MAX_DEPTH) {
1192ca987d46SWarner Losh 			printf("max depth exceeded: %d\n", depth);
1193ca987d46SWarner Losh 			continue;
1194ca987d46SWarner Losh 		}
1195ca987d46SWarner Losh 
1196ca987d46SWarner Losh 		prevname[depth] = name;
1197ca987d46SWarner Losh 
1198ca987d46SWarner Losh 		/* Skip root (i = 1) when printing devices */
1199ca987d46SWarner Losh 		for (i = 1; i <= depth; i++) {
1200ca987d46SWarner Losh 			if (prevname[i] == NULL)
1201ca987d46SWarner Losh 				break;
1202ca987d46SWarner Losh 
1203ca987d46SWarner Losh 			if (strcmp(cwd, "/") == 0)
1204ca987d46SWarner Losh 				printf("/");
1205ca987d46SWarner Losh 			printf("%s", prevname[i]);
1206ca987d46SWarner Losh 		}
1207ca987d46SWarner Losh 		printf("\n");
1208ca987d46SWarner Losh 	}
1209ca987d46SWarner Losh 
1210ca987d46SWarner Losh 	return (CMD_OK);
1211ca987d46SWarner Losh }
1212ca987d46SWarner Losh 
1213ca987d46SWarner Losh static __inline int
1214ca987d46SWarner Losh isprint(int c)
1215ca987d46SWarner Losh {
1216ca987d46SWarner Losh 
1217ca987d46SWarner Losh 	return (c >= ' ' && c <= 0x7e);
1218ca987d46SWarner Losh }
1219ca987d46SWarner Losh 
1220ca987d46SWarner Losh static int
1221ca987d46SWarner Losh fdt_isprint(const void *data, int len, int *count)
1222ca987d46SWarner Losh {
1223ca987d46SWarner Losh 	const char *d;
1224ca987d46SWarner Losh 	char ch;
1225ca987d46SWarner Losh 	int yesno, i;
1226ca987d46SWarner Losh 
1227ca987d46SWarner Losh 	if (len == 0)
1228ca987d46SWarner Losh 		return (0);
1229ca987d46SWarner Losh 
1230ca987d46SWarner Losh 	d = (const char *)data;
1231ca987d46SWarner Losh 	if (d[len - 1] != '\0')
1232ca987d46SWarner Losh 		return (0);
1233ca987d46SWarner Losh 
1234ca987d46SWarner Losh 	*count = 0;
1235ca987d46SWarner Losh 	yesno = 1;
1236ca987d46SWarner Losh 	for (i = 0; i < len; i++) {
1237ca987d46SWarner Losh 		ch = *(d + i);
1238ca987d46SWarner Losh 		if (isprint(ch) || (ch == '\0' && i > 0)) {
1239ca987d46SWarner Losh 			/* Count strings */
1240ca987d46SWarner Losh 			if (ch == '\0')
1241ca987d46SWarner Losh 				(*count)++;
1242ca987d46SWarner Losh 			continue;
1243ca987d46SWarner Losh 		}
1244ca987d46SWarner Losh 
1245ca987d46SWarner Losh 		yesno = 0;
1246ca987d46SWarner Losh 		break;
1247ca987d46SWarner Losh 	}
1248ca987d46SWarner Losh 
1249ca987d46SWarner Losh 	return (yesno);
1250ca987d46SWarner Losh }
1251ca987d46SWarner Losh 
1252ca987d46SWarner Losh static int
1253ca987d46SWarner Losh fdt_data_str(const void *data, int len, int count, char **buf)
1254ca987d46SWarner Losh {
1255ca987d46SWarner Losh 	char *b, *tmp;
1256ca987d46SWarner Losh 	const char *d;
1257ca987d46SWarner Losh 	int buf_len, i, l;
1258ca987d46SWarner Losh 
1259ca987d46SWarner Losh 	/*
1260ca987d46SWarner Losh 	 * Calculate the length for the string and allocate memory.
1261ca987d46SWarner Losh 	 *
1262ca987d46SWarner Losh 	 * Note that 'len' already includes at least one terminator.
1263ca987d46SWarner Losh 	 */
1264ca987d46SWarner Losh 	buf_len = len;
1265ca987d46SWarner Losh 	if (count > 1) {
1266ca987d46SWarner Losh 		/*
1267ca987d46SWarner Losh 		 * Each token had already a terminator buried in 'len', but we
1268ca987d46SWarner Losh 		 * only need one eventually, don't count space for these.
1269ca987d46SWarner Losh 		 */
1270ca987d46SWarner Losh 		buf_len -= count - 1;
1271ca987d46SWarner Losh 
1272ca987d46SWarner Losh 		/* Each consecutive token requires a ", " separator. */
1273ca987d46SWarner Losh 		buf_len += count * 2;
1274ca987d46SWarner Losh 	}
1275ca987d46SWarner Losh 
1276ca987d46SWarner Losh 	/* Add some space for surrounding double quotes. */
1277ca987d46SWarner Losh 	buf_len += count * 2;
1278ca987d46SWarner Losh 
1279ca987d46SWarner Losh 	/* Note that string being put in 'tmp' may be as big as 'buf_len'. */
1280ca987d46SWarner Losh 	b = (char *)malloc(buf_len);
1281ca987d46SWarner Losh 	tmp = (char *)malloc(buf_len);
1282ca987d46SWarner Losh 	if (b == NULL)
1283ca987d46SWarner Losh 		goto error;
1284ca987d46SWarner Losh 
1285ca987d46SWarner Losh 	if (tmp == NULL) {
1286ca987d46SWarner Losh 		free(b);
1287ca987d46SWarner Losh 		goto error;
1288ca987d46SWarner Losh 	}
1289ca987d46SWarner Losh 
1290ca987d46SWarner Losh 	b[0] = '\0';
1291ca987d46SWarner Losh 
1292ca987d46SWarner Losh 	/*
1293ca987d46SWarner Losh 	 * Now that we have space, format the string.
1294ca987d46SWarner Losh 	 */
1295ca987d46SWarner Losh 	i = 0;
1296ca987d46SWarner Losh 	do {
1297ca987d46SWarner Losh 		d = (const char *)data + i;
1298ca987d46SWarner Losh 		l = strlen(d) + 1;
1299ca987d46SWarner Losh 
1300ca987d46SWarner Losh 		sprintf(tmp, "\"%s\"%s", d,
1301ca987d46SWarner Losh 		    (i + l) < len ?  ", " : "");
1302ca987d46SWarner Losh 		strcat(b, tmp);
1303ca987d46SWarner Losh 
1304ca987d46SWarner Losh 		i += l;
1305ca987d46SWarner Losh 
1306ca987d46SWarner Losh 	} while (i < len);
1307ca987d46SWarner Losh 	*buf = b;
1308ca987d46SWarner Losh 
1309ca987d46SWarner Losh 	free(tmp);
1310ca987d46SWarner Losh 
1311ca987d46SWarner Losh 	return (0);
1312ca987d46SWarner Losh error:
1313ca987d46SWarner Losh 	return (1);
1314ca987d46SWarner Losh }
1315ca987d46SWarner Losh 
1316ca987d46SWarner Losh static int
1317ca987d46SWarner Losh fdt_data_cell(const void *data, int len, char **buf)
1318ca987d46SWarner Losh {
1319ca987d46SWarner Losh 	char *b, *tmp;
1320ca987d46SWarner Losh 	const uint32_t *c;
1321ca987d46SWarner Losh 	int count, i, l;
1322ca987d46SWarner Losh 
1323ca987d46SWarner Losh 	/* Number of cells */
1324ca987d46SWarner Losh 	count = len / 4;
1325ca987d46SWarner Losh 
1326ca987d46SWarner Losh 	/*
1327ca987d46SWarner Losh 	 * Calculate the length for the string and allocate memory.
1328ca987d46SWarner Losh 	 */
1329ca987d46SWarner Losh 
1330ca987d46SWarner Losh 	/* Each byte translates to 2 output characters */
1331ca987d46SWarner Losh 	l = len * 2;
1332ca987d46SWarner Losh 	if (count > 1) {
1333ca987d46SWarner Losh 		/* Each consecutive cell requires a " " separator. */
1334ca987d46SWarner Losh 		l += (count - 1) * 1;
1335ca987d46SWarner Losh 	}
1336ca987d46SWarner Losh 	/* Each cell will have a "0x" prefix */
1337ca987d46SWarner Losh 	l += count * 2;
1338ca987d46SWarner Losh 	/* Space for surrounding <> and terminator */
1339ca987d46SWarner Losh 	l += 3;
1340ca987d46SWarner Losh 
1341ca987d46SWarner Losh 	b = (char *)malloc(l);
1342ca987d46SWarner Losh 	tmp = (char *)malloc(l);
1343ca987d46SWarner Losh 	if (b == NULL)
1344ca987d46SWarner Losh 		goto error;
1345ca987d46SWarner Losh 
1346ca987d46SWarner Losh 	if (tmp == NULL) {
1347ca987d46SWarner Losh 		free(b);
1348ca987d46SWarner Losh 		goto error;
1349ca987d46SWarner Losh 	}
1350ca987d46SWarner Losh 
1351ca987d46SWarner Losh 	b[0] = '\0';
1352ca987d46SWarner Losh 	strcat(b, "<");
1353ca987d46SWarner Losh 
1354ca987d46SWarner Losh 	for (i = 0; i < len; i += 4) {
1355ca987d46SWarner Losh 		c = (const uint32_t *)((const uint8_t *)data + i);
1356ca987d46SWarner Losh 		sprintf(tmp, "0x%08x%s", fdt32_to_cpu(*c),
1357ca987d46SWarner Losh 		    i < (len - 4) ? " " : "");
1358ca987d46SWarner Losh 		strcat(b, tmp);
1359ca987d46SWarner Losh 	}
1360ca987d46SWarner Losh 	strcat(b, ">");
1361ca987d46SWarner Losh 	*buf = b;
1362ca987d46SWarner Losh 
1363ca987d46SWarner Losh 	free(tmp);
1364ca987d46SWarner Losh 
1365ca987d46SWarner Losh 	return (0);
1366ca987d46SWarner Losh error:
1367ca987d46SWarner Losh 	return (1);
1368ca987d46SWarner Losh }
1369ca987d46SWarner Losh 
1370ca987d46SWarner Losh static int
1371ca987d46SWarner Losh fdt_data_bytes(const void *data, int len, char **buf)
1372ca987d46SWarner Losh {
1373ca987d46SWarner Losh 	char *b, *tmp;
1374ca987d46SWarner Losh 	const char *d;
1375ca987d46SWarner Losh 	int i, l;
1376ca987d46SWarner Losh 
1377ca987d46SWarner Losh 	/*
1378ca987d46SWarner Losh 	 * Calculate the length for the string and allocate memory.
1379ca987d46SWarner Losh 	 */
1380ca987d46SWarner Losh 
1381ca987d46SWarner Losh 	/* Each byte translates to 2 output characters */
1382ca987d46SWarner Losh 	l = len * 2;
1383ca987d46SWarner Losh 	if (len > 1)
1384ca987d46SWarner Losh 		/* Each consecutive byte requires a " " separator. */
1385ca987d46SWarner Losh 		l += (len - 1) * 1;
1386ca987d46SWarner Losh 	/* Each byte will have a "0x" prefix */
1387ca987d46SWarner Losh 	l += len * 2;
1388ca987d46SWarner Losh 	/* Space for surrounding [] and terminator. */
1389ca987d46SWarner Losh 	l += 3;
1390ca987d46SWarner Losh 
1391ca987d46SWarner Losh 	b = (char *)malloc(l);
1392ca987d46SWarner Losh 	tmp = (char *)malloc(l);
1393ca987d46SWarner Losh 	if (b == NULL)
1394ca987d46SWarner Losh 		goto error;
1395ca987d46SWarner Losh 
1396ca987d46SWarner Losh 	if (tmp == NULL) {
1397ca987d46SWarner Losh 		free(b);
1398ca987d46SWarner Losh 		goto error;
1399ca987d46SWarner Losh 	}
1400ca987d46SWarner Losh 
1401ca987d46SWarner Losh 	b[0] = '\0';
1402ca987d46SWarner Losh 	strcat(b, "[");
1403ca987d46SWarner Losh 
1404ca987d46SWarner Losh 	for (i = 0, d = data; i < len; i++) {
1405ca987d46SWarner Losh 		sprintf(tmp, "0x%02x%s", d[i], i < len - 1 ? " " : "");
1406ca987d46SWarner Losh 		strcat(b, tmp);
1407ca987d46SWarner Losh 	}
1408ca987d46SWarner Losh 	strcat(b, "]");
1409ca987d46SWarner Losh 	*buf = b;
1410ca987d46SWarner Losh 
1411ca987d46SWarner Losh 	free(tmp);
1412ca987d46SWarner Losh 
1413ca987d46SWarner Losh 	return (0);
1414ca987d46SWarner Losh error:
1415ca987d46SWarner Losh 	return (1);
1416ca987d46SWarner Losh }
1417ca987d46SWarner Losh 
1418ca987d46SWarner Losh static int
1419ca987d46SWarner Losh fdt_data_fmt(const void *data, int len, char **buf)
1420ca987d46SWarner Losh {
1421ca987d46SWarner Losh 	int count;
1422ca987d46SWarner Losh 
1423ca987d46SWarner Losh 	if (len == 0) {
1424ca987d46SWarner Losh 		*buf = NULL;
1425ca987d46SWarner Losh 		return (1);
1426ca987d46SWarner Losh 	}
1427ca987d46SWarner Losh 
1428ca987d46SWarner Losh 	if (fdt_isprint(data, len, &count))
1429ca987d46SWarner Losh 		return (fdt_data_str(data, len, count, buf));
1430ca987d46SWarner Losh 
1431ca987d46SWarner Losh 	else if ((len % 4) == 0)
1432ca987d46SWarner Losh 		return (fdt_data_cell(data, len, buf));
1433ca987d46SWarner Losh 
1434ca987d46SWarner Losh 	else
1435ca987d46SWarner Losh 		return (fdt_data_bytes(data, len, buf));
1436ca987d46SWarner Losh }
1437ca987d46SWarner Losh 
1438ca987d46SWarner Losh static int
1439ca987d46SWarner Losh fdt_prop(int offset)
1440ca987d46SWarner Losh {
1441ca987d46SWarner Losh 	char *line, *buf;
1442ca987d46SWarner Losh 	const struct fdt_property *prop;
1443ca987d46SWarner Losh 	const char *name;
1444ca987d46SWarner Losh 	const void *data;
1445ca987d46SWarner Losh 	int len, rv;
1446ca987d46SWarner Losh 
1447ca987d46SWarner Losh 	line = NULL;
1448ca987d46SWarner Losh 	prop = fdt_offset_ptr(fdtp, offset, sizeof(*prop));
1449ca987d46SWarner Losh 	if (prop == NULL)
1450ca987d46SWarner Losh 		return (1);
1451ca987d46SWarner Losh 
1452ca987d46SWarner Losh 	name = fdt_string(fdtp, fdt32_to_cpu(prop->nameoff));
1453ca987d46SWarner Losh 	len = fdt32_to_cpu(prop->len);
1454ca987d46SWarner Losh 
1455ca987d46SWarner Losh 	rv = 0;
1456ca987d46SWarner Losh 	buf = NULL;
1457ca987d46SWarner Losh 	if (len == 0) {
1458ca987d46SWarner Losh 		/* Property without value */
1459ca987d46SWarner Losh 		line = (char *)malloc(strlen(name) + 2);
1460ca987d46SWarner Losh 		if (line == NULL) {
1461ca987d46SWarner Losh 			rv = 2;
1462ca987d46SWarner Losh 			goto out2;
1463ca987d46SWarner Losh 		}
1464ca987d46SWarner Losh 		sprintf(line, "%s\n", name);
1465ca987d46SWarner Losh 		goto out1;
1466ca987d46SWarner Losh 	}
1467ca987d46SWarner Losh 
1468ca987d46SWarner Losh 	/*
1469ca987d46SWarner Losh 	 * Process property with value
1470ca987d46SWarner Losh 	 */
1471ca987d46SWarner Losh 	data = prop->data;
1472ca987d46SWarner Losh 
1473ca987d46SWarner Losh 	if (fdt_data_fmt(data, len, &buf) != 0) {
1474ca987d46SWarner Losh 		rv = 3;
1475ca987d46SWarner Losh 		goto out2;
1476ca987d46SWarner Losh 	}
1477ca987d46SWarner Losh 
1478ca987d46SWarner Losh 	line = (char *)malloc(strlen(name) + strlen(FDT_PROP_SEP) +
1479ca987d46SWarner Losh 	    strlen(buf) + 2);
1480ca987d46SWarner Losh 	if (line == NULL) {
1481ca987d46SWarner Losh 		sprintf(command_errbuf, "could not allocate space for string");
1482ca987d46SWarner Losh 		rv = 4;
1483ca987d46SWarner Losh 		goto out2;
1484ca987d46SWarner Losh 	}
1485ca987d46SWarner Losh 
1486ca987d46SWarner Losh 	sprintf(line, "%s" FDT_PROP_SEP "%s\n", name, buf);
1487ca987d46SWarner Losh 
1488ca987d46SWarner Losh out1:
1489ca987d46SWarner Losh 	pager_open();
1490ca987d46SWarner Losh 	pager_output(line);
1491ca987d46SWarner Losh 	pager_close();
1492ca987d46SWarner Losh 
1493ca987d46SWarner Losh out2:
1494ca987d46SWarner Losh 	if (buf)
1495ca987d46SWarner Losh 		free(buf);
1496ca987d46SWarner Losh 
1497ca987d46SWarner Losh 	if (line)
1498ca987d46SWarner Losh 		free(line);
1499ca987d46SWarner Losh 
1500ca987d46SWarner Losh 	return (rv);
1501ca987d46SWarner Losh }
1502ca987d46SWarner Losh 
1503ca987d46SWarner Losh static int
1504ca987d46SWarner Losh fdt_modprop(int nodeoff, char *propname, void *value, char mode)
1505ca987d46SWarner Losh {
1506ca987d46SWarner Losh 	uint32_t cells[100];
1507ca987d46SWarner Losh 	const char *buf;
1508ca987d46SWarner Losh 	int len, rv;
1509ca987d46SWarner Losh 	const struct fdt_property *p;
1510ca987d46SWarner Losh 
1511ca987d46SWarner Losh 	p = fdt_get_property(fdtp, nodeoff, propname, NULL);
1512ca987d46SWarner Losh 
1513ca987d46SWarner Losh 	if (p != NULL) {
1514ca987d46SWarner Losh 		if (mode == 1) {
1515ca987d46SWarner Losh 			 /* Adding inexistant value in mode 1 is forbidden */
1516ca987d46SWarner Losh 			sprintf(command_errbuf, "property already exists!");
1517ca987d46SWarner Losh 			return (CMD_ERROR);
1518ca987d46SWarner Losh 		}
1519ca987d46SWarner Losh 	} else if (mode == 0) {
1520ca987d46SWarner Losh 		sprintf(command_errbuf, "property does not exist!");
1521ca987d46SWarner Losh 		return (CMD_ERROR);
1522ca987d46SWarner Losh 	}
1523ca987d46SWarner Losh 	rv = 0;
1524ca987d46SWarner Losh 	buf = value;
1525ca987d46SWarner Losh 
1526ca987d46SWarner Losh 	switch (*buf) {
1527ca987d46SWarner Losh 	case '&':
1528ca987d46SWarner Losh 		/* phandles */
1529ca987d46SWarner Losh 		break;
1530ca987d46SWarner Losh 	case '<':
1531ca987d46SWarner Losh 		/* Data cells */
1532ca987d46SWarner Losh 		len = fdt_strtovect(buf, (void *)&cells, 100,
1533ca987d46SWarner Losh 		    sizeof(uint32_t));
1534ca987d46SWarner Losh 
1535ca987d46SWarner Losh 		rv = fdt_setprop(fdtp, nodeoff, propname, &cells,
1536ca987d46SWarner Losh 		    len * sizeof(uint32_t));
1537ca987d46SWarner Losh 		break;
1538ca987d46SWarner Losh 	case '[':
1539ca987d46SWarner Losh 		/* Data bytes */
1540ca987d46SWarner Losh 		len = fdt_strtovect(buf, (void *)&cells, 100,
1541ca987d46SWarner Losh 		    sizeof(uint8_t));
1542ca987d46SWarner Losh 
1543ca987d46SWarner Losh 		rv = fdt_setprop(fdtp, nodeoff, propname, &cells,
1544ca987d46SWarner Losh 		    len * sizeof(uint8_t));
1545ca987d46SWarner Losh 		break;
1546ca987d46SWarner Losh 	case '"':
1547ca987d46SWarner Losh 	default:
1548ca987d46SWarner Losh 		/* Default -- string */
1549ca987d46SWarner Losh 		rv = fdt_setprop_string(fdtp, nodeoff, propname, value);
1550ca987d46SWarner Losh 		break;
1551ca987d46SWarner Losh 	}
1552ca987d46SWarner Losh 
1553ca987d46SWarner Losh 	if (rv != 0) {
1554ca987d46SWarner Losh 		if (rv == -FDT_ERR_NOSPACE)
1555ca987d46SWarner Losh 			sprintf(command_errbuf,
1556ca987d46SWarner Losh 			    "Device tree blob is too small!\n");
1557ca987d46SWarner Losh 		else
1558ca987d46SWarner Losh 			sprintf(command_errbuf,
1559ca987d46SWarner Losh 			    "Could not add/modify property!\n");
1560ca987d46SWarner Losh 	}
1561ca987d46SWarner Losh 	return (rv);
1562ca987d46SWarner Losh }
1563ca987d46SWarner Losh 
1564ca987d46SWarner Losh /* Merge strings from argv into a single string */
1565ca987d46SWarner Losh static int
1566ca987d46SWarner Losh fdt_merge_strings(int argc, char *argv[], int start, char **buffer)
1567ca987d46SWarner Losh {
1568ca987d46SWarner Losh 	char *buf;
1569ca987d46SWarner Losh 	int i, idx, sz;
1570ca987d46SWarner Losh 
1571ca987d46SWarner Losh 	*buffer = NULL;
1572ca987d46SWarner Losh 	sz = 0;
1573ca987d46SWarner Losh 
1574ca987d46SWarner Losh 	for (i = start; i < argc; i++)
1575ca987d46SWarner Losh 		sz += strlen(argv[i]);
1576ca987d46SWarner Losh 
1577ca987d46SWarner Losh 	/* Additional bytes for whitespaces between args */
1578ca987d46SWarner Losh 	sz += argc - start;
1579ca987d46SWarner Losh 
1580ca987d46SWarner Losh 	buf = (char *)malloc(sizeof(char) * sz);
1581ca987d46SWarner Losh 	if (buf == NULL) {
1582ca987d46SWarner Losh 		sprintf(command_errbuf, "could not allocate space "
1583ca987d46SWarner Losh 		    "for string");
1584ca987d46SWarner Losh 		return (1);
1585ca987d46SWarner Losh 	}
1586ca987d46SWarner Losh 	bzero(buf, sizeof(char) * sz);
1587ca987d46SWarner Losh 
1588ca987d46SWarner Losh 	idx = 0;
1589ca987d46SWarner Losh 	for (i = start, idx = 0; i < argc; i++) {
1590ca987d46SWarner Losh 		strcpy(buf + idx, argv[i]);
1591ca987d46SWarner Losh 		idx += strlen(argv[i]);
1592ca987d46SWarner Losh 		buf[idx] = ' ';
1593ca987d46SWarner Losh 		idx++;
1594ca987d46SWarner Losh 	}
1595ca987d46SWarner Losh 	buf[sz - 1] = '\0';
1596ca987d46SWarner Losh 	*buffer = buf;
1597ca987d46SWarner Losh 	return (0);
1598ca987d46SWarner Losh }
1599ca987d46SWarner Losh 
1600ca987d46SWarner Losh /* Extract offset and name of node/property from a given path */
1601ca987d46SWarner Losh static int
1602ca987d46SWarner Losh fdt_extract_nameloc(char **pathp, char **namep, int *nodeoff)
1603ca987d46SWarner Losh {
1604ca987d46SWarner Losh 	int o;
1605ca987d46SWarner Losh 	char *path = *pathp, *name = NULL, *subpath = NULL;
1606ca987d46SWarner Losh 
1607ca987d46SWarner Losh 	subpath = strrchr(path, '/');
1608ca987d46SWarner Losh 	if (subpath == NULL) {
1609ca987d46SWarner Losh 		o = fdt_path_offset(fdtp, cwd);
1610ca987d46SWarner Losh 		name = path;
1611ca987d46SWarner Losh 		path = (char *)&cwd;
1612ca987d46SWarner Losh 	} else {
1613ca987d46SWarner Losh 		*subpath = '\0';
1614ca987d46SWarner Losh 		if (strlen(path) == 0)
1615ca987d46SWarner Losh 			path = cwd;
1616ca987d46SWarner Losh 
1617ca987d46SWarner Losh 		name = subpath + 1;
1618ca987d46SWarner Losh 		o = fdt_path_offset(fdtp, path);
1619ca987d46SWarner Losh 	}
1620ca987d46SWarner Losh 
1621ca987d46SWarner Losh 	if (strlen(name) == 0) {
1622ca987d46SWarner Losh 		sprintf(command_errbuf, "name not specified");
1623ca987d46SWarner Losh 		return (1);
1624ca987d46SWarner Losh 	}
1625ca987d46SWarner Losh 	if (o < 0) {
1626ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
1627ca987d46SWarner Losh 		    "could not find node: '%s'", path);
1628ca987d46SWarner Losh 		return (1);
1629ca987d46SWarner Losh 	}
1630ca987d46SWarner Losh 	*namep = name;
1631ca987d46SWarner Losh 	*nodeoff = o;
1632ca987d46SWarner Losh 	*pathp = path;
1633ca987d46SWarner Losh 	return (0);
1634ca987d46SWarner Losh }
1635ca987d46SWarner Losh 
1636ca987d46SWarner Losh static int
1637ca987d46SWarner Losh fdt_cmd_prop(int argc, char *argv[])
1638ca987d46SWarner Losh {
1639ca987d46SWarner Losh 	char *path, *propname, *value;
1640ca987d46SWarner Losh 	int o, next, depth, rv;
1641ca987d46SWarner Losh 	uint32_t tag;
1642ca987d46SWarner Losh 
1643ca987d46SWarner Losh 	path = (argc > 2) ? argv[2] : NULL;
1644ca987d46SWarner Losh 
1645ca987d46SWarner Losh 	value = NULL;
1646ca987d46SWarner Losh 
1647ca987d46SWarner Losh 	if (argc > 3) {
1648ca987d46SWarner Losh 		/* Merge property value strings into one */
1649ca987d46SWarner Losh 		if (fdt_merge_strings(argc, argv, 3, &value) != 0)
1650ca987d46SWarner Losh 			return (CMD_ERROR);
1651ca987d46SWarner Losh 	} else
1652ca987d46SWarner Losh 		value = NULL;
1653ca987d46SWarner Losh 
1654ca987d46SWarner Losh 	if (path == NULL)
1655ca987d46SWarner Losh 		path = cwd;
1656ca987d46SWarner Losh 
1657ca987d46SWarner Losh 	rv = CMD_OK;
1658ca987d46SWarner Losh 
1659ca987d46SWarner Losh 	if (value) {
1660ca987d46SWarner Losh 		/* If value is specified -- try to modify prop. */
1661ca987d46SWarner Losh 		if (fdt_extract_nameloc(&path, &propname, &o) != 0)
1662ca987d46SWarner Losh 			return (CMD_ERROR);
1663ca987d46SWarner Losh 
1664ca987d46SWarner Losh 		rv = fdt_modprop(o, propname, value, 0);
1665ca987d46SWarner Losh 		if (rv)
1666ca987d46SWarner Losh 			return (CMD_ERROR);
1667ca987d46SWarner Losh 		return (CMD_OK);
1668ca987d46SWarner Losh 
1669ca987d46SWarner Losh 	}
1670ca987d46SWarner Losh 	/* User wants to display properties */
1671ca987d46SWarner Losh 	o = fdt_path_offset(fdtp, path);
1672ca987d46SWarner Losh 
1673ca987d46SWarner Losh 	if (o < 0) {
1674ca987d46SWarner Losh 		snprintf(command_errbuf, sizeof(command_errbuf),
1675ca987d46SWarner Losh 		    "could not find node: '%s'", path);
1676ca987d46SWarner Losh 		rv = CMD_ERROR;
1677ca987d46SWarner Losh 		goto out;
1678ca987d46SWarner Losh 	}
1679ca987d46SWarner Losh 
1680ca987d46SWarner Losh 	depth = 0;
1681ca987d46SWarner Losh 	while (depth >= 0) {
1682ca987d46SWarner Losh 		tag = fdt_next_tag(fdtp, o, &next);
1683ca987d46SWarner Losh 		switch (tag) {
1684ca987d46SWarner Losh 		case FDT_NOP:
1685ca987d46SWarner Losh 			break;
1686ca987d46SWarner Losh 		case FDT_PROP:
1687ca987d46SWarner Losh 			if (depth > 1)
1688ca987d46SWarner Losh 				/* Don't process properties of nested nodes */
1689ca987d46SWarner Losh 				break;
1690ca987d46SWarner Losh 
1691ca987d46SWarner Losh 			if (fdt_prop(o) != 0) {
1692ca987d46SWarner Losh 				sprintf(command_errbuf, "could not process "
1693ca987d46SWarner Losh 				    "property");
1694ca987d46SWarner Losh 				rv = CMD_ERROR;
1695ca987d46SWarner Losh 				goto out;
1696ca987d46SWarner Losh 			}
1697ca987d46SWarner Losh 			break;
1698ca987d46SWarner Losh 		case FDT_BEGIN_NODE:
1699ca987d46SWarner Losh 			depth++;
1700ca987d46SWarner Losh 			if (depth > FDT_MAX_DEPTH) {
1701ca987d46SWarner Losh 				printf("warning: nesting too deep: %d\n",
1702ca987d46SWarner Losh 				    depth);
1703ca987d46SWarner Losh 				goto out;
1704ca987d46SWarner Losh 			}
1705ca987d46SWarner Losh 			break;
1706ca987d46SWarner Losh 		case FDT_END_NODE:
1707ca987d46SWarner Losh 			depth--;
1708ca987d46SWarner Losh 			if (depth == 0)
1709ca987d46SWarner Losh 				/*
1710ca987d46SWarner Losh 				 * This is the end of our starting node, force
1711ca987d46SWarner Losh 				 * the loop finish.
1712ca987d46SWarner Losh 				 */
1713ca987d46SWarner Losh 				depth--;
1714ca987d46SWarner Losh 			break;
1715ca987d46SWarner Losh 		}
1716ca987d46SWarner Losh 		o = next;
1717ca987d46SWarner Losh 	}
1718ca987d46SWarner Losh out:
1719ca987d46SWarner Losh 	return (rv);
1720ca987d46SWarner Losh }
1721ca987d46SWarner Losh 
1722ca987d46SWarner Losh static int
1723ca987d46SWarner Losh fdt_cmd_mkprop(int argc, char *argv[])
1724ca987d46SWarner Losh {
1725ca987d46SWarner Losh 	int o;
1726ca987d46SWarner Losh 	char *path, *propname, *value;
1727ca987d46SWarner Losh 
1728ca987d46SWarner Losh 	path = (argc > 2) ? argv[2] : NULL;
1729ca987d46SWarner Losh 
1730ca987d46SWarner Losh 	value = NULL;
1731ca987d46SWarner Losh 
1732ca987d46SWarner Losh 	if (argc > 3) {
1733ca987d46SWarner Losh 		/* Merge property value strings into one */
1734ca987d46SWarner Losh 		if (fdt_merge_strings(argc, argv, 3, &value) != 0)
1735ca987d46SWarner Losh 			return (CMD_ERROR);
1736ca987d46SWarner Losh 	} else
1737ca987d46SWarner Losh 		value = NULL;
1738ca987d46SWarner Losh 
1739ca987d46SWarner Losh 	if (fdt_extract_nameloc(&path, &propname, &o) != 0)
1740ca987d46SWarner Losh 		return (CMD_ERROR);
1741ca987d46SWarner Losh 
1742ca987d46SWarner Losh 	if (fdt_modprop(o, propname, value, 1))
1743ca987d46SWarner Losh 		return (CMD_ERROR);
1744ca987d46SWarner Losh 
1745ca987d46SWarner Losh 	return (CMD_OK);
1746ca987d46SWarner Losh }
1747ca987d46SWarner Losh 
1748ca987d46SWarner Losh static int
1749ca987d46SWarner Losh fdt_cmd_rm(int argc, char *argv[])
1750ca987d46SWarner Losh {
1751ca987d46SWarner Losh 	int o, rv;
1752ca987d46SWarner Losh 	char *path = NULL, *propname;
1753ca987d46SWarner Losh 
1754ca987d46SWarner Losh 	if (argc > 2)
1755ca987d46SWarner Losh 		path = argv[2];
1756ca987d46SWarner Losh 	else {
1757ca987d46SWarner Losh 		sprintf(command_errbuf, "no node/property name specified");
1758ca987d46SWarner Losh 		return (CMD_ERROR);
1759ca987d46SWarner Losh 	}
1760ca987d46SWarner Losh 
1761ca987d46SWarner Losh 	o = fdt_path_offset(fdtp, path);
1762ca987d46SWarner Losh 	if (o < 0) {
1763ca987d46SWarner Losh 		/* If node not found -- try to find & delete property */
1764ca987d46SWarner Losh 		if (fdt_extract_nameloc(&path, &propname, &o) != 0)
1765ca987d46SWarner Losh 			return (CMD_ERROR);
1766ca987d46SWarner Losh 
1767ca987d46SWarner Losh 		if ((rv = fdt_delprop(fdtp, o, propname)) != 0) {
1768ca987d46SWarner Losh 			snprintf(command_errbuf, sizeof(command_errbuf),
1769ca987d46SWarner Losh 			    "could not delete %s\n",
1770ca987d46SWarner Losh 			    (rv == -FDT_ERR_NOTFOUND) ?
1771ca987d46SWarner Losh 			    "(property/node does not exist)" : "");
1772ca987d46SWarner Losh 			return (CMD_ERROR);
1773ca987d46SWarner Losh 
1774ca987d46SWarner Losh 		} else
1775ca987d46SWarner Losh 			return (CMD_OK);
1776ca987d46SWarner Losh 	}
1777ca987d46SWarner Losh 	/* If node exists -- remove node */
1778ca987d46SWarner Losh 	rv = fdt_del_node(fdtp, o);
1779ca987d46SWarner Losh 	if (rv) {
1780ca987d46SWarner Losh 		sprintf(command_errbuf, "could not delete node");
1781ca987d46SWarner Losh 		return (CMD_ERROR);
1782ca987d46SWarner Losh 	}
1783ca987d46SWarner Losh 	return (CMD_OK);
1784ca987d46SWarner Losh }
1785ca987d46SWarner Losh 
1786ca987d46SWarner Losh static int
1787ca987d46SWarner Losh fdt_cmd_mknode(int argc, char *argv[])
1788ca987d46SWarner Losh {
1789ca987d46SWarner Losh 	int o, rv;
1790ca987d46SWarner Losh 	char *path = NULL, *nodename = NULL;
1791ca987d46SWarner Losh 
1792ca987d46SWarner Losh 	if (argc > 2)
1793ca987d46SWarner Losh 		path = argv[2];
1794ca987d46SWarner Losh 	else {
1795ca987d46SWarner Losh 		sprintf(command_errbuf, "no node name specified");
1796ca987d46SWarner Losh 		return (CMD_ERROR);
1797ca987d46SWarner Losh 	}
1798ca987d46SWarner Losh 
1799ca987d46SWarner Losh 	if (fdt_extract_nameloc(&path, &nodename, &o) != 0)
1800ca987d46SWarner Losh 		return (CMD_ERROR);
1801ca987d46SWarner Losh 
1802ca987d46SWarner Losh 	rv = fdt_add_subnode(fdtp, o, nodename);
1803ca987d46SWarner Losh 
1804ca987d46SWarner Losh 	if (rv < 0) {
1805ca987d46SWarner Losh 		if (rv == -FDT_ERR_NOSPACE)
1806ca987d46SWarner Losh 			sprintf(command_errbuf,
1807ca987d46SWarner Losh 			    "Device tree blob is too small!\n");
1808ca987d46SWarner Losh 		else
1809ca987d46SWarner Losh 			sprintf(command_errbuf,
1810ca987d46SWarner Losh 			    "Could not add node!\n");
1811ca987d46SWarner Losh 		return (CMD_ERROR);
1812ca987d46SWarner Losh 	}
1813ca987d46SWarner Losh 	return (CMD_OK);
1814ca987d46SWarner Losh }
1815ca987d46SWarner Losh 
1816ca987d46SWarner Losh static int
1817ca987d46SWarner Losh fdt_cmd_pwd(int argc, char *argv[])
1818ca987d46SWarner Losh {
1819ca987d46SWarner Losh 	char line[FDT_CWD_LEN];
1820ca987d46SWarner Losh 
1821ca987d46SWarner Losh 	pager_open();
1822ca987d46SWarner Losh 	sprintf(line, "%s\n", cwd);
1823ca987d46SWarner Losh 	pager_output(line);
1824ca987d46SWarner Losh 	pager_close();
1825ca987d46SWarner Losh 	return (CMD_OK);
1826ca987d46SWarner Losh }
1827ca987d46SWarner Losh 
1828ca987d46SWarner Losh static int
1829ca987d46SWarner Losh fdt_cmd_mres(int argc, char *argv[])
1830ca987d46SWarner Losh {
1831ca987d46SWarner Losh 	uint64_t start, size;
1832ca987d46SWarner Losh 	int i, total;
1833ca987d46SWarner Losh 	char line[80];
1834ca987d46SWarner Losh 
1835ca987d46SWarner Losh 	pager_open();
1836ca987d46SWarner Losh 	total = fdt_num_mem_rsv(fdtp);
1837ca987d46SWarner Losh 	if (total > 0) {
1838ca987d46SWarner Losh 		if (pager_output("Reserved memory regions:\n"))
1839ca987d46SWarner Losh 			goto out;
1840ca987d46SWarner Losh 		for (i = 0; i < total; i++) {
1841ca987d46SWarner Losh 			fdt_get_mem_rsv(fdtp, i, &start, &size);
1842ca987d46SWarner Losh 			sprintf(line, "reg#%d: (start: 0x%jx, size: 0x%jx)\n",
1843ca987d46SWarner Losh 			    i, start, size);
1844ca987d46SWarner Losh 			if (pager_output(line))
1845ca987d46SWarner Losh 				goto out;
1846ca987d46SWarner Losh 		}
1847ca987d46SWarner Losh 	} else
1848ca987d46SWarner Losh 		pager_output("No reserved memory regions\n");
1849ca987d46SWarner Losh out:
1850ca987d46SWarner Losh 	pager_close();
1851ca987d46SWarner Losh 
1852ca987d46SWarner Losh 	return (CMD_OK);
1853ca987d46SWarner Losh }
1854ca987d46SWarner Losh 
1855ca987d46SWarner Losh static int
1856ca987d46SWarner Losh fdt_cmd_nyi(int argc, char *argv[])
1857ca987d46SWarner Losh {
1858ca987d46SWarner Losh 
1859ca987d46SWarner Losh 	printf("command not yet implemented\n");
1860ca987d46SWarner Losh 	return (CMD_ERROR);
1861ca987d46SWarner Losh }
1862