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