xref: /openbmc/qemu/scripts/qapi/common.py (revision 0ca7b11709b8e81e5e8f28245b5b594b597b062b)
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