]>
Commit | Line | Data |
---|---|---|
4549e789 | 1 | /* SPDX-License-Identifier: GPL-2.0+ OR BSD-2-Clause */ |
cf2064d7 SG |
2 | /* |
3 | * pylibfdt - Flat Device Tree manipulation in Python | |
4 | * Copyright (C) 2017 Google, Inc. | |
5 | * Written by Simon Glass <[email protected]> | |
cf2064d7 SG |
6 | */ |
7 | ||
8 | %module libfdt | |
9 | ||
c79d18c4 SG |
10 | %include <stdint.i> |
11 | ||
cf2064d7 SG |
12 | %{ |
13 | #define SWIG_FILE_WITH_INIT | |
14 | #include "libfdt.h" | |
3def0cf2 SG |
15 | |
16 | /* | |
17 | * We rename this function here to avoid problems with swig, since we also have | |
18 | * a struct called fdt_property. That struct causes swig to create a class in | |
19 | * libfdt.py called fdt_property(), which confuses things. | |
20 | */ | |
50c59522 SG |
21 | static int fdt_property_stub(void *fdt, const char *name, const char *val, |
22 | int len) | |
3def0cf2 SG |
23 | { |
24 | return fdt_property(fdt, name, val, len); | |
25 | } | |
26 | ||
cf2064d7 SG |
27 | %} |
28 | ||
29 | %pythoncode %{ | |
30 | ||
31 | import struct | |
32 | ||
33 | # Error codes, corresponding to FDT_ERR_... in libfdt.h | |
34 | (NOTFOUND, | |
35 | EXISTS, | |
36 | NOSPACE, | |
37 | BADOFFSET, | |
38 | BADPATH, | |
39 | BADPHANDLE, | |
40 | BADSTATE, | |
41 | TRUNCATED, | |
42 | BADMAGIC, | |
43 | BADVERSION, | |
44 | BADSTRUCTURE, | |
45 | BADLAYOUT, | |
46 | INTERNAL, | |
47 | BADNCELLS, | |
48 | BADVALUE, | |
49 | BADOVERLAY, | |
50 | NOPHANDLES) = QUIET_ALL = range(1, 18) | |
51 | # QUIET_ALL can be passed as the 'quiet' parameter to avoid exceptions | |
52 | # altogether. All # functions passed this value will return an error instead | |
53 | # of raising an exception. | |
54 | ||
55 | # Pass this as the 'quiet' parameter to return -ENOTFOUND on NOTFOUND errors, | |
56 | # instead of raising an exception. | |
57 | QUIET_NOTFOUND = (NOTFOUND,) | |
50c59522 | 58 | QUIET_NOSPACE = (NOSPACE,) |
cf2064d7 SG |
59 | |
60 | ||
61 | class FdtException(Exception): | |
62 | """An exception caused by an error such as one of the codes above""" | |
63 | def __init__(self, err): | |
64 | self.err = err | |
65 | ||
66 | def __str__(self): | |
67 | return 'pylibfdt error %d: %s' % (self.err, fdt_strerror(self.err)) | |
68 | ||
69 | def strerror(fdt_err): | |
70 | """Get the string for an error number | |
71 | ||
72 | Args: | |
73 | fdt_err: Error number (-ve) | |
74 | ||
75 | Returns: | |
76 | String containing the associated error | |
77 | """ | |
78 | return fdt_strerror(fdt_err) | |
79 | ||
80 | def check_err(val, quiet=()): | |
81 | """Raise an error if the return value is -ve | |
82 | ||
83 | This is used to check for errors returned by libfdt C functions. | |
84 | ||
85 | Args: | |
86 | val: Return value from a libfdt function | |
87 | quiet: Errors to ignore (empty to raise on all errors) | |
88 | ||
89 | Returns: | |
90 | val if val >= 0 | |
91 | ||
92 | Raises | |
93 | FdtException if val < 0 | |
94 | """ | |
95 | if val < 0: | |
96 | if -val not in quiet: | |
97 | raise FdtException(val) | |
98 | return val | |
99 | ||
100 | def check_err_null(val, quiet=()): | |
101 | """Raise an error if the return value is NULL | |
102 | ||
103 | This is used to check for a NULL return value from certain libfdt C | |
104 | functions | |
105 | ||
106 | Args: | |
107 | val: Return value from a libfdt function | |
108 | quiet: Errors to ignore (empty to raise on all errors) | |
109 | ||
110 | Returns: | |
111 | val if val is a list, None if not | |
112 | ||
113 | Raises | |
114 | FdtException if val indicates an error was reported and the error | |
115 | is not in @quiet. | |
116 | """ | |
117 | # Normally a list is returned which contains the data and its length. | |
118 | # If we get just an integer error code, it means the function failed. | |
119 | if not isinstance(val, list): | |
120 | if -val not in quiet: | |
121 | raise FdtException(val) | |
122 | return val | |
123 | ||
50c59522 SG |
124 | class FdtRo(object): |
125 | """Class for a read-only device-tree | |
3def0cf2 | 126 | |
50c59522 SG |
127 | This is a base class used by FdtRw (read-write access) and FdtSw |
128 | (sequential-write access). It implements read-only access to the | |
129 | device tree. | |
cf2064d7 | 130 | |
50c59522 | 131 | Here are the three classes and when you should use them: |
cf2064d7 | 132 | |
50c59522 SG |
133 | FdtRo - read-only access to an existing FDT |
134 | FdtRw - read-write access to an existing FDT (most common case) | |
135 | FdtSw - for creating a new FDT, as well as allowing read-only access | |
cf2064d7 SG |
136 | """ |
137 | def __init__(self, data): | |
138 | self._fdt = bytearray(data) | |
139 | check_err(fdt_check_header(self._fdt)); | |
140 | ||
3def0cf2 SG |
141 | def as_bytearray(self): |
142 | """Get the device tree contents as a bytearray | |
143 | ||
144 | This can be passed directly to libfdt functions that access a | |
145 | const void * for the device tree. | |
146 | ||
147 | Returns: | |
148 | bytearray containing the device tree | |
149 | """ | |
150 | return bytearray(self._fdt) | |
151 | ||
152 | def next_node(self, nodeoffset, depth, quiet=()): | |
153 | """Find the next subnode | |
154 | ||
155 | Args: | |
156 | nodeoffset: Node offset of previous node | |
50c59522 SG |
157 | depth: The depth of the node at nodeoffset. This is used to |
158 | calculate the depth of the returned node | |
3def0cf2 SG |
159 | quiet: Errors to ignore (empty to raise on all errors) |
160 | ||
161 | Returns: | |
50c59522 SG |
162 | Typle: |
163 | Offset of the next node, if any, else a -ve error | |
164 | Depth of the returned node, if any, else undefined | |
3def0cf2 SG |
165 | |
166 | Raises: | |
167 | FdtException if no more nodes found or other error occurs | |
168 | """ | |
169 | return check_err(fdt_next_node(self._fdt, nodeoffset, depth), quiet) | |
170 | ||
171 | def first_subnode(self, nodeoffset, quiet=()): | |
172 | """Find the first subnode of a parent node | |
173 | ||
174 | Args: | |
175 | nodeoffset: Node offset of parent node | |
176 | quiet: Errors to ignore (empty to raise on all errors) | |
177 | ||
178 | Returns: | |
179 | The offset of the first subnode, if any | |
180 | ||
181 | Raises: | |
182 | FdtException if no subnodes found or other error occurs | |
183 | """ | |
184 | return check_err(fdt_first_subnode(self._fdt, nodeoffset), quiet) | |
185 | ||
186 | def next_subnode(self, nodeoffset, quiet=()): | |
187 | """Find the next subnode | |
188 | ||
189 | Args: | |
190 | nodeoffset: Node offset of previous subnode | |
191 | quiet: Errors to ignore (empty to raise on all errors) | |
192 | ||
193 | Returns: | |
194 | The offset of the next subnode, if any | |
195 | ||
196 | Raises: | |
197 | FdtException if no more subnodes found or other error occurs | |
198 | """ | |
199 | return check_err(fdt_next_subnode(self._fdt, nodeoffset), quiet) | |
200 | ||
201 | def magic(self): | |
202 | """Return the magic word from the header | |
203 | ||
204 | Returns: | |
205 | Magic word | |
206 | """ | |
50c59522 | 207 | return fdt_magic(self._fdt) |
3def0cf2 SG |
208 | |
209 | def totalsize(self): | |
210 | """Return the total size of the device tree | |
211 | ||
212 | Returns: | |
213 | Total tree size in bytes | |
214 | """ | |
50c59522 | 215 | return fdt_totalsize(self._fdt) |
3def0cf2 SG |
216 | |
217 | def off_dt_struct(self): | |
218 | """Return the start of the device-tree struct area | |
219 | ||
220 | Returns: | |
221 | Start offset of struct area | |
222 | """ | |
50c59522 | 223 | return fdt_off_dt_struct(self._fdt) |
3def0cf2 SG |
224 | |
225 | def off_dt_strings(self): | |
226 | """Return the start of the device-tree string area | |
227 | ||
228 | Returns: | |
229 | Start offset of string area | |
230 | """ | |
50c59522 | 231 | return fdt_off_dt_strings(self._fdt) |
3def0cf2 SG |
232 | |
233 | def off_mem_rsvmap(self): | |
234 | """Return the start of the memory reserve map | |
235 | ||
236 | Returns: | |
237 | Start offset of memory reserve map | |
238 | """ | |
50c59522 | 239 | return fdt_off_mem_rsvmap(self._fdt) |
3def0cf2 SG |
240 | |
241 | def version(self): | |
242 | """Return the version of the device tree | |
243 | ||
244 | Returns: | |
245 | Version number of the device tree | |
246 | """ | |
50c59522 | 247 | return fdt_version(self._fdt) |
3def0cf2 SG |
248 | |
249 | def last_comp_version(self): | |
250 | """Return the last compatible version of the device tree | |
251 | ||
252 | Returns: | |
253 | Last compatible version number of the device tree | |
254 | """ | |
50c59522 | 255 | return fdt_last_comp_version(self._fdt) |
3def0cf2 SG |
256 | |
257 | def boot_cpuid_phys(self): | |
258 | """Return the physical boot CPU ID | |
259 | ||
260 | Returns: | |
261 | Physical boot CPU ID | |
262 | """ | |
50c59522 | 263 | return fdt_boot_cpuid_phys(self._fdt) |
3def0cf2 SG |
264 | |
265 | def size_dt_strings(self): | |
266 | """Return the start of the device-tree string area | |
267 | ||
268 | Returns: | |
269 | Start offset of string area | |
270 | """ | |
50c59522 | 271 | return fdt_size_dt_strings(self._fdt) |
3def0cf2 SG |
272 | |
273 | def size_dt_struct(self): | |
274 | """Return the start of the device-tree struct area | |
275 | ||
276 | Returns: | |
277 | Start offset of struct area | |
278 | """ | |
50c59522 | 279 | return fdt_size_dt_struct(self._fdt) |
3def0cf2 SG |
280 | |
281 | def num_mem_rsv(self, quiet=()): | |
282 | """Return the number of memory reserve-map records | |
283 | ||
284 | Returns: | |
285 | Number of memory reserve-map records | |
286 | """ | |
287 | return check_err(fdt_num_mem_rsv(self._fdt), quiet) | |
288 | ||
289 | def get_mem_rsv(self, index, quiet=()): | |
290 | """Return the indexed memory reserve-map record | |
291 | ||
292 | Args: | |
293 | index: Record to return (0=first) | |
294 | ||
295 | Returns: | |
296 | Number of memory reserve-map records | |
297 | """ | |
298 | return check_err(fdt_get_mem_rsv(self._fdt, index), quiet) | |
299 | ||
93c94b88 SG |
300 | def subnode_offset(self, parentoffset, name, quiet=()): |
301 | """Get the offset of a named subnode | |
302 | ||
303 | Args: | |
304 | parentoffset: Offset of the parent node to check | |
305 | name: Name of the required subnode, e.g. 'subnode@1' | |
306 | quiet: Errors to ignore (empty to raise on all errors) | |
307 | ||
308 | Returns: | |
309 | The node offset of the found node, if any | |
310 | ||
311 | Raises | |
312 | FdtException if there is no node with that name, or other error | |
313 | """ | |
314 | return check_err(fdt_subnode_offset(self._fdt, parentoffset, name), | |
315 | quiet) | |
316 | ||
cf2064d7 SG |
317 | def path_offset(self, path, quiet=()): |
318 | """Get the offset for a given path | |
319 | ||
320 | Args: | |
321 | path: Path to the required node, e.g. '/node@3/subnode@1' | |
322 | quiet: Errors to ignore (empty to raise on all errors) | |
323 | ||
324 | Returns: | |
325 | Node offset | |
326 | ||
327 | Raises | |
328 | FdtException if the path is not valid or not found | |
329 | """ | |
330 | return check_err(fdt_path_offset(self._fdt, path), quiet) | |
331 | ||
3def0cf2 SG |
332 | def get_name(self, nodeoffset): |
333 | """Get the name of a node | |
334 | ||
335 | Args: | |
336 | nodeoffset: Offset of node to check | |
337 | ||
338 | Returns: | |
339 | Node name | |
340 | ||
341 | Raises: | |
342 | FdtException on error (e.g. nodeoffset is invalid) | |
343 | """ | |
344 | return check_err_null(fdt_get_name(self._fdt, nodeoffset))[0] | |
345 | ||
cf2064d7 SG |
346 | def first_property_offset(self, nodeoffset, quiet=()): |
347 | """Get the offset of the first property in a node offset | |
348 | ||
349 | Args: | |
350 | nodeoffset: Offset to the node to check | |
351 | quiet: Errors to ignore (empty to raise on all errors) | |
352 | ||
353 | Returns: | |
354 | Offset of the first property | |
355 | ||
356 | Raises | |
357 | FdtException if the associated node has no properties, or some | |
358 | other error occurred | |
359 | """ | |
360 | return check_err(fdt_first_property_offset(self._fdt, nodeoffset), | |
361 | quiet) | |
362 | ||
363 | def next_property_offset(self, prop_offset, quiet=()): | |
364 | """Get the next property in a node | |
365 | ||
366 | Args: | |
367 | prop_offset: Offset of the previous property | |
368 | quiet: Errors to ignore (empty to raise on all errors) | |
369 | ||
370 | Returns: | |
371 | Offset of the next property | |
372 | ||
373 | Raises: | |
374 | FdtException if the associated node has no more properties, or | |
375 | some other error occurred | |
376 | """ | |
377 | return check_err(fdt_next_property_offset(self._fdt, prop_offset), | |
378 | quiet) | |
379 | ||
cf2064d7 SG |
380 | def get_property_by_offset(self, prop_offset, quiet=()): |
381 | """Obtains a property that can be examined | |
382 | ||
383 | Args: | |
384 | prop_offset: Offset of property (e.g. from first_property_offset()) | |
385 | quiet: Errors to ignore (empty to raise on all errors) | |
386 | ||
387 | Returns: | |
388 | Property object, or None if not found | |
389 | ||
390 | Raises: | |
391 | FdtException on error (e.g. invalid prop_offset or device | |
392 | tree format) | |
393 | """ | |
394 | pdata = check_err_null( | |
395 | fdt_get_property_by_offset(self._fdt, prop_offset), quiet) | |
396 | if isinstance(pdata, (int)): | |
397 | return pdata | |
398 | return Property(pdata[0], pdata[1]) | |
399 | ||
50c59522 SG |
400 | def getprop(self, nodeoffset, prop_name, quiet=()): |
401 | """Get a property from a node | |
c640ed0c SG |
402 | |
403 | Args: | |
50c59522 | 404 | nodeoffset: Node offset containing property to get |
c640ed0c SG |
405 | prop_name: Name of property to get |
406 | quiet: Errors to ignore (empty to raise on all errors) | |
407 | ||
408 | Returns: | |
50c59522 SG |
409 | Value of property as a Property object (which can be used as a |
410 | bytearray/string), or -ve error number. On failure, returns an | |
411 | integer error | |
c640ed0c SG |
412 | |
413 | Raises: | |
50c59522 | 414 | FdtError if any error occurs (e.g. the property is not found) |
c640ed0c | 415 | """ |
50c59522 SG |
416 | pdata = check_err_null(fdt_getprop(self._fdt, nodeoffset, prop_name), |
417 | quiet) | |
c640ed0c SG |
418 | if isinstance(pdata, (int)): |
419 | return pdata | |
50c59522 SG |
420 | return Property(prop_name, bytearray(pdata[0])) |
421 | ||
422 | def get_phandle(self, nodeoffset): | |
423 | """Get the phandle of a node | |
424 | ||
425 | Args: | |
426 | nodeoffset: Node offset to check | |
427 | ||
428 | Returns: | |
429 | phandle of node, or 0 if the node has no phandle or another error | |
430 | occurs | |
431 | """ | |
432 | return fdt_get_phandle(self._fdt, nodeoffset) | |
433 | ||
434 | def parent_offset(self, nodeoffset, quiet=()): | |
435 | """Get the offset of a node's parent | |
436 | ||
437 | Args: | |
438 | nodeoffset: Node offset to check | |
439 | quiet: Errors to ignore (empty to raise on all errors) | |
440 | ||
441 | Returns: | |
442 | The offset of the parent node, if any | |
443 | ||
444 | Raises: | |
445 | FdtException if no parent found or other error occurs | |
446 | """ | |
447 | return check_err(fdt_parent_offset(self._fdt, nodeoffset), quiet) | |
448 | ||
449 | def node_offset_by_phandle(self, phandle, quiet=()): | |
450 | """Get the offset of a node with the given phandle | |
451 | ||
452 | Args: | |
453 | phandle: Phandle to search for | |
454 | quiet: Errors to ignore (empty to raise on all errors) | |
455 | ||
456 | Returns: | |
457 | The offset of node with that phandle, if any | |
458 | ||
459 | Raises: | |
460 | FdtException if no node found or other error occurs | |
461 | """ | |
462 | return check_err(fdt_node_offset_by_phandle(self._fdt, phandle), quiet) | |
463 | ||
464 | ||
465 | class Fdt(FdtRo): | |
466 | """Device tree class, supporting all operations | |
467 | ||
468 | The Fdt object is created is created from a device tree binary file, | |
469 | e.g. with something like: | |
470 | ||
471 | fdt = Fdt(open("filename.dtb").read()) | |
472 | ||
473 | Operations can then be performed using the methods in this class. Each | |
474 | method xxx(args...) corresponds to a libfdt function fdt_xxx(fdt, args...). | |
475 | ||
476 | All methods raise an FdtException if an error occurs. To avoid this | |
477 | behaviour a 'quiet' parameter is provided for some functions. This | |
478 | defaults to empty, but you can pass a list of errors that you expect. | |
479 | If one of these errors occurs, the function will return an error number | |
480 | (e.g. -NOTFOUND). | |
481 | """ | |
482 | def __init__(self, data): | |
483 | FdtRo.__init__(self, data) | |
c640ed0c | 484 | |
3def0cf2 SG |
485 | @staticmethod |
486 | def create_empty_tree(size, quiet=()): | |
487 | """Create an empty device tree ready for use | |
cf2064d7 SG |
488 | |
489 | Args: | |
3def0cf2 | 490 | size: Size of device tree in bytes |
cf2064d7 SG |
491 | |
492 | Returns: | |
3def0cf2 | 493 | Fdt object containing the device tree |
cf2064d7 | 494 | """ |
3def0cf2 SG |
495 | data = bytearray(size) |
496 | err = check_err(fdt_create_empty_tree(data, size), quiet) | |
497 | if err: | |
498 | return err | |
499 | return Fdt(data) | |
cf2064d7 | 500 | |
50c59522 | 501 | def resize(self, size, quiet=()): |
3def0cf2 | 502 | """Move the device tree into a larger or smaller space |
cf2064d7 | 503 | |
3def0cf2 SG |
504 | This creates a new device tree of size @size and moves the existing |
505 | device tree contents over to that. It can be used to create more space | |
50c59522 SG |
506 | in a device tree. Note that the Fdt object remains the same, but it |
507 | now has a new bytearray holding the contents. | |
cf2064d7 | 508 | |
3def0cf2 SG |
509 | Args: |
510 | size: Required new size of device tree in bytes | |
cf2064d7 | 511 | """ |
3def0cf2 | 512 | fdt = bytearray(size) |
3def0cf2 SG |
513 | err = check_err(fdt_open_into(self._fdt, fdt, size), quiet) |
514 | if err: | |
515 | return err | |
516 | self._fdt = fdt | |
cf2064d7 SG |
517 | |
518 | def pack(self, quiet=()): | |
519 | """Pack the device tree to remove unused space | |
520 | ||
521 | This adjusts the tree in place. | |
522 | ||
523 | Args: | |
524 | quiet: Errors to ignore (empty to raise on all errors) | |
525 | ||
50c59522 SG |
526 | Returns: |
527 | Error code, or 0 if OK | |
528 | ||
cf2064d7 SG |
529 | Raises: |
530 | FdtException if any error occurs | |
531 | """ | |
a1e00855 SG |
532 | err = check_err(fdt_pack(self._fdt), quiet) |
533 | if err: | |
534 | return err | |
535 | del self._fdt[self.totalsize():] | |
536 | return err | |
cf2064d7 | 537 | |
3def0cf2 SG |
538 | def set_name(self, nodeoffset, name, quiet=()): |
539 | """Set the name of a node | |
540 | ||
541 | Args: | |
542 | nodeoffset: Node offset of node to update | |
50c59522 | 543 | name: New node name (string without \0) |
3def0cf2 SG |
544 | |
545 | Returns: | |
546 | Error code, or 0 if OK | |
547 | ||
548 | Raises: | |
549 | FdtException if no parent found or other error occurs | |
550 | """ | |
50c59522 SG |
551 | if chr(0) in name: |
552 | raise ValueError('Property contains embedded nul characters') | |
3def0cf2 SG |
553 | return check_err(fdt_set_name(self._fdt, nodeoffset, name), quiet) |
554 | ||
555 | def setprop(self, nodeoffset, prop_name, val, quiet=()): | |
556 | """Set the value of a property | |
557 | ||
558 | Args: | |
559 | nodeoffset: Node offset containing the property to create/update | |
560 | prop_name: Name of property | |
561 | val: Value to write (string or bytearray) | |
562 | quiet: Errors to ignore (empty to raise on all errors) | |
563 | ||
564 | Returns: | |
565 | Error code, or 0 if OK | |
566 | ||
567 | Raises: | |
568 | FdtException if no parent found or other error occurs | |
569 | """ | |
570 | return check_err(fdt_setprop(self._fdt, nodeoffset, prop_name, val, | |
571 | len(val)), quiet) | |
572 | ||
573 | def setprop_u32(self, nodeoffset, prop_name, val, quiet=()): | |
574 | """Set the value of a property | |
575 | ||
576 | Args: | |
577 | nodeoffset: Node offset containing the property to create/update | |
578 | prop_name: Name of property | |
579 | val: Value to write (integer) | |
580 | quiet: Errors to ignore (empty to raise on all errors) | |
581 | ||
582 | Returns: | |
583 | Error code, or 0 if OK | |
584 | ||
585 | Raises: | |
586 | FdtException if no parent found or other error occurs | |
587 | """ | |
588 | return check_err(fdt_setprop_u32(self._fdt, nodeoffset, prop_name, val), | |
589 | quiet) | |
590 | ||
591 | def setprop_u64(self, nodeoffset, prop_name, val, quiet=()): | |
592 | """Set the value of a property | |
593 | ||
594 | Args: | |
595 | nodeoffset: Node offset containing the property to create/update | |
596 | prop_name: Name of property | |
597 | val: Value to write (integer) | |
598 | quiet: Errors to ignore (empty to raise on all errors) | |
599 | ||
600 | Returns: | |
601 | Error code, or 0 if OK | |
602 | ||
603 | Raises: | |
604 | FdtException if no parent found or other error occurs | |
605 | """ | |
606 | return check_err(fdt_setprop_u64(self._fdt, nodeoffset, prop_name, val), | |
607 | quiet) | |
608 | ||
609 | def setprop_str(self, nodeoffset, prop_name, val, quiet=()): | |
610 | """Set the string value of a property | |
611 | ||
612 | The property is set to the string, with a nul terminator added | |
613 | ||
614 | Args: | |
615 | nodeoffset: Node offset containing the property to create/update | |
616 | prop_name: Name of property | |
50c59522 SG |
617 | val: Value to write (string without nul terminator). Unicode is |
618 | supposed by encoding to UTF-8 | |
3def0cf2 SG |
619 | quiet: Errors to ignore (empty to raise on all errors) |
620 | ||
621 | Returns: | |
622 | Error code, or 0 if OK | |
623 | ||
624 | Raises: | |
625 | FdtException if no parent found or other error occurs | |
626 | """ | |
50c59522 | 627 | val = val.encode('utf-8') + '\0' |
3def0cf2 SG |
628 | return check_err(fdt_setprop(self._fdt, nodeoffset, prop_name, |
629 | val, len(val)), quiet) | |
630 | ||
34967c26 | 631 | def delprop(self, nodeoffset, prop_name, quiet=()): |
3def0cf2 SG |
632 | """Delete a property from a node |
633 | ||
634 | Args: | |
635 | nodeoffset: Node offset containing property to delete | |
636 | prop_name: Name of property to delete | |
34967c26 SG |
637 | quiet: Errors to ignore (empty to raise on all errors) |
638 | ||
639 | Returns: | |
640 | Error code, or 0 if OK | |
3def0cf2 SG |
641 | |
642 | Raises: | |
643 | FdtError if the property does not exist, or another error occurs | |
644 | """ | |
34967c26 SG |
645 | return check_err(fdt_delprop(self._fdt, nodeoffset, prop_name), quiet) |
646 | ||
647 | def add_subnode(self, parentoffset, name, quiet=()): | |
648 | """Add a new subnode to a node | |
649 | ||
650 | Args: | |
651 | parentoffset: Parent offset to add the subnode to | |
652 | name: Name of node to add | |
653 | ||
654 | Returns: | |
655 | offset of the node created, or negative error code on failure | |
656 | ||
657 | Raises: | |
658 | FdtError if there is not enough space, or another error occurs | |
659 | """ | |
660 | return check_err(fdt_add_subnode(self._fdt, parentoffset, name), quiet) | |
3def0cf2 | 661 | |
34967c26 | 662 | def del_node(self, nodeoffset, quiet=()): |
c640ed0c SG |
663 | """Delete a node |
664 | ||
665 | Args: | |
34967c26 SG |
666 | nodeoffset: Offset of node to delete |
667 | ||
668 | Returns: | |
669 | Error code, or 0 if OK | |
c640ed0c SG |
670 | |
671 | Raises: | |
34967c26 | 672 | FdtError if an error occurs |
c640ed0c | 673 | """ |
34967c26 | 674 | return check_err(fdt_del_node(self._fdt, nodeoffset), quiet) |
c640ed0c | 675 | |
3def0cf2 SG |
676 | |
677 | class Property(bytearray): | |
cf2064d7 SG |
678 | """Holds a device tree property name and value. |
679 | ||
680 | This holds a copy of a property taken from the device tree. It does not | |
681 | reference the device tree, so if anything changes in the device tree, | |
682 | a Property object will remain valid. | |
683 | ||
684 | Properties: | |
685 | name: Property name | |
3def0cf2 | 686 | value: Property value as a bytearray |
cf2064d7 SG |
687 | """ |
688 | def __init__(self, name, value): | |
3def0cf2 | 689 | bytearray.__init__(self, value) |
cf2064d7 | 690 | self.name = name |
3def0cf2 SG |
691 | |
692 | def as_cell(self, fmt): | |
693 | return struct.unpack('>' + fmt, self)[0] | |
694 | ||
695 | def as_uint32(self): | |
696 | return self.as_cell('L') | |
697 | ||
698 | def as_int32(self): | |
699 | return self.as_cell('l') | |
700 | ||
701 | def as_uint64(self): | |
702 | return self.as_cell('Q') | |
703 | ||
704 | def as_int64(self): | |
705 | return self.as_cell('q') | |
706 | ||
707 | def as_str(self): | |
50c59522 SG |
708 | """Unicode is supported by decoding from UTF-8""" |
709 | if self[-1] != 0: | |
710 | raise ValueError('Property lacks nul termination') | |
711 | if 0 in self[:-1]: | |
712 | raise ValueError('Property contains embedded nul characters') | |
713 | return self[:-1].decode('utf-8') | |
3def0cf2 SG |
714 | |
715 | ||
50c59522 | 716 | class FdtSw(FdtRo): |
3def0cf2 SG |
717 | """Software interface to create a device tree from scratch |
718 | ||
719 | The methods in this class work by adding to an existing 'partial' device | |
720 | tree buffer of a fixed size created by instantiating this class. When the | |
50c59522 | 721 | tree is complete, call as_fdt() to obtain a device tree ready to be used. |
3def0cf2 SG |
722 | |
723 | Similarly with nodes, a new node is started with begin_node() and finished | |
724 | with end_node(). | |
725 | ||
726 | The context manager functions can be used to make this a bit easier: | |
727 | ||
728 | # First create the device tree with a node and property: | |
50c59522 SG |
729 | sw = FdtSw() |
730 | with sw.add_node('node'): | |
731 | sw.property_u32('reg', 2) | |
732 | fdt = sw.as_fdt() | |
3def0cf2 SG |
733 | |
734 | # Now we can use it as a real device tree | |
735 | fdt.setprop_u32(0, 'reg', 3) | |
50c59522 SG |
736 | |
737 | The size hint provides a starting size for the space to be used by the | |
738 | device tree. This will be increased automatically as needed as new items | |
739 | are added to the tree. | |
3def0cf2 | 740 | """ |
50c59522 | 741 | INC_SIZE = 1024 # Expand size by this much when out of space |
3def0cf2 | 742 | |
50c59522 SG |
743 | def __init__(self, size_hint=None): |
744 | """Create a new FdtSw object | |
3def0cf2 | 745 | |
50c59522 SG |
746 | Args: |
747 | size_hint: A hint as to the initial size to use | |
748 | ||
749 | Raises: | |
750 | ValueError if size_hint is negative | |
3def0cf2 | 751 | |
50c59522 SG |
752 | Returns: |
753 | FdtSw object on success, else integer error code (if not raising) | |
754 | """ | |
755 | if not size_hint: | |
756 | size_hint = self.INC_SIZE | |
757 | fdtsw = bytearray(size_hint) | |
758 | err = check_err(fdt_create(fdtsw, size_hint)) | |
759 | if err: | |
760 | return err | |
761 | self._fdt = fdtsw | |
762 | ||
763 | def as_fdt(self): | |
3def0cf2 SG |
764 | """Convert a FdtSw into an Fdt so it can be accessed as normal |
765 | ||
50c59522 SG |
766 | Creates a new Fdt object from the work-in-progress device tree. This |
767 | does not call fdt_finish() on the current object, so it is possible to | |
768 | add more nodes/properties and call as_fdt() again to get an updated | |
769 | tree. | |
3def0cf2 SG |
770 | |
771 | Returns: | |
772 | Fdt object allowing access to the newly created device tree | |
773 | """ | |
50c59522 SG |
774 | fdtsw = bytearray(self._fdt) |
775 | check_err(fdt_finish(fdtsw)) | |
776 | return Fdt(fdtsw) | |
3def0cf2 | 777 | |
50c59522 SG |
778 | def check_space(self, val): |
779 | """Check if we need to add more space to the FDT | |
780 | ||
781 | This should be called with the error code from an operation. If this is | |
782 | -NOSPACE then the FDT will be expanded to have more space, and True will | |
783 | be returned, indicating that the operation needs to be tried again. | |
784 | ||
785 | Args: | |
786 | val: Return value from the operation that was attempted | |
787 | ||
788 | Returns: | |
789 | True if the operation must be retried, else False | |
790 | """ | |
791 | if check_err(val, QUIET_NOSPACE) < 0: | |
792 | self.resize(len(self._fdt) + self.INC_SIZE) | |
793 | return True | |
794 | return False | |
795 | ||
796 | def resize(self, size): | |
3def0cf2 SG |
797 | """Resize the buffer to accommodate a larger tree |
798 | ||
799 | Args: | |
800 | size: New size of tree | |
3def0cf2 SG |
801 | |
802 | Raises: | |
50c59522 | 803 | FdtException on any error |
3def0cf2 SG |
804 | """ |
805 | fdt = bytearray(size) | |
50c59522 SG |
806 | err = check_err(fdt_resize(self._fdt, fdt, size)) |
807 | self._fdt = fdt | |
3def0cf2 | 808 | |
50c59522 | 809 | def add_reservemap_entry(self, addr, size): |
3def0cf2 SG |
810 | """Add a new memory reserve map entry |
811 | ||
812 | Once finished adding, you must call finish_reservemap(). | |
813 | ||
814 | Args: | |
815 | addr: 64-bit start address | |
816 | size: 64-bit size | |
3def0cf2 SG |
817 | |
818 | Raises: | |
50c59522 | 819 | FdtException on any error |
3def0cf2 | 820 | """ |
50c59522 SG |
821 | while self.check_space(fdt_add_reservemap_entry(self._fdt, addr, |
822 | size)): | |
823 | pass | |
3def0cf2 | 824 | |
50c59522 | 825 | def finish_reservemap(self): |
3def0cf2 SG |
826 | """Indicate that there are no more reserve map entries to add |
827 | ||
3def0cf2 | 828 | Raises: |
50c59522 | 829 | FdtException on any error |
3def0cf2 | 830 | """ |
50c59522 SG |
831 | while self.check_space(fdt_finish_reservemap(self._fdt)): |
832 | pass | |
3def0cf2 | 833 | |
50c59522 | 834 | def begin_node(self, name): |
3def0cf2 SG |
835 | """Begin a new node |
836 | ||
837 | Use this before adding properties to the node. Then call end_node() to | |
838 | finish it. You can also use the context manager as shown in the FdtSw | |
839 | class comment. | |
840 | ||
841 | Args: | |
842 | name: Name of node to begin | |
3def0cf2 SG |
843 | |
844 | Raises: | |
50c59522 | 845 | FdtException on any error |
3def0cf2 | 846 | """ |
50c59522 SG |
847 | while self.check_space(fdt_begin_node(self._fdt, name)): |
848 | pass | |
3def0cf2 | 849 | |
50c59522 | 850 | def property_string(self, name, string): |
3def0cf2 SG |
851 | """Add a property with a string value |
852 | ||
853 | The string will be nul-terminated when written to the device tree | |
854 | ||
855 | Args: | |
856 | name: Name of property to add | |
857 | string: String value of property | |
3def0cf2 SG |
858 | |
859 | Raises: | |
50c59522 | 860 | FdtException on any error |
3def0cf2 | 861 | """ |
50c59522 SG |
862 | while self.check_space(fdt_property_string(self._fdt, name, string)): |
863 | pass | |
3def0cf2 | 864 | |
50c59522 | 865 | def property_u32(self, name, val): |
3def0cf2 SG |
866 | """Add a property with a 32-bit value |
867 | ||
868 | Write a single-cell value to the device tree | |
869 | ||
870 | Args: | |
871 | name: Name of property to add | |
872 | val: Value of property | |
3def0cf2 SG |
873 | |
874 | Raises: | |
50c59522 | 875 | FdtException on any error |
3def0cf2 | 876 | """ |
50c59522 SG |
877 | while self.check_space(fdt_property_u32(self._fdt, name, val)): |
878 | pass | |
3def0cf2 | 879 | |
50c59522 | 880 | def property_u64(self, name, val): |
3def0cf2 SG |
881 | """Add a property with a 64-bit value |
882 | ||
883 | Write a double-cell value to the device tree in big-endian format | |
884 | ||
885 | Args: | |
886 | name: Name of property to add | |
887 | val: Value of property | |
3def0cf2 SG |
888 | |
889 | Raises: | |
50c59522 | 890 | FdtException on any error |
3def0cf2 | 891 | """ |
50c59522 SG |
892 | while self.check_space(fdt_property_u64(self._fdt, name, val)): |
893 | pass | |
3def0cf2 | 894 | |
50c59522 | 895 | def property_cell(self, name, val): |
3def0cf2 SG |
896 | """Add a property with a single-cell value |
897 | ||
898 | Write a single-cell value to the device tree | |
899 | ||
900 | Args: | |
901 | name: Name of property to add | |
902 | val: Value of property | |
903 | quiet: Errors to ignore (empty to raise on all errors) | |
904 | ||
905 | Raises: | |
50c59522 | 906 | FdtException on any error |
3def0cf2 | 907 | """ |
50c59522 SG |
908 | while self.check_space(fdt_property_cell(self._fdt, name, val)): |
909 | pass | |
3def0cf2 | 910 | |
50c59522 | 911 | def property(self, name, val): |
3def0cf2 SG |
912 | """Add a property |
913 | ||
914 | Write a new property with the given value to the device tree. The value | |
915 | is taken as is and is not nul-terminated | |
916 | ||
917 | Args: | |
918 | name: Name of property to add | |
919 | val: Value of property | |
920 | quiet: Errors to ignore (empty to raise on all errors) | |
921 | ||
922 | Raises: | |
50c59522 | 923 | FdtException on any error |
3def0cf2 | 924 | """ |
50c59522 SG |
925 | while self.check_space(fdt_property_stub(self._fdt, name, val, |
926 | len(val))): | |
927 | pass | |
3def0cf2 | 928 | |
50c59522 | 929 | def end_node(self): |
3def0cf2 SG |
930 | """End a node |
931 | ||
932 | Use this after adding properties to a node to close it off. You can also | |
933 | use the context manager as shown in the FdtSw class comment. | |
934 | ||
935 | Args: | |
936 | quiet: Errors to ignore (empty to raise on all errors) | |
937 | ||
938 | Raises: | |
50c59522 | 939 | FdtException on any error |
3def0cf2 | 940 | """ |
50c59522 SG |
941 | while self.check_space(fdt_end_node(self._fdt)): |
942 | pass | |
3def0cf2 | 943 | |
50c59522 | 944 | def add_node(self, name): |
3def0cf2 SG |
945 | """Create a new context for adding a node |
946 | ||
947 | When used in a 'with' clause this starts a new node and finishes it | |
948 | afterward. | |
949 | ||
950 | Args: | |
951 | name: Name of node to add | |
952 | """ | |
50c59522 | 953 | return NodeAdder(self, name) |
3def0cf2 SG |
954 | |
955 | ||
956 | class NodeAdder(): | |
957 | """Class to provide a node context | |
958 | ||
959 | This allows you to add nodes in a more natural way: | |
960 | ||
50c59522 | 961 | with fdtsw.add_node('name'): |
3def0cf2 SG |
962 | fdtsw.property_string('test', 'value') |
963 | ||
964 | The node is automatically completed with a call to end_node() when the | |
965 | context exits. | |
966 | """ | |
50c59522 SG |
967 | def __init__(self, fdtsw, name): |
968 | self._fdt = fdtsw | |
3def0cf2 SG |
969 | self._name = name |
970 | ||
971 | def __enter__(self): | |
50c59522 | 972 | self._fdt.begin_node(self._name) |
3def0cf2 SG |
973 | |
974 | def __exit__(self, type, value, traceback): | |
50c59522 | 975 | self._fdt.end_node() |
cf2064d7 SG |
976 | %} |
977 | ||
978 | %rename(fdt_property) fdt_property_func; | |
979 | ||
50c59522 SG |
980 | /* |
981 | * fdt32_t is a big-endian 32-bit value defined to uint32_t in libfdt_env.h | |
982 | * so use the same type here. | |
983 | */ | |
984 | typedef uint32_t fdt32_t; | |
cf2064d7 SG |
985 | |
986 | %include "libfdt/fdt.h" | |
987 | ||
988 | %include "typemaps.i" | |
989 | ||
990 | /* Most functions don't change the device tree, so use a const void * */ | |
991 | %typemap(in) (const void *)(const void *fdt) { | |
992 | if (!PyByteArray_Check($input)) { | |
993 | SWIG_exception_fail(SWIG_TypeError, "in method '" "$symname" | |
994 | "', argument " "$argnum"" of type '" "$type""'"); | |
995 | } | |
996 | $1 = (void *)PyByteArray_AsString($input); | |
997 | fdt = $1; | |
998 | fdt = fdt; /* avoid unused variable warning */ | |
999 | } | |
1000 | ||
1001 | /* Some functions do change the device tree, so use void * */ | |
1002 | %typemap(in) (void *)(const void *fdt) { | |
1003 | if (!PyByteArray_Check($input)) { | |
1004 | SWIG_exception_fail(SWIG_TypeError, "in method '" "$symname" | |
1005 | "', argument " "$argnum"" of type '" "$type""'"); | |
1006 | } | |
1007 | $1 = PyByteArray_AsString($input); | |
1008 | fdt = $1; | |
1009 | fdt = fdt; /* avoid unused variable warning */ | |
1010 | } | |
1011 | ||
3def0cf2 | 1012 | /* typemap used for fdt_get_property_by_offset() */ |
cf2064d7 SG |
1013 | %typemap(out) (struct fdt_property *) { |
1014 | PyObject *buff; | |
1015 | ||
1016 | if ($1) { | |
1017 | resultobj = PyString_FromString( | |
1018 | fdt_string(fdt1, fdt32_to_cpu($1->nameoff))); | |
1019 | buff = PyByteArray_FromStringAndSize( | |
1020 | (const char *)($1 + 1), fdt32_to_cpu($1->len)); | |
1021 | resultobj = SWIG_Python_AppendOutput(resultobj, buff); | |
1022 | } | |
1023 | } | |
1024 | ||
1025 | %apply int *OUTPUT { int *lenp }; | |
1026 | ||
1027 | /* typemap used for fdt_getprop() */ | |
1028 | %typemap(out) (const void *) { | |
1029 | if (!$1) | |
1030 | $result = Py_None; | |
1031 | else | |
1032 | $result = Py_BuildValue("s#", $1, *arg4); | |
1033 | } | |
1034 | ||
3def0cf2 SG |
1035 | /* typemap used for fdt_setprop() */ |
1036 | %typemap(in) (const void *val) { | |
1037 | $1 = PyString_AsString($input); /* char *str */ | |
1038 | } | |
1039 | ||
1040 | /* typemap used for fdt_add_reservemap_entry() */ | |
1041 | %typemap(in) uint64_t { | |
1042 | $1 = PyLong_AsUnsignedLong($input); | |
1043 | } | |
1044 | ||
1045 | /* typemaps used for fdt_next_node() */ | |
1046 | %typemap(in, numinputs=1) int *depth (int depth) { | |
1047 | depth = (int) PyInt_AsLong($input); | |
1048 | $1 = &depth; | |
1049 | } | |
1050 | ||
1051 | %typemap(argout) int *depth { | |
1052 | PyObject *val = Py_BuildValue("i", *arg$argnum); | |
1053 | resultobj = SWIG_Python_AppendOutput(resultobj, val); | |
1054 | } | |
1055 | ||
1056 | %apply int *depth { int *depth }; | |
1057 | ||
1058 | /* typemaps for fdt_get_mem_rsv */ | |
1059 | %typemap(in, numinputs=0) uint64_t * (uint64_t temp) { | |
1060 | $1 = &temp; | |
1061 | } | |
1062 | ||
1063 | %typemap(argout) uint64_t * { | |
1064 | PyObject *val = PyLong_FromUnsignedLong(*arg$argnum); | |
1065 | if (!result) { | |
1066 | if (PyTuple_GET_SIZE(resultobj) == 0) | |
1067 | resultobj = val; | |
1068 | else | |
1069 | resultobj = SWIG_Python_AppendOutput(resultobj, val); | |
1070 | } | |
1071 | } | |
1072 | ||
cf2064d7 SG |
1073 | /* We have both struct fdt_property and a function fdt_property() */ |
1074 | %warnfilter(302) fdt_property; | |
1075 | ||
1076 | /* These are macros in the header so have to be redefined here */ | |
50c59522 SG |
1077 | uint32_t fdt_magic(const void *fdt); |
1078 | uint32_t fdt_totalsize(const void *fdt); | |
1079 | uint32_t fdt_off_dt_struct(const void *fdt); | |
1080 | uint32_t fdt_off_dt_strings(const void *fdt); | |
1081 | uint32_t fdt_off_mem_rsvmap(const void *fdt); | |
1082 | uint32_t fdt_version(const void *fdt); | |
1083 | uint32_t fdt_last_comp_version(const void *fdt); | |
1084 | uint32_t fdt_boot_cpuid_phys(const void *fdt); | |
1085 | uint32_t fdt_size_dt_strings(const void *fdt); | |
1086 | uint32_t fdt_size_dt_struct(const void *fdt); | |
1087 | ||
3def0cf2 SG |
1088 | int fdt_property_string(void *fdt, const char *name, const char *val); |
1089 | int fdt_property_cell(void *fdt, const char *name, uint32_t val); | |
1090 | ||
1091 | /* | |
1092 | * This function has a stub since the name fdt_property is used for both a | |
1093 | * function and a struct, which confuses SWIG. | |
1094 | */ | |
50c59522 | 1095 | int fdt_property_stub(void *fdt, const char *name, const char *val, int len); |
cf2064d7 SG |
1096 | |
1097 | %include <../libfdt/libfdt.h> |