1ec564b47SSimon Glass#!/usr/bin/python 2ec564b47SSimon Glass# 3ec564b47SSimon Glass# Copyright (C) 2016 Google, Inc 4ec564b47SSimon Glass# Written by Simon Glass <sjg@chromium.org> 5ec564b47SSimon Glass# 6ec564b47SSimon Glass# SPDX-License-Identifier: GPL-2.0+ 7ec564b47SSimon Glass# 8ec564b47SSimon Glass 9355c67c3SSimon Glassimport os 10ec564b47SSimon Glassimport struct 11355c67c3SSimon Glassimport tempfile 12355c67c3SSimon Glass 13355c67c3SSimon Glassimport command 14355c67c3SSimon Glassimport tools 15ec564b47SSimon Glass 16ec564b47SSimon Glassdef fdt32_to_cpu(val): 17ec564b47SSimon Glass """Convert a device tree cell to an integer 18ec564b47SSimon Glass 19ec564b47SSimon Glass Args: 20ec564b47SSimon Glass Value to convert (4-character string representing the cell value) 21ec564b47SSimon Glass 22ec564b47SSimon Glass Return: 23ec564b47SSimon Glass A native-endian integer value 24ec564b47SSimon Glass """ 2520024daeSSimon Glass return struct.unpack('>I', val)[0] 26355c67c3SSimon Glass 27355c67c3SSimon Glassdef EnsureCompiled(fname): 28355c67c3SSimon Glass """Compile an fdt .dts source file into a .dtb binary blob if needed. 29355c67c3SSimon Glass 30355c67c3SSimon Glass Args: 31355c67c3SSimon Glass fname: Filename (if .dts it will be compiled). It not it will be 32355c67c3SSimon Glass left alone 33355c67c3SSimon Glass 34355c67c3SSimon Glass Returns: 35355c67c3SSimon Glass Filename of resulting .dtb file 36355c67c3SSimon Glass """ 37355c67c3SSimon Glass _, ext = os.path.splitext(fname) 38355c67c3SSimon Glass if ext != '.dts': 39355c67c3SSimon Glass return fname 40355c67c3SSimon Glass 41355c67c3SSimon Glass dts_input = tools.GetOutputFilename('source.dts') 42355c67c3SSimon Glass dtb_output = tools.GetOutputFilename('source.dtb') 43355c67c3SSimon Glass 44355c67c3SSimon Glass search_paths = [os.path.join(os.getcwd(), 'include')] 45355c67c3SSimon Glass root, _ = os.path.splitext(fname) 46355c67c3SSimon Glass args = ['-E', '-P', '-x', 'assembler-with-cpp', '-D__ASSEMBLY__'] 47355c67c3SSimon Glass args += ['-Ulinux'] 48355c67c3SSimon Glass for path in search_paths: 49355c67c3SSimon Glass args.extend(['-I', path]) 50355c67c3SSimon Glass args += ['-o', dts_input, fname] 51355c67c3SSimon Glass command.Run('cc', *args) 52355c67c3SSimon Glass 53355c67c3SSimon Glass # If we don't have a directory, put it in the tools tempdir 54355c67c3SSimon Glass search_list = [] 55355c67c3SSimon Glass for path in search_paths: 56355c67c3SSimon Glass search_list.extend(['-i', path]) 57355c67c3SSimon Glass args = ['-I', 'dts', '-o', dtb_output, '-O', 'dtb'] 58355c67c3SSimon Glass args.extend(search_list) 59355c67c3SSimon Glass args.append(dts_input) 60355c67c3SSimon Glass command.Run('dtc', *args) 61355c67c3SSimon Glass return dtb_output 62*8f224b37SSimon Glass 63*8f224b37SSimon Glassdef GetInt(node, propname, default=None): 64*8f224b37SSimon Glass prop = node.props.get(propname) 65*8f224b37SSimon Glass if not prop: 66*8f224b37SSimon Glass return default 67*8f224b37SSimon Glass value = fdt32_to_cpu(prop.value) 68*8f224b37SSimon Glass if type(value) == type(list): 69*8f224b37SSimon Glass raise ValueError("Node '%s' property '%' has list value: expecting" 70*8f224b37SSimon Glass "a single integer" % (node.name, propname)) 71*8f224b37SSimon Glass return value 72*8f224b37SSimon Glass 73*8f224b37SSimon Glassdef GetString(node, propname, default=None): 74*8f224b37SSimon Glass prop = node.props.get(propname) 75*8f224b37SSimon Glass if not prop: 76*8f224b37SSimon Glass return default 77*8f224b37SSimon Glass value = prop.value 78*8f224b37SSimon Glass if type(value) == type(list): 79*8f224b37SSimon Glass raise ValueError("Node '%s' property '%' has list value: expecting" 80*8f224b37SSimon Glass "a single string" % (node.name, propname)) 81*8f224b37SSimon Glass return value 82*8f224b37SSimon Glass 83*8f224b37SSimon Glassdef GetBool(node, propname, default=False): 84*8f224b37SSimon Glass if propname in node.props: 85*8f224b37SSimon Glass return True 86*8f224b37SSimon Glass return default 87