import stdapi
+from dispatch import Dispatcher
def interface_wrap_name(interface):
print ' static const Trace::StructSig sig = {'
print ' %u, "%s", %u, members' % (int(struct.id), struct.name, len(struct.members))
print ' };'
- print ' Trace::BeginStruct(&sig);'
+ print ' __writer.beginStruct(&sig);'
for type, name in struct.members:
dump_instance(type, 'value.%s' % (name,))
- print ' Trace::EndStruct();'
+ print ' __writer.endStruct();'
print '}'
print
def visit_blob(self, array):
pass
+ __enum_id = 0
+
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 ' static const Trace::EnumSig sig%u = {%u, "%s", %s};' % (i, DumpDeclarator.__enum_id, value, value)
+ DumpDeclarator.__enum_id += 1
print ' const Trace::EnumSig *sig;'
print ' switch(value) {'
for i in range(n):
print ' sig = &sig%u;' % i
print ' break;'
print ' default:'
- print ' Trace::LiteralSInt(value);'
+ print ' __writer.writeSInt(value);'
print ' return;'
print ' }'
- print ' Trace::LiteralEnum(sig);'
+ print ' __writer.writeEnum(sig);'
print '}'
print
def visit_bitmask(self, bitmask):
- print 'static const Trace::BitmaskVal __bitmask%s_vals[] = {' % (bitmask.id)
+ print 'static const Trace::BitmaskFlag __bitmask%s_flags[] = {' % (bitmask.id)
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 ' %u, %u, __bitmask%s_flags' % (int(bitmask.id), len(bitmask.values), bitmask.id)
print '};'
print
'''Dump an instance.'''
def visit_literal(self, literal, instance):
- print ' Trace::Literal%s(%s);' % (literal.format, instance)
+ print ' __writer.write%s(%s);' % (literal.format, instance)
def visit_string(self, string, instance):
if string.length is not None:
- print ' Trace::LiteralString((const char *)%s, %s);' % (instance, string.length)
+ print ' __writer.writeString((const char *)%s, %s);' % (instance, string.length)
else:
- print ' Trace::LiteralString((const char *)%s);' % instance
+ print ' __writer.writeString((const char *)%s);' % instance
def visit_const(self, const, instance):
self.visit(const.type, instance)
print ' __traceStruct%s(%s);' % (struct.id, instance)
def visit_array(self, array, instance):
- print ' if(%s) {' % instance
+ length = '__c' + array.type.id
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 ' if (%s) {' % instance
+ print ' size_t %s = %s;' % (length, array.length)
+ print ' __writer.beginArray(%s);' % length
+ print ' for (size_t %s = 0; %s < %s; ++%s) {' % (index, index, length, index)
+ print ' __writer.beginElement();'
self.visit(array.type, '(%s)[%s]' % (instance, index))
- print ' Trace::EndElement();'
+ print ' __writer.endElement();'
print ' }'
- print ' Trace::EndArray();'
+ print ' __writer.endArray();'
+ print ' } else {'
+ print ' __writer.writeNull();'
print ' }'
- print ' else'
- print ' Trace::LiteralNull();'
def visit_blob(self, blob, instance):
- print ' Trace::LiteralBlob(%s, %s);' % (instance, blob.size)
+ print ' __writer.writeBlob(%s, %s);' % (instance, blob.size)
def visit_enum(self, enum, instance):
print ' __traceEnum%s(%s);' % (enum.id, instance)
def visit_bitmask(self, bitmask, instance):
- print ' Trace::LiteralBitmask(__bitmask%s_sig, %s);' % (bitmask.id, instance)
+ print ' __writer.writeBitmask(&__bitmask%s_sig, %s);' % (bitmask.id, instance)
def visit_pointer(self, pointer, instance):
- print ' if(%s) {' % instance
- print ' Trace::BeginArray(1);'
- print ' Trace::BeginElement();'
+ print ' if (%s) {' % instance
+ print ' __writer.beginArray(1);'
+ print ' __writer.beginElement();'
dump_instance(pointer.type, "*" + instance)
- print ' Trace::EndElement();'
- print ' Trace::EndArray();'
+ print ' __writer.endElement();'
+ print ' __writer.endArray();'
+ print ' } else {'
+ print ' __writer.writeNull();'
print ' }'
- print ' else'
- print ' Trace::LiteralNull();'
def visit_handle(self, handle, instance):
self.visit(handle.type, instance)
self.visit(alias.type, instance)
def visit_opaque(self, opaque, instance):
- print ' Trace::LiteralOpaque((const void *)%s);' % instance
+ print ' __writer.writeOpaque((const void *)%s);' % instance
def visit_interface(self, interface, instance):
- print ' Trace::LiteralOpaque((const void *)&%s);' % instance
+ print ' __writer.writeOpaque((const void *)&%s);' % instance
dump_instance = DumpImplementer().visit
pass
def visit_pointer(self, pointer, instance):
+ print " if (%s) {" % instance
self.visit(pointer.type, "*" + instance)
+ print " }"
def visit_handle(self, handle, instance):
self.visit(handle.type, instance)
def visit_interface(self, interface, instance):
assert instance.startswith('*')
instance = instance[1:]
- print " if(%s)" % instance
+ print " if (%s) {" % instance
print " %s = new %s(%s);" % (instance, interface_wrap_name(interface), instance)
+ print " }"
class Unwrapper(Wrapper):
def visit_interface(self, interface, instance):
assert instance.startswith('*')
instance = instance[1:]
- print " if(%s)" % instance
+ print " if (%s) {" % instance
print " %s = static_cast<%s *>(%s)->m_pInstance;" % (instance, interface_wrap_name(interface), instance)
+ print " }"
wrap_instance = Wrapper().visit
class Tracer:
+ def __init__(self):
+ self.api = None
+
def trace_api(self, api):
+ self.api = api
+
self.header(api)
# Includes
self.footer(api)
def header(self, api):
- pass
+ print 'Trace::LocalWriter __writer;'
+ print
def footer(self, api):
pass
- def function_pointer_type(self, function):
- return 'P' + function.name
-
- def function_pointer_value(self, function):
- return 'p' + function.name
-
def trace_function_decl(self, function):
- ptype = self.function_pointer_type(function)
- pvalue = self.function_pointer_value(function)
- print 'typedef ' + function.prototype('* %s' % ptype) + ';'
- print 'static %s %s = NULL;' % (ptype, pvalue)
+ # 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]))
+ 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 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
- def trace_function_fail(self, function):
- if function.fail is not None:
- if function.type is stdapi.Void:
- assert function.fail == ''
- print ' return;'
- else:
- assert function.fail != ''
- print ' return %s;' % function.fail
- else:
- print ' Trace::Abort();'
-
- def get_function_address(self, function):
- raise NotImplementedError
+ def get_dispatch_function(self, function):
+ return '__' + function.name
- def _get_true_pointer(self, function):
- ptype = self.function_pointer_type(function)
- pvalue = self.function_pointer_value(function)
- print ' if(!%s) {' % (pvalue,)
- print ' %s = (%s)%s;' % (pvalue, ptype, self.get_function_address(function))
- print ' if(!%s)' % (pvalue,)
- self.trace_function_fail(function)
- print ' }'
+ def is_public_function(self, function):
+ return True
def trace_function_impl(self, function):
- pvalue = self.function_pointer_value(function)
- print 'extern "C" ' + function.prototype() + ' {'
- if function.type is stdapi.Void:
- result = ''
+ if self.is_public_function(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
- result = '__result = '
- self._get_true_pointer(function)
- print ' unsigned __call = Trace::BeginEnter(__%s_sig);' % (function.name,)
+ self.trace_function_impl_body(function)
+ if function.type is not stdapi.Void:
+ self.wrap_ret(function, "__result")
+ print ' return __result;'
+ print '}'
+ print
+
+ def trace_function_impl_body(self, function):
+ print ' unsigned __call = __writer.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();'
- print ' %s%s(%s);' % (result, pvalue, ', '.join([str(arg.name) for arg in function.args]))
- print ' Trace::BeginLeave(__call);'
+ print ' __writer.endEnter();'
+ self.dispatch_function(function)
+ print ' __writer.beginLeave(__call);'
for arg in function.args:
if arg.output:
self.dump_arg(function, arg)
self.wrap_arg(function, arg)
if function.type is not stdapi.Void:
self.dump_ret(function, "__result")
- print ' Trace::EndLeave();'
- if function.type is not stdapi.Void:
- self.wrap_ret(function, "__result")
- print ' return __result;'
- print '}'
- print
+ print ' __writer.endLeave();'
+
+ def dispatch_function(self, function):
+ if function.type is stdapi.Void:
+ result = ''
+ else:
+ result = '__result = '
+ dispatch = self.get_dispatch_function(function)
+ 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,)
+ print ' __writer.beginArg(%u);' % (arg.index,)
+ self.dump_arg_instance(function, arg)
+ print ' __writer.endArg();'
+
+ def dump_arg_instance(self, function, arg):
dump_instance(arg.type, arg.name)
- print ' Trace::EndArg();'
def wrap_arg(self, function, arg):
wrap_instance(arg.type, arg.name)
unwrap_instance(arg.type, arg.name)
def dump_ret(self, function, instance):
- print ' Trace::BeginReturn();'
+ print ' __writer.beginReturn();'
dump_instance(function.type, instance)
- print ' Trace::EndReturn();'
+ print ' __writer.endReturn();'
def wrap_ret(self, function, instance):
wrap_instance(function.type, instance)
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();'
+ print ' unsigned __call = __writer.beginEnter(&__sig);'
+ print ' __writer.beginArg(0);'
+ print ' __writer.writeOpaque((const void *)m_pInstance);'
+ print ' __writer.endArg();'
for arg in method.args:
if not arg.output:
self.unwrap_arg(method, arg)
else:
print ' %s __result;' % method.type
result = '__result = '
- print ' Trace::EndEnter();'
+ print ' __writer.endEnter();'
print ' %sm_pInstance->%s(%s);' % (result, method.name, ', '.join([str(arg.name) for arg in method.args]))
- print ' Trace::BeginLeave(__call);'
+ print ' __writer.beginLeave(__call);'
for arg in method.args:
if arg.output:
self.dump_arg(method, arg)
self.wrap_arg(method, arg)
if method.type is not stdapi.Void:
- print ' Trace::BeginReturn();'
+ print ' __writer.beginReturn();'
dump_instance(method.type, "__result")
- print ' Trace::EndReturn();'
+ print ' __writer.endReturn();'
wrap_instance(method.type, '__result')
- print ' Trace::EndLeave();'
+ print ' __writer.endLeave();'
if method.name == 'QueryInterface':
- print ' if (*ppvObj == m_pInstance)'
- print ' *ppvObj = this;'
+ print ' if (ppvObj && *ppvObj) {'
+ print ' if (*ppvObj == m_pInstance) {'
+ print ' *ppvObj = this;'
+ print ' }'
+ for iface in self.api.interfaces:
+ print ' else if (riid == IID_%s) {' % iface.name
+ print ' *ppvObj = new Wrap%s((%s *) *ppvObj);' % (iface.name, iface.name)
+ print ' }'
+ print ' }'
if method.name == 'Release':
assert method.type is not stdapi.Void
print ' if (!__result)'
def __init__(self, dllname):
self.dllname = dllname
- def get_function_address(self, function):
- return '__GetProcAddress("%s")' % (function.name,)
-
-
def header(self, api):
- Tracer.header(self, api)
-
print '''
static HINSTANCE g_hDll = NULL;
static PROC
-__GetProcAddress(LPCSTR lpProcName)
+__getPublicProcAddress(LPCSTR lpProcName)
{
if (!g_hDll) {
char szDll[MAX_PATH] = {0};
return GetProcAddress(g_hDll, lpProcName);
}
- ''' % self.dllname
+''' % self.dllname
+
+ dispatcher = Dispatcher()
+ dispatcher.dispatch_api(api)
+
+ Tracer.header(self, api)