1 // SPDX-License-Identifier: LGPL-2.1-or-later
2 /*
3 * libfdt - Flat Device Tree manipulation
4 * Testcase for fdt_nop_node()
5 * Copyright (C) 2006 David Gibson, IBM Corporation.
6 */
7
8 #include <stdlib.h>
9 #include <stdio.h>
10 #include <string.h>
11 #include <ctype.h>
12 #include <stdint.h>
13
14 #include <libfdt.h>
15
16 #include "tests.h"
17 #include "testdata.h"
18
19 #define SPACE 65536
20
21 static enum {
22 FIXED = 0,
23 RESIZE,
24 REALLOC,
25 NEWALLOC,
26 } alloc_mode;
27
realloc_fdt(void ** fdt,size_t * size,bool created)28 static void realloc_fdt(void **fdt, size_t *size, bool created)
29 {
30 int err;
31
32 switch (alloc_mode) {
33 case FIXED:
34 if (!(*fdt))
35 *fdt = xmalloc(*size);
36 else
37 FAIL("Ran out of space");
38 return;
39
40 case RESIZE:
41 if (!(*fdt)) {
42 *fdt = xmalloc(SPACE);
43 } else if (*size < SPACE) {
44 *size += 1;
45 err = fdt_resize(*fdt, *fdt, *size);
46 if (err < 0)
47 FAIL("fdt_resize() failed: %s",
48 fdt_strerror(err));
49 } else {
50 FAIL("Ran out of space");
51 }
52 return;
53
54 case REALLOC:
55 *size += 1;
56 *fdt = xrealloc(*fdt, *size);
57 if (created) {
58 err = fdt_resize(*fdt, *fdt, *size);
59 if (err < 0)
60 FAIL("fdt_resize() failed: %s",
61 fdt_strerror(err));
62 }
63 return;
64
65 case NEWALLOC: {
66 void *buf;
67
68 *size += 1;
69 buf = xmalloc(*size);
70 if (created) {
71 err = fdt_resize(*fdt, buf, *size);
72 if (err < 0)
73 FAIL("fdt_resize() failed: %s",
74 fdt_strerror(err));
75 }
76 free(*fdt);
77 *fdt = buf;
78 return;
79 }
80
81 default:
82 CONFIG("Bad allocation mode");
83 }
84 }
85
86 #define CHECK(code) \
87 do { \
88 err = (code); \
89 if (err == -FDT_ERR_NOSPACE) \
90 realloc_fdt(&fdt, &size, created); \
91 else if (err) \
92 FAIL(#code ": %s", fdt_strerror(err)); \
93 } while (err != 0)
94
main(int argc,char * argv[])95 int main(int argc, char *argv[])
96 {
97 void *fdt = NULL;
98 size_t size;
99 int err;
100 bool created = false;
101 void *place;
102 const char place_str[] = "this is a placeholder string\0string2";
103 int place_len = sizeof(place_str);
104 int create_flags;
105
106 test_init(argc, argv);
107
108 alloc_mode = FIXED;
109 size = SPACE;
110 create_flags = 0;
111
112 if (argc == 2 || argc == 3) {
113 if (streq(argv[1], "fixed")) {
114 alloc_mode = FIXED;
115 size = SPACE;
116 } else if (streq(argv[1], "resize")) {
117 alloc_mode = REALLOC;
118 size = 0;
119 } else if (streq(argv[1], "realloc")) {
120 alloc_mode = REALLOC;
121 size = 0;
122 } else if (streq(argv[1], "newalloc")) {
123 alloc_mode = NEWALLOC;
124 size = 0;
125 } else {
126 char *endp;
127
128 size = strtoul(argv[1], &endp, 0);
129 if (*endp == '\0')
130 alloc_mode = FIXED;
131 else
132 CONFIG("Bad allocation mode \"%s\" specified",
133 argv[1]);
134 }
135 }
136 if (argc == 3) {
137 char *str = argv[2], *saveptr, *tok;
138 bool default_flag = false;
139
140 while ((tok = strtok_r(str, ",", &saveptr)) != NULL) {
141 str = NULL;
142 if (streq(tok, "default")) {
143 default_flag = true;
144 } else if (streq(tok, "no_name_dedup")) {
145 create_flags |= FDT_CREATE_FLAG_NO_NAME_DEDUP;
146 } else if (streq(tok, "bad")) {
147 create_flags |= 0xffffffff;
148 } else {
149 CONFIG("Bad creation flags \"%s\" specified",
150 argv[2]);
151 }
152 }
153
154 if (default_flag && create_flags != 0)
155 CONFIG("Bad creation flags \"%s\" specified",
156 argv[2]);
157 }
158
159 if (argc > 3) {
160 CONFIG("sw_tree1 [<allocation mode>] [<create flags>]");
161 }
162
163 fdt = xmalloc(size);
164 CHECK(fdt_create_with_flags(fdt, size, create_flags));
165
166 created = true;
167
168 CHECK(fdt_add_reservemap_entry(fdt, TEST_ADDR_1, TEST_SIZE_1));
169
170 CHECK(fdt_add_reservemap_entry(fdt, TEST_ADDR_2, TEST_SIZE_2));
171 CHECK(fdt_finish_reservemap(fdt));
172
173 CHECK(fdt_begin_node(fdt, ""));
174 CHECK(fdt_property_string(fdt, "compatible", "test_tree1"));
175 CHECK(fdt_property_u32(fdt, "prop-int", TEST_VALUE_1));
176 CHECK(fdt_property_u64(fdt, "prop-int64", TEST_VALUE64_1));
177 CHECK(fdt_property_string(fdt, "prop-str", TEST_STRING_1));
178 CHECK(fdt_property_u32(fdt, "#address-cells", 1));
179 CHECK(fdt_property_u32(fdt, "#size-cells", 0));
180
181 CHECK(fdt_begin_node(fdt, "subnode@1"));
182 CHECK(fdt_property_string(fdt, "compatible", "subnode1"));
183 CHECK(fdt_property_u32(fdt, "reg", 1));
184 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_1));
185 CHECK(fdt_begin_node(fdt, "subsubnode"));
186 CHECK(fdt_property(fdt, "compatible", "subsubnode1\0subsubnode",
187 23));
188 CHECK(fdt_property_placeholder(fdt, "placeholder", place_len, &place));
189 memcpy(place, place_str, place_len);
190 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_1));
191 CHECK(fdt_end_node(fdt));
192 CHECK(fdt_begin_node(fdt, "ss1"));
193 CHECK(fdt_end_node(fdt));
194 CHECK(fdt_end_node(fdt));
195
196 CHECK(fdt_begin_node(fdt, "subnode@2"));
197 CHECK(fdt_property_u32(fdt, "reg", 2));
198 CHECK(fdt_property_cell(fdt, "linux,phandle", PHANDLE_1));
199 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_2));
200 CHECK(fdt_property_u32(fdt, "#address-cells", 1));
201 CHECK(fdt_property_u32(fdt, "#size-cells", 0));
202 CHECK(fdt_begin_node(fdt, "subsubnode@0"));
203 CHECK(fdt_property_u32(fdt, "reg", 0));
204 CHECK(fdt_property_cell(fdt, "phandle", PHANDLE_2));
205 CHECK(fdt_property(fdt, "compatible", "subsubnode2\0subsubnode",
206 23));
207 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_2));
208 CHECK(fdt_end_node(fdt));
209 CHECK(fdt_begin_node(fdt, "ss2"));
210 CHECK(fdt_end_node(fdt));
211
212 CHECK(fdt_end_node(fdt));
213
214 CHECK(fdt_end_node(fdt));
215
216 save_blob("unfinished_tree1.test.dtb", fdt);
217
218 CHECK(fdt_finish(fdt));
219
220 verbose_printf("Completed tree, totalsize = %d\n",
221 fdt_totalsize(fdt));
222
223 save_blob("sw_tree1.test.dtb", fdt);
224
225 PASS();
226 }
227