Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 1 | #!/usr/bin/python |
Tom Rini | 83d290c | 2018-05-06 17:58:06 -0400 | [diff] [blame] | 2 | # SPDX-License-Identifier: GPL-2.0+ |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 3 | # |
| 4 | # Copyright (C) 2016 Google, Inc |
| 5 | # Written by Simon Glass <sjg@chromium.org> |
| 6 | # |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 7 | |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 8 | from enum import IntEnum |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 9 | import struct |
| 10 | import sys |
| 11 | |
Simon Glass | bf77667 | 2020-04-17 18:09:04 -0600 | [diff] [blame] | 12 | from dtoc import fdt_util |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 13 | import libfdt |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 14 | from libfdt import QUIET_NOTFOUND |
Simon Glass | 4583c00 | 2023-02-23 18:18:04 -0700 | [diff] [blame] | 15 | from u_boot_pylib import tools |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 16 | from u_boot_pylib import tout |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 17 | |
| 18 | # This deals with a device tree, presenting it as an assortment of Node and |
| 19 | # Prop objects, representing nodes and properties, respectively. This file |
Simon Glass | 99ed4a2 | 2017-05-27 07:38:30 -0600 | [diff] [blame] | 20 | # contains the base classes and defines the high-level API. You can use |
| 21 | # FdtScan() as a convenience function to create and scan an Fdt. |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 22 | |
| 23 | # This implementation uses a libfdt Python library to access the device tree, |
| 24 | # so it is fairly efficient. |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 25 | |
Simon Glass | bc1dea3 | 2016-07-25 18:59:05 -0600 | [diff] [blame] | 26 | # A list of types we support |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 27 | class Type(IntEnum): |
Simon Glass | df82de8 | 2021-07-28 19:23:09 -0600 | [diff] [blame] | 28 | # Types in order from widest to narrowest |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 29 | (BYTE, INT, STRING, BOOL, INT64) = range(5) |
| 30 | |
Simon Glass | df82de8 | 2021-07-28 19:23:09 -0600 | [diff] [blame] | 31 | def needs_widening(self, other): |
| 32 | """Check if this type needs widening to hold a value from another type |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 33 | |
Simon Glass | df82de8 | 2021-07-28 19:23:09 -0600 | [diff] [blame] | 34 | A wider type is one that can hold a wider array of information than |
| 35 | another one, or is less restrictive, so it can hold the information of |
| 36 | another type as well as its own. This is similar to the concept of |
| 37 | type-widening in C. |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 38 | |
| 39 | This uses a simple arithmetic comparison, since type values are in order |
Simon Glass | df82de8 | 2021-07-28 19:23:09 -0600 | [diff] [blame] | 40 | from widest (BYTE) to narrowest (INT64). |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 41 | |
| 42 | Args: |
| 43 | other: Other type to compare against |
| 44 | |
| 45 | Return: |
| 46 | True if the other type is wider |
| 47 | """ |
| 48 | return self.value > other.value |
Simon Glass | bc1dea3 | 2016-07-25 18:59:05 -0600 | [diff] [blame] | 49 | |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 50 | def CheckErr(errnum, msg): |
| 51 | if errnum: |
| 52 | raise ValueError('Error %d: %s: %s' % |
| 53 | (errnum, libfdt.fdt_strerror(errnum), msg)) |
| 54 | |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 55 | |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 56 | def BytesToValue(data): |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 57 | """Converts a string of bytes into a type and value |
| 58 | |
| 59 | Args: |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 60 | A bytes value (which on Python 2 is an alias for str) |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 61 | |
| 62 | Return: |
| 63 | A tuple: |
| 64 | Type of data |
| 65 | Data, either a single element or a list of elements. Each element |
| 66 | is one of: |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 67 | Type.STRING: str/bytes value from the property |
| 68 | Type.INT: a byte-swapped integer stored as a 4-byte str/bytes |
| 69 | Type.BYTE: a byte stored as a single-byte str/bytes |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 70 | """ |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 71 | data = bytes(data) |
| 72 | size = len(data) |
| 73 | strings = data.split(b'\0') |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 74 | is_string = True |
| 75 | count = len(strings) - 1 |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 76 | if count > 0 and not len(strings[-1]): |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 77 | for string in strings[:-1]: |
| 78 | if not string: |
| 79 | is_string = False |
| 80 | break |
| 81 | for ch in string: |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 82 | if ch < 32 or ch > 127: |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 83 | is_string = False |
| 84 | break |
| 85 | else: |
| 86 | is_string = False |
| 87 | if is_string: |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 88 | if count == 1: |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 89 | return Type.STRING, strings[0].decode() |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 90 | else: |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 91 | return Type.STRING, [s.decode() for s in strings[:-1]] |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 92 | if size % 4: |
| 93 | if size == 1: |
Simon Glass | 479dd30 | 2020-11-08 20:36:20 -0700 | [diff] [blame] | 94 | return Type.BYTE, chr(data[0]) |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 95 | else: |
Simon Glass | 479dd30 | 2020-11-08 20:36:20 -0700 | [diff] [blame] | 96 | return Type.BYTE, [chr(ch) for ch in list(data)] |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 97 | val = [] |
| 98 | for i in range(0, size, 4): |
Simon Glass | 2b6ed5e | 2019-05-17 22:00:35 -0600 | [diff] [blame] | 99 | val.append(data[i:i + 4]) |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 100 | if size == 4: |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 101 | return Type.INT, val[0] |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 102 | else: |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 103 | return Type.INT, val |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 104 | |
| 105 | |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 106 | class Prop: |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 107 | """A device tree property |
| 108 | |
| 109 | Properties: |
Simon Glass | 37ba984 | 2021-03-21 18:24:35 +1300 | [diff] [blame] | 110 | node: Node containing this property |
| 111 | offset: Offset of the property (None if still to be synced) |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 112 | name: Property name (as per the device tree) |
| 113 | value: Property value as a string of bytes, or a list of strings of |
| 114 | bytes |
| 115 | type: Value type |
| 116 | """ |
Simon Glass | 928527f | 2019-05-17 22:00:37 -0600 | [diff] [blame] | 117 | def __init__(self, node, offset, name, data): |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 118 | self._node = node |
| 119 | self._offset = offset |
| 120 | self.name = name |
| 121 | self.value = None |
Simon Glass | 928527f | 2019-05-17 22:00:37 -0600 | [diff] [blame] | 122 | self.bytes = bytes(data) |
Simon Glass | 37ba984 | 2021-03-21 18:24:35 +1300 | [diff] [blame] | 123 | self.dirty = offset is None |
Simon Glass | 928527f | 2019-05-17 22:00:37 -0600 | [diff] [blame] | 124 | if not data: |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 125 | self.type = Type.BOOL |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 126 | self.value = True |
| 127 | return |
Simon Glass | 928527f | 2019-05-17 22:00:37 -0600 | [diff] [blame] | 128 | self.type, self.value = BytesToValue(bytes(data)) |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 129 | |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 130 | def RefreshOffset(self, poffset): |
| 131 | self._offset = poffset |
| 132 | |
Simon Glass | c322a85 | 2016-07-25 18:59:06 -0600 | [diff] [blame] | 133 | def Widen(self, newprop): |
| 134 | """Figure out which property type is more general |
| 135 | |
| 136 | Given a current property and a new property, this function returns the |
| 137 | one that is less specific as to type. The less specific property will |
| 138 | be ble to represent the data in the more specific property. This is |
| 139 | used for things like: |
| 140 | |
| 141 | node1 { |
| 142 | compatible = "fred"; |
| 143 | value = <1>; |
| 144 | }; |
| 145 | node1 { |
| 146 | compatible = "fred"; |
| 147 | value = <1 2>; |
| 148 | }; |
| 149 | |
| 150 | He we want to use an int array for 'value'. The first property |
| 151 | suggests that a single int is enough, but the second one shows that |
| 152 | it is not. Calling this function with these two propertes would |
| 153 | update the current property to be like the second, since it is less |
| 154 | specific. |
| 155 | """ |
Simon Glass | df82de8 | 2021-07-28 19:23:09 -0600 | [diff] [blame] | 156 | if self.type.needs_widening(newprop.type): |
Simon Glass | eec44c7 | 2021-07-28 19:23:11 -0600 | [diff] [blame] | 157 | |
| 158 | # A boolean has an empty value: if it exists it is True and if not |
| 159 | # it is False. So when widening we always start with an empty list |
| 160 | # since the only valid integer property would be an empty list of |
| 161 | # integers. |
| 162 | # e.g. this is a boolean: |
| 163 | # some-prop; |
| 164 | # and it would be widened to int list by: |
| 165 | # some-prop = <1 2>; |
| 166 | if self.type == Type.BOOL: |
| 167 | self.type = Type.INT |
| 168 | self.value = [self.GetEmpty(self.type)] |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 169 | if self.type == Type.INT and newprop.type == Type.BYTE: |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 170 | if type(self.value) == list: |
| 171 | new_value = [] |
| 172 | for val in self.value: |
Simon Glass | 479dd30 | 2020-11-08 20:36:20 -0700 | [diff] [blame] | 173 | new_value += [chr(by) for by in val] |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 174 | else: |
Simon Glass | 479dd30 | 2020-11-08 20:36:20 -0700 | [diff] [blame] | 175 | new_value = [chr(by) for by in self.value] |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 176 | self.value = new_value |
Simon Glass | c322a85 | 2016-07-25 18:59:06 -0600 | [diff] [blame] | 177 | self.type = newprop.type |
| 178 | |
Simon Glass | ca04494 | 2021-07-28 19:23:10 -0600 | [diff] [blame] | 179 | if type(newprop.value) == list: |
| 180 | if type(self.value) != list: |
| 181 | self.value = [self.value] |
Simon Glass | c322a85 | 2016-07-25 18:59:06 -0600 | [diff] [blame] | 182 | |
Simon Glass | ca04494 | 2021-07-28 19:23:10 -0600 | [diff] [blame] | 183 | if len(newprop.value) > len(self.value): |
| 184 | val = self.GetEmpty(self.type) |
| 185 | while len(self.value) < len(newprop.value): |
| 186 | self.value.append(val) |
Simon Glass | c322a85 | 2016-07-25 18:59:06 -0600 | [diff] [blame] | 187 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 188 | @classmethod |
Simon Glass | bc1dea3 | 2016-07-25 18:59:05 -0600 | [diff] [blame] | 189 | def GetEmpty(self, type): |
| 190 | """Get an empty / zero value of the given type |
| 191 | |
| 192 | Returns: |
| 193 | A single value of the given type |
| 194 | """ |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 195 | if type == Type.BYTE: |
Simon Glass | bc1dea3 | 2016-07-25 18:59:05 -0600 | [diff] [blame] | 196 | return chr(0) |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 197 | elif type == Type.INT: |
Simon Glass | af53f5a | 2018-09-14 04:57:14 -0600 | [diff] [blame] | 198 | return struct.pack('>I', 0); |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 199 | elif type == Type.STRING: |
Simon Glass | bc1dea3 | 2016-07-25 18:59:05 -0600 | [diff] [blame] | 200 | return '' |
| 201 | else: |
| 202 | return True |
| 203 | |
Simon Glass | babdbde | 2016-07-25 18:59:16 -0600 | [diff] [blame] | 204 | def GetOffset(self): |
| 205 | """Get the offset of a property |
| 206 | |
Simon Glass | babdbde | 2016-07-25 18:59:16 -0600 | [diff] [blame] | 207 | Returns: |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 208 | The offset of the property (struct fdt_property) within the file |
Simon Glass | babdbde | 2016-07-25 18:59:16 -0600 | [diff] [blame] | 209 | """ |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 210 | self._node._fdt.CheckCache() |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 211 | return self._node._fdt.GetStructOffset(self._offset) |
Simon Glass | babdbde | 2016-07-25 18:59:16 -0600 | [diff] [blame] | 212 | |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 213 | def SetInt(self, val): |
| 214 | """Set the integer value of the property |
| 215 | |
| 216 | The device tree is marked dirty so that the value will be written to |
| 217 | the block on the next sync. |
| 218 | |
| 219 | Args: |
| 220 | val: Integer value (32-bit, single cell) |
| 221 | """ |
| 222 | self.bytes = struct.pack('>I', val); |
Simon Glass | 41b781d | 2018-10-01 12:22:49 -0600 | [diff] [blame] | 223 | self.value = self.bytes |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 224 | self.type = Type.INT |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 225 | self.dirty = True |
| 226 | |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 227 | def SetData(self, bytes): |
| 228 | """Set the value of a property as bytes |
| 229 | |
| 230 | Args: |
| 231 | bytes: New property value to set |
| 232 | """ |
Simon Glass | f6b6481 | 2019-05-17 22:00:36 -0600 | [diff] [blame] | 233 | self.bytes = bytes |
Simon Glass | 7e6952d | 2019-05-17 22:00:34 -0600 | [diff] [blame] | 234 | self.type, self.value = BytesToValue(bytes) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 235 | self.dirty = True |
| 236 | |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 237 | def Sync(self, auto_resize=False): |
| 238 | """Sync property changes back to the device tree |
| 239 | |
| 240 | This updates the device tree blob with any changes to this property |
| 241 | since the last sync. |
| 242 | |
| 243 | Args: |
| 244 | auto_resize: Resize the device tree automatically if it does not |
| 245 | have enough space for the update |
| 246 | |
| 247 | Raises: |
| 248 | FdtException if auto_resize is False and there is not enough space |
| 249 | """ |
Simon Glass | 37ba984 | 2021-03-21 18:24:35 +1300 | [diff] [blame] | 250 | if self.dirty: |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 251 | node = self._node |
| 252 | fdt_obj = node._fdt._fdt_obj |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 253 | node_name = fdt_obj.get_name(node._offset) |
| 254 | if node_name and node_name != node.name: |
| 255 | raise ValueError("Internal error, node '%s' name mismatch '%s'" % |
| 256 | (node.path, node_name)) |
| 257 | |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 258 | if auto_resize: |
| 259 | while fdt_obj.setprop(node.Offset(), self.name, self.bytes, |
| 260 | (libfdt.NOSPACE,)) == -libfdt.NOSPACE: |
Simon Glass | c063917 | 2020-07-09 18:39:44 -0600 | [diff] [blame] | 261 | fdt_obj.resize(fdt_obj.totalsize() + 1024 + |
| 262 | len(self.bytes)) |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 263 | fdt_obj.setprop(node.Offset(), self.name, self.bytes) |
| 264 | else: |
| 265 | fdt_obj.setprop(node.Offset(), self.name, self.bytes) |
Simon Glass | 37ba984 | 2021-03-21 18:24:35 +1300 | [diff] [blame] | 266 | self.dirty = False |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 267 | |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 268 | def purge(self): |
| 269 | """Set a property offset to None |
| 270 | |
| 271 | The property remains in the tree structure and will be recreated when |
| 272 | the FDT is synced |
| 273 | """ |
| 274 | self._offset = None |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 275 | |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 276 | class Node: |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 277 | """A device tree node |
| 278 | |
| 279 | Properties: |
Simon Glass | 37ba984 | 2021-03-21 18:24:35 +1300 | [diff] [blame] | 280 | parent: Parent Node |
| 281 | offset: Integer offset in the device tree (None if to be synced) |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 282 | name: Device tree node tname |
| 283 | path: Full path to node, along with the node name itself |
| 284 | _fdt: Device tree object |
| 285 | subnodes: A list of subnodes for this node, each a Node object |
| 286 | props: A dict of properties for this node, each a Prop object. |
| 287 | Keyed by property name |
| 288 | """ |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 289 | def __init__(self, fdt, parent, offset, name, path): |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 290 | self._fdt = fdt |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 291 | self.parent = parent |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 292 | self._offset = offset |
| 293 | self.name = name |
| 294 | self.path = path |
| 295 | self.subnodes = [] |
| 296 | self.props = {} |
| 297 | |
Simon Glass | 94a7c60 | 2018-07-17 13:25:46 -0600 | [diff] [blame] | 298 | def GetFdt(self): |
| 299 | """Get the Fdt object for this node |
| 300 | |
| 301 | Returns: |
| 302 | Fdt object |
| 303 | """ |
| 304 | return self._fdt |
| 305 | |
Simon Glass | 1d85888 | 2018-07-17 13:25:41 -0600 | [diff] [blame] | 306 | def FindNode(self, name): |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 307 | """Find a node given its name |
| 308 | |
| 309 | Args: |
| 310 | name: Node name to look for |
| 311 | Returns: |
| 312 | Node object if found, else None |
| 313 | """ |
| 314 | for subnode in self.subnodes: |
| 315 | if subnode.name == name: |
| 316 | return subnode |
| 317 | return None |
| 318 | |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 319 | def Offset(self): |
| 320 | """Returns the offset of a node, after checking the cache |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 321 | |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 322 | This should be used instead of self._offset directly, to ensure that |
| 323 | the cache does not contain invalid offsets. |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 324 | """ |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 325 | self._fdt.CheckCache() |
| 326 | return self._offset |
| 327 | |
| 328 | def Scan(self): |
| 329 | """Scan a node's properties and subnodes |
| 330 | |
| 331 | This fills in the props and subnodes properties, recursively |
| 332 | searching into subnodes so that the entire tree is built. |
| 333 | """ |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 334 | fdt_obj = self._fdt._fdt_obj |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 335 | self.props = self._fdt.GetProps(self) |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 336 | phandle = fdt_obj.get_phandle(self.Offset()) |
Simon Glass | 09264e0 | 2017-08-29 14:15:52 -0600 | [diff] [blame] | 337 | if phandle: |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 338 | self._fdt.phandle_to_node[phandle] = self |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 339 | |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 340 | offset = fdt_obj.first_subnode(self.Offset(), QUIET_NOTFOUND) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 341 | while offset >= 0: |
| 342 | sep = '' if self.path[-1] == '/' else '/' |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 343 | name = fdt_obj.get_name(offset) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 344 | path = self.path + sep + name |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 345 | node = Node(self._fdt, self, offset, name, path) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 346 | self.subnodes.append(node) |
| 347 | |
| 348 | node.Scan() |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 349 | offset = fdt_obj.next_subnode(offset, QUIET_NOTFOUND) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 350 | |
| 351 | def Refresh(self, my_offset): |
| 352 | """Fix up the _offset for each node, recursively |
| 353 | |
| 354 | Note: This does not take account of property offsets - these will not |
| 355 | be updated. |
| 356 | """ |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 357 | fdt_obj = self._fdt._fdt_obj |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 358 | if self._offset != my_offset: |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 359 | self._offset = my_offset |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 360 | name = fdt_obj.get_name(self._offset) |
| 361 | if name and self.name != name: |
| 362 | raise ValueError("Internal error, node '%s' name mismatch '%s'" % |
| 363 | (self.path, name)) |
| 364 | |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 365 | offset = fdt_obj.first_subnode(self._offset, QUIET_NOTFOUND) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 366 | for subnode in self.subnodes: |
Simon Glass | dd857ee | 2022-02-08 11:49:52 -0700 | [diff] [blame] | 367 | if subnode._offset is None: |
| 368 | continue |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 369 | if subnode.name != fdt_obj.get_name(offset): |
| 370 | raise ValueError('Internal error, node name mismatch %s != %s' % |
| 371 | (subnode.name, fdt_obj.get_name(offset))) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 372 | subnode.Refresh(offset) |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 373 | offset = fdt_obj.next_subnode(offset, QUIET_NOTFOUND) |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 374 | if offset != -libfdt.FDT_ERR_NOTFOUND: |
| 375 | raise ValueError('Internal error, offset == %d' % offset) |
| 376 | |
| 377 | poffset = fdt_obj.first_property_offset(self._offset, QUIET_NOTFOUND) |
| 378 | while poffset >= 0: |
| 379 | p = fdt_obj.get_property_by_offset(poffset) |
| 380 | prop = self.props.get(p.name) |
| 381 | if not prop: |
Simon Glass | acd9861 | 2021-03-21 18:24:34 +1300 | [diff] [blame] | 382 | raise ValueError("Internal error, node '%s' property '%s' missing, " |
| 383 | 'offset %d' % (self.path, p.name, poffset)) |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 384 | prop.RefreshOffset(poffset) |
| 385 | poffset = fdt_obj.next_property_offset(poffset, QUIET_NOTFOUND) |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 386 | |
Simon Glass | 2a70d89 | 2016-07-25 18:59:14 -0600 | [diff] [blame] | 387 | def DeleteProp(self, prop_name): |
| 388 | """Delete a property of a node |
| 389 | |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 390 | The property is deleted and the offset cache is invalidated. |
Simon Glass | 2a70d89 | 2016-07-25 18:59:14 -0600 | [diff] [blame] | 391 | |
| 392 | Args: |
| 393 | prop_name: Name of the property to delete |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 394 | Raises: |
| 395 | ValueError if the property does not exist |
Simon Glass | 2a70d89 | 2016-07-25 18:59:14 -0600 | [diff] [blame] | 396 | """ |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 397 | CheckErr(self._fdt._fdt_obj.delprop(self.Offset(), prop_name), |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 398 | "Node '%s': delete property: '%s'" % (self.path, prop_name)) |
| 399 | del self.props[prop_name] |
| 400 | self._fdt.Invalidate() |
Simon Glass | 2a70d89 | 2016-07-25 18:59:14 -0600 | [diff] [blame] | 401 | |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 402 | def AddZeroProp(self, prop_name): |
| 403 | """Add a new property to the device tree with an integer value of 0. |
| 404 | |
| 405 | Args: |
| 406 | prop_name: Name of property |
| 407 | """ |
Simon Glass | 194b8d5 | 2019-05-17 22:00:33 -0600 | [diff] [blame] | 408 | self.props[prop_name] = Prop(self, None, prop_name, |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 409 | tools.get_bytes(0, 4)) |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 410 | |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 411 | def AddEmptyProp(self, prop_name, len): |
| 412 | """Add a property with a fixed data size, for filling in later |
| 413 | |
| 414 | The device tree is marked dirty so that the value will be written to |
| 415 | the blob on the next sync. |
| 416 | |
| 417 | Args: |
| 418 | prop_name: Name of property |
| 419 | len: Length of data in property |
| 420 | """ |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 421 | value = tools.get_bytes(0, len) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 422 | self.props[prop_name] = Prop(self, None, prop_name, value) |
| 423 | |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 424 | def _CheckProp(self, prop_name): |
| 425 | """Check if a property is present |
| 426 | |
| 427 | Args: |
| 428 | prop_name: Name of property |
| 429 | |
| 430 | Returns: |
| 431 | self |
| 432 | |
| 433 | Raises: |
| 434 | ValueError if the property is missing |
| 435 | """ |
| 436 | if prop_name not in self.props: |
| 437 | raise ValueError("Fdt '%s', node '%s': Missing property '%s'" % |
| 438 | (self._fdt._fname, self.path, prop_name)) |
| 439 | return self |
| 440 | |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 441 | def SetInt(self, prop_name, val): |
| 442 | """Update an integer property int the device tree. |
| 443 | |
| 444 | This is not allowed to change the size of the FDT. |
| 445 | |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 446 | The device tree is marked dirty so that the value will be written to |
| 447 | the blob on the next sync. |
| 448 | |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 449 | Args: |
| 450 | prop_name: Name of property |
| 451 | val: Value to set |
| 452 | """ |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 453 | self._CheckProp(prop_name).props[prop_name].SetInt(val) |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 454 | |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 455 | def SetData(self, prop_name, val): |
| 456 | """Set the data value of a property |
| 457 | |
| 458 | The device tree is marked dirty so that the value will be written to |
| 459 | the blob on the next sync. |
| 460 | |
| 461 | Args: |
| 462 | prop_name: Name of property to set |
| 463 | val: Data value to set |
| 464 | """ |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 465 | self._CheckProp(prop_name).props[prop_name].SetData(val) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 466 | |
| 467 | def SetString(self, prop_name, val): |
| 468 | """Set the string value of a property |
| 469 | |
| 470 | The device tree is marked dirty so that the value will be written to |
| 471 | the blob on the next sync. |
| 472 | |
| 473 | Args: |
| 474 | prop_name: Name of property to set |
| 475 | val: String value to set (will be \0-terminated in DT) |
| 476 | """ |
Simon Glass | a90df2b | 2019-10-31 07:43:04 -0600 | [diff] [blame] | 477 | if type(val) == str: |
| 478 | val = val.encode('utf-8') |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 479 | self._CheckProp(prop_name).props[prop_name].SetData(val + b'\0') |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 480 | |
Simon Glass | c063917 | 2020-07-09 18:39:44 -0600 | [diff] [blame] | 481 | def AddData(self, prop_name, val): |
| 482 | """Add a new property to a node |
| 483 | |
| 484 | The device tree is marked dirty so that the value will be written to |
| 485 | the blob on the next sync. |
| 486 | |
| 487 | Args: |
| 488 | prop_name: Name of property to add |
| 489 | val: Bytes value of property |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 490 | |
| 491 | Returns: |
| 492 | Prop added |
Simon Glass | c063917 | 2020-07-09 18:39:44 -0600 | [diff] [blame] | 493 | """ |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 494 | prop = Prop(self, None, prop_name, val) |
| 495 | self.props[prop_name] = prop |
| 496 | return prop |
Simon Glass | c063917 | 2020-07-09 18:39:44 -0600 | [diff] [blame] | 497 | |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 498 | def AddString(self, prop_name, val): |
| 499 | """Add a new string property to a node |
| 500 | |
| 501 | The device tree is marked dirty so that the value will be written to |
| 502 | the blob on the next sync. |
| 503 | |
| 504 | Args: |
| 505 | prop_name: Name of property to add |
| 506 | val: String value of property |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 507 | |
| 508 | Returns: |
| 509 | Prop added |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 510 | """ |
Simon Glass | 9fc6ebd | 2021-01-06 21:35:10 -0700 | [diff] [blame] | 511 | val = bytes(val, 'utf-8') |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 512 | return self.AddData(prop_name, val + b'\0') |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 513 | |
Simon Glass | bc11602 | 2022-02-08 11:49:50 -0700 | [diff] [blame] | 514 | def AddStringList(self, prop_name, val): |
| 515 | """Add a new string-list property to a node |
| 516 | |
| 517 | The device tree is marked dirty so that the value will be written to |
| 518 | the blob on the next sync. |
| 519 | |
| 520 | Args: |
| 521 | prop_name: Name of property to add |
| 522 | val (list of str): List of strings to add |
| 523 | |
| 524 | Returns: |
| 525 | Prop added |
| 526 | """ |
Simon Glass | 0ded4d4 | 2022-03-05 20:18:56 -0700 | [diff] [blame] | 527 | out = b'\0'.join(bytes(s, 'utf-8') for s in val) + b'\0' if val else b'' |
Simon Glass | bc11602 | 2022-02-08 11:49:50 -0700 | [diff] [blame] | 528 | return self.AddData(prop_name, out) |
| 529 | |
Simon Glass | 6eb9932 | 2021-01-06 21:35:18 -0700 | [diff] [blame] | 530 | def AddInt(self, prop_name, val): |
| 531 | """Add a new integer property to a node |
| 532 | |
| 533 | The device tree is marked dirty so that the value will be written to |
| 534 | the blob on the next sync. |
| 535 | |
| 536 | Args: |
| 537 | prop_name: Name of property to add |
| 538 | val: Integer value of property |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 539 | |
| 540 | Returns: |
| 541 | Prop added |
Simon Glass | 6eb9932 | 2021-01-06 21:35:18 -0700 | [diff] [blame] | 542 | """ |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 543 | return self.AddData(prop_name, struct.pack('>I', val)) |
Simon Glass | 6eb9932 | 2021-01-06 21:35:18 -0700 | [diff] [blame] | 544 | |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 545 | def Subnode(self, name): |
| 546 | """Create new subnode for the node |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 547 | |
| 548 | Args: |
| 549 | name: name of node to add |
| 550 | |
| 551 | Returns: |
| 552 | New subnode that was created |
| 553 | """ |
Simon Glass | e21c27a | 2018-09-14 04:57:15 -0600 | [diff] [blame] | 554 | path = self.path + '/' + name |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 555 | return Node(self._fdt, self, None, name, path) |
| 556 | |
| 557 | def AddSubnode(self, name): |
| 558 | """Add a new subnode to the node, after all other subnodes |
| 559 | |
| 560 | Args: |
| 561 | name: name of node to add |
| 562 | |
| 563 | Returns: |
| 564 | New subnode that was created |
| 565 | """ |
| 566 | subnode = self.Subnode(name) |
Simon Glass | e21c27a | 2018-09-14 04:57:15 -0600 | [diff] [blame] | 567 | self.subnodes.append(subnode) |
| 568 | return subnode |
| 569 | |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 570 | def insert_subnode(self, name): |
| 571 | """Add a new subnode to the node, before all other subnodes |
| 572 | |
| 573 | This deletes other subnodes and sets their offset to None, so that they |
| 574 | will be recreated after this one. |
| 575 | |
| 576 | Args: |
| 577 | name: name of node to add |
| 578 | |
| 579 | Returns: |
| 580 | New subnode that was created |
| 581 | """ |
| 582 | # Deleting a node invalidates the offsets of all following nodes, so |
| 583 | # process in reverse order so that the offset of each node remains valid |
| 584 | # until deletion. |
| 585 | for subnode in reversed(self.subnodes): |
| 586 | subnode.purge(True) |
| 587 | subnode = self.Subnode(name) |
| 588 | self.subnodes.insert(0, subnode) |
| 589 | return subnode |
| 590 | |
| 591 | def purge(self, delete_it=False): |
| 592 | """Purge this node, setting offset to None and deleting from FDT""" |
| 593 | if self._offset is not None: |
| 594 | if delete_it: |
| 595 | CheckErr(self._fdt._fdt_obj.del_node(self.Offset()), |
| 596 | "Node '%s': delete" % self.path) |
| 597 | self._offset = None |
| 598 | self._fdt.Invalidate() |
| 599 | |
| 600 | for prop in self.props.values(): |
| 601 | prop.purge() |
| 602 | |
| 603 | for subnode in self.subnodes: |
| 604 | subnode.purge(False) |
| 605 | |
| 606 | def move_to_first(self): |
| 607 | """Move the current node to first in its parent's node list""" |
| 608 | parent = self.parent |
| 609 | if parent.subnodes and parent.subnodes[0] == self: |
| 610 | return |
| 611 | for subnode in reversed(parent.subnodes): |
| 612 | subnode.purge(True) |
| 613 | |
| 614 | new_subnodes = [self] |
| 615 | for subnode in parent.subnodes: |
| 616 | #subnode.purge(False) |
| 617 | if subnode != self: |
| 618 | new_subnodes.append(subnode) |
| 619 | parent.subnodes = new_subnodes |
| 620 | |
Simon Glass | a30c39f | 2022-02-08 11:49:51 -0700 | [diff] [blame] | 621 | def Delete(self): |
| 622 | """Delete a node |
| 623 | |
| 624 | The node is deleted and the offset cache is invalidated. |
| 625 | |
| 626 | Args: |
| 627 | node (Node): Node to delete |
| 628 | |
| 629 | Raises: |
| 630 | ValueError if the node does not exist |
| 631 | """ |
| 632 | CheckErr(self._fdt._fdt_obj.del_node(self.Offset()), |
| 633 | "Node '%s': delete" % self.path) |
| 634 | parent = self.parent |
| 635 | self._fdt.Invalidate() |
| 636 | parent.subnodes.remove(self) |
| 637 | |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 638 | def Sync(self, auto_resize=False): |
| 639 | """Sync node changes back to the device tree |
| 640 | |
| 641 | This updates the device tree blob with any changes to this node and its |
| 642 | subnodes since the last sync. |
| 643 | |
| 644 | Args: |
| 645 | auto_resize: Resize the device tree automatically if it does not |
| 646 | have enough space for the update |
| 647 | |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 648 | Returns: |
| 649 | True if the node had to be added, False if it already existed |
| 650 | |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 651 | Raises: |
| 652 | FdtException if auto_resize is False and there is not enough space |
| 653 | """ |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 654 | added = False |
Simon Glass | e21c27a | 2018-09-14 04:57:15 -0600 | [diff] [blame] | 655 | if self._offset is None: |
| 656 | # The subnode doesn't exist yet, so add it |
| 657 | fdt_obj = self._fdt._fdt_obj |
| 658 | if auto_resize: |
| 659 | while True: |
| 660 | offset = fdt_obj.add_subnode(self.parent._offset, self.name, |
| 661 | (libfdt.NOSPACE,)) |
| 662 | if offset != -libfdt.NOSPACE: |
| 663 | break |
| 664 | fdt_obj.resize(fdt_obj.totalsize() + 1024) |
| 665 | else: |
| 666 | offset = fdt_obj.add_subnode(self.parent._offset, self.name) |
| 667 | self._offset = offset |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 668 | added = True |
Simon Glass | e21c27a | 2018-09-14 04:57:15 -0600 | [diff] [blame] | 669 | |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 670 | # Sync the existing subnodes first, so that we can rely on the offsets |
| 671 | # being correct. As soon as we add new subnodes, it pushes all the |
| 672 | # existing subnodes up. |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 673 | for node in reversed(self.subnodes): |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 674 | if node._offset is not None: |
| 675 | node.Sync(auto_resize) |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 676 | |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 677 | # Sync subnodes in reverse so that we get the expected order. Each |
| 678 | # new node goes at the start of the subnode list. This avoids an O(n^2) |
| 679 | # rescan of node offsets. |
| 680 | num_added = 0 |
| 681 | for node in reversed(self.subnodes): |
| 682 | if node.Sync(auto_resize): |
| 683 | num_added += 1 |
| 684 | if num_added: |
| 685 | # Reorder our list of nodes to put the new ones first, since that's |
| 686 | # what libfdt does |
| 687 | old_count = len(self.subnodes) - num_added |
| 688 | subnodes = self.subnodes[old_count:] + self.subnodes[:old_count] |
| 689 | self.subnodes = subnodes |
| 690 | |
| 691 | # Sync properties now, whose offsets should not have been disturbed, |
| 692 | # since properties come before subnodes. This is done after all the |
| 693 | # subnode processing above, since updating properties can disturb the |
| 694 | # offsets of those subnodes. |
| 695 | # Properties are synced in reverse order, with new properties added |
| 696 | # before existing properties are synced. This ensures that the offsets |
| 697 | # of earlier properties are not disturbed. |
| 698 | # Note that new properties will have an offset of None here, which |
| 699 | # Python cannot sort against int. So use a large value instead so that |
| 700 | # new properties are added first. |
Simon Glass | 6351805 | 2019-05-17 22:00:38 -0600 | [diff] [blame] | 701 | prop_list = sorted(self.props.values(), |
| 702 | key=lambda prop: prop._offset or 1 << 31, |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 703 | reverse=True) |
| 704 | for prop in prop_list: |
| 705 | prop.Sync(auto_resize) |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 706 | return added |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 707 | |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 708 | def merge_props(self, src): |
| 709 | """Copy missing properties (except 'phandle') from another node |
| 710 | |
| 711 | Args: |
| 712 | src (Node): Node containing properties to copy |
| 713 | |
| 714 | Adds properties which are present in src but not in this node. Any |
| 715 | 'phandle' property is not copied since this might result in two nodes |
| 716 | with the same phandle, thus making phandle references ambiguous. |
| 717 | """ |
| 718 | for name, src_prop in src.props.items(): |
| 719 | if name != 'phandle' and name not in self.props: |
| 720 | self.props[name] = Prop(self, None, name, src_prop.bytes) |
| 721 | |
| 722 | def copy_node(self, src): |
| 723 | """Copy a node and all its subnodes into this node |
| 724 | |
| 725 | Args: |
| 726 | src (Node): Node to copy |
| 727 | |
| 728 | Returns: |
| 729 | Node: Resulting destination node |
| 730 | |
| 731 | This works recursively. |
| 732 | |
| 733 | The new node is put before all other nodes. If the node already |
| 734 | exists, just its subnodes and properties are copied, placing them before |
| 735 | any existing subnodes. Properties which exist in the destination node |
| 736 | already are not copied. |
| 737 | """ |
| 738 | dst = self.FindNode(src.name) |
| 739 | if dst: |
| 740 | dst.move_to_first() |
| 741 | else: |
| 742 | dst = self.insert_subnode(src.name) |
| 743 | dst.merge_props(src) |
| 744 | |
| 745 | # Process in reverse order so that they appear correctly in the result, |
| 746 | # since copy_node() puts the node first in the list |
| 747 | for node in reversed(src.subnodes): |
| 748 | dst.copy_node(node) |
| 749 | return dst |
| 750 | |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 751 | |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 752 | class Fdt: |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 753 | """Provides simple access to a flat device tree blob using libfdts. |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 754 | |
| 755 | Properties: |
| 756 | fname: Filename of fdt |
| 757 | _root: Root of device tree (a Node object) |
Simon Glass | 880e9ee | 2019-07-20 12:23:38 -0600 | [diff] [blame] | 758 | name: Helpful name for this Fdt for the user (useful when creating the |
| 759 | DT from data rather than a file) |
Simon Glass | a06a34b | 2016-07-25 18:59:04 -0600 | [diff] [blame] | 760 | """ |
| 761 | def __init__(self, fname): |
| 762 | self._fname = fname |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 763 | self._cached_offsets = False |
Simon Glass | 09264e0 | 2017-08-29 14:15:52 -0600 | [diff] [blame] | 764 | self.phandle_to_node = {} |
Simon Glass | 880e9ee | 2019-07-20 12:23:38 -0600 | [diff] [blame] | 765 | self.name = '' |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 766 | if self._fname: |
Simon Glass | 880e9ee | 2019-07-20 12:23:38 -0600 | [diff] [blame] | 767 | self.name = self._fname |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 768 | self._fname = fdt_util.EnsureCompiled(self._fname) |
| 769 | |
Simon Glass | 3e4b51e | 2019-05-14 15:53:43 -0600 | [diff] [blame] | 770 | with open(self._fname, 'rb') as fd: |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 771 | self._fdt_obj = libfdt.Fdt(fd.read()) |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 772 | |
Simon Glass | 746aee3 | 2018-09-14 04:57:17 -0600 | [diff] [blame] | 773 | @staticmethod |
Simon Glass | 880e9ee | 2019-07-20 12:23:38 -0600 | [diff] [blame] | 774 | def FromData(data, name=''): |
Simon Glass | 746aee3 | 2018-09-14 04:57:17 -0600 | [diff] [blame] | 775 | """Create a new Fdt object from the given data |
| 776 | |
| 777 | Args: |
| 778 | data: Device-tree data blob |
Simon Glass | 880e9ee | 2019-07-20 12:23:38 -0600 | [diff] [blame] | 779 | name: Helpful name for this Fdt for the user |
Simon Glass | 746aee3 | 2018-09-14 04:57:17 -0600 | [diff] [blame] | 780 | |
| 781 | Returns: |
| 782 | Fdt object containing the data |
| 783 | """ |
| 784 | fdt = Fdt(None) |
Simon Glass | f6b6481 | 2019-05-17 22:00:36 -0600 | [diff] [blame] | 785 | fdt._fdt_obj = libfdt.Fdt(bytes(data)) |
Simon Glass | 880e9ee | 2019-07-20 12:23:38 -0600 | [diff] [blame] | 786 | fdt.name = name |
Simon Glass | 746aee3 | 2018-09-14 04:57:17 -0600 | [diff] [blame] | 787 | return fdt |
| 788 | |
Simon Glass | 94a7c60 | 2018-07-17 13:25:46 -0600 | [diff] [blame] | 789 | def LookupPhandle(self, phandle): |
| 790 | """Look up a phandle |
| 791 | |
| 792 | Args: |
| 793 | phandle: Phandle to look up (int) |
| 794 | |
| 795 | Returns: |
| 796 | Node object the phandle points to |
| 797 | """ |
| 798 | return self.phandle_to_node.get(phandle) |
| 799 | |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 800 | def Scan(self, root='/'): |
| 801 | """Scan a device tree, building up a tree of Node objects |
| 802 | |
| 803 | This fills in the self._root property |
| 804 | |
| 805 | Args: |
| 806 | root: Ignored |
| 807 | |
| 808 | TODO(sjg@chromium.org): Implement the 'root' parameter |
| 809 | """ |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 810 | self._cached_offsets = True |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 811 | self._root = self.Node(self, None, 0, '/', '/') |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 812 | self._root.Scan() |
| 813 | |
| 814 | def GetRoot(self): |
| 815 | """Get the root Node of the device tree |
| 816 | |
| 817 | Returns: |
| 818 | The root Node object |
| 819 | """ |
| 820 | return self._root |
| 821 | |
| 822 | def GetNode(self, path): |
| 823 | """Look up a node from its path |
| 824 | |
| 825 | Args: |
| 826 | path: Path to look up, e.g. '/microcode/update@0' |
| 827 | Returns: |
| 828 | Node object, or None if not found |
| 829 | """ |
| 830 | node = self._root |
Simon Glass | b9066ff | 2018-07-06 10:27:30 -0600 | [diff] [blame] | 831 | parts = path.split('/') |
| 832 | if len(parts) < 2: |
| 833 | return None |
Simon Glass | e44bc83 | 2019-07-20 12:23:39 -0600 | [diff] [blame] | 834 | if len(parts) == 2 and parts[1] == '': |
| 835 | return node |
Simon Glass | b9066ff | 2018-07-06 10:27:30 -0600 | [diff] [blame] | 836 | for part in parts[1:]: |
Simon Glass | 1d85888 | 2018-07-17 13:25:41 -0600 | [diff] [blame] | 837 | node = node.FindNode(part) |
Simon Glass | f7a2aee | 2016-07-25 18:59:07 -0600 | [diff] [blame] | 838 | if not node: |
| 839 | return None |
| 840 | return node |
| 841 | |
Simon Glass | da5f749 | 2016-07-25 18:59:15 -0600 | [diff] [blame] | 842 | def Flush(self): |
| 843 | """Flush device tree changes back to the file |
| 844 | |
| 845 | If the device tree has changed in memory, write it back to the file. |
Simon Glass | da5f749 | 2016-07-25 18:59:15 -0600 | [diff] [blame] | 846 | """ |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 847 | with open(self._fname, 'wb') as fd: |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 848 | fd.write(self._fdt_obj.as_bytearray()) |
Simon Glass | da5f749 | 2016-07-25 18:59:15 -0600 | [diff] [blame] | 849 | |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 850 | def Sync(self, auto_resize=False): |
| 851 | """Make sure any DT changes are written to the blob |
| 852 | |
| 853 | Args: |
| 854 | auto_resize: Resize the device tree automatically if it does not |
| 855 | have enough space for the update |
| 856 | |
| 857 | Raises: |
| 858 | FdtException if auto_resize is False and there is not enough space |
| 859 | """ |
Simon Glass | 71719e1 | 2021-03-21 18:24:36 +1300 | [diff] [blame] | 860 | self.CheckCache() |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 861 | self._root.Sync(auto_resize) |
Simon Glass | 71719e1 | 2021-03-21 18:24:36 +1300 | [diff] [blame] | 862 | self.Refresh() |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 863 | |
Simon Glass | da5f749 | 2016-07-25 18:59:15 -0600 | [diff] [blame] | 864 | def Pack(self): |
| 865 | """Pack the device tree down to its minimum size |
| 866 | |
| 867 | When nodes and properties shrink or are deleted, wasted space can |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 868 | build up in the device tree binary. |
Simon Glass | da5f749 | 2016-07-25 18:59:15 -0600 | [diff] [blame] | 869 | """ |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 870 | CheckErr(self._fdt_obj.pack(), 'pack') |
Simon Glass | 71719e1 | 2021-03-21 18:24:36 +1300 | [diff] [blame] | 871 | self.Refresh() |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 872 | |
Simon Glass | 9606624 | 2018-07-06 10:27:27 -0600 | [diff] [blame] | 873 | def GetContents(self): |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 874 | """Get the contents of the FDT |
| 875 | |
| 876 | Returns: |
| 877 | The FDT contents as a string of bytes |
| 878 | """ |
Simon Glass | f6b6481 | 2019-05-17 22:00:36 -0600 | [diff] [blame] | 879 | return bytes(self._fdt_obj.as_bytearray()) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 880 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 881 | def GetFdtObj(self): |
| 882 | """Get the contents of the FDT |
| 883 | |
| 884 | Returns: |
| 885 | The FDT contents as a libfdt.Fdt object |
| 886 | """ |
| 887 | return self._fdt_obj |
| 888 | |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 889 | def GetProps(self, node): |
| 890 | """Get all properties from a node. |
| 891 | |
| 892 | Args: |
| 893 | node: Full path to node name to look in. |
| 894 | |
| 895 | Returns: |
| 896 | A dictionary containing all the properties, indexed by node name. |
| 897 | The entries are Prop objects. |
| 898 | |
| 899 | Raises: |
| 900 | ValueError: if the node does not exist. |
| 901 | """ |
| 902 | props_dict = {} |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 903 | poffset = self._fdt_obj.first_property_offset(node._offset, |
| 904 | QUIET_NOTFOUND) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 905 | while poffset >= 0: |
| 906 | p = self._fdt_obj.get_property_by_offset(poffset) |
Simon Glass | 3def0cf | 2018-07-06 10:27:20 -0600 | [diff] [blame] | 907 | prop = Prop(node, poffset, p.name, p) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 908 | props_dict[prop.name] = prop |
| 909 | |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 910 | poffset = self._fdt_obj.next_property_offset(poffset, |
| 911 | QUIET_NOTFOUND) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 912 | return props_dict |
| 913 | |
| 914 | def Invalidate(self): |
| 915 | """Mark our offset cache as invalid""" |
| 916 | self._cached_offsets = False |
| 917 | |
| 918 | def CheckCache(self): |
| 919 | """Refresh the offset cache if needed""" |
| 920 | if self._cached_offsets: |
| 921 | return |
| 922 | self.Refresh() |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 923 | |
| 924 | def Refresh(self): |
| 925 | """Refresh the offset cache""" |
| 926 | self._root.Refresh(0) |
Simon Glass | 71719e1 | 2021-03-21 18:24:36 +1300 | [diff] [blame] | 927 | self._cached_offsets = True |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 928 | |
| 929 | def GetStructOffset(self, offset): |
| 930 | """Get the file offset of a given struct offset |
| 931 | |
| 932 | Args: |
| 933 | offset: Offset within the 'struct' region of the device tree |
| 934 | Returns: |
| 935 | Position of @offset within the device tree binary |
| 936 | """ |
Simon Glass | 117f57b | 2018-07-06 10:27:26 -0600 | [diff] [blame] | 937 | return self._fdt_obj.off_dt_struct() + offset |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 938 | |
| 939 | @classmethod |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 940 | def Node(self, fdt, parent, offset, name, path): |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 941 | """Create a new node |
| 942 | |
| 943 | This is used by Fdt.Scan() to create a new node using the correct |
| 944 | class. |
| 945 | |
| 946 | Args: |
| 947 | fdt: Fdt object |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 948 | parent: Parent node, or None if this is the root node |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 949 | offset: Offset of node |
| 950 | name: Node name |
| 951 | path: Full path to node |
| 952 | """ |
Simon Glass | 979ab02 | 2017-08-29 14:15:47 -0600 | [diff] [blame] | 953 | node = Node(fdt, parent, offset, name, path) |
Simon Glass | 7b75b44 | 2017-05-27 07:38:28 -0600 | [diff] [blame] | 954 | return node |
Simon Glass | 99ed4a2 | 2017-05-27 07:38:30 -0600 | [diff] [blame] | 955 | |
Simon Glass | f6e0249 | 2019-07-20 12:24:08 -0600 | [diff] [blame] | 956 | def GetFilename(self): |
| 957 | """Get the filename of the device tree |
| 958 | |
| 959 | Returns: |
| 960 | String filename |
| 961 | """ |
| 962 | return self._fname |
| 963 | |
Simon Glass | 99ed4a2 | 2017-05-27 07:38:30 -0600 | [diff] [blame] | 964 | def FdtScan(fname): |
Simon Glass | dfe5f5b | 2018-07-06 10:27:32 -0600 | [diff] [blame] | 965 | """Returns a new Fdt object""" |
Simon Glass | 99ed4a2 | 2017-05-27 07:38:30 -0600 | [diff] [blame] | 966 | dtb = Fdt(fname) |
| 967 | dtb.Scan() |
| 968 | return dtb |