Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
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