scripts/qapi-types.py

   1 #
   2 # QAPI types generator
   3 #
   4 # Copyright IBM, Corp. 2011
   5 # Copyright (c) 2013-2015 Red Hat Inc.
   6 #
   7 # Authors:
   8 #  Anthony Liguori <aliguori@us.ibm.com>
   9 #  Markus Armbruster <armbru@redhat.com>
  10 #
  11 # This work is licensed under the terms of the GNU GPL, version 2.
  12 # See the COPYING file in the top-level directory.
  13
  14 from qapi import *
  15
  16
  17 def gen_fwd_object_or_array(name):
  18     return mcgen('''
  19
  20 typedef struct %(c_name)s %(c_name)s;
  21 ''',
  22                  c_name=c_name(name))
  23
  24
  25 def gen_array(name, element_type):
  26     return mcgen('''
  27
  28 struct %(c_name)s {
  29     union {
  30         %(c_type)s value;
  31         uint64_t padding;
  32     };
  33     %(c_name)s *next;
  34 };
  35 ''',
  36                  c_name=c_name(name), c_type=element_type.c_type())
  37
  38
  39 def gen_struct_field(member):
  40     ret = ''
  41
  42     if member.optional:
  43         ret += mcgen('''
  44     bool has_%(c_name)s;
  45 ''',
  46                      c_name=c_name(member.name))
  47     ret += mcgen('''
  48     %(c_type)s %(c_name)s;
  49 ''',
  50                  c_type=member.type.c_type(), c_name=c_name(member.name))
  51     return ret
  52
  53
  54 def gen_struct_fields(local_members, base=None):
  55     ret = ''
  56
  57     if base:
  58         ret += mcgen('''
  59     /* Members inherited from %(c_name)s: */
  60 ''',
  61                      c_name=base.c_name())
  62         for memb in base.members:
  63             ret += gen_struct_field(memb)
  64         ret += mcgen('''
  65     /* Own members: */
  66 ''')
  67
  68     for memb in local_members:
  69         ret += gen_struct_field(memb)
  70     return ret
  71
  72
  73 def gen_struct(name, base, members):
  74     ret = mcgen('''
  75
  76 struct %(c_name)s {
  77 ''',
  78                 c_name=c_name(name))
  79
  80     ret += gen_struct_fields(members, base)
  81
  82     # Make sure that all structs have at least one field; this avoids
  83     # potential issues with attempting to malloc space for zero-length
  84     # structs in C, and also incompatibility with C++ (where an empty
  85     # struct is size 1).
  86     if not (base and base.members) and not members:
  87         ret += mcgen('''
  88     char qapi_dummy_field_for_empty_struct;
  89 ''')
  90
  91     ret += mcgen('''
  92 };
  93 ''')
  94
  95     return ret
  96
  97
  98 def gen_upcast(name, base):
  99     # C makes const-correctness ugly.  We have to cast away const to let
 100     # this function work for both const and non-const obj.
 101     return mcgen('''
 102
 103 static inline %(base)s *qapi_%(c_name)s_base(const %(c_name)s *obj)
 104 {
 105     return (%(base)s *)obj;
 106 }
 107 ''',
 108                  c_name=c_name(name), base=base.c_name())
 109
 110
 111 def gen_alternate_qtypes_decl(name):
 112     return mcgen('''
 113
 114 extern const int %(c_name)s_qtypes[];
 115 ''',
 116                  c_name=c_name(name))
 117
 118
 119 def gen_alternate_qtypes(name, variants):
 120     ret = mcgen('''
 121
 122 const int %(c_name)s_qtypes[QTYPE_MAX] = {
 123 ''',
 124                 c_name=c_name(name))
 125
 126     for var in variants.variants:
 127         qtype = var.type.alternate_qtype()
 128         assert qtype
 129
 130         ret += mcgen('''
 131     [%(qtype)s] = %(enum_const)s,
 132 ''',
 133                      qtype=qtype,
 134                      enum_const=c_enum_const(variants.tag_member.type.name,
 135                                              var.name))
 136
 137     ret += mcgen('''
 138 };
 139 ''')
 140     return ret
 141
 142
 143 def gen_union(name, base, variants):
 144     ret = mcgen('''
 145
 146 struct %(c_name)s {
 147 ''',
 148                 c_name=c_name(name))
 149     if base:
 150         ret += gen_struct_fields([], base)
 151     else:
 152         ret += gen_struct_field(variants.tag_member)
 153
 154     # FIXME: What purpose does data serve, besides preventing a union that
 155     # has a branch named 'data'? We use it in qapi-visit.py to decide
 156     # whether to bypass the switch statement if visiting the discriminator
 157     # failed; but since we 0-initialize structs, and cannot tell what
 158     # branch of the union is in use if the discriminator is invalid, there
 159     # should not be any data leaks even without a data pointer.  Or, if
 160     # 'data' is merely added to guarantee we don't have an empty union,
 161     # shouldn't we enforce that at .json parse time?
 162     ret += mcgen('''
 163     union { /* union tag is @%(c_name)s */
 164         void *data;
 165 ''',
 166                  c_name=c_name(variants.tag_member.name))
 167
 168     for var in variants.variants:
 169         # Ugly special case for simple union TODO get rid of it
 170         typ = var.simple_union_type() or var.type
 171         ret += mcgen('''
 172         %(c_type)s %(c_name)s;
 173 ''',
 174                      c_type=typ.c_type(),
 175                      c_name=c_name(var.name))
 176
 177     ret += mcgen('''
 178     } u;
 179 };
 180 ''')
 181
 182     return ret
 183
 184
 185 def gen_type_cleanup_decl(name):
 186     ret = mcgen('''
 187
 188 void qapi_free_%(c_name)s(%(c_name)s *obj);
 189 ''',
 190                 c_name=c_name(name))
 191     return ret
 192
 193
 194 def gen_type_cleanup(name):
 195     ret = mcgen('''
 196
 197 void qapi_free_%(c_name)s(%(c_name)s *obj)
 198 {
 199     QapiDeallocVisitor *qdv;
 200     Visitor *v;
 201
 202     if (!obj) {
 203         return;
 204     }
 205
 206     qdv = qapi_dealloc_visitor_new();
 207     v = qapi_dealloc_get_visitor(qdv);
 208     visit_type_%(c_name)s(v, &obj, NULL, NULL);
 209     qapi_dealloc_visitor_cleanup(qdv);
 210 }
 211 ''',
 212                 c_name=c_name(name))
 213     return ret
 214
 215
 216 class QAPISchemaGenTypeVisitor(QAPISchemaVisitor):
 217     def __init__(self):
 218         self.decl = None
 219         self.defn = None
 220         self._fwdecl = None
 221         self._fwdefn = None
 222         self._btin = None
 223
 224     def visit_begin(self, schema):
 225         self.decl = ''
 226         self.defn = ''
 227         self._fwdecl = ''
 228         self._fwdefn = ''
 229         self._btin = guardstart('QAPI_TYPES_BUILTIN')
 230
 231     def visit_end(self):
 232         self.decl = self._fwdecl + self.decl
 233         self._fwdecl = None
 234         self.defn = self._fwdefn + self.defn
 235         self._fwdefn = None
 236         # To avoid header dependency hell, we always generate
 237         # declarations for built-in types in our header files and
 238         # simply guard them.  See also do_builtins (command line
 239         # option -b).
 240         self._btin += guardend('QAPI_TYPES_BUILTIN')
 241         self.decl = self._btin + self.decl
 242         self._btin = None
 243
 244     def visit_needed(self, entity):
 245         # Visit everything except implicit objects
 246         return not (entity.is_implicit() and
 247                     isinstance(entity, QAPISchemaObjectType))
 248
 249     def _gen_type_cleanup(self, name):
 250         self.decl += gen_type_cleanup_decl(name)
 251         self.defn += gen_type_cleanup(name)
 252
 253     def visit_enum_type(self, name, info, values, prefix):
 254         self._fwdecl += gen_enum(name, values, prefix)
 255         self._fwdefn += gen_enum_lookup(name, values, prefix)
 256
 257     def visit_array_type(self, name, info, element_type):
 258         if isinstance(element_type, QAPISchemaBuiltinType):
 259             self._btin += gen_fwd_object_or_array(name)
 260             self._btin += gen_array(name, element_type)
 261             self._btin += gen_type_cleanup_decl(name)
 262             if do_builtins:
 263                 self.defn += gen_type_cleanup(name)
 264         else:
 265             self._fwdecl += gen_fwd_object_or_array(name)
 266             self.decl += gen_array(name, element_type)
 267             self._gen_type_cleanup(name)
 268
 269     def visit_object_type(self, name, info, base, members, variants):
 270         self._fwdecl += gen_fwd_object_or_array(name)
 271         if variants:
 272             assert not members      # not implemented
 273             self.decl += gen_union(name, base, variants)
 274         else:
 275             self.decl += gen_struct(name, base, members)
 276         if base:
 277             self.decl += gen_upcast(name, base)
 278         self._gen_type_cleanup(name)
 279
 280     def visit_alternate_type(self, name, info, variants):
 281         self._fwdecl += gen_fwd_object_or_array(name)
 282         self._fwdefn += gen_alternate_qtypes(name, variants)
 283         self.decl += gen_union(name, None, variants)
 284         self.decl += gen_alternate_qtypes_decl(name)
 285         self._gen_type_cleanup(name)
 286
 287 # If you link code generated from multiple schemata, you want only one
 288 # instance of the code for built-in types.  Generate it only when
 289 # do_builtins, enabled by command line option -b.  See also
 290 # QAPISchemaGenTypeVisitor.visit_end().
 291 do_builtins = False
 292
 293 (input_file, output_dir, do_c, do_h, prefix, opts) = \
 294     parse_command_line("b", ["builtins"])
 295
 296 for o, a in opts:
 297     if o in ("-b", "--builtins"):
 298         do_builtins = True
 299
 300 c_comment = '''
 301 /*
 302  * deallocation functions for schema-defined QAPI types
 303  *
 304  * Copyright IBM, Corp. 2011
 305  *
 306  * Authors:
 307  *  Anthony Liguori   <aliguori@us.ibm.com>
 308  *  Michael Roth      <mdroth@linux.vnet.ibm.com>
 309  *
 310  * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
 311  * See the COPYING.LIB file in the top-level directory.
 312  *
 313  */
 314 '''
 315 h_comment = '''
 316 /*
 317  * schema-defined QAPI types
 318  *
 319  * Copyright IBM, Corp. 2011
 320  *
 321  * Authors:
 322  *  Anthony Liguori   <aliguori@us.ibm.com>
 323  *
 324  * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
 325  * See the COPYING.LIB file in the top-level directory.
 326  *
 327  */
 328 '''
 329
 330 (fdef, fdecl) = open_output(output_dir, do_c, do_h, prefix,
 331                             'qapi-types.c', 'qapi-types.h',
 332                             c_comment, h_comment)
 333
 334 fdef.write(mcgen('''
 335 #include "qapi/dealloc-visitor.h"
 336 #include "%(prefix)sqapi-types.h"
 337 #include "%(prefix)sqapi-visit.h"
 338 ''',
 339                  prefix=prefix))
 340
 341 fdecl.write(mcgen('''
 342 #include <stdbool.h>
 343 #include <stdint.h>
 344 #include "qapi/qmp/qobject.h"
 345 '''))
 346
 347 schema = QAPISchema(input_file)
 348 gen = QAPISchemaGenTypeVisitor()
 349 schema.visit(gen)
 350 fdef.write(gen.defn)
 351 fdecl.write(gen.decl)
 352
 353 close_output(fdef, fdecl)