1 /*
2 * libfdt - Flat Device Tree manipulation
3 * Testcase/tool for rearranging blocks of a dtb
4 * Copyright (C) 2006 David Gibson, IBM Corporation.
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public License
8 * as published by the Free Software Foundation; either version 2.1 of
9 * the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <string.h>
24 #include <limits.h>
25 #include <stdint.h>
26
27 #include <libfdt.h>
28
29 #include "tests.h"
30 #include "testdata.h"
31
32 struct bufstate {
33 char *buf;
34 int size;
35 };
36
expand_buf(struct bufstate * buf,int newsize)37 static void expand_buf(struct bufstate *buf, int newsize)
38 {
39 buf->buf = realloc(buf->buf, newsize);
40 if (!buf->buf)
41 CONFIG("Allocation failure");
42 buf->size = newsize;
43 }
44
new_header(struct bufstate * buf,int version,const void * fdt)45 static void new_header(struct bufstate *buf, int version, const void *fdt)
46 {
47 int hdrsize;
48
49 if (version == 16)
50 hdrsize = FDT_V16_SIZE;
51 else if (version == 17)
52 hdrsize = FDT_V17_SIZE;
53 else
54 CONFIG("Bad version %d", version);
55
56 expand_buf(buf, hdrsize);
57 memset(buf->buf, 0, hdrsize);
58
59 fdt_set_magic(buf->buf, FDT_MAGIC);
60 fdt_set_version(buf->buf, version);
61 fdt_set_last_comp_version(buf->buf, 16);
62 fdt_set_boot_cpuid_phys(buf->buf, fdt_boot_cpuid_phys(fdt));
63 }
64
add_block(struct bufstate * buf,int version,char block,const void * fdt)65 static void add_block(struct bufstate *buf, int version, char block, const void *fdt)
66 {
67 int align, size, oldsize;
68 const void *src;
69 int offset;
70
71 switch (block) {
72 case 'm':
73 /* Memory reserve map */
74 align = 8;
75 src = (const char *)fdt + fdt_off_mem_rsvmap(fdt);
76 size = (fdt_num_mem_rsv(fdt) + 1)
77 * sizeof(struct fdt_reserve_entry);
78 break;
79
80 case 't':
81 /* Structure block */
82 align = 4;
83 src = (const char *)fdt + fdt_off_dt_struct(fdt);
84 size = fdt_size_dt_struct(fdt);
85 break;
86
87 case 's':
88 /* Strings block */
89 align = 1;
90 src = (const char *)fdt + fdt_off_dt_strings(fdt);
91 size = fdt_size_dt_strings(fdt);
92 break;
93 default:
94 CONFIG("Bad block '%c'", block);
95 }
96
97 oldsize = buf->size;
98 offset = ALIGN(oldsize, align);
99 expand_buf(buf, offset+size);
100 memset(buf->buf + oldsize, 0, offset - oldsize);
101
102 memcpy(buf->buf + offset, src, size);
103
104 switch (block) {
105 case 'm':
106 fdt_set_off_mem_rsvmap(buf->buf, offset);
107 break;
108
109 case 't':
110 fdt_set_off_dt_struct(buf->buf, offset);
111 if (version >= 17)
112 fdt_set_size_dt_struct(buf->buf, size);
113 break;
114
115 case 's':
116 fdt_set_off_dt_strings(buf->buf, offset);
117 fdt_set_size_dt_strings(buf->buf, size);
118 break;
119 }
120 }
121
main(int argc,char * argv[])122 int main(int argc, char *argv[])
123 {
124 void *fdt;
125 int version;
126 const char *blockorder;
127 struct bufstate buf = {NULL, 0};
128 int err;
129 const char *inname;
130 char outname[PATH_MAX];
131
132 test_init(argc, argv);
133 if (argc != 4)
134 CONFIG("Usage: %s <dtb file> <version> <block order>", argv[0]);
135
136 inname = argv[1];
137 fdt = load_blob(argv[1]);
138 version = atoi(argv[2]);
139 blockorder = argv[3];
140 sprintf(outname, "v%d.%s.%s", version, blockorder, inname);
141
142 if ((version != 16) && (version != 17))
143 CONFIG("Version must be 16 or 17");
144
145 if (fdt_version(fdt) < 17)
146 CONFIG("Input tree must be v17");
147
148 new_header(&buf, version, fdt);
149
150 while (*blockorder) {
151 add_block(&buf, version, *blockorder, fdt);
152 blockorder++;
153 }
154
155 fdt_set_totalsize(buf.buf, buf.size);
156
157 err = fdt_check_header(buf.buf);
158 if (err)
159 FAIL("Output tree fails check: %s", fdt_strerror(err));
160
161 save_blob(outname, buf.buf);
162
163 PASS();
164 }
165