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 11 12from generators.constant import XdrConstantGenerator 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 _XdrConstant, _XdrEnum, _XdrPointer 24from xdr_ast import _XdrTypedef, _XdrStruct, _XdrUnion 25from xdr_parse import xdr_parser, set_xdr_annotate 26from xdr_parse import make_error_handler, XdrParseError 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 ast = transform_parse_tree(parse_tree) 67 68 gen = XdrHeaderTopGenerator(args.language, args.peer) 69 gen.emit_declaration(args.filename, ast) 70 71 emit_header_declarations(ast, args.language, args.peer) 72 73 gen = XdrHeaderBottomGenerator(args.language, args.peer) 74 gen.emit_declaration(args.filename, ast) 75 76 return 0 77