xref: /openbmc/u-boot/tools/dtoc/dtoc.py (revision cbd2fba1) 
1#!/usr/bin/env python2
2# SPDX-License-Identifier: GPL-2.0+
3#
4# Copyright (C) 2016 Google, Inc
5# Written by Simon Glass <sjg@chromium.org>
6#
7
8"""Device tree to C tool
9
10This tool converts a device tree binary file (.dtb) into two C files. The
11indent is to allow a C program to access data from the device tree without
12having to link against libfdt. By putting the data from the device tree into
13C structures, normal C code can be used. This helps to reduce the size of the
14compiled program.
15
16Dtoc produces two output files:
17
18   dt-structs.h  - contains struct definitions
19   dt-platdata.c - contains data from the device tree using the struct
20                      definitions, as well as U-Boot driver definitions.
21
22This tool is used in U-Boot to provide device tree data to SPL without
23increasing the code size of SPL. This supports the CONFIG_SPL_OF_PLATDATA
24options. For more information about the use of this options and tool please
25see doc/driver-model/of-plat.txt
26"""
27
28from optparse import OptionParser
29import os
30import sys
31import unittest
32
33# Bring in the patman libraries
34our_path = os.path.dirname(os.path.realpath(__file__))
35sys.path.append(os.path.join(our_path, '../patman'))
36
37# Bring in the libfdt module
38sys.path.insert(0, 'scripts/dtc/pylibfdt')
39sys.path.insert(0, os.path.join(our_path,
40                '../../build-sandbox_spl/scripts/dtc/pylibfdt'))
41
42import dtb_platdata
43import test_util
44
45def run_tests(args):
46    """Run all the test we have for dtoc
47
48    Args:
49        args: List of positional args provided to dtoc. This can hold a test
50            name to execute (as in 'dtoc -t test_empty_file', for example)
51    """
52    import test_dtoc
53
54    result = unittest.TestResult()
55    sys.argv = [sys.argv[0]]
56    test_name = args and args[0] or None
57    for module in (test_dtoc.TestDtoc,):
58        if test_name:
59            try:
60                suite = unittest.TestLoader().loadTestsFromName(test_name, module)
61            except AttributeError:
62                continue
63        else:
64            suite = unittest.TestLoader().loadTestsFromTestCase(module)
65        suite.run(result)
66
67    print result
68    for _, err in result.errors:
69        print err
70    for _, err in result.failures:
71        print err
72
73def RunTestCoverage():
74    """Run the tests and check that we get 100% coverage"""
75    sys.argv = [sys.argv[0]]
76    test_util.RunTestCoverage('tools/dtoc/dtoc.py', '/dtoc.py',
77            ['tools/patman/*.py', '*/fdt*', '*test*'], options.build_dir)
78
79
80if __name__ != '__main__':
81    sys.exit(1)
82
83parser = OptionParser()
84parser.add_option('-B', '--build-dir', type='string', default='b',
85        help='Directory containing the build output')
86parser.add_option('-d', '--dtb-file', action='store',
87                  help='Specify the .dtb input file')
88parser.add_option('--include-disabled', action='store_true',
89                  help='Include disabled nodes')
90parser.add_option('-o', '--output', action='store', default='-',
91                  help='Select output filename')
92parser.add_option('-P', '--processes', type=int,
93                  help='set number of processes to use for running tests')
94parser.add_option('-t', '--test', action='store_true', dest='test',
95                  default=False, help='run tests')
96parser.add_option('-T', '--test-coverage', action='store_true',
97                default=False, help='run tests and check for 100% coverage')
98(options, args) = parser.parse_args()
99
100# Run our meagre tests
101if options.test:
102    run_tests(args)
103
104elif options.test_coverage:
105    RunTestCoverage()
106
107else:
108    dtb_platdata.run_steps(args, options.dtb_file, options.include_disabled,
109                           options.output)
110