1
0
mirror of https://gitlab.com/libvirt/libvirt.git synced 2024-12-22 17:34:18 +03:00
libvirt/scripts/rpcgen/main.py
Daniel P. Berrangé 086fa214bb rpcgen: add g_auto function support
This will eliminate the need to call xdr_free to clear
pointers from data structures.

Reviewed-by: Michal Privoznik <mprivozn@redhat.com>
Signed-off-by: Daniel P. Berrangé <berrange@redhat.com>
2023-11-03 14:06:35 -04:00

89 lines
2.8 KiB
Python
Executable File

#!/usr/bin/env python3
# SPDX-License-Identifier: LGPL-2.1-or-later
import argparse
import os
import sys
from rpcgen.parser import XDRParser
from rpcgen.generator import (
XDRTypeDeclarationGenerator,
XDRTypeImplementationGenerator,
XDRMarshallDeclarationGenerator,
XDRMarshallImplementationGenerator,
)
def parse_cli():
parser = argparse.ArgumentParser("RPC code generator")
parser.add_argument(
"-m",
"--mode",
choices=["header", "source", "repr"],
help="Output generation mode",
)
parser.add_argument(
"-r", "--header", default=[], action="append", help="Extra headers to include"
)
parser.add_argument("input", default="-", nargs="?", help="XDR input protocol file")
parser.add_argument("output", default="-", nargs="?", help="Generated output file")
return parser.parse_args()
def main():
args = parse_cli()
infp = sys.stdin
outfp = sys.stdout
if args.input != "-":
infp = open(args.input, "r")
if args.output != "-":
# the old genprotocol.pl wrapper would make the
# output files mode 0444, which will prevent us
# from writing directly do them. Explicitly
# unlinking first gets rid of any old possibly
# read-only copy
#
# We can delete this in a few years, once we
# know users won't have a previously generated
# readonly copy lieing around.
try:
os.unlink(args.output)
except Exception:
pass
outfp = open(args.output, "w")
parser = XDRParser(infp)
spec = parser.parse()
if args.mode == "header":
print("/* This file is auto-generated from %s */\n" % args.input, file=outfp)
print("#include <rpc/rpc.h>", file=outfp)
print('#include "internal.h"', file=outfp)
for h in args.header:
print('#include "%s"' % h, file=outfp)
print("", file=outfp)
print("#pragma once\n", file=outfp)
generator = XDRTypeDeclarationGenerator(spec)
print(generator.visit(), file=outfp)
generator = XDRMarshallDeclarationGenerator(spec)
print(generator.visit(), file=outfp)
elif args.mode == "source":
print("/* This file is auto-generated from %s */\n" % args.input, file=outfp)
print("#include <config.h>", file=outfp)
for h in args.header:
print('#include "%s"' % h, file=outfp)
print("", file=outfp)
generator = XDRTypeImplementationGenerator(spec)
print(generator.visit(), file=outfp)
generator = XDRMarshallImplementationGenerator(spec)
print(generator.visit(), file=outfp)
elif args.mode == "repr":
print(spec, file=outfp)
else:
pass # Just validates XDR input syntax
main()