xref: /openbmc/u-boot/tools/dtoc/fdt_util.py (revision cd1cc31f) 
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
9import os
10import struct
11import sys
12import tempfile
13
14import command
15import tools
16
17def 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    """
26    if sys.version_info > (3, 0):
27        if isinstance(val, bytes):
28            val = val.decode('utf-8')
29        val = val.encode('raw_unicode_escape')
30    return struct.unpack('>I', val)[0]
31
32def 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    """
41    if not cells:
42        return 0
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
48def 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])
78    args = ['-I', 'dts', '-o', dtb_output, '-O', 'dtb',
79            '-W', 'no-unit_address_vs_reg']
80    args.extend(search_list)
81    args.append(dts_input)
82    dtc = os.environ.get('DTC') or 'dtc'
83    command.Run(dtc, *args)
84    return dtb_output
85
86def 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
96def 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
106def GetBool(node, propname, default=False):
107    if propname in node.props:
108        return True
109    return default
110