"""Common trace code generation."""
-import stdapi
-from dispatch import Dispatcher
+import specs.stdapi as stdapi
-def interface_wrap_name(interface):
+def getWrapperInterfaceName(interface):
return "Wrap" + interface.expr
-class DumpDeclarator(stdapi.OnceVisitor):
- '''Declare helper functions to dump complex types.'''
+class ComplexValueSerializer(stdapi.OnceVisitor):
+ '''Type visitors which generates serialization functions for
+ complex types.
+
+ Simple types are serialized inline.
+ '''
+
+ def __init__(self, serializer):
+ stdapi.OnceVisitor.__init__(self)
+ self.serializer = serializer
- def visit_void(self, literal):
+ def visitVoid(self, literal):
pass
- def visit_literal(self, literal):
+ def visitLiteral(self, literal):
pass
- def visit_string(self, string):
+ def visitString(self, string):
pass
- def visit_const(self, const):
+ def visitConst(self, const):
self.visit(const.type)
- def visit_struct(self, struct):
+ def visitStruct(self, struct):
for type, name in struct.members:
self.visit(type)
- print 'static void __traceStruct%s(const %s &value) {' % (struct.id, struct.expr)
+ print 'static void _write__%s(const %s &value) {' % (struct.tag, struct.expr)
print ' static const char * members[%u] = {' % (len(struct.members),)
for type, name, in struct.members:
print ' "%s",' % (name,)
print ' };'
- print ' static const Trace::StructSig sig = {'
- print ' %u, "%s", %u, members' % (int(struct.id), struct.name, len(struct.members))
+ print ' static const trace::StructSig sig = {'
+ print ' %u, "%s", %u, members' % (struct.id, struct.name, len(struct.members))
print ' };'
- print ' Trace::BeginStruct(&sig);'
+ print ' trace::localWriter.beginStruct(&sig);'
for type, name in struct.members:
- dump_instance(type, 'value.%s' % (name,))
- print ' Trace::EndStruct();'
+ self.serializer.visit(type, 'value.%s' % (name,))
+ print ' trace::localWriter.endStruct();'
print '}'
print
- def visit_array(self, array):
+ def visitArray(self, array):
self.visit(array.type)
- def visit_blob(self, array):
+ def visitBlob(self, array):
pass
- def visit_enum(self, enum):
- print 'static void __traceEnum%s(const %s value) {' % (enum.id, enum.expr)
- n = len(enum.values)
- for i in range(n):
- value = enum.values[i]
- print ' static const Trace::EnumSig sig%u = {%u, "%s", %s};' % (i, enum.vid + i, value, value)
- print ' const Trace::EnumSig *sig;'
- print ' switch(value) {'
- for i in range(n):
- value = enum.values[i]
- print ' case %s:' % value
- print ' sig = &sig%u;' % i
- print ' break;'
- print ' default:'
- print ' Trace::LiteralSInt(value);'
- print ' return;'
- print ' }'
- print ' Trace::LiteralEnum(sig);'
- print '}'
+ def visitEnum(self, enum):
+ 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 '};'
print
- def visit_bitmask(self, bitmask):
- print 'static const Trace::BitmaskVal __bitmask%s_vals[] = {' % (bitmask.id)
+ def visitBitmask(self, bitmask):
+ 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.id)
- print ' %u, %u, __bitmask%s_vals' % (int(bitmask.id), len(bitmask.values), bitmask.id)
+ 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
- def visit_pointer(self, pointer):
+ def visitPointer(self, pointer):
+ self.visit(pointer.type)
+
+ def visitIntPointer(self, pointer):
+ pass
+
+ def visitLinearPointer(self, pointer):
self.visit(pointer.type)
- def visit_handle(self, handle):
+ def visitHandle(self, handle):
self.visit(handle.type)
- def visit_alias(self, alias):
+ def visitAlias(self, alias):
self.visit(alias.type)
- def visit_opaque(self, opaque):
+ def visitOpaque(self, opaque):
pass
- def visit_interface(self, interface):
- print "class %s : public %s " % (interface_wrap_name(interface), interface.name)
- print "{"
- print "public:"
- print " %s(%s * pInstance);" % (interface_wrap_name(interface), interface.name)
- print " virtual ~%s();" % interface_wrap_name(interface)
- print
- for method in interface.itermethods():
- print " " + method.prototype() + ";"
- print
- #print "private:"
- print " %s * m_pInstance;" % (interface.name,)
- print "};"
- print
-
+ def visitInterface(self, interface):
+ pass
-class DumpImplementer(stdapi.Visitor):
- '''Dump an instance.'''
+ def visitPolymorphic(self, polymorphic):
+ print 'static void _write__%s(int selector, const %s & value) {' % (polymorphic.tag, polymorphic.expr)
+ print ' switch (selector) {'
+ for cases, type in polymorphic.iterSwitch():
+ for case in cases:
+ print ' %s:' % case
+ self.serializer.visit(type, 'static_cast<%s>(value)' % (type,))
+ print ' break;'
+ print ' }'
+ print '}'
+ print
- def visit_literal(self, literal, instance):
- print ' Trace::Literal%s(%s);' % (literal.format, instance)
- def visit_string(self, string, instance):
+class ValueSerializer(stdapi.Visitor):
+ '''Visitor which generates code to serialize any type.
+
+ Simple types are serialized inline here, whereas the serialization of
+ complex types is dispatched to the serialization functions generated by
+ ComplexValueSerializer visitor above.
+ '''
+
+ def visitLiteral(self, literal, instance):
+ print ' trace::localWriter.write%s(%s);' % (literal.kind, instance)
+
+ def visitString(self, string, instance):
+ if string.kind == 'String':
+ cast = 'const char *'
+ elif string.kind == 'WString':
+ cast = 'const wchar_t *'
+ else:
+ assert False
+ if cast != string.expr:
+ # reinterpret_cast is necessary for GLubyte * <=> char *
+ instance = 'reinterpret_cast<%s>(%s)' % (cast, instance)
if string.length is not None:
- print ' Trace::LiteralString((const char *)%s, %s);' % (instance, string.length)
+ length = ', %s' % string.length
else:
- print ' Trace::LiteralString((const char *)%s);' % instance
+ length = ''
+ print ' trace::localWriter.write%s(%s%s);' % (string.kind, instance, length)
- def visit_const(self, const, instance):
+ def visitConst(self, const, instance):
self.visit(const.type, instance)
- def visit_struct(self, struct, instance):
- print ' __traceStruct%s(%s);' % (struct.id, instance)
+ def visitStruct(self, struct, instance):
+ print ' _write__%s(%s);' % (struct.tag, instance)
- def visit_array(self, array, instance):
+ def visitArray(self, array, instance):
+ length = '__c' + array.type.tag
+ index = '__i' + array.type.tag
print ' if (%s) {' % instance
- index = '__i' + array.type.id
- print ' Trace::BeginArray(%s);' % (array.length,)
- print ' for (int %s = 0; %s < %s; ++%s) {' % (index, index, array.length, index)
- print ' Trace::BeginElement();'
+ print ' size_t %s = %s;' % (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();'
self.visit(array.type, '(%s)[%s]' % (instance, index))
- print ' Trace::EndElement();'
+ print ' trace::localWriter.endElement();'
print ' }'
- print ' Trace::EndArray();'
+ print ' trace::localWriter.endArray();'
+ print ' } else {'
+ print ' trace::localWriter.writeNull();'
print ' }'
- print ' else'
- print ' Trace::LiteralNull();'
- def visit_blob(self, blob, instance):
- print ' Trace::LiteralBlob(%s, %s);' % (instance, blob.size)
+ def visitBlob(self, blob, instance):
+ print ' trace::localWriter.writeBlob(%s, %s);' % (instance, blob.size)
- def visit_enum(self, enum, instance):
- print ' __traceEnum%s(%s);' % (enum.id, instance)
+ def visitEnum(self, enum, instance):
+ print ' trace::localWriter.writeEnum(&__enum%s_sig, %s);' % (enum.tag, instance)
- def visit_bitmask(self, bitmask, instance):
- print ' Trace::LiteralBitmask(__bitmask%s_sig, %s);' % (bitmask.id, instance)
+ def visitBitmask(self, bitmask, instance):
+ print ' trace::localWriter.writeBitmask(&__bitmask%s_sig, %s);' % (bitmask.tag, instance)
- def visit_pointer(self, pointer, instance):
+ def visitPointer(self, pointer, instance):
print ' if (%s) {' % instance
- print ' Trace::BeginArray(1);'
- print ' Trace::BeginElement();'
- dump_instance(pointer.type, "*" + instance)
- print ' Trace::EndElement();'
- print ' Trace::EndArray();'
+ print ' trace::localWriter.beginArray(1);'
+ print ' trace::localWriter.beginElement();'
+ self.visit(pointer.type, "*" + instance)
+ print ' trace::localWriter.endElement();'
+ print ' trace::localWriter.endArray();'
+ print ' } else {'
+ print ' trace::localWriter.writeNull();'
print ' }'
- print ' else'
- print ' Trace::LiteralNull();'
- def visit_handle(self, handle, instance):
- self.visit(handle.type, instance)
+ def visitIntPointer(self, pointer, instance):
+ print ' trace::localWriter.writeOpaque((const void *)%s);' % instance
- def visit_alias(self, alias, instance):
- self.visit(alias.type, instance)
+ def visitLinearPointer(self, pointer, instance):
+ print ' trace::localWriter.writeOpaque((const void *)%s);' % instance
- def visit_opaque(self, opaque, instance):
- print ' Trace::LiteralOpaque((const void *)%s);' % instance
+ def visitHandle(self, handle, instance):
+ self.visit(handle.type, instance)
- def visit_interface(self, interface, instance):
- print ' Trace::LiteralOpaque((const void *)&%s);' % instance
+ def visitAlias(self, alias, instance):
+ self.visit(alias.type, instance)
+ def visitOpaque(self, opaque, instance):
+ print ' trace::localWriter.writeOpaque((const void *)%s);' % instance
-dump_instance = DumpImplementer().visit
+ def visitInterface(self, interface, instance):
+ print ' trace::localWriter.writeOpaque((const void *)&%s);' % instance
+ def visitPolymorphic(self, polymorphic, instance):
+ print ' _write__%s(%s, %s);' % (polymorphic.tag, polymorphic.switchExpr, instance)
-class Wrapper(stdapi.Visitor):
- '''Wrap an instance.'''
+class ValueWrapper(stdapi.Visitor):
+ '''Type visitor which will generate the code to wrap an instance.
+
+ Wrapping is necessary mostly for interfaces, however interface pointers can
+ appear anywhere inside complex types.
+ '''
- def visit_void(self, type, instance):
+ def visitVoid(self, type, instance):
raise NotImplementedError
- def visit_literal(self, type, instance):
+ def visitLiteral(self, type, instance):
pass
- def visit_string(self, type, instance):
+ def visitString(self, type, instance):
pass
- def visit_const(self, type, instance):
+ def visitConst(self, type, instance):
pass
- def visit_struct(self, struct, instance):
+ def visitStruct(self, struct, instance):
for type, name in struct.members:
self.visit(type, "(%s).%s" % (instance, name))
- def visit_array(self, array, instance):
+ def visitArray(self, array, instance):
# XXX: actually it is possible to return an array of pointers
pass
- def visit_blob(self, blob, instance):
+ def visitBlob(self, blob, instance):
pass
- def visit_enum(self, enum, instance):
+ def visitEnum(self, enum, instance):
pass
- def visit_bitmask(self, bitmask, instance):
+ def visitBitmask(self, bitmask, instance):
pass
- def visit_pointer(self, pointer, instance):
+ def visitPointer(self, pointer, instance):
+ print " if (%s) {" % instance
self.visit(pointer.type, "*" + instance)
+ print " }"
+
+ def visitIntPointer(self, pointer, instance):
+ pass
+
+ def visitLinearPointer(self, pointer, instance):
+ pass
- def visit_handle(self, handle, instance):
+ def visitHandle(self, handle, instance):
self.visit(handle.type, instance)
- def visit_alias(self, alias, instance):
+ def visitAlias(self, alias, instance):
self.visit(alias.type, instance)
- def visit_opaque(self, opaque, instance):
+ def visitOpaque(self, opaque, instance):
pass
- def visit_interface(self, interface, instance):
+ def visitInterface(self, interface, instance):
assert instance.startswith('*')
instance = instance[1:]
- print " if (%s)" % instance
- print " %s = new %s(%s);" % (instance, interface_wrap_name(interface), instance)
+ print " if (%s) {" % instance
+ print " %s = new %s(%s);" % (instance, getWrapperInterfaceName(interface), instance)
+ print " }"
+
+ def visitPolymorphic(self, type, instance):
+ # XXX: There might be polymorphic values that need wrapping in the future
+ pass
-class Unwrapper(Wrapper):
+class ValueUnwrapper(ValueWrapper):
+ '''Reverse of ValueWrapper.'''
- def visit_interface(self, interface, instance):
+ def visitInterface(self, interface, instance):
assert instance.startswith('*')
instance = instance[1:]
- print " if (%s)" % instance
- print " %s = static_cast<%s *>(%s)->m_pInstance;" % (instance, interface_wrap_name(interface), instance)
+ print " if (%s) {" % instance
+ print " %s = static_cast<%s *>(%s)->m_pInstance;" % (instance, getWrapperInterfaceName(interface), instance)
+ print " }"
-wrap_instance = Wrapper().visit
-unwrap_instance = Unwrapper().visit
+class Tracer:
+ '''Base class to orchestrate the code generation of API tracing.'''
+ def __init__(self):
+ self.api = None
-class Tracer:
+ def serializerFactory(self):
+ '''Create a serializer.
+
+ Can be overriden by derived classes to inject their own serialzer.
+ '''
+
+ return ValueSerializer()
def trace_api(self, api):
+ self.api = api
+
self.header(api)
# Includes
print header
print
- # Type dumpers
- types = api.all_types()
- visitor = DumpDeclarator()
+ # Generate the serializer functions
+ types = api.getAllTypes()
+ visitor = ComplexValueSerializer(self.serializerFactory())
map(visitor.visit, types)
print
# Interfaces wrapers
- interfaces = [type for type in types if isinstance(type, stdapi.Interface)]
- map(self.interface_wrap_impl, interfaces)
+ interfaces = api.getAllInterfaces()
+ map(self.declareWrapperInterface, interfaces)
+ map(self.implementWrapperInterface, interfaces)
print
# Function wrappers
- map(self.trace_function_decl, api.functions)
- map(self.trace_function_impl, api.functions)
+ map(self.traceFunctionDecl, api.functions)
+ map(self.traceFunctionImpl, api.functions)
print
self.footer(api)
def footer(self, api):
pass
- def trace_function_decl(self, function):
+ 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, int(function.id), function.name, len(function.args), 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 get_dispatch_function(self, function):
- return '__' + function.name
+ def isFunctionPublic(self, function):
+ return True
- def trace_function_impl(self, function):
- print 'extern "C" ' + function.prototype() + ' {'
+ def traceFunctionImpl(self, function):
+ if self.isFunctionPublic(function):
+ print 'extern "C" PUBLIC'
+ else:
+ print 'extern "C" PRIVATE'
+ print function.prototype() + ' {'
if function.type is not stdapi.Void:
print ' %s __result;' % function.type
- self.trace_function_impl_body(function)
+ self.traceFunctionImplBody(function)
if function.type is not stdapi.Void:
- self.wrap_ret(function, "__result")
+ self.wrapRet(function, "__result")
print ' return __result;'
print '}'
print
- def trace_function_impl_body(self, function):
- print ' unsigned __call = Trace::BeginEnter(__%s_sig);' % (function.name,)
+ def traceFunctionImplBody(self, function):
+ print ' unsigned __call = trace::localWriter.beginEnter(&__%s_sig);' % (function.name,)
for arg in function.args:
if not arg.output:
- self.unwrap_arg(function, arg)
- self.dump_arg(function, arg)
- print ' Trace::EndEnter();'
- self.dispatch_function(function)
- print ' Trace::BeginLeave(__call);'
+ 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.dump_arg(function, arg)
- self.wrap_arg(function, arg)
+ self.serializeArg(function, arg)
+ self.wrapArg(function, arg)
if function.type is not stdapi.Void:
- self.dump_ret(function, "__result")
- print ' Trace::EndLeave();'
+ self.serializeRet(function, "__result")
+ print ' trace::localWriter.endLeave();'
- def dispatch_function(self, function):
+ def invokeFunction(self, function, prefix='__', suffix=''):
if function.type is stdapi.Void:
result = ''
else:
result = '__result = '
- dispatch = self.get_dispatch_function(function)
+ dispatch = prefix + function.name + suffix
print ' %s%s(%s);' % (result, dispatch, ', '.join([str(arg.name) for arg in function.args]))
- def dump_arg(self, function, arg):
- print ' Trace::BeginArg(%u);' % (arg.index,)
- self.dump_arg_instance(function, arg)
- print ' Trace::EndArg();'
+ def serializeArg(self, function, arg):
+ print ' trace::localWriter.beginArg(%u);' % (arg.index,)
+ self.serializeArgValue(function, arg)
+ print ' trace::localWriter.endArg();'
+
+ def serializeArgValue(self, function, arg):
+ self.serializeValue(arg.type, arg.name)
- def dump_arg_instance(self, function, arg):
- dump_instance(arg.type, arg.name)
+ def wrapArg(self, function, arg):
+ self.wrapValue(arg.type, arg.name)
- def wrap_arg(self, function, arg):
- wrap_instance(arg.type, arg.name)
+ def unwrapArg(self, function, arg):
+ self.unwrapValue(arg.type, arg.name)
- def unwrap_arg(self, function, arg):
- unwrap_instance(arg.type, arg.name)
+ def serializeRet(self, function, instance):
+ print ' trace::localWriter.beginReturn();'
+ self.serializeValue(function.type, instance)
+ print ' trace::localWriter.endReturn();'
- def dump_ret(self, function, instance):
- print ' Trace::BeginReturn();'
- dump_instance(function.type, instance)
- print ' Trace::EndReturn();'
+ def serializeValue(self, type, instance):
+ serializer = self.serializerFactory()
+ serializer.visit(type, instance)
- def wrap_ret(self, function, instance):
- wrap_instance(function.type, instance)
+ def wrapRet(self, function, instance):
+ self.wrapValue(function.type, instance)
- def unwrap_ret(self, function, instance):
- unwrap_instance(function.type, instance)
+ def unwrapRet(self, function, instance):
+ self.unwrapValue(function.type, instance)
- def interface_wrap_impl(self, interface):
- print '%s::%s(%s * pInstance) {' % (interface_wrap_name(interface), interface_wrap_name(interface), interface.name)
+ def wrapValue(self, type, instance):
+ visitor = ValueWrapper()
+ visitor.visit(type, instance)
+
+ def unwrapValue(self, type, instance):
+ visitor = ValueUnwrapper()
+ visitor.visit(type, instance)
+
+ def declareWrapperInterface(self, interface):
+ print "class %s : public %s " % (getWrapperInterfaceName(interface), interface.name)
+ print "{"
+ print "public:"
+ print " %s(%s * pInstance);" % (getWrapperInterfaceName(interface), interface.name)
+ print " virtual ~%s();" % getWrapperInterfaceName(interface)
+ print
+ for method in interface.iterMethods():
+ print " " + method.prototype() + ";"
+ print
+ self.declareWrapperInterfaceVariables(interface)
+ print "};"
+ print
+
+ def declareWrapperInterfaceVariables(self, interface):
+ #print "private:"
+ print " %s * m_pInstance;" % (interface.name,)
+
+ def implementWrapperInterface(self, interface):
+ print '%s::%s(%s * pInstance) {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface), interface.name)
print ' m_pInstance = pInstance;'
print '}'
print
- print '%s::~%s() {' % (interface_wrap_name(interface), interface_wrap_name(interface))
+ print '%s::~%s() {' % (getWrapperInterfaceName(interface), getWrapperInterfaceName(interface))
print '}'
print
- for method in interface.itermethods():
- self.trace_method(interface, method)
+ for method in interface.iterMethods():
+ self.implementWrapperInterfaceMethod(interface, method)
print
- def trace_method(self, interface, method):
- print method.prototype(interface_wrap_name(interface) + '::' + method.name) + ' {'
- 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};' % (int(method.id), interface.name + '::' + method.name, len(method.args) + 1)
- print ' unsigned __call = Trace::BeginEnter(__sig);'
- print ' Trace::BeginArg(0);'
- print ' Trace::LiteralOpaque((const void *)m_pInstance);'
- print ' Trace::EndArg();'
- for arg in method.args:
- if not arg.output:
- self.unwrap_arg(method, arg)
- self.dump_arg(method, arg)
- if method.type is stdapi.Void:
- result = ''
- else:
- print ' %s __result;' % method.type
- result = '__result = '
- print ' Trace::EndEnter();'
- print ' %sm_pInstance->%s(%s);' % (result, method.name, ', '.join([str(arg.name) for arg in method.args]))
- print ' Trace::BeginLeave(__call);'
- for arg in method.args:
- if arg.output:
- self.dump_arg(method, arg)
- self.wrap_arg(method, arg)
+ def implementWrapperInterfaceMethod(self, interface, method):
+ print method.prototype(getWrapperInterfaceName(interface) + '::' + method.name) + ' {'
if method.type is not stdapi.Void:
- print ' Trace::BeginReturn();'
- dump_instance(method.type, "__result")
- print ' Trace::EndReturn();'
- wrap_instance(method.type, '__result')
- print ' Trace::EndLeave();'
- if method.name == 'QueryInterface':
- print ' if (*ppvObj == m_pInstance)'
- print ' *ppvObj = this;'
- if method.name == 'Release':
- assert method.type is not stdapi.Void
- print ' if (!__result)'
- print ' delete this;'
+ print ' %s __result;' % method.type
+
+ self.implementWrapperInterfaceMethodBody(interface, method)
+
if method.type is not stdapi.Void:
print ' return __result;'
print '}'
print
+ def implementWrapperInterfaceMethodBody(self, interface, 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);'
+ print ' trace::localWriter.beginArg(0);'
+ print ' trace::localWriter.writeOpaque((const void *)m_pInstance);'
+ print ' trace::localWriter.endArg();'
-class DllTracer(Tracer):
+ from specs.winapi import REFIID
+ from specs.stdapi import Pointer, Opaque
- def __init__(self, dllname):
- self.dllname = dllname
-
- def get_function_address(self, function):
- return '__%s' % (function.name,)
-
- def header(self, api):
- print '''
-static HINSTANCE g_hDll = NULL;
-
-static PROC
-__getPublicProcAddress(LPCSTR lpProcName)
-{
- if (!g_hDll) {
- char szDll[MAX_PATH] = {0};
-
- if (!GetSystemDirectoryA(szDll, MAX_PATH)) {
- return NULL;
- }
-
- strcat(szDll, "\\\\%s");
-
- g_hDll = LoadLibraryA(szDll);
- if (!g_hDll) {
- return NULL;
- }
- }
+ riid = None
+ for arg in method.args:
+ if not arg.output:
+ self.unwrapArg(method, arg)
+ self.serializeArg(method, arg)
+ if arg.type is REFIID:
+ riid = arg
+ print ' trace::localWriter.endEnter();'
- return GetProcAddress(g_hDll, lpProcName);
-}
-
- ''' % self.dllname
+ self.invokeMethod(interface, method)
- dispatcher = Dispatcher()
- dispatcher.dispatch_api(api)
+ print ' trace::localWriter.beginLeave(__call);'
+ for arg in method.args:
+ if arg.output:
+ self.serializeArg(method, arg)
+ self.wrapArg(method, arg)
+ if riid is not None and isinstance(arg.type, Pointer):
+ assert isinstance(arg.type.type, Opaque)
+ self.wrapIid(interface, method, riid, arg)
- Tracer.header(self, api)
+ 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')
+ print ' trace::localWriter.endLeave();'
+ if method.name == 'Release':
+ assert method.type is not stdapi.Void
+ print ' if (!__result)'
+ print ' delete this;'
+ def wrapIid(self, interface, method, riid, out):
+ print ' if (%s && *%s) {' % (out.name, out.name)
+ print ' if (*%s == m_pInstance) {' % (out.name,)
+ print ' *%s = this;' % (out.name,)
+ print ' }'
+ for iface in self.api.interfaces:
+ print r' else if (%s == IID_%s) {' % (riid.name, iface.name)
+ print r' *%s = new Wrap%s((%s *) *%s);' % (out.name, iface.name, iface.name, out.name)
+ print r' }'
+ print r' else {'
+ print r' os::log("apitrace: warning: unknown REFIID {0x%08lX,0x%04X,0x%04X,{0x%02X,0x%02X,0x%02X,0x%02X,0x%02X,0x%02X,0x%02X,0x%02X}}\n",'
+ print r' %s.Data1, %s.Data2, %s.Data3,' % (riid.name, riid.name, riid.name)
+ print r' %s.Data4[0],' % (riid.name,)
+ print r' %s.Data4[1],' % (riid.name,)
+ print r' %s.Data4[2],' % (riid.name,)
+ print r' %s.Data4[3],' % (riid.name,)
+ print r' %s.Data4[4],' % (riid.name,)
+ print r' %s.Data4[5],' % (riid.name,)
+ print r' %s.Data4[6],' % (riid.name,)
+ print r' %s.Data4[7]);' % (riid.name,)
+ print r' }'
+ print ' }'
+
+ def invokeMethod(self, interface, method):
+ if method.type is stdapi.Void:
+ result = ''
+ else:
+ result = '__result = '
+ print ' %sm_pInstance->%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 ' trace::localWriter.beginArg(0);'
+ print ' trace::localWriter.writeOpaque(%s);' % dest
+ print ' trace::localWriter.endArg();'
+ print ' trace::localWriter.beginArg(1);'
+ print ' trace::localWriter.writeBlob(%s, %s);' % (src, length)
+ print ' trace::localWriter.endArg();'
+ print ' trace::localWriter.beginArg(2);'
+ print ' trace::localWriter.writeUInt(%s);' % length
+ print ' trace::localWriter.endArg();'
+ print ' trace::localWriter.endEnter();'
+ print ' trace::localWriter.beginLeave(__call);'
+ print ' trace::localWriter.endLeave();'
+