José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 1 | ########################################################################## |
| 2 | # |
| 3 | # Copyright 2008-2010 VMware, Inc. |
| 4 | # All Rights Reserved. |
| 5 | # |
| 6 | # Permission is hereby granted, free of charge, to any person obtaining a copy |
| 7 | # of this software and associated documentation files (the "Software"), to deal |
| 8 | # in the Software without restriction, including without limitation the rights |
| 9 | # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
| 10 | # copies of the Software, and to permit persons to whom the Software is |
| 11 | # furnished to do so, subject to the following conditions: |
| 12 | # |
| 13 | # The above copyright notice and this permission notice shall be included in |
| 14 | # all copies or substantial portions of the Software. |
| 15 | # |
| 16 | # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 17 | # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 18 | # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| 19 | # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| 20 | # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| 21 | # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
| 22 | # THE SOFTWARE. |
| 23 | # |
| 24 | ##########################################################################/ |
| 25 | |
José Fonseca | 73d0a96 | 2010-11-29 14:49:34 +0000 | [diff] [blame] | 26 | """Common trace code generation.""" |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 27 | |
| 28 | |
José Fonseca | 452d325 | 2012-04-14 15:55:40 +0100 | [diff] [blame] | 29 | # Adjust path |
| 30 | import os.path |
| 31 | import sys |
| 32 | sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..')) |
| 33 | |
| 34 | |
José Fonseca | bd86a22 | 2011-09-27 09:21:38 +0100 | [diff] [blame] | 35 | import specs.stdapi as stdapi |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 36 | |
| 37 | |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 38 | def getWrapperInterfaceName(interface): |
José Fonseca | 87d1cc6 | 2010-11-29 15:57:25 +0000 | [diff] [blame] | 39 | return "Wrap" + interface.expr |
| 40 | |
José Fonseca | 6fac5ae | 2010-11-29 16:09:13 +0000 | [diff] [blame] | 41 | |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 42 | |
| 43 | class ExpanderMixin: |
| 44 | '''Mixin class that provides a bunch of methods to expand C expressions |
| 45 | from the specifications.''' |
| 46 | |
| 47 | __structs = None |
| 48 | __indices = None |
| 49 | |
| 50 | def expand(self, expr): |
| 51 | # Expand a C expression, replacing certain variables |
| 52 | if not isinstance(expr, basestring): |
| 53 | return expr |
| 54 | variables = {} |
| 55 | |
| 56 | if self.__structs is not None: |
| 57 | variables['self'] = '(%s)' % self.__structs[0] |
| 58 | if self.__indices is not None: |
| 59 | variables['i'] = self.__indices[0] |
| 60 | |
| 61 | expandedExpr = expr.format(**variables) |
| 62 | if expandedExpr != expr and 0: |
| 63 | sys.stderr.write(" %r -> %r\n" % (expr, expandedExpr)) |
| 64 | return expandedExpr |
| 65 | |
| 66 | def visitMember(self, structInstance, member_type, *args, **kwargs): |
| 67 | self.__structs = (structInstance, self.__structs) |
| 68 | try: |
| 69 | return self.visit(member_type, *args, **kwargs) |
| 70 | finally: |
| 71 | _, self.__structs = self.__structs |
| 72 | |
| 73 | def visitElement(self, element_index, element_type, *args, **kwargs): |
| 74 | self.__indices = (element_index, self.__indices) |
| 75 | try: |
| 76 | return self.visit(element_type, *args, **kwargs) |
| 77 | finally: |
| 78 | _, self.__indices = self.__indices |
| 79 | |
| 80 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 81 | class ComplexValueSerializer(stdapi.OnceVisitor): |
| 82 | '''Type visitors which generates serialization functions for |
| 83 | complex types. |
| 84 | |
| 85 | Simple types are serialized inline. |
| 86 | ''' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 87 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 88 | def __init__(self, serializer): |
| 89 | stdapi.OnceVisitor.__init__(self) |
| 90 | self.serializer = serializer |
| 91 | |
| 92 | def visitVoid(self, literal): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 93 | pass |
| 94 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 95 | def visitLiteral(self, literal): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 96 | pass |
| 97 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 98 | def visitString(self, string): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 99 | pass |
| 100 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 101 | def visitConst(self, const): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 102 | self.visit(const.type) |
| 103 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 104 | def visitStruct(self, struct): |
José Fonseca | 30fb4c3 | 2012-11-04 11:07:45 +0000 | [diff] [blame] | 105 | print 'static const char * _struct%s_members[%u] = {' % (struct.tag, len(struct.members)) |
José Fonseca | 7eef8ce | 2010-11-26 15:46:36 +0000 | [diff] [blame] | 106 | for type, name, in struct.members: |
José Fonseca | 30fb4c3 | 2012-11-04 11:07:45 +0000 | [diff] [blame] | 107 | print ' "%s",' % (name,) |
| 108 | print '};' |
| 109 | print 'static const trace::StructSig _struct%s_sig = {' % (struct.tag,) |
| 110 | print ' %u, "%s", %u, _struct%s_members' % (struct.id, struct.name, len(struct.members), struct.tag) |
| 111 | print '};' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 112 | print |
| 113 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 114 | def visitArray(self, array): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 115 | self.visit(array.type) |
| 116 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 117 | def visitBlob(self, array): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 118 | pass |
| 119 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 120 | def visitEnum(self, enum): |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 121 | print 'static const trace::EnumValue _enum%s_values[] = {' % (enum.tag) |
José Fonseca | eb64451 | 2011-12-11 10:33:55 +0000 | [diff] [blame] | 122 | for value in enum.values: |
| 123 | print ' {"%s", %s},' % (value, value) |
| 124 | print '};' |
| 125 | print |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 126 | print 'static const trace::EnumSig _enum%s_sig = {' % (enum.tag) |
| 127 | print ' %u, %u, _enum%s_values' % (enum.id, len(enum.values), enum.tag) |
José Fonseca | eb64451 | 2011-12-11 10:33:55 +0000 | [diff] [blame] | 128 | print '};' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 129 | print |
| 130 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 131 | def visitBitmask(self, bitmask): |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 132 | print 'static const trace::BitmaskFlag _bitmask%s_flags[] = {' % (bitmask.tag) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 133 | for value in bitmask.values: |
José Fonseca | d35973c | 2010-11-26 14:14:45 +0000 | [diff] [blame] | 134 | print ' {"%s", %s},' % (value, value) |
| 135 | print '};' |
| 136 | print |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 137 | print 'static const trace::BitmaskSig _bitmask%s_sig = {' % (bitmask.tag) |
| 138 | print ' %u, %u, _bitmask%s_flags' % (bitmask.id, len(bitmask.values), bitmask.tag) |
José Fonseca | d35973c | 2010-11-26 14:14:45 +0000 | [diff] [blame] | 139 | print '};' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 140 | print |
| 141 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 142 | def visitPointer(self, pointer): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 143 | self.visit(pointer.type) |
| 144 | |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 145 | def visitIntPointer(self, pointer): |
| 146 | pass |
| 147 | |
José Fonseca | fbcf683 | 2012-04-05 07:10:30 +0100 | [diff] [blame] | 148 | def visitObjPointer(self, pointer): |
| 149 | self.visit(pointer.type) |
| 150 | |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 151 | def visitLinearPointer(self, pointer): |
| 152 | self.visit(pointer.type) |
| 153 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 154 | def visitHandle(self, handle): |
José Fonseca | 50d78d8 | 2010-11-23 22:13:14 +0000 | [diff] [blame] | 155 | self.visit(handle.type) |
| 156 | |
José Fonseca | b89c593 | 2012-04-01 22:47:11 +0200 | [diff] [blame] | 157 | def visitReference(self, reference): |
| 158 | self.visit(reference.type) |
| 159 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 160 | def visitAlias(self, alias): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 161 | self.visit(alias.type) |
| 162 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 163 | def visitOpaque(self, opaque): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 164 | pass |
| 165 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 166 | def visitInterface(self, interface): |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 167 | pass |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 168 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 169 | def visitPolymorphic(self, polymorphic): |
José Fonseca | b95e372 | 2012-04-16 14:01:15 +0100 | [diff] [blame] | 170 | if not polymorphic.contextLess: |
| 171 | return |
José Fonseca | 06e8519 | 2011-10-16 14:15:36 +0100 | [diff] [blame] | 172 | print 'static void _write__%s(int selector, const %s & value) {' % (polymorphic.tag, polymorphic.expr) |
José Fonseca | c636b9d | 2011-10-14 10:15:02 +0100 | [diff] [blame] | 173 | print ' switch (selector) {' |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 174 | for cases, type in polymorphic.iterSwitch(): |
José Fonseca | c636b9d | 2011-10-14 10:15:02 +0100 | [diff] [blame] | 175 | for case in cases: |
| 176 | print ' %s:' % case |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 177 | self.serializer.visit(type, 'static_cast<%s>(value)' % (type,)) |
José Fonseca | c636b9d | 2011-10-14 10:15:02 +0100 | [diff] [blame] | 178 | print ' break;' |
| 179 | print ' }' |
| 180 | print '}' |
| 181 | print |
José Fonseca | 16d46dd | 2011-10-13 09:52:52 +0100 | [diff] [blame] | 182 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 183 | |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 184 | class ValueSerializer(stdapi.Visitor, ExpanderMixin): |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 185 | '''Visitor which generates code to serialize any type. |
| 186 | |
| 187 | Simple types are serialized inline here, whereas the serialization of |
| 188 | complex types is dispatched to the serialization functions generated by |
| 189 | ComplexValueSerializer visitor above. |
| 190 | ''' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 191 | |
José Fonseca | 30fb4c3 | 2012-11-04 11:07:45 +0000 | [diff] [blame] | 192 | def __init__(self): |
| 193 | #stdapi.Visitor.__init__(self) |
| 194 | self.indices = [] |
| 195 | self.instances = [] |
| 196 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 197 | def visitLiteral(self, literal, instance): |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 198 | print ' trace::localWriter.write%s(%s);' % (literal.kind, instance) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 199 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 200 | def visitString(self, string, instance): |
José Fonseca | bcfc81b | 2012-08-07 21:07:22 +0100 | [diff] [blame] | 201 | if not string.wide: |
José Fonseca | 280a176 | 2012-01-31 15:10:13 +0000 | [diff] [blame] | 202 | cast = 'const char *' |
José Fonseca | bcfc81b | 2012-08-07 21:07:22 +0100 | [diff] [blame] | 203 | suffix = 'String' |
José Fonseca | e6a50bd | 2010-11-24 10:12:22 +0000 | [diff] [blame] | 204 | else: |
José Fonseca | bcfc81b | 2012-08-07 21:07:22 +0100 | [diff] [blame] | 205 | cast = 'const wchar_t *' |
| 206 | suffix = 'WString' |
José Fonseca | 280a176 | 2012-01-31 15:10:13 +0000 | [diff] [blame] | 207 | if cast != string.expr: |
| 208 | # reinterpret_cast is necessary for GLubyte * <=> char * |
| 209 | instance = 'reinterpret_cast<%s>(%s)' % (cast, instance) |
| 210 | if string.length is not None: |
| 211 | length = ', %s' % string.length |
| 212 | else: |
| 213 | length = '' |
José Fonseca | bcfc81b | 2012-08-07 21:07:22 +0100 | [diff] [blame] | 214 | print ' trace::localWriter.write%s(%s%s);' % (suffix, instance, length) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 215 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 216 | def visitConst(self, const, instance): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 217 | self.visit(const.type, instance) |
| 218 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 219 | def visitStruct(self, struct, instance): |
José Fonseca | 30fb4c3 | 2012-11-04 11:07:45 +0000 | [diff] [blame] | 220 | print ' trace::localWriter.beginStruct(&_struct%s_sig);' % (struct.tag,) |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 221 | for type, name in struct.members: |
| 222 | self.visitMember(instance, type, '(%s).%s' % (instance, name,)) |
José Fonseca | 30fb4c3 | 2012-11-04 11:07:45 +0000 | [diff] [blame] | 223 | print ' trace::localWriter.endStruct();' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 224 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 225 | def visitArray(self, array, instance): |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 226 | length = '_c' + array.type.tag |
| 227 | index = '_i' + array.type.tag |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 228 | array_length = self.expand(array.length) |
José Fonseca | fd34e4e | 2011-06-03 19:34:29 +0100 | [diff] [blame] | 229 | print ' if (%s) {' % instance |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 230 | print ' size_t %s = %s > 0 ? %s : 0;' % (length, array_length, array_length) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 231 | print ' trace::localWriter.beginArray(%s);' % length |
José Fonseca | fd34e4e | 2011-06-03 19:34:29 +0100 | [diff] [blame] | 232 | print ' for (size_t %s = 0; %s < %s; ++%s) {' % (index, index, length, index) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 233 | print ' trace::localWriter.beginElement();' |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 234 | self.visitElement(index, array.type, '(%s)[%s]' % (instance, index)) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 235 | print ' trace::localWriter.endElement();' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 236 | print ' }' |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 237 | print ' trace::localWriter.endArray();' |
José Fonseca | fd34e4e | 2011-06-03 19:34:29 +0100 | [diff] [blame] | 238 | print ' } else {' |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 239 | print ' trace::localWriter.writeNull();' |
José Fonseca | fd34e4e | 2011-06-03 19:34:29 +0100 | [diff] [blame] | 240 | print ' }' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 241 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 242 | def visitBlob(self, blob, instance): |
José Fonseca | 30fb4c3 | 2012-11-04 11:07:45 +0000 | [diff] [blame] | 243 | print ' trace::localWriter.writeBlob(%s, %s);' % (instance, self.expand(blob.size)) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 244 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 245 | def visitEnum(self, enum, instance): |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 246 | print ' trace::localWriter.writeEnum(&_enum%s_sig, %s);' % (enum.tag, instance) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 247 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 248 | def visitBitmask(self, bitmask, instance): |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 249 | print ' trace::localWriter.writeBitmask(&_bitmask%s_sig, %s);' % (bitmask.tag, instance) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 250 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 251 | def visitPointer(self, pointer, instance): |
José Fonseca | dbaae49 | 2011-04-21 09:28:10 +0100 | [diff] [blame] | 252 | print ' if (%s) {' % instance |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 253 | print ' trace::localWriter.beginArray(1);' |
| 254 | print ' trace::localWriter.beginElement();' |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 255 | self.visit(pointer.type, "*" + instance) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 256 | print ' trace::localWriter.endElement();' |
| 257 | print ' trace::localWriter.endArray();' |
José Fonseca | fd34e4e | 2011-06-03 19:34:29 +0100 | [diff] [blame] | 258 | print ' } else {' |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 259 | print ' trace::localWriter.writeNull();' |
José Fonseca | fd34e4e | 2011-06-03 19:34:29 +0100 | [diff] [blame] | 260 | print ' }' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 261 | |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 262 | def visitIntPointer(self, pointer, instance): |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 263 | print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 264 | |
José Fonseca | fbcf683 | 2012-04-05 07:10:30 +0100 | [diff] [blame] | 265 | def visitObjPointer(self, pointer, instance): |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 266 | print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance |
José Fonseca | fbcf683 | 2012-04-05 07:10:30 +0100 | [diff] [blame] | 267 | |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 268 | def visitLinearPointer(self, pointer, instance): |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 269 | print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 270 | |
José Fonseca | b89c593 | 2012-04-01 22:47:11 +0200 | [diff] [blame] | 271 | def visitReference(self, reference, instance): |
| 272 | self.visit(reference.type, instance) |
| 273 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 274 | def visitHandle(self, handle, instance): |
José Fonseca | 50d78d8 | 2010-11-23 22:13:14 +0000 | [diff] [blame] | 275 | self.visit(handle.type, instance) |
| 276 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 277 | def visitAlias(self, alias, instance): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 278 | self.visit(alias.type, instance) |
| 279 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 280 | def visitOpaque(self, opaque, instance): |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 281 | print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 282 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 283 | def visitInterface(self, interface, instance): |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 284 | assert False |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 285 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 286 | def visitPolymorphic(self, polymorphic, instance): |
José Fonseca | b95e372 | 2012-04-16 14:01:15 +0100 | [diff] [blame] | 287 | if polymorphic.contextLess: |
| 288 | print ' _write__%s(%s, %s);' % (polymorphic.tag, polymorphic.switchExpr, instance) |
| 289 | else: |
| 290 | print ' switch (%s) {' % polymorphic.switchExpr |
| 291 | for cases, type in polymorphic.iterSwitch(): |
| 292 | for case in cases: |
| 293 | print ' %s:' % case |
| 294 | self.visit(type, 'static_cast<%s>(%s)' % (type, instance)) |
| 295 | print ' break;' |
| 296 | print ' }' |
José Fonseca | 16d46dd | 2011-10-13 09:52:52 +0100 | [diff] [blame] | 297 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 298 | |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 299 | class WrapDecider(stdapi.Traverser): |
| 300 | '''Type visitor which will decide wheter this type will need wrapping or not. |
| 301 | |
| 302 | For complex types (arrays, structures), we need to know this before hand. |
| 303 | ''' |
| 304 | |
| 305 | def __init__(self): |
| 306 | self.needsWrapping = False |
| 307 | |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 308 | def visitLinearPointer(self, void): |
| 309 | pass |
| 310 | |
| 311 | def visitInterface(self, interface): |
| 312 | self.needsWrapping = True |
| 313 | |
| 314 | |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 315 | class ValueWrapper(stdapi.Traverser, ExpanderMixin): |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 316 | '''Type visitor which will generate the code to wrap an instance. |
| 317 | |
| 318 | Wrapping is necessary mostly for interfaces, however interface pointers can |
| 319 | appear anywhere inside complex types. |
| 320 | ''' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 321 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 322 | def visitStruct(self, struct, instance): |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 323 | for type, name in struct.members: |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 324 | self.visitMember(instance, type, "(%s).%s" % (instance, name)) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 325 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 326 | def visitArray(self, array, instance): |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 327 | array_length = self.expand(array.length) |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 328 | print " if (%s) {" % instance |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 329 | print " for (size_t _i = 0, _s = %s; _i < _s; ++_i) {" % array_length |
| 330 | self.visitElement('_i', array.type, instance + "[_i]") |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 331 | print " }" |
| 332 | print " }" |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 333 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 334 | def visitPointer(self, pointer, instance): |
José Fonseca | 3a2a476 | 2011-05-26 11:37:30 +0100 | [diff] [blame] | 335 | print " if (%s) {" % instance |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 336 | self.visit(pointer.type, "*" + instance) |
José Fonseca | 3a2a476 | 2011-05-26 11:37:30 +0100 | [diff] [blame] | 337 | print " }" |
José Fonseca | 59ee88e | 2012-01-15 14:24:10 +0000 | [diff] [blame] | 338 | |
José Fonseca | fbcf683 | 2012-04-05 07:10:30 +0100 | [diff] [blame] | 339 | def visitObjPointer(self, pointer, instance): |
José Fonseca | 9782b29 | 2012-04-14 22:02:42 +0100 | [diff] [blame] | 340 | elem_type = pointer.type.mutable() |
| 341 | if isinstance(elem_type, stdapi.Interface): |
| 342 | self.visitInterfacePointer(elem_type, instance) |
| 343 | else: |
José Fonseca | 467a42a | 2012-05-04 11:49:19 +0100 | [diff] [blame] | 344 | self.visitPointer(pointer, instance) |
José Fonseca | fbcf683 | 2012-04-05 07:10:30 +0100 | [diff] [blame] | 345 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 346 | def visitInterface(self, interface, instance): |
José Fonseca | 9782b29 | 2012-04-14 22:02:42 +0100 | [diff] [blame] | 347 | raise NotImplementedError |
| 348 | |
| 349 | def visitInterfacePointer(self, interface, instance): |
José Fonseca | 3a2a476 | 2011-05-26 11:37:30 +0100 | [diff] [blame] | 350 | print " if (%s) {" % instance |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 351 | print " %s = new %s(%s);" % (instance, getWrapperInterfaceName(interface), instance) |
José Fonseca | 3a2a476 | 2011-05-26 11:37:30 +0100 | [diff] [blame] | 352 | print " }" |
José Fonseca | 16d46dd | 2011-10-13 09:52:52 +0100 | [diff] [blame] | 353 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 354 | def visitPolymorphic(self, type, instance): |
José Fonseca | 16d46dd | 2011-10-13 09:52:52 +0100 | [diff] [blame] | 355 | # XXX: There might be polymorphic values that need wrapping in the future |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 356 | raise NotImplementedError |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 357 | |
| 358 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 359 | class ValueUnwrapper(ValueWrapper): |
| 360 | '''Reverse of ValueWrapper.''' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 361 | |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 362 | allocated = False |
| 363 | |
| 364 | def visitArray(self, array, instance): |
| 365 | if self.allocated or isinstance(instance, stdapi.Interface): |
| 366 | return ValueWrapper.visitArray(self, array, instance) |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 367 | array_length = self.expand(array.length) |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 368 | elem_type = array.type.mutable() |
José Fonseca | f6b0513 | 2012-11-06 00:16:28 +0000 | [diff] [blame^] | 369 | print " if (%s && %s) {" % (instance, array_length) |
| 370 | print " %s * _t = static_cast<%s *>(alloca(%s * sizeof *_t));" % (elem_type, elem_type, array_length) |
| 371 | print " for (size_t _i = 0, _s = %s; _i < _s; ++_i) {" % array_length |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 372 | print " _t[_i] = %s[_i];" % instance |
| 373 | self.allocated = True |
| 374 | self.visit(array.type, "_t[_i]") |
| 375 | print " }" |
| 376 | print " %s = _t;" % instance |
| 377 | print " }" |
| 378 | |
José Fonseca | 9782b29 | 2012-04-14 22:02:42 +0100 | [diff] [blame] | 379 | def visitInterfacePointer(self, interface, instance): |
José Fonseca | 7aa8690 | 2012-01-31 20:14:31 +0000 | [diff] [blame] | 380 | print r' if (%s) {' % instance |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 381 | print r' const %s *pWrapper = static_cast<const %s*>(%s);' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface), instance) |
José Fonseca | 7aa8690 | 2012-01-31 20:14:31 +0000 | [diff] [blame] | 382 | print r' if (pWrapper && pWrapper->m_dwMagic == 0xd8365d6c) {' |
| 383 | print r' %s = pWrapper->m_pInstance;' % (instance,) |
| 384 | print r' } else {' |
| 385 | print r' os::log("apitrace: warning: %%s: unexpected %%s pointer\n", __FUNCTION__, "%s");' % interface.name |
| 386 | print r' }' |
| 387 | print r' }' |
José Fonseca | 87d1cc6 | 2010-11-29 15:57:25 +0000 | [diff] [blame] | 388 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 389 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 390 | class Tracer: |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 391 | '''Base class to orchestrate the code generation of API tracing.''' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 392 | |
José Fonseca | bb8760b | 2011-05-25 23:21:18 +0100 | [diff] [blame] | 393 | def __init__(self): |
| 394 | self.api = None |
| 395 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 396 | def serializerFactory(self): |
| 397 | '''Create a serializer. |
| 398 | |
| 399 | Can be overriden by derived classes to inject their own serialzer. |
| 400 | ''' |
| 401 | |
| 402 | return ValueSerializer() |
| 403 | |
José Fonseca | 1b6c875 | 2012-04-15 14:33:00 +0100 | [diff] [blame] | 404 | def traceApi(self, api): |
José Fonseca | bb8760b | 2011-05-25 23:21:18 +0100 | [diff] [blame] | 405 | self.api = api |
| 406 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 407 | self.header(api) |
| 408 | |
| 409 | # Includes |
| 410 | for header in api.headers: |
José Fonseca | e6a50bd | 2010-11-24 10:12:22 +0000 | [diff] [blame] | 411 | print header |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 412 | print |
| 413 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 414 | # Generate the serializer functions |
José Fonseca | 4470382 | 2012-01-31 10:48:58 +0000 | [diff] [blame] | 415 | types = api.getAllTypes() |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 416 | visitor = ComplexValueSerializer(self.serializerFactory()) |
José Fonseca | e6a50bd | 2010-11-24 10:12:22 +0000 | [diff] [blame] | 417 | map(visitor.visit, types) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 418 | print |
| 419 | |
| 420 | # Interfaces wrapers |
José Fonseca | 143e925 | 2012-04-15 09:31:18 +0100 | [diff] [blame] | 421 | self.traceInterfaces(api) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 422 | |
| 423 | # Function wrappers |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 424 | self.interface = None |
| 425 | self.base = None |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 426 | map(self.traceFunctionDecl, api.functions) |
| 427 | map(self.traceFunctionImpl, api.functions) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 428 | print |
| 429 | |
| 430 | self.footer(api) |
| 431 | |
| 432 | def header(self, api): |
José Fonseca | ba2f08c | 2012-05-10 17:28:31 +0100 | [diff] [blame] | 433 | print '#ifdef _WIN32' |
| 434 | print '# include <malloc.h> // alloca' |
| 435 | print '# ifndef alloca' |
| 436 | print '# define alloca _alloca' |
| 437 | print '# endif' |
| 438 | print '#else' |
| 439 | print '# include <alloca.h> // alloca' |
| 440 | print '#endif' |
José Fonseca | 537c507 | 2012-07-07 12:54:09 +0100 | [diff] [blame] | 441 | print |
| 442 | print '#include "trace.hpp"' |
| 443 | print |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 444 | |
| 445 | def footer(self, api): |
| 446 | pass |
| 447 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 448 | def traceFunctionDecl(self, function): |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 449 | # Per-function declarations |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 450 | |
José Fonseca | 84cea3b | 2012-05-09 21:12:30 +0100 | [diff] [blame] | 451 | if not function.internal: |
| 452 | if function.args: |
| 453 | print 'static const char * _%s_args[%u] = {%s};' % (function.name, len(function.args), ', '.join(['"%s"' % arg.name for arg in function.args])) |
| 454 | else: |
| 455 | print 'static const char ** _%s_args = NULL;' % (function.name,) |
| 456 | print 'static const trace::FunctionSig _%s_sig = {%u, "%s", %u, _%s_args};' % (function.name, function.id, function.name, len(function.args), function.name) |
| 457 | print |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 458 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 459 | def isFunctionPublic(self, function): |
José Fonseca | 2369129 | 2011-04-22 10:40:25 +0100 | [diff] [blame] | 460 | return True |
| 461 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 462 | def traceFunctionImpl(self, function): |
| 463 | if self.isFunctionPublic(function): |
José Fonseca | 2369129 | 2011-04-22 10:40:25 +0100 | [diff] [blame] | 464 | print 'extern "C" PUBLIC' |
| 465 | else: |
| 466 | print 'extern "C" PRIVATE' |
| 467 | print function.prototype() + ' {' |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 468 | if function.type is not stdapi.Void: |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 469 | print ' %s _result;' % function.type |
José Fonseca | 537c507 | 2012-07-07 12:54:09 +0100 | [diff] [blame] | 470 | |
| 471 | # No-op if tracing is disabled |
| 472 | print ' if (!trace::isTracingEnabled()) {' |
Imre Deak | 1242ab5 | 2012-03-30 15:46:26 +0300 | [diff] [blame] | 473 | Tracer.invokeFunction(self, function) |
| 474 | if function.type is not stdapi.Void: |
| 475 | print ' return _result;' |
| 476 | else: |
| 477 | print ' return;' |
| 478 | print ' }' |
José Fonseca | 537c507 | 2012-07-07 12:54:09 +0100 | [diff] [blame] | 479 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 480 | self.traceFunctionImplBody(function) |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 481 | if function.type is not stdapi.Void: |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 482 | print ' return _result;' |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 483 | print '}' |
| 484 | print |
| 485 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 486 | def traceFunctionImplBody(self, function): |
José Fonseca | 84cea3b | 2012-05-09 21:12:30 +0100 | [diff] [blame] | 487 | if not function.internal: |
| 488 | print ' unsigned _call = trace::localWriter.beginEnter(&_%s_sig);' % (function.name,) |
| 489 | for arg in function.args: |
| 490 | if not arg.output: |
| 491 | self.unwrapArg(function, arg) |
| 492 | self.serializeArg(function, arg) |
| 493 | print ' trace::localWriter.endEnter();' |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 494 | self.invokeFunction(function) |
José Fonseca | 84cea3b | 2012-05-09 21:12:30 +0100 | [diff] [blame] | 495 | if not function.internal: |
| 496 | print ' trace::localWriter.beginLeave(_call);' |
| 497 | for arg in function.args: |
| 498 | if arg.output: |
| 499 | self.serializeArg(function, arg) |
| 500 | self.wrapArg(function, arg) |
| 501 | if function.type is not stdapi.Void: |
| 502 | self.serializeRet(function, "_result") |
| 503 | print ' trace::localWriter.endLeave();' |
| 504 | if function.type is not stdapi.Void: |
| 505 | self.wrapRet(function, "_result") |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 506 | |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 507 | def invokeFunction(self, function, prefix='_', suffix=''): |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 508 | if function.type is stdapi.Void: |
| 509 | result = '' |
| 510 | else: |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 511 | result = '_result = ' |
José Fonseca | a08d275 | 2011-08-25 13:26:43 +0100 | [diff] [blame] | 512 | dispatch = prefix + function.name + suffix |
José Fonseca | 14c21bc | 2011-02-20 23:32:22 +0000 | [diff] [blame] | 513 | print ' %s%s(%s);' % (result, dispatch, ', '.join([str(arg.name) for arg in function.args])) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 514 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 515 | def serializeArg(self, function, arg): |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 516 | print ' trace::localWriter.beginArg(%u);' % (arg.index,) |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 517 | self.serializeArgValue(function, arg) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 518 | print ' trace::localWriter.endArg();' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 519 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 520 | def serializeArgValue(self, function, arg): |
| 521 | self.serializeValue(arg.type, arg.name) |
José Fonseca | 9922183 | 2011-03-22 22:15:46 +0000 | [diff] [blame] | 522 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 523 | def wrapArg(self, function, arg): |
José Fonseca | 9782b29 | 2012-04-14 22:02:42 +0100 | [diff] [blame] | 524 | assert not isinstance(arg.type, stdapi.ObjPointer) |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 525 | |
| 526 | from specs.winapi import REFIID |
| 527 | riid = None |
| 528 | for other_arg in function.args: |
| 529 | if not other_arg.output and other_arg.type is REFIID: |
| 530 | riid = other_arg |
José Fonseca | c328b8c | 2012-04-28 21:45:38 +0100 | [diff] [blame] | 531 | if riid is not None \ |
| 532 | and isinstance(arg.type, stdapi.Pointer) \ |
| 533 | and isinstance(arg.type.type, stdapi.ObjPointer): |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 534 | self.wrapIid(function, riid, arg) |
| 535 | return |
| 536 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 537 | self.wrapValue(arg.type, arg.name) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 538 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 539 | def unwrapArg(self, function, arg): |
| 540 | self.unwrapValue(arg.type, arg.name) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 541 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 542 | def serializeRet(self, function, instance): |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 543 | print ' trace::localWriter.beginReturn();' |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 544 | self.serializeValue(function.type, instance) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 545 | print ' trace::localWriter.endReturn();' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 546 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 547 | def serializeValue(self, type, instance): |
| 548 | serializer = self.serializerFactory() |
| 549 | serializer.visit(type, instance) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 550 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 551 | def wrapRet(self, function, instance): |
| 552 | self.wrapValue(function.type, instance) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 553 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 554 | def unwrapRet(self, function, instance): |
| 555 | self.unwrapValue(function.type, instance) |
| 556 | |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 557 | def needsWrapping(self, type): |
| 558 | visitor = WrapDecider() |
| 559 | visitor.visit(type) |
| 560 | return visitor.needsWrapping |
| 561 | |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 562 | def wrapValue(self, type, instance): |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 563 | if self.needsWrapping(type): |
| 564 | visitor = ValueWrapper() |
| 565 | visitor.visit(type, instance) |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 566 | |
| 567 | def unwrapValue(self, type, instance): |
José Fonseca | 0075f15 | 2012-04-14 20:25:52 +0100 | [diff] [blame] | 568 | if self.needsWrapping(type): |
| 569 | visitor = ValueUnwrapper() |
| 570 | visitor.visit(type, instance) |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 571 | |
José Fonseca | 143e925 | 2012-04-15 09:31:18 +0100 | [diff] [blame] | 572 | def traceInterfaces(self, api): |
| 573 | interfaces = api.getAllInterfaces() |
| 574 | if not interfaces: |
| 575 | return |
| 576 | map(self.declareWrapperInterface, interfaces) |
| 577 | self.implementIidWrapper(api) |
| 578 | map(self.implementWrapperInterface, interfaces) |
| 579 | print |
| 580 | |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 581 | def declareWrapperInterface(self, interface): |
| 582 | print "class %s : public %s " % (getWrapperInterfaceName(interface), interface.name) |
| 583 | print "{" |
| 584 | print "public:" |
| 585 | print " %s(%s * pInstance);" % (getWrapperInterfaceName(interface), interface.name) |
| 586 | print " virtual ~%s();" % getWrapperInterfaceName(interface) |
| 587 | print |
| 588 | for method in interface.iterMethods(): |
| 589 | print " " + method.prototype() + ";" |
| 590 | print |
José Fonseca | acc9062 | 2012-05-02 13:10:07 +0100 | [diff] [blame] | 591 | #print "private:" |
| 592 | for type, name, value in self.enumWrapperInterfaceVariables(interface): |
| 593 | print ' %s %s;' % (type, name) |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 594 | print "};" |
| 595 | print |
| 596 | |
José Fonseca | acc9062 | 2012-05-02 13:10:07 +0100 | [diff] [blame] | 597 | def enumWrapperInterfaceVariables(self, interface): |
| 598 | return [ |
| 599 | ("DWORD", "m_dwMagic", "0xd8365d6c"), |
| 600 | ("%s *" % interface.name, "m_pInstance", "pInstance"), |
| 601 | ] |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 602 | |
| 603 | def implementWrapperInterface(self, interface): |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 604 | self.interface = interface |
| 605 | |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 606 | print '%s::%s(%s * pInstance) {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface), interface.name) |
José Fonseca | acc9062 | 2012-05-02 13:10:07 +0100 | [diff] [blame] | 607 | for type, name, value in self.enumWrapperInterfaceVariables(interface): |
| 608 | print ' %s = %s;' % (name, value) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 609 | print '}' |
| 610 | print |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 611 | print '%s::~%s() {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface)) |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 612 | print '}' |
| 613 | print |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 614 | |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 615 | for base, method in interface.iterBaseMethods(): |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 616 | self.base = base |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 617 | self.implementWrapperInterfaceMethod(interface, base, method) |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 618 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 619 | print |
| 620 | |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 621 | def implementWrapperInterfaceMethod(self, interface, base, method): |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 622 | print method.prototype(getWrapperInterfaceName(interface) + '::' + method.name) + ' {' |
| 623 | if method.type is not stdapi.Void: |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 624 | print ' %s _result;' % method.type |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 625 | |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 626 | self.implementWrapperInterfaceMethodBody(interface, base, method) |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 627 | |
| 628 | if method.type is not stdapi.Void: |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 629 | print ' return _result;' |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 630 | print '}' |
| 631 | print |
| 632 | |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 633 | def implementWrapperInterfaceMethodBody(self, interface, base, method): |
José Fonseca | 84cea3b | 2012-05-09 21:12:30 +0100 | [diff] [blame] | 634 | assert not method.internal |
| 635 | |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 636 | print ' static const char * _args[%u] = {%s};' % (len(method.args) + 1, ', '.join(['"this"'] + ['"%s"' % arg.name for arg in method.args])) |
| 637 | print ' static const trace::FunctionSig _sig = {%u, "%s", %u, _args};' % (method.id, interface.name + '::' + method.name, len(method.args) + 1) |
José Fonseca | a0e9786 | 2012-04-29 23:22:52 +0100 | [diff] [blame] | 638 | |
| 639 | print ' %s *_this = static_cast<%s *>(m_pInstance);' % (base, base) |
| 640 | |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 641 | print ' unsigned _call = trace::localWriter.beginEnter(&_sig);' |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 642 | print ' trace::localWriter.beginArg(0);' |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 643 | print ' trace::localWriter.writePointer((uintptr_t)m_pInstance);' |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 644 | print ' trace::localWriter.endArg();' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 645 | for arg in method.args: |
| 646 | if not arg.output: |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 647 | self.unwrapArg(method, arg) |
| 648 | self.serializeArg(method, arg) |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 649 | print ' trace::localWriter.endEnter();' |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 650 | |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 651 | self.invokeMethod(interface, base, method) |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 652 | |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 653 | print ' trace::localWriter.beginLeave(_call);' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 654 | for arg in method.args: |
| 655 | if arg.output: |
José Fonseca | 54f304a | 2012-01-14 19:33:08 +0000 | [diff] [blame] | 656 | self.serializeArg(method, arg) |
| 657 | self.wrapArg(method, arg) |
José Fonseca | 1ce52f0 | 2012-01-31 12:19:57 +0000 | [diff] [blame] | 658 | |
José Fonseca | 87d1cc6 | 2010-11-29 15:57:25 +0000 | [diff] [blame] | 659 | if method.type is not stdapi.Void: |
José Fonseca | 3a259b8 | 2012-05-09 19:33:33 +0100 | [diff] [blame] | 660 | self.serializeRet(method, '_result') |
José Fonseca | b4a3d14 | 2011-10-27 07:43:19 +0100 | [diff] [blame] | 661 | print ' trace::localWriter.endLeave();' |
José Fonseca | 3a259b8 | 2012-05-09 19:33:33 +0100 | [diff] [blame] | 662 | if method.type is not stdapi.Void: |
| 663 | self.wrapRet(method, '_result') |
| 664 | |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 665 | if method.name == 'Release': |
José Fonseca | 87d1cc6 | 2010-11-29 15:57:25 +0000 | [diff] [blame] | 666 | assert method.type is not stdapi.Void |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 667 | print ' if (!_result)' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 668 | print ' delete this;' |
José Fonseca | 8fbdd3a | 2010-11-23 20:55:07 +0000 | [diff] [blame] | 669 | |
José Fonseca | 143e925 | 2012-04-15 09:31:18 +0100 | [diff] [blame] | 670 | def implementIidWrapper(self, api): |
| 671 | print r'static void' |
| 672 | print r'warnIID(const char *functionName, REFIID riid, const char *reason) {' |
| 673 | print r' os::log("apitrace: warning: %s: %s IID {0x%08lX,0x%04X,0x%04X,{0x%02X,0x%02X,0x%02X,0x%02X,0x%02X,0x%02X,0x%02X,0x%02X}}\n",' |
| 674 | print r' functionName, reason,' |
| 675 | print r' riid.Data1, riid.Data2, riid.Data3,' |
| 676 | print r' riid.Data4[0], riid.Data4[1], riid.Data4[2], riid.Data4[3], riid.Data4[4], riid.Data4[5], riid.Data4[6], riid.Data4[7]);' |
| 677 | print r'}' |
| 678 | print |
| 679 | print r'static void' |
| 680 | print r'wrapIID(const char *functionName, REFIID riid, void * * ppvObj) {' |
| 681 | print r' if (!ppvObj || !*ppvObj) {' |
| 682 | print r' return;' |
| 683 | print r' }' |
| 684 | else_ = '' |
| 685 | for iface in api.getAllInterfaces(): |
| 686 | print r' %sif (riid == IID_%s) {' % (else_, iface.name) |
| 687 | print r' *ppvObj = new Wrap%s((%s *) *ppvObj);' % (iface.name, iface.name) |
| 688 | print r' }' |
| 689 | else_ = 'else ' |
| 690 | print r' %s{' % else_ |
| 691 | print r' warnIID(functionName, riid, "unknown");' |
| 692 | print r' }' |
| 693 | print r'}' |
| 694 | print |
| 695 | |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 696 | def wrapIid(self, function, riid, out): |
José Fonseca | c328b8c | 2012-04-28 21:45:38 +0100 | [diff] [blame] | 697 | # Cast output arg to `void **` if necessary |
| 698 | out_name = out.name |
| 699 | obj_type = out.type.type.type |
| 700 | if not obj_type is stdapi.Void: |
| 701 | assert isinstance(obj_type, stdapi.Interface) |
| 702 | out_name = 'reinterpret_cast<void * *>(%s)' % out_name |
| 703 | |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 704 | print r' if (%s && *%s) {' % (out.name, out.name) |
José Fonseca | 143e925 | 2012-04-15 09:31:18 +0100 | [diff] [blame] | 705 | functionName = function.name |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 706 | else_ = '' |
| 707 | if self.interface is not None: |
José Fonseca | 143e925 | 2012-04-15 09:31:18 +0100 | [diff] [blame] | 708 | functionName = self.interface.name + '::' + functionName |
José Fonseca | c328b8c | 2012-04-28 21:45:38 +0100 | [diff] [blame] | 709 | print r' if (*%s == m_pInstance &&' % (out_name,) |
José Fonseca | 828bf10 | 2012-04-16 20:48:59 +0100 | [diff] [blame] | 710 | print r' (%s)) {' % ' || '.join('%s == IID_%s' % (riid.name, iface.name) for iface in self.interface.iterBases()) |
José Fonseca | c328b8c | 2012-04-28 21:45:38 +0100 | [diff] [blame] | 711 | print r' *%s = this;' % (out_name,) |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 712 | print r' }' |
| 713 | else_ = 'else ' |
| 714 | print r' %s{' % else_ |
José Fonseca | c328b8c | 2012-04-28 21:45:38 +0100 | [diff] [blame] | 715 | print r' wrapIID("%s", %s, %s);' % (functionName, riid.name, out_name) |
José Fonseca | bcb26b2 | 2012-04-15 08:42:25 +0100 | [diff] [blame] | 716 | print r' }' |
| 717 | print r' }' |
José Fonseca | 1ce52f0 | 2012-01-31 12:19:57 +0000 | [diff] [blame] | 718 | |
José Fonseca | 4220b1b | 2012-02-03 19:05:29 +0000 | [diff] [blame] | 719 | def invokeMethod(self, interface, base, method): |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 720 | if method.type is stdapi.Void: |
| 721 | result = '' |
| 722 | else: |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 723 | result = '_result = ' |
José Fonseca | a0e9786 | 2012-04-29 23:22:52 +0100 | [diff] [blame] | 724 | print ' %s_this->%s(%s);' % (result, method.name, ', '.join([str(arg.name) for arg in method.args])) |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 725 | |
| 726 | def emit_memcpy(self, dest, src, length): |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 727 | print ' unsigned _call = trace::localWriter.beginEnter(&trace::memcpy_sig);' |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 728 | print ' trace::localWriter.beginArg(0);' |
José Fonseca | d559f02 | 2012-04-15 16:13:51 +0100 | [diff] [blame] | 729 | print ' trace::localWriter.writePointer((uintptr_t)%s);' % dest |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 730 | print ' trace::localWriter.endArg();' |
| 731 | print ' trace::localWriter.beginArg(1);' |
| 732 | print ' trace::localWriter.writeBlob(%s, %s);' % (src, length) |
| 733 | print ' trace::localWriter.endArg();' |
| 734 | print ' trace::localWriter.beginArg(2);' |
| 735 | print ' trace::localWriter.writeUInt(%s);' % length |
| 736 | print ' trace::localWriter.endArg();' |
| 737 | print ' trace::localWriter.endEnter();' |
José Fonseca | 632a78d | 2012-04-19 07:18:59 +0100 | [diff] [blame] | 738 | print ' trace::localWriter.beginLeave(_call);' |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 739 | print ' trace::localWriter.endLeave();' |
José Fonseca | 122c987 | 2012-08-02 08:42:24 +0100 | [diff] [blame] | 740 | |
| 741 | def fake_call(self, function, args): |
| 742 | print ' unsigned _fake_call = trace::localWriter.beginEnter(&_%s_sig);' % (function.name,) |
| 743 | for arg, instance in zip(function.args, args): |
| 744 | assert not arg.output |
| 745 | print ' trace::localWriter.beginArg(%u);' % (arg.index,) |
| 746 | self.serializeValue(arg.type, instance) |
| 747 | print ' trace::localWriter.endArg();' |
| 748 | print ' trace::localWriter.endEnter();' |
| 749 | print ' trace::localWriter.beginLeave(_fake_call);' |
| 750 | print ' trace::localWriter.endLeave();' |
José Fonseca | 0423d2c | 2012-01-20 19:16:17 +0000 | [diff] [blame] | 751 | |