1*e3e2c0c8SPhilippe Mathieu-Daudé#!/usr/bin/env python3
2*e3e2c0c8SPhilippe Mathieu-Daudé#
3*e3e2c0c8SPhilippe Mathieu-Daudé# Generate a Coccinelle semantic patch to remove pointless QOM cast.
4*e3e2c0c8SPhilippe Mathieu-Daudé#
5*e3e2c0c8SPhilippe Mathieu-Daudé# Usage:
6*e3e2c0c8SPhilippe Mathieu-Daudé#
7*e3e2c0c8SPhilippe Mathieu-Daudé# $ qom-cast-macro-clean-cocci-gen.py $(git ls-files) > qom_pointless_cast.cocci
8*e3e2c0c8SPhilippe Mathieu-Daudé# $ spatch \
9*e3e2c0c8SPhilippe Mathieu-Daudé#           --macro-file scripts/cocci-macro-file.h \
10*e3e2c0c8SPhilippe Mathieu-Daudé#           --sp-file qom_pointless_cast.cocci \
11*e3e2c0c8SPhilippe Mathieu-Daudé#           --keep-comments \
12*e3e2c0c8SPhilippe Mathieu-Daudé#           --use-gitgrep \
13*e3e2c0c8SPhilippe Mathieu-Daudé#           --in-place \
14*e3e2c0c8SPhilippe Mathieu-Daudé#           --dir .
15*e3e2c0c8SPhilippe Mathieu-Daudé#
16*e3e2c0c8SPhilippe Mathieu-Daudé# SPDX-FileContributor: Philippe Mathieu-Daudé <philmd@linaro.org>
17*e3e2c0c8SPhilippe Mathieu-Daudé# SPDX-FileCopyrightText: 2023 Linaro Ltd.
18*e3e2c0c8SPhilippe Mathieu-Daudé# SPDX-License-Identifier: GPL-2.0-or-later
19*e3e2c0c8SPhilippe Mathieu-Daudé
20*e3e2c0c8SPhilippe Mathieu-Daudéimport re
21*e3e2c0c8SPhilippe Mathieu-Daudéimport sys
22*e3e2c0c8SPhilippe Mathieu-Daudé
23*e3e2c0c8SPhilippe Mathieu-Daudéassert len(sys.argv) > 0
24*e3e2c0c8SPhilippe Mathieu-Daudé
25*e3e2c0c8SPhilippe Mathieu-Daudédef print_cocci_rule(qom_typedef, qom_cast_macro):
26*e3e2c0c8SPhilippe Mathieu-Daudé    print(f'''@@
27*e3e2c0c8SPhilippe Mathieu-Daudétypedef {qom_typedef};
28*e3e2c0c8SPhilippe Mathieu-Daudé{qom_typedef} *obj;
29*e3e2c0c8SPhilippe Mathieu-Daudé@@
30*e3e2c0c8SPhilippe Mathieu-Daudé-    {qom_cast_macro}(obj)
31*e3e2c0c8SPhilippe Mathieu-Daudé+    obj
32*e3e2c0c8SPhilippe Mathieu-Daudé''')
33*e3e2c0c8SPhilippe Mathieu-Daudé
34*e3e2c0c8SPhilippe Mathieu-Daudépatterns = [
35*e3e2c0c8SPhilippe Mathieu-Daudé    r'DECLARE_INSTANCE_CHECKER\((\w+),\W*(\w+),\W*TYPE_\w+\)',
36*e3e2c0c8SPhilippe Mathieu-Daudé    r'DECLARE_OBJ_CHECKERS\((\w+),\W*\w+,\W*(\w+),\W*TYPE_\w+\)',
37*e3e2c0c8SPhilippe Mathieu-Daudé    r'OBJECT_DECLARE_TYPE\((\w+),\W*\w+,\W*(\w+)\)',
38*e3e2c0c8SPhilippe Mathieu-Daudé    r'OBJECT_DECLARE_SIMPLE_TYPE\((\w+),\W*(\w+)\)',
39*e3e2c0c8SPhilippe Mathieu-Daudé    r'INTERFACE_CHECK\((\w+),\W*\(\w+\),\W*TYPE_(\w+)\)',
40*e3e2c0c8SPhilippe Mathieu-Daudé]
41*e3e2c0c8SPhilippe Mathieu-Daudé
42*e3e2c0c8SPhilippe Mathieu-Daudéfor fn in sys.argv[1:]:
43*e3e2c0c8SPhilippe Mathieu-Daudé    try:
44*e3e2c0c8SPhilippe Mathieu-Daudé        content = open(fn, 'rt').read()
45*e3e2c0c8SPhilippe Mathieu-Daudé    except:
46*e3e2c0c8SPhilippe Mathieu-Daudé        continue
47*e3e2c0c8SPhilippe Mathieu-Daudé    for pattern in patterns:
48*e3e2c0c8SPhilippe Mathieu-Daudé        for match in re.findall(pattern, content):
49*e3e2c0c8SPhilippe Mathieu-Daudé            print_cocci_rule(match[0], match[1])
50