fdt_rw.c 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505
  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. #include <fdt.h>
  53. #include <libfdt.h>
  54. #include "libfdt_internal.h"
  55. static int fdt_blocks_misordered_(const void *fdt,
  56. int mem_rsv_size, int struct_size)
  57. {
  58. return (fdt_off_mem_rsvmap(fdt) < FDT_ALIGN(sizeof(struct fdt_header), 8))
  59. || (fdt_off_dt_struct(fdt) <
  60. (fdt_off_mem_rsvmap(fdt) + mem_rsv_size))
  61. || (fdt_off_dt_strings(fdt) <
  62. (fdt_off_dt_struct(fdt) + struct_size))
  63. || (fdt_totalsize(fdt) <
  64. (fdt_off_dt_strings(fdt) + fdt_size_dt_strings(fdt)));
  65. }
  66. static int fdt_rw_check_header_(void *fdt)
  67. {
  68. FDT_CHECK_HEADER(fdt);
  69. if (fdt_version(fdt) < 17)
  70. return -FDT_ERR_BADVERSION;
  71. if (fdt_blocks_misordered_(fdt, sizeof(struct fdt_reserve_entry),
  72. fdt_size_dt_struct(fdt)))
  73. return -FDT_ERR_BADLAYOUT;
  74. if (fdt_version(fdt) > 17)
  75. fdt_set_version(fdt, 17);
  76. return 0;
  77. }
  78. #define FDT_RW_CHECK_HEADER(fdt) \
  79. { \
  80. int err_; \
  81. if ((err_ = fdt_rw_check_header_(fdt)) != 0) \
  82. return err_; \
  83. }
  84. static inline int fdt_data_size_(void *fdt)
  85. {
  86. return fdt_off_dt_strings(fdt) + fdt_size_dt_strings(fdt);
  87. }
  88. static int fdt_splice_(void *fdt, void *splicepoint, int oldlen, int newlen)
  89. {
  90. char *p = splicepoint;
  91. char *end = (char *)fdt + fdt_data_size_(fdt);
  92. if (((p + oldlen) < p) || ((p + oldlen) > end))
  93. return -FDT_ERR_BADOFFSET;
  94. if ((p < (char *)fdt) || ((end - oldlen + newlen) < (char *)fdt))
  95. return -FDT_ERR_BADOFFSET;
  96. if ((end - oldlen + newlen) > ((char *)fdt + fdt_totalsize(fdt)))
  97. return -FDT_ERR_NOSPACE;
  98. memmove(p + newlen, p + oldlen, end - p - oldlen);
  99. return 0;
  100. }
  101. static int fdt_splice_mem_rsv_(void *fdt, struct fdt_reserve_entry *p,
  102. int oldn, int newn)
  103. {
  104. int delta = (newn - oldn) * sizeof(*p);
  105. int err;
  106. err = fdt_splice_(fdt, p, oldn * sizeof(*p), newn * sizeof(*p));
  107. if (err)
  108. return err;
  109. fdt_set_off_dt_struct(fdt, fdt_off_dt_struct(fdt) + delta);
  110. fdt_set_off_dt_strings(fdt, fdt_off_dt_strings(fdt) + delta);
  111. return 0;
  112. }
  113. static int fdt_splice_struct_(void *fdt, void *p,
  114. int oldlen, int newlen)
  115. {
  116. int delta = newlen - oldlen;
  117. int err;
  118. if ((err = fdt_splice_(fdt, p, oldlen, newlen)))
  119. return err;
  120. fdt_set_size_dt_struct(fdt, fdt_size_dt_struct(fdt) + delta);
  121. fdt_set_off_dt_strings(fdt, fdt_off_dt_strings(fdt) + delta);
  122. return 0;
  123. }
  124. static int fdt_splice_string_(void *fdt, int newlen)
  125. {
  126. void *p = (char *)fdt
  127. + fdt_off_dt_strings(fdt) + fdt_size_dt_strings(fdt);
  128. int err;
  129. if ((err = fdt_splice_(fdt, p, 0, newlen)))
  130. return err;
  131. fdt_set_size_dt_strings(fdt, fdt_size_dt_strings(fdt) + newlen);
  132. return 0;
  133. }
  134. static int fdt_find_add_string_(void *fdt, const char *s)
  135. {
  136. char *strtab = (char *)fdt + fdt_off_dt_strings(fdt);
  137. const char *p;
  138. char *new;
  139. int len = strlen(s) + 1;
  140. int err;
  141. p = fdt_find_string_(strtab, fdt_size_dt_strings(fdt), s);
  142. if (p)
  143. /* found it */
  144. return (p - strtab);
  145. new = strtab + fdt_size_dt_strings(fdt);
  146. err = fdt_splice_string_(fdt, len);
  147. if (err)
  148. return err;
  149. memcpy(new, s, len);
  150. return (new - strtab);
  151. }
  152. int fdt_add_mem_rsv(void *fdt, uint64_t address, uint64_t size)
  153. {
  154. struct fdt_reserve_entry *re;
  155. int err;
  156. FDT_RW_CHECK_HEADER(fdt);
  157. re = fdt_mem_rsv_w_(fdt, fdt_num_mem_rsv(fdt));
  158. err = fdt_splice_mem_rsv_(fdt, re, 0, 1);
  159. if (err)
  160. return err;
  161. re->address = cpu_to_fdt64(address);
  162. re->size = cpu_to_fdt64(size);
  163. return 0;
  164. }
  165. int fdt_del_mem_rsv(void *fdt, int n)
  166. {
  167. struct fdt_reserve_entry *re = fdt_mem_rsv_w_(fdt, n);
  168. FDT_RW_CHECK_HEADER(fdt);
  169. if (n >= fdt_num_mem_rsv(fdt))
  170. return -FDT_ERR_NOTFOUND;
  171. return fdt_splice_mem_rsv_(fdt, re, 1, 0);
  172. }
  173. static int fdt_resize_property_(void *fdt, int nodeoffset, const char *name,
  174. int len, struct fdt_property **prop)
  175. {
  176. int oldlen;
  177. int err;
  178. *prop = fdt_get_property_w(fdt, nodeoffset, name, &oldlen);
  179. if (!*prop)
  180. return oldlen;
  181. if ((err = fdt_splice_struct_(fdt, (*prop)->data, FDT_TAGALIGN(oldlen),
  182. FDT_TAGALIGN(len))))
  183. return err;
  184. (*prop)->len = cpu_to_fdt32(len);
  185. return 0;
  186. }
  187. static int fdt_add_property_(void *fdt, int nodeoffset, const char *name,
  188. int len, struct fdt_property **prop)
  189. {
  190. int proplen;
  191. int nextoffset;
  192. int namestroff;
  193. int err;
  194. if ((nextoffset = fdt_check_node_offset_(fdt, nodeoffset)) < 0)
  195. return nextoffset;
  196. namestroff = fdt_find_add_string_(fdt, name);
  197. if (namestroff < 0)
  198. return namestroff;
  199. *prop = fdt_offset_ptr_w_(fdt, nextoffset);
  200. proplen = sizeof(**prop) + FDT_TAGALIGN(len);
  201. err = fdt_splice_struct_(fdt, *prop, 0, proplen);
  202. if (err)
  203. return err;
  204. (*prop)->tag = cpu_to_fdt32(FDT_PROP);
  205. (*prop)->nameoff = cpu_to_fdt32(namestroff);
  206. (*prop)->len = cpu_to_fdt32(len);
  207. return 0;
  208. }
  209. int fdt_set_name(void *fdt, int nodeoffset, const char *name)
  210. {
  211. char *namep;
  212. int oldlen, newlen;
  213. int err;
  214. FDT_RW_CHECK_HEADER(fdt);
  215. namep = (char *)(uintptr_t)fdt_get_name(fdt, nodeoffset, &oldlen);
  216. if (!namep)
  217. return oldlen;
  218. newlen = strlen(name);
  219. err = fdt_splice_struct_(fdt, namep, FDT_TAGALIGN(oldlen+1),
  220. FDT_TAGALIGN(newlen+1));
  221. if (err)
  222. return err;
  223. memcpy(namep, name, newlen+1);
  224. return 0;
  225. }
  226. int fdt_setprop_placeholder(void *fdt, int nodeoffset, const char *name,
  227. int len, void **prop_data)
  228. {
  229. struct fdt_property *prop;
  230. int err;
  231. FDT_RW_CHECK_HEADER(fdt);
  232. err = fdt_resize_property_(fdt, nodeoffset, name, len, &prop);
  233. if (err == -FDT_ERR_NOTFOUND)
  234. err = fdt_add_property_(fdt, nodeoffset, name, len, &prop);
  235. if (err)
  236. return err;
  237. *prop_data = prop->data;
  238. return 0;
  239. }
  240. int fdt_setprop(void *fdt, int nodeoffset, const char *name,
  241. const void *val, int len)
  242. {
  243. void *prop_data;
  244. int err;
  245. err = fdt_setprop_placeholder(fdt, nodeoffset, name, len, &prop_data);
  246. if (err)
  247. return err;
  248. if (len)
  249. memcpy(prop_data, val, len);
  250. return 0;
  251. }
  252. int fdt_appendprop(void *fdt, int nodeoffset, const char *name,
  253. const void *val, int len)
  254. {
  255. struct fdt_property *prop;
  256. int err, oldlen, newlen;
  257. FDT_RW_CHECK_HEADER(fdt);
  258. prop = fdt_get_property_w(fdt, nodeoffset, name, &oldlen);
  259. if (prop) {
  260. newlen = len + oldlen;
  261. err = fdt_splice_struct_(fdt, prop->data,
  262. FDT_TAGALIGN(oldlen),
  263. FDT_TAGALIGN(newlen));
  264. if (err)
  265. return err;
  266. prop->len = cpu_to_fdt32(newlen);
  267. memcpy(prop->data + oldlen, val, len);
  268. } else {
  269. err = fdt_add_property_(fdt, nodeoffset, name, len, &prop);
  270. if (err)
  271. return err;
  272. memcpy(prop->data, val, len);
  273. }
  274. return 0;
  275. }
  276. int fdt_delprop(void *fdt, int nodeoffset, const char *name)
  277. {
  278. struct fdt_property *prop;
  279. int len, proplen;
  280. FDT_RW_CHECK_HEADER(fdt);
  281. prop = fdt_get_property_w(fdt, nodeoffset, name, &len);
  282. if (!prop)
  283. return len;
  284. proplen = sizeof(*prop) + FDT_TAGALIGN(len);
  285. return fdt_splice_struct_(fdt, prop, proplen, 0);
  286. }
  287. int fdt_add_subnode_namelen(void *fdt, int parentoffset,
  288. const char *name, int namelen)
  289. {
  290. struct fdt_node_header *nh;
  291. int offset, nextoffset;
  292. int nodelen;
  293. int err;
  294. uint32_t tag;
  295. fdt32_t *endtag;
  296. FDT_RW_CHECK_HEADER(fdt);
  297. offset = fdt_subnode_offset_namelen(fdt, parentoffset, name, namelen);
  298. if (offset >= 0)
  299. return -FDT_ERR_EXISTS;
  300. else if (offset != -FDT_ERR_NOTFOUND)
  301. return offset;
  302. /* Try to place the new node after the parent's properties */
  303. fdt_next_tag(fdt, parentoffset, &nextoffset); /* skip the BEGIN_NODE */
  304. do {
  305. offset = nextoffset;
  306. tag = fdt_next_tag(fdt, offset, &nextoffset);
  307. } while ((tag == FDT_PROP) || (tag == FDT_NOP));
  308. nh = fdt_offset_ptr_w_(fdt, offset);
  309. nodelen = sizeof(*nh) + FDT_TAGALIGN(namelen+1) + FDT_TAGSIZE;
  310. err = fdt_splice_struct_(fdt, nh, 0, nodelen);
  311. if (err)
  312. return err;
  313. nh->tag = cpu_to_fdt32(FDT_BEGIN_NODE);
  314. memset(nh->name, 0, FDT_TAGALIGN(namelen+1));
  315. memcpy(nh->name, name, namelen);
  316. endtag = (fdt32_t *)((char *)nh + nodelen - FDT_TAGSIZE);
  317. *endtag = cpu_to_fdt32(FDT_END_NODE);
  318. return offset;
  319. }
  320. int fdt_add_subnode(void *fdt, int parentoffset, const char *name)
  321. {
  322. return fdt_add_subnode_namelen(fdt, parentoffset, name, strlen(name));
  323. }
  324. int fdt_del_node(void *fdt, int nodeoffset)
  325. {
  326. int endoffset;
  327. FDT_RW_CHECK_HEADER(fdt);
  328. endoffset = fdt_node_end_offset_(fdt, nodeoffset);
  329. if (endoffset < 0)
  330. return endoffset;
  331. return fdt_splice_struct_(fdt, fdt_offset_ptr_w_(fdt, nodeoffset),
  332. endoffset - nodeoffset, 0);
  333. }
  334. static void fdt_packblocks_(const char *old, char *new,
  335. int mem_rsv_size, int struct_size)
  336. {
  337. int mem_rsv_off, struct_off, strings_off;
  338. mem_rsv_off = FDT_ALIGN(sizeof(struct fdt_header), 8);
  339. struct_off = mem_rsv_off + mem_rsv_size;
  340. strings_off = struct_off + struct_size;
  341. memmove(new + mem_rsv_off, old + fdt_off_mem_rsvmap(old), mem_rsv_size);
  342. fdt_set_off_mem_rsvmap(new, mem_rsv_off);
  343. memmove(new + struct_off, old + fdt_off_dt_struct(old), struct_size);
  344. fdt_set_off_dt_struct(new, struct_off);
  345. fdt_set_size_dt_struct(new, struct_size);
  346. memmove(new + strings_off, old + fdt_off_dt_strings(old),
  347. fdt_size_dt_strings(old));
  348. fdt_set_off_dt_strings(new, strings_off);
  349. fdt_set_size_dt_strings(new, fdt_size_dt_strings(old));
  350. }
  351. int fdt_open_into(const void *fdt, void *buf, int bufsize)
  352. {
  353. int err;
  354. int mem_rsv_size, struct_size;
  355. int newsize;
  356. const char *fdtstart = fdt;
  357. const char *fdtend = fdtstart + fdt_totalsize(fdt);
  358. char *tmp;
  359. FDT_CHECK_HEADER(fdt);
  360. mem_rsv_size = (fdt_num_mem_rsv(fdt)+1)
  361. * sizeof(struct fdt_reserve_entry);
  362. if (fdt_version(fdt) >= 17) {
  363. struct_size = fdt_size_dt_struct(fdt);
  364. } else {
  365. struct_size = 0;
  366. while (fdt_next_tag(fdt, struct_size, &struct_size) != FDT_END)
  367. ;
  368. if (struct_size < 0)
  369. return struct_size;
  370. }
  371. if (!fdt_blocks_misordered_(fdt, mem_rsv_size, struct_size)) {
  372. /* no further work necessary */
  373. err = fdt_move(fdt, buf, bufsize);
  374. if (err)
  375. return err;
  376. fdt_set_version(buf, 17);
  377. fdt_set_size_dt_struct(buf, struct_size);
  378. fdt_set_totalsize(buf, bufsize);
  379. return 0;
  380. }
  381. /* Need to reorder */
  382. newsize = FDT_ALIGN(sizeof(struct fdt_header), 8) + mem_rsv_size
  383. + struct_size + fdt_size_dt_strings(fdt);
  384. if (bufsize < newsize)
  385. return -FDT_ERR_NOSPACE;
  386. /* First attempt to build converted tree at beginning of buffer */
  387. tmp = buf;
  388. /* But if that overlaps with the old tree... */
  389. if (((tmp + newsize) > fdtstart) && (tmp < fdtend)) {
  390. /* Try right after the old tree instead */
  391. tmp = (char *)(uintptr_t)fdtend;
  392. if ((tmp + newsize) > ((char *)buf + bufsize))
  393. return -FDT_ERR_NOSPACE;
  394. }
  395. fdt_packblocks_(fdt, tmp, mem_rsv_size, struct_size);
  396. memmove(buf, tmp, newsize);
  397. fdt_set_magic(buf, FDT_MAGIC);
  398. fdt_set_totalsize(buf, bufsize);
  399. fdt_set_version(buf, 17);
  400. fdt_set_last_comp_version(buf, 16);
  401. fdt_set_boot_cpuid_phys(buf, fdt_boot_cpuid_phys(fdt));
  402. return 0;
  403. }
  404. int fdt_pack(void *fdt)
  405. {
  406. int mem_rsv_size;
  407. FDT_RW_CHECK_HEADER(fdt);
  408. mem_rsv_size = (fdt_num_mem_rsv(fdt)+1)
  409. * sizeof(struct fdt_reserve_entry);
  410. fdt_packblocks_(fdt, fdt, mem_rsv_size, fdt_size_dt_struct(fdt));
  411. fdt_set_totalsize(fdt, fdt_data_size_(fdt));
  412. return 0;
  413. }