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 11*c4c5f9eeSPaul Burtonimport sys 12355c67c3SSimon Glassimport tempfile 13355c67c3SSimon Glass 14355c67c3SSimon Glassimport command 15355c67c3SSimon Glassimport tools 16ec564b47SSimon Glass 17ec564b47SSimon Glassdef fdt32_to_cpu(val): 18ec564b47SSimon Glass """Convert a device tree cell to an integer 19ec564b47SSimon Glass 20ec564b47SSimon Glass Args: 21ec564b47SSimon Glass Value to convert (4-character string representing the cell value) 22ec564b47SSimon Glass 23ec564b47SSimon Glass Return: 24ec564b47SSimon Glass A native-endian integer value 25ec564b47SSimon Glass """ 26*c4c5f9eeSPaul Burton if sys.version_info > (3, 0): 27*c4c5f9eeSPaul Burton val = val.encode('raw_unicode_escape') 2820024daeSSimon Glass return struct.unpack('>I', val)[0] 29355c67c3SSimon Glass 30355c67c3SSimon Glassdef EnsureCompiled(fname): 31355c67c3SSimon Glass """Compile an fdt .dts source file into a .dtb binary blob if needed. 32355c67c3SSimon Glass 33355c67c3SSimon Glass Args: 34355c67c3SSimon Glass fname: Filename (if .dts it will be compiled). It not it will be 35355c67c3SSimon Glass left alone 36355c67c3SSimon Glass 37355c67c3SSimon Glass Returns: 38355c67c3SSimon Glass Filename of resulting .dtb file 39355c67c3SSimon Glass """ 40355c67c3SSimon Glass _, ext = os.path.splitext(fname) 41355c67c3SSimon Glass if ext != '.dts': 42355c67c3SSimon Glass return fname 43355c67c3SSimon Glass 44355c67c3SSimon Glass dts_input = tools.GetOutputFilename('source.dts') 45355c67c3SSimon Glass dtb_output = tools.GetOutputFilename('source.dtb') 46355c67c3SSimon Glass 47355c67c3SSimon Glass search_paths = [os.path.join(os.getcwd(), 'include')] 48355c67c3SSimon Glass root, _ = os.path.splitext(fname) 49355c67c3SSimon Glass args = ['-E', '-P', '-x', 'assembler-with-cpp', '-D__ASSEMBLY__'] 50355c67c3SSimon Glass args += ['-Ulinux'] 51355c67c3SSimon Glass for path in search_paths: 52355c67c3SSimon Glass args.extend(['-I', path]) 53355c67c3SSimon Glass args += ['-o', dts_input, fname] 54355c67c3SSimon Glass command.Run('cc', *args) 55355c67c3SSimon Glass 56355c67c3SSimon Glass # If we don't have a directory, put it in the tools tempdir 57355c67c3SSimon Glass search_list = [] 58355c67c3SSimon Glass for path in search_paths: 59355c67c3SSimon Glass search_list.extend(['-i', path]) 60355c67c3SSimon Glass args = ['-I', 'dts', '-o', dtb_output, '-O', 'dtb'] 61355c67c3SSimon Glass args.extend(search_list) 62355c67c3SSimon Glass args.append(dts_input) 63355c67c3SSimon Glass command.Run('dtc', *args) 64355c67c3SSimon Glass return dtb_output 658f224b37SSimon Glass 668f224b37SSimon Glassdef GetInt(node, propname, default=None): 678f224b37SSimon Glass prop = node.props.get(propname) 688f224b37SSimon Glass if not prop: 698f224b37SSimon Glass return default 708f224b37SSimon Glass value = fdt32_to_cpu(prop.value) 718f224b37SSimon Glass if type(value) == type(list): 728f224b37SSimon Glass raise ValueError("Node '%s' property '%' has list value: expecting" 738f224b37SSimon Glass "a single integer" % (node.name, propname)) 748f224b37SSimon Glass return value 758f224b37SSimon Glass 768f224b37SSimon Glassdef GetString(node, propname, default=None): 778f224b37SSimon Glass prop = node.props.get(propname) 788f224b37SSimon Glass if not prop: 798f224b37SSimon Glass return default 808f224b37SSimon Glass value = prop.value 818f224b37SSimon Glass if type(value) == type(list): 828f224b37SSimon Glass raise ValueError("Node '%s' property '%' has list value: expecting" 838f224b37SSimon Glass "a single string" % (node.name, propname)) 848f224b37SSimon Glass return value 858f224b37SSimon Glass 868f224b37SSimon Glassdef GetBool(node, propname, default=False): 878f224b37SSimon Glass if propname in node.props: 888f224b37SSimon Glass return True 898f224b37SSimon Glass return default 90