schema.py 41 KB

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