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 assert not isinstance(self.element_type, QAPISchemaArrayType) 1366 1367 @property 1368 def ifcond(self): 1369 assert self._checked 1370 return self.element_type.ifcond 1371 1372 @property 1373 def module(self): 1374 assert self._checked 1375 return self.element_type.module 1376 1377 def is_implicit(self): 1378 return True 1379 1380 def c_type(self): 1381 return c_name(self.name) + pointer_suffix 1382 1383 def json_type(self): 1384 return 'array' 1385 1386 def doc_type(self): 1387 elt_doc_type = self.element_type.doc_type() 1388 if not elt_doc_type: 1389 return None 1390 return 'array of ' + elt_doc_type 1391 1392 def visit(self, visitor): 1393 QAPISchemaType.visit(self, visitor) 1394 visitor.visit_array_type(self.name, self.info, self.ifcond, 1395 self.element_type) 1396 1397 1398class QAPISchemaObjectType(QAPISchemaType): 1399 def __init__(self, name, info, doc, ifcond, 1400 base, local_members, variants, features): 1401 # struct has local_members, optional base, and no variants 1402 # flat union has base, variants, and no local_members 1403 # simple union has local_members, variants, and no base 1404 QAPISchemaType.__init__(self, name, info, doc, ifcond) 1405 assert base is None or isinstance(base, str) 1406 for m in local_members: 1407 assert isinstance(m, QAPISchemaObjectTypeMember) 1408 m.set_owner(name) 1409 if variants is not None: 1410 assert isinstance(variants, QAPISchemaObjectTypeVariants) 1411 variants.set_owner(name) 1412 for f in features: 1413 assert isinstance(f, QAPISchemaFeature) 1414 f.set_owner(name) 1415 self._base_name = base 1416 self.base = None 1417 self.local_members = local_members 1418 self.variants = variants 1419 self.members = None 1420 self.features = features 1421 1422 def check(self, schema): 1423 # This calls another type T's .check() exactly when the C 1424 # struct emitted by gen_object() contains that T's C struct 1425 # (pointers don't count). 1426 if self.members is not None: 1427 # A previous .check() completed: nothing to do 1428 return 1429 if self._checked: 1430 # Recursed: C struct contains itself 1431 raise QAPISemError(self.info, 1432 "Object %s contains itself" % self.name) 1433 1434 QAPISchemaType.check(self, schema) 1435 assert self._checked and self.members is None 1436 1437 seen = OrderedDict() 1438 if self._base_name: 1439 self.base = schema.lookup_type(self._base_name) 1440 assert isinstance(self.base, QAPISchemaObjectType) 1441 self.base.check(schema) 1442 self.base.check_clash(self.info, seen) 1443 for m in self.local_members: 1444 m.check(schema) 1445 m.check_clash(self.info, seen) 1446 if self.doc: 1447 self.doc.connect_member(m) 1448 members = seen.values() 1449 1450 if self.variants: 1451 self.variants.check(schema, seen) 1452 assert self.variants.tag_member in members 1453 self.variants.check_clash(self.info, seen) 1454 1455 # Features are in a name space separate from members 1456 seen = {} 1457 for f in self.features: 1458 f.check_clash(self.info, seen) 1459 1460 if self.doc: 1461 self.doc.check() 1462 1463 self.members = members # mark completed 1464 1465 # Check that the members of this type do not cause duplicate JSON members, 1466 # and update seen to track the members seen so far. Report any errors 1467 # on behalf of info, which is not necessarily self.info 1468 def check_clash(self, info, seen): 1469 assert self._checked 1470 assert not self.variants # not implemented 1471 for m in self.members: 1472 m.check_clash(info, seen) 1473 1474 @property 1475 def ifcond(self): 1476 assert self._checked 1477 if isinstance(self._ifcond, QAPISchemaType): 1478 # Simple union wrapper type inherits from wrapped type; 1479 # see _make_implicit_object_type() 1480 return self._ifcond.ifcond 1481 return self._ifcond 1482 1483 def is_implicit(self): 1484 # See QAPISchema._make_implicit_object_type(), as well as 1485 # _def_predefineds() 1486 return self.name.startswith('q_') 1487 1488 def is_empty(self): 1489 assert self.members is not None 1490 return not self.members and not self.variants 1491 1492 def c_name(self): 1493 assert self.name != 'q_empty' 1494 return QAPISchemaType.c_name(self) 1495 1496 def c_type(self): 1497 assert not self.is_implicit() 1498 return c_name(self.name) + pointer_suffix 1499 1500 def c_unboxed_type(self): 1501 return c_name(self.name) 1502 1503 def json_type(self): 1504 return 'object' 1505 1506 def visit(self, visitor): 1507 QAPISchemaType.visit(self, visitor) 1508 visitor.visit_object_type(self.name, self.info, self.ifcond, 1509 self.base, self.local_members, self.variants, 1510 self.features) 1511 visitor.visit_object_type_flat(self.name, self.info, self.ifcond, 1512 self.members, self.variants, 1513 self.features) 1514 1515 1516class QAPISchemaMember(object): 1517 """ Represents object members, enum members and features """ 1518 role = 'member' 1519 1520 def __init__(self, name, ifcond=None): 1521 assert isinstance(name, str) 1522 self.name = name 1523 self.ifcond = ifcond or [] 1524 self.owner = None 1525 1526 def set_owner(self, name): 1527 assert not self.owner 1528 self.owner = name 1529 1530 def check_clash(self, info, seen): 1531 cname = c_name(self.name) 1532 if cname.lower() != cname and self.owner not in name_case_whitelist: 1533 raise QAPISemError(info, 1534 "%s should not use uppercase" % self.describe()) 1535 if cname in seen: 1536 raise QAPISemError(info, "%s collides with %s" % 1537 (self.describe(), seen[cname].describe())) 1538 seen[cname] = self 1539 1540 def _pretty_owner(self): 1541 owner = self.owner 1542 if owner.startswith('q_obj_'): 1543 # See QAPISchema._make_implicit_object_type() - reverse the 1544 # mapping there to create a nice human-readable description 1545 owner = owner[6:] 1546 if owner.endswith('-arg'): 1547 return '(parameter of %s)' % owner[:-4] 1548 elif owner.endswith('-base'): 1549 return '(base of %s)' % owner[:-5] 1550 else: 1551 assert owner.endswith('-wrapper') 1552 # Unreachable and not implemented 1553 assert False 1554 if owner.endswith('Kind'): 1555 # See QAPISchema._make_implicit_enum_type() 1556 return '(branch of %s)' % owner[:-4] 1557 return '(%s of %s)' % (self.role, owner) 1558 1559 def describe(self): 1560 return "'%s' %s" % (self.name, self._pretty_owner()) 1561 1562 1563class QAPISchemaEnumMember(QAPISchemaMember): 1564 role = 'value' 1565 1566 1567class QAPISchemaFeature(QAPISchemaMember): 1568 role = 'feature' 1569 1570 1571class QAPISchemaObjectTypeMember(QAPISchemaMember): 1572 def __init__(self, name, typ, optional, ifcond=None): 1573 QAPISchemaMember.__init__(self, name, ifcond) 1574 assert isinstance(typ, str) 1575 assert isinstance(optional, bool) 1576 self._type_name = typ 1577 self.type = None 1578 self.optional = optional 1579 1580 def check(self, schema): 1581 assert self.owner 1582 self.type = schema.lookup_type(self._type_name) 1583 assert self.type 1584 1585 1586class QAPISchemaObjectTypeVariants(object): 1587 def __init__(self, tag_name, tag_member, variants): 1588 # Flat unions pass tag_name but not tag_member. 1589 # Simple unions and alternates pass tag_member but not tag_name. 1590 # After check(), tag_member is always set, and tag_name remains 1591 # a reliable witness of being used by a flat union. 1592 assert bool(tag_member) != bool(tag_name) 1593 assert (isinstance(tag_name, str) or 1594 isinstance(tag_member, QAPISchemaObjectTypeMember)) 1595 for v in variants: 1596 assert isinstance(v, QAPISchemaObjectTypeVariant) 1597 self._tag_name = tag_name 1598 self.tag_member = tag_member 1599 self.variants = variants 1600 1601 def set_owner(self, name): 1602 for v in self.variants: 1603 v.set_owner(name) 1604 1605 def check(self, schema, seen): 1606 if not self.tag_member: # flat union 1607 self.tag_member = seen[c_name(self._tag_name)] 1608 assert self._tag_name == self.tag_member.name 1609 assert isinstance(self.tag_member.type, QAPISchemaEnumType) 1610 assert not self.tag_member.optional 1611 assert self.tag_member.ifcond == [] 1612 if self._tag_name: # flat union 1613 # branches that are not explicitly covered get an empty type 1614 cases = set([v.name for v in self.variants]) 1615 for m in self.tag_member.type.members: 1616 if m.name not in cases: 1617 v = QAPISchemaObjectTypeVariant(m.name, 'q_empty', 1618 m.ifcond) 1619 v.set_owner(self.tag_member.owner) 1620 self.variants.append(v) 1621 assert self.variants 1622 for v in self.variants: 1623 v.check(schema) 1624 # Union names must match enum values; alternate names are 1625 # checked separately. Use 'seen' to tell the two apart. 1626 if seen: 1627 assert v.name in self.tag_member.type.member_names() 1628 assert (isinstance(v.type, QAPISchemaObjectType) 1629 and not v.type.variants) 1630 v.type.check(schema) 1631 1632 def check_clash(self, info, seen): 1633 for v in self.variants: 1634 # Reset seen map for each variant, since qapi names from one 1635 # branch do not affect another branch 1636 v.type.check_clash(info, dict(seen)) 1637 1638 1639class QAPISchemaObjectTypeVariant(QAPISchemaObjectTypeMember): 1640 role = 'branch' 1641 1642 def __init__(self, name, typ, ifcond=None): 1643 QAPISchemaObjectTypeMember.__init__(self, name, typ, False, ifcond) 1644 1645 1646class QAPISchemaAlternateType(QAPISchemaType): 1647 def __init__(self, name, info, doc, ifcond, variants): 1648 QAPISchemaType.__init__(self, name, info, doc, ifcond) 1649 assert isinstance(variants, QAPISchemaObjectTypeVariants) 1650 assert variants.tag_member 1651 variants.set_owner(name) 1652 variants.tag_member.set_owner(self.name) 1653 self.variants = variants 1654 1655 def check(self, schema): 1656 QAPISchemaType.check(self, schema) 1657 self.variants.tag_member.check(schema) 1658 # Not calling self.variants.check_clash(), because there's nothing 1659 # to clash with 1660 self.variants.check(schema, {}) 1661 # Alternate branch names have no relation to the tag enum values; 1662 # so we have to check for potential name collisions ourselves. 1663 seen = {} 1664 for v in self.variants.variants: 1665 v.check_clash(self.info, seen) 1666 # TODO check conflicting qtypes 1667 if self.doc: 1668 self.doc.connect_member(v) 1669 if self.doc: 1670 self.doc.check() 1671 1672 def c_type(self): 1673 return c_name(self.name) + pointer_suffix 1674 1675 def json_type(self): 1676 return 'value' 1677 1678 def visit(self, visitor): 1679 QAPISchemaType.visit(self, visitor) 1680 visitor.visit_alternate_type(self.name, self.info, self.ifcond, 1681 self.variants) 1682 1683 1684class QAPISchemaCommand(QAPISchemaEntity): 1685 def __init__(self, name, info, doc, ifcond, arg_type, ret_type, 1686 gen, success_response, boxed, allow_oob, allow_preconfig): 1687 QAPISchemaEntity.__init__(self, name, info, doc, ifcond) 1688 assert not arg_type or isinstance(arg_type, str) 1689 assert not ret_type or isinstance(ret_type, str) 1690 self._arg_type_name = arg_type 1691 self.arg_type = None 1692 self._ret_type_name = ret_type 1693 self.ret_type = None 1694 self.gen = gen 1695 self.success_response = success_response 1696 self.boxed = boxed 1697 self.allow_oob = allow_oob 1698 self.allow_preconfig = allow_preconfig 1699 1700 def check(self, schema): 1701 QAPISchemaEntity.check(self, schema) 1702 if self._arg_type_name: 1703 self.arg_type = schema.lookup_type(self._arg_type_name) 1704 assert isinstance(self.arg_type, QAPISchemaObjectType) 1705 assert not self.arg_type.variants or self.boxed 1706 elif self.boxed: 1707 raise QAPISemError(self.info, "Use of 'boxed' requires 'data'") 1708 if self._ret_type_name: 1709 self.ret_type = schema.lookup_type(self._ret_type_name) 1710 assert isinstance(self.ret_type, QAPISchemaType) 1711 1712 def visit(self, visitor): 1713 QAPISchemaEntity.visit(self, visitor) 1714 visitor.visit_command(self.name, self.info, self.ifcond, 1715 self.arg_type, self.ret_type, 1716 self.gen, self.success_response, 1717 self.boxed, self.allow_oob, 1718 self.allow_preconfig) 1719 1720 1721class QAPISchemaEvent(QAPISchemaEntity): 1722 def __init__(self, name, info, doc, ifcond, arg_type, boxed): 1723 QAPISchemaEntity.__init__(self, name, info, doc, ifcond) 1724 assert not arg_type or isinstance(arg_type, str) 1725 self._arg_type_name = arg_type 1726 self.arg_type = None 1727 self.boxed = boxed 1728 1729 def check(self, schema): 1730 QAPISchemaEntity.check(self, schema) 1731 if self._arg_type_name: 1732 self.arg_type = schema.lookup_type(self._arg_type_name) 1733 assert isinstance(self.arg_type, QAPISchemaObjectType) 1734 assert not self.arg_type.variants or self.boxed 1735 elif self.boxed: 1736 raise QAPISemError(self.info, "Use of 'boxed' requires 'data'") 1737 1738 def visit(self, visitor): 1739 QAPISchemaEntity.visit(self, visitor) 1740 visitor.visit_event(self.name, self.info, self.ifcond, 1741 self.arg_type, self.boxed) 1742 1743 1744class QAPISchema(object): 1745 def __init__(self, fname): 1746 self.fname = fname 1747 if sys.version_info[0] >= 3: 1748 f = open(fname, 'r', encoding='utf-8') 1749 else: 1750 f = open(fname, 'r') 1751 parser = QAPISchemaParser(f) 1752 exprs = check_exprs(parser.exprs) 1753 self.docs = parser.docs 1754 self._entity_list = [] 1755 self._entity_dict = {} 1756 self._predefining = True 1757 self._def_predefineds() 1758 self._predefining = False 1759 self._def_exprs(exprs) 1760 self.check() 1761 1762 def _def_entity(self, ent): 1763 # Only the predefined types are allowed to not have info 1764 assert ent.info or self._predefining 1765 assert ent.name is None or ent.name not in self._entity_dict 1766 self._entity_list.append(ent) 1767 if ent.name is not None: 1768 self._entity_dict[ent.name] = ent 1769 1770 def lookup_entity(self, name, typ=None): 1771 ent = self._entity_dict.get(name) 1772 if typ and not isinstance(ent, typ): 1773 return None 1774 return ent 1775 1776 def lookup_type(self, name): 1777 return self.lookup_entity(name, QAPISchemaType) 1778 1779 def _def_include(self, expr, info, doc): 1780 include = expr['include'] 1781 assert doc is None 1782 main_info = info 1783 while main_info['parent']: 1784 main_info = main_info['parent'] 1785 fname = os.path.relpath(include, os.path.dirname(main_info['file'])) 1786 self._def_entity(QAPISchemaInclude(fname, info)) 1787 1788 def _def_builtin_type(self, name, json_type, c_type): 1789 self._def_entity(QAPISchemaBuiltinType(name, json_type, c_type)) 1790 # Instantiating only the arrays that are actually used would 1791 # be nice, but we can't as long as their generated code 1792 # (qapi-builtin-types.[ch]) may be shared by some other 1793 # schema. 1794 self._make_array_type(name, None) 1795 1796 def _def_predefineds(self): 1797 for t in [('str', 'string', 'char' + pointer_suffix), 1798 ('number', 'number', 'double'), 1799 ('int', 'int', 'int64_t'), 1800 ('int8', 'int', 'int8_t'), 1801 ('int16', 'int', 'int16_t'), 1802 ('int32', 'int', 'int32_t'), 1803 ('int64', 'int', 'int64_t'), 1804 ('uint8', 'int', 'uint8_t'), 1805 ('uint16', 'int', 'uint16_t'), 1806 ('uint32', 'int', 'uint32_t'), 1807 ('uint64', 'int', 'uint64_t'), 1808 ('size', 'int', 'uint64_t'), 1809 ('bool', 'boolean', 'bool'), 1810 ('any', 'value', 'QObject' + pointer_suffix), 1811 ('null', 'null', 'QNull' + pointer_suffix)]: 1812 self._def_builtin_type(*t) 1813 self.the_empty_object_type = QAPISchemaObjectType( 1814 'q_empty', None, None, None, None, [], None, []) 1815 self._def_entity(self.the_empty_object_type) 1816 1817 qtypes = ['none', 'qnull', 'qnum', 'qstring', 'qdict', 'qlist', 1818 'qbool'] 1819 qtype_values = self._make_enum_members([{'name': n} for n in qtypes]) 1820 1821 self._def_entity(QAPISchemaEnumType('QType', None, None, None, 1822 qtype_values, 'QTYPE')) 1823 1824 def _make_features(self, features): 1825 return [QAPISchemaFeature(f['name'], f.get('if')) for f in features] 1826 1827 def _make_enum_members(self, values): 1828 return [QAPISchemaEnumMember(v['name'], v.get('if')) 1829 for v in values] 1830 1831 def _make_implicit_enum_type(self, name, info, ifcond, values): 1832 # See also QAPISchemaObjectTypeMember._pretty_owner() 1833 name = name + 'Kind' # Use namespace reserved by add_name() 1834 self._def_entity(QAPISchemaEnumType( 1835 name, info, None, ifcond, self._make_enum_members(values), None)) 1836 return name 1837 1838 def _make_array_type(self, element_type, info): 1839 name = element_type + 'List' # Use namespace reserved by add_name() 1840 if not self.lookup_type(name): 1841 self._def_entity(QAPISchemaArrayType(name, info, element_type)) 1842 return name 1843 1844 def _make_implicit_object_type(self, name, info, doc, ifcond, 1845 role, members): 1846 if not members: 1847 return None 1848 # See also QAPISchemaObjectTypeMember._pretty_owner() 1849 name = 'q_obj_%s-%s' % (name, role) 1850 typ = self.lookup_entity(name, QAPISchemaObjectType) 1851 if typ: 1852 # The implicit object type has multiple users. This can 1853 # happen only for simple unions' implicit wrapper types. 1854 # Its ifcond should be the disjunction of its user's 1855 # ifconds. Not implemented. Instead, we always pass the 1856 # wrapped type's ifcond, which is trivially the same for all 1857 # users. It's also necessary for the wrapper to compile. 1858 # But it's not tight: the disjunction need not imply it. We 1859 # may end up compiling useless wrapper types. 1860 # TODO kill simple unions or implement the disjunction 1861 assert ifcond == typ._ifcond # pylint: disable=protected-access 1862 else: 1863 self._def_entity(QAPISchemaObjectType(name, info, doc, ifcond, 1864 None, members, None, [])) 1865 return name 1866 1867 def _def_enum_type(self, expr, info, doc): 1868 name = expr['enum'] 1869 data = expr['data'] 1870 prefix = expr.get('prefix') 1871 ifcond = expr.get('if') 1872 self._def_entity(QAPISchemaEnumType( 1873 name, info, doc, ifcond, 1874 self._make_enum_members(data), prefix)) 1875 1876 def _make_member(self, name, typ, ifcond, info): 1877 optional = False 1878 if name.startswith('*'): 1879 name = name[1:] 1880 optional = True 1881 if isinstance(typ, list): 1882 assert len(typ) == 1 1883 typ = self._make_array_type(typ[0], info) 1884 return QAPISchemaObjectTypeMember(name, typ, optional, ifcond) 1885 1886 def _make_members(self, data, info): 1887 return [self._make_member(key, value['type'], value.get('if'), info) 1888 for (key, value) in data.items()] 1889 1890 def _def_struct_type(self, expr, info, doc): 1891 name = expr['struct'] 1892 base = expr.get('base') 1893 data = expr['data'] 1894 ifcond = expr.get('if') 1895 features = expr.get('features', []) 1896 self._def_entity(QAPISchemaObjectType(name, info, doc, ifcond, base, 1897 self._make_members(data, info), 1898 None, 1899 self._make_features(features))) 1900 1901 def _make_variant(self, case, typ, ifcond): 1902 return QAPISchemaObjectTypeVariant(case, typ, ifcond) 1903 1904 def _make_simple_variant(self, case, typ, ifcond, info): 1905 if isinstance(typ, list): 1906 assert len(typ) == 1 1907 typ = self._make_array_type(typ[0], info) 1908 typ = self._make_implicit_object_type( 1909 typ, info, None, self.lookup_type(typ), 1910 'wrapper', [self._make_member('data', typ, None, info)]) 1911 return QAPISchemaObjectTypeVariant(case, typ, ifcond) 1912 1913 def _def_union_type(self, expr, info, doc): 1914 name = expr['union'] 1915 data = expr['data'] 1916 base = expr.get('base') 1917 ifcond = expr.get('if') 1918 tag_name = expr.get('discriminator') 1919 tag_member = None 1920 if isinstance(base, dict): 1921 base = self._make_implicit_object_type( 1922 name, info, doc, ifcond, 1923 'base', self._make_members(base, info)) 1924 if tag_name: 1925 variants = [self._make_variant(key, value['type'], value.get('if')) 1926 for (key, value) in data.items()] 1927 members = [] 1928 else: 1929 variants = [self._make_simple_variant(key, value['type'], 1930 value.get('if'), info) 1931 for (key, value) in data.items()] 1932 enum = [{'name': v.name, 'if': v.ifcond} for v in variants] 1933 typ = self._make_implicit_enum_type(name, info, ifcond, enum) 1934 tag_member = QAPISchemaObjectTypeMember('type', typ, False) 1935 members = [tag_member] 1936 self._def_entity( 1937 QAPISchemaObjectType(name, info, doc, ifcond, base, members, 1938 QAPISchemaObjectTypeVariants(tag_name, 1939 tag_member, 1940 variants), [])) 1941 1942 def _def_alternate_type(self, expr, info, doc): 1943 name = expr['alternate'] 1944 data = expr['data'] 1945 ifcond = expr.get('if') 1946 variants = [self._make_variant(key, value['type'], value.get('if')) 1947 for (key, value) in data.items()] 1948 tag_member = QAPISchemaObjectTypeMember('type', 'QType', False) 1949 self._def_entity( 1950 QAPISchemaAlternateType(name, info, doc, ifcond, 1951 QAPISchemaObjectTypeVariants(None, 1952 tag_member, 1953 variants))) 1954 1955 def _def_command(self, expr, info, doc): 1956 name = expr['command'] 1957 data = expr.get('data') 1958 rets = expr.get('returns') 1959 gen = expr.get('gen', True) 1960 success_response = expr.get('success-response', True) 1961 boxed = expr.get('boxed', False) 1962 allow_oob = expr.get('allow-oob', False) 1963 allow_preconfig = expr.get('allow-preconfig', False) 1964 ifcond = expr.get('if') 1965 if isinstance(data, OrderedDict): 1966 data = self._make_implicit_object_type( 1967 name, info, doc, ifcond, 'arg', self._make_members(data, info)) 1968 if isinstance(rets, list): 1969 assert len(rets) == 1 1970 rets = self._make_array_type(rets[0], info) 1971 self._def_entity(QAPISchemaCommand(name, info, doc, ifcond, data, rets, 1972 gen, success_response, 1973 boxed, allow_oob, allow_preconfig)) 1974 1975 def _def_event(self, expr, info, doc): 1976 name = expr['event'] 1977 data = expr.get('data') 1978 boxed = expr.get('boxed', False) 1979 ifcond = expr.get('if') 1980 if isinstance(data, OrderedDict): 1981 data = self._make_implicit_object_type( 1982 name, info, doc, ifcond, 'arg', self._make_members(data, info)) 1983 self._def_entity(QAPISchemaEvent(name, info, doc, ifcond, data, boxed)) 1984 1985 def _def_exprs(self, exprs): 1986 for expr_elem in exprs: 1987 expr = expr_elem['expr'] 1988 info = expr_elem['info'] 1989 doc = expr_elem.get('doc') 1990 if 'enum' in expr: 1991 self._def_enum_type(expr, info, doc) 1992 elif 'struct' in expr: 1993 self._def_struct_type(expr, info, doc) 1994 elif 'union' in expr: 1995 self._def_union_type(expr, info, doc) 1996 elif 'alternate' in expr: 1997 self._def_alternate_type(expr, info, doc) 1998 elif 'command' in expr: 1999 self._def_command(expr, info, doc) 2000 elif 'event' in expr: 2001 self._def_event(expr, info, doc) 2002 elif 'include' in expr: 2003 self._def_include(expr, info, doc) 2004 else: 2005 assert False 2006 2007 def check(self): 2008 for ent in self._entity_list: 2009 ent.check(self) 2010 2011 def visit(self, visitor): 2012 visitor.visit_begin(self) 2013 module = None 2014 visitor.visit_module(module) 2015 for entity in self._entity_list: 2016 if visitor.visit_needed(entity): 2017 if entity.module != module: 2018 module = entity.module 2019 visitor.visit_module(module) 2020 entity.visit(visitor) 2021 visitor.visit_end() 2022 2023 2024# 2025# Code generation helpers 2026# 2027 2028def camel_case(name): 2029 new_name = '' 2030 first = True 2031 for ch in name: 2032 if ch in ['_', '-']: 2033 first = True 2034 elif first: 2035 new_name += ch.upper() 2036 first = False 2037 else: 2038 new_name += ch.lower() 2039 return new_name 2040 2041 2042# ENUMName -> ENUM_NAME, EnumName1 -> ENUM_NAME1 2043# ENUM_NAME -> ENUM_NAME, ENUM_NAME1 -> ENUM_NAME1, ENUM_Name2 -> ENUM_NAME2 2044# ENUM24_Name -> ENUM24_NAME 2045def camel_to_upper(value): 2046 c_fun_str = c_name(value, False) 2047 if value.isupper(): 2048 return c_fun_str 2049 2050 new_name = '' 2051 length = len(c_fun_str) 2052 for i in range(length): 2053 c = c_fun_str[i] 2054 # When c is upper and no '_' appears before, do more checks 2055 if c.isupper() and (i > 0) and c_fun_str[i - 1] != '_': 2056 if i < length - 1 and c_fun_str[i + 1].islower(): 2057 new_name += '_' 2058 elif c_fun_str[i - 1].isdigit(): 2059 new_name += '_' 2060 new_name += c 2061 return new_name.lstrip('_').upper() 2062 2063 2064def c_enum_const(type_name, const_name, prefix=None): 2065 if prefix is not None: 2066 type_name = prefix 2067 return camel_to_upper(type_name) + '_' + c_name(const_name, False).upper() 2068 2069 2070if hasattr(str, 'maketrans'): 2071 c_name_trans = str.maketrans('.-', '__') 2072else: 2073 c_name_trans = string.maketrans('.-', '__') 2074 2075 2076# Map @name to a valid C identifier. 2077# If @protect, avoid returning certain ticklish identifiers (like 2078# C keywords) by prepending 'q_'. 2079# 2080# Used for converting 'name' from a 'name':'type' qapi definition 2081# into a generated struct member, as well as converting type names 2082# into substrings of a generated C function name. 2083# '__a.b_c' -> '__a_b_c', 'x-foo' -> 'x_foo' 2084# protect=True: 'int' -> 'q_int'; protect=False: 'int' -> 'int' 2085def c_name(name, protect=True): 2086 # ANSI X3J11/88-090, 3.1.1 2087 c89_words = set(['auto', 'break', 'case', 'char', 'const', 'continue', 2088 'default', 'do', 'double', 'else', 'enum', 'extern', 2089 'float', 'for', 'goto', 'if', 'int', 'long', 'register', 2090 'return', 'short', 'signed', 'sizeof', 'static', 2091 'struct', 'switch', 'typedef', 'union', 'unsigned', 2092 'void', 'volatile', 'while']) 2093 # ISO/IEC 9899:1999, 6.4.1 2094 c99_words = set(['inline', 'restrict', '_Bool', '_Complex', '_Imaginary']) 2095 # ISO/IEC 9899:2011, 6.4.1 2096 c11_words = set(['_Alignas', '_Alignof', '_Atomic', '_Generic', 2097 '_Noreturn', '_Static_assert', '_Thread_local']) 2098 # GCC http://gcc.gnu.org/onlinedocs/gcc-4.7.1/gcc/C-Extensions.html 2099 # excluding _.* 2100 gcc_words = set(['asm', 'typeof']) 2101 # C++ ISO/IEC 14882:2003 2.11 2102 cpp_words = set(['bool', 'catch', 'class', 'const_cast', 'delete', 2103 'dynamic_cast', 'explicit', 'false', 'friend', 'mutable', 2104 'namespace', 'new', 'operator', 'private', 'protected', 2105 'public', 'reinterpret_cast', 'static_cast', 'template', 2106 'this', 'throw', 'true', 'try', 'typeid', 'typename', 2107 'using', 'virtual', 'wchar_t', 2108 # alternative representations 2109 'and', 'and_eq', 'bitand', 'bitor', 'compl', 'not', 2110 'not_eq', 'or', 'or_eq', 'xor', 'xor_eq']) 2111 # namespace pollution: 2112 polluted_words = set(['unix', 'errno', 'mips', 'sparc', 'i386']) 2113 name = name.translate(c_name_trans) 2114 if protect and (name in c89_words | c99_words | c11_words | gcc_words 2115 | cpp_words | polluted_words): 2116 return 'q_' + name 2117 return name 2118 2119 2120eatspace = '\033EATSPACE.' 2121pointer_suffix = ' *' + eatspace 2122 2123 2124def genindent(count): 2125 ret = '' 2126 for _ in range(count): 2127 ret += ' ' 2128 return ret 2129 2130 2131indent_level = 0 2132 2133 2134def push_indent(indent_amount=4): 2135 global indent_level 2136 indent_level += indent_amount 2137 2138 2139def pop_indent(indent_amount=4): 2140 global indent_level 2141 indent_level -= indent_amount 2142 2143 2144# Generate @code with @kwds interpolated. 2145# Obey indent_level, and strip eatspace. 2146def cgen(code, **kwds): 2147 raw = code % kwds 2148 if indent_level: 2149 indent = genindent(indent_level) 2150 # re.subn() lacks flags support before Python 2.7, use re.compile() 2151 raw = re.subn(re.compile(r'^(?!(#|$))', re.MULTILINE), 2152 indent, raw) 2153 raw = raw[0] 2154 return re.sub(re.escape(eatspace) + r' *', '', raw) 2155 2156 2157def mcgen(code, **kwds): 2158 if code[0] == '\n': 2159 code = code[1:] 2160 return cgen(code, **kwds) 2161 2162 2163def c_fname(filename): 2164 return re.sub(r'[^A-Za-z0-9_]', '_', filename) 2165 2166 2167def guardstart(name): 2168 return mcgen(''' 2169#ifndef %(name)s 2170#define %(name)s 2171 2172''', 2173 name=c_fname(name).upper()) 2174 2175 2176def guardend(name): 2177 return mcgen(''' 2178 2179#endif /* %(name)s */ 2180''', 2181 name=c_fname(name).upper()) 2182 2183 2184def gen_if(ifcond): 2185 ret = '' 2186 for ifc in ifcond: 2187 ret += mcgen(''' 2188#if %(cond)s 2189''', cond=ifc) 2190 return ret 2191 2192 2193def gen_endif(ifcond): 2194 ret = '' 2195 for ifc in reversed(ifcond): 2196 ret += mcgen(''' 2197#endif /* %(cond)s */ 2198''', cond=ifc) 2199 return ret 2200 2201 2202def _wrap_ifcond(ifcond, before, after): 2203 if before == after: 2204 return after # suppress empty #if ... #endif 2205 2206 assert after.startswith(before) 2207 out = before 2208 added = after[len(before):] 2209 if added[0] == '\n': 2210 out += '\n' 2211 added = added[1:] 2212 out += gen_if(ifcond) 2213 out += added 2214 out += gen_endif(ifcond) 2215 return out 2216 2217 2218def gen_enum_lookup(name, members, prefix=None): 2219 ret = mcgen(''' 2220 2221const QEnumLookup %(c_name)s_lookup = { 2222 .array = (const char *const[]) { 2223''', 2224 c_name=c_name(name)) 2225 for m in members: 2226 ret += gen_if(m.ifcond) 2227 index = c_enum_const(name, m.name, prefix) 2228 ret += mcgen(''' 2229 [%(index)s] = "%(name)s", 2230''', 2231 index=index, name=m.name) 2232 ret += gen_endif(m.ifcond) 2233 2234 ret += mcgen(''' 2235 }, 2236 .size = %(max_index)s 2237}; 2238''', 2239 max_index=c_enum_const(name, '_MAX', prefix)) 2240 return ret 2241 2242 2243def gen_enum(name, members, prefix=None): 2244 # append automatically generated _MAX value 2245 enum_members = members + [QAPISchemaEnumMember('_MAX')] 2246 2247 ret = mcgen(''' 2248 2249typedef enum %(c_name)s { 2250''', 2251 c_name=c_name(name)) 2252 2253 for m in enum_members: 2254 ret += gen_if(m.ifcond) 2255 ret += mcgen(''' 2256 %(c_enum)s, 2257''', 2258 c_enum=c_enum_const(name, m.name, prefix)) 2259 ret += gen_endif(m.ifcond) 2260 2261 ret += mcgen(''' 2262} %(c_name)s; 2263''', 2264 c_name=c_name(name)) 2265 2266 ret += mcgen(''' 2267 2268#define %(c_name)s_str(val) \\ 2269 qapi_enum_lookup(&%(c_name)s_lookup, (val)) 2270 2271extern const QEnumLookup %(c_name)s_lookup; 2272''', 2273 c_name=c_name(name)) 2274 return ret 2275 2276 2277def build_params(arg_type, boxed, extra=None): 2278 ret = '' 2279 sep = '' 2280 if boxed: 2281 assert arg_type 2282 ret += '%s arg' % arg_type.c_param_type() 2283 sep = ', ' 2284 elif arg_type: 2285 assert not arg_type.variants 2286 for memb in arg_type.members: 2287 ret += sep 2288 sep = ', ' 2289 if memb.optional: 2290 ret += 'bool has_%s, ' % c_name(memb.name) 2291 ret += '%s %s' % (memb.type.c_param_type(), 2292 c_name(memb.name)) 2293 if extra: 2294 ret += sep + extra 2295 return ret if ret else 'void' 2296 2297 2298# 2299# Accumulate and write output 2300# 2301 2302class QAPIGen(object): 2303 2304 def __init__(self, fname): 2305 self.fname = fname 2306 self._preamble = '' 2307 self._body = '' 2308 2309 def preamble_add(self, text): 2310 self._preamble += text 2311 2312 def add(self, text): 2313 self._body += text 2314 2315 def get_content(self): 2316 return self._top() + self._preamble + self._body + self._bottom() 2317 2318 def _top(self): 2319 return '' 2320 2321 def _bottom(self): 2322 return '' 2323 2324 def write(self, output_dir): 2325 pathname = os.path.join(output_dir, self.fname) 2326 dir = os.path.dirname(pathname) 2327 if dir: 2328 try: 2329 os.makedirs(dir) 2330 except os.error as e: 2331 if e.errno != errno.EEXIST: 2332 raise 2333 fd = os.open(pathname, os.O_RDWR | os.O_CREAT, 0o666) 2334 if sys.version_info[0] >= 3: 2335 f = open(fd, 'r+', encoding='utf-8') 2336 else: 2337 f = os.fdopen(fd, 'r+') 2338 text = self.get_content() 2339 oldtext = f.read(len(text) + 1) 2340 if text != oldtext: 2341 f.seek(0) 2342 f.truncate(0) 2343 f.write(text) 2344 f.close() 2345 2346 2347@contextmanager 2348def ifcontext(ifcond, *args): 2349 """A 'with' statement context manager to wrap with start_if()/end_if() 2350 2351 *args: any number of QAPIGenCCode 2352 2353 Example:: 2354 2355 with ifcontext(ifcond, self._genh, self._genc): 2356 modify self._genh and self._genc ... 2357 2358 Is equivalent to calling:: 2359 2360 self._genh.start_if(ifcond) 2361 self._genc.start_if(ifcond) 2362 modify self._genh and self._genc ... 2363 self._genh.end_if() 2364 self._genc.end_if() 2365 """ 2366 for arg in args: 2367 arg.start_if(ifcond) 2368 yield 2369 for arg in args: 2370 arg.end_if() 2371 2372 2373class QAPIGenCCode(QAPIGen): 2374 2375 def __init__(self, fname): 2376 QAPIGen.__init__(self, fname) 2377 self._start_if = None 2378 2379 def start_if(self, ifcond): 2380 assert self._start_if is None 2381 self._start_if = (ifcond, self._body, self._preamble) 2382 2383 def end_if(self): 2384 assert self._start_if 2385 self._wrap_ifcond() 2386 self._start_if = None 2387 2388 def _wrap_ifcond(self): 2389 self._body = _wrap_ifcond(self._start_if[0], 2390 self._start_if[1], self._body) 2391 self._preamble = _wrap_ifcond(self._start_if[0], 2392 self._start_if[2], self._preamble) 2393 2394 def get_content(self): 2395 assert self._start_if is None 2396 return QAPIGen.get_content(self) 2397 2398 2399class QAPIGenC(QAPIGenCCode): 2400 2401 def __init__(self, fname, blurb, pydoc): 2402 QAPIGenCCode.__init__(self, fname) 2403 self._blurb = blurb 2404 self._copyright = '\n * '.join(re.findall(r'^Copyright .*', pydoc, 2405 re.MULTILINE)) 2406 2407 def _top(self): 2408 return mcgen(''' 2409/* AUTOMATICALLY GENERATED, DO NOT MODIFY */ 2410 2411/* 2412%(blurb)s 2413 * 2414 * %(copyright)s 2415 * 2416 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later. 2417 * See the COPYING.LIB file in the top-level directory. 2418 */ 2419 2420''', 2421 blurb=self._blurb, copyright=self._copyright) 2422 2423 def _bottom(self): 2424 return mcgen(''' 2425 2426/* Dummy declaration to prevent empty .o file */ 2427char qapi_dummy_%(name)s; 2428''', 2429 name=c_fname(self.fname)) 2430 2431 2432class QAPIGenH(QAPIGenC): 2433 2434 def _top(self): 2435 return QAPIGenC._top(self) + guardstart(self.fname) 2436 2437 def _bottom(self): 2438 return guardend(self.fname) 2439 2440 2441class QAPIGenDoc(QAPIGen): 2442 2443 def _top(self): 2444 return (QAPIGen._top(self) 2445 + '@c AUTOMATICALLY GENERATED, DO NOT MODIFY\n\n') 2446 2447 2448class QAPISchemaMonolithicCVisitor(QAPISchemaVisitor): 2449 2450 def __init__(self, prefix, what, blurb, pydoc): 2451 self._prefix = prefix 2452 self._what = what 2453 self._genc = QAPIGenC(self._prefix + self._what + '.c', 2454 blurb, pydoc) 2455 self._genh = QAPIGenH(self._prefix + self._what + '.h', 2456 blurb, pydoc) 2457 2458 def write(self, output_dir): 2459 self._genc.write(output_dir) 2460 self._genh.write(output_dir) 2461 2462 2463class QAPISchemaModularCVisitor(QAPISchemaVisitor): 2464 2465 def __init__(self, prefix, what, blurb, pydoc): 2466 self._prefix = prefix 2467 self._what = what 2468 self._blurb = blurb 2469 self._pydoc = pydoc 2470 self._genc = None 2471 self._genh = None 2472 self._module = {} 2473 self._main_module = None 2474 2475 @staticmethod 2476 def _is_user_module(name): 2477 return name and not name.startswith('./') 2478 2479 @staticmethod 2480 def _is_builtin_module(name): 2481 return not name 2482 2483 def _module_dirname(self, what, name): 2484 if self._is_user_module(name): 2485 return os.path.dirname(name) 2486 return '' 2487 2488 def _module_basename(self, what, name): 2489 ret = '' if self._is_builtin_module(name) else self._prefix 2490 if self._is_user_module(name): 2491 basename = os.path.basename(name) 2492 ret += what 2493 if name != self._main_module: 2494 ret += '-' + os.path.splitext(basename)[0] 2495 else: 2496 name = name[2:] if name else 'builtin' 2497 ret += re.sub(r'-', '-' + name + '-', what) 2498 return ret 2499 2500 def _module_filename(self, what, name): 2501 return os.path.join(self._module_dirname(what, name), 2502 self._module_basename(what, name)) 2503 2504 def _add_module(self, name, blurb): 2505 basename = self._module_filename(self._what, name) 2506 genc = QAPIGenC(basename + '.c', blurb, self._pydoc) 2507 genh = QAPIGenH(basename + '.h', blurb, self._pydoc) 2508 self._module[name] = (genc, genh) 2509 self._set_module(name) 2510 2511 def _add_user_module(self, name, blurb): 2512 assert self._is_user_module(name) 2513 if self._main_module is None: 2514 self._main_module = name 2515 self._add_module(name, blurb) 2516 2517 def _add_system_module(self, name, blurb): 2518 self._add_module(name and './' + name, blurb) 2519 2520 def _set_module(self, name): 2521 self._genc, self._genh = self._module[name] 2522 2523 def write(self, output_dir, opt_builtins=False): 2524 for name in self._module: 2525 if self._is_builtin_module(name) and not opt_builtins: 2526 continue 2527 (genc, genh) = self._module[name] 2528 genc.write(output_dir) 2529 genh.write(output_dir) 2530 2531 def _begin_user_module(self, name): 2532 pass 2533 2534 def visit_module(self, name): 2535 if name in self._module: 2536 self._set_module(name) 2537 elif self._is_builtin_module(name): 2538 # The built-in module has not been created. No code may 2539 # be generated. 2540 self._genc = None 2541 self._genh = None 2542 else: 2543 self._add_user_module(name, self._blurb) 2544 self._begin_user_module(name) 2545 2546 def visit_include(self, name, info): 2547 relname = os.path.relpath(self._module_filename(self._what, name), 2548 os.path.dirname(self._genh.fname)) 2549 self._genh.preamble_add(mcgen(''' 2550#include "%(relname)s.h" 2551''', 2552 relname=relname)) 2553