schema.py 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086
  1. # -*- coding: utf-8 -*-
  2. #
  3. # QAPI schema internal representation
  4. #
  5. # Copyright (c) 2015-2019 Red Hat Inc.
  6. #
  7. # Authors:
  8. # Markus Armbruster <armbru@redhat.com>
  9. # Eric Blake <eblake@redhat.com>
  10. # Marc-André Lureau <marcandre.lureau@redhat.com>
  11. #
  12. # This work is licensed under the terms of the GNU GPL, version 2.
  13. # See the COPYING file in the top-level directory.
  14. # TODO catching name collisions in generated code would be nice
  15. import os
  16. import re
  17. from collections import OrderedDict
  18. from qapi.common import c_name, pointer_suffix
  19. from qapi.error import QAPIError, QAPIParseError, QAPISemError
  20. from qapi.expr import check_exprs
  21. from qapi.parser import QAPISchemaParser
  22. class QAPISchemaEntity(object):
  23. meta = None
  24. def __init__(self, name, info, doc, ifcond=None):
  25. assert name is None or isinstance(name, str)
  26. self.name = name
  27. self._module = None
  28. # For explicitly defined entities, info points to the (explicit)
  29. # definition. For builtins (and their arrays), info is None.
  30. # For implicitly defined entities, info points to a place that
  31. # triggered the implicit definition (there may be more than one
  32. # such place).
  33. self.info = info
  34. self.doc = doc
  35. self._ifcond = ifcond or []
  36. self._checked = False
  37. def c_name(self):
  38. return c_name(self.name)
  39. def check(self, schema):
  40. assert not self._checked
  41. if self.info:
  42. self._module = os.path.relpath(self.info.fname,
  43. os.path.dirname(schema.fname))
  44. self._checked = True
  45. def connect_doc(self, doc=None):
  46. pass
  47. def check_doc(self):
  48. pass
  49. @property
  50. def ifcond(self):
  51. assert self._checked
  52. return self._ifcond
  53. @property
  54. def module(self):
  55. assert self._checked
  56. return self._module
  57. def is_implicit(self):
  58. return not self.info
  59. def visit(self, visitor):
  60. assert self._checked
  61. def describe(self):
  62. assert self.meta
  63. return "%s '%s'" % (self.meta, self.name)
  64. class QAPISchemaVisitor(object):
  65. def visit_begin(self, schema):
  66. pass
  67. def visit_end(self):
  68. pass
  69. def visit_module(self, fname):
  70. pass
  71. def visit_needed(self, entity):
  72. # Default to visiting everything
  73. return True
  74. def visit_include(self, fname, info):
  75. pass
  76. def visit_builtin_type(self, name, info, json_type):
  77. pass
  78. def visit_enum_type(self, name, info, ifcond, members, prefix):
  79. pass
  80. def visit_array_type(self, name, info, ifcond, element_type):
  81. pass
  82. def visit_object_type(self, name, info, ifcond, base, members, variants,
  83. features):
  84. pass
  85. def visit_object_type_flat(self, name, info, ifcond, members, variants,
  86. features):
  87. pass
  88. def visit_alternate_type(self, name, info, ifcond, variants):
  89. pass
  90. def visit_command(self, name, info, ifcond, arg_type, ret_type, gen,
  91. success_response, boxed, allow_oob, allow_preconfig,
  92. features):
  93. pass
  94. def visit_event(self, name, info, ifcond, arg_type, boxed):
  95. pass
  96. class QAPISchemaInclude(QAPISchemaEntity):
  97. def __init__(self, fname, info):
  98. QAPISchemaEntity.__init__(self, None, info, None)
  99. self.fname = fname
  100. def visit(self, visitor):
  101. QAPISchemaEntity.visit(self, visitor)
  102. visitor.visit_include(self.fname, self.info)
  103. class QAPISchemaType(QAPISchemaEntity):
  104. # Return the C type for common use.
  105. # For the types we commonly box, this is a pointer type.
  106. def c_type(self):
  107. pass
  108. # Return the C type to be used in a parameter list.
  109. def c_param_type(self):
  110. return self.c_type()
  111. # Return the C type to be used where we suppress boxing.
  112. def c_unboxed_type(self):
  113. return self.c_type()
  114. def json_type(self):
  115. pass
  116. def alternate_qtype(self):
  117. json2qtype = {
  118. 'null': 'QTYPE_QNULL',
  119. 'string': 'QTYPE_QSTRING',
  120. 'number': 'QTYPE_QNUM',
  121. 'int': 'QTYPE_QNUM',
  122. 'boolean': 'QTYPE_QBOOL',
  123. 'object': 'QTYPE_QDICT'
  124. }
  125. return json2qtype.get(self.json_type())
  126. def doc_type(self):
  127. if self.is_implicit():
  128. return None
  129. return self.name
  130. def describe(self):
  131. assert self.meta
  132. return "%s type '%s'" % (self.meta, self.name)
  133. class QAPISchemaBuiltinType(QAPISchemaType):
  134. meta = 'built-in'
  135. def __init__(self, name, json_type, c_type):
  136. QAPISchemaType.__init__(self, name, None, None)
  137. assert not c_type or isinstance(c_type, str)
  138. assert json_type in ('string', 'number', 'int', 'boolean', 'null',
  139. 'value')
  140. self._json_type_name = json_type
  141. self._c_type_name = c_type
  142. def c_name(self):
  143. return self.name
  144. def c_type(self):
  145. return self._c_type_name
  146. def c_param_type(self):
  147. if self.name == 'str':
  148. return 'const ' + self._c_type_name
  149. return self._c_type_name
  150. def json_type(self):
  151. return self._json_type_name
  152. def doc_type(self):
  153. return self.json_type()
  154. def visit(self, visitor):
  155. QAPISchemaType.visit(self, visitor)
  156. visitor.visit_builtin_type(self.name, self.info, self.json_type())
  157. class QAPISchemaEnumType(QAPISchemaType):
  158. meta = 'enum'
  159. def __init__(self, name, info, doc, ifcond, members, prefix):
  160. QAPISchemaType.__init__(self, name, info, doc, ifcond)
  161. for m in members:
  162. assert isinstance(m, QAPISchemaEnumMember)
  163. m.set_defined_in(name)
  164. assert prefix is None or isinstance(prefix, str)
  165. self.members = members
  166. self.prefix = prefix
  167. def check(self, schema):
  168. QAPISchemaType.check(self, schema)
  169. seen = {}
  170. for m in self.members:
  171. m.check_clash(self.info, seen)
  172. def connect_doc(self, doc=None):
  173. doc = doc or self.doc
  174. if doc:
  175. for m in self.members:
  176. doc.connect_member(m)
  177. def check_doc(self):
  178. if self.doc:
  179. self.doc.check()
  180. def is_implicit(self):
  181. # See QAPISchema._make_implicit_enum_type() and ._def_predefineds()
  182. return self.name.endswith('Kind') or self.name == 'QType'
  183. def c_type(self):
  184. return c_name(self.name)
  185. def member_names(self):
  186. return [m.name for m in self.members]
  187. def json_type(self):
  188. return 'string'
  189. def visit(self, visitor):
  190. QAPISchemaType.visit(self, visitor)
  191. visitor.visit_enum_type(self.name, self.info, self.ifcond,
  192. self.members, self.prefix)
  193. class QAPISchemaArrayType(QAPISchemaType):
  194. meta = 'array'
  195. def __init__(self, name, info, element_type):
  196. QAPISchemaType.__init__(self, name, info, None, None)
  197. assert isinstance(element_type, str)
  198. self._element_type_name = element_type
  199. self.element_type = None
  200. def check(self, schema):
  201. QAPISchemaType.check(self, schema)
  202. self.element_type = schema.resolve_type(
  203. self._element_type_name, self.info,
  204. self.info and self.info.defn_meta)
  205. assert not isinstance(self.element_type, QAPISchemaArrayType)
  206. @property
  207. def ifcond(self):
  208. assert self._checked
  209. return self.element_type.ifcond
  210. @property
  211. def module(self):
  212. assert self._checked
  213. return self.element_type.module
  214. def is_implicit(self):
  215. return True
  216. def c_type(self):
  217. return c_name(self.name) + pointer_suffix
  218. def json_type(self):
  219. return 'array'
  220. def doc_type(self):
  221. elt_doc_type = self.element_type.doc_type()
  222. if not elt_doc_type:
  223. return None
  224. return 'array of ' + elt_doc_type
  225. def visit(self, visitor):
  226. QAPISchemaType.visit(self, visitor)
  227. visitor.visit_array_type(self.name, self.info, self.ifcond,
  228. self.element_type)
  229. def describe(self):
  230. assert self.meta
  231. return "%s type ['%s']" % (self.meta, self._element_type_name)
  232. class QAPISchemaObjectType(QAPISchemaType):
  233. def __init__(self, name, info, doc, ifcond,
  234. base, local_members, variants, features):
  235. # struct has local_members, optional base, and no variants
  236. # flat union has base, variants, and no local_members
  237. # simple union has local_members, variants, and no base
  238. QAPISchemaType.__init__(self, name, info, doc, ifcond)
  239. self.meta = 'union' if variants else 'struct'
  240. assert base is None or isinstance(base, str)
  241. for m in local_members:
  242. assert isinstance(m, QAPISchemaObjectTypeMember)
  243. m.set_defined_in(name)
  244. if variants is not None:
  245. assert isinstance(variants, QAPISchemaObjectTypeVariants)
  246. variants.set_defined_in(name)
  247. for f in features:
  248. assert isinstance(f, QAPISchemaFeature)
  249. f.set_defined_in(name)
  250. self._base_name = base
  251. self.base = None
  252. self.local_members = local_members
  253. self.variants = variants
  254. self.members = None
  255. self.features = features
  256. def check(self, schema):
  257. # This calls another type T's .check() exactly when the C
  258. # struct emitted by gen_object() contains that T's C struct
  259. # (pointers don't count).
  260. if self.members is not None:
  261. # A previous .check() completed: nothing to do
  262. return
  263. if self._checked:
  264. # Recursed: C struct contains itself
  265. raise QAPISemError(self.info,
  266. "object %s contains itself" % self.name)
  267. QAPISchemaType.check(self, schema)
  268. assert self._checked and self.members is None
  269. seen = OrderedDict()
  270. if self._base_name:
  271. self.base = schema.resolve_type(self._base_name, self.info,
  272. "'base'")
  273. if (not isinstance(self.base, QAPISchemaObjectType)
  274. or self.base.variants):
  275. raise QAPISemError(
  276. self.info,
  277. "'base' requires a struct type, %s isn't"
  278. % self.base.describe())
  279. self.base.check(schema)
  280. self.base.check_clash(self.info, seen)
  281. for m in self.local_members:
  282. m.check(schema)
  283. m.check_clash(self.info, seen)
  284. members = seen.values()
  285. if self.variants:
  286. self.variants.check(schema, seen)
  287. self.variants.check_clash(self.info, seen)
  288. # Features are in a name space separate from members
  289. seen = {}
  290. for f in self.features:
  291. f.check_clash(self.info, seen)
  292. self.members = members # mark completed
  293. # Check that the members of this type do not cause duplicate JSON members,
  294. # and update seen to track the members seen so far. Report any errors
  295. # on behalf of info, which is not necessarily self.info
  296. def check_clash(self, info, seen):
  297. assert self._checked
  298. assert not self.variants # not implemented
  299. for m in self.members:
  300. m.check_clash(info, seen)
  301. def connect_doc(self, doc=None):
  302. doc = doc or self.doc
  303. if doc:
  304. if self.base and self.base.is_implicit():
  305. self.base.connect_doc(doc)
  306. for m in self.local_members:
  307. doc.connect_member(m)
  308. def check_doc(self):
  309. if self.doc:
  310. self.doc.check()
  311. @property
  312. def ifcond(self):
  313. assert self._checked
  314. if isinstance(self._ifcond, QAPISchemaType):
  315. # Simple union wrapper type inherits from wrapped type;
  316. # see _make_implicit_object_type()
  317. return self._ifcond.ifcond
  318. return self._ifcond
  319. def is_implicit(self):
  320. # See QAPISchema._make_implicit_object_type(), as well as
  321. # _def_predefineds()
  322. return self.name.startswith('q_')
  323. def is_empty(self):
  324. assert self.members is not None
  325. return not self.members and not self.variants
  326. def c_name(self):
  327. assert self.name != 'q_empty'
  328. return QAPISchemaType.c_name(self)
  329. def c_type(self):
  330. assert not self.is_implicit()
  331. return c_name(self.name) + pointer_suffix
  332. def c_unboxed_type(self):
  333. return c_name(self.name)
  334. def json_type(self):
  335. return 'object'
  336. def visit(self, visitor):
  337. QAPISchemaType.visit(self, visitor)
  338. visitor.visit_object_type(self.name, self.info, self.ifcond,
  339. self.base, self.local_members, self.variants,
  340. self.features)
  341. visitor.visit_object_type_flat(self.name, self.info, self.ifcond,
  342. self.members, self.variants,
  343. self.features)
  344. class QAPISchemaMember(object):
  345. """ Represents object members, enum members and features """
  346. role = 'member'
  347. def __init__(self, name, info, ifcond=None):
  348. assert isinstance(name, str)
  349. self.name = name
  350. self.info = info
  351. self.ifcond = ifcond or []
  352. self.defined_in = None
  353. def set_defined_in(self, name):
  354. assert not self.defined_in
  355. self.defined_in = name
  356. def check_clash(self, info, seen):
  357. cname = c_name(self.name)
  358. if cname in seen:
  359. raise QAPISemError(
  360. info,
  361. "%s collides with %s"
  362. % (self.describe(info), seen[cname].describe(info)))
  363. seen[cname] = self
  364. def describe(self, info):
  365. role = self.role
  366. defined_in = self.defined_in
  367. assert defined_in
  368. if defined_in.startswith('q_obj_'):
  369. # See QAPISchema._make_implicit_object_type() - reverse the
  370. # mapping there to create a nice human-readable description
  371. defined_in = defined_in[6:]
  372. if defined_in.endswith('-arg'):
  373. # Implicit type created for a command's dict 'data'
  374. assert role == 'member'
  375. role = 'parameter'
  376. elif defined_in.endswith('-base'):
  377. # Implicit type created for a flat union's dict 'base'
  378. role = 'base ' + role
  379. else:
  380. # Implicit type created for a simple union's branch
  381. assert defined_in.endswith('-wrapper')
  382. # Unreachable and not implemented
  383. assert False
  384. elif defined_in.endswith('Kind'):
  385. # See QAPISchema._make_implicit_enum_type()
  386. # Implicit enum created for simple union's branches
  387. assert role == 'value'
  388. role = 'branch'
  389. elif defined_in != info.defn_name:
  390. return "%s '%s' of type '%s'" % (role, self.name, defined_in)
  391. return "%s '%s'" % (role, self.name)
  392. class QAPISchemaEnumMember(QAPISchemaMember):
  393. role = 'value'
  394. class QAPISchemaFeature(QAPISchemaMember):
  395. role = 'feature'
  396. class QAPISchemaObjectTypeMember(QAPISchemaMember):
  397. def __init__(self, name, info, typ, optional, ifcond=None):
  398. QAPISchemaMember.__init__(self, name, info, ifcond)
  399. assert isinstance(typ, str)
  400. assert isinstance(optional, bool)
  401. self._type_name = typ
  402. self.type = None
  403. self.optional = optional
  404. def check(self, schema):
  405. assert self.defined_in
  406. self.type = schema.resolve_type(self._type_name, self.info,
  407. self.describe)
  408. class QAPISchemaObjectTypeVariants(object):
  409. def __init__(self, tag_name, info, tag_member, variants):
  410. # Flat unions pass tag_name but not tag_member.
  411. # Simple unions and alternates pass tag_member but not tag_name.
  412. # After check(), tag_member is always set, and tag_name remains
  413. # a reliable witness of being used by a flat union.
  414. assert bool(tag_member) != bool(tag_name)
  415. assert (isinstance(tag_name, str) or
  416. isinstance(tag_member, QAPISchemaObjectTypeMember))
  417. for v in variants:
  418. assert isinstance(v, QAPISchemaObjectTypeVariant)
  419. self._tag_name = tag_name
  420. self.info = info
  421. self.tag_member = tag_member
  422. self.variants = variants
  423. def set_defined_in(self, name):
  424. for v in self.variants:
  425. v.set_defined_in(name)
  426. def check(self, schema, seen):
  427. if not self.tag_member: # flat union
  428. self.tag_member = seen.get(c_name(self._tag_name))
  429. base = "'base'"
  430. # Pointing to the base type when not implicit would be
  431. # nice, but we don't know it here
  432. if not self.tag_member or self._tag_name != self.tag_member.name:
  433. raise QAPISemError(
  434. self.info,
  435. "discriminator '%s' is not a member of %s"
  436. % (self._tag_name, base))
  437. # Here we do:
  438. base_type = schema.lookup_type(self.tag_member.defined_in)
  439. assert base_type
  440. if not base_type.is_implicit():
  441. base = "base type '%s'" % self.tag_member.defined_in
  442. if not isinstance(self.tag_member.type, QAPISchemaEnumType):
  443. raise QAPISemError(
  444. self.info,
  445. "discriminator member '%s' of %s must be of enum type"
  446. % (self._tag_name, base))
  447. if self.tag_member.optional:
  448. raise QAPISemError(
  449. self.info,
  450. "discriminator member '%s' of %s must not be optional"
  451. % (self._tag_name, base))
  452. if self.tag_member.ifcond:
  453. raise QAPISemError(
  454. self.info,
  455. "discriminator member '%s' of %s must not be conditional"
  456. % (self._tag_name, base))
  457. else: # simple union
  458. assert isinstance(self.tag_member.type, QAPISchemaEnumType)
  459. assert not self.tag_member.optional
  460. assert self.tag_member.ifcond == []
  461. if self._tag_name: # flat union
  462. # branches that are not explicitly covered get an empty type
  463. cases = set([v.name for v in self.variants])
  464. for m in self.tag_member.type.members:
  465. if m.name not in cases:
  466. v = QAPISchemaObjectTypeVariant(m.name, self.info,
  467. 'q_empty', m.ifcond)
  468. v.set_defined_in(self.tag_member.defined_in)
  469. self.variants.append(v)
  470. if not self.variants:
  471. raise QAPISemError(self.info, "union has no branches")
  472. for v in self.variants:
  473. v.check(schema)
  474. # Union names must match enum values; alternate names are
  475. # checked separately. Use 'seen' to tell the two apart.
  476. if seen:
  477. if v.name not in self.tag_member.type.member_names():
  478. raise QAPISemError(
  479. self.info,
  480. "branch '%s' is not a value of %s"
  481. % (v.name, self.tag_member.type.describe()))
  482. if (not isinstance(v.type, QAPISchemaObjectType)
  483. or v.type.variants):
  484. raise QAPISemError(
  485. self.info,
  486. "%s cannot use %s"
  487. % (v.describe(self.info), v.type.describe()))
  488. v.type.check(schema)
  489. def check_clash(self, info, seen):
  490. for v in self.variants:
  491. # Reset seen map for each variant, since qapi names from one
  492. # branch do not affect another branch
  493. v.type.check_clash(info, dict(seen))
  494. class QAPISchemaObjectTypeVariant(QAPISchemaObjectTypeMember):
  495. role = 'branch'
  496. def __init__(self, name, info, typ, ifcond=None):
  497. QAPISchemaObjectTypeMember.__init__(self, name, info, typ,
  498. False, ifcond)
  499. class QAPISchemaAlternateType(QAPISchemaType):
  500. meta = 'alternate'
  501. def __init__(self, name, info, doc, ifcond, variants):
  502. QAPISchemaType.__init__(self, name, info, doc, ifcond)
  503. assert isinstance(variants, QAPISchemaObjectTypeVariants)
  504. assert variants.tag_member
  505. variants.set_defined_in(name)
  506. variants.tag_member.set_defined_in(self.name)
  507. self.variants = variants
  508. def check(self, schema):
  509. QAPISchemaType.check(self, schema)
  510. self.variants.tag_member.check(schema)
  511. # Not calling self.variants.check_clash(), because there's nothing
  512. # to clash with
  513. self.variants.check(schema, {})
  514. # Alternate branch names have no relation to the tag enum values;
  515. # so we have to check for potential name collisions ourselves.
  516. seen = {}
  517. types_seen = {}
  518. for v in self.variants.variants:
  519. v.check_clash(self.info, seen)
  520. qtype = v.type.alternate_qtype()
  521. if not qtype:
  522. raise QAPISemError(
  523. self.info,
  524. "%s cannot use %s"
  525. % (v.describe(self.info), v.type.describe()))
  526. conflicting = set([qtype])
  527. if qtype == 'QTYPE_QSTRING':
  528. if isinstance(v.type, QAPISchemaEnumType):
  529. for m in v.type.members:
  530. if m.name in ['on', 'off']:
  531. conflicting.add('QTYPE_QBOOL')
  532. if re.match(r'[-+0-9.]', m.name):
  533. # lazy, could be tightened
  534. conflicting.add('QTYPE_QNUM')
  535. else:
  536. conflicting.add('QTYPE_QNUM')
  537. conflicting.add('QTYPE_QBOOL')
  538. for qt in conflicting:
  539. if qt in types_seen:
  540. raise QAPISemError(
  541. self.info,
  542. "%s can't be distinguished from '%s'"
  543. % (v.describe(self.info), types_seen[qt]))
  544. types_seen[qt] = v.name
  545. def connect_doc(self, doc=None):
  546. doc = doc or self.doc
  547. if doc:
  548. for v in self.variants.variants:
  549. doc.connect_member(v)
  550. def check_doc(self):
  551. if self.doc:
  552. self.doc.check()
  553. def c_type(self):
  554. return c_name(self.name) + pointer_suffix
  555. def json_type(self):
  556. return 'value'
  557. def visit(self, visitor):
  558. QAPISchemaType.visit(self, visitor)
  559. visitor.visit_alternate_type(self.name, self.info, self.ifcond,
  560. self.variants)
  561. class QAPISchemaCommand(QAPISchemaEntity):
  562. meta = 'command'
  563. def __init__(self, name, info, doc, ifcond, arg_type, ret_type,
  564. gen, success_response, boxed, allow_oob, allow_preconfig,
  565. features):
  566. QAPISchemaEntity.__init__(self, name, info, doc, ifcond)
  567. assert not arg_type or isinstance(arg_type, str)
  568. assert not ret_type or isinstance(ret_type, str)
  569. for f in features:
  570. assert isinstance(f, QAPISchemaFeature)
  571. f.set_defined_in(name)
  572. self._arg_type_name = arg_type
  573. self.arg_type = None
  574. self._ret_type_name = ret_type
  575. self.ret_type = None
  576. self.gen = gen
  577. self.success_response = success_response
  578. self.boxed = boxed
  579. self.allow_oob = allow_oob
  580. self.allow_preconfig = allow_preconfig
  581. self.features = features
  582. def check(self, schema):
  583. QAPISchemaEntity.check(self, schema)
  584. if self._arg_type_name:
  585. self.arg_type = schema.resolve_type(
  586. self._arg_type_name, self.info, "command's 'data'")
  587. if not isinstance(self.arg_type, QAPISchemaObjectType):
  588. raise QAPISemError(
  589. self.info,
  590. "command's 'data' cannot take %s"
  591. % self.arg_type.describe())
  592. if self.arg_type.variants and not self.boxed:
  593. raise QAPISemError(
  594. self.info,
  595. "command's 'data' can take %s only with 'boxed': true"
  596. % self.arg_type.describe())
  597. if self._ret_type_name:
  598. self.ret_type = schema.resolve_type(
  599. self._ret_type_name, self.info, "command's 'returns'")
  600. if self.name not in self.info.pragma.returns_whitelist:
  601. if not (isinstance(self.ret_type, QAPISchemaObjectType)
  602. or (isinstance(self.ret_type, QAPISchemaArrayType)
  603. and isinstance(self.ret_type.element_type,
  604. QAPISchemaObjectType))):
  605. raise QAPISemError(
  606. self.info,
  607. "command's 'returns' cannot take %s"
  608. % self.ret_type.describe())
  609. # Features are in a name space separate from members
  610. seen = {}
  611. for f in self.features:
  612. f.check_clash(self.info, seen)
  613. def visit(self, visitor):
  614. QAPISchemaEntity.visit(self, visitor)
  615. visitor.visit_command(self.name, self.info, self.ifcond,
  616. self.arg_type, self.ret_type,
  617. self.gen, self.success_response,
  618. self.boxed, self.allow_oob,
  619. self.allow_preconfig,
  620. self.features)
  621. class QAPISchemaEvent(QAPISchemaEntity):
  622. meta = 'event'
  623. def __init__(self, name, info, doc, ifcond, arg_type, boxed):
  624. QAPISchemaEntity.__init__(self, name, info, doc, ifcond)
  625. assert not arg_type or isinstance(arg_type, str)
  626. self._arg_type_name = arg_type
  627. self.arg_type = None
  628. self.boxed = boxed
  629. def check(self, schema):
  630. QAPISchemaEntity.check(self, schema)
  631. if self._arg_type_name:
  632. self.arg_type = schema.resolve_type(
  633. self._arg_type_name, self.info, "event's 'data'")
  634. if not isinstance(self.arg_type, QAPISchemaObjectType):
  635. raise QAPISemError(
  636. self.info,
  637. "event's 'data' cannot take %s"
  638. % self.arg_type.describe())
  639. if self.arg_type.variants and not self.boxed:
  640. raise QAPISemError(
  641. self.info,
  642. "event's 'data' can take %s only with 'boxed': true"
  643. % self.arg_type.describe())
  644. def visit(self, visitor):
  645. QAPISchemaEntity.visit(self, visitor)
  646. visitor.visit_event(self.name, self.info, self.ifcond,
  647. self.arg_type, self.boxed)
  648. class QAPISchema(object):
  649. def __init__(self, fname):
  650. self.fname = fname
  651. parser = QAPISchemaParser(fname)
  652. exprs = check_exprs(parser.exprs)
  653. self.docs = parser.docs
  654. self._entity_list = []
  655. self._entity_dict = {}
  656. self._predefining = True
  657. self._def_predefineds()
  658. self._predefining = False
  659. self._def_exprs(exprs)
  660. self.check()
  661. def _def_entity(self, ent):
  662. # Only the predefined types are allowed to not have info
  663. assert ent.info or self._predefining
  664. self._entity_list.append(ent)
  665. if ent.name is None:
  666. return
  667. # TODO reject names that differ only in '_' vs. '.' vs. '-',
  668. # because they're liable to clash in generated C.
  669. other_ent = self._entity_dict.get(ent.name)
  670. if other_ent:
  671. if other_ent.info:
  672. where = QAPIError(other_ent.info, None, "previous definition")
  673. raise QAPISemError(
  674. ent.info,
  675. "'%s' is already defined\n%s" % (ent.name, where))
  676. raise QAPISemError(
  677. ent.info, "%s is already defined" % other_ent.describe())
  678. self._entity_dict[ent.name] = ent
  679. def lookup_entity(self, name, typ=None):
  680. ent = self._entity_dict.get(name)
  681. if typ and not isinstance(ent, typ):
  682. return None
  683. return ent
  684. def lookup_type(self, name):
  685. return self.lookup_entity(name, QAPISchemaType)
  686. def resolve_type(self, name, info, what):
  687. typ = self.lookup_type(name)
  688. if not typ:
  689. if callable(what):
  690. what = what(info)
  691. raise QAPISemError(
  692. info, "%s uses unknown type '%s'" % (what, name))
  693. return typ
  694. def _def_include(self, expr, info, doc):
  695. include = expr['include']
  696. assert doc is None
  697. main_info = info
  698. while main_info.parent:
  699. main_info = main_info.parent
  700. fname = os.path.relpath(include, os.path.dirname(main_info.fname))
  701. self._def_entity(QAPISchemaInclude(fname, info))
  702. def _def_builtin_type(self, name, json_type, c_type):
  703. self._def_entity(QAPISchemaBuiltinType(name, json_type, c_type))
  704. # Instantiating only the arrays that are actually used would
  705. # be nice, but we can't as long as their generated code
  706. # (qapi-builtin-types.[ch]) may be shared by some other
  707. # schema.
  708. self._make_array_type(name, None)
  709. def _def_predefineds(self):
  710. for t in [('str', 'string', 'char' + pointer_suffix),
  711. ('number', 'number', 'double'),
  712. ('int', 'int', 'int64_t'),
  713. ('int8', 'int', 'int8_t'),
  714. ('int16', 'int', 'int16_t'),
  715. ('int32', 'int', 'int32_t'),
  716. ('int64', 'int', 'int64_t'),
  717. ('uint8', 'int', 'uint8_t'),
  718. ('uint16', 'int', 'uint16_t'),
  719. ('uint32', 'int', 'uint32_t'),
  720. ('uint64', 'int', 'uint64_t'),
  721. ('size', 'int', 'uint64_t'),
  722. ('bool', 'boolean', 'bool'),
  723. ('any', 'value', 'QObject' + pointer_suffix),
  724. ('null', 'null', 'QNull' + pointer_suffix)]:
  725. self._def_builtin_type(*t)
  726. self.the_empty_object_type = QAPISchemaObjectType(
  727. 'q_empty', None, None, None, None, [], None, [])
  728. self._def_entity(self.the_empty_object_type)
  729. qtypes = ['none', 'qnull', 'qnum', 'qstring', 'qdict', 'qlist',
  730. 'qbool']
  731. qtype_values = self._make_enum_members(
  732. [{'name': n} for n in qtypes], None)
  733. self._def_entity(QAPISchemaEnumType('QType', None, None, None,
  734. qtype_values, 'QTYPE'))
  735. def _make_features(self, features, info):
  736. return [QAPISchemaFeature(f['name'], info, f.get('if'))
  737. for f in features]
  738. def _make_enum_members(self, values, info):
  739. return [QAPISchemaEnumMember(v['name'], info, v.get('if'))
  740. for v in values]
  741. def _make_implicit_enum_type(self, name, info, ifcond, values):
  742. # See also QAPISchemaObjectTypeMember.describe()
  743. name = name + 'Kind' # reserved by check_defn_name_str()
  744. self._def_entity(QAPISchemaEnumType(
  745. name, info, None, ifcond, self._make_enum_members(values, info),
  746. None))
  747. return name
  748. def _make_array_type(self, element_type, info):
  749. name = element_type + 'List' # reserved by check_defn_name_str()
  750. if not self.lookup_type(name):
  751. self._def_entity(QAPISchemaArrayType(name, info, element_type))
  752. return name
  753. def _make_implicit_object_type(self, name, info, doc, ifcond,
  754. role, members):
  755. if not members:
  756. return None
  757. # See also QAPISchemaObjectTypeMember.describe()
  758. name = 'q_obj_%s-%s' % (name, role)
  759. typ = self.lookup_entity(name, QAPISchemaObjectType)
  760. if typ:
  761. # The implicit object type has multiple users. This can
  762. # happen only for simple unions' implicit wrapper types.
  763. # Its ifcond should be the disjunction of its user's
  764. # ifconds. Not implemented. Instead, we always pass the
  765. # wrapped type's ifcond, which is trivially the same for all
  766. # users. It's also necessary for the wrapper to compile.
  767. # But it's not tight: the disjunction need not imply it. We
  768. # may end up compiling useless wrapper types.
  769. # TODO kill simple unions or implement the disjunction
  770. assert (ifcond or []) == typ._ifcond # pylint: disable=protected-access
  771. else:
  772. self._def_entity(QAPISchemaObjectType(name, info, doc, ifcond,
  773. None, members, None, []))
  774. return name
  775. def _def_enum_type(self, expr, info, doc):
  776. name = expr['enum']
  777. data = expr['data']
  778. prefix = expr.get('prefix')
  779. ifcond = expr.get('if')
  780. self._def_entity(QAPISchemaEnumType(
  781. name, info, doc, ifcond,
  782. self._make_enum_members(data, info), prefix))
  783. def _make_member(self, name, typ, ifcond, info):
  784. optional = False
  785. if name.startswith('*'):
  786. name = name[1:]
  787. optional = True
  788. if isinstance(typ, list):
  789. assert len(typ) == 1
  790. typ = self._make_array_type(typ[0], info)
  791. return QAPISchemaObjectTypeMember(name, info, typ, optional, ifcond)
  792. def _make_members(self, data, info):
  793. return [self._make_member(key, value['type'], value.get('if'), info)
  794. for (key, value) in data.items()]
  795. def _def_struct_type(self, expr, info, doc):
  796. name = expr['struct']
  797. base = expr.get('base')
  798. data = expr['data']
  799. ifcond = expr.get('if')
  800. features = expr.get('features', [])
  801. self._def_entity(QAPISchemaObjectType(
  802. name, info, doc, ifcond, base,
  803. self._make_members(data, info),
  804. None,
  805. self._make_features(features, info)))
  806. def _make_variant(self, case, typ, ifcond, info):
  807. return QAPISchemaObjectTypeVariant(case, info, typ, ifcond)
  808. def _make_simple_variant(self, case, typ, ifcond, info):
  809. if isinstance(typ, list):
  810. assert len(typ) == 1
  811. typ = self._make_array_type(typ[0], info)
  812. typ = self._make_implicit_object_type(
  813. typ, info, None, self.lookup_type(typ),
  814. 'wrapper', [self._make_member('data', typ, None, info)])
  815. return QAPISchemaObjectTypeVariant(case, info, typ, ifcond)
  816. def _def_union_type(self, expr, info, doc):
  817. name = expr['union']
  818. data = expr['data']
  819. base = expr.get('base')
  820. ifcond = expr.get('if')
  821. tag_name = expr.get('discriminator')
  822. tag_member = None
  823. if isinstance(base, dict):
  824. base = self._make_implicit_object_type(
  825. name, info, None, ifcond,
  826. 'base', self._make_members(base, info))
  827. if tag_name:
  828. variants = [self._make_variant(key, value['type'],
  829. value.get('if'), info)
  830. for (key, value) in data.items()]
  831. members = []
  832. else:
  833. variants = [self._make_simple_variant(key, value['type'],
  834. value.get('if'), info)
  835. for (key, value) in data.items()]
  836. enum = [{'name': v.name, 'if': v.ifcond} for v in variants]
  837. typ = self._make_implicit_enum_type(name, info, ifcond, enum)
  838. tag_member = QAPISchemaObjectTypeMember('type', info, typ, False)
  839. members = [tag_member]
  840. self._def_entity(
  841. QAPISchemaObjectType(name, info, doc, ifcond, base, members,
  842. QAPISchemaObjectTypeVariants(
  843. tag_name, info, tag_member, variants),
  844. []))
  845. def _def_alternate_type(self, expr, info, doc):
  846. name = expr['alternate']
  847. data = expr['data']
  848. ifcond = expr.get('if')
  849. variants = [self._make_variant(key, value['type'], value.get('if'),
  850. info)
  851. for (key, value) in data.items()]
  852. tag_member = QAPISchemaObjectTypeMember('type', info, 'QType', False)
  853. self._def_entity(
  854. QAPISchemaAlternateType(name, info, doc, ifcond,
  855. QAPISchemaObjectTypeVariants(
  856. None, info, tag_member, variants)))
  857. def _def_command(self, expr, info, doc):
  858. name = expr['command']
  859. data = expr.get('data')
  860. rets = expr.get('returns')
  861. gen = expr.get('gen', True)
  862. success_response = expr.get('success-response', True)
  863. boxed = expr.get('boxed', False)
  864. allow_oob = expr.get('allow-oob', False)
  865. allow_preconfig = expr.get('allow-preconfig', False)
  866. ifcond = expr.get('if')
  867. features = expr.get('features', [])
  868. if isinstance(data, OrderedDict):
  869. data = self._make_implicit_object_type(
  870. name, info, doc, ifcond, 'arg', self._make_members(data, info))
  871. if isinstance(rets, list):
  872. assert len(rets) == 1
  873. rets = self._make_array_type(rets[0], info)
  874. self._def_entity(QAPISchemaCommand(name, info, doc, ifcond, data, rets,
  875. gen, success_response,
  876. boxed, allow_oob, allow_preconfig,
  877. self._make_features(features, info)))
  878. def _def_event(self, expr, info, doc):
  879. name = expr['event']
  880. data = expr.get('data')
  881. boxed = expr.get('boxed', False)
  882. ifcond = expr.get('if')
  883. if isinstance(data, OrderedDict):
  884. data = self._make_implicit_object_type(
  885. name, info, doc, ifcond, 'arg', self._make_members(data, info))
  886. self._def_entity(QAPISchemaEvent(name, info, doc, ifcond, data, boxed))
  887. def _def_exprs(self, exprs):
  888. for expr_elem in exprs:
  889. expr = expr_elem['expr']
  890. info = expr_elem['info']
  891. doc = expr_elem.get('doc')
  892. if 'enum' in expr:
  893. self._def_enum_type(expr, info, doc)
  894. elif 'struct' in expr:
  895. self._def_struct_type(expr, info, doc)
  896. elif 'union' in expr:
  897. self._def_union_type(expr, info, doc)
  898. elif 'alternate' in expr:
  899. self._def_alternate_type(expr, info, doc)
  900. elif 'command' in expr:
  901. self._def_command(expr, info, doc)
  902. elif 'event' in expr:
  903. self._def_event(expr, info, doc)
  904. elif 'include' in expr:
  905. self._def_include(expr, info, doc)
  906. else:
  907. assert False
  908. def check(self):
  909. for ent in self._entity_list:
  910. ent.check(self)
  911. ent.connect_doc()
  912. ent.check_doc()
  913. def visit(self, visitor):
  914. visitor.visit_begin(self)
  915. module = None
  916. visitor.visit_module(module)
  917. for entity in self._entity_list:
  918. if visitor.visit_needed(entity):
  919. if entity.module != module:
  920. module = entity.module
  921. visitor.visit_module(module)
  922. entity.visit(visitor)
  923. visitor.visit_end()