1 // SPDX-License-Identifier: LGPL-2.1-or-later
2 /*
3 * libfdt - Flat Device Tree manipulation
4 * Testcase for error handling with sequential write states
5 * Copyright (C) 2018 David Gibson, Red Hat Inc.
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 #define CHECK_OK(code) \
22 do { \
23 verbose_printf(" OK: %s\n", #code); \
24 err = (code); \
25 if (err) \
26 FAIL(#code ": %s", fdt_strerror(err)); \
27 } while (0)
28
29 #define CHECK_BADSTATE(code) \
30 do { \
31 verbose_printf("BAD: %s\n", #code); \
32 err = (code); \
33 if (err == 0) \
34 FAIL(#code ": succeeded in bad state"); \
35 else if (err != -FDT_ERR_BADSTATE) \
36 FAIL(#code ": %s", fdt_strerror(err)); \
37 } while (0)
38
main(int argc,char * argv[])39 int main(int argc, char *argv[])
40 {
41 void *fdt = NULL;
42 int err;
43
44 test_init(argc, argv);
45
46 fdt = xmalloc(SPACE);
47
48 err = fdt_create(fdt, SPACE);
49 if (err)
50 FAIL("fdt_create(): %s", fdt_strerror(err));
51
52 /* Memory reserve state */
53
54 CHECK_BADSTATE(fdt_begin_node(fdt, ""));
55 CHECK_BADSTATE(fdt_property_string(fdt, "bad-str", "TEST_STRING_1"));
56 CHECK_BADSTATE(fdt_end_node(fdt));
57 CHECK_BADSTATE(fdt_finish(fdt));
58
59 CHECK_OK(fdt_add_reservemap_entry(fdt, TEST_ADDR_1, TEST_SIZE_1));
60 CHECK_OK(fdt_add_reservemap_entry(fdt, TEST_ADDR_2, TEST_SIZE_2));
61
62 CHECK_BADSTATE(fdt_begin_node(fdt, ""));
63 CHECK_BADSTATE(fdt_property_string(fdt, "bad-str", "TEST_STRING_1"));
64 CHECK_BADSTATE(fdt_end_node(fdt));
65 CHECK_BADSTATE(fdt_finish(fdt));
66
67 CHECK_OK(fdt_finish_reservemap(fdt));
68
69 /* Structure state */
70
71 CHECK_BADSTATE(fdt_add_reservemap_entry(fdt, TEST_ADDR_1, TEST_SIZE_1));
72 CHECK_BADSTATE(fdt_finish_reservemap(fdt));
73
74 CHECK_OK(fdt_begin_node(fdt, ""));
75 CHECK_OK(fdt_property_string(fdt, "compatible", "test_tree1"));
76 CHECK_OK(fdt_property_u32(fdt, "prop-int", TEST_VALUE_1));
77 CHECK_OK(fdt_property_u64(fdt, "prop-int64", TEST_VALUE64_1));
78 CHECK_OK(fdt_property_string(fdt, "prop-str", TEST_STRING_1));
79 CHECK_OK(fdt_property_u32(fdt, "#address-cells", 1));
80 CHECK_OK(fdt_property_u32(fdt, "#size-cells", 0));
81
82 CHECK_OK(fdt_begin_node(fdt, "subnode@1"));
83 CHECK_OK(fdt_property_string(fdt, "compatible", "subnode1"));
84 CHECK_OK(fdt_property_u32(fdt, "reg", 1));
85 CHECK_OK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_1));
86 CHECK_OK(fdt_begin_node(fdt, "subsubnode"));
87 CHECK_OK(fdt_property(fdt, "compatible", "subsubnode1\0subsubnode",
88 23));
89 CHECK_OK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_1));
90 CHECK_OK(fdt_end_node(fdt));
91 CHECK_OK(fdt_begin_node(fdt, "ss1"));
92 CHECK_OK(fdt_end_node(fdt));
93 CHECK_OK(fdt_end_node(fdt));
94
95 CHECK_OK(fdt_begin_node(fdt, "subnode@2"));
96 CHECK_OK(fdt_property_u32(fdt, "reg", 2));
97 CHECK_OK(fdt_property_cell(fdt, "linux,phandle", PHANDLE_1));
98 CHECK_OK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_2));
99 CHECK_OK(fdt_property_u32(fdt, "#address-cells", 1));
100 CHECK_OK(fdt_property_u32(fdt, "#size-cells", 0));
101 CHECK_OK(fdt_begin_node(fdt, "subsubnode@0"));
102 CHECK_OK(fdt_property_u32(fdt, "reg", 0));
103 CHECK_OK(fdt_property_cell(fdt, "phandle", PHANDLE_2));
104 CHECK_OK(fdt_property(fdt, "compatible", "subsubnode2\0subsubnode",
105 23));
106 CHECK_OK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_2));
107 CHECK_OK(fdt_end_node(fdt));
108 CHECK_OK(fdt_begin_node(fdt, "ss2"));
109 CHECK_OK(fdt_end_node(fdt));
110
111 CHECK_OK(fdt_end_node(fdt));
112
113 CHECK_OK(fdt_end_node(fdt));
114
115 CHECK_OK(fdt_finish(fdt));
116
117 /* Completed state */
118
119 CHECK_BADSTATE(fdt_add_reservemap_entry(fdt, TEST_ADDR_1, TEST_SIZE_1));
120 CHECK_BADSTATE(fdt_finish_reservemap(fdt));
121 CHECK_BADSTATE(fdt_begin_node(fdt, ""));
122 CHECK_BADSTATE(fdt_property_string(fdt, "bad-str", "TEST_STRING_1"));
123 CHECK_BADSTATE(fdt_end_node(fdt));
124 CHECK_BADSTATE(fdt_finish(fdt));
125
126 PASS();
127 }
128