[qemu.git] / scripts / qapi-commands.py

#
# QAPI command marshaller generator
#
# Copyright IBM, Corp. 2011
#
# Authors:
#  Anthony Liguori <aliguori@us.ibm.com>
#  Michael Roth    <mdroth@linux.vnet.ibm.com>
#
# This work is licensed under the terms of the GNU GPLv2.
# See the COPYING.LIB file in the top-level directory.

from ordereddict import OrderedDict
from qapi import *
import sys
import os
import getopt
import errno

def type_visitor(name):
    if type(name) == list:
        return 'visit_type_%sList' % name[0]
    else:
        return 'visit_type_%s' % name

def generate_decl_enum(name, members, genlist=True):
    return mcgen('''

void %(visitor)s(Visitor *m, %(name)s * obj, const char *name, Error **errp);
''',
                 visitor=type_visitor(name))

def generate_command_decl(name, args, ret_type):
    arglist=""
    for argname, argtype, optional, structured in parse_args(args):
        argtype = c_type(argtype)
        if argtype == "char *":
            argtype = "const char *"
        if optional:
            arglist += "bool has_%s, " % c_var(argname)
        arglist += "%s %s, " % (argtype, c_var(argname))
    return mcgen('''
%(ret_type)s qmp_%(name)s(%(args)sError **errp);
''',
                 ret_type=c_type(ret_type), name=c_fun(name), args=arglist).strip()

def gen_sync_call(name, args, ret_type, indent=0):
    ret = ""
    arglist=""
    retval=""
    if ret_type:
        retval = "retval = "
    for argname, argtype, optional, structured in parse_args(args):
        if optional:
            arglist += "has_%s, " % c_var(argname)
        arglist += "%s, " % (c_var(argname))
    push_indent(indent)
    ret = mcgen('''
%(retval)sqmp_%(name)s(%(args)serrp);

''',
                name=c_fun(name), args=arglist, retval=retval).rstrip()
    if ret_type:
        ret += "\n" + mcgen(''''
if (!error_is_set(errp)) {
    %(marshal_output_call)s
}
''',
                            marshal_output_call=gen_marshal_output_call(name, ret_type)).rstrip()
    pop_indent(indent)
    return ret.rstrip()


def gen_marshal_output_call(name, ret_type):
    if not ret_type:
        return ""
    return "qmp_marshal_output_%s(retval, ret, errp);" % c_fun(name)

def gen_visitor_output_containers_decl(ret_type):
    ret = ""
    push_indent()
    if ret_type:
        ret += mcgen('''
QmpOutputVisitor *mo;
QapiDeallocVisitor *md;
Visitor *v;
''')
    pop_indent()

    return ret

def gen_visitor_input_containers_decl(args):
    ret = ""

    push_indent()
    if len(args) > 0:
        ret += mcgen('''
QmpInputVisitor *mi;
QapiDeallocVisitor *md;
Visitor *v;
''')
    pop_indent()

    return ret.rstrip()

def gen_visitor_input_vars_decl(args):
    ret = ""
    push_indent()
    for argname, argtype, optional, structured in parse_args(args):
        if optional:
            ret += mcgen('''
bool has_%(argname)s = false;
''',
                         argname=c_var(argname))
        if c_type(argtype).endswith("*"):
            ret += mcgen('''
%(argtype)s %(argname)s = NULL;
''',
                         argname=c_var(argname), argtype=c_type(argtype))
        else:
            ret += mcgen('''
%(argtype)s %(argname)s;
''',
                         argname=c_var(argname), argtype=c_type(argtype))

    pop_indent()
    return ret.rstrip()

def gen_visitor_input_block(args, obj, dealloc=False):
    ret = ""
    if len(args) == 0:
        return ret

    push_indent()

    if dealloc:
        ret += mcgen('''
md = qapi_dealloc_visitor_new();
v = qapi_dealloc_get_visitor(md);
''')
    else:
        ret += mcgen('''
mi = qmp_input_visitor_new_strict(%(obj)s);
v = qmp_input_get_visitor(mi);
''',
                     obj=obj)

    for argname, argtype, optional, structured in parse_args(args):
        if optional:
            ret += mcgen('''
visit_start_optional(v, &has_%(c_name)s, "%(name)s", errp);
if (has_%(c_name)s) {
''',
                         c_name=c_var(argname), name=argname)
            push_indent()
        ret += mcgen('''
%(visitor)s(v, &%(c_name)s, "%(name)s", errp);
''',
                     c_name=c_var(argname), name=argname, argtype=argtype,
                     visitor=type_visitor(argtype))
        if optional:
            pop_indent()
            ret += mcgen('''
}
visit_end_optional(v, errp);
''')

    if dealloc:
        ret += mcgen('''
qapi_dealloc_visitor_cleanup(md);
''')
    else:
        ret += mcgen('''
qmp_input_visitor_cleanup(mi);
''')
    pop_indent()
    return ret.rstrip()

def gen_marshal_output(name, args, ret_type, middle_mode):
    if not ret_type:
        return ""

    ret = mcgen('''
static void qmp_marshal_output_%(c_name)s(%(c_ret_type)s ret_in, QObject **ret_out, Error **errp)
{
    QapiDeallocVisitor *md = qapi_dealloc_visitor_new();
    QmpOutputVisitor *mo = qmp_output_visitor_new();
    Visitor *v;

    v = qmp_output_get_visitor(mo);
    %(visitor)s(v, &ret_in, "unused", errp);
    if (!error_is_set(errp)) {
        *ret_out = qmp_output_get_qobject(mo);
    }
    qmp_output_visitor_cleanup(mo);
    v = qapi_dealloc_get_visitor(md);
    %(visitor)s(v, &ret_in, "unused", errp);
    qapi_dealloc_visitor_cleanup(md);
}
''',
                c_ret_type=c_type(ret_type), c_name=c_fun(name),
                visitor=type_visitor(ret_type))

    return ret

def gen_marshal_input_decl(name, args, ret_type, middle_mode):
    if middle_mode:
        return 'int qmp_marshal_input_%s(Monitor *mon, const QDict *qdict, QObject **ret)' % c_fun(name)
    else:
        return 'static void qmp_marshal_input_%s(QDict *args, QObject **ret, Error **errp)' % c_fun(name)


def gen_marshal_input(name, args, ret_type, middle_mode):
    hdr = gen_marshal_input_decl(name, args, ret_type, middle_mode)

    ret = mcgen('''
%(header)s
{
''',
                header=hdr)

    if middle_mode:
        ret += mcgen('''
    Error *local_err = NULL;
    Error **errp = &local_err;
    QDict *args = (QDict *)qdict;
''')

    if ret_type:
        if c_type(ret_type).endswith("*"):
            retval = "    %s retval = NULL;" % c_type(ret_type)
        else:
            retval = "    %s retval;" % c_type(ret_type)
        ret += mcgen('''
%(retval)s
''',
                     retval=retval)

    if len(args) > 0:
        ret += mcgen('''
%(visitor_input_containers_decl)s
%(visitor_input_vars_decl)s

%(visitor_input_block)s

''',
                     visitor_input_containers_decl=gen_visitor_input_containers_decl(args),
                     visitor_input_vars_decl=gen_visitor_input_vars_decl(args),
                     visitor_input_block=gen_visitor_input_block(args, "QOBJECT(args)"))
    else:
        ret += mcgen('''
    (void)args;
''')

    ret += mcgen('''
    if (error_is_set(errp)) {
        goto out;
    }
%(sync_call)s
''',
                 sync_call=gen_sync_call(name, args, ret_type, indent=4))
    ret += mcgen('''

out:
''')
    ret += mcgen('''
%(visitor_input_block_cleanup)s
''',
                 visitor_input_block_cleanup=gen_visitor_input_block(args, None,
                                                                     dealloc=True))

    if middle_mode:
        ret += mcgen('''

    if (local_err) {
        qerror_report_err(local_err);
        error_free(local_err);
        return -1;
    }
    return 0;
''')
    else:
        ret += mcgen('''
    return;
''')

    ret += mcgen('''
}
''')

    return ret

def option_value_matches(opt, val, cmd):
    if opt in cmd and cmd[opt] == val:
        return True
    return False

def gen_registry(commands):
    registry=""
    push_indent()
    for cmd in commands:
        options = 'QCO_NO_OPTIONS'
        if option_value_matches('success-response', 'no', cmd):
            options = 'QCO_NO_SUCCESS_RESP'

        registry += mcgen('''
qmp_register_command("%(name)s", qmp_marshal_input_%(c_name)s, %(opts)s);
''',
                     name=cmd['command'], c_name=c_fun(cmd['command']),
                     opts=options)
    pop_indent()
    ret = mcgen('''
static void qmp_init_marshal(void)
{
%(registry)s
}

qapi_init(qmp_init_marshal);
''',
                registry=registry.rstrip())
    return ret

def gen_command_decl_prologue(header, guard, prefix=""):
    ret = mcgen('''
/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */

/*
 * schema-defined QAPI function prototypes
 *
 * Copyright IBM, Corp. 2011
 *
 * Authors:
 *  Anthony Liguori   <aliguori@us.ibm.com>
 *
 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
 * See the COPYING.LIB file in the top-level directory.
 *
 */

#ifndef %(guard)s
#define %(guard)s

#include "%(prefix)sqapi-types.h"
#include "qapi/qmp/qdict.h"
#include "qapi/error.h"

''',
                 header=basename(header), guard=guardname(header), prefix=prefix)
    return ret

def gen_command_def_prologue(prefix="", proxy=False):
    ret = mcgen('''
/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */

/*
 * schema-defined QMP->QAPI command dispatch
 *
 * Copyright IBM, Corp. 2011
 *
 * Authors:
 *  Anthony Liguori   <aliguori@us.ibm.com>
 *
 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
 * See the COPYING.LIB file in the top-level directory.
 *
 */

#include "qemu-common.h"
#include "qemu/module.h"
#include "qapi/qmp/qerror.h"
#include "qapi/qmp/types.h"
#include "qapi/qmp/dispatch.h"
#include "qapi/visitor.h"
#include "qapi/qmp-output-visitor.h"
#include "qapi/qmp-input-visitor.h"
#include "qapi/dealloc-visitor.h"
#include "%(prefix)sqapi-types.h"
#include "%(prefix)sqapi-visit.h"

''',
                prefix=prefix)
    if not proxy:
        ret += '#include "%sqmp-commands.h"' % prefix
    return ret + "\n\n"


try:
    opts, args = getopt.gnu_getopt(sys.argv[1:], "chp:o:m",
                                   ["source", "header", "prefix=",
                                    "output-dir=", "type=", "middle"])
except getopt.GetoptError, err:
    print str(err)
    sys.exit(1)

output_dir = ""
prefix = ""
dispatch_type = "sync"
c_file = 'qmp-marshal.c'
h_file = 'qmp-commands.h'
middle_mode = False

do_c = False
do_h = False

for o, a in opts:
    if o in ("-p", "--prefix"):
        prefix = a
    elif o in ("-o", "--output-dir"):
        output_dir = a + "/"
    elif o in ("-t", "--type"):
        dispatch_type = a
    elif o in ("-m", "--middle"):
        middle_mode = True
    elif o in ("-c", "--source"):
        do_c = True
    elif o in ("-h", "--header"):
        do_h = True

if not do_c and not do_h:
    do_c = True
    do_h = True

c_file = output_dir + prefix + c_file
h_file = output_dir + prefix + h_file

def maybe_open(really, name, opt):
    if really:
        return open(name, opt)
    else:
        import StringIO
        return StringIO.StringIO()

try:
    os.makedirs(output_dir)
except os.error, e:
    if e.errno != errno.EEXIST:
        raise

exprs = parse_schema(sys.stdin)
commands = filter(lambda expr: expr.has_key('command'), exprs)
commands = filter(lambda expr: not expr.has_key('gen'), commands)

if dispatch_type == "sync":
    fdecl = maybe_open(do_h, h_file, 'w')
    fdef = maybe_open(do_c, c_file, 'w')
    ret = gen_command_decl_prologue(header=basename(h_file), guard=guardname(h_file), prefix=prefix)
    fdecl.write(ret)
    ret = gen_command_def_prologue(prefix=prefix)
    fdef.write(ret)

    for cmd in commands:
        arglist = []
        ret_type = None
        if cmd.has_key('data'):
            arglist = cmd['data']
        if cmd.has_key('returns'):
            ret_type = cmd['returns']
        ret = generate_command_decl(cmd['command'], arglist, ret_type) + "\n"
        fdecl.write(ret)
        if ret_type:
            ret = gen_marshal_output(cmd['command'], arglist, ret_type, middle_mode) + "\n"
            fdef.write(ret)

        if middle_mode:
            fdecl.write('%s;\n' % gen_marshal_input_decl(cmd['command'], arglist, ret_type, middle_mode))

        ret = gen_marshal_input(cmd['command'], arglist, ret_type, middle_mode) + "\n"
        fdef.write(ret)

    fdecl.write("\n#endif\n");

    if not middle_mode:
        ret = gen_registry(commands)
        fdef.write(ret)

    fdef.flush()
    fdef.close()
    fdecl.flush()
    fdecl.close()
Commit	Line	Data
c17d9908 MR	1	#
	2	# QAPI command marshaller generator
	3	#
	4	# Copyright IBM, Corp. 2011
	5	#
	6	# Authors:
	7	# Anthony Liguori <aliguori@us.ibm.com>
	8	# Michael Roth <mdroth@linux.vnet.ibm.com>
	9	#
	10	# This work is licensed under the terms of the GNU GPLv2.
	11	# See the COPYING.LIB file in the top-level directory.
	12
	13	from ordereddict import OrderedDict
	14	from qapi import *
	15	import sys
	16	import os
	17	import getopt
	18	import errno
	19
15e43e64 AL	20	def type_visitor(name):
	21	if type(name) == list:
	22	return 'visit_type_%sList' % name[0]
	23	else:
	24	return 'visit_type_%s' % name
	25
c17d9908 MR	26	def generate_decl_enum(name, members, genlist=True):
	27	return mcgen('''
	28
15e43e64	29	void %(visitor)s(Visitor m, %(name)s obj, const char name, Error *errp);
c17d9908	30	''',
15e43e64	31	visitor=type_visitor(name))
c17d9908 MR	32
	33	def generate_command_decl(name, args, ret_type):
	34	arglist=""
	35	for argname, argtype, optional, structured in parse_args(args):
	36	argtype = c_type(argtype)
	37	if argtype == "char *":
	38	argtype = "const char *"
	39	if optional:
	40	arglist += "bool has_%s, " % c_var(argname)
	41	arglist += "%s %s, " % (argtype, c_var(argname))
	42	return mcgen('''
	43	%(ret_type)s qmp_%(name)s(%(args)sError **errp);
	44	''',
c9da228b	45	ret_type=c_type(ret_type), name=c_fun(name), args=arglist).strip()
c17d9908 MR	46
	47	def gen_sync_call(name, args, ret_type, indent=0):
	48	ret = ""
	49	arglist=""
	50	retval=""
	51	if ret_type:
	52	retval = "retval = "
	53	for argname, argtype, optional, structured in parse_args(args):
	54	if optional:
	55	arglist += "has_%s, " % c_var(argname)
	56	arglist += "%s, " % (c_var(argname))
	57	push_indent(indent)
	58	ret = mcgen('''
	59	%(retval)sqmp_%(name)s(%(args)serrp);
	60
	61	''',
c9da228b	62	name=c_fun(name), args=arglist, retval=retval).rstrip()
c17d9908 MR	63	if ret_type:
c17d9908 MR	64	ret += "\n" + mcgen(''''
694a099a LC	65	if (!error_is_set(errp)) {
	66	%(marshal_output_call)s
	67	}
c17d9908 MR	68	''',
	69	marshal_output_call=gen_marshal_output_call(name, ret_type)).rstrip()
	70	pop_indent(indent)
	71	return ret.rstrip()
	72
	73
	74	def gen_marshal_output_call(name, ret_type):
	75	if not ret_type:
	76	return ""
c9da228b	77	return "qmp_marshal_output_%s(retval, ret, errp);" % c_fun(name)
c17d9908 MR	78
	79	def gen_visitor_output_containers_decl(ret_type):
	80	ret = ""
	81	push_indent()
	82	if ret_type:
	83	ret += mcgen('''
	84	QmpOutputVisitor *mo;
	85	QapiDeallocVisitor *md;
	86	Visitor *v;
	87	''')
	88	pop_indent()
	89
	90	return ret
	91
	92	def gen_visitor_input_containers_decl(args):
	93	ret = ""
	94
	95	push_indent()
	96	if len(args) > 0:
	97	ret += mcgen('''
	98	QmpInputVisitor *mi;
	99	QapiDeallocVisitor *md;
	100	Visitor *v;
	101	''')
	102	pop_indent()
	103
	104	return ret.rstrip()
	105
	106	def gen_visitor_input_vars_decl(args):
	107	ret = ""
	108	push_indent()
	109	for argname, argtype, optional, structured in parse_args(args):
	110	if optional:
	111	ret += mcgen('''
	112	bool has_%(argname)s = false;
	113	''',
	114	argname=c_var(argname))
	115	if c_type(argtype).endswith("*"):
	116	ret += mcgen('''
	117	%(argtype)s %(argname)s = NULL;
	118	''',
	119	argname=c_var(argname), argtype=c_type(argtype))
	120	else:
	121	ret += mcgen('''
	122	%(argtype)s %(argname)s;
	123	''',
	124	argname=c_var(argname), argtype=c_type(argtype))
	125
	126	pop_indent()
	127	return ret.rstrip()
	128
	129	def gen_visitor_input_block(args, obj, dealloc=False):
	130	ret = ""
	131	if len(args) == 0:
	132	return ret
	133
	134	push_indent()
	135
	136	if dealloc:
	137	ret += mcgen('''
	138	md = qapi_dealloc_visitor_new();
	139	v = qapi_dealloc_get_visitor(md);
	140	''')
	141	else:
142	ret += mcgen('''
6d36d7dc	143	mi = qmp_input_visitor_new_strict(%(obj)s);
c17d9908 MR	144	v = qmp_input_get_visitor(mi);
	145	''',
	146	obj=obj)
	147
	148	for argname, argtype, optional, structured in parse_args(args):
	149	if optional:
	150	ret += mcgen('''
	151	visit_start_optional(v, &has_%(c_name)s, "%(name)s", errp);
	152	if (has_%(c_name)s) {
	153	''',
	154	c_name=c_var(argname), name=argname)
	155	push_indent()
	156	ret += mcgen('''
15e43e64	157	%(visitor)s(v, &%(c_name)s, "%(name)s", errp);
c17d9908	158	''',
15e43e64 AL	159	c_name=c_var(argname), name=argname, argtype=argtype,
15e43e64 AL	160	visitor=type_visitor(argtype))
c17d9908 MR	161	if optional:
	162	pop_indent()
	163	ret += mcgen('''
	164	}
	165	visit_end_optional(v, errp);
	166	''')
	167
	168	if dealloc:
	169	ret += mcgen('''
	170	qapi_dealloc_visitor_cleanup(md);
	171	''')
	172	else:
	173	ret += mcgen('''
	174	qmp_input_visitor_cleanup(mi);
	175	''')
	176	pop_indent()
	177	return ret.rstrip()
	178
776574d6	179	def gen_marshal_output(name, args, ret_type, middle_mode):
c17d9908 MR	180	if not ret_type:
c17d9908 MR	181	return ""
776574d6	182
c17d9908 MR	183	ret = mcgen('''
	184	static void qmp_marshal_output_%(c_name)s(%(c_ret_type)s ret_in, QObject ret_out, Error errp)
	185	{
	186	QapiDeallocVisitor *md = qapi_dealloc_visitor_new();
	187	QmpOutputVisitor *mo = qmp_output_visitor_new();
	188	Visitor *v;
	189
	190	v = qmp_output_get_visitor(mo);
15e43e64	191	%(visitor)s(v, &ret_in, "unused", errp);
c17d9908 MR	192	if (!error_is_set(errp)) {
	193	*ret_out = qmp_output_get_qobject(mo);
	194	}
	195	qmp_output_visitor_cleanup(mo);
	196	v = qapi_dealloc_get_visitor(md);
15e43e64	197	%(visitor)s(v, &ret_in, "unused", errp);
c17d9908 MR	198	qapi_dealloc_visitor_cleanup(md);
	199	}
	200	''',
c9da228b	201	c_ret_type=c_type(ret_type), c_name=c_fun(name),
15e43e64	202	visitor=type_visitor(ret_type))
c17d9908 MR	203
	204	return ret
	205
776574d6 AL	206	def gen_marshal_input_decl(name, args, ret_type, middle_mode):
776574d6 AL	207	if middle_mode:
c9da228b	208	return 'int qmp_marshal_input_%s(Monitor mon, const QDict qdict, QObject **ret)' % c_fun(name)
776574d6	209	else:
c9da228b	210	return 'static void qmp_marshal_input_%s(QDict args, QObject ret, Error *errp)' % c_fun(name)
776574d6 AL	211
	212
	213
	214	def gen_marshal_input(name, args, ret_type, middle_mode):
	215	hdr = gen_marshal_input_decl(name, args, ret_type, middle_mode)
	216
c17d9908	217	ret = mcgen('''
776574d6	218	%(header)s
c17d9908 MR	219	{
c17d9908 MR	220	''',
776574d6 AL	221	header=hdr)
	222
	223	if middle_mode:
	224	ret += mcgen('''
	225	Error *local_err = NULL;
	226	Error **errp = &local_err;
	227	QDict args = (QDict )qdict;
	228	''')
c17d9908 MR	229
	230	if ret_type:
	231	if c_type(ret_type).endswith("*"):
	232	retval = " %s retval = NULL;" % c_type(ret_type)
	233	else:
	234	retval = " %s retval;" % c_type(ret_type)
	235	ret += mcgen('''
	236	%(retval)s
	237	''',
	238	retval=retval)
	239
	240	if len(args) > 0:
	241	ret += mcgen('''
	242	%(visitor_input_containers_decl)s
	243	%(visitor_input_vars_decl)s
	244
	245	%(visitor_input_block)s
	246
	247	''',
	248	visitor_input_containers_decl=gen_visitor_input_containers_decl(args),
	249	visitor_input_vars_decl=gen_visitor_input_vars_decl(args),
	250	visitor_input_block=gen_visitor_input_block(args, "QOBJECT(args)"))
776574d6 AL	251	else:
	252	ret += mcgen('''
	253	(void)args;
	254	''')
c17d9908 MR	255
	256	ret += mcgen('''
	257	if (error_is_set(errp)) {
	258	goto out;
	259	}
	260	%(sync_call)s
	261	''',
	262	sync_call=gen_sync_call(name, args, ret_type, indent=4))
	263	ret += mcgen('''
	264
	265	out:
	266	''')
	267	ret += mcgen('''
	268	%(visitor_input_block_cleanup)s
776574d6 AL	269	''',
	270	visitor_input_block_cleanup=gen_visitor_input_block(args, None,
	271	dealloc=True))
	272
	273	if middle_mode:
	274	ret += mcgen('''
	275
	276	if (local_err) {
	277	qerror_report_err(local_err);
	278	error_free(local_err);
	279	return -1;
	280	}
	281	return 0;
	282	''')
	283	else:
	284	ret += mcgen('''
c17d9908	285	return;
776574d6 AL	286	''')
	287
	288	ret += mcgen('''
c17d9908	289	}
776574d6 AL	290	''')
776574d6 AL	291
c17d9908 MR	292	return ret
c17d9908 MR	293
d34b867d LC	294	def option_value_matches(opt, val, cmd):
	295	if opt in cmd and cmd[opt] == val:
	296	return True
	297	return False
	298
c17d9908 MR	299	def gen_registry(commands):
	300	registry=""
	301	push_indent()
	302	for cmd in commands:
d34b867d LC	303	options = 'QCO_NO_OPTIONS'
	304	if option_value_matches('success-response', 'no', cmd):
	305	options = 'QCO_NO_SUCCESS_RESP'
	306
c17d9908	307	registry += mcgen('''
d34b867d	308	qmp_register_command("%(name)s", qmp_marshal_input_%(c_name)s, %(opts)s);
c17d9908	309	''',
d34b867d LC	310	name=cmd['command'], c_name=c_fun(cmd['command']),
d34b867d LC	311	opts=options)
c17d9908 MR	312	pop_indent()
	313	ret = mcgen('''
	314	static void qmp_init_marshal(void)
	315	{
	316	%(registry)s
	317	}
	318
	319	qapi_init(qmp_init_marshal);
	320	''',
	321	registry=registry.rstrip())
	322	return ret
	323
	324	def gen_command_decl_prologue(header, guard, prefix=""):
	325	ret = mcgen('''
	326	/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */
	327
	328	/*
	329	* schema-defined QAPI function prototypes
	330	*
	331	* Copyright IBM, Corp. 2011
	332	*
	333	* Authors:
	334	* Anthony Liguori <aliguori@us.ibm.com>
	335	*
	336	* This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
	337	* See the COPYING.LIB file in the top-level directory.
	338	*
	339	*/
	340
	341	#ifndef %(guard)s
	342	#define %(guard)s
	343
	344	#include "%(prefix)sqapi-types.h"
7b1b5d19 PB	345	#include "qapi/qmp/qdict.h"
7b1b5d19 PB	346	#include "qapi/error.h"
c17d9908 MR	347
c17d9908 MR	348	''',
776574d6	349	header=basename(header), guard=guardname(header), prefix=prefix)
c17d9908 MR	350	return ret
	351
	352	def gen_command_def_prologue(prefix="", proxy=False):
	353	ret = mcgen('''
	354	/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */
	355
	356	/*
	357	* schema-defined QMP->QAPI command dispatch
	358	*
	359	* Copyright IBM, Corp. 2011
	360	*
	361	* Authors:
	362	* Anthony Liguori <aliguori@us.ibm.com>
	363	*
	364	* This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
	365	* See the COPYING.LIB file in the top-level directory.
	366	*
	367	*/
	368
79ee7df8	369	#include "qemu-common.h"
1de7afc9	370	#include "qemu/module.h"
7b1b5d19 PB	371	#include "qapi/qmp/qerror.h"
	372	#include "qapi/qmp/types.h"
	373	#include "qapi/qmp/dispatch.h"
	374	#include "qapi/visitor.h"
c17d9908 MR	375	#include "qapi/qmp-output-visitor.h"
c17d9908 MR	376	#include "qapi/qmp-input-visitor.h"
7b1b5d19	377	#include "qapi/dealloc-visitor.h"
c17d9908 MR	378	#include "%(prefix)sqapi-types.h"
	379	#include "%(prefix)sqapi-visit.h"
	380
	381	''',
	382	prefix=prefix)
	383	if not proxy:
	384	ret += '#include "%sqmp-commands.h"' % prefix
776574d6	385	return ret + "\n\n"
c17d9908 MR	386
	387
	388	try:
8d3bc517 AK	389	opts, args = getopt.gnu_getopt(sys.argv[1:], "chp:o:m",
	390	["source", "header", "prefix=",
	391	"output-dir=", "type=", "middle"])
c17d9908 MR	392	except getopt.GetoptError, err:
	393	print str(err)
	394	sys.exit(1)
	395
	396	output_dir = ""
	397	prefix = ""
	398	dispatch_type = "sync"
	399	c_file = 'qmp-marshal.c'
	400	h_file = 'qmp-commands.h'
776574d6	401	middle_mode = False
c17d9908	402
8d3bc517 AK	403	do_c = False
	404	do_h = False
	405
c17d9908 MR	406	for o, a in opts:
	407	if o in ("-p", "--prefix"):
	408	prefix = a
	409	elif o in ("-o", "--output-dir"):
	410	output_dir = a + "/"
	411	elif o in ("-t", "--type"):
	412	dispatch_type = a
776574d6 AL	413	elif o in ("-m", "--middle"):
776574d6 AL	414	middle_mode = True
8d3bc517	415	elif o in ("-c", "--source"):
8d3bc517	416	do_c = True
19bf7c87 AK	417	elif o in ("-h", "--header"):
19bf7c87 AK	418	do_h = True
8d3bc517 AK	419
	420	if not do_c and not do_h:
	421	do_c = True
	422	do_h = True
c17d9908 MR	423
	424	c_file = output_dir + prefix + c_file
	425	h_file = output_dir + prefix + h_file
	426
8d3bc517	427	def maybe_open(really, name, opt):
8d3bc517 AK	428	if really:
	429	return open(name, opt)
	430	else:
19bf7c87 AK	431	import StringIO
19bf7c87 AK	432	return StringIO.StringIO()
8d3bc517	433
c17d9908 MR	434	try:
	435	os.makedirs(output_dir)
	436	except os.error, e:
	437	if e.errno != errno.EEXIST:
	438	raise
	439
	440	exprs = parse_schema(sys.stdin)
	441	commands = filter(lambda expr: expr.has_key('command'), exprs)
5dbee474	442	commands = filter(lambda expr: not expr.has_key('gen'), commands)
c17d9908 MR	443
c17d9908 MR	444	if dispatch_type == "sync":
8d3bc517 AK	445	fdecl = maybe_open(do_h, h_file, 'w')
8d3bc517 AK	446	fdef = maybe_open(do_c, c_file, 'w')
c17d9908 MR	447	ret = gen_command_decl_prologue(header=basename(h_file), guard=guardname(h_file), prefix=prefix)
	448	fdecl.write(ret)
	449	ret = gen_command_def_prologue(prefix=prefix)
	450	fdef.write(ret)
	451
	452	for cmd in commands:
	453	arglist = []
	454	ret_type = None
	455	if cmd.has_key('data'):
	456	arglist = cmd['data']
	457	if cmd.has_key('returns'):
	458	ret_type = cmd['returns']
	459	ret = generate_command_decl(cmd['command'], arglist, ret_type) + "\n"
	460	fdecl.write(ret)
	461	if ret_type:
776574d6	462	ret = gen_marshal_output(cmd['command'], arglist, ret_type, middle_mode) + "\n"
c17d9908	463	fdef.write(ret)
776574d6 AL	464
	465	if middle_mode:
	466	fdecl.write('%s;\n' % gen_marshal_input_decl(cmd['command'], arglist, ret_type, middle_mode))
	467
	468	ret = gen_marshal_input(cmd['command'], arglist, ret_type, middle_mode) + "\n"
c17d9908 MR	469	fdef.write(ret)
c17d9908 MR	470
7534ba01	471	fdecl.write("\n#endif\n");
776574d6 AL	472
	473	if not middle_mode:
	474	ret = gen_registry(commands)
	475	fdef.write(ret)
c17d9908 MR	476
	477	fdef.flush()
	478	fdef.close()
	479	fdecl.flush()
	480	fdecl.close()