1 // SPDX-License-Identifier: LGPL-2.1-or-later
2 /*
3 * libfdt - Flat Device Tree manipulation
4 * Testcase/tool for rearranging blocks of a dtb
5 * Copyright (C) 2006 David Gibson, IBM Corporation.
6 */
7
8 #include <stdlib.h>
9 #include <stdio.h>
10 #include <string.h>
11 #include <limits.h>
12 #include <stdint.h>
13
14 #include <libfdt.h>
15
16 #include "tests.h"
17 #include "testdata.h"
18
19 struct bufstate {
20 char *buf;
21 int size;
22 };
23
expand_buf(struct bufstate * buf,int newsize)24 static void expand_buf(struct bufstate *buf, int newsize)
25 {
26 buf->buf = realloc(buf->buf, newsize);
27 if (!buf->buf)
28 CONFIG("Allocation failure");
29 buf->size = newsize;
30 }
31
new_header(struct bufstate * buf,int version,const void * fdt)32 static void new_header(struct bufstate *buf, int version, const void *fdt)
33 {
34 int hdrsize = fdt_header_size_(version);
35
36 if ((version != 16) && (version != 17))
37 CONFIG("Bad version %d", version);
38
39 expand_buf(buf, hdrsize);
40 memset(buf->buf, 0, hdrsize);
41
42 fdt_set_magic(buf->buf, FDT_MAGIC);
43 fdt_set_version(buf->buf, version);
44 fdt_set_last_comp_version(buf->buf, 16);
45 fdt_set_boot_cpuid_phys(buf->buf, fdt_boot_cpuid_phys(fdt));
46 }
47
add_block(struct bufstate * buf,int version,char block,const void * fdt)48 static void add_block(struct bufstate *buf, int version, char block, const void *fdt)
49 {
50 int align, size, oldsize;
51 const void *src;
52 int offset;
53
54 switch (block) {
55 case 'm':
56 /* Memory reserve map */
57 align = 8;
58 src = (const char *)fdt + fdt_off_mem_rsvmap(fdt);
59 size = (fdt_num_mem_rsv(fdt) + 1)
60 * sizeof(struct fdt_reserve_entry);
61 break;
62
63 case 't':
64 /* Structure block */
65 align = 4;
66 src = (const char *)fdt + fdt_off_dt_struct(fdt);
67 size = fdt_size_dt_struct(fdt);
68 break;
69
70 case 's':
71 /* Strings block */
72 align = 1;
73 src = (const char *)fdt + fdt_off_dt_strings(fdt);
74 size = fdt_size_dt_strings(fdt);
75 break;
76 default:
77 CONFIG("Bad block '%c'", block);
78 }
79
80 oldsize = buf->size;
81 offset = ALIGN(oldsize, align);
82 expand_buf(buf, offset+size);
83 memset(buf->buf + oldsize, 0, offset - oldsize);
84
85 memcpy(buf->buf + offset, src, size);
86
87 switch (block) {
88 case 'm':
89 fdt_set_off_mem_rsvmap(buf->buf, offset);
90 break;
91
92 case 't':
93 fdt_set_off_dt_struct(buf->buf, offset);
94 if (version >= 17)
95 fdt_set_size_dt_struct(buf->buf, size);
96 break;
97
98 case 's':
99 fdt_set_off_dt_strings(buf->buf, offset);
100 fdt_set_size_dt_strings(buf->buf, size);
101 break;
102 }
103 }
104
main(int argc,char * argv[])105 int main(int argc, char *argv[])
106 {
107 void *fdt;
108 int version;
109 const char *blockorder;
110 struct bufstate buf = {NULL, 0};
111 int err;
112 const char *inname;
113 char outname[PATH_MAX];
114
115 test_init(argc, argv);
116 if (argc != 4)
117 CONFIG("Usage: %s <dtb file> <version> <block order>", argv[0]);
118
119 inname = argv[1];
120 fdt = load_blob(argv[1]);
121 version = atoi(argv[2]);
122 blockorder = argv[3];
123 sprintf(outname, "v%d.%s.%s", version, blockorder, inname);
124
125 if ((version != 16) && (version != 17))
126 CONFIG("Version must be 16 or 17");
127
128 if (fdt_version(fdt) < 17)
129 CONFIG("Input tree must be v17");
130
131 new_header(&buf, version, fdt);
132
133 while (*blockorder) {
134 add_block(&buf, version, *blockorder, fdt);
135 blockorder++;
136 }
137
138 fdt_set_totalsize(buf.buf, buf.size);
139
140 err = fdt_check_header(buf.buf);
141 if (err)
142 FAIL("Output tree fails check: %s", fdt_strerror(err));
143
144 save_blob(outname, buf.buf);
145
146 PASS();
147 }
148