1/* 2 * libfdt - Flat Device Tree manipulation 3 * Testcase for fdt_nop_node() 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 <ctype.h> 25#include <stdint.h> 26 27#include <libfdt.h> 28 29#include "tests.h" 30#include "testdata.h" 31 32#define SPACE 65536 33 34static enum { 35 FIXED = 0, 36 RESIZE, 37 REALLOC, 38} alloc_mode; 39 40static void realloc_fdt(void **fdt, size_t *size, bool created) 41{ 42 switch (alloc_mode) { 43 case FIXED: 44 if (!(*fdt)) 45 fdt = xmalloc(*size); 46 else 47 FAIL("Ran out of space"); 48 return; 49 50 case RESIZE: 51 if (!(*fdt)) { 52 fdt = xmalloc(SPACE); 53 } else if (*size < SPACE) { 54 *size += 1; 55 fdt_resize(*fdt, *fdt, *size); 56 } else { 57 FAIL("Ran out of space"); 58 } 59 return; 60 61 case REALLOC: 62 *size += 1; 63 *fdt = xrealloc(*fdt, *size); 64 if (created) 65 fdt_resize(*fdt, *fdt, *size); 66 return; 67 68 default: 69 CONFIG("Bad allocation mode"); 70 } 71} 72 73#define CHECK(code) \ 74 do { \ 75 err = (code); \ 76 if (err == -FDT_ERR_NOSPACE) \ 77 realloc_fdt(&fdt, &size, created); \ 78 else if (err) \ 79 FAIL(#code ": %s", fdt_strerror(err)); \ 80 } while (err != 0) 81 82int main(int argc, char *argv[]) 83{ 84 void *fdt = NULL; 85 size_t size; 86 int err; 87 bool created = false; 88 89 test_init(argc, argv); 90 91 if (argc == 1) { 92 alloc_mode = FIXED; 93 size = SPACE; 94 } else if (argc == 2) { 95 if (streq(argv[1], "resize")) { 96 alloc_mode = REALLOC; 97 size = 0; 98 } else if (streq(argv[1], "realloc")) { 99 alloc_mode = REALLOC; 100 size = 0; 101 } else { 102 char *endp; 103 104 size = strtoul(argv[1], &endp, 0); 105 if (*endp == '\0') 106 alloc_mode = FIXED; 107 else 108 CONFIG("Bad allocation mode \"%s\" specified", 109 argv[1]); 110 } 111 } 112 113 fdt = xmalloc(size); 114 CHECK(fdt_create(fdt, size)); 115 116 created = true; 117 118 CHECK(fdt_add_reservemap_entry(fdt, TEST_ADDR_1, TEST_SIZE_1)); 119 120 CHECK(fdt_add_reservemap_entry(fdt, TEST_ADDR_2, TEST_SIZE_2)); 121 CHECK(fdt_finish_reservemap(fdt)); 122 123 CHECK(fdt_begin_node(fdt, "")); 124 CHECK(fdt_property_string(fdt, "compatible", "test_tree1")); 125 CHECK(fdt_property_u32(fdt, "prop-int", TEST_VALUE_1)); 126 CHECK(fdt_property_u64(fdt, "prop-int64", TEST_VALUE64_1)); 127 CHECK(fdt_property_string(fdt, "prop-str", TEST_STRING_1)); 128 CHECK(fdt_property_u32(fdt, "#address-cells", 1)); 129 CHECK(fdt_property_u32(fdt, "#size-cells", 0)); 130 131 CHECK(fdt_begin_node(fdt, "subnode@1")); 132 CHECK(fdt_property_string(fdt, "compatible", "subnode1")); 133 CHECK(fdt_property_u32(fdt, "reg", 1)); 134 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_1)); 135 CHECK(fdt_begin_node(fdt, "subsubnode")); 136 CHECK(fdt_property(fdt, "compatible", "subsubnode1\0subsubnode", 137 23)); 138 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_1)); 139 CHECK(fdt_end_node(fdt)); 140 CHECK(fdt_begin_node(fdt, "ss1")); 141 CHECK(fdt_end_node(fdt)); 142 CHECK(fdt_end_node(fdt)); 143 144 CHECK(fdt_begin_node(fdt, "subnode@2")); 145 CHECK(fdt_property_u32(fdt, "reg", 2)); 146 CHECK(fdt_property_cell(fdt, "linux,phandle", PHANDLE_1)); 147 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_2)); 148 CHECK(fdt_property_u32(fdt, "#address-cells", 1)); 149 CHECK(fdt_property_u32(fdt, "#size-cells", 0)); 150 CHECK(fdt_begin_node(fdt, "subsubnode@0")); 151 CHECK(fdt_property_u32(fdt, "reg", 0)); 152 CHECK(fdt_property_cell(fdt, "phandle", PHANDLE_2)); 153 CHECK(fdt_property(fdt, "compatible", "subsubnode2\0subsubnode", 154 23)); 155 CHECK(fdt_property_cell(fdt, "prop-int", TEST_VALUE_2)); 156 CHECK(fdt_end_node(fdt)); 157 CHECK(fdt_begin_node(fdt, "ss2")); 158 CHECK(fdt_end_node(fdt)); 159 160 CHECK(fdt_end_node(fdt)); 161 162 CHECK(fdt_end_node(fdt)); 163 164 save_blob("unfinished_tree1.test.dtb", fdt); 165 166 CHECK(fdt_finish(fdt)); 167 168 verbose_printf("Completed tree, totalsize = %d\n", 169 fdt_totalsize(fdt)); 170 171 save_blob("sw_tree1.test.dtb", fdt); 172 173 PASS(); 174} 175