schema.py 39 KB

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