common.py 88 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565
  1. #
  2. # QAPI helper library
  3. #
  4. # Copyright IBM, Corp. 2011
  5. # Copyright (c) 2013-2018 Red Hat Inc.
  6. #
  7. # Authors:
  8. # Anthony Liguori <aliguori@us.ibm.com>
  9. # Markus Armbruster <armbru@redhat.com>
  10. #
  11. # This work is licensed under the terms of the GNU GPL, version 2.
  12. # See the COPYING file in the top-level directory.
  13. from __future__ import print_function
  14. from contextlib import contextmanager
  15. import errno
  16. import os
  17. import re
  18. import string
  19. import sys
  20. from collections import OrderedDict
  21. builtin_types = {
  22. 'null': 'QTYPE_QNULL',
  23. 'str': 'QTYPE_QSTRING',
  24. 'int': 'QTYPE_QNUM',
  25. 'number': 'QTYPE_QNUM',
  26. 'bool': 'QTYPE_QBOOL',
  27. 'int8': 'QTYPE_QNUM',
  28. 'int16': 'QTYPE_QNUM',
  29. 'int32': 'QTYPE_QNUM',
  30. 'int64': 'QTYPE_QNUM',
  31. 'uint8': 'QTYPE_QNUM',
  32. 'uint16': 'QTYPE_QNUM',
  33. 'uint32': 'QTYPE_QNUM',
  34. 'uint64': 'QTYPE_QNUM',
  35. 'size': 'QTYPE_QNUM',
  36. 'any': None, # any QType possible, actually
  37. 'QType': 'QTYPE_QSTRING',
  38. }
  39. # Are documentation comments required?
  40. doc_required = False
  41. # Whitelist of commands allowed to return a non-dictionary
  42. returns_whitelist = []
  43. # Whitelist of entities allowed to violate case conventions
  44. name_case_whitelist = []
  45. enum_types = {}
  46. struct_types = {}
  47. union_types = {}
  48. all_names = {}
  49. #
  50. # Parsing the schema into expressions
  51. #
  52. def error_path(parent):
  53. res = ''
  54. while parent:
  55. res = ('In file included from %s:%d:\n' % (parent['file'],
  56. parent['line'])) + res
  57. parent = parent['parent']
  58. return res
  59. class QAPIError(Exception):
  60. def __init__(self, fname, line, col, incl_info, msg):
  61. Exception.__init__(self)
  62. self.fname = fname
  63. self.line = line
  64. self.col = col
  65. self.info = incl_info
  66. self.msg = msg
  67. def __str__(self):
  68. loc = '%s:%d' % (self.fname, self.line)
  69. if self.col is not None:
  70. loc += ':%s' % self.col
  71. return error_path(self.info) + '%s: %s' % (loc, self.msg)
  72. class QAPIParseError(QAPIError):
  73. def __init__(self, parser, msg):
  74. col = 1
  75. for ch in parser.src[parser.line_pos:parser.pos]:
  76. if ch == '\t':
  77. col = (col + 7) % 8 + 1
  78. else:
  79. col += 1
  80. QAPIError.__init__(self, parser.fname, parser.line, col,
  81. parser.incl_info, msg)
  82. class QAPISemError(QAPIError):
  83. def __init__(self, info, msg):
  84. QAPIError.__init__(self, info['file'], info['line'], None,
  85. info['parent'], msg)
  86. class QAPIDoc(object):
  87. """
  88. A documentation comment block, either expression or free-form
  89. Expression documentation blocks consist of
  90. * a body section: one line naming the expression, followed by an
  91. overview (any number of lines)
  92. * argument sections: a description of each argument (for commands
  93. and events) or member (for structs, unions and alternates)
  94. * features sections: a description of each feature flag
  95. * additional (non-argument) sections, possibly tagged
  96. Free-form documentation blocks consist only of a body section.
  97. """
  98. class Section(object):
  99. def __init__(self, name=None):
  100. # optional section name (argument/member or section name)
  101. self.name = name
  102. # the list of lines for this section
  103. self.text = ''
  104. def append(self, line):
  105. self.text += line.rstrip() + '\n'
  106. class ArgSection(Section):
  107. def __init__(self, name):
  108. QAPIDoc.Section.__init__(self, name)
  109. self.member = None
  110. def connect(self, member):
  111. self.member = member
  112. def __init__(self, parser, info):
  113. # self._parser is used to report errors with QAPIParseError. The
  114. # resulting error position depends on the state of the parser.
  115. # It happens to be the beginning of the comment. More or less
  116. # servicable, but action at a distance.
  117. self._parser = parser
  118. self.info = info
  119. self.symbol = None
  120. self.body = QAPIDoc.Section()
  121. # dict mapping parameter name to ArgSection
  122. self.args = OrderedDict()
  123. self.features = OrderedDict()
  124. # a list of Section
  125. self.sections = []
  126. # the current section
  127. self._section = self.body
  128. self._append_line = self._append_body_line
  129. def has_section(self, name):
  130. """Return True if we have a section with this name."""
  131. for i in self.sections:
  132. if i.name == name:
  133. return True
  134. return False
  135. def append(self, line):
  136. """
  137. Parse a comment line and add it to the documentation.
  138. The way that the line is dealt with depends on which part of
  139. the documentation we're parsing right now:
  140. * The body section: ._append_line is ._append_body_line
  141. * An argument section: ._append_line is ._append_args_line
  142. * A features section: ._append_line is ._append_features_line
  143. * An additional section: ._append_line is ._append_various_line
  144. """
  145. line = line[1:]
  146. if not line:
  147. self._append_freeform(line)
  148. return
  149. if line[0] != ' ':
  150. raise QAPIParseError(self._parser, "Missing space after #")
  151. line = line[1:]
  152. self._append_line(line)
  153. def end_comment(self):
  154. self._end_section()
  155. @staticmethod
  156. def _is_section_tag(name):
  157. return name in ('Returns:', 'Since:',
  158. # those are often singular or plural
  159. 'Note:', 'Notes:',
  160. 'Example:', 'Examples:',
  161. 'TODO:')
  162. def _append_body_line(self, line):
  163. """
  164. Process a line of documentation text in the body section.
  165. If this a symbol line and it is the section's first line, this
  166. is an expression documentation block for that symbol.
  167. If it's an expression documentation block, another symbol line
  168. begins the argument section for the argument named by it, and
  169. a section tag begins an additional section. Start that
  170. section and append the line to it.
  171. Else, append the line to the current section.
  172. """
  173. name = line.split(' ', 1)[0]
  174. # FIXME not nice: things like '# @foo:' and '# @foo: ' aren't
  175. # recognized, and get silently treated as ordinary text
  176. if not self.symbol and not self.body.text and line.startswith('@'):
  177. if not line.endswith(':'):
  178. raise QAPIParseError(self._parser, "Line should end with :")
  179. self.symbol = line[1:-1]
  180. # FIXME invalid names other than the empty string aren't flagged
  181. if not self.symbol:
  182. raise QAPIParseError(self._parser, "Invalid name")
  183. elif self.symbol:
  184. # This is an expression documentation block
  185. if name.startswith('@') and name.endswith(':'):
  186. self._append_line = self._append_args_line
  187. self._append_args_line(line)
  188. elif line == 'Features:':
  189. self._append_line = self._append_features_line
  190. elif self._is_section_tag(name):
  191. self._append_line = self._append_various_line
  192. self._append_various_line(line)
  193. else:
  194. self._append_freeform(line.strip())
  195. else:
  196. # This is a free-form documentation block
  197. self._append_freeform(line.strip())
  198. def _append_args_line(self, line):
  199. """
  200. Process a line of documentation text in an argument section.
  201. A symbol line begins the next argument section, a section tag
  202. section or a non-indented line after a blank line begins an
  203. additional section. Start that section and append the line to
  204. it.
  205. Else, append the line to the current section.
  206. """
  207. name = line.split(' ', 1)[0]
  208. if name.startswith('@') and name.endswith(':'):
  209. line = line[len(name)+1:]
  210. self._start_args_section(name[1:-1])
  211. elif self._is_section_tag(name):
  212. self._append_line = self._append_various_line
  213. self._append_various_line(line)
  214. return
  215. elif (self._section.text.endswith('\n\n')
  216. and line and not line[0].isspace()):
  217. if line == 'Features:':
  218. self._append_line = self._append_features_line
  219. else:
  220. self._start_section()
  221. self._append_line = self._append_various_line
  222. self._append_various_line(line)
  223. return
  224. self._append_freeform(line.strip())
  225. def _append_features_line(self, line):
  226. name = line.split(' ', 1)[0]
  227. if name.startswith('@') and name.endswith(':'):
  228. line = line[len(name)+1:]
  229. self._start_features_section(name[1:-1])
  230. elif self._is_section_tag(name):
  231. self._append_line = self._append_various_line
  232. self._append_various_line(line)
  233. return
  234. elif (self._section.text.endswith('\n\n')
  235. and line and not line[0].isspace()):
  236. self._start_section()
  237. self._append_line = self._append_various_line
  238. self._append_various_line(line)
  239. return
  240. self._append_freeform(line.strip())
  241. def _append_various_line(self, line):
  242. """
  243. Process a line of documentation text in an additional section.
  244. A symbol line is an error.
  245. A section tag begins an additional section. Start that
  246. section and append the line to it.
  247. Else, append the line to the current section.
  248. """
  249. name = line.split(' ', 1)[0]
  250. if name.startswith('@') and name.endswith(':'):
  251. raise QAPIParseError(self._parser,
  252. "'%s' can't follow '%s' section"
  253. % (name, self.sections[0].name))
  254. elif self._is_section_tag(name):
  255. line = line[len(name)+1:]
  256. self._start_section(name[:-1])
  257. if (not self._section.name or
  258. not self._section.name.startswith('Example')):
  259. line = line.strip()
  260. self._append_freeform(line)
  261. def _start_symbol_section(self, symbols_dict, name):
  262. # FIXME invalid names other than the empty string aren't flagged
  263. if not name:
  264. raise QAPIParseError(self._parser, "Invalid parameter name")
  265. if name in symbols_dict:
  266. raise QAPIParseError(self._parser,
  267. "'%s' parameter name duplicated" % name)
  268. assert not self.sections
  269. self._end_section()
  270. self._section = QAPIDoc.ArgSection(name)
  271. symbols_dict[name] = self._section
  272. def _start_args_section(self, name):
  273. self._start_symbol_section(self.args, name)
  274. def _start_features_section(self, name):
  275. self._start_symbol_section(self.features, name)
  276. def _start_section(self, name=None):
  277. if name in ('Returns', 'Since') and self.has_section(name):
  278. raise QAPIParseError(self._parser,
  279. "Duplicated '%s' section" % name)
  280. self._end_section()
  281. self._section = QAPIDoc.Section(name)
  282. self.sections.append(self._section)
  283. def _end_section(self):
  284. if self._section:
  285. text = self._section.text = self._section.text.strip()
  286. if self._section.name and (not text or text.isspace()):
  287. raise QAPIParseError(self._parser, "Empty doc section '%s'"
  288. % self._section.name)
  289. self._section = None
  290. def _append_freeform(self, line):
  291. match = re.match(r'(@\S+:)', line)
  292. if match:
  293. raise QAPIParseError(self._parser,
  294. "'%s' not allowed in free-form documentation"
  295. % match.group(1))
  296. self._section.append(line)
  297. def connect_member(self, member):
  298. if member.name not in self.args:
  299. # Undocumented TODO outlaw
  300. self.args[member.name] = QAPIDoc.ArgSection(member.name)
  301. self.args[member.name].connect(member)
  302. def check_expr(self, expr):
  303. if self.has_section('Returns') and 'command' not in expr:
  304. raise QAPISemError(self.info,
  305. "'Returns:' is only valid for commands")
  306. def check(self):
  307. bogus = [name for name, section in self.args.items()
  308. if not section.member]
  309. if bogus:
  310. raise QAPISemError(
  311. self.info,
  312. "The following documented members are not in "
  313. "the declaration: %s" % ", ".join(bogus))
  314. class QAPISchemaParser(object):
  315. def __init__(self, fp, previously_included=[], incl_info=None):
  316. self.fname = fp.name
  317. previously_included.append(os.path.abspath(fp.name))
  318. self.incl_info = incl_info
  319. self.src = fp.read()
  320. if self.src == '' or self.src[-1] != '\n':
  321. self.src += '\n'
  322. self.cursor = 0
  323. self.line = 1
  324. self.line_pos = 0
  325. self.exprs = []
  326. self.docs = []
  327. self.accept()
  328. cur_doc = None
  329. while self.tok is not None:
  330. info = {'file': self.fname, 'line': self.line,
  331. 'parent': self.incl_info}
  332. if self.tok == '#':
  333. self.reject_expr_doc(cur_doc)
  334. cur_doc = self.get_doc(info)
  335. self.docs.append(cur_doc)
  336. continue
  337. expr = self.get_expr(False)
  338. if 'include' in expr:
  339. self.reject_expr_doc(cur_doc)
  340. if len(expr) != 1:
  341. raise QAPISemError(info, "Invalid 'include' directive")
  342. include = expr['include']
  343. if not isinstance(include, str):
  344. raise QAPISemError(info,
  345. "Value of 'include' must be a string")
  346. incl_fname = os.path.join(os.path.dirname(self.fname),
  347. include)
  348. self.exprs.append({'expr': {'include': incl_fname},
  349. 'info': info})
  350. exprs_include = self._include(include, info, incl_fname,
  351. previously_included)
  352. if exprs_include:
  353. self.exprs.extend(exprs_include.exprs)
  354. self.docs.extend(exprs_include.docs)
  355. elif "pragma" in expr:
  356. self.reject_expr_doc(cur_doc)
  357. if len(expr) != 1:
  358. raise QAPISemError(info, "Invalid 'pragma' directive")
  359. pragma = expr['pragma']
  360. if not isinstance(pragma, dict):
  361. raise QAPISemError(
  362. info, "Value of 'pragma' must be a dictionary")
  363. for name, value in pragma.items():
  364. self._pragma(name, value, info)
  365. else:
  366. expr_elem = {'expr': expr,
  367. 'info': info}
  368. if cur_doc:
  369. if not cur_doc.symbol:
  370. raise QAPISemError(
  371. cur_doc.info, "Expression documentation required")
  372. expr_elem['doc'] = cur_doc
  373. self.exprs.append(expr_elem)
  374. cur_doc = None
  375. self.reject_expr_doc(cur_doc)
  376. @staticmethod
  377. def reject_expr_doc(doc):
  378. if doc and doc.symbol:
  379. raise QAPISemError(
  380. doc.info,
  381. "Documentation for '%s' is not followed by the definition"
  382. % doc.symbol)
  383. def _include(self, include, info, incl_fname, previously_included):
  384. incl_abs_fname = os.path.abspath(incl_fname)
  385. # catch inclusion cycle
  386. inf = info
  387. while inf:
  388. if incl_abs_fname == os.path.abspath(inf['file']):
  389. raise QAPISemError(info, "Inclusion loop for %s" % include)
  390. inf = inf['parent']
  391. # skip multiple include of the same file
  392. if incl_abs_fname in previously_included:
  393. return None
  394. try:
  395. if sys.version_info[0] >= 3:
  396. fobj = open(incl_fname, 'r', encoding='utf-8')
  397. else:
  398. fobj = open(incl_fname, 'r')
  399. except IOError as e:
  400. raise QAPISemError(info, '%s: %s' % (e.strerror, incl_fname))
  401. return QAPISchemaParser(fobj, previously_included, info)
  402. def _pragma(self, name, value, info):
  403. global doc_required, returns_whitelist, name_case_whitelist
  404. if name == 'doc-required':
  405. if not isinstance(value, bool):
  406. raise QAPISemError(info,
  407. "Pragma 'doc-required' must be boolean")
  408. doc_required = value
  409. elif name == 'returns-whitelist':
  410. if (not isinstance(value, list)
  411. or any([not isinstance(elt, str) for elt in value])):
  412. raise QAPISemError(info,
  413. "Pragma returns-whitelist must be"
  414. " a list of strings")
  415. returns_whitelist = value
  416. elif name == 'name-case-whitelist':
  417. if (not isinstance(value, list)
  418. or any([not isinstance(elt, str) for elt in value])):
  419. raise QAPISemError(info,
  420. "Pragma name-case-whitelist must be"
  421. " a list of strings")
  422. name_case_whitelist = value
  423. else:
  424. raise QAPISemError(info, "Unknown pragma '%s'" % name)
  425. def accept(self, skip_comment=True):
  426. while True:
  427. self.tok = self.src[self.cursor]
  428. self.pos = self.cursor
  429. self.cursor += 1
  430. self.val = None
  431. if self.tok == '#':
  432. if self.src[self.cursor] == '#':
  433. # Start of doc comment
  434. skip_comment = False
  435. self.cursor = self.src.find('\n', self.cursor)
  436. if not skip_comment:
  437. self.val = self.src[self.pos:self.cursor]
  438. return
  439. elif self.tok in '{}:,[]':
  440. return
  441. elif self.tok == "'":
  442. # Note: we accept only printable ASCII
  443. string = ''
  444. esc = False
  445. while True:
  446. ch = self.src[self.cursor]
  447. self.cursor += 1
  448. if ch == '\n':
  449. raise QAPIParseError(self, 'Missing terminating "\'"')
  450. if esc:
  451. # Note: we don't recognize escape sequences
  452. # for control characters
  453. if ch == 'u':
  454. value = 0
  455. for _ in range(0, 4):
  456. ch = self.src[self.cursor]
  457. self.cursor += 1
  458. if ch not in '0123456789abcdefABCDEF':
  459. raise QAPIParseError(self,
  460. '\\u escape needs 4 '
  461. 'hex digits')
  462. value = (value << 4) + int(ch, 16)
  463. # If Python 2 and 3 didn't disagree so much on
  464. # how to handle Unicode, then we could allow
  465. # Unicode string defaults. But most of QAPI is
  466. # ASCII-only, so we aren't losing much for now.
  467. if not value or value > 0x7f:
  468. raise QAPIParseError(self,
  469. 'For now, \\u escape '
  470. 'only supports non-zero '
  471. 'values up to \\u007f')
  472. ch = chr(value)
  473. elif ch not in '\\/\'"':
  474. raise QAPIParseError(self,
  475. "Unknown escape \\%s" % ch)
  476. esc = False
  477. elif ch == '\\':
  478. esc = True
  479. continue
  480. elif ch == "'":
  481. self.val = string
  482. return
  483. if ord(ch) < 32 or ord(ch) >= 127:
  484. raise QAPIParseError(
  485. self, "Funny character in string")
  486. string += ch
  487. elif self.src.startswith('true', self.pos):
  488. self.val = True
  489. self.cursor += 3
  490. return
  491. elif self.src.startswith('false', self.pos):
  492. self.val = False
  493. self.cursor += 4
  494. return
  495. elif self.src.startswith('null', self.pos):
  496. self.val = None
  497. self.cursor += 3
  498. return
  499. elif self.tok == '\n':
  500. if self.cursor == len(self.src):
  501. self.tok = None
  502. return
  503. self.line += 1
  504. self.line_pos = self.cursor
  505. elif not self.tok.isspace():
  506. raise QAPIParseError(self, 'Stray "%s"' % self.tok)
  507. def get_members(self):
  508. expr = OrderedDict()
  509. if self.tok == '}':
  510. self.accept()
  511. return expr
  512. if self.tok != "'":
  513. raise QAPIParseError(self, 'Expected string or "}"')
  514. while True:
  515. key = self.val
  516. self.accept()
  517. if self.tok != ':':
  518. raise QAPIParseError(self, 'Expected ":"')
  519. self.accept()
  520. if key in expr:
  521. raise QAPIParseError(self, 'Duplicate key "%s"' % key)
  522. expr[key] = self.get_expr(True)
  523. if self.tok == '}':
  524. self.accept()
  525. return expr
  526. if self.tok != ',':
  527. raise QAPIParseError(self, 'Expected "," or "}"')
  528. self.accept()
  529. if self.tok != "'":
  530. raise QAPIParseError(self, 'Expected string')
  531. def get_values(self):
  532. expr = []
  533. if self.tok == ']':
  534. self.accept()
  535. return expr
  536. if self.tok not in "{['tfn":
  537. raise QAPIParseError(self, 'Expected "{", "[", "]", string, '
  538. 'boolean or "null"')
  539. while True:
  540. expr.append(self.get_expr(True))
  541. if self.tok == ']':
  542. self.accept()
  543. return expr
  544. if self.tok != ',':
  545. raise QAPIParseError(self, 'Expected "," or "]"')
  546. self.accept()
  547. def get_expr(self, nested):
  548. if self.tok != '{' and not nested:
  549. raise QAPIParseError(self, 'Expected "{"')
  550. if self.tok == '{':
  551. self.accept()
  552. expr = self.get_members()
  553. elif self.tok == '[':
  554. self.accept()
  555. expr = self.get_values()
  556. elif self.tok in "'tfn":
  557. expr = self.val
  558. self.accept()
  559. else:
  560. raise QAPIParseError(self, 'Expected "{", "[", string, '
  561. 'boolean or "null"')
  562. return expr
  563. def get_doc(self, info):
  564. if self.val != '##':
  565. raise QAPIParseError(self, "Junk after '##' at start of "
  566. "documentation comment")
  567. doc = QAPIDoc(self, info)
  568. self.accept(False)
  569. while self.tok == '#':
  570. if self.val.startswith('##'):
  571. # End of doc comment
  572. if self.val != '##':
  573. raise QAPIParseError(self, "Junk after '##' at end of "
  574. "documentation comment")
  575. doc.end_comment()
  576. self.accept()
  577. return doc
  578. else:
  579. doc.append(self.val)
  580. self.accept(False)
  581. raise QAPIParseError(self, "Documentation comment must end with '##'")
  582. #
  583. # Semantic analysis of schema expressions
  584. # TODO fold into QAPISchema
  585. # TODO catching name collisions in generated code would be nice
  586. #
  587. def find_base_members(base):
  588. if isinstance(base, dict):
  589. return base
  590. base_struct_define = struct_types.get(base)
  591. if not base_struct_define:
  592. return None
  593. return base_struct_define['data']
  594. # Return the qtype of an alternate branch, or None on error.
  595. def find_alternate_member_qtype(qapi_type):
  596. if qapi_type in builtin_types:
  597. return builtin_types[qapi_type]
  598. elif qapi_type in struct_types:
  599. return 'QTYPE_QDICT'
  600. elif qapi_type in enum_types:
  601. return 'QTYPE_QSTRING'
  602. elif qapi_type in union_types:
  603. return 'QTYPE_QDICT'
  604. return None
  605. # Return the discriminator enum define if discriminator is specified as an
  606. # enum type, otherwise return None.
  607. def discriminator_find_enum_define(expr):
  608. base = expr.get('base')
  609. discriminator = expr.get('discriminator')
  610. if not (discriminator and base):
  611. return None
  612. base_members = find_base_members(base)
  613. if not base_members:
  614. return None
  615. discriminator_value = base_members.get(discriminator)
  616. if not discriminator_value:
  617. return None
  618. return enum_types.get(discriminator_value['type'])
  619. # Names must be letters, numbers, -, and _. They must start with letter,
  620. # except for downstream extensions which must start with __RFQDN_.
  621. # Dots are only valid in the downstream extension prefix.
  622. valid_name = re.compile(r'^(__[a-zA-Z0-9.-]+_)?'
  623. '[a-zA-Z][a-zA-Z0-9_-]*$')
  624. def check_name(info, source, name, allow_optional=False,
  625. enum_member=False):
  626. global valid_name
  627. membername = name
  628. if not isinstance(name, str):
  629. raise QAPISemError(info, "%s requires a string name" % source)
  630. if name.startswith('*'):
  631. membername = name[1:]
  632. if not allow_optional:
  633. raise QAPISemError(info, "%s does not allow optional name '%s'"
  634. % (source, name))
  635. # Enum members can start with a digit, because the generated C
  636. # code always prefixes it with the enum name
  637. if enum_member and membername[0].isdigit():
  638. membername = 'D' + membername
  639. # Reserve the entire 'q_' namespace for c_name(), and for 'q_empty'
  640. # and 'q_obj_*' implicit type names.
  641. if not valid_name.match(membername) or \
  642. c_name(membername, False).startswith('q_'):
  643. raise QAPISemError(info, "%s uses invalid name '%s'" % (source, name))
  644. def add_name(name, info, meta, implicit=False):
  645. global all_names
  646. check_name(info, "'%s'" % meta, name)
  647. # FIXME should reject names that differ only in '_' vs. '.'
  648. # vs. '-', because they're liable to clash in generated C.
  649. if name in all_names:
  650. raise QAPISemError(info, "%s '%s' is already defined"
  651. % (all_names[name], name))
  652. if not implicit and (name.endswith('Kind') or name.endswith('List')):
  653. raise QAPISemError(info, "%s '%s' should not end in '%s'"
  654. % (meta, name, name[-4:]))
  655. all_names[name] = meta
  656. def check_if(expr, info):
  657. def check_if_str(ifcond, info):
  658. if not isinstance(ifcond, str):
  659. raise QAPISemError(
  660. info, "'if' condition must be a string or a list of strings")
  661. if ifcond == '':
  662. raise QAPISemError(info, "'if' condition '' makes no sense")
  663. ifcond = expr.get('if')
  664. if ifcond is None:
  665. return
  666. if isinstance(ifcond, list):
  667. if ifcond == []:
  668. raise QAPISemError(info, "'if' condition [] is useless")
  669. for elt in ifcond:
  670. check_if_str(elt, info)
  671. else:
  672. check_if_str(ifcond, info)
  673. def check_type(info, source, value,
  674. allow_array=False, allow_dict=False, allow_metas=[]):
  675. global all_names
  676. if value is None:
  677. return
  678. # Check if array type for value is okay
  679. if isinstance(value, list):
  680. if not allow_array:
  681. raise QAPISemError(info, "%s cannot be an array" % source)
  682. if len(value) != 1 or not isinstance(value[0], str):
  683. raise QAPISemError(info,
  684. "%s: array type must contain single type name" %
  685. source)
  686. value = value[0]
  687. # Check if type name for value is okay
  688. if isinstance(value, str):
  689. if value not in all_names:
  690. raise QAPISemError(info, "%s uses unknown type '%s'"
  691. % (source, value))
  692. if not all_names[value] in allow_metas:
  693. raise QAPISemError(info, "%s cannot use %s type '%s'" %
  694. (source, all_names[value], value))
  695. return
  696. if not allow_dict:
  697. raise QAPISemError(info, "%s should be a type name" % source)
  698. if not isinstance(value, OrderedDict):
  699. raise QAPISemError(info,
  700. "%s should be a dictionary or type name" % source)
  701. # value is a dictionary, check that each member is okay
  702. for (key, arg) in value.items():
  703. check_name(info, "Member of %s" % source, key,
  704. allow_optional=True)
  705. if c_name(key, False) == 'u' or c_name(key, False).startswith('has_'):
  706. raise QAPISemError(info, "Member of %s uses reserved name '%s'"
  707. % (source, key))
  708. # Todo: allow dictionaries to represent default values of
  709. # an optional argument.
  710. check_known_keys(info, "member '%s' of %s" % (key, source),
  711. arg, ['type'], ['if'])
  712. check_type(info, "Member '%s' of %s" % (key, source),
  713. arg['type'], allow_array=True,
  714. allow_metas=['built-in', 'union', 'alternate', 'struct',
  715. 'enum'])
  716. def check_command(expr, info):
  717. name = expr['command']
  718. boxed = expr.get('boxed', False)
  719. args_meta = ['struct']
  720. if boxed:
  721. args_meta += ['union']
  722. check_type(info, "'data' for command '%s'" % name,
  723. expr.get('data'), allow_dict=not boxed,
  724. allow_metas=args_meta)
  725. returns_meta = ['union', 'struct']
  726. if name in returns_whitelist:
  727. returns_meta += ['built-in', 'alternate', 'enum']
  728. check_type(info, "'returns' for command '%s'" % name,
  729. expr.get('returns'), allow_array=True,
  730. allow_metas=returns_meta)
  731. def check_event(expr, info):
  732. name = expr['event']
  733. boxed = expr.get('boxed', False)
  734. meta = ['struct']
  735. if boxed:
  736. meta += ['union']
  737. check_type(info, "'data' for event '%s'" % name,
  738. expr.get('data'), allow_dict=not boxed,
  739. allow_metas=meta)
  740. def enum_get_names(expr):
  741. return [e['name'] for e in expr['data']]
  742. def check_union(expr, info):
  743. name = expr['union']
  744. base = expr.get('base')
  745. discriminator = expr.get('discriminator')
  746. members = expr['data']
  747. # Two types of unions, determined by discriminator.
  748. # With no discriminator it is a simple union.
  749. if discriminator is None:
  750. enum_define = None
  751. allow_metas = ['built-in', 'union', 'alternate', 'struct', 'enum']
  752. if base is not None:
  753. raise QAPISemError(info, "Simple union '%s' must not have a base" %
  754. name)
  755. # Else, it's a flat union.
  756. else:
  757. # The object must have a string or dictionary 'base'.
  758. check_type(info, "'base' for union '%s'" % name,
  759. base, allow_dict=True,
  760. allow_metas=['struct'])
  761. if not base:
  762. raise QAPISemError(info, "Flat union '%s' must have a base"
  763. % name)
  764. base_members = find_base_members(base)
  765. assert base_members is not None
  766. # The value of member 'discriminator' must name a non-optional
  767. # member of the base struct.
  768. check_name(info, "Discriminator of flat union '%s'" % name,
  769. discriminator)
  770. discriminator_value = base_members.get(discriminator)
  771. if not discriminator_value:
  772. raise QAPISemError(info,
  773. "Discriminator '%s' is not a member of base "
  774. "struct '%s'"
  775. % (discriminator, base))
  776. if discriminator_value.get('if'):
  777. raise QAPISemError(info, 'The discriminator %s.%s for union %s '
  778. 'must not be conditional' %
  779. (base, discriminator, name))
  780. enum_define = enum_types.get(discriminator_value['type'])
  781. allow_metas = ['struct']
  782. # Do not allow string discriminator
  783. if not enum_define:
  784. raise QAPISemError(info,
  785. "Discriminator '%s' must be of enumeration "
  786. "type" % discriminator)
  787. # Check every branch; don't allow an empty union
  788. if len(members) == 0:
  789. raise QAPISemError(info, "Union '%s' cannot have empty 'data'" % name)
  790. for (key, value) in members.items():
  791. check_name(info, "Member of union '%s'" % name, key)
  792. check_known_keys(info, "member '%s' of union '%s'" % (key, name),
  793. value, ['type'], ['if'])
  794. # Each value must name a known type
  795. check_type(info, "Member '%s' of union '%s'" % (key, name),
  796. value['type'],
  797. allow_array=not base, allow_metas=allow_metas)
  798. # If the discriminator names an enum type, then all members
  799. # of 'data' must also be members of the enum type.
  800. if enum_define:
  801. if key not in enum_get_names(enum_define):
  802. raise QAPISemError(info,
  803. "Discriminator value '%s' is not found in "
  804. "enum '%s'"
  805. % (key, enum_define['enum']))
  806. def check_alternate(expr, info):
  807. name = expr['alternate']
  808. members = expr['data']
  809. types_seen = {}
  810. # Check every branch; require at least two branches
  811. if len(members) < 2:
  812. raise QAPISemError(info,
  813. "Alternate '%s' should have at least two branches "
  814. "in 'data'" % name)
  815. for (key, value) in members.items():
  816. check_name(info, "Member of alternate '%s'" % name, key)
  817. check_known_keys(info,
  818. "member '%s' of alternate '%s'" % (key, name),
  819. value, ['type'], ['if'])
  820. typ = value['type']
  821. # Ensure alternates have no type conflicts.
  822. check_type(info, "Member '%s' of alternate '%s'" % (key, name), typ,
  823. allow_metas=['built-in', 'union', 'struct', 'enum'])
  824. qtype = find_alternate_member_qtype(typ)
  825. if not qtype:
  826. raise QAPISemError(info, "Alternate '%s' member '%s' cannot use "
  827. "type '%s'" % (name, key, typ))
  828. conflicting = set([qtype])
  829. if qtype == 'QTYPE_QSTRING':
  830. enum_expr = enum_types.get(typ)
  831. if enum_expr:
  832. for v in enum_get_names(enum_expr):
  833. if v in ['on', 'off']:
  834. conflicting.add('QTYPE_QBOOL')
  835. if re.match(r'[-+0-9.]', v): # lazy, could be tightened
  836. conflicting.add('QTYPE_QNUM')
  837. else:
  838. conflicting.add('QTYPE_QNUM')
  839. conflicting.add('QTYPE_QBOOL')
  840. for qt in conflicting:
  841. if qt in types_seen:
  842. raise QAPISemError(info, "Alternate '%s' member '%s' can't "
  843. "be distinguished from member '%s'"
  844. % (name, key, types_seen[qt]))
  845. types_seen[qt] = key
  846. def check_enum(expr, info):
  847. name = expr['enum']
  848. members = expr['data']
  849. prefix = expr.get('prefix')
  850. if not isinstance(members, list):
  851. raise QAPISemError(info,
  852. "Enum '%s' requires an array for 'data'" % name)
  853. if prefix is not None and not isinstance(prefix, str):
  854. raise QAPISemError(info,
  855. "Enum '%s' requires a string for 'prefix'" % name)
  856. for member in members:
  857. source = "dictionary member of enum '%s'" % name
  858. check_known_keys(info, source, member, ['name'], ['if'])
  859. check_if(member, info)
  860. check_name(info, "Member of enum '%s'" % name, member['name'],
  861. enum_member=True)
  862. def check_struct(expr, info):
  863. name = expr['struct']
  864. members = expr['data']
  865. features = expr.get('features')
  866. check_type(info, "'data' for struct '%s'" % name, members,
  867. allow_dict=True)
  868. check_type(info, "'base' for struct '%s'" % name, expr.get('base'),
  869. allow_metas=['struct'])
  870. if features:
  871. if not isinstance(features, list):
  872. raise QAPISemError(info,
  873. "Struct '%s' requires an array for 'features'" %
  874. name)
  875. for f in features:
  876. assert isinstance(f, dict)
  877. check_known_keys(info, "feature of struct %s" % name, f,
  878. ['name'], ['if'])
  879. check_if(f, info)
  880. check_name(info, "Feature of struct %s" % name, f['name'])
  881. def check_known_keys(info, source, keys, required, optional):
  882. def pprint(elems):
  883. return ', '.join("'" + e + "'" for e in sorted(elems))
  884. missing = set(required) - set(keys)
  885. if missing:
  886. raise QAPISemError(info, "Key%s %s %s missing from %s"
  887. % ('s' if len(missing) > 1 else '', pprint(missing),
  888. 'are' if len(missing) > 1 else 'is', source))
  889. allowed = set(required + optional)
  890. unknown = set(keys) - allowed
  891. if unknown:
  892. raise QAPISemError(info, "Unknown key%s %s in %s\nValid keys are %s."
  893. % ('s' if len(unknown) > 1 else '', pprint(unknown),
  894. source, pprint(allowed)))
  895. def check_keys(expr_elem, meta, required, optional=[]):
  896. expr = expr_elem['expr']
  897. info = expr_elem['info']
  898. name = expr[meta]
  899. if not isinstance(name, str):
  900. raise QAPISemError(info, "'%s' key must have a string value" % meta)
  901. required = required + [meta]
  902. source = "%s '%s'" % (meta, name)
  903. check_known_keys(info, source, expr.keys(), required, optional)
  904. for (key, value) in expr.items():
  905. if key in ['gen', 'success-response'] and value is not False:
  906. raise QAPISemError(info,
  907. "'%s' of %s '%s' should only use false value"
  908. % (key, meta, name))
  909. if (key in ['boxed', 'allow-oob', 'allow-preconfig']
  910. and value is not True):
  911. raise QAPISemError(info,
  912. "'%s' of %s '%s' should only use true value"
  913. % (key, meta, name))
  914. if key == 'if':
  915. check_if(expr, info)
  916. def normalize_enum(expr):
  917. if isinstance(expr['data'], list):
  918. expr['data'] = [m if isinstance(m, dict) else {'name': m}
  919. for m in expr['data']]
  920. def normalize_members(members):
  921. if isinstance(members, OrderedDict):
  922. for key, arg in members.items():
  923. if isinstance(arg, dict):
  924. continue
  925. members[key] = {'type': arg}
  926. def normalize_features(features):
  927. if isinstance(features, list):
  928. features[:] = [f if isinstance(f, dict) else {'name': f}
  929. for f in features]
  930. def check_exprs(exprs):
  931. global all_names
  932. # Populate name table with names of built-in types
  933. for builtin in builtin_types.keys():
  934. all_names[builtin] = 'built-in'
  935. # Learn the types and check for valid expression keys
  936. for expr_elem in exprs:
  937. expr = expr_elem['expr']
  938. info = expr_elem['info']
  939. doc = expr_elem.get('doc')
  940. if 'include' in expr:
  941. continue
  942. if not doc and doc_required:
  943. raise QAPISemError(info,
  944. "Expression missing documentation comment")
  945. if 'enum' in expr:
  946. meta = 'enum'
  947. check_keys(expr_elem, 'enum', ['data'], ['if', 'prefix'])
  948. normalize_enum(expr)
  949. enum_types[expr[meta]] = expr
  950. elif 'union' in expr:
  951. meta = 'union'
  952. check_keys(expr_elem, 'union', ['data'],
  953. ['base', 'discriminator', 'if'])
  954. normalize_members(expr.get('base'))
  955. normalize_members(expr['data'])
  956. union_types[expr[meta]] = expr
  957. elif 'alternate' in expr:
  958. meta = 'alternate'
  959. check_keys(expr_elem, 'alternate', ['data'], ['if'])
  960. normalize_members(expr['data'])
  961. elif 'struct' in expr:
  962. meta = 'struct'
  963. check_keys(expr_elem, 'struct', ['data'],
  964. ['base', 'if', 'features'])
  965. normalize_members(expr['data'])
  966. normalize_features(expr.get('features'))
  967. struct_types[expr[meta]] = expr
  968. elif 'command' in expr:
  969. meta = 'command'
  970. check_keys(expr_elem, 'command', [],
  971. ['data', 'returns', 'gen', 'success-response',
  972. 'boxed', 'allow-oob', 'allow-preconfig', 'if'])
  973. normalize_members(expr.get('data'))
  974. elif 'event' in expr:
  975. meta = 'event'
  976. check_keys(expr_elem, 'event', [], ['data', 'boxed', 'if'])
  977. normalize_members(expr.get('data'))
  978. else:
  979. raise QAPISemError(expr_elem['info'],
  980. "Expression is missing metatype")
  981. name = expr[meta]
  982. add_name(name, info, meta)
  983. if doc and doc.symbol != name:
  984. raise QAPISemError(info, "Definition of '%s' follows documentation"
  985. " for '%s'" % (name, doc.symbol))
  986. # Try again for hidden UnionKind enum
  987. for expr_elem in exprs:
  988. expr = expr_elem['expr']
  989. if 'include' in expr:
  990. continue
  991. if 'union' in expr and not discriminator_find_enum_define(expr):
  992. name = '%sKind' % expr['union']
  993. elif 'alternate' in expr:
  994. name = '%sKind' % expr['alternate']
  995. else:
  996. continue
  997. enum_types[name] = {'enum': name}
  998. add_name(name, info, 'enum', implicit=True)
  999. # Validate that exprs make sense
  1000. for expr_elem in exprs:
  1001. expr = expr_elem['expr']
  1002. info = expr_elem['info']
  1003. doc = expr_elem.get('doc')
  1004. if 'include' in expr:
  1005. continue
  1006. if 'enum' in expr:
  1007. check_enum(expr, info)
  1008. elif 'union' in expr:
  1009. check_union(expr, info)
  1010. elif 'alternate' in expr:
  1011. check_alternate(expr, info)
  1012. elif 'struct' in expr:
  1013. check_struct(expr, info)
  1014. elif 'command' in expr:
  1015. check_command(expr, info)
  1016. elif 'event' in expr:
  1017. check_event(expr, info)
  1018. else:
  1019. assert False, 'unexpected meta type'
  1020. if doc:
  1021. doc.check_expr(expr)
  1022. return exprs
  1023. #
  1024. # Schema compiler frontend
  1025. #
  1026. def listify_cond(ifcond):
  1027. if not ifcond:
  1028. return []
  1029. if not isinstance(ifcond, list):
  1030. return [ifcond]
  1031. return ifcond
  1032. class QAPISchemaEntity(object):
  1033. def __init__(self, name, info, doc, ifcond=None):
  1034. assert name is None or isinstance(name, str)
  1035. self.name = name
  1036. self.module = None
  1037. # For explicitly defined entities, info points to the (explicit)
  1038. # definition. For builtins (and their arrays), info is None.
  1039. # For implicitly defined entities, info points to a place that
  1040. # triggered the implicit definition (there may be more than one
  1041. # such place).
  1042. self.info = info
  1043. self.doc = doc
  1044. self._ifcond = ifcond # self.ifcond is set only after .check()
  1045. def c_name(self):
  1046. return c_name(self.name)
  1047. def check(self, schema):
  1048. if isinstance(self._ifcond, QAPISchemaType):
  1049. # inherit the condition from a type
  1050. typ = self._ifcond
  1051. typ.check(schema)
  1052. self.ifcond = typ.ifcond
  1053. else:
  1054. self.ifcond = listify_cond(self._ifcond)
  1055. if self.info:
  1056. self.module = os.path.relpath(self.info['file'],
  1057. os.path.dirname(schema.fname))
  1058. def is_implicit(self):
  1059. return not self.info
  1060. def visit(self, visitor):
  1061. pass
  1062. class QAPISchemaVisitor(object):
  1063. def visit_begin(self, schema):
  1064. pass
  1065. def visit_end(self):
  1066. pass
  1067. def visit_module(self, fname):
  1068. pass
  1069. def visit_needed(self, entity):
  1070. # Default to visiting everything
  1071. return True
  1072. def visit_include(self, fname, info):
  1073. pass
  1074. def visit_builtin_type(self, name, info, json_type):
  1075. pass
  1076. def visit_enum_type(self, name, info, ifcond, members, prefix):
  1077. pass
  1078. def visit_array_type(self, name, info, ifcond, element_type):
  1079. pass
  1080. def visit_object_type(self, name, info, ifcond, base, members, variants,
  1081. features):
  1082. pass
  1083. def visit_object_type_flat(self, name, info, ifcond, members, variants,
  1084. features):
  1085. pass
  1086. def visit_alternate_type(self, name, info, ifcond, variants):
  1087. pass
  1088. def visit_command(self, name, info, ifcond, arg_type, ret_type, gen,
  1089. success_response, boxed, allow_oob, allow_preconfig):
  1090. pass
  1091. def visit_event(self, name, info, ifcond, arg_type, boxed):
  1092. pass
  1093. class QAPISchemaInclude(QAPISchemaEntity):
  1094. def __init__(self, fname, info):
  1095. QAPISchemaEntity.__init__(self, None, info, None)
  1096. self.fname = fname
  1097. def visit(self, visitor):
  1098. visitor.visit_include(self.fname, self.info)
  1099. class QAPISchemaType(QAPISchemaEntity):
  1100. # Return the C type for common use.
  1101. # For the types we commonly box, this is a pointer type.
  1102. def c_type(self):
  1103. pass
  1104. # Return the C type to be used in a parameter list.
  1105. def c_param_type(self):
  1106. return self.c_type()
  1107. # Return the C type to be used where we suppress boxing.
  1108. def c_unboxed_type(self):
  1109. return self.c_type()
  1110. def json_type(self):
  1111. pass
  1112. def alternate_qtype(self):
  1113. json2qtype = {
  1114. 'null': 'QTYPE_QNULL',
  1115. 'string': 'QTYPE_QSTRING',
  1116. 'number': 'QTYPE_QNUM',
  1117. 'int': 'QTYPE_QNUM',
  1118. 'boolean': 'QTYPE_QBOOL',
  1119. 'object': 'QTYPE_QDICT'
  1120. }
  1121. return json2qtype.get(self.json_type())
  1122. def doc_type(self):
  1123. if self.is_implicit():
  1124. return None
  1125. return self.name
  1126. class QAPISchemaBuiltinType(QAPISchemaType):
  1127. def __init__(self, name, json_type, c_type):
  1128. QAPISchemaType.__init__(self, name, None, None)
  1129. assert not c_type or isinstance(c_type, str)
  1130. assert json_type in ('string', 'number', 'int', 'boolean', 'null',
  1131. 'value')
  1132. self._json_type_name = json_type
  1133. self._c_type_name = c_type
  1134. def c_name(self):
  1135. return self.name
  1136. def c_type(self):
  1137. return self._c_type_name
  1138. def c_param_type(self):
  1139. if self.name == 'str':
  1140. return 'const ' + self._c_type_name
  1141. return self._c_type_name
  1142. def json_type(self):
  1143. return self._json_type_name
  1144. def doc_type(self):
  1145. return self.json_type()
  1146. def visit(self, visitor):
  1147. visitor.visit_builtin_type(self.name, self.info, self.json_type())
  1148. class QAPISchemaEnumType(QAPISchemaType):
  1149. def __init__(self, name, info, doc, ifcond, members, prefix):
  1150. QAPISchemaType.__init__(self, name, info, doc, ifcond)
  1151. for m in members:
  1152. assert isinstance(m, QAPISchemaMember)
  1153. m.set_owner(name)
  1154. assert prefix is None or isinstance(prefix, str)
  1155. self.members = members
  1156. self.prefix = prefix
  1157. def check(self, schema):
  1158. QAPISchemaType.check(self, schema)
  1159. seen = {}
  1160. for m in self.members:
  1161. m.check_clash(self.info, seen)
  1162. if self.doc:
  1163. self.doc.connect_member(m)
  1164. def is_implicit(self):
  1165. # See QAPISchema._make_implicit_enum_type() and ._def_predefineds()
  1166. return self.name.endswith('Kind') or self.name == 'QType'
  1167. def c_type(self):
  1168. return c_name(self.name)
  1169. def member_names(self):
  1170. return [m.name for m in self.members]
  1171. def json_type(self):
  1172. return 'string'
  1173. def visit(self, visitor):
  1174. visitor.visit_enum_type(self.name, self.info, self.ifcond,
  1175. self.members, self.prefix)
  1176. class QAPISchemaArrayType(QAPISchemaType):
  1177. def __init__(self, name, info, element_type):
  1178. QAPISchemaType.__init__(self, name, info, None, None)
  1179. assert isinstance(element_type, str)
  1180. self._element_type_name = element_type
  1181. self.element_type = None
  1182. def check(self, schema):
  1183. QAPISchemaType.check(self, schema)
  1184. self.element_type = schema.lookup_type(self._element_type_name)
  1185. assert self.element_type
  1186. self.element_type.check(schema)
  1187. self.module = self.element_type.module
  1188. self.ifcond = self.element_type.ifcond
  1189. def is_implicit(self):
  1190. return True
  1191. def c_type(self):
  1192. return c_name(self.name) + pointer_suffix
  1193. def json_type(self):
  1194. return 'array'
  1195. def doc_type(self):
  1196. elt_doc_type = self.element_type.doc_type()
  1197. if not elt_doc_type:
  1198. return None
  1199. return 'array of ' + elt_doc_type
  1200. def visit(self, visitor):
  1201. visitor.visit_array_type(self.name, self.info, self.ifcond,
  1202. self.element_type)
  1203. class QAPISchemaObjectType(QAPISchemaType):
  1204. def __init__(self, name, info, doc, ifcond,
  1205. base, local_members, variants, features):
  1206. # struct has local_members, optional base, and no variants
  1207. # flat union has base, variants, and no local_members
  1208. # simple union has local_members, variants, and no base
  1209. QAPISchemaType.__init__(self, name, info, doc, ifcond)
  1210. assert base is None or isinstance(base, str)
  1211. for m in local_members:
  1212. assert isinstance(m, QAPISchemaObjectTypeMember)
  1213. m.set_owner(name)
  1214. if variants is not None:
  1215. assert isinstance(variants, QAPISchemaObjectTypeVariants)
  1216. variants.set_owner(name)
  1217. for f in features:
  1218. assert isinstance(f, QAPISchemaFeature)
  1219. f.set_owner(name)
  1220. self._base_name = base
  1221. self.base = None
  1222. self.local_members = local_members
  1223. self.variants = variants
  1224. self.members = None
  1225. self.features = features
  1226. def check(self, schema):
  1227. QAPISchemaType.check(self, schema)
  1228. if self.members is False: # check for cycles
  1229. raise QAPISemError(self.info,
  1230. "Object %s contains itself" % self.name)
  1231. if self.members:
  1232. return
  1233. self.members = False # mark as being checked
  1234. seen = OrderedDict()
  1235. if self._base_name:
  1236. self.base = schema.lookup_type(self._base_name)
  1237. assert isinstance(self.base, QAPISchemaObjectType)
  1238. self.base.check(schema)
  1239. self.base.check_clash(self.info, seen)
  1240. for m in self.local_members:
  1241. m.check(schema)
  1242. m.check_clash(self.info, seen)
  1243. if self.doc:
  1244. self.doc.connect_member(m)
  1245. self.members = seen.values()
  1246. if self.variants:
  1247. self.variants.check(schema, seen)
  1248. assert self.variants.tag_member in self.members
  1249. self.variants.check_clash(self.info, seen)
  1250. # Features are in a name space separate from members
  1251. seen = {}
  1252. for f in self.features:
  1253. f.check_clash(self.info, seen)
  1254. if self.doc:
  1255. self.doc.check()
  1256. # Check that the members of this type do not cause duplicate JSON members,
  1257. # and update seen to track the members seen so far. Report any errors
  1258. # on behalf of info, which is not necessarily self.info
  1259. def check_clash(self, info, seen):
  1260. assert not self.variants # not implemented
  1261. for m in self.members:
  1262. m.check_clash(info, seen)
  1263. def is_implicit(self):
  1264. # See QAPISchema._make_implicit_object_type(), as well as
  1265. # _def_predefineds()
  1266. return self.name.startswith('q_')
  1267. def is_empty(self):
  1268. assert self.members is not None
  1269. return not self.members and not self.variants
  1270. def c_name(self):
  1271. assert self.name != 'q_empty'
  1272. return QAPISchemaType.c_name(self)
  1273. def c_type(self):
  1274. assert not self.is_implicit()
  1275. return c_name(self.name) + pointer_suffix
  1276. def c_unboxed_type(self):
  1277. return c_name(self.name)
  1278. def json_type(self):
  1279. return 'object'
  1280. def visit(self, visitor):
  1281. visitor.visit_object_type(self.name, self.info, self.ifcond,
  1282. self.base, self.local_members, self.variants,
  1283. self.features)
  1284. visitor.visit_object_type_flat(self.name, self.info, self.ifcond,
  1285. self.members, self.variants,
  1286. self.features)
  1287. class QAPISchemaMember(object):
  1288. """ Represents object members, enum members and features """
  1289. role = 'member'
  1290. def __init__(self, name, ifcond=None):
  1291. assert isinstance(name, str)
  1292. self.name = name
  1293. self.ifcond = listify_cond(ifcond)
  1294. self.owner = None
  1295. def set_owner(self, name):
  1296. assert not self.owner
  1297. self.owner = name
  1298. def check_clash(self, info, seen):
  1299. cname = c_name(self.name)
  1300. if cname.lower() != cname and self.owner not in name_case_whitelist:
  1301. raise QAPISemError(info,
  1302. "%s should not use uppercase" % self.describe())
  1303. if cname in seen:
  1304. raise QAPISemError(info, "%s collides with %s" %
  1305. (self.describe(), seen[cname].describe()))
  1306. seen[cname] = self
  1307. def _pretty_owner(self):
  1308. owner = self.owner
  1309. if owner.startswith('q_obj_'):
  1310. # See QAPISchema._make_implicit_object_type() - reverse the
  1311. # mapping there to create a nice human-readable description
  1312. owner = owner[6:]
  1313. if owner.endswith('-arg'):
  1314. return '(parameter of %s)' % owner[:-4]
  1315. elif owner.endswith('-base'):
  1316. return '(base of %s)' % owner[:-5]
  1317. else:
  1318. assert owner.endswith('-wrapper')
  1319. # Unreachable and not implemented
  1320. assert False
  1321. if owner.endswith('Kind'):
  1322. # See QAPISchema._make_implicit_enum_type()
  1323. return '(branch of %s)' % owner[:-4]
  1324. return '(%s of %s)' % (self.role, owner)
  1325. def describe(self):
  1326. return "'%s' %s" % (self.name, self._pretty_owner())
  1327. class QAPISchemaFeature(QAPISchemaMember):
  1328. role = 'feature'
  1329. class QAPISchemaObjectTypeMember(QAPISchemaMember):
  1330. def __init__(self, name, typ, optional, ifcond=None):
  1331. QAPISchemaMember.__init__(self, name, ifcond)
  1332. assert isinstance(typ, str)
  1333. assert isinstance(optional, bool)
  1334. self._type_name = typ
  1335. self.type = None
  1336. self.optional = optional
  1337. def check(self, schema):
  1338. assert self.owner
  1339. self.type = schema.lookup_type(self._type_name)
  1340. assert self.type
  1341. class QAPISchemaObjectTypeVariants(object):
  1342. def __init__(self, tag_name, tag_member, variants):
  1343. # Flat unions pass tag_name but not tag_member.
  1344. # Simple unions and alternates pass tag_member but not tag_name.
  1345. # After check(), tag_member is always set, and tag_name remains
  1346. # a reliable witness of being used by a flat union.
  1347. assert bool(tag_member) != bool(tag_name)
  1348. assert (isinstance(tag_name, str) or
  1349. isinstance(tag_member, QAPISchemaObjectTypeMember))
  1350. assert len(variants) > 0
  1351. for v in variants:
  1352. assert isinstance(v, QAPISchemaObjectTypeVariant)
  1353. self._tag_name = tag_name
  1354. self.tag_member = tag_member
  1355. self.variants = variants
  1356. def set_owner(self, name):
  1357. for v in self.variants:
  1358. v.set_owner(name)
  1359. def check(self, schema, seen):
  1360. if not self.tag_member: # flat union
  1361. self.tag_member = seen[c_name(self._tag_name)]
  1362. assert self._tag_name == self.tag_member.name
  1363. assert isinstance(self.tag_member.type, QAPISchemaEnumType)
  1364. if self._tag_name: # flat union
  1365. # branches that are not explicitly covered get an empty type
  1366. cases = set([v.name for v in self.variants])
  1367. for m in self.tag_member.type.members:
  1368. if m.name not in cases:
  1369. v = QAPISchemaObjectTypeVariant(m.name, 'q_empty',
  1370. m.ifcond)
  1371. v.set_owner(self.tag_member.owner)
  1372. self.variants.append(v)
  1373. for v in self.variants:
  1374. v.check(schema)
  1375. # Union names must match enum values; alternate names are
  1376. # checked separately. Use 'seen' to tell the two apart.
  1377. if seen:
  1378. assert v.name in self.tag_member.type.member_names()
  1379. assert isinstance(v.type, QAPISchemaObjectType)
  1380. v.type.check(schema)
  1381. def check_clash(self, info, seen):
  1382. for v in self.variants:
  1383. # Reset seen map for each variant, since qapi names from one
  1384. # branch do not affect another branch
  1385. assert isinstance(v.type, QAPISchemaObjectType)
  1386. v.type.check_clash(info, dict(seen))
  1387. class QAPISchemaObjectTypeVariant(QAPISchemaObjectTypeMember):
  1388. role = 'branch'
  1389. def __init__(self, name, typ, ifcond=None):
  1390. QAPISchemaObjectTypeMember.__init__(self, name, typ, False, ifcond)
  1391. class QAPISchemaAlternateType(QAPISchemaType):
  1392. def __init__(self, name, info, doc, ifcond, variants):
  1393. QAPISchemaType.__init__(self, name, info, doc, ifcond)
  1394. assert isinstance(variants, QAPISchemaObjectTypeVariants)
  1395. assert variants.tag_member
  1396. variants.set_owner(name)
  1397. variants.tag_member.set_owner(self.name)
  1398. self.variants = variants
  1399. def check(self, schema):
  1400. QAPISchemaType.check(self, schema)
  1401. self.variants.tag_member.check(schema)
  1402. # Not calling self.variants.check_clash(), because there's nothing
  1403. # to clash with
  1404. self.variants.check(schema, {})
  1405. # Alternate branch names have no relation to the tag enum values;
  1406. # so we have to check for potential name collisions ourselves.
  1407. seen = {}
  1408. for v in self.variants.variants:
  1409. v.check_clash(self.info, seen)
  1410. if self.doc:
  1411. self.doc.connect_member(v)
  1412. if self.doc:
  1413. self.doc.check()
  1414. def c_type(self):
  1415. return c_name(self.name) + pointer_suffix
  1416. def json_type(self):
  1417. return 'value'
  1418. def visit(self, visitor):
  1419. visitor.visit_alternate_type(self.name, self.info, self.ifcond,
  1420. self.variants)
  1421. class QAPISchemaCommand(QAPISchemaEntity):
  1422. def __init__(self, name, info, doc, ifcond, arg_type, ret_type,
  1423. gen, success_response, boxed, allow_oob, allow_preconfig):
  1424. QAPISchemaEntity.__init__(self, name, info, doc, ifcond)
  1425. assert not arg_type or isinstance(arg_type, str)
  1426. assert not ret_type or isinstance(ret_type, str)
  1427. self._arg_type_name = arg_type
  1428. self.arg_type = None
  1429. self._ret_type_name = ret_type
  1430. self.ret_type = None
  1431. self.gen = gen
  1432. self.success_response = success_response
  1433. self.boxed = boxed
  1434. self.allow_oob = allow_oob
  1435. self.allow_preconfig = allow_preconfig
  1436. def check(self, schema):
  1437. QAPISchemaEntity.check(self, schema)
  1438. if self._arg_type_name:
  1439. self.arg_type = schema.lookup_type(self._arg_type_name)
  1440. assert isinstance(self.arg_type, QAPISchemaObjectType)
  1441. self.arg_type.check(schema)
  1442. if self.boxed:
  1443. if self.arg_type.is_empty():
  1444. raise QAPISemError(self.info,
  1445. "Cannot use 'boxed' with empty type")
  1446. else:
  1447. assert not self.arg_type.variants
  1448. elif self.boxed:
  1449. raise QAPISemError(self.info, "Use of 'boxed' requires 'data'")
  1450. if self._ret_type_name:
  1451. self.ret_type = schema.lookup_type(self._ret_type_name)
  1452. assert isinstance(self.ret_type, QAPISchemaType)
  1453. def visit(self, visitor):
  1454. visitor.visit_command(self.name, self.info, self.ifcond,
  1455. self.arg_type, self.ret_type,
  1456. self.gen, self.success_response,
  1457. self.boxed, self.allow_oob,
  1458. self.allow_preconfig)
  1459. class QAPISchemaEvent(QAPISchemaEntity):
  1460. def __init__(self, name, info, doc, ifcond, arg_type, boxed):
  1461. QAPISchemaEntity.__init__(self, name, info, doc, ifcond)
  1462. assert not arg_type or isinstance(arg_type, str)
  1463. self._arg_type_name = arg_type
  1464. self.arg_type = None
  1465. self.boxed = boxed
  1466. def check(self, schema):
  1467. QAPISchemaEntity.check(self, schema)
  1468. if self._arg_type_name:
  1469. self.arg_type = schema.lookup_type(self._arg_type_name)
  1470. assert isinstance(self.arg_type, QAPISchemaObjectType)
  1471. self.arg_type.check(schema)
  1472. if self.boxed:
  1473. if self.arg_type.is_empty():
  1474. raise QAPISemError(self.info,
  1475. "Cannot use 'boxed' with empty type")
  1476. else:
  1477. assert not self.arg_type.variants
  1478. elif self.boxed:
  1479. raise QAPISemError(self.info, "Use of 'boxed' requires 'data'")
  1480. def visit(self, visitor):
  1481. visitor.visit_event(self.name, self.info, self.ifcond,
  1482. self.arg_type, self.boxed)
  1483. class QAPISchema(object):
  1484. def __init__(self, fname):
  1485. self.fname = fname
  1486. if sys.version_info[0] >= 3:
  1487. f = open(fname, 'r', encoding='utf-8')
  1488. else:
  1489. f = open(fname, 'r')
  1490. parser = QAPISchemaParser(f)
  1491. exprs = check_exprs(parser.exprs)
  1492. self.docs = parser.docs
  1493. self._entity_list = []
  1494. self._entity_dict = {}
  1495. self._predefining = True
  1496. self._def_predefineds()
  1497. self._predefining = False
  1498. self._def_exprs(exprs)
  1499. self.check()
  1500. def _def_entity(self, ent):
  1501. # Only the predefined types are allowed to not have info
  1502. assert ent.info or self._predefining
  1503. assert ent.name is None or ent.name not in self._entity_dict
  1504. self._entity_list.append(ent)
  1505. if ent.name is not None:
  1506. self._entity_dict[ent.name] = ent
  1507. def lookup_entity(self, name, typ=None):
  1508. ent = self._entity_dict.get(name)
  1509. if typ and not isinstance(ent, typ):
  1510. return None
  1511. return ent
  1512. def lookup_type(self, name):
  1513. return self.lookup_entity(name, QAPISchemaType)
  1514. def _def_include(self, expr, info, doc):
  1515. include = expr['include']
  1516. assert doc is None
  1517. main_info = info
  1518. while main_info['parent']:
  1519. main_info = main_info['parent']
  1520. fname = os.path.relpath(include, os.path.dirname(main_info['file']))
  1521. self._def_entity(QAPISchemaInclude(fname, info))
  1522. def _def_builtin_type(self, name, json_type, c_type):
  1523. self._def_entity(QAPISchemaBuiltinType(name, json_type, c_type))
  1524. # Instantiating only the arrays that are actually used would
  1525. # be nice, but we can't as long as their generated code
  1526. # (qapi-builtin-types.[ch]) may be shared by some other
  1527. # schema.
  1528. self._make_array_type(name, None)
  1529. def _def_predefineds(self):
  1530. for t in [('str', 'string', 'char' + pointer_suffix),
  1531. ('number', 'number', 'double'),
  1532. ('int', 'int', 'int64_t'),
  1533. ('int8', 'int', 'int8_t'),
  1534. ('int16', 'int', 'int16_t'),
  1535. ('int32', 'int', 'int32_t'),
  1536. ('int64', 'int', 'int64_t'),
  1537. ('uint8', 'int', 'uint8_t'),
  1538. ('uint16', 'int', 'uint16_t'),
  1539. ('uint32', 'int', 'uint32_t'),
  1540. ('uint64', 'int', 'uint64_t'),
  1541. ('size', 'int', 'uint64_t'),
  1542. ('bool', 'boolean', 'bool'),
  1543. ('any', 'value', 'QObject' + pointer_suffix),
  1544. ('null', 'null', 'QNull' + pointer_suffix)]:
  1545. self._def_builtin_type(*t)
  1546. self.the_empty_object_type = QAPISchemaObjectType(
  1547. 'q_empty', None, None, None, None, [], None, [])
  1548. self._def_entity(self.the_empty_object_type)
  1549. qtypes = ['none', 'qnull', 'qnum', 'qstring', 'qdict', 'qlist',
  1550. 'qbool']
  1551. qtype_values = self._make_enum_members([{'name': n} for n in qtypes])
  1552. self._def_entity(QAPISchemaEnumType('QType', None, None, None,
  1553. qtype_values, 'QTYPE'))
  1554. def _make_features(self, features):
  1555. return [QAPISchemaFeature(f['name'], f.get('if')) for f in features]
  1556. def _make_enum_members(self, values):
  1557. return [QAPISchemaMember(v['name'], v.get('if')) for v in values]
  1558. def _make_implicit_enum_type(self, name, info, ifcond, values):
  1559. # See also QAPISchemaObjectTypeMember._pretty_owner()
  1560. name = name + 'Kind' # Use namespace reserved by add_name()
  1561. self._def_entity(QAPISchemaEnumType(
  1562. name, info, None, ifcond, self._make_enum_members(values), None))
  1563. return name
  1564. def _make_array_type(self, element_type, info):
  1565. name = element_type + 'List' # Use namespace reserved by add_name()
  1566. if not self.lookup_type(name):
  1567. self._def_entity(QAPISchemaArrayType(name, info, element_type))
  1568. return name
  1569. def _make_implicit_object_type(self, name, info, doc, ifcond,
  1570. role, members):
  1571. if not members:
  1572. return None
  1573. # See also QAPISchemaObjectTypeMember._pretty_owner()
  1574. name = 'q_obj_%s-%s' % (name, role)
  1575. typ = self.lookup_entity(name, QAPISchemaObjectType)
  1576. if typ:
  1577. # The implicit object type has multiple users. This can
  1578. # happen only for simple unions' implicit wrapper types.
  1579. # Its ifcond should be the disjunction of its user's
  1580. # ifconds. Not implemented. Instead, we always pass the
  1581. # wrapped type's ifcond, which is trivially the same for all
  1582. # users. It's also necessary for the wrapper to compile.
  1583. # But it's not tight: the disjunction need not imply it. We
  1584. # may end up compiling useless wrapper types.
  1585. # TODO kill simple unions or implement the disjunction
  1586. assert ifcond == typ._ifcond # pylint: disable=protected-access
  1587. else:
  1588. self._def_entity(QAPISchemaObjectType(name, info, doc, ifcond,
  1589. None, members, None, []))
  1590. return name
  1591. def _def_enum_type(self, expr, info, doc):
  1592. name = expr['enum']
  1593. data = expr['data']
  1594. prefix = expr.get('prefix')
  1595. ifcond = expr.get('if')
  1596. self._def_entity(QAPISchemaEnumType(
  1597. name, info, doc, ifcond,
  1598. self._make_enum_members(data), prefix))
  1599. def _make_member(self, name, typ, ifcond, info):
  1600. optional = False
  1601. if name.startswith('*'):
  1602. name = name[1:]
  1603. optional = True
  1604. if isinstance(typ, list):
  1605. assert len(typ) == 1
  1606. typ = self._make_array_type(typ[0], info)
  1607. return QAPISchemaObjectTypeMember(name, typ, optional, ifcond)
  1608. def _make_members(self, data, info):
  1609. return [self._make_member(key, value['type'], value.get('if'), info)
  1610. for (key, value) in data.items()]
  1611. def _def_struct_type(self, expr, info, doc):
  1612. name = expr['struct']
  1613. base = expr.get('base')
  1614. data = expr['data']
  1615. ifcond = expr.get('if')
  1616. features = expr.get('features', [])
  1617. self._def_entity(QAPISchemaObjectType(name, info, doc, ifcond, base,
  1618. self._make_members(data, info),
  1619. None,
  1620. self._make_features(features)))
  1621. def _make_variant(self, case, typ, ifcond):
  1622. return QAPISchemaObjectTypeVariant(case, typ, ifcond)
  1623. def _make_simple_variant(self, case, typ, ifcond, info):
  1624. if isinstance(typ, list):
  1625. assert len(typ) == 1
  1626. typ = self._make_array_type(typ[0], info)
  1627. typ = self._make_implicit_object_type(
  1628. typ, info, None, self.lookup_type(typ),
  1629. 'wrapper', [self._make_member('data', typ, None, info)])
  1630. return QAPISchemaObjectTypeVariant(case, typ, ifcond)
  1631. def _def_union_type(self, expr, info, doc):
  1632. name = expr['union']
  1633. data = expr['data']
  1634. base = expr.get('base')
  1635. ifcond = expr.get('if')
  1636. tag_name = expr.get('discriminator')
  1637. tag_member = None
  1638. if isinstance(base, dict):
  1639. base = self._make_implicit_object_type(
  1640. name, info, doc, ifcond,
  1641. 'base', self._make_members(base, info))
  1642. if tag_name:
  1643. variants = [self._make_variant(key, value['type'], value.get('if'))
  1644. for (key, value) in data.items()]
  1645. members = []
  1646. else:
  1647. variants = [self._make_simple_variant(key, value['type'],
  1648. value.get('if'), info)
  1649. for (key, value) in data.items()]
  1650. enum = [{'name': v.name, 'if': v.ifcond} for v in variants]
  1651. typ = self._make_implicit_enum_type(name, info, ifcond, enum)
  1652. tag_member = QAPISchemaObjectTypeMember('type', typ, False)
  1653. members = [tag_member]
  1654. self._def_entity(
  1655. QAPISchemaObjectType(name, info, doc, ifcond, base, members,
  1656. QAPISchemaObjectTypeVariants(tag_name,
  1657. tag_member,
  1658. variants), []))
  1659. def _def_alternate_type(self, expr, info, doc):
  1660. name = expr['alternate']
  1661. data = expr['data']
  1662. ifcond = expr.get('if')
  1663. variants = [self._make_variant(key, value['type'], value.get('if'))
  1664. for (key, value) in data.items()]
  1665. tag_member = QAPISchemaObjectTypeMember('type', 'QType', False)
  1666. self._def_entity(
  1667. QAPISchemaAlternateType(name, info, doc, ifcond,
  1668. QAPISchemaObjectTypeVariants(None,
  1669. tag_member,
  1670. variants)))
  1671. def _def_command(self, expr, info, doc):
  1672. name = expr['command']
  1673. data = expr.get('data')
  1674. rets = expr.get('returns')
  1675. gen = expr.get('gen', True)
  1676. success_response = expr.get('success-response', True)
  1677. boxed = expr.get('boxed', False)
  1678. allow_oob = expr.get('allow-oob', False)
  1679. allow_preconfig = expr.get('allow-preconfig', False)
  1680. ifcond = expr.get('if')
  1681. if isinstance(data, OrderedDict):
  1682. data = self._make_implicit_object_type(
  1683. name, info, doc, ifcond, 'arg', self._make_members(data, info))
  1684. if isinstance(rets, list):
  1685. assert len(rets) == 1
  1686. rets = self._make_array_type(rets[0], info)
  1687. self._def_entity(QAPISchemaCommand(name, info, doc, ifcond, data, rets,
  1688. gen, success_response,
  1689. boxed, allow_oob, allow_preconfig))
  1690. def _def_event(self, expr, info, doc):
  1691. name = expr['event']
  1692. data = expr.get('data')
  1693. boxed = expr.get('boxed', False)
  1694. ifcond = expr.get('if')
  1695. if isinstance(data, OrderedDict):
  1696. data = self._make_implicit_object_type(
  1697. name, info, doc, ifcond, 'arg', self._make_members(data, info))
  1698. self._def_entity(QAPISchemaEvent(name, info, doc, ifcond, data, boxed))
  1699. def _def_exprs(self, exprs):
  1700. for expr_elem in exprs:
  1701. expr = expr_elem['expr']
  1702. info = expr_elem['info']
  1703. doc = expr_elem.get('doc')
  1704. if 'enum' in expr:
  1705. self._def_enum_type(expr, info, doc)
  1706. elif 'struct' in expr:
  1707. self._def_struct_type(expr, info, doc)
  1708. elif 'union' in expr:
  1709. self._def_union_type(expr, info, doc)
  1710. elif 'alternate' in expr:
  1711. self._def_alternate_type(expr, info, doc)
  1712. elif 'command' in expr:
  1713. self._def_command(expr, info, doc)
  1714. elif 'event' in expr:
  1715. self._def_event(expr, info, doc)
  1716. elif 'include' in expr:
  1717. self._def_include(expr, info, doc)
  1718. else:
  1719. assert False
  1720. def check(self):
  1721. for ent in self._entity_list:
  1722. ent.check(self)
  1723. def visit(self, visitor):
  1724. visitor.visit_begin(self)
  1725. module = None
  1726. visitor.visit_module(module)
  1727. for entity in self._entity_list:
  1728. if visitor.visit_needed(entity):
  1729. if entity.module != module:
  1730. module = entity.module
  1731. visitor.visit_module(module)
  1732. entity.visit(visitor)
  1733. visitor.visit_end()
  1734. #
  1735. # Code generation helpers
  1736. #
  1737. def camel_case(name):
  1738. new_name = ''
  1739. first = True
  1740. for ch in name:
  1741. if ch in ['_', '-']:
  1742. first = True
  1743. elif first:
  1744. new_name += ch.upper()
  1745. first = False
  1746. else:
  1747. new_name += ch.lower()
  1748. return new_name
  1749. # ENUMName -> ENUM_NAME, EnumName1 -> ENUM_NAME1
  1750. # ENUM_NAME -> ENUM_NAME, ENUM_NAME1 -> ENUM_NAME1, ENUM_Name2 -> ENUM_NAME2
  1751. # ENUM24_Name -> ENUM24_NAME
  1752. def camel_to_upper(value):
  1753. c_fun_str = c_name(value, False)
  1754. if value.isupper():
  1755. return c_fun_str
  1756. new_name = ''
  1757. length = len(c_fun_str)
  1758. for i in range(length):
  1759. c = c_fun_str[i]
  1760. # When c is upper and no '_' appears before, do more checks
  1761. if c.isupper() and (i > 0) and c_fun_str[i - 1] != '_':
  1762. if i < length - 1 and c_fun_str[i + 1].islower():
  1763. new_name += '_'
  1764. elif c_fun_str[i - 1].isdigit():
  1765. new_name += '_'
  1766. new_name += c
  1767. return new_name.lstrip('_').upper()
  1768. def c_enum_const(type_name, const_name, prefix=None):
  1769. if prefix is not None:
  1770. type_name = prefix
  1771. return camel_to_upper(type_name) + '_' + c_name(const_name, False).upper()
  1772. if hasattr(str, 'maketrans'):
  1773. c_name_trans = str.maketrans('.-', '__')
  1774. else:
  1775. c_name_trans = string.maketrans('.-', '__')
  1776. # Map @name to a valid C identifier.
  1777. # If @protect, avoid returning certain ticklish identifiers (like
  1778. # C keywords) by prepending 'q_'.
  1779. #
  1780. # Used for converting 'name' from a 'name':'type' qapi definition
  1781. # into a generated struct member, as well as converting type names
  1782. # into substrings of a generated C function name.
  1783. # '__a.b_c' -> '__a_b_c', 'x-foo' -> 'x_foo'
  1784. # protect=True: 'int' -> 'q_int'; protect=False: 'int' -> 'int'
  1785. def c_name(name, protect=True):
  1786. # ANSI X3J11/88-090, 3.1.1
  1787. c89_words = set(['auto', 'break', 'case', 'char', 'const', 'continue',
  1788. 'default', 'do', 'double', 'else', 'enum', 'extern',
  1789. 'float', 'for', 'goto', 'if', 'int', 'long', 'register',
  1790. 'return', 'short', 'signed', 'sizeof', 'static',
  1791. 'struct', 'switch', 'typedef', 'union', 'unsigned',
  1792. 'void', 'volatile', 'while'])
  1793. # ISO/IEC 9899:1999, 6.4.1
  1794. c99_words = set(['inline', 'restrict', '_Bool', '_Complex', '_Imaginary'])
  1795. # ISO/IEC 9899:2011, 6.4.1
  1796. c11_words = set(['_Alignas', '_Alignof', '_Atomic', '_Generic',
  1797. '_Noreturn', '_Static_assert', '_Thread_local'])
  1798. # GCC http://gcc.gnu.org/onlinedocs/gcc-4.7.1/gcc/C-Extensions.html
  1799. # excluding _.*
  1800. gcc_words = set(['asm', 'typeof'])
  1801. # C++ ISO/IEC 14882:2003 2.11
  1802. cpp_words = set(['bool', 'catch', 'class', 'const_cast', 'delete',
  1803. 'dynamic_cast', 'explicit', 'false', 'friend', 'mutable',
  1804. 'namespace', 'new', 'operator', 'private', 'protected',
  1805. 'public', 'reinterpret_cast', 'static_cast', 'template',
  1806. 'this', 'throw', 'true', 'try', 'typeid', 'typename',
  1807. 'using', 'virtual', 'wchar_t',
  1808. # alternative representations
  1809. 'and', 'and_eq', 'bitand', 'bitor', 'compl', 'not',
  1810. 'not_eq', 'or', 'or_eq', 'xor', 'xor_eq'])
  1811. # namespace pollution:
  1812. polluted_words = set(['unix', 'errno', 'mips', 'sparc', 'i386'])
  1813. name = name.translate(c_name_trans)
  1814. if protect and (name in c89_words | c99_words | c11_words | gcc_words
  1815. | cpp_words | polluted_words):
  1816. return 'q_' + name
  1817. return name
  1818. eatspace = '\033EATSPACE.'
  1819. pointer_suffix = ' *' + eatspace
  1820. def genindent(count):
  1821. ret = ''
  1822. for _ in range(count):
  1823. ret += ' '
  1824. return ret
  1825. indent_level = 0
  1826. def push_indent(indent_amount=4):
  1827. global indent_level
  1828. indent_level += indent_amount
  1829. def pop_indent(indent_amount=4):
  1830. global indent_level
  1831. indent_level -= indent_amount
  1832. # Generate @code with @kwds interpolated.
  1833. # Obey indent_level, and strip eatspace.
  1834. def cgen(code, **kwds):
  1835. raw = code % kwds
  1836. if indent_level:
  1837. indent = genindent(indent_level)
  1838. # re.subn() lacks flags support before Python 2.7, use re.compile()
  1839. raw = re.subn(re.compile(r'^(?!(#|$))', re.MULTILINE),
  1840. indent, raw)
  1841. raw = raw[0]
  1842. return re.sub(re.escape(eatspace) + r' *', '', raw)
  1843. def mcgen(code, **kwds):
  1844. if code[0] == '\n':
  1845. code = code[1:]
  1846. return cgen(code, **kwds)
  1847. def c_fname(filename):
  1848. return re.sub(r'[^A-Za-z0-9_]', '_', filename)
  1849. def guardstart(name):
  1850. return mcgen('''
  1851. #ifndef %(name)s
  1852. #define %(name)s
  1853. ''',
  1854. name=c_fname(name).upper())
  1855. def guardend(name):
  1856. return mcgen('''
  1857. #endif /* %(name)s */
  1858. ''',
  1859. name=c_fname(name).upper())
  1860. def gen_if(ifcond):
  1861. ret = ''
  1862. for ifc in ifcond:
  1863. ret += mcgen('''
  1864. #if %(cond)s
  1865. ''', cond=ifc)
  1866. return ret
  1867. def gen_endif(ifcond):
  1868. ret = ''
  1869. for ifc in reversed(ifcond):
  1870. ret += mcgen('''
  1871. #endif /* %(cond)s */
  1872. ''', cond=ifc)
  1873. return ret
  1874. def _wrap_ifcond(ifcond, before, after):
  1875. if before == after:
  1876. return after # suppress empty #if ... #endif
  1877. assert after.startswith(before)
  1878. out = before
  1879. added = after[len(before):]
  1880. if added[0] == '\n':
  1881. out += '\n'
  1882. added = added[1:]
  1883. out += gen_if(ifcond)
  1884. out += added
  1885. out += gen_endif(ifcond)
  1886. return out
  1887. def gen_enum_lookup(name, members, prefix=None):
  1888. ret = mcgen('''
  1889. const QEnumLookup %(c_name)s_lookup = {
  1890. .array = (const char *const[]) {
  1891. ''',
  1892. c_name=c_name(name))
  1893. for m in members:
  1894. ret += gen_if(m.ifcond)
  1895. index = c_enum_const(name, m.name, prefix)
  1896. ret += mcgen('''
  1897. [%(index)s] = "%(name)s",
  1898. ''',
  1899. index=index, name=m.name)
  1900. ret += gen_endif(m.ifcond)
  1901. ret += mcgen('''
  1902. },
  1903. .size = %(max_index)s
  1904. };
  1905. ''',
  1906. max_index=c_enum_const(name, '_MAX', prefix))
  1907. return ret
  1908. def gen_enum(name, members, prefix=None):
  1909. # append automatically generated _MAX value
  1910. enum_members = members + [QAPISchemaMember('_MAX')]
  1911. ret = mcgen('''
  1912. typedef enum %(c_name)s {
  1913. ''',
  1914. c_name=c_name(name))
  1915. for m in enum_members:
  1916. ret += gen_if(m.ifcond)
  1917. ret += mcgen('''
  1918. %(c_enum)s,
  1919. ''',
  1920. c_enum=c_enum_const(name, m.name, prefix))
  1921. ret += gen_endif(m.ifcond)
  1922. ret += mcgen('''
  1923. } %(c_name)s;
  1924. ''',
  1925. c_name=c_name(name))
  1926. ret += mcgen('''
  1927. #define %(c_name)s_str(val) \\
  1928. qapi_enum_lookup(&%(c_name)s_lookup, (val))
  1929. extern const QEnumLookup %(c_name)s_lookup;
  1930. ''',
  1931. c_name=c_name(name))
  1932. return ret
  1933. def build_params(arg_type, boxed, extra=None):
  1934. ret = ''
  1935. sep = ''
  1936. if boxed:
  1937. assert arg_type
  1938. ret += '%s arg' % arg_type.c_param_type()
  1939. sep = ', '
  1940. elif arg_type:
  1941. assert not arg_type.variants
  1942. for memb in arg_type.members:
  1943. ret += sep
  1944. sep = ', '
  1945. if memb.optional:
  1946. ret += 'bool has_%s, ' % c_name(memb.name)
  1947. ret += '%s %s' % (memb.type.c_param_type(),
  1948. c_name(memb.name))
  1949. if extra:
  1950. ret += sep + extra
  1951. return ret if ret else 'void'
  1952. #
  1953. # Accumulate and write output
  1954. #
  1955. class QAPIGen(object):
  1956. def __init__(self, fname):
  1957. self.fname = fname
  1958. self._preamble = ''
  1959. self._body = ''
  1960. def preamble_add(self, text):
  1961. self._preamble += text
  1962. def add(self, text):
  1963. self._body += text
  1964. def get_content(self):
  1965. return self._top() + self._preamble + self._body + self._bottom()
  1966. def _top(self):
  1967. return ''
  1968. def _bottom(self):
  1969. return ''
  1970. def write(self, output_dir):
  1971. pathname = os.path.join(output_dir, self.fname)
  1972. dir = os.path.dirname(pathname)
  1973. if dir:
  1974. try:
  1975. os.makedirs(dir)
  1976. except os.error as e:
  1977. if e.errno != errno.EEXIST:
  1978. raise
  1979. fd = os.open(pathname, os.O_RDWR | os.O_CREAT, 0o666)
  1980. if sys.version_info[0] >= 3:
  1981. f = open(fd, 'r+', encoding='utf-8')
  1982. else:
  1983. f = os.fdopen(fd, 'r+')
  1984. text = self.get_content()
  1985. oldtext = f.read(len(text) + 1)
  1986. if text != oldtext:
  1987. f.seek(0)
  1988. f.truncate(0)
  1989. f.write(text)
  1990. f.close()
  1991. @contextmanager
  1992. def ifcontext(ifcond, *args):
  1993. """A 'with' statement context manager to wrap with start_if()/end_if()
  1994. *args: any number of QAPIGenCCode
  1995. Example::
  1996. with ifcontext(ifcond, self._genh, self._genc):
  1997. modify self._genh and self._genc ...
  1998. Is equivalent to calling::
  1999. self._genh.start_if(ifcond)
  2000. self._genc.start_if(ifcond)
  2001. modify self._genh and self._genc ...
  2002. self._genh.end_if()
  2003. self._genc.end_if()
  2004. """
  2005. for arg in args:
  2006. arg.start_if(ifcond)
  2007. yield
  2008. for arg in args:
  2009. arg.end_if()
  2010. class QAPIGenCCode(QAPIGen):
  2011. def __init__(self, fname):
  2012. QAPIGen.__init__(self, fname)
  2013. self._start_if = None
  2014. def start_if(self, ifcond):
  2015. assert self._start_if is None
  2016. self._start_if = (ifcond, self._body, self._preamble)
  2017. def end_if(self):
  2018. assert self._start_if
  2019. self._wrap_ifcond()
  2020. self._start_if = None
  2021. def _wrap_ifcond(self):
  2022. self._body = _wrap_ifcond(self._start_if[0],
  2023. self._start_if[1], self._body)
  2024. self._preamble = _wrap_ifcond(self._start_if[0],
  2025. self._start_if[2], self._preamble)
  2026. def get_content(self):
  2027. assert self._start_if is None
  2028. return QAPIGen.get_content(self)
  2029. class QAPIGenC(QAPIGenCCode):
  2030. def __init__(self, fname, blurb, pydoc):
  2031. QAPIGenCCode.__init__(self, fname)
  2032. self._blurb = blurb
  2033. self._copyright = '\n * '.join(re.findall(r'^Copyright .*', pydoc,
  2034. re.MULTILINE))
  2035. def _top(self):
  2036. return mcgen('''
  2037. /* AUTOMATICALLY GENERATED, DO NOT MODIFY */
  2038. /*
  2039. %(blurb)s
  2040. *
  2041. * %(copyright)s
  2042. *
  2043. * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
  2044. * See the COPYING.LIB file in the top-level directory.
  2045. */
  2046. ''',
  2047. blurb=self._blurb, copyright=self._copyright)
  2048. def _bottom(self):
  2049. return mcgen('''
  2050. /* Dummy declaration to prevent empty .o file */
  2051. char qapi_dummy_%(name)s;
  2052. ''',
  2053. name=c_fname(self.fname))
  2054. class QAPIGenH(QAPIGenC):
  2055. def _top(self):
  2056. return QAPIGenC._top(self) + guardstart(self.fname)
  2057. def _bottom(self):
  2058. return guardend(self.fname)
  2059. class QAPIGenDoc(QAPIGen):
  2060. def _top(self):
  2061. return (QAPIGen._top(self)
  2062. + '@c AUTOMATICALLY GENERATED, DO NOT MODIFY\n\n')
  2063. class QAPISchemaMonolithicCVisitor(QAPISchemaVisitor):
  2064. def __init__(self, prefix, what, blurb, pydoc):
  2065. self._prefix = prefix
  2066. self._what = what
  2067. self._genc = QAPIGenC(self._prefix + self._what + '.c',
  2068. blurb, pydoc)
  2069. self._genh = QAPIGenH(self._prefix + self._what + '.h',
  2070. blurb, pydoc)
  2071. def write(self, output_dir):
  2072. self._genc.write(output_dir)
  2073. self._genh.write(output_dir)
  2074. class QAPISchemaModularCVisitor(QAPISchemaVisitor):
  2075. def __init__(self, prefix, what, blurb, pydoc):
  2076. self._prefix = prefix
  2077. self._what = what
  2078. self._blurb = blurb
  2079. self._pydoc = pydoc
  2080. self._genc = None
  2081. self._genh = None
  2082. self._module = {}
  2083. self._main_module = None
  2084. @staticmethod
  2085. def _is_user_module(name):
  2086. return name and not name.startswith('./')
  2087. @staticmethod
  2088. def _is_builtin_module(name):
  2089. return not name
  2090. def _module_dirname(self, what, name):
  2091. if self._is_user_module(name):
  2092. return os.path.dirname(name)
  2093. return ''
  2094. def _module_basename(self, what, name):
  2095. ret = '' if self._is_builtin_module(name) else self._prefix
  2096. if self._is_user_module(name):
  2097. basename = os.path.basename(name)
  2098. ret += what
  2099. if name != self._main_module:
  2100. ret += '-' + os.path.splitext(basename)[0]
  2101. else:
  2102. name = name[2:] if name else 'builtin'
  2103. ret += re.sub(r'-', '-' + name + '-', what)
  2104. return ret
  2105. def _module_filename(self, what, name):
  2106. return os.path.join(self._module_dirname(what, name),
  2107. self._module_basename(what, name))
  2108. def _add_module(self, name, blurb):
  2109. basename = self._module_filename(self._what, name)
  2110. genc = QAPIGenC(basename + '.c', blurb, self._pydoc)
  2111. genh = QAPIGenH(basename + '.h', blurb, self._pydoc)
  2112. self._module[name] = (genc, genh)
  2113. self._set_module(name)
  2114. def _add_user_module(self, name, blurb):
  2115. assert self._is_user_module(name)
  2116. if self._main_module is None:
  2117. self._main_module = name
  2118. self._add_module(name, blurb)
  2119. def _add_system_module(self, name, blurb):
  2120. self._add_module(name and './' + name, blurb)
  2121. def _set_module(self, name):
  2122. self._genc, self._genh = self._module[name]
  2123. def write(self, output_dir, opt_builtins=False):
  2124. for name in self._module:
  2125. if self._is_builtin_module(name) and not opt_builtins:
  2126. continue
  2127. (genc, genh) = self._module[name]
  2128. genc.write(output_dir)
  2129. genh.write(output_dir)
  2130. def _begin_user_module(self, name):
  2131. pass
  2132. def visit_module(self, name):
  2133. if name in self._module:
  2134. self._set_module(name)
  2135. elif self._is_builtin_module(name):
  2136. # The built-in module has not been created. No code may
  2137. # be generated.
  2138. self._genc = None
  2139. self._genh = None
  2140. else:
  2141. self._add_user_module(name, self._blurb)
  2142. self._begin_user_module(name)
  2143. def visit_include(self, name, info):
  2144. relname = os.path.relpath(self._module_filename(self._what, name),
  2145. os.path.dirname(self._genh.fname))
  2146. self._genh.preamble_add(mcgen('''
  2147. #include "%(relname)s.h"
  2148. ''',
  2149. relname=relname))