xref: /linux/tools/net/sunrpc/xdrgen/subcmds/declarations.py (revision 2831fa8b8bcf1083f9526aa0c41fafb0796cf874) 
1#!/usr/bin/env python3
2# ex: set filetype=python:
3
4"""Translate an XDR specification into executable code that
5can be compiled for the Linux kernel."""
6
7import logging
8
9from argparse import Namespace
10from lark import logger
11from lark.exceptions import VisitError
12
13from generators.enum import XdrEnumGenerator
14from generators.header_bottom import XdrHeaderBottomGenerator
15from generators.header_top import XdrHeaderTopGenerator
16from generators.pointer import XdrPointerGenerator
17from generators.program import XdrProgramGenerator
18from generators.typedef import XdrTypedefGenerator
19from generators.struct import XdrStructGenerator
20from generators.union import XdrUnionGenerator
21
22from xdr_ast import transform_parse_tree, _RpcProgram, Specification
23from xdr_ast import _XdrEnum, _XdrPointer, _XdrTypedef, _XdrStruct, _XdrUnion
24from xdr_parse import xdr_parser, set_xdr_annotate
25from xdr_parse import make_error_handler, XdrParseError
26from xdr_parse import handle_transform_error
27
28logger.setLevel(logging.INFO)
29
30
31def emit_header_declarations(
32    root: Specification, language: str, peer: str
33) -> None:
34    """Emit header declarations"""
35    for definition in root.definitions:
36        if isinstance(definition.value, _XdrEnum):
37            gen = XdrEnumGenerator(language, peer)
38        elif isinstance(definition.value, _XdrPointer):
39            gen = XdrPointerGenerator(language, peer)
40        elif isinstance(definition.value, _XdrTypedef):
41            gen = XdrTypedefGenerator(language, peer)
42        elif isinstance(definition.value, _XdrStruct):
43            gen = XdrStructGenerator(language, peer)
44        elif isinstance(definition.value, _XdrUnion):
45            gen = XdrUnionGenerator(language, peer)
46        elif isinstance(definition.value, _RpcProgram):
47            gen = XdrProgramGenerator(language, peer)
48        else:
49            continue
50        gen.emit_declaration(definition.value)
51
52
53def subcmd(args: Namespace) -> int:
54    """Generate definitions and declarations"""
55
56    set_xdr_annotate(args.annotate)
57    parser = xdr_parser()
58    with open(args.filename, encoding="utf-8") as f:
59        source = f.read()
60        try:
61            parse_tree = parser.parse(
62                source, on_error=make_error_handler(source, args.filename)
63            )
64        except XdrParseError:
65            return 1
66        try:
67            ast = transform_parse_tree(parse_tree)
68        except VisitError as e:
69            handle_transform_error(e, source, args.filename)
70            return 1
71
72        gen = XdrHeaderTopGenerator(args.language, args.peer)
73        gen.emit_declaration(args.filename, ast)
74
75        emit_header_declarations(ast, args.language, args.peer)
76
77        gen = XdrHeaderBottomGenerator(args.language, args.peer)
78        gen.emit_declaration(args.filename, ast)
79
80    return 0
81