X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=wrappers%2Ftrace.py;h=aab99d793c1ce433bf6ad19d97d7afd89bfbac15;hb=1242ab5cbdc409e2fc2d787edc28c6ac2a8439d1;hp=6cb7934bfc684e4c153cca24fd3ef7b05561befe;hpb=1b6c87507810fe2d72c81fc109110c1ca25ce94f;p=apitrace diff --git a/wrappers/trace.py b/wrappers/trace.py index 6cb7934..aab99d7 100644 --- a/wrappers/trace.py +++ b/wrappers/trace.py @@ -87,24 +87,24 @@ class ComplexValueSerializer(stdapi.OnceVisitor): pass def visitEnum(self, enum): - print 'static const trace::EnumValue __enum%s_values[] = {' % (enum.tag) + print 'static const trace::EnumValue _enum%s_values[] = {' % (enum.tag) for value in enum.values: print ' {"%s", %s},' % (value, value) print '};' print - print 'static const trace::EnumSig __enum%s_sig = {' % (enum.tag) - print ' %u, %u, __enum%s_values' % (enum.id, len(enum.values), enum.tag) + print 'static const trace::EnumSig _enum%s_sig = {' % (enum.tag) + print ' %u, %u, _enum%s_values' % (enum.id, len(enum.values), enum.tag) print '};' print def visitBitmask(self, bitmask): - print 'static const trace::BitmaskFlag __bitmask%s_flags[] = {' % (bitmask.tag) + print 'static const trace::BitmaskFlag _bitmask%s_flags[] = {' % (bitmask.tag) for value in bitmask.values: print ' {"%s", %s},' % (value, value) print '};' print - print 'static const trace::BitmaskSig __bitmask%s_sig = {' % (bitmask.tag) - print ' %u, %u, __bitmask%s_flags' % (bitmask.id, len(bitmask.values), bitmask.tag) + print 'static const trace::BitmaskSig _bitmask%s_sig = {' % (bitmask.tag) + print ' %u, %u, _bitmask%s_flags' % (bitmask.id, len(bitmask.values), bitmask.tag) print '};' print @@ -136,6 +136,8 @@ class ComplexValueSerializer(stdapi.OnceVisitor): pass def visitPolymorphic(self, polymorphic): + if not polymorphic.contextLess: + return print 'static void _write__%s(int selector, const %s & value) {' % (polymorphic.tag, polymorphic.expr) print ' switch (selector) {' for cases, type in polymorphic.iterSwitch(): @@ -182,10 +184,10 @@ class ValueSerializer(stdapi.Visitor): print ' _write__%s(%s);' % (struct.tag, instance) def visitArray(self, array, instance): - length = '__c' + array.type.tag - index = '__i' + array.type.tag + length = '_c' + array.type.tag + index = '_i' + array.type.tag print ' if (%s) {' % instance - print ' size_t %s = %s;' % (length, array.length) + print ' size_t %s = %s > 0 ? %s : 0;' % (length, array.length, array.length) print ' trace::localWriter.beginArray(%s);' % length print ' for (size_t %s = 0; %s < %s; ++%s) {' % (index, index, length, index) print ' trace::localWriter.beginElement();' @@ -201,10 +203,10 @@ class ValueSerializer(stdapi.Visitor): print ' trace::localWriter.writeBlob(%s, %s);' % (instance, blob.size) def visitEnum(self, enum, instance): - print ' trace::localWriter.writeEnum(&__enum%s_sig, %s);' % (enum.tag, instance) + print ' trace::localWriter.writeEnum(&_enum%s_sig, %s);' % (enum.tag, instance) def visitBitmask(self, bitmask, instance): - print ' trace::localWriter.writeBitmask(&__bitmask%s_sig, %s);' % (bitmask.tag, instance) + print ' trace::localWriter.writeBitmask(&_bitmask%s_sig, %s);' % (bitmask.tag, instance) def visitPointer(self, pointer, instance): print ' if (%s) {' % instance @@ -218,13 +220,13 @@ class ValueSerializer(stdapi.Visitor): print ' }' def visitIntPointer(self, pointer, instance): - print ' trace::localWriter.writeOpaque((const void *)%s);' % instance + print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance def visitObjPointer(self, pointer, instance): - print ' trace::localWriter.writeOpaque((const void *)%s);' % instance + print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance def visitLinearPointer(self, pointer, instance): - print ' trace::localWriter.writeOpaque((const void *)%s);' % instance + print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance def visitReference(self, reference, instance): self.visit(reference.type, instance) @@ -236,13 +238,22 @@ class ValueSerializer(stdapi.Visitor): self.visit(alias.type, instance) def visitOpaque(self, opaque, instance): - print ' trace::localWriter.writeOpaque((const void *)%s);' % instance + print ' trace::localWriter.writePointer((uintptr_t)%s);' % instance def visitInterface(self, interface, instance): - print ' trace::localWriter.writeOpaque((const void *)&%s);' % instance + assert False def visitPolymorphic(self, polymorphic, instance): - print ' _write__%s(%s, %s);' % (polymorphic.tag, polymorphic.switchExpr, instance) + if polymorphic.contextLess: + print ' _write__%s(%s, %s);' % (polymorphic.tag, polymorphic.switchExpr, instance) + else: + print ' switch (%s) {' % polymorphic.switchExpr + for cases, type in polymorphic.iterSwitch(): + for case in cases: + print ' %s:' % case + self.visit(type, 'static_cast<%s>(%s)' % (type, instance)) + print ' break;' + print ' }' class WrapDecider(stdapi.Traverser): @@ -289,7 +300,7 @@ class ValueWrapper(stdapi.Traverser): if isinstance(elem_type, stdapi.Interface): self.visitInterfacePointer(elem_type, instance) else: - self.visitPointer(self, pointer, instance) + self.visitPointer(pointer, instance) def visitInterface(self, interface, instance): raise NotImplementedError @@ -377,7 +388,14 @@ class Tracer: self.footer(api) def header(self, api): - pass + print '#ifdef _WIN32' + print '# include // alloca' + print '# ifndef alloca' + print '# define alloca _alloca' + print '# endif' + print '#else' + print '# include // alloca' + print '#endif' def footer(self, api): pass @@ -385,12 +403,13 @@ class Tracer: def traceFunctionDecl(self, function): # Per-function declarations - if function.args: - print 'static const char * __%s_args[%u] = {%s};' % (function.name, len(function.args), ', '.join(['"%s"' % arg.name for arg in function.args])) - else: - print 'static const char ** __%s_args = NULL;' % (function.name,) - print 'static const trace::FunctionSig __%s_sig = {%u, "%s", %u, __%s_args};' % (function.name, function.id, function.name, len(function.args), function.name) - print + if not function.internal: + if function.args: + print 'static const char * _%s_args[%u] = {%s};' % (function.name, len(function.args), ', '.join(['"%s"' % arg.name for arg in function.args])) + else: + print 'static const char ** _%s_args = NULL;' % (function.name,) + print 'static const trace::FunctionSig _%s_sig = {%u, "%s", %u, _%s_args};' % (function.name, function.id, function.name, len(function.args), function.name) + print def isFunctionPublic(self, function): return True @@ -402,36 +421,47 @@ class Tracer: print 'extern "C" PRIVATE' print function.prototype() + ' {' if function.type is not stdapi.Void: - print ' %s __result;' % function.type + print ' %s _result;' % function.type + print ' if (!os::apitrace_enabled()) {' + Tracer.invokeFunction(self, function) + if function.type is not stdapi.Void: + print ' return _result;' + else: + print ' return;' + print ' }' + print self.traceFunctionImplBody(function) if function.type is not stdapi.Void: - self.wrapRet(function, "__result") - print ' return __result;' + print ' return _result;' print '}' print def traceFunctionImplBody(self, function): - print ' unsigned __call = trace::localWriter.beginEnter(&__%s_sig);' % (function.name,) - for arg in function.args: - if not arg.output: - self.unwrapArg(function, arg) - self.serializeArg(function, arg) - print ' trace::localWriter.endEnter();' + if not function.internal: + print ' unsigned _call = trace::localWriter.beginEnter(&_%s_sig);' % (function.name,) + for arg in function.args: + if not arg.output: + self.unwrapArg(function, arg) + self.serializeArg(function, arg) + print ' trace::localWriter.endEnter();' self.invokeFunction(function) - print ' trace::localWriter.beginLeave(__call);' - for arg in function.args: - if arg.output: - self.serializeArg(function, arg) - self.wrapArg(function, arg) - if function.type is not stdapi.Void: - self.serializeRet(function, "__result") - print ' trace::localWriter.endLeave();' - - def invokeFunction(self, function, prefix='__', suffix=''): + if not function.internal: + print ' trace::localWriter.beginLeave(_call);' + for arg in function.args: + if arg.output: + self.serializeArg(function, arg) + self.wrapArg(function, arg) + if function.type is not stdapi.Void: + self.serializeRet(function, "_result") + print ' trace::localWriter.endLeave();' + if function.type is not stdapi.Void: + self.wrapRet(function, "_result") + + def invokeFunction(self, function, prefix='_', suffix=''): if function.type is stdapi.Void: result = '' else: - result = '__result = ' + result = '_result = ' dispatch = prefix + function.name + suffix print ' %s%s(%s);' % (result, dispatch, ', '.join([str(arg.name) for arg in function.args])) @@ -451,10 +481,9 @@ class Tracer: for other_arg in function.args: if not other_arg.output and other_arg.type is REFIID: riid = other_arg - if riid is not None and isinstance(arg.type, stdapi.Pointer): - assert isinstance(arg.type.type, stdapi.ObjPointer) - obj_type = arg.type.type.type - assert obj_type is stdapi.Void + if riid is not None \ + and isinstance(arg.type, stdapi.Pointer) \ + and isinstance(arg.type.type, stdapi.ObjPointer): self.wrapIid(function, riid, arg) return @@ -512,21 +541,24 @@ class Tracer: for method in interface.iterMethods(): print " " + method.prototype() + ";" print - self.declareWrapperInterfaceVariables(interface) + #print "private:" + for type, name, value in self.enumWrapperInterfaceVariables(interface): + print ' %s %s;' % (type, name) print "};" print - def declareWrapperInterfaceVariables(self, interface): - #print "private:" - print " DWORD m_dwMagic;" - print " %s * m_pInstance;" % (interface.name,) + def enumWrapperInterfaceVariables(self, interface): + return [ + ("DWORD", "m_dwMagic", "0xd8365d6c"), + ("%s *" % interface.name, "m_pInstance", "pInstance"), + ] def implementWrapperInterface(self, interface): self.interface = interface print '%s::%s(%s * pInstance) {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface), interface.name) - print ' m_dwMagic = 0xd8365d6c;' - print ' m_pInstance = pInstance;' + for type, name, value in self.enumWrapperInterfaceVariables(interface): + print ' %s = %s;' % (name, value) print '}' print print '%s::~%s() {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface)) @@ -542,21 +574,26 @@ class Tracer: def implementWrapperInterfaceMethod(self, interface, base, method): print method.prototype(getWrapperInterfaceName(interface) + '::' + method.name) + ' {' if method.type is not stdapi.Void: - print ' %s __result;' % method.type + print ' %s _result;' % method.type self.implementWrapperInterfaceMethodBody(interface, base, method) if method.type is not stdapi.Void: - print ' return __result;' + print ' return _result;' print '}' print def implementWrapperInterfaceMethodBody(self, interface, base, method): - print ' static const char * __args[%u] = {%s};' % (len(method.args) + 1, ', '.join(['"this"'] + ['"%s"' % arg.name for arg in method.args])) - print ' static const trace::FunctionSig __sig = {%u, "%s", %u, __args};' % (method.id, interface.name + '::' + method.name, len(method.args) + 1) - print ' unsigned __call = trace::localWriter.beginEnter(&__sig);' + assert not method.internal + + print ' static const char * _args[%u] = {%s};' % (len(method.args) + 1, ', '.join(['"this"'] + ['"%s"' % arg.name for arg in method.args])) + print ' static const trace::FunctionSig _sig = {%u, "%s", %u, _args};' % (method.id, interface.name + '::' + method.name, len(method.args) + 1) + + print ' %s *_this = static_cast<%s *>(m_pInstance);' % (base, base) + + print ' unsigned _call = trace::localWriter.beginEnter(&_sig);' print ' trace::localWriter.beginArg(0);' - print ' trace::localWriter.writeOpaque((const void *)m_pInstance);' + print ' trace::localWriter.writePointer((uintptr_t)m_pInstance);' print ' trace::localWriter.endArg();' for arg in method.args: if not arg.output: @@ -566,21 +603,21 @@ class Tracer: self.invokeMethod(interface, base, method) - print ' trace::localWriter.beginLeave(__call);' + print ' trace::localWriter.beginLeave(_call);' for arg in method.args: if arg.output: self.serializeArg(method, arg) self.wrapArg(method, arg) if method.type is not stdapi.Void: - print ' trace::localWriter.beginReturn();' - self.serializeValue(method.type, "__result") - print ' trace::localWriter.endReturn();' - self.wrapValue(method.type, '__result') + self.serializeRet(method, '_result') print ' trace::localWriter.endLeave();' + if method.type is not stdapi.Void: + self.wrapRet(method, '_result') + if method.name == 'Release': assert method.type is not stdapi.Void - print ' if (!__result)' + print ' if (!_result)' print ' delete this;' def implementIidWrapper(self, api): @@ -610,20 +647,25 @@ class Tracer: print def wrapIid(self, function, riid, out): + # Cast output arg to `void **` if necessary + out_name = out.name + obj_type = out.type.type.type + if not obj_type is stdapi.Void: + assert isinstance(obj_type, stdapi.Interface) + out_name = 'reinterpret_cast(%s)' % out_name + print r' if (%s && *%s) {' % (out.name, out.name) functionName = function.name else_ = '' if self.interface is not None: functionName = self.interface.name + '::' + functionName - print r' %sif (*%s == m_pInstance) {' % (else_, out.name,) - print r' *%s = this;' % (out.name,) - print r' if (%s) {' % ' && '.join('%s != IID_%s' % (riid.name, iface.name) for iface in self.interface.iterBases()) - print r' warnIID("%s", %s, "unexpected");' % (functionName, riid.name) - print r' }' + print r' if (*%s == m_pInstance &&' % (out_name,) + print r' (%s)) {' % ' || '.join('%s == IID_%s' % (riid.name, iface.name) for iface in self.interface.iterBases()) + print r' *%s = this;' % (out_name,) print r' }' else_ = 'else ' print r' %s{' % else_ - print r' wrapIID("%s", %s, %s);' % (functionName, riid.name, out.name) + print r' wrapIID("%s", %s, %s);' % (functionName, riid.name, out_name) print r' }' print r' }' @@ -631,13 +673,13 @@ class Tracer: if method.type is stdapi.Void: result = '' else: - result = '__result = ' - print ' %sstatic_cast<%s *>(m_pInstance)->%s(%s);' % (result, base, method.name, ', '.join([str(arg.name) for arg in method.args])) + result = '_result = ' + print ' %s_this->%s(%s);' % (result, method.name, ', '.join([str(arg.name) for arg in method.args])) def emit_memcpy(self, dest, src, length): - print ' unsigned __call = trace::localWriter.beginEnter(&trace::memcpy_sig);' + print ' unsigned _call = trace::localWriter.beginEnter(&trace::memcpy_sig);' print ' trace::localWriter.beginArg(0);' - print ' trace::localWriter.writeOpaque(%s);' % dest + print ' trace::localWriter.writePointer((uintptr_t)%s);' % dest print ' trace::localWriter.endArg();' print ' trace::localWriter.beginArg(1);' print ' trace::localWriter.writeBlob(%s, %s);' % (src, length) @@ -646,6 +688,6 @@ class Tracer: print ' trace::localWriter.writeUInt(%s);' % length print ' trace::localWriter.endArg();' print ' trace::localWriter.endEnter();' - print ' trace::localWriter.beginLeave(__call);' + print ' trace::localWriter.beginLeave(_call);' print ' trace::localWriter.endLeave();'