]>
Commit | Line | Data |
---|---|---|
ec564b47 SG |
1 | #!/usr/bin/python |
2 | # | |
3 | # Copyright (C) 2016 Google, Inc | |
4 | # Written by Simon Glass <sjg@chromium.org> | |
5 | # | |
6 | # SPDX-License-Identifier: GPL-2.0+ | |
7 | # | |
8 | ||
355c67c3 | 9 | import os |
ec564b47 | 10 | import struct |
c4c5f9ee | 11 | import sys |
355c67c3 SG |
12 | import tempfile |
13 | ||
14 | import command | |
15 | import tools | |
ec564b47 | 16 | |
ec564b47 SG |
17 | def fdt32_to_cpu(val): |
18 | """Convert a device tree cell to an integer | |
19 | ||
20 | Args: | |
21 | Value to convert (4-character string representing the cell value) | |
22 | ||
23 | Return: | |
24 | A native-endian integer value | |
25 | """ | |
c4c5f9ee | 26 | if sys.version_info > (3, 0): |
f156b5b5 GM |
27 | if isinstance(val, bytes): |
28 | val = val.decode('utf-8') | |
c4c5f9ee | 29 | val = val.encode('raw_unicode_escape') |
20024dae | 30 | return struct.unpack('>I', val)[0] |
355c67c3 | 31 | |
fbdfd228 SG |
32 | def fdt_cells_to_cpu(val, cells): |
33 | """Convert one or two cells to a long integer | |
34 | ||
35 | Args: | |
36 | Value to convert (array of one or more 4-character strings) | |
37 | ||
38 | Return: | |
39 | A native-endian long value | |
40 | """ | |
c20ee0ed SG |
41 | if not cells: |
42 | return 0 | |
fbdfd228 SG |
43 | out = long(fdt32_to_cpu(val[0])) |
44 | if cells == 2: | |
45 | out = out << 32 | fdt32_to_cpu(val[1]) | |
46 | return out | |
47 | ||
355c67c3 SG |
48 | def EnsureCompiled(fname): |
49 | """Compile an fdt .dts source file into a .dtb binary blob if needed. | |
50 | ||
51 | Args: | |
52 | fname: Filename (if .dts it will be compiled). It not it will be | |
53 | left alone | |
54 | ||
55 | Returns: | |
56 | Filename of resulting .dtb file | |
57 | """ | |
58 | _, ext = os.path.splitext(fname) | |
59 | if ext != '.dts': | |
60 | return fname | |
61 | ||
62 | dts_input = tools.GetOutputFilename('source.dts') | |
63 | dtb_output = tools.GetOutputFilename('source.dtb') | |
64 | ||
65 | search_paths = [os.path.join(os.getcwd(), 'include')] | |
66 | root, _ = os.path.splitext(fname) | |
67 | args = ['-E', '-P', '-x', 'assembler-with-cpp', '-D__ASSEMBLY__'] | |
68 | args += ['-Ulinux'] | |
69 | for path in search_paths: | |
70 | args.extend(['-I', path]) | |
71 | args += ['-o', dts_input, fname] | |
72 | command.Run('cc', *args) | |
73 | ||
74 | # If we don't have a directory, put it in the tools tempdir | |
75 | search_list = [] | |
76 | for path in search_paths: | |
77 | search_list.extend(['-i', path]) | |
d09682ef SG |
78 | args = ['-I', 'dts', '-o', dtb_output, '-O', 'dtb', |
79 | '-W', 'no-unit_address_vs_reg'] | |
355c67c3 SG |
80 | args.extend(search_list) |
81 | args.append(dts_input) | |
3ed0de31 SG |
82 | dtc = os.environ.get('DTC') or 'dtc' |
83 | command.Run(dtc, *args) | |
355c67c3 | 84 | return dtb_output |
8f224b37 SG |
85 | |
86 | def GetInt(node, propname, default=None): | |
87 | prop = node.props.get(propname) | |
88 | if not prop: | |
89 | return default | |
90 | value = fdt32_to_cpu(prop.value) | |
91 | if type(value) == type(list): | |
92 | raise ValueError("Node '%s' property '%' has list value: expecting" | |
93 | "a single integer" % (node.name, propname)) | |
94 | return value | |
95 | ||
96 | def GetString(node, propname, default=None): | |
97 | prop = node.props.get(propname) | |
98 | if not prop: | |
99 | return default | |
100 | value = prop.value | |
101 | if type(value) == type(list): | |
102 | raise ValueError("Node '%s' property '%' has list value: expecting" | |
103 | "a single string" % (node.name, propname)) | |
104 | return value | |
105 | ||
106 | def GetBool(node, propname, default=False): | |
107 | if propname in node.props: | |
108 | return True | |
109 | return default |