xref: /openbmc/linux/scripts/dtc/dt-extract-compatibles (revision c35bcede4ffa43a581460ff79ec969d535db0b84)
1#!/usr/bin/env python3
2# SPDX-License-Identifier: GPL-2.0-only
3
4import os
5import glob
6import re
7import argparse
8
9
10def parse_of_declare_macros(data):
11	""" Find all compatible strings in OF_DECLARE() style macros """
12	compat_list = []
13	# CPU_METHOD_OF_DECLARE does not have a compatible string
14	for m in re.finditer(r'(?<!CPU_METHOD_)(IRQCHIP|OF)_(DECLARE|MATCH)(_DRIVER)?\(.*?\)', data):
15		try:
16			compat = re.search(r'"(.*?)"', m[0])[1]
17		except:
18			# Fails on compatible strings in #define, so just skip
19			continue
20		compat_list += [compat]
21
22	return compat_list
23
24
25def parse_of_device_id(data):
26	""" Find all compatible strings in of_device_id structs """
27	compat_list = []
28	for m in re.finditer(r'of_device_id(\s+\S+)?\s+\S+\[\](\s+\S+)?\s*=\s*({.*?);', data):
29		compat_list += re.findall(r'\.compatible\s+=\s+"(\S+)"', m[3])
30
31	return compat_list
32
33
34def parse_compatibles(file):
35	with open(file, 'r', encoding='utf-8') as f:
36		data = f.read().replace('\n', '')
37
38	compat_list = parse_of_declare_macros(data)
39	compat_list += parse_of_device_id(data)
40
41	return compat_list
42
43def print_compat(filename, compatibles):
44	if not compatibles:
45		return
46	if show_filename:
47		compat_str = ' '.join(compatibles)
48		print(filename + ": compatible(s): " + compat_str)
49	else:
50		print(*compatibles, sep='\n')
51
52def files_to_parse(path_args):
53	for f in path_args:
54		if os.path.isdir(f):
55			for filename in glob.iglob(f + "/**/*.c", recursive=True):
56				yield filename
57		else:
58			yield f
59
60show_filename = False
61
62if __name__ == "__main__":
63	ap = argparse.ArgumentParser()
64	ap.add_argument("cfile", type=str, nargs='*', help="C source files or directories to parse")
65	ap.add_argument('-H', '--with-filename', help="Print filename with compatibles", action="store_true")
66	args = ap.parse_args()
67
68	show_filename = args.with_filename
69
70	for f in files_to_parse(args.cfile):
71		compat_list = parse_compatibles(f)
72		print_compat(f, compat_list)
73