schema.py 41 KB

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