[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 generate_decl_enum(name, members, genlist=True):
    return mcgen('''

void visit_type_%(name)s(Visitor *m, %(name)s * obj, const char *name, Error **errp);
''',
                name=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_var(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_var(name), args=arglist, retval=retval).rstrip()
    if ret_type:
        ret += "\n" + mcgen(''''
%(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_var(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(%(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('''
visit_type_%(argtype)s(v, &%(c_name)s, "%(name)s", errp);
''',
                      c_name=c_var(argname), name=argname, argtype=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);
    visit_type_%(ret_type)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);
    visit_type_%(ret_type)s(v, &ret_in, "unused", errp);
    qapi_dealloc_visitor_cleanup(md);
}
''',
                 c_ret_type=c_type(ret_type), c_name=c_var(name),
                 ret_type=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_var(name)
    else:
        return 'static void qmp_marshal_input_%s(QDict *args, QObject **ret, Error **errp)' % c_var(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 gen_registry(commands):
    registry=""
    push_indent()
    for cmd in commands:
        registry += mcgen('''
qmp_register_command("%(name)s", qmp_marshal_input_%(c_name)s);
''',
                     name=cmd['command'], c_name=c_var(cmd['command']))
    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 "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-objects.h"
#include "qapi/qmp-core.h"
#include "qapi/qapi-visit-core.h"
#include "qapi/qmp-output-visitor.h"
#include "qapi/qmp-input-visitor.h"
#include "qapi/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:], "p:o:m", ["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

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

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

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)

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