git.cworth.org Git - apitrace/blob - dispatch.py

   1 ##########################################################################
   2 #
   3 # Copyright 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
  26
  27 """Generate DLL/SO dispatching functions.
  28 """
  29
  30
  31 import stdapi
  32
  33
  34 def function_pointer_type(function):
  35     return '__PFN' + function.name.upper()
  36
  37
  38 def function_pointer_value(function):
  39     return '__' + function.name + '_ptr'
  40
  41
  42 class Dispatcher:
  43
  44     def header(self):
  45         # Must be implemented by derived classes, which should define, declare,
  46         # or implement something like:
  47         #
  48         #  typedef void (*__PROC)(void);
  49         #
  50         #  static __PROC __getPublicProcAddress(const char *name);
  51         #  static __PROC __getPrivateProcAddress(const char *name);
  52         #
  53         raise NotImplementedError
  54
  55     def dispatch_api(self, api):
  56         for function in api.functions:
  57             self.dispatch_function(function)
  58
  59         # define standard name aliases for convenience, but only when not
  60         # tracing, as that would cause symbol clashing with the tracing
  61         # functions
  62         print '#ifdef RETRACE'
  63         for function in api.functions:
  64             if self.is_public_function(function):
  65                 print '#define __%s %s' % (function.name, function.name)
  66             else:
  67                 print '#define %s __%s' % (function.name, function.name)
  68         print '#endif /* RETRACE */'
  69         print
  70
  71     def dispatch_function(self, function):
  72         if self.is_public_function(function):
  73             print '#ifndef RETRACE'
  74             print
  75         ptype = function_pointer_type(function)
  76         pvalue = function_pointer_value(function)
  77         print 'typedef ' + function.prototype('* %s' % ptype) + ';'
  78         print 'static %s %s = NULL;' % (ptype, pvalue)
  79         print
  80         print 'static inline ' + function.prototype('__' + function.name) + ' {'
  81         if function.type is stdapi.Void:
  82             ret = ''
  83         else:
  84             ret = 'return '
  85         self.get_true_pointer(function)
  86         print '    %s%s(%s);' % (ret, pvalue, ', '.join([str(arg.name) for arg in function.args]))
  87         print '}'
  88         print
  89         if self.is_public_function(function):
  90             print '#endif /* !RETRACE */'
  91             print
  92
  93     def is_public_function(self, function):
  94         return True
  95
  96     def get_true_pointer(self, function):
  97         ptype = function_pointer_type(function)
  98         pvalue = function_pointer_value(function)
  99         if self.is_public_function(function):
 100             get_proc_address = '__getPublicProcAddress'
 101         else:
 102             get_proc_address = '__getPrivateProcAddress'
 103         print '    if (!%s) {' % (pvalue,)
 104         print '        %s = (%s)%s("%s");' % (pvalue, ptype, get_proc_address, function.name)
 105         print '        if (!%s) {' % (pvalue,)
 106         self.fail_function(function)
 107         print '        }'
 108         print '    }'
 109
 110     def fail_function(self, function):
 111         print '            OS::DebugMessage("error: unavailable function \\"%s\\"\\n");' % function.name
 112         if function.fail is not None:
 113             if function.type is stdapi.Void:
 114                 assert function.fail == ''
 115                 print '            return;'
 116             else:
 117                 assert function.fail != ''
 118                 print '            return %s;' % function.fail
 119         else:
 120             print '            OS::Abort();'
 121
 122