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.constant import XdrConstantGenerator 14from generators.enum import XdrEnumGenerator 15from generators.header_bottom import XdrHeaderBottomGenerator 16from generators.header_top import XdrHeaderTopGenerator 17from generators.pointer import XdrPointerGenerator 18from generators.program import XdrProgramGenerator 19from generators.typedef import XdrTypedefGenerator 20from generators.struct import XdrStructGenerator 21from generators.union import XdrUnionGenerator 22 23from xdr_ast import transform_parse_tree, _RpcProgram, Specification 24from xdr_ast import _XdrConstant, _XdrEnum, _XdrPointer 25from xdr_ast import _XdrTypedef, _XdrStruct, _XdrUnion 26from xdr_parse import xdr_parser, set_xdr_annotate 27from xdr_parse import make_error_handler, XdrParseError 28from xdr_parse import handle_transform_error 29 30logger.setLevel(logging.INFO) 31 32 33def emit_header_declarations( 34 root: Specification, language: str, peer: str 35) -> None: 36 """Emit header declarations""" 37 for definition in root.definitions: 38 if isinstance(definition.value, _XdrEnum): 39 gen = XdrEnumGenerator(language, peer) 40 elif isinstance(definition.value, _XdrPointer): 41 gen = XdrPointerGenerator(language, peer) 42 elif isinstance(definition.value, _XdrTypedef): 43 gen = XdrTypedefGenerator(language, peer) 44 elif isinstance(definition.value, _XdrStruct): 45 gen = XdrStructGenerator(language, peer) 46 elif isinstance(definition.value, _XdrUnion): 47 gen = XdrUnionGenerator(language, peer) 48 elif isinstance(definition.value, _RpcProgram): 49 gen = XdrProgramGenerator(language, peer) 50 else: 51 continue 52 gen.emit_declaration(definition.value) 53 54 55def subcmd(args: Namespace) -> int: 56 """Generate definitions and declarations""" 57 58 set_xdr_annotate(args.annotate) 59 parser = xdr_parser() 60 with open(args.filename, encoding="utf-8") as f: 61 source = f.read() 62 try: 63 parse_tree = parser.parse( 64 source, on_error=make_error_handler(source, args.filename) 65 ) 66 except XdrParseError: 67 return 1 68 try: 69 ast = transform_parse_tree(parse_tree) 70 except VisitError as e: 71 handle_transform_error(e, source, args.filename) 72 return 1 73 74 gen = XdrHeaderTopGenerator(args.language, args.peer) 75 gen.emit_declaration(args.filename, ast) 76 77 emit_header_declarations(ast, args.language, args.peer) 78 79 gen = XdrHeaderBottomGenerator(args.language, args.peer) 80 gen.emit_declaration(args.filename, ast) 81 82 return 0 83