schema.py 41 KB

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