schema.py 42 KB

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