blob: 61561625007ac3f4e11d7f7f996a5f75fd10332b [file] [log] [blame]
Michael Roth06d64c62011-07-19 14:50:41 -05001#
2# QAPI visitor generator
3#
4# Copyright IBM, Corp. 2011
Eric Blake6540e9f2015-04-10 15:07:59 -06005# Copyright (C) 2014-2015 Red Hat, Inc.
Michael Roth06d64c62011-07-19 14:50:41 -05006#
7# Authors:
8# Anthony Liguori <aliguori@us.ibm.com>
9# Michael Roth <mdroth@linux.vnet.ibm.com>
Markus Armbruster297a3642014-05-07 09:53:54 +020010# Markus Armbruster <armbru@redhat.com>
Michael Roth06d64c62011-07-19 14:50:41 -050011#
Markus Armbruster678e48a2014-03-01 08:40:34 +010012# This work is licensed under the terms of the GNU GPL, version 2.
13# See the COPYING file in the top-level directory.
Michael Roth06d64c62011-07-19 14:50:41 -050014
15from ordereddict import OrderedDict
16from qapi import *
Markus Armbruster297a3642014-05-07 09:53:54 +020017import re
Michael Roth06d64c62011-07-19 14:50:41 -050018import sys
19import os
20import getopt
21import errno
22
Markus Armbrusterbe3c7712014-05-07 09:53:50 +020023implicit_structs = []
24
25def generate_visit_implicit_struct(type):
26 global implicit_structs
27 if type in implicit_structs:
28 return ''
29 implicit_structs.append(type)
30 return mcgen('''
31
32static void visit_type_implicit_%(c_type)s(Visitor *m, %(c_type)s **obj, Error **errp)
33{
34 Error *err = NULL;
35
36 visit_start_implicit_struct(m, (void **)obj, sizeof(%(c_type)s), &err);
37 if (!err) {
Markus Armbruster297a3642014-05-07 09:53:54 +020038 visit_type_%(c_type)s_fields(m, obj, errp);
Markus Armbrusterbe3c7712014-05-07 09:53:50 +020039 visit_end_implicit_struct(m, &err);
40 }
41 error_propagate(errp, err);
42}
43''',
44 c_type=type_name(type))
45
Kevin Wolf622f5572013-09-19 11:56:36 +020046def generate_visit_struct_fields(name, field_prefix, fn_prefix, members, base = None):
Kevin Wolf50f2bdc2013-07-03 15:58:57 +020047 substructs = []
Kevin Wolfd131c892013-07-02 16:18:35 +020048 ret = ''
Richard Henderson7b75d9d2013-10-31 13:26:01 -070049 if not fn_prefix:
50 full_name = name
51 else:
52 full_name = "%s_%s" % (name, fn_prefix)
Kevin Wolf50f2bdc2013-07-03 15:58:57 +020053
Markus Armbrusterbe3c7712014-05-07 09:53:50 +020054 if base:
55 ret += generate_visit_implicit_struct(base)
56
Kevin Wolf50f2bdc2013-07-03 15:58:57 +020057 ret += mcgen('''
58
Amos Kong638ca8a2014-06-10 19:25:51 +080059static void visit_type_%(full_name)s_fields(Visitor *m, %(name)s **obj, Error **errp)
Kevin Wolf50f2bdc2013-07-03 15:58:57 +020060{
61 Error *err = NULL;
62''',
63 name=name, full_name=full_name)
64 push_indent()
Paolo Bonzinid1953252012-07-17 16:17:04 +020065
Kevin Wolf622f5572013-09-19 11:56:36 +020066 if base:
67 ret += mcgen('''
Markus Armbrusterbe3c7712014-05-07 09:53:50 +020068visit_type_implicit_%(type)s(m, &(*obj)->%(c_prefix)s%(c_name)s, &err);
Markus Armbruster297a3642014-05-07 09:53:54 +020069if (err) {
70 goto out;
71}
Kevin Wolf622f5572013-09-19 11:56:36 +020072''',
73 c_prefix=c_var(field_prefix),
74 type=type_name(base), c_name=c_var('base'))
75
Eric Blake6b5abc72015-05-04 09:05:33 -060076 for argname, argentry, optional in parse_args(members):
Michael Roth06d64c62011-07-19 14:50:41 -050077 if optional:
78 ret += mcgen('''
Markus Armbrustere2cd0f42014-05-07 09:53:46 +020079visit_optional(m, &(*obj)->%(c_prefix)shas_%(c_name)s, "%(name)s", &err);
Markus Armbruster297a3642014-05-07 09:53:54 +020080if (!err && (*obj)->%(prefix)shas_%(c_name)s) {
Michael Roth06d64c62011-07-19 14:50:41 -050081''',
82 c_prefix=c_var(field_prefix), prefix=field_prefix,
83 c_name=c_var(argname), name=argname)
84 push_indent()
85
Eric Blake6b5abc72015-05-04 09:05:33 -060086 ret += mcgen('''
Markus Armbruster949ceeb2014-03-01 08:40:37 +010087visit_type_%(type)s(m, &(*obj)->%(c_prefix)s%(c_name)s, "%(name)s", &err);
Michael Roth06d64c62011-07-19 14:50:41 -050088''',
Eric Blake6b5abc72015-05-04 09:05:33 -060089 c_prefix=c_var(field_prefix), prefix=field_prefix,
90 type=type_name(argentry), c_name=c_var(argname),
91 name=argname)
Michael Roth06d64c62011-07-19 14:50:41 -050092
93 if optional:
94 pop_indent()
95 ret += mcgen('''
96}
Paolo Bonzinid1953252012-07-17 16:17:04 +020097''')
Markus Armbruster297a3642014-05-07 09:53:54 +020098 ret += mcgen('''
99if (err) {
100 goto out;
101}
102''')
Paolo Bonzinid1953252012-07-17 16:17:04 +0200103
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200104 pop_indent()
Markus Armbruster297a3642014-05-07 09:53:54 +0200105 if re.search('^ *goto out\\;', ret, re.MULTILINE):
106 ret += mcgen('''
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200107
Markus Armbruster297a3642014-05-07 09:53:54 +0200108out:
109''')
110 ret += mcgen('''
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200111 error_propagate(errp, err);
112}
113''')
Kevin Wolfd131c892013-07-02 16:18:35 +0200114 return ret
115
116
117def generate_visit_struct_body(field_prefix, name, members):
118 ret = mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200119 Error *err = NULL;
120
Kevin Wolfd131c892013-07-02 16:18:35 +0200121''')
Kevin Wolfd131c892013-07-02 16:18:35 +0200122
Richard Henderson7b75d9d2013-10-31 13:26:01 -0700123 if not field_prefix:
124 full_name = name
125 else:
126 full_name = "%s_%s" % (field_prefix, name)
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200127
Kevin Wolfd131c892013-07-02 16:18:35 +0200128 if len(field_prefix):
Kevin Wolfd131c892013-07-02 16:18:35 +0200129 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200130 visit_start_struct(m, NULL, "", "%(name)s", 0, &err);
Kevin Wolfd131c892013-07-02 16:18:35 +0200131''',
132 name=name)
133 else:
134 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200135 visit_start_struct(m, (void **)obj, "%(name)s", name, sizeof(%(name)s), &err);
Kevin Wolfd131c892013-07-02 16:18:35 +0200136''',
137 name=name)
138
139 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200140 if (!err) {
141 if (*obj) {
142 visit_type_%(name)s_fields(m, obj, errp);
143 }
144 visit_end_struct(m, &err);
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200145 }
Markus Armbruster297a3642014-05-07 09:53:54 +0200146 error_propagate(errp, err);
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200147''',
148 name=full_name)
Kevin Wolfd131c892013-07-02 16:18:35 +0200149
Michael Roth06d64c62011-07-19 14:50:41 -0500150 return ret
151
Kevin Wolf14d36302013-09-18 17:22:02 +0200152def generate_visit_struct(expr):
153
Eric Blakefd41dd42015-05-04 09:05:25 -0600154 name = expr['struct']
Kevin Wolf14d36302013-09-18 17:22:02 +0200155 members = expr['data']
Kevin Wolf622f5572013-09-19 11:56:36 +0200156 base = expr.get('base')
Kevin Wolf14d36302013-09-18 17:22:02 +0200157
Kevin Wolf622f5572013-09-19 11:56:36 +0200158 ret = generate_visit_struct_fields(name, "", "", members, base)
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200159
160 ret += mcgen('''
Michael Roth06d64c62011-07-19 14:50:41 -0500161
Amos Kong638ca8a2014-06-10 19:25:51 +0800162void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp)
Michael Roth06d64c62011-07-19 14:50:41 -0500163{
Michael Roth06d64c62011-07-19 14:50:41 -0500164''',
165 name=name)
Paolo Bonzinid1953252012-07-17 16:17:04 +0200166
Paolo Bonzinid1953252012-07-17 16:17:04 +0200167 ret += generate_visit_struct_body("", name, members)
Michael Roth06d64c62011-07-19 14:50:41 -0500168
169 ret += mcgen('''
Michael Roth06d64c62011-07-19 14:50:41 -0500170}
171''')
172 return ret
173
174def generate_visit_list(name, members):
175 return mcgen('''
176
Amos Kong638ca8a2014-06-10 19:25:51 +0800177void visit_type_%(name)sList(Visitor *m, %(name)sList **obj, const char *name, Error **errp)
Michael Roth06d64c62011-07-19 14:50:41 -0500178{
Paolo Bonzinid1953252012-07-17 16:17:04 +0200179 Error *err = NULL;
Markus Armbruster297a3642014-05-07 09:53:54 +0200180 GenericList *i, **prev;
Michael Roth06d64c62011-07-19 14:50:41 -0500181
Markus Armbruster297a3642014-05-07 09:53:54 +0200182 visit_start_list(m, name, &err);
183 if (err) {
184 goto out;
Paolo Bonzinib6f04742012-03-22 12:51:06 +0100185 }
Markus Armbruster297a3642014-05-07 09:53:54 +0200186
187 for (prev = (GenericList **)obj;
188 !err && (i = visit_next_list(m, prev, &err)) != NULL;
189 prev = &i) {
190 %(name)sList *native_i = (%(name)sList *)i;
191 visit_type_%(name)s(m, &native_i->value, NULL, &err);
192 }
193
194 error_propagate(errp, err);
195 err = NULL;
196 visit_end_list(m, &err);
197out:
198 error_propagate(errp, err);
Michael Roth06d64c62011-07-19 14:50:41 -0500199}
200''',
201 name=name)
202
203def generate_visit_enum(name, members):
204 return mcgen('''
205
Amos Kong638ca8a2014-06-10 19:25:51 +0800206void visit_type_%(name)s(Visitor *m, %(name)s *obj, const char *name, Error **errp)
Michael Roth06d64c62011-07-19 14:50:41 -0500207{
208 visit_type_enum(m, (int *)obj, %(name)s_lookup, "%(name)s", name, errp);
209}
210''',
211 name=name)
212
Eric Blake811d04f2015-05-04 09:05:10 -0600213def generate_visit_alternate(name, members):
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200214 ret = mcgen('''
215
Amos Kong638ca8a2014-06-10 19:25:51 +0800216void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp)
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200217{
218 Error *err = NULL;
219
Markus Armbruster297a3642014-05-07 09:53:54 +0200220 visit_start_implicit_struct(m, (void**) obj, sizeof(%(name)s), &err);
221 if (err) {
222 goto out;
223 }
224 visit_get_next_type(m, (int*) &(*obj)->kind, %(name)s_qtypes, name, &err);
225 if (err) {
226 goto out_end;
227 }
228 switch ((*obj)->kind) {
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200229''',
230 name=name)
231
Eric Blakeab916fa2015-05-04 09:05:13 -0600232 # For alternate, always use the default enum type automatically generated
Wenchao Xiab0b58192014-03-04 18:44:36 -0800233 # as "'%sKind' % (name)"
234 disc_type = '%sKind' % (name)
235
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200236 for key in members:
Eric Blakeb52c4b92015-05-04 09:05:00 -0600237 assert (members[key] in builtin_types.keys()
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200238 or find_struct(members[key])
Max Reitze775ba72014-08-20 19:59:34 +0200239 or find_union(members[key])
Eric Blakeab916fa2015-05-04 09:05:13 -0600240 or find_enum(members[key])), "Invalid alternate member"
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200241
Wenchao Xiab0b58192014-03-04 18:44:36 -0800242 enum_full_value = generate_enum_full_value(disc_type, key)
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200243 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200244 case %(enum_full_value)s:
245 visit_type_%(c_type)s(m, &(*obj)->%(c_name)s, name, &err);
246 break;
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200247''',
Wenchao Xiab0b58192014-03-04 18:44:36 -0800248 enum_full_value = enum_full_value,
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200249 c_type = type_name(members[key]),
250 c_name = c_fun(key))
251
252 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200253 default:
254 abort();
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200255 }
Markus Armbruster297a3642014-05-07 09:53:54 +0200256out_end:
257 error_propagate(errp, err);
258 err = NULL;
259 visit_end_implicit_struct(m, &err);
260out:
261 error_propagate(errp, err);
Kevin Wolf69dd62d2013-07-08 16:14:21 +0200262}
263''')
264
265 return ret
266
267
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200268def generate_visit_union(expr):
269
270 name = expr['union']
271 members = expr['data']
272
273 base = expr.get('base')
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200274 discriminator = expr.get('discriminator')
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200275
Wenchao Xiabceae762014-03-06 17:08:56 -0800276 enum_define = discriminator_find_enum_define(expr)
277 if enum_define:
278 # Use the enum type as discriminator
279 ret = ""
280 disc_type = enum_define['enum_name']
281 else:
Eric Blakea8d4a2e2015-05-04 09:05:07 -0600282 # There will always be a discriminator in the C switch code, by default
283 # it is an enum type generated silently as "'%sKind' % (name)"
Wenchao Xiabceae762014-03-06 17:08:56 -0800284 ret = generate_visit_enum('%sKind' % name, members.keys())
285 disc_type = '%sKind' % (name)
Michael Roth06d64c62011-07-19 14:50:41 -0500286
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200287 if base:
Eric Blakea8d4a2e2015-05-04 09:05:07 -0600288 assert discriminator
289 base_fields = find_struct(base)['data'].copy()
290 del base_fields[discriminator]
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200291 ret += generate_visit_struct_fields(name, "", "", base_fields)
292
Markus Armbrusterbe3c7712014-05-07 09:53:50 +0200293 if discriminator:
294 for key in members:
295 ret += generate_visit_implicit_struct(members[key])
296
Michael Roth06d64c62011-07-19 14:50:41 -0500297 ret += mcgen('''
298
Amos Kong638ca8a2014-06-10 19:25:51 +0800299void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp)
Michael Roth06d64c62011-07-19 14:50:41 -0500300{
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100301 Error *err = NULL;
302
Markus Armbruster297a3642014-05-07 09:53:54 +0200303 visit_start_struct(m, (void **)obj, "%(name)s", name, sizeof(%(name)s), &err);
304 if (err) {
305 goto out;
306 }
307 if (*obj) {
Michael Roth06d64c62011-07-19 14:50:41 -0500308''',
309 name=name)
310
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200311 if base:
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200312 ret += mcgen('''
Markus Armbruster468866b2014-05-07 09:53:47 +0200313 visit_type_%(name)s_fields(m, obj, &err);
Markus Armbruster297a3642014-05-07 09:53:54 +0200314 if (err) {
315 goto out_obj;
316 }
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200317''',
318 name=name)
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200319
Richard Henderson7b75d9d2013-10-31 13:26:01 -0700320 if not discriminator:
Wenchao Xiabceae762014-03-06 17:08:56 -0800321 disc_key = "type"
Richard Henderson7b75d9d2013-10-31 13:26:01 -0700322 else:
Wenchao Xiabceae762014-03-06 17:08:56 -0800323 disc_key = discriminator
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200324 ret += mcgen('''
Wenchao Xiabceae762014-03-06 17:08:56 -0800325 visit_type_%(disc_type)s(m, &(*obj)->kind, "%(disc_key)s", &err);
Markus Armbruster297a3642014-05-07 09:53:54 +0200326 if (err) {
327 goto out_obj;
328 }
Michael Rothcee2ded2014-09-18 15:36:40 -0500329 if (!visit_start_union(m, !!(*obj)->data, &err) || err) {
330 goto out_obj;
331 }
Markus Armbruster297a3642014-05-07 09:53:54 +0200332 switch ((*obj)->kind) {
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200333''',
Wenchao Xiabceae762014-03-06 17:08:56 -0800334 disc_type = disc_type,
335 disc_key = disc_key)
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200336
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100337 for key in members:
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200338 if not discriminator:
339 fmt = 'visit_type_%(c_type)s(m, &(*obj)->%(c_name)s, "data", &err);'
340 else:
Markus Armbrusterbe3c7712014-05-07 09:53:50 +0200341 fmt = 'visit_type_implicit_%(c_type)s(m, &(*obj)->%(c_name)s, &err);'
Kevin Wolf50f2bdc2013-07-03 15:58:57 +0200342
Wenchao Xiab0b58192014-03-04 18:44:36 -0800343 enum_full_value = generate_enum_full_value(disc_type, key)
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100344 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200345 case %(enum_full_value)s:
346 ''' + fmt + '''
347 break;
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100348''',
Wenchao Xiab0b58192014-03-04 18:44:36 -0800349 enum_full_value = enum_full_value,
Michael Rothc664aef2013-05-10 17:46:01 -0500350 c_type=type_name(members[key]),
Federico Simoncellic9da2282012-03-20 13:54:35 +0000351 c_name=c_fun(key))
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100352
353 ret += mcgen('''
Markus Armbruster297a3642014-05-07 09:53:54 +0200354 default:
355 abort();
Paolo Bonzinid1953252012-07-17 16:17:04 +0200356 }
Markus Armbruster297a3642014-05-07 09:53:54 +0200357out_obj:
Paolo Bonzinid1953252012-07-17 16:17:04 +0200358 error_propagate(errp, err);
359 err = NULL;
Michael Rothcee2ded2014-09-18 15:36:40 -0500360 visit_end_union(m, !!(*obj)->data, &err);
361 error_propagate(errp, err);
362 err = NULL;
Markus Armbruster468866b2014-05-07 09:53:47 +0200363 }
Markus Armbruster297a3642014-05-07 09:53:54 +0200364 visit_end_struct(m, &err);
365out:
366 error_propagate(errp, err);
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100367}
368''')
369
Michael Roth06d64c62011-07-19 14:50:41 -0500370 return ret
371
Eric Blake6540e9f2015-04-10 15:07:59 -0600372def generate_declaration(name, members, builtin_type=False):
Michael Roth7c946bc2013-05-10 17:46:02 -0500373 ret = ""
374 if not builtin_type:
375 ret += mcgen('''
Michael Roth06d64c62011-07-19 14:50:41 -0500376
Amos Kong638ca8a2014-06-10 19:25:51 +0800377void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp);
Michael Roth06d64c62011-07-19 14:50:41 -0500378''',
Eric Blake6540e9f2015-04-10 15:07:59 -0600379 name=name)
Michael Roth06d64c62011-07-19 14:50:41 -0500380
Eric Blake6540e9f2015-04-10 15:07:59 -0600381 ret += mcgen('''
Amos Kong638ca8a2014-06-10 19:25:51 +0800382void visit_type_%(name)sList(Visitor *m, %(name)sList **obj, const char *name, Error **errp);
Michael Roth06d64c62011-07-19 14:50:41 -0500383''',
384 name=name)
385
386 return ret
387
Eric Blake6540e9f2015-04-10 15:07:59 -0600388def generate_enum_declaration(name, members):
389 ret = mcgen('''
Amos Kong638ca8a2014-06-10 19:25:51 +0800390void visit_type_%(name)sList(Visitor *m, %(name)sList **obj, const char *name, Error **errp);
Amos Kongb9c4b482012-08-31 10:56:23 +0800391''',
Eric Blake6540e9f2015-04-10 15:07:59 -0600392 name=name)
Amos Kongb9c4b482012-08-31 10:56:23 +0800393
394 return ret
395
Eric Blake6540e9f2015-04-10 15:07:59 -0600396def generate_decl_enum(name, members):
Michael Roth06d64c62011-07-19 14:50:41 -0500397 return mcgen('''
398
Amos Kong638ca8a2014-06-10 19:25:51 +0800399void visit_type_%(name)s(Visitor *m, %(name)s *obj, const char *name, Error **errp);
Michael Roth06d64c62011-07-19 14:50:41 -0500400''',
401 name=name)
402
403try:
LluĂ­s Vilanova33aaad52014-05-02 15:52:35 +0200404 opts, args = getopt.gnu_getopt(sys.argv[1:], "chbp:i:o:",
Michael Roth7c946bc2013-05-10 17:46:02 -0500405 ["source", "header", "builtins", "prefix=",
LluĂ­s Vilanova33aaad52014-05-02 15:52:35 +0200406 "input-file=", "output-dir="])
Michael Roth06d64c62011-07-19 14:50:41 -0500407except getopt.GetoptError, err:
408 print str(err)
409 sys.exit(1)
410
LluĂ­s Vilanova33aaad52014-05-02 15:52:35 +0200411input_file = ""
Michael Roth06d64c62011-07-19 14:50:41 -0500412output_dir = ""
413prefix = ""
414c_file = 'qapi-visit.c'
415h_file = 'qapi-visit.h'
416
Avi Kivity8d3bc512011-12-27 16:02:16 +0200417do_c = False
418do_h = False
Michael Roth7c946bc2013-05-10 17:46:02 -0500419do_builtins = False
Avi Kivity8d3bc512011-12-27 16:02:16 +0200420
Michael Roth06d64c62011-07-19 14:50:41 -0500421for o, a in opts:
422 if o in ("-p", "--prefix"):
423 prefix = a
LluĂ­s Vilanova33aaad52014-05-02 15:52:35 +0200424 elif o in ("-i", "--input-file"):
425 input_file = a
Michael Roth06d64c62011-07-19 14:50:41 -0500426 elif o in ("-o", "--output-dir"):
427 output_dir = a + "/"
Avi Kivity8d3bc512011-12-27 16:02:16 +0200428 elif o in ("-c", "--source"):
Avi Kivity8d3bc512011-12-27 16:02:16 +0200429 do_c = True
Avi Kivity19bf7c82011-12-28 12:26:58 +0200430 elif o in ("-h", "--header"):
431 do_h = True
Michael Roth7c946bc2013-05-10 17:46:02 -0500432 elif o in ("-b", "--builtins"):
433 do_builtins = True
Avi Kivity8d3bc512011-12-27 16:02:16 +0200434
435if not do_c and not do_h:
436 do_c = True
437 do_h = True
Michael Roth06d64c62011-07-19 14:50:41 -0500438
439c_file = output_dir + prefix + c_file
440h_file = output_dir + prefix + h_file
441
442try:
443 os.makedirs(output_dir)
444except os.error, e:
445 if e.errno != errno.EEXIST:
446 raise
447
Avi Kivity8d3bc512011-12-27 16:02:16 +0200448def maybe_open(really, name, opt):
Avi Kivity8d3bc512011-12-27 16:02:16 +0200449 if really:
450 return open(name, opt)
Avi Kivity19bf7c82011-12-28 12:26:58 +0200451 else:
452 import StringIO
453 return StringIO.StringIO()
Avi Kivity8d3bc512011-12-27 16:02:16 +0200454
455fdef = maybe_open(do_c, c_file, 'w')
456fdecl = maybe_open(do_h, h_file, 'w')
Michael Roth06d64c62011-07-19 14:50:41 -0500457
458fdef.write(mcgen('''
459/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */
460
461/*
462 * schema-defined QAPI visitor functions
463 *
464 * Copyright IBM, Corp. 2011
465 *
466 * Authors:
467 * Anthony Liguori <aliguori@us.ibm.com>
468 *
469 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
470 * See the COPYING.LIB file in the top-level directory.
471 *
472 */
473
Paolo Bonzini79ee7df2012-12-06 11:22:34 +0100474#include "qemu-common.h"
Michael Roth06d64c62011-07-19 14:50:41 -0500475#include "%(header)s"
476''',
477 header=basename(h_file)))
478
479fdecl.write(mcgen('''
480/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */
481
482/*
Markus Armbruster297a3642014-05-07 09:53:54 +0200483 * schema-defined QAPI visitor functions
Michael Roth06d64c62011-07-19 14:50:41 -0500484 *
485 * Copyright IBM, Corp. 2011
486 *
487 * Authors:
488 * Anthony Liguori <aliguori@us.ibm.com>
489 *
490 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
491 * See the COPYING.LIB file in the top-level directory.
492 *
493 */
494
495#ifndef %(guard)s
496#define %(guard)s
497
Paolo Bonzini7b1b5d12012-12-17 18:19:43 +0100498#include "qapi/visitor.h"
Michael Roth06d64c62011-07-19 14:50:41 -0500499#include "%(prefix)sqapi-types.h"
Michael Roth7c946bc2013-05-10 17:46:02 -0500500
Michael Roth06d64c62011-07-19 14:50:41 -0500501''',
502 prefix=prefix, guard=guardname(h_file)))
503
LluĂ­s Vilanova33aaad52014-05-02 15:52:35 +0200504exprs = parse_schema(input_file)
Michael Roth06d64c62011-07-19 14:50:41 -0500505
Michael Roth7c946bc2013-05-10 17:46:02 -0500506# to avoid header dependency hell, we always generate declarations
507# for built-in types in our header files and simply guard them
508fdecl.write(guardstart("QAPI_VISIT_BUILTIN_VISITOR_DECL"))
Eric Blakeb52c4b92015-05-04 09:05:00 -0600509for typename in builtin_types.keys():
Eric Blake6540e9f2015-04-10 15:07:59 -0600510 fdecl.write(generate_declaration(typename, None, builtin_type=True))
Michael Roth7c946bc2013-05-10 17:46:02 -0500511fdecl.write(guardend("QAPI_VISIT_BUILTIN_VISITOR_DECL"))
512
513# ...this doesn't work for cases where we link in multiple objects that
514# have the functions defined, so we use -b option to provide control
515# over these cases
516if do_builtins:
Eric Blakeb52c4b92015-05-04 09:05:00 -0600517 for typename in builtin_types.keys():
Michael Roth7c946bc2013-05-10 17:46:02 -0500518 fdef.write(generate_visit_list(typename, None))
Michael Roth7c946bc2013-05-10 17:46:02 -0500519
Michael Roth06d64c62011-07-19 14:50:41 -0500520for expr in exprs:
Eric Blakefd41dd42015-05-04 09:05:25 -0600521 if expr.has_key('struct'):
Kevin Wolf14d36302013-09-18 17:22:02 +0200522 ret = generate_visit_struct(expr)
Eric Blakefd41dd42015-05-04 09:05:25 -0600523 ret += generate_visit_list(expr['struct'], expr['data'])
Michael Roth06d64c62011-07-19 14:50:41 -0500524 fdef.write(ret)
525
Eric Blakefd41dd42015-05-04 09:05:25 -0600526 ret = generate_declaration(expr['struct'], expr['data'])
Michael Roth06d64c62011-07-19 14:50:41 -0500527 fdecl.write(ret)
528 elif expr.has_key('union'):
Kevin Wolf0aef92b2013-07-02 16:20:04 +0200529 ret = generate_visit_union(expr)
Paolo Bonzinidc8fb6d2012-03-06 18:55:56 +0100530 ret += generate_visit_list(expr['union'], expr['data'])
Michael Roth06d64c62011-07-19 14:50:41 -0500531 fdef.write(ret)
532
Wenchao Xiabceae762014-03-06 17:08:56 -0800533 enum_define = discriminator_find_enum_define(expr)
534 ret = ""
535 if not enum_define:
536 ret = generate_decl_enum('%sKind' % expr['union'],
537 expr['data'].keys())
Michael Roth06d64c62011-07-19 14:50:41 -0500538 ret += generate_declaration(expr['union'], expr['data'])
539 fdecl.write(ret)
Eric Blakeab916fa2015-05-04 09:05:13 -0600540 elif expr.has_key('alternate'):
541 ret = generate_visit_alternate(expr['alternate'], expr['data'])
542 ret += generate_visit_list(expr['alternate'], expr['data'])
543 fdef.write(ret)
544
545 ret = generate_decl_enum('%sKind' % expr['alternate'],
546 expr['data'].keys())
547 ret += generate_declaration(expr['alternate'], expr['data'])
548 fdecl.write(ret)
Michael Roth06d64c62011-07-19 14:50:41 -0500549 elif expr.has_key('enum'):
Amos Kongb9c4b482012-08-31 10:56:23 +0800550 ret = generate_visit_list(expr['enum'], expr['data'])
551 ret += generate_visit_enum(expr['enum'], expr['data'])
Michael Roth06d64c62011-07-19 14:50:41 -0500552 fdef.write(ret)
553
554 ret = generate_decl_enum(expr['enum'], expr['data'])
Amos Kongb9c4b482012-08-31 10:56:23 +0800555 ret += generate_enum_declaration(expr['enum'], expr['data'])
Michael Roth06d64c62011-07-19 14:50:41 -0500556 fdecl.write(ret)
557
558fdecl.write('''
559#endif
560''')
561
562fdecl.flush()
563fdecl.close()
564
565fdef.flush()
566fdef.close()