schema.py 41 KB

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