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