Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 1 | # |
| 2 | # QAPI visitor generator |
| 3 | # |
| 4 | # Copyright IBM, Corp. 2011 |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 5 | # Copyright (C) 2014-2015 Red Hat, Inc. |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 6 | # |
| 7 | # Authors: |
| 8 | # Anthony Liguori <aliguori@us.ibm.com> |
| 9 | # Michael Roth <mdroth@linux.vnet.ibm.com> |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 10 | # Markus Armbruster <armbru@redhat.com> |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 11 | # |
Markus Armbruster | 678e48a | 2014-03-01 08:40:34 +0100 | [diff] [blame] | 12 | # This work is licensed under the terms of the GNU GPL, version 2. |
| 13 | # See the COPYING file in the top-level directory. |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 14 | |
| 15 | from ordereddict import OrderedDict |
| 16 | from qapi import * |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 17 | import re |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 18 | import sys |
| 19 | import os |
| 20 | import getopt |
| 21 | import errno |
| 22 | |
Markus Armbruster | be3c771 | 2014-05-07 09:53:50 +0200 | [diff] [blame] | 23 | implicit_structs = [] |
| 24 | |
| 25 | def 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 | |
| 32 | static 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 Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 38 | visit_type_%(c_type)s_fields(m, obj, errp); |
Markus Armbruster | be3c771 | 2014-05-07 09:53:50 +0200 | [diff] [blame] | 39 | visit_end_implicit_struct(m, &err); |
| 40 | } |
| 41 | error_propagate(errp, err); |
| 42 | } |
| 43 | ''', |
| 44 | c_type=type_name(type)) |
| 45 | |
Kevin Wolf | 622f557 | 2013-09-19 11:56:36 +0200 | [diff] [blame] | 46 | def generate_visit_struct_fields(name, field_prefix, fn_prefix, members, base = None): |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 47 | substructs = [] |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 48 | ret = '' |
Richard Henderson | 7b75d9d | 2013-10-31 13:26:01 -0700 | [diff] [blame] | 49 | if not fn_prefix: |
| 50 | full_name = name |
| 51 | else: |
| 52 | full_name = "%s_%s" % (name, fn_prefix) |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 53 | |
Markus Armbruster | be3c771 | 2014-05-07 09:53:50 +0200 | [diff] [blame] | 54 | if base: |
| 55 | ret += generate_visit_implicit_struct(base) |
| 56 | |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 57 | ret += mcgen(''' |
| 58 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 59 | static void visit_type_%(full_name)s_fields(Visitor *m, %(name)s **obj, Error **errp) |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 60 | { |
| 61 | Error *err = NULL; |
| 62 | ''', |
| 63 | name=name, full_name=full_name) |
| 64 | push_indent() |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 65 | |
Kevin Wolf | 622f557 | 2013-09-19 11:56:36 +0200 | [diff] [blame] | 66 | if base: |
| 67 | ret += mcgen(''' |
Markus Armbruster | be3c771 | 2014-05-07 09:53:50 +0200 | [diff] [blame] | 68 | visit_type_implicit_%(type)s(m, &(*obj)->%(c_prefix)s%(c_name)s, &err); |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 69 | if (err) { |
| 70 | goto out; |
| 71 | } |
Kevin Wolf | 622f557 | 2013-09-19 11:56:36 +0200 | [diff] [blame] | 72 | ''', |
| 73 | c_prefix=c_var(field_prefix), |
| 74 | type=type_name(base), c_name=c_var('base')) |
| 75 | |
Eric Blake | 6b5abc7 | 2015-05-04 09:05:33 -0600 | [diff] [blame^] | 76 | for argname, argentry, optional in parse_args(members): |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 77 | if optional: |
| 78 | ret += mcgen(''' |
Markus Armbruster | e2cd0f4 | 2014-05-07 09:53:46 +0200 | [diff] [blame] | 79 | visit_optional(m, &(*obj)->%(c_prefix)shas_%(c_name)s, "%(name)s", &err); |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 80 | if (!err && (*obj)->%(prefix)shas_%(c_name)s) { |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 81 | ''', |
| 82 | c_prefix=c_var(field_prefix), prefix=field_prefix, |
| 83 | c_name=c_var(argname), name=argname) |
| 84 | push_indent() |
| 85 | |
Eric Blake | 6b5abc7 | 2015-05-04 09:05:33 -0600 | [diff] [blame^] | 86 | ret += mcgen(''' |
Markus Armbruster | 949ceeb | 2014-03-01 08:40:37 +0100 | [diff] [blame] | 87 | visit_type_%(type)s(m, &(*obj)->%(c_prefix)s%(c_name)s, "%(name)s", &err); |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 88 | ''', |
Eric Blake | 6b5abc7 | 2015-05-04 09:05:33 -0600 | [diff] [blame^] | 89 | c_prefix=c_var(field_prefix), prefix=field_prefix, |
| 90 | type=type_name(argentry), c_name=c_var(argname), |
| 91 | name=argname) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 92 | |
| 93 | if optional: |
| 94 | pop_indent() |
| 95 | ret += mcgen(''' |
| 96 | } |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 97 | ''') |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 98 | ret += mcgen(''' |
| 99 | if (err) { |
| 100 | goto out; |
| 101 | } |
| 102 | ''') |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 103 | |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 104 | pop_indent() |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 105 | if re.search('^ *goto out\\;', ret, re.MULTILINE): |
| 106 | ret += mcgen(''' |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 107 | |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 108 | out: |
| 109 | ''') |
| 110 | ret += mcgen(''' |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 111 | error_propagate(errp, err); |
| 112 | } |
| 113 | ''') |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 114 | return ret |
| 115 | |
| 116 | |
| 117 | def generate_visit_struct_body(field_prefix, name, members): |
| 118 | ret = mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 119 | Error *err = NULL; |
| 120 | |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 121 | ''') |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 122 | |
Richard Henderson | 7b75d9d | 2013-10-31 13:26:01 -0700 | [diff] [blame] | 123 | if not field_prefix: |
| 124 | full_name = name |
| 125 | else: |
| 126 | full_name = "%s_%s" % (field_prefix, name) |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 127 | |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 128 | if len(field_prefix): |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 129 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 130 | visit_start_struct(m, NULL, "", "%(name)s", 0, &err); |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 131 | ''', |
| 132 | name=name) |
| 133 | else: |
| 134 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 135 | visit_start_struct(m, (void **)obj, "%(name)s", name, sizeof(%(name)s), &err); |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 136 | ''', |
| 137 | name=name) |
| 138 | |
| 139 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 140 | if (!err) { |
| 141 | if (*obj) { |
| 142 | visit_type_%(name)s_fields(m, obj, errp); |
| 143 | } |
| 144 | visit_end_struct(m, &err); |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 145 | } |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 146 | error_propagate(errp, err); |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 147 | ''', |
| 148 | name=full_name) |
Kevin Wolf | d131c89 | 2013-07-02 16:18:35 +0200 | [diff] [blame] | 149 | |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 150 | return ret |
| 151 | |
Kevin Wolf | 14d3630 | 2013-09-18 17:22:02 +0200 | [diff] [blame] | 152 | def generate_visit_struct(expr): |
| 153 | |
Eric Blake | fd41dd4 | 2015-05-04 09:05:25 -0600 | [diff] [blame] | 154 | name = expr['struct'] |
Kevin Wolf | 14d3630 | 2013-09-18 17:22:02 +0200 | [diff] [blame] | 155 | members = expr['data'] |
Kevin Wolf | 622f557 | 2013-09-19 11:56:36 +0200 | [diff] [blame] | 156 | base = expr.get('base') |
Kevin Wolf | 14d3630 | 2013-09-18 17:22:02 +0200 | [diff] [blame] | 157 | |
Kevin Wolf | 622f557 | 2013-09-19 11:56:36 +0200 | [diff] [blame] | 158 | ret = generate_visit_struct_fields(name, "", "", members, base) |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 159 | |
| 160 | ret += mcgen(''' |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 161 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 162 | void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 163 | { |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 164 | ''', |
| 165 | name=name) |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 166 | |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 167 | ret += generate_visit_struct_body("", name, members) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 168 | |
| 169 | ret += mcgen(''' |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 170 | } |
| 171 | ''') |
| 172 | return ret |
| 173 | |
| 174 | def generate_visit_list(name, members): |
| 175 | return mcgen(''' |
| 176 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 177 | void visit_type_%(name)sList(Visitor *m, %(name)sList **obj, const char *name, Error **errp) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 178 | { |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 179 | Error *err = NULL; |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 180 | GenericList *i, **prev; |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 181 | |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 182 | visit_start_list(m, name, &err); |
| 183 | if (err) { |
| 184 | goto out; |
Paolo Bonzini | b6f0474 | 2012-03-22 12:51:06 +0100 | [diff] [blame] | 185 | } |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 186 | |
| 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); |
| 197 | out: |
| 198 | error_propagate(errp, err); |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 199 | } |
| 200 | ''', |
| 201 | name=name) |
| 202 | |
| 203 | def generate_visit_enum(name, members): |
| 204 | return mcgen(''' |
| 205 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 206 | void visit_type_%(name)s(Visitor *m, %(name)s *obj, const char *name, Error **errp) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 207 | { |
| 208 | visit_type_enum(m, (int *)obj, %(name)s_lookup, "%(name)s", name, errp); |
| 209 | } |
| 210 | ''', |
| 211 | name=name) |
| 212 | |
Eric Blake | 811d04f | 2015-05-04 09:05:10 -0600 | [diff] [blame] | 213 | def generate_visit_alternate(name, members): |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 214 | ret = mcgen(''' |
| 215 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 216 | void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp) |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 217 | { |
| 218 | Error *err = NULL; |
| 219 | |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 220 | 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 Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 229 | ''', |
| 230 | name=name) |
| 231 | |
Eric Blake | ab916fa | 2015-05-04 09:05:13 -0600 | [diff] [blame] | 232 | # For alternate, always use the default enum type automatically generated |
Wenchao Xia | b0b5819 | 2014-03-04 18:44:36 -0800 | [diff] [blame] | 233 | # as "'%sKind' % (name)" |
| 234 | disc_type = '%sKind' % (name) |
| 235 | |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 236 | for key in members: |
Eric Blake | b52c4b9 | 2015-05-04 09:05:00 -0600 | [diff] [blame] | 237 | assert (members[key] in builtin_types.keys() |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 238 | or find_struct(members[key]) |
Max Reitz | e775ba7 | 2014-08-20 19:59:34 +0200 | [diff] [blame] | 239 | or find_union(members[key]) |
Eric Blake | ab916fa | 2015-05-04 09:05:13 -0600 | [diff] [blame] | 240 | or find_enum(members[key])), "Invalid alternate member" |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 241 | |
Wenchao Xia | b0b5819 | 2014-03-04 18:44:36 -0800 | [diff] [blame] | 242 | enum_full_value = generate_enum_full_value(disc_type, key) |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 243 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 244 | case %(enum_full_value)s: |
| 245 | visit_type_%(c_type)s(m, &(*obj)->%(c_name)s, name, &err); |
| 246 | break; |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 247 | ''', |
Wenchao Xia | b0b5819 | 2014-03-04 18:44:36 -0800 | [diff] [blame] | 248 | enum_full_value = enum_full_value, |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 249 | c_type = type_name(members[key]), |
| 250 | c_name = c_fun(key)) |
| 251 | |
| 252 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 253 | default: |
| 254 | abort(); |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 255 | } |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 256 | out_end: |
| 257 | error_propagate(errp, err); |
| 258 | err = NULL; |
| 259 | visit_end_implicit_struct(m, &err); |
| 260 | out: |
| 261 | error_propagate(errp, err); |
Kevin Wolf | 69dd62d | 2013-07-08 16:14:21 +0200 | [diff] [blame] | 262 | } |
| 263 | ''') |
| 264 | |
| 265 | return ret |
| 266 | |
| 267 | |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 268 | def generate_visit_union(expr): |
| 269 | |
| 270 | name = expr['union'] |
| 271 | members = expr['data'] |
| 272 | |
| 273 | base = expr.get('base') |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 274 | discriminator = expr.get('discriminator') |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 275 | |
Wenchao Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 276 | 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 Blake | a8d4a2e | 2015-05-04 09:05:07 -0600 | [diff] [blame] | 282 | # 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 Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 284 | ret = generate_visit_enum('%sKind' % name, members.keys()) |
| 285 | disc_type = '%sKind' % (name) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 286 | |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 287 | if base: |
Eric Blake | a8d4a2e | 2015-05-04 09:05:07 -0600 | [diff] [blame] | 288 | assert discriminator |
| 289 | base_fields = find_struct(base)['data'].copy() |
| 290 | del base_fields[discriminator] |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 291 | ret += generate_visit_struct_fields(name, "", "", base_fields) |
| 292 | |
Markus Armbruster | be3c771 | 2014-05-07 09:53:50 +0200 | [diff] [blame] | 293 | if discriminator: |
| 294 | for key in members: |
| 295 | ret += generate_visit_implicit_struct(members[key]) |
| 296 | |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 297 | ret += mcgen(''' |
| 298 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 299 | void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 300 | { |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 301 | Error *err = NULL; |
| 302 | |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 303 | visit_start_struct(m, (void **)obj, "%(name)s", name, sizeof(%(name)s), &err); |
| 304 | if (err) { |
| 305 | goto out; |
| 306 | } |
| 307 | if (*obj) { |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 308 | ''', |
| 309 | name=name) |
| 310 | |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 311 | if base: |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 312 | ret += mcgen(''' |
Markus Armbruster | 468866b | 2014-05-07 09:53:47 +0200 | [diff] [blame] | 313 | visit_type_%(name)s_fields(m, obj, &err); |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 314 | if (err) { |
| 315 | goto out_obj; |
| 316 | } |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 317 | ''', |
| 318 | name=name) |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 319 | |
Richard Henderson | 7b75d9d | 2013-10-31 13:26:01 -0700 | [diff] [blame] | 320 | if not discriminator: |
Wenchao Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 321 | disc_key = "type" |
Richard Henderson | 7b75d9d | 2013-10-31 13:26:01 -0700 | [diff] [blame] | 322 | else: |
Wenchao Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 323 | disc_key = discriminator |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 324 | ret += mcgen(''' |
Wenchao Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 325 | visit_type_%(disc_type)s(m, &(*obj)->kind, "%(disc_key)s", &err); |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 326 | if (err) { |
| 327 | goto out_obj; |
| 328 | } |
Michael Roth | cee2ded | 2014-09-18 15:36:40 -0500 | [diff] [blame] | 329 | if (!visit_start_union(m, !!(*obj)->data, &err) || err) { |
| 330 | goto out_obj; |
| 331 | } |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 332 | switch ((*obj)->kind) { |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 333 | ''', |
Wenchao Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 334 | disc_type = disc_type, |
| 335 | disc_key = disc_key) |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 336 | |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 337 | for key in members: |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 338 | if not discriminator: |
| 339 | fmt = 'visit_type_%(c_type)s(m, &(*obj)->%(c_name)s, "data", &err);' |
| 340 | else: |
Markus Armbruster | be3c771 | 2014-05-07 09:53:50 +0200 | [diff] [blame] | 341 | fmt = 'visit_type_implicit_%(c_type)s(m, &(*obj)->%(c_name)s, &err);' |
Kevin Wolf | 50f2bdc | 2013-07-03 15:58:57 +0200 | [diff] [blame] | 342 | |
Wenchao Xia | b0b5819 | 2014-03-04 18:44:36 -0800 | [diff] [blame] | 343 | enum_full_value = generate_enum_full_value(disc_type, key) |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 344 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 345 | case %(enum_full_value)s: |
| 346 | ''' + fmt + ''' |
| 347 | break; |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 348 | ''', |
Wenchao Xia | b0b5819 | 2014-03-04 18:44:36 -0800 | [diff] [blame] | 349 | enum_full_value = enum_full_value, |
Michael Roth | c664aef | 2013-05-10 17:46:01 -0500 | [diff] [blame] | 350 | c_type=type_name(members[key]), |
Federico Simoncelli | c9da228 | 2012-03-20 13:54:35 +0000 | [diff] [blame] | 351 | c_name=c_fun(key)) |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 352 | |
| 353 | ret += mcgen(''' |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 354 | default: |
| 355 | abort(); |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 356 | } |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 357 | out_obj: |
Paolo Bonzini | d195325 | 2012-07-17 16:17:04 +0200 | [diff] [blame] | 358 | error_propagate(errp, err); |
| 359 | err = NULL; |
Michael Roth | cee2ded | 2014-09-18 15:36:40 -0500 | [diff] [blame] | 360 | visit_end_union(m, !!(*obj)->data, &err); |
| 361 | error_propagate(errp, err); |
| 362 | err = NULL; |
Markus Armbruster | 468866b | 2014-05-07 09:53:47 +0200 | [diff] [blame] | 363 | } |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 364 | visit_end_struct(m, &err); |
| 365 | out: |
| 366 | error_propagate(errp, err); |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 367 | } |
| 368 | ''') |
| 369 | |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 370 | return ret |
| 371 | |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 372 | def generate_declaration(name, members, builtin_type=False): |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 373 | ret = "" |
| 374 | if not builtin_type: |
| 375 | ret += mcgen(''' |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 376 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 377 | void visit_type_%(name)s(Visitor *m, %(name)s **obj, const char *name, Error **errp); |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 378 | ''', |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 379 | name=name) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 380 | |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 381 | ret += mcgen(''' |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 382 | void visit_type_%(name)sList(Visitor *m, %(name)sList **obj, const char *name, Error **errp); |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 383 | ''', |
| 384 | name=name) |
| 385 | |
| 386 | return ret |
| 387 | |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 388 | def generate_enum_declaration(name, members): |
| 389 | ret = mcgen(''' |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 390 | void visit_type_%(name)sList(Visitor *m, %(name)sList **obj, const char *name, Error **errp); |
Amos Kong | b9c4b48 | 2012-08-31 10:56:23 +0800 | [diff] [blame] | 391 | ''', |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 392 | name=name) |
Amos Kong | b9c4b48 | 2012-08-31 10:56:23 +0800 | [diff] [blame] | 393 | |
| 394 | return ret |
| 395 | |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 396 | def generate_decl_enum(name, members): |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 397 | return mcgen(''' |
| 398 | |
Amos Kong | 638ca8a | 2014-06-10 19:25:51 +0800 | [diff] [blame] | 399 | void visit_type_%(name)s(Visitor *m, %(name)s *obj, const char *name, Error **errp); |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 400 | ''', |
| 401 | name=name) |
| 402 | |
| 403 | try: |
LluĂs Vilanova | 33aaad5 | 2014-05-02 15:52:35 +0200 | [diff] [blame] | 404 | opts, args = getopt.gnu_getopt(sys.argv[1:], "chbp:i:o:", |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 405 | ["source", "header", "builtins", "prefix=", |
LluĂs Vilanova | 33aaad5 | 2014-05-02 15:52:35 +0200 | [diff] [blame] | 406 | "input-file=", "output-dir="]) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 407 | except getopt.GetoptError, err: |
| 408 | print str(err) |
| 409 | sys.exit(1) |
| 410 | |
LluĂs Vilanova | 33aaad5 | 2014-05-02 15:52:35 +0200 | [diff] [blame] | 411 | input_file = "" |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 412 | output_dir = "" |
| 413 | prefix = "" |
| 414 | c_file = 'qapi-visit.c' |
| 415 | h_file = 'qapi-visit.h' |
| 416 | |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 417 | do_c = False |
| 418 | do_h = False |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 419 | do_builtins = False |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 420 | |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 421 | for o, a in opts: |
| 422 | if o in ("-p", "--prefix"): |
| 423 | prefix = a |
LluĂs Vilanova | 33aaad5 | 2014-05-02 15:52:35 +0200 | [diff] [blame] | 424 | elif o in ("-i", "--input-file"): |
| 425 | input_file = a |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 426 | elif o in ("-o", "--output-dir"): |
| 427 | output_dir = a + "/" |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 428 | elif o in ("-c", "--source"): |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 429 | do_c = True |
Avi Kivity | 19bf7c8 | 2011-12-28 12:26:58 +0200 | [diff] [blame] | 430 | elif o in ("-h", "--header"): |
| 431 | do_h = True |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 432 | elif o in ("-b", "--builtins"): |
| 433 | do_builtins = True |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 434 | |
| 435 | if not do_c and not do_h: |
| 436 | do_c = True |
| 437 | do_h = True |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 438 | |
| 439 | c_file = output_dir + prefix + c_file |
| 440 | h_file = output_dir + prefix + h_file |
| 441 | |
| 442 | try: |
| 443 | os.makedirs(output_dir) |
| 444 | except os.error, e: |
| 445 | if e.errno != errno.EEXIST: |
| 446 | raise |
| 447 | |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 448 | def maybe_open(really, name, opt): |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 449 | if really: |
| 450 | return open(name, opt) |
Avi Kivity | 19bf7c8 | 2011-12-28 12:26:58 +0200 | [diff] [blame] | 451 | else: |
| 452 | import StringIO |
| 453 | return StringIO.StringIO() |
Avi Kivity | 8d3bc51 | 2011-12-27 16:02:16 +0200 | [diff] [blame] | 454 | |
| 455 | fdef = maybe_open(do_c, c_file, 'w') |
| 456 | fdecl = maybe_open(do_h, h_file, 'w') |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 457 | |
| 458 | fdef.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 Bonzini | 79ee7df | 2012-12-06 11:22:34 +0100 | [diff] [blame] | 474 | #include "qemu-common.h" |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 475 | #include "%(header)s" |
| 476 | ''', |
| 477 | header=basename(h_file))) |
| 478 | |
| 479 | fdecl.write(mcgen(''' |
| 480 | /* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */ |
| 481 | |
| 482 | /* |
Markus Armbruster | 297a364 | 2014-05-07 09:53:54 +0200 | [diff] [blame] | 483 | * schema-defined QAPI visitor functions |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 484 | * |
| 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 Bonzini | 7b1b5d1 | 2012-12-17 18:19:43 +0100 | [diff] [blame] | 498 | #include "qapi/visitor.h" |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 499 | #include "%(prefix)sqapi-types.h" |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 500 | |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 501 | ''', |
| 502 | prefix=prefix, guard=guardname(h_file))) |
| 503 | |
LluĂs Vilanova | 33aaad5 | 2014-05-02 15:52:35 +0200 | [diff] [blame] | 504 | exprs = parse_schema(input_file) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 505 | |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 506 | # to avoid header dependency hell, we always generate declarations |
| 507 | # for built-in types in our header files and simply guard them |
| 508 | fdecl.write(guardstart("QAPI_VISIT_BUILTIN_VISITOR_DECL")) |
Eric Blake | b52c4b9 | 2015-05-04 09:05:00 -0600 | [diff] [blame] | 509 | for typename in builtin_types.keys(): |
Eric Blake | 6540e9f | 2015-04-10 15:07:59 -0600 | [diff] [blame] | 510 | fdecl.write(generate_declaration(typename, None, builtin_type=True)) |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 511 | fdecl.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 |
| 516 | if do_builtins: |
Eric Blake | b52c4b9 | 2015-05-04 09:05:00 -0600 | [diff] [blame] | 517 | for typename in builtin_types.keys(): |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 518 | fdef.write(generate_visit_list(typename, None)) |
Michael Roth | 7c946bc | 2013-05-10 17:46:02 -0500 | [diff] [blame] | 519 | |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 520 | for expr in exprs: |
Eric Blake | fd41dd4 | 2015-05-04 09:05:25 -0600 | [diff] [blame] | 521 | if expr.has_key('struct'): |
Kevin Wolf | 14d3630 | 2013-09-18 17:22:02 +0200 | [diff] [blame] | 522 | ret = generate_visit_struct(expr) |
Eric Blake | fd41dd4 | 2015-05-04 09:05:25 -0600 | [diff] [blame] | 523 | ret += generate_visit_list(expr['struct'], expr['data']) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 524 | fdef.write(ret) |
| 525 | |
Eric Blake | fd41dd4 | 2015-05-04 09:05:25 -0600 | [diff] [blame] | 526 | ret = generate_declaration(expr['struct'], expr['data']) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 527 | fdecl.write(ret) |
| 528 | elif expr.has_key('union'): |
Kevin Wolf | 0aef92b | 2013-07-02 16:20:04 +0200 | [diff] [blame] | 529 | ret = generate_visit_union(expr) |
Paolo Bonzini | dc8fb6d | 2012-03-06 18:55:56 +0100 | [diff] [blame] | 530 | ret += generate_visit_list(expr['union'], expr['data']) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 531 | fdef.write(ret) |
| 532 | |
Wenchao Xia | bceae76 | 2014-03-06 17:08:56 -0800 | [diff] [blame] | 533 | 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 Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 538 | ret += generate_declaration(expr['union'], expr['data']) |
| 539 | fdecl.write(ret) |
Eric Blake | ab916fa | 2015-05-04 09:05:13 -0600 | [diff] [blame] | 540 | 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 Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 549 | elif expr.has_key('enum'): |
Amos Kong | b9c4b48 | 2012-08-31 10:56:23 +0800 | [diff] [blame] | 550 | ret = generate_visit_list(expr['enum'], expr['data']) |
| 551 | ret += generate_visit_enum(expr['enum'], expr['data']) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 552 | fdef.write(ret) |
| 553 | |
| 554 | ret = generate_decl_enum(expr['enum'], expr['data']) |
Amos Kong | b9c4b48 | 2012-08-31 10:56:23 +0800 | [diff] [blame] | 555 | ret += generate_enum_declaration(expr['enum'], expr['data']) |
Michael Roth | 06d64c6 | 2011-07-19 14:50:41 -0500 | [diff] [blame] | 556 | fdecl.write(ret) |
| 557 | |
| 558 | fdecl.write(''' |
| 559 | #endif |
| 560 | ''') |
| 561 | |
| 562 | fdecl.flush() |
| 563 | fdecl.close() |
| 564 | |
| 565 | fdef.flush() |
| 566 | fdef.close() |