src/mapi/glapi/gen/gl_marshal.py

   1
   2 # Copyright (C) 2012 Intel Corporation
   3 #
   4 # Permission is hereby granted, free of charge, to any person obtaining a
   5 # copy of this software and associated documentation files (the "Software"),
   6 # to deal in the Software without restriction, including without limitation
   7 # the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8 # and/or sell copies of the Software, and to permit persons to whom the
   9 # Software is furnished to do so, subject to the following conditions:
  10 #
  11 # The above copyright notice and this permission notice (including the next
  12 # paragraph) shall be included in all copies or substantial portions of the
  13 # Software.
  14 #
  15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18 # THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21 # IN THE SOFTWARE.
  22
  23 from __future__ import print_function
  24
  25 import contextlib
  26 import getopt
  27 import gl_XML
  28 import license
  29 import marshal_XML
  30 import sys
  31
  32 header = """
  33 #include "api_exec.h"
  34 #include "context.h"
  35 #include "dispatch.h"
  36 #include "glthread.h"
  37 #include "marshal.h"
  38
  39 #define COMPAT (ctx->API != API_OPENGL_CORE)
  40 """
  41
  42
  43 current_indent = 0
  44
  45
  46 def out(str):
  47     if str:
  48         print(' '*current_indent + str)
  49     else:
  50         print('')
  51
  52
  53 @contextlib.contextmanager
  54 def indent(delta = 3):
  55     global current_indent
  56     current_indent += delta
  57     yield
  58     current_indent -= delta
  59
  60
  61 class PrintCode(gl_XML.gl_print_base):
  62     def __init__(self):
  63         super(PrintCode, self).__init__()
  64
  65         self.name = 'gl_marshal.py'
  66         self.license = license.bsd_license_template % (
  67             'Copyright (C) 2012 Intel Corporation', 'INTEL CORPORATION')
  68
  69     def printRealHeader(self):
  70         print(header)
  71         print('static inline int safe_mul(int a, int b)')
  72         print('{')
  73         print('    if (a < 0 || b < 0) return -1;')
  74         print('    if (a == 0 || b == 0) return 0;')
  75         print('    if (a > INT_MAX / b) return -1;')
  76         print('    return a * b;')
  77         print('}')
  78         print()
  79
  80     def printRealFooter(self):
  81         pass
  82
  83     def print_sync_call(self, func, unmarshal = 0):
  84         call = 'CALL_{0}(ctx->CurrentServerDispatch, ({1}))'.format(
  85             func.name, func.get_called_parameter_string())
  86         if func.return_type == 'void':
  87             out('{0};'.format(call))
  88             if func.marshal_call_after and not unmarshal:
  89                 out(func.marshal_call_after);
  90         else:
  91             out('return {0};'.format(call))
  92             assert not func.marshal_call_after
  93
  94     def print_sync_dispatch(self, func):
  95         self.print_sync_call(func)
  96
  97     def print_sync_body(self, func):
  98         out('/* {0}: marshalled synchronously */'.format(func.name))
  99         out('static {0} GLAPIENTRY'.format(func.return_type))
 100         out('_mesa_marshal_{0}({1})'.format(func.name, func.get_parameter_string()))
 101         out('{')
 102         with indent():
 103             out('GET_CURRENT_CONTEXT(ctx);')
 104             out('_mesa_glthread_finish_before(ctx, "{0}");'.format(func.name))
 105             self.print_sync_call(func)
 106         out('}')
 107         out('')
 108         out('')
 109
 110     def print_async_dispatch(self, func):
 111         out('cmd = _mesa_glthread_allocate_command(ctx, '
 112             'DISPATCH_CMD_{0}, cmd_size);'.format(func.name))
 113         for p in func.fixed_params:
 114             if p.count:
 115                 out('memcpy(cmd->{0}, {0}, {1});'.format(
 116                         p.name, p.size_string()))
 117             else:
 118                 out('cmd->{0} = {0};'.format(p.name))
 119         if func.variable_params:
 120             out('char *variable_data = (char *) (cmd + 1);')
 121             i = 1
 122             for p in func.variable_params:
 123                 if p.img_null_flag:
 124                     out('cmd->{0}_null = !{0};'.format(p.name))
 125                     out('if (!cmd->{0}_null) {{'.format(p.name))
 126                     with indent():
 127                         out(('memcpy(variable_data, {0}, {0}_size);').format(p.name))
 128                         if i < len(func.variable_params):
 129                             out('variable_data += {0}_size;'.format(p.name))
 130                     out('}')
 131                 else:
 132                     out(('memcpy(variable_data, {0}, {0}_size);').format(p.name))
 133                     if i < len(func.variable_params):
 134                         out('variable_data += {0}_size;'.format(p.name))
 135                 i += 1
 136
 137         if not func.fixed_params and not func.variable_params:
 138             out('(void) cmd;')
 139
 140         if func.marshal_call_after:
 141             out(func.marshal_call_after);
 142
 143         # Uncomment this if you want to call _mesa_glthread_finish for debugging
 144         #out('_mesa_glthread_finish(ctx);')
 145
 146     def print_async_struct(self, func):
 147         out('struct marshal_cmd_{0}'.format(func.name))
 148         out('{')
 149         with indent():
 150             out('struct marshal_cmd_base cmd_base;')
 151             for p in func.fixed_params:
 152                 if p.count:
 153                     out('{0} {1}[{2}];'.format(
 154                             p.get_base_type_string(), p.name, p.count))
 155                 else:
 156                     out('{0} {1};'.format(p.type_string(), p.name))
 157
 158             for p in func.variable_params:
 159                 if p.img_null_flag:
 160                     out('bool {0}_null; /* If set, no data follows '
 161                         'for "{0}" */'.format(p.name))
 162
 163             for p in func.variable_params:
 164                 if p.count_scale != 1:
 165                     out(('/* Next {0} bytes are '
 166                          '{1} {2}[{3}][{4}] */').format(
 167                             p.size_string(marshal = 1), p.get_base_type_string(),
 168                             p.name, p.counter, p.count_scale))
 169                 else:
 170                     out(('/* Next {0} bytes are '
 171                          '{1} {2}[{3}] */').format(
 172                             p.size_string(marshal = 1), p.get_base_type_string(),
 173                             p.name, p.counter))
 174         out('};')
 175
 176     def print_async_unmarshal(self, func):
 177         out('static inline void')
 178         out(('_mesa_unmarshal_{0}(struct gl_context *ctx, '
 179              'const struct marshal_cmd_{0} *cmd)').format(func.name))
 180         out('{')
 181         with indent():
 182             for p in func.fixed_params:
 183                 if p.count:
 184                     p_decl = '{0} * {1} = cmd->{1};'.format(
 185                             p.get_base_type_string(), p.name)
 186                 else:
 187                     p_decl = '{0} {1} = cmd->{1};'.format(
 188                             p.type_string(), p.name)
 189
 190                 if not p_decl.startswith('const '):
 191                     # Declare all local function variables as const, even if
 192                     # the original parameter is not const.
 193                     p_decl = 'const ' + p_decl
 194
 195                 out(p_decl)
 196
 197             if func.variable_params:
 198                 for p in func.variable_params:
 199                     out('{0} * {1};'.format(
 200                             p.get_base_type_string(), p.name))
 201                 out('const char *variable_data = (const char *) (cmd + 1);')
 202                 i = 1
 203                 for p in func.variable_params:
 204                     out('{0} = ({1} *) variable_data;'.format(
 205                             p.name, p.get_base_type_string()))
 206
 207                     if p.img_null_flag:
 208                         out('if (cmd->{0}_null)'.format(p.name))
 209                         with indent():
 210                             out('{0} = NULL;'.format(p.name))
 211                         if i < len(func.variable_params):
 212                             out('else')
 213                             with indent():
 214                                 out('variable_data += {0};'.format(p.size_string(False, marshal = 1)))
 215                     elif i < len(func.variable_params):
 216                         out('variable_data += {0};'.format(p.size_string(False, marshal = 1)))
 217                     i += 1
 218
 219             self.print_sync_call(func, unmarshal = 1)
 220         out('}')
 221
 222     def validate_count_or_fallback(self, func):
 223         # Check that any counts for variable-length arguments might be < 0, in
 224         # which case the command alloc or the memcpy would blow up before we
 225         # get to the validation in Mesa core.
 226         list = []
 227         for p in func.parameters:
 228             if p.is_variable_length():
 229                 list.append('{0}_size < 0'.format(p.name))
 230                 list.append('({0}_size > 0 && !{0})'.format(p.name))
 231
 232         if len(list) == 0:
 233             return
 234
 235         list.append('(unsigned)cmd_size > MARSHAL_MAX_CMD_SIZE')
 236
 237         out('if (unlikely({0})) {{'.format(' || '.join(list)))
 238         with indent():
 239             out('_mesa_glthread_finish_before(ctx, "{0}");'.format(func.name))
 240             self.print_sync_dispatch(func)
 241             out('return;')
 242         out('}')
 243
 244     def print_async_marshal(self, func):
 245         out('static void GLAPIENTRY')
 246         out('_mesa_marshal_{0}({1})'.format(
 247                 func.name, func.get_parameter_string()))
 248         out('{')
 249         with indent():
 250             out('GET_CURRENT_CONTEXT(ctx);')
 251             for p in func.variable_params:
 252                 out('int {0}_size = {1};'.format(p.name, p.size_string(marshal = 1)))
 253
 254             struct = 'struct marshal_cmd_{0}'.format(func.name)
 255             size_terms = ['sizeof({0})'.format(struct)]
 256             for p in func.variable_params:
 257                 if p.img_null_flag:
 258                     size_terms.append('({0} ? {0}_size : 0)'.format(p.name))
 259                 else:
 260                     size_terms.append('{0}_size'.format(p.name))
 261             out('int cmd_size = {0};'.format(' + '.join(size_terms)))
 262             out('{0} *cmd;'.format(struct))
 263
 264             self.validate_count_or_fallback(func)
 265
 266             if func.marshal_fail:
 267                 out('if ({0}) {{'.format(func.marshal_fail))
 268                 with indent():
 269                     out('_mesa_glthread_disable(ctx, "{0}");'.format(func.name))
 270                     self.print_sync_dispatch(func)
 271                     out('return;')
 272                 out('}')
 273
 274             if func.marshal_sync:
 275                 out('if ({0}) {{'.format(func.marshal_sync))
 276                 with indent():
 277                     out('_mesa_glthread_finish_before(ctx, "{0}");'.format(func.name))
 278                     self.print_sync_dispatch(func)
 279                     out('return;')
 280                 out('}')
 281
 282         with indent():
 283             self.print_async_dispatch(func)
 284         out('}')
 285
 286     def print_async_body(self, func):
 287         out('/* {0}: marshalled asynchronously */'.format(func.name))
 288         self.print_async_struct(func)
 289         self.print_async_unmarshal(func)
 290         self.print_async_marshal(func)
 291         out('')
 292         out('')
 293
 294     def print_unmarshal_dispatch_cmd(self, api):
 295         out('const _mesa_unmarshal_func _mesa_unmarshal_dispatch[NUM_DISPATCH_CMD] = {')
 296         with indent():
 297             for func in api.functionIterateAll():
 298                 flavor = func.marshal_flavor()
 299                 if flavor in ('skip', 'sync'):
 300                     continue
 301                 out('[DISPATCH_CMD_{0}] = (_mesa_unmarshal_func)_mesa_unmarshal_{0},'.format(func.name))
 302         out('};')
 303         out('')
 304         out('')
 305
 306     def print_create_marshal_table(self, api):
 307         out('struct _glapi_table *')
 308         out('_mesa_create_marshal_table(const struct gl_context *ctx)')
 309         out('{')
 310         with indent():
 311             out('struct _glapi_table *table;')
 312             out('')
 313             out('table = _mesa_alloc_dispatch_table();')
 314             out('if (table == NULL)')
 315             with indent():
 316                 out('return NULL;')
 317             out('')
 318             for func in api.functionIterateAll():
 319                 if func.marshal_flavor() == 'skip':
 320                     continue
 321                 out('SET_{0}(table, _mesa_marshal_{0});'.format(func.name))
 322             out('')
 323             out('return table;')
 324         out('}')
 325         out('')
 326         out('')
 327
 328     def printBody(self, api):
 329         async_funcs = []
 330         for func in api.functionIterateAll():
 331             flavor = func.marshal_flavor()
 332             if flavor in ('skip', 'custom'):
 333                 continue
 334             elif flavor == 'async':
 335                 self.print_async_body(func)
 336                 async_funcs.append(func)
 337             elif flavor == 'sync':
 338                 self.print_sync_body(func)
 339         self.print_unmarshal_dispatch_cmd(api)
 340         self.print_create_marshal_table(api)
 341
 342
 343 def show_usage():
 344     print('Usage: %s [-f input_file_name]' % sys.argv[0])
 345     sys.exit(1)
 346
 347
 348 if __name__ == '__main__':
 349     file_name = 'gl_API.xml'
 350
 351     try:
 352         (args, trail) = getopt.getopt(sys.argv[1:], 'm:f:')
 353     except Exception:
 354         show_usage()
 355
 356     for (arg,val) in args:
 357         if arg == '-f':
 358             file_name = val
 359
 360     printer = PrintCode()
 361
 362     api = gl_XML.parse_GL_API(file_name, marshal_XML.marshal_item_factory())
 363     printer.Print(api)