]>
Commit | Line | Data |
---|---|---|
39a18158 MA |
1 | # |
2 | # QAPI introspection generator | |
3 | # | |
9167ebd9 | 4 | # Copyright (C) 2015-2016 Red Hat, Inc. |
39a18158 MA |
5 | # |
6 | # Authors: | |
7 | # Markus Armbruster <armbru@redhat.com> | |
8 | # | |
9 | # This work is licensed under the terms of the GNU GPL, version 2. | |
10 | # See the COPYING file in the top-level directory. | |
11 | ||
12 | from qapi import * | |
13 | ||
14 | ||
15 | # Caveman's json.dumps() replacement (we're stuck at Python 2.4) | |
16 | # TODO try to use json.dumps() once we get unstuck | |
17 | def to_json(obj, level=0): | |
18 | if obj is None: | |
19 | ret = 'null' | |
20 | elif isinstance(obj, str): | |
21 | ret = '"' + obj.replace('"', r'\"') + '"' | |
22 | elif isinstance(obj, list): | |
23 | elts = [to_json(elt, level + 1) | |
24 | for elt in obj] | |
25 | ret = '[' + ', '.join(elts) + ']' | |
26 | elif isinstance(obj, dict): | |
27 | elts = ['"%s": %s' % (key.replace('"', r'\"'), | |
28 | to_json(obj[key], level + 1)) | |
29 | for key in sorted(obj.keys())] | |
30 | ret = '{' + ', '.join(elts) + '}' | |
31 | else: | |
32 | assert False # not implemented | |
33 | if level == 1: | |
34 | ret = '\n' + ret | |
35 | return ret | |
36 | ||
37 | ||
38 | def to_c_string(string): | |
39 | return '"' + string.replace('\\', r'\\').replace('"', r'\"') + '"' | |
40 | ||
41 | ||
42 | class QAPISchemaGenIntrospectVisitor(QAPISchemaVisitor): | |
1a9a507b MA |
43 | def __init__(self, unmask): |
44 | self._unmask = unmask | |
39a18158 MA |
45 | self.defn = None |
46 | self.decl = None | |
47 | self._schema = None | |
48 | self._jsons = None | |
49 | self._used_types = None | |
1a9a507b | 50 | self._name_map = None |
39a18158 MA |
51 | |
52 | def visit_begin(self, schema): | |
53 | self._schema = schema | |
54 | self._jsons = [] | |
55 | self._used_types = [] | |
1a9a507b | 56 | self._name_map = {} |
39a18158 MA |
57 | |
58 | def visit_end(self): | |
59 | # visit the types that are actually used | |
1a9a507b MA |
60 | jsons = self._jsons |
61 | self._jsons = [] | |
39a18158 MA |
62 | for typ in self._used_types: |
63 | typ.visit(self) | |
39a18158 MA |
64 | # generate C |
65 | # TODO can generate awfully long lines | |
1a9a507b | 66 | jsons.extend(self._jsons) |
9b0c9a63 | 67 | name = c_name(prefix, protect=False) + 'qmp_schema_json' |
39a18158 MA |
68 | self.decl = mcgen(''' |
69 | extern const char %(c_name)s[]; | |
70 | ''', | |
71 | c_name=c_name(name)) | |
1a9a507b | 72 | lines = to_json(jsons).split('\n') |
39a18158 MA |
73 | c_string = '\n '.join([to_c_string(line) for line in lines]) |
74 | self.defn = mcgen(''' | |
75 | const char %(c_name)s[] = %(c_string)s; | |
76 | ''', | |
77 | c_name=c_name(name), | |
78 | c_string=c_string) | |
79 | self._schema = None | |
80 | self._jsons = None | |
81 | self._used_types = None | |
1a9a507b MA |
82 | self._name_map = None |
83 | ||
25a0d9c9 EB |
84 | def visit_needed(self, entity): |
85 | # Ignore types on first pass; visit_end() will pick up used types | |
86 | return not isinstance(entity, QAPISchemaType) | |
87 | ||
1a9a507b MA |
88 | def _name(self, name): |
89 | if self._unmask: | |
90 | return name | |
91 | if name not in self._name_map: | |
92 | self._name_map[name] = '%d' % len(self._name_map) | |
93 | return self._name_map[name] | |
39a18158 MA |
94 | |
95 | def _use_type(self, typ): | |
96 | # Map the various integer types to plain int | |
97 | if typ.json_type() == 'int': | |
98 | typ = self._schema.lookup_type('int') | |
99 | elif (isinstance(typ, QAPISchemaArrayType) and | |
100 | typ.element_type.json_type() == 'int'): | |
101 | typ = self._schema.lookup_type('intList') | |
102 | # Add type to work queue if new | |
103 | if typ not in self._used_types: | |
104 | self._used_types.append(typ) | |
1a9a507b MA |
105 | # Clients should examine commands and events, not types. Hide |
106 | # type names to reduce the temptation. Also saves a few | |
107 | # characters. | |
108 | if isinstance(typ, QAPISchemaBuiltinType): | |
109 | return typ.name | |
ce5fcb47 EB |
110 | if isinstance(typ, QAPISchemaArrayType): |
111 | return '[' + self._use_type(typ.element_type) + ']' | |
1a9a507b | 112 | return self._name(typ.name) |
39a18158 MA |
113 | |
114 | def _gen_json(self, name, mtype, obj): | |
ce5fcb47 | 115 | if mtype not in ('command', 'event', 'builtin', 'array'): |
1a9a507b | 116 | name = self._name(name) |
39a18158 MA |
117 | obj['name'] = name |
118 | obj['meta-type'] = mtype | |
119 | self._jsons.append(obj) | |
120 | ||
121 | def _gen_member(self, member): | |
122 | ret = {'name': member.name, 'type': self._use_type(member.type)} | |
123 | if member.optional: | |
124 | ret['default'] = None | |
125 | return ret | |
126 | ||
127 | def _gen_variants(self, tag_name, variants): | |
128 | return {'tag': tag_name, | |
129 | 'variants': [self._gen_variant(v) for v in variants]} | |
130 | ||
131 | def _gen_variant(self, variant): | |
132 | return {'case': variant.name, 'type': self._use_type(variant.type)} | |
133 | ||
134 | def visit_builtin_type(self, name, info, json_type): | |
135 | self._gen_json(name, 'builtin', {'json-type': json_type}) | |
136 | ||
137 | def visit_enum_type(self, name, info, values, prefix): | |
138 | self._gen_json(name, 'enum', {'values': values}) | |
139 | ||
140 | def visit_array_type(self, name, info, element_type): | |
ce5fcb47 EB |
141 | element = self._use_type(element_type) |
142 | self._gen_json('[' + element + ']', 'array', {'element-type': element}) | |
39a18158 MA |
143 | |
144 | def visit_object_type_flat(self, name, info, members, variants): | |
145 | obj = {'members': [self._gen_member(m) for m in members]} | |
146 | if variants: | |
147 | obj.update(self._gen_variants(variants.tag_member.name, | |
148 | variants.variants)) | |
149 | self._gen_json(name, 'object', obj) | |
150 | ||
151 | def visit_alternate_type(self, name, info, variants): | |
152 | self._gen_json(name, 'alternate', | |
153 | {'members': [{'type': self._use_type(m.type)} | |
154 | for m in variants.variants]}) | |
155 | ||
156 | def visit_command(self, name, info, arg_type, ret_type, | |
48825ca4 | 157 | gen, success_response, boxed): |
39a18158 MA |
158 | arg_type = arg_type or self._schema.the_empty_object_type |
159 | ret_type = ret_type or self._schema.the_empty_object_type | |
160 | self._gen_json(name, 'command', | |
161 | {'arg-type': self._use_type(arg_type), | |
162 | 'ret-type': self._use_type(ret_type)}) | |
163 | ||
48825ca4 | 164 | def visit_event(self, name, info, arg_type, boxed): |
39a18158 MA |
165 | arg_type = arg_type or self._schema.the_empty_object_type |
166 | self._gen_json(name, 'event', {'arg-type': self._use_type(arg_type)}) | |
167 | ||
1a9a507b MA |
168 | # Debugging aid: unmask QAPI schema's type names |
169 | # We normally mask them, because they're not QMP wire ABI | |
170 | opt_unmask = False | |
171 | ||
172 | (input_file, output_dir, do_c, do_h, prefix, opts) = \ | |
ef801a9b | 173 | parse_command_line('u', ['unmask-non-abi-names']) |
1a9a507b MA |
174 | |
175 | for o, a in opts: | |
ef801a9b | 176 | if o in ('-u', '--unmask-non-abi-names'): |
1a9a507b | 177 | opt_unmask = True |
39a18158 MA |
178 | |
179 | c_comment = ''' | |
180 | /* | |
181 | * QAPI/QMP schema introspection | |
182 | * | |
183 | * Copyright (C) 2015 Red Hat, Inc. | |
184 | * | |
185 | * This work is licensed under the terms of the GNU LGPL, version 2.1 or later. | |
186 | * See the COPYING.LIB file in the top-level directory. | |
187 | * | |
188 | */ | |
189 | ''' | |
190 | h_comment = ''' | |
191 | /* | |
192 | * QAPI/QMP schema introspection | |
193 | * | |
194 | * Copyright (C) 2015 Red Hat, Inc. | |
195 | * | |
196 | * This work is licensed under the terms of the GNU LGPL, version 2.1 or later. | |
197 | * See the COPYING.LIB file in the top-level directory. | |
198 | * | |
199 | */ | |
200 | ''' | |
201 | ||
202 | (fdef, fdecl) = open_output(output_dir, do_c, do_h, prefix, | |
203 | 'qmp-introspect.c', 'qmp-introspect.h', | |
204 | c_comment, h_comment) | |
205 | ||
206 | fdef.write(mcgen(''' | |
9167ebd9 | 207 | #include "qemu/osdep.h" |
39a18158 MA |
208 | #include "%(prefix)sqmp-introspect.h" |
209 | ||
210 | ''', | |
211 | prefix=prefix)) | |
212 | ||
213 | schema = QAPISchema(input_file) | |
1a9a507b | 214 | gen = QAPISchemaGenIntrospectVisitor(opt_unmask) |
39a18158 MA |
215 | schema.visit(gen) |
216 | fdef.write(gen.defn) | |
217 | fdecl.write(gen.decl) | |
218 | ||
219 | close_output(fdef, fdecl) |