David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 1 | /* |
| 2 | * libfdt - Flat Device Tree manipulation |
| 3 | * Copyright (C) 2006 David Gibson, IBM Corporation. |
| 4 | * |
| 5 | * libfdt is dual licensed: you can use it either under the terms of |
| 6 | * the GPL, or the BSD license, at your option. |
| 7 | * |
| 8 | * a) This library is free software; you can redistribute it and/or |
| 9 | * modify it under the terms of the GNU General Public License as |
| 10 | * published by the Free Software Foundation; either version 2 of the |
| 11 | * License, or (at your option) any later version. |
| 12 | * |
| 13 | * This library is distributed in the hope that it will be useful, |
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 16 | * GNU General Public License for more details. |
| 17 | * |
| 18 | * You should have received a copy of the GNU General Public |
| 19 | * License along with this library; if not, write to the Free |
| 20 | * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, |
| 21 | * MA 02110-1301 USA |
| 22 | * |
| 23 | * Alternatively, |
| 24 | * |
| 25 | * b) Redistribution and use in source and binary forms, with or |
| 26 | * without modification, are permitted provided that the following |
| 27 | * conditions are met: |
| 28 | * |
| 29 | * 1. Redistributions of source code must retain the above |
| 30 | * copyright notice, this list of conditions and the following |
| 31 | * disclaimer. |
| 32 | * 2. Redistributions in binary form must reproduce the above |
| 33 | * copyright notice, this list of conditions and the following |
| 34 | * disclaimer in the documentation and/or other materials |
| 35 | * provided with the distribution. |
| 36 | * |
| 37 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND |
| 38 | * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, |
| 39 | * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
| 40 | * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| 41 | * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
| 42 | * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 43 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
| 44 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
| 45 | * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
| 46 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| 47 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR |
| 48 | * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, |
| 49 | * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 50 | */ |
| 51 | #include "libfdt_env.h" |
| 52 | |
| 53 | #include <fdt.h> |
| 54 | #include <libfdt.h> |
| 55 | |
| 56 | #include "libfdt_internal.h" |
| 57 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 58 | static int _fdt_sw_check_header(void *fdt) |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 59 | { |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 60 | if (fdt_magic(fdt) != FDT_SW_MAGIC) |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 61 | return -FDT_ERR_BADMAGIC; |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 62 | /* FIXME: should check more details about the header state */ |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 63 | return 0; |
| 64 | } |
| 65 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 66 | #define FDT_SW_CHECK_HEADER(fdt) \ |
| 67 | { \ |
| 68 | int err; \ |
| 69 | if ((err = _fdt_sw_check_header(fdt)) != 0) \ |
| 70 | return err; \ |
| 71 | } |
| 72 | |
Stephen Warren | cd29672 | 2012-09-28 21:25:59 +0000 | [diff] [blame] | 73 | static void *_fdt_grab_space(void *fdt, size_t len) |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 74 | { |
| 75 | int offset = fdt_size_dt_struct(fdt); |
| 76 | int spaceleft; |
| 77 | |
| 78 | spaceleft = fdt_totalsize(fdt) - fdt_off_dt_struct(fdt) |
| 79 | - fdt_size_dt_strings(fdt); |
| 80 | |
| 81 | if ((offset + len < offset) || (offset + len > spaceleft)) |
| 82 | return NULL; |
| 83 | |
| 84 | fdt_set_size_dt_struct(fdt, offset + len); |
Stephen Warren | cd29672 | 2012-09-28 21:25:59 +0000 | [diff] [blame] | 85 | return _fdt_offset_ptr_w(fdt, offset); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 86 | } |
| 87 | |
| 88 | int fdt_create(void *buf, int bufsize) |
| 89 | { |
| 90 | void *fdt = buf; |
| 91 | |
| 92 | if (bufsize < sizeof(struct fdt_header)) |
| 93 | return -FDT_ERR_NOSPACE; |
| 94 | |
| 95 | memset(buf, 0, bufsize); |
| 96 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 97 | fdt_set_magic(fdt, FDT_SW_MAGIC); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 98 | fdt_set_version(fdt, FDT_LAST_SUPPORTED_VERSION); |
| 99 | fdt_set_last_comp_version(fdt, FDT_FIRST_SUPPORTED_VERSION); |
| 100 | fdt_set_totalsize(fdt, bufsize); |
| 101 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 102 | fdt_set_off_mem_rsvmap(fdt, FDT_ALIGN(sizeof(struct fdt_header), |
| 103 | sizeof(struct fdt_reserve_entry))); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 104 | fdt_set_off_dt_struct(fdt, fdt_off_mem_rsvmap(fdt)); |
| 105 | fdt_set_off_dt_strings(fdt, bufsize); |
| 106 | |
| 107 | return 0; |
| 108 | } |
| 109 | |
Rob Herring | 4760597 | 2015-04-29 16:00:05 -0500 | [diff] [blame] | 110 | int fdt_resize(void *fdt, void *buf, int bufsize) |
| 111 | { |
| 112 | size_t headsize, tailsize; |
| 113 | char *oldtail, *newtail; |
| 114 | |
| 115 | FDT_SW_CHECK_HEADER(fdt); |
| 116 | |
| 117 | headsize = fdt_off_dt_struct(fdt); |
| 118 | tailsize = fdt_size_dt_strings(fdt); |
| 119 | |
| 120 | if ((headsize + tailsize) > bufsize) |
| 121 | return -FDT_ERR_NOSPACE; |
| 122 | |
| 123 | oldtail = (char *)fdt + fdt_totalsize(fdt) - tailsize; |
| 124 | newtail = (char *)buf + bufsize - tailsize; |
| 125 | |
| 126 | /* Two cases to avoid clobbering data if the old and new |
| 127 | * buffers partially overlap */ |
| 128 | if (buf <= fdt) { |
| 129 | memmove(buf, fdt, headsize); |
| 130 | memmove(newtail, oldtail, tailsize); |
| 131 | } else { |
| 132 | memmove(newtail, oldtail, tailsize); |
| 133 | memmove(buf, fdt, headsize); |
| 134 | } |
| 135 | |
| 136 | fdt_set_off_dt_strings(buf, bufsize); |
| 137 | fdt_set_totalsize(buf, bufsize); |
| 138 | |
| 139 | return 0; |
| 140 | } |
| 141 | |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 142 | int fdt_add_reservemap_entry(void *fdt, uint64_t addr, uint64_t size) |
| 143 | { |
| 144 | struct fdt_reserve_entry *re; |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 145 | int offset; |
| 146 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 147 | FDT_SW_CHECK_HEADER(fdt); |
| 148 | |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 149 | if (fdt_size_dt_struct(fdt)) |
| 150 | return -FDT_ERR_BADSTATE; |
| 151 | |
| 152 | offset = fdt_off_dt_struct(fdt); |
| 153 | if ((offset + sizeof(*re)) > fdt_totalsize(fdt)) |
| 154 | return -FDT_ERR_NOSPACE; |
| 155 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 156 | re = (struct fdt_reserve_entry *)((char *)fdt + offset); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 157 | re->address = cpu_to_fdt64(addr); |
| 158 | re->size = cpu_to_fdt64(size); |
| 159 | |
| 160 | fdt_set_off_dt_struct(fdt, offset + sizeof(*re)); |
| 161 | |
| 162 | return 0; |
| 163 | } |
| 164 | |
| 165 | int fdt_finish_reservemap(void *fdt) |
| 166 | { |
| 167 | return fdt_add_reservemap_entry(fdt, 0, 0); |
| 168 | } |
| 169 | |
| 170 | int fdt_begin_node(void *fdt, const char *name) |
| 171 | { |
| 172 | struct fdt_node_header *nh; |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 173 | int namelen = strlen(name) + 1; |
| 174 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 175 | FDT_SW_CHECK_HEADER(fdt); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 176 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 177 | nh = _fdt_grab_space(fdt, sizeof(*nh) + FDT_TAGALIGN(namelen)); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 178 | if (! nh) |
| 179 | return -FDT_ERR_NOSPACE; |
| 180 | |
| 181 | nh->tag = cpu_to_fdt32(FDT_BEGIN_NODE); |
| 182 | memcpy(nh->name, name, namelen); |
| 183 | return 0; |
| 184 | } |
| 185 | |
| 186 | int fdt_end_node(void *fdt) |
| 187 | { |
Rob Herring | 4760597 | 2015-04-29 16:00:05 -0500 | [diff] [blame] | 188 | fdt32_t *en; |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 189 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 190 | FDT_SW_CHECK_HEADER(fdt); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 191 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 192 | en = _fdt_grab_space(fdt, FDT_TAGSIZE); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 193 | if (! en) |
| 194 | return -FDT_ERR_NOSPACE; |
| 195 | |
| 196 | *en = cpu_to_fdt32(FDT_END_NODE); |
| 197 | return 0; |
| 198 | } |
| 199 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 200 | static int _fdt_find_add_string(void *fdt, const char *s) |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 201 | { |
| 202 | char *strtab = (char *)fdt + fdt_totalsize(fdt); |
| 203 | const char *p; |
| 204 | int strtabsize = fdt_size_dt_strings(fdt); |
| 205 | int len = strlen(s) + 1; |
| 206 | int struct_top, offset; |
| 207 | |
| 208 | p = _fdt_find_string(strtab - strtabsize, strtabsize, s); |
| 209 | if (p) |
| 210 | return p - strtab; |
| 211 | |
| 212 | /* Add it */ |
| 213 | offset = -strtabsize - len; |
| 214 | struct_top = fdt_off_dt_struct(fdt) + fdt_size_dt_struct(fdt); |
| 215 | if (fdt_totalsize(fdt) + offset < struct_top) |
| 216 | return 0; /* no more room :( */ |
| 217 | |
| 218 | memcpy(strtab + offset, s, len); |
| 219 | fdt_set_size_dt_strings(fdt, strtabsize + len); |
| 220 | return offset; |
| 221 | } |
| 222 | |
| 223 | int fdt_property(void *fdt, const char *name, const void *val, int len) |
| 224 | { |
| 225 | struct fdt_property *prop; |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 226 | int nameoff; |
| 227 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 228 | FDT_SW_CHECK_HEADER(fdt); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 229 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 230 | nameoff = _fdt_find_add_string(fdt, name); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 231 | if (nameoff == 0) |
| 232 | return -FDT_ERR_NOSPACE; |
| 233 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 234 | prop = _fdt_grab_space(fdt, sizeof(*prop) + FDT_TAGALIGN(len)); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 235 | if (! prop) |
| 236 | return -FDT_ERR_NOSPACE; |
| 237 | |
| 238 | prop->tag = cpu_to_fdt32(FDT_PROP); |
| 239 | prop->nameoff = cpu_to_fdt32(nameoff); |
| 240 | prop->len = cpu_to_fdt32(len); |
| 241 | memcpy(prop->data, val, len); |
| 242 | return 0; |
| 243 | } |
| 244 | |
| 245 | int fdt_finish(void *fdt) |
| 246 | { |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 247 | char *p = (char *)fdt; |
Rob Herring | 4760597 | 2015-04-29 16:00:05 -0500 | [diff] [blame] | 248 | fdt32_t *end; |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 249 | int oldstroffset, newstroffset; |
| 250 | uint32_t tag; |
| 251 | int offset, nextoffset; |
| 252 | |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 253 | FDT_SW_CHECK_HEADER(fdt); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 254 | |
| 255 | /* Add terminator */ |
David Gibson | ed95d74 | 2008-08-07 12:24:17 +1000 | [diff] [blame] | 256 | end = _fdt_grab_space(fdt, sizeof(*end)); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 257 | if (! end) |
| 258 | return -FDT_ERR_NOSPACE; |
| 259 | *end = cpu_to_fdt32(FDT_END); |
| 260 | |
| 261 | /* Relocate the string table */ |
| 262 | oldstroffset = fdt_totalsize(fdt) - fdt_size_dt_strings(fdt); |
| 263 | newstroffset = fdt_off_dt_struct(fdt) + fdt_size_dt_struct(fdt); |
| 264 | memmove(p + newstroffset, p + oldstroffset, fdt_size_dt_strings(fdt)); |
| 265 | fdt_set_off_dt_strings(fdt, newstroffset); |
| 266 | |
| 267 | /* Walk the structure, correcting string offsets */ |
| 268 | offset = 0; |
| 269 | while ((tag = fdt_next_tag(fdt, offset, &nextoffset)) != FDT_END) { |
| 270 | if (tag == FDT_PROP) { |
| 271 | struct fdt_property *prop = |
Stephen Warren | cd29672 | 2012-09-28 21:25:59 +0000 | [diff] [blame] | 272 | _fdt_offset_ptr_w(fdt, offset); |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 273 | int nameoff; |
| 274 | |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 275 | nameoff = fdt32_to_cpu(prop->nameoff); |
| 276 | nameoff += fdt_size_dt_strings(fdt); |
| 277 | prop->nameoff = cpu_to_fdt32(nameoff); |
| 278 | } |
| 279 | offset = nextoffset; |
| 280 | } |
Stephen Warren | cd29672 | 2012-09-28 21:25:59 +0000 | [diff] [blame] | 281 | if (nextoffset < 0) |
| 282 | return nextoffset; |
David Gibson | 1cade99 | 2007-12-10 14:28:39 +1100 | [diff] [blame] | 283 | |
| 284 | /* Finally, adjust the header */ |
| 285 | fdt_set_totalsize(fdt, newstroffset + fdt_size_dt_strings(fdt)); |
| 286 | fdt_set_magic(fdt, FDT_MAGIC); |
| 287 | return 0; |
| 288 | } |