xref: /qemu/scripts/qapi/main.py (revision 8afc43ea) 
1# This work is licensed under the terms of the GNU GPL, version 2 or later.
2# See the COPYING file in the top-level directory.
3
4"""
5QAPI Generator
6
7This is the main entry point for generating C code from the QAPI schema.
8"""
9
10import argparse
11import sys
12from typing import Optional
13
14from .commands import gen_commands
15from .common import must_match
16from .error import QAPIError
17from .events import gen_events
18from .introspect import gen_introspect
19from .schema import QAPISchema
20from .types import gen_types
21from .visit import gen_visit
22
23
24def invalid_prefix_char(prefix: str) -> Optional[str]:
25    match = must_match(r'([A-Za-z_.-][A-Za-z0-9_.-]*)?', prefix)
26    if match.end() != len(prefix):
27        return prefix[match.end()]
28    return None
29
30
31def generate(schema_file: str,
32             output_dir: str,
33             prefix: str,
34             unmask: bool = False,
35             builtins: bool = False,
36             gen_tracing: bool = False) -> None:
37    """
38    Generate C code for the given schema into the target directory.
39
40    :param schema_file: The primary QAPI schema file.
41    :param output_dir: The output directory to store generated code.
42    :param prefix: Optional C-code prefix for symbol names.
43    :param unmask: Expose non-ABI names through introspection?
44    :param builtins: Generate code for built-in types?
45
46    :raise QAPIError: On failures.
47    """
48    assert invalid_prefix_char(prefix) is None
49
50    schema = QAPISchema(schema_file)
51    gen_types(schema, output_dir, prefix, builtins)
52    gen_visit(schema, output_dir, prefix, builtins)
53    gen_commands(schema, output_dir, prefix, gen_tracing)
54    gen_events(schema, output_dir, prefix)
55    gen_introspect(schema, output_dir, prefix, unmask)
56
57
58def main() -> int:
59    """
60    gapi-gen executable entry point.
61    Expects arguments via sys.argv, see --help for details.
62
63    :return: int, 0 on success, 1 on failure.
64    """
65    parser = argparse.ArgumentParser(
66        description='Generate code from a QAPI schema')
67    parser.add_argument('-b', '--builtins', action='store_true',
68                        help="generate code for built-in types")
69    parser.add_argument('-o', '--output-dir', action='store',
70                        default='',
71                        help="write output to directory OUTPUT_DIR")
72    parser.add_argument('-p', '--prefix', action='store',
73                        default='',
74                        help="prefix for symbols")
75    parser.add_argument('-u', '--unmask-non-abi-names', action='store_true',
76                        dest='unmask',
77                        help="expose non-ABI names in introspection")
78
79    # Option --suppress-tracing exists so we can avoid solving build system
80    # problems.  TODO Drop it when we no longer need it.
81    parser.add_argument('--suppress-tracing', action='store_true',
82                        help="suppress adding trace events to qmp marshals")
83
84    parser.add_argument('schema', action='store')
85    args = parser.parse_args()
86
87    funny_char = invalid_prefix_char(args.prefix)
88    if funny_char:
89        msg = f"funny character '{funny_char}' in argument of --prefix"
90        print(f"{sys.argv[0]}: {msg}", file=sys.stderr)
91        return 1
92
93    try:
94        generate(args.schema,
95                 output_dir=args.output_dir,
96                 prefix=args.prefix,
97                 unmask=args.unmask,
98                 builtins=args.builtins,
99                 gen_tracing=not args.suppress_tracing)
100    except QAPIError as err:
101        print(err, file=sys.stderr)
102        return 1
103    return 0
104