1 // SPDX-License-Identifier: LGPL-2.1-or-later
2 /*
3  * libfdt - Flat Device Tree manipulation
4  *	Tests if two given dtbs are structurally equal (including order)
5  * Copyright (C) 2007 David Gibson, IBM Corporation.
6  */
7 
8 #include <stdlib.h>
9 #include <stdio.h>
10 #include <string.h>
11 #include <stdint.h>
12 
13 #include <libfdt.h>
14 
15 #include "tests.h"
16 #include "testdata.h"
17 
18 static int notequal; /* = 0 */
19 
20 #define MISMATCH(fmt, ...)			\
21 	do { \
22 		if (notequal) \
23 			PASS(); \
24 		else \
25 			FAIL(fmt, ##__VA_ARGS__);	\
26 	} while (0)
27 
28 #define MATCH()			\
29 	do { \
30 		if (!notequal) \
31 			PASS(); \
32 		else \
33 			FAIL("Trees match which shouldn't");	\
34 	} while (0)
35 
36 #define CHECK(code) \
37 	{ \
38 		err = (code); \
39 		if (err) \
40 			FAIL(#code ": %s", fdt_strerror(err)); \
41 	}
42 
compare_mem_rsv(const void * fdt1,const void * fdt2)43 static void compare_mem_rsv(const void *fdt1, const void *fdt2)
44 {
45 	int i;
46 	uint64_t addr1, size1, addr2, size2;
47 	int err;
48 
49 	if (fdt_num_mem_rsv(fdt1) != fdt_num_mem_rsv(fdt2))
50 		MISMATCH("Trees have different number of reserve entries");
51 	for (i = 0; i < fdt_num_mem_rsv(fdt1); i++) {
52 		CHECK(fdt_get_mem_rsv(fdt1, i, &addr1, &size1));
53 		CHECK(fdt_get_mem_rsv(fdt2, i, &addr2, &size2));
54 
55 		if ((addr1 != addr2) || (size1 != size2))
56 			MISMATCH("Mismatch in reserve entry %d: "
57 				 "(0x%llx, 0x%llx) != (0x%llx, 0x%llx)", i,
58 				 (unsigned long long)addr1,
59 				 (unsigned long long)size1,
60 				 (unsigned long long)addr2,
61 				 (unsigned long long)size2);
62 	}
63 }
64 
compare_structure(const void * fdt1,const void * fdt2)65 static void compare_structure(const void *fdt1, const void *fdt2)
66 {
67 	int nextoffset1 = 0, nextoffset2 = 0;
68 	int offset1, offset2;
69 	uint32_t tag1, tag2;
70 	const char *name1, *name2;
71 	int err;
72 	const struct fdt_property *prop1, *prop2;
73 	int len1, len2;
74 
75 	while (1) {
76 		do {
77 			offset1 = nextoffset1;
78 			tag1 = fdt_next_tag(fdt1, offset1, &nextoffset1);
79 		} while (tag1 == FDT_NOP);
80 		do {
81 			offset2 = nextoffset2;
82 			tag2 = fdt_next_tag(fdt2, offset2, &nextoffset2);
83 		} while (tag2 == FDT_NOP);
84 
85 		if (tag1 != tag2)
86 			MISMATCH("Tag mismatch (%d != %d) at (%d, %d)",
87 			     tag1, tag2, offset1, offset2);
88 
89 		switch (tag1) {
90 		case FDT_BEGIN_NODE:
91 			name1 = fdt_get_name(fdt1, offset1, &err);
92 			if (!name1)
93 				FAIL("fdt_get_name(fdt1, %d, ..): %s",
94 				     offset1, fdt_strerror(err));
95 			name2 = fdt_get_name(fdt2, offset2, NULL);
96 			if (!name2)
97 				FAIL("fdt_get_name(fdt2, %d, ..): %s",
98 				     offset2, fdt_strerror(err));
99 
100 			if (!streq(name1, name2))
101 			    MISMATCH("Name mismatch (\"%s\" != \"%s\") at (%d, %d)",
102 				     name1, name2, offset1, offset2);
103 			break;
104 
105 		case FDT_PROP:
106 			prop1 = fdt_offset_ptr(fdt1, offset1, sizeof(*prop1));
107 			if (!prop1)
108 				FAIL("Could get fdt1 property at %d", offset1);
109 			prop2 = fdt_offset_ptr(fdt2, offset2, sizeof(*prop2));
110 			if (!prop2)
111 				FAIL("Could get fdt2 property at %d", offset2);
112 
113 			name1 = fdt_string(fdt1, fdt32_to_cpu(prop1->nameoff));
114 			name2 = fdt_string(fdt2, fdt32_to_cpu(prop2->nameoff));
115 			if (!streq(name1, name2))
116 				MISMATCH("Property name mismatch \"%s\" != \"%s\" "
117 					 "at (%d, %d)", name1, name2, offset1, offset2);
118 			len1 = fdt32_to_cpu(prop1->len);
119 			len2 = fdt32_to_cpu(prop2->len);
120 			if (len1 != len2)
121 				MISMATCH("Property length mismatch %u != %u "
122 					 "at (%d, %d)", len1, len2, offset1, offset2);
123 
124 			if (memcmp(prop1->data, prop2->data, len1) != 0)
125 				MISMATCH("Property value mismatch at (%d, %d)",
126 					 offset1, offset2);
127 			break;
128 
129 		case FDT_END:
130 			return;
131 		}
132 	}
133 }
134 
main(int argc,char * argv[])135 int main(int argc, char *argv[])
136 {
137 	void *fdt1, *fdt2;
138 	uint32_t cpuid1, cpuid2;
139 
140 	test_init(argc, argv);
141 	if ((argc != 3)
142 	    && ((argc != 4) || !streq(argv[1], "-n")))
143 		CONFIG("Usage: %s [-n] <dtb file> <dtb file>", argv[0]);
144 	if (argc == 4)
145 		notequal = 1;
146 
147 	fdt1 = load_blob(argv[argc-2]);
148 	fdt2 = load_blob(argv[argc-1]);
149 
150 	compare_mem_rsv(fdt1, fdt2);
151 	compare_structure(fdt1, fdt2);
152 
153 	cpuid1 = fdt_boot_cpuid_phys(fdt1);
154 	cpuid2 = fdt_boot_cpuid_phys(fdt2);
155 	if (cpuid1 != cpuid2)
156 		MISMATCH("boot_cpuid_phys mismatch 0x%x != 0x%x",
157 			 cpuid1, cpuid2);
158 
159 	MATCH();
160 }
161