Pepper: Autogenerate thunk for PPB_URL_Loader.

ppapi/generators/idl_thunk.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """ Generator for C++ style thunks """
 
 import glob
 import os
 import re
@@ skipping 187 matching lines @@
   arg_list = ', '.join([a[1] for a in args])
   if member.GetProperty('create_func'):
     create_func = member.GetProperty('create_func')
   else:
     create_func = _GetCreateFuncName(interface)
   body += 'return enter.functions()->%s(%s);' % (create_func,
                                                  arg_list)
   return body
 
 
+def _GetOutputParams(member, release):
+  """Returns output parameters (and their types) for a member function.
+
+  Args:
+    member - IDLNode for the member function
+    release - Release to get output parameters for
+  Returns:
+    A list of (name, type) tuples for all output parameters of the member
+    function.
+  """
+  out_params = []
+  callnode = member.GetOneOf('Callspec')
+  if callnode:
+    cgen = CGen()
+    for param in callnode.GetListOf('Param'):
+      mode = cgen.GetParamMode(param)
+      if mode == 'out':
+        # We use the 'store' mode when getting the parameter type, since we
+        # need to call sizeof() for memset().
+        ptype, pname, _, _ = cgen.GetComponents(param, release, 'store')
+        out_params.append((pname, ptype))
+  return out_params
+
+
 def _MakeNormalMemberBody(filenode, release, node, member, rtype, args,
                           include_version, meta):
   """Returns the body of a typical function.
 
   Args:
     filenode - IDLNode for the file
     release - release to generate body for
     node - IDLNode for the interface
     member - IDLNode for the member function
     rtype - Return type for the member function
@@ skipping 19 matching lines @@
   function_name = member.GetName()
   if include_version:
     version = node.GetVersion(release).replace('.', '_')
     function_name += version
 
   invocation = 'enter.%s()->%s(%s)' % (function_container,
                                        function_name,
                                        call_arglist)
 
   handle_errors = not (member.GetProperty('report_errors') == 'False')
+  out_params = _GetOutputParams(member, release)
   if is_callback_func:
+    # TODO(teravest): Reduce code duplication below.
     body = '%s\n' % _MakeEnterLine(filenode, node, args[0], handle_errors,
                                    args[len(args) - 1][1], meta)
-    body += 'if (enter.failed())\n'
     value = member.GetProperty('on_failure')
     if value is None:
       value = 'enter.retval()'
-    body += '  return %s;\n' % value
-    body += 'return enter.SetResult(%s);' % invocation
+    if member.GetProperty('always_set_output_parameters'):
+      body += 'if (enter.failed()) {\n'
+      for param in out_params:
+        body += '  memset(%s, 0, sizeof(%s));\n' % param
+      body += '  return %s;\n' % value
+      body += '}\n'
+      body += 'return enter.SetResult(%s);' % invocation
+      meta.AddBuiltinInclude('string.h')
+    else:
+      body += 'if (enter.failed())\n'
+      body += '  return %s;\n' % value
+      body += 'return enter.SetResult(%s);' % invocation
   elif rtype == 'void':
-    # On failure, zero out all output parameters.
-    out_params = []
-    callnode = member.GetOneOf('Callspec')
-    if callnode:
-      cgen = CGen()
-      for param in callnode.GetListOf('Param'):
-        mode = cgen.GetParamMode(param)
-        if mode == 'out':
-          # We use the 'store' mode when getting the parameter type, since we
-          # need to call sizeof() for memset().
-          ptype, pname, _, _ = cgen.GetComponents(param, release, 'store')
-          out_params.append((pname, ptype))
-
     body = '%s\n' % _MakeEnterLine(filenode, node, args[0], handle_errors,
                                    None, meta)
-    if not out_params:
-      body += 'if (enter.succeeded())\n'
-      body += '  %s;' % invocation
-    else:
+    if member.GetProperty('always_set_output_parameters'):
       body += 'if (enter.succeeded()) {\n'
       body += '  %s;\n' % invocation
       body += '  return;\n'
       body += '}'
       for param in out_params:
         body += '\nmemset(%s, 0, sizeof(%s));' % param
       meta.AddBuiltinInclude('string.h')
+    else:
+      body += 'if (enter.succeeded())\n'
+      body += '  %s;' % invocation
 
   else:
     value = member.GetProperty('on_failure')
     if value is None:
       value = _GetDefaultFailureValue(rtype)
     if value is None:
       raise TGenError('No default value for rtype %s' % rtype)
 
     body = '%s\n' % _MakeEnterLine(filenode, node, args[0], handle_errors,
                                    None, meta)
-    body += 'if (enter.failed())\n'
-    body += '  return %s;\n' % value
-    body += 'return %s;' % invocation
+    if member.GetProperty('always_set_output_parameters'):
+      body += 'if (enter.failed()) {\n'
+      for param in out_params:
+        body += '  memset(%s, 0, sizeof(%s));\n' % param
+      body += '  return %s;\n' % value
+      body += '}\n'
+      body += 'return %s;' % invocation
+      meta.AddBuiltinInclude('string.h')
+    else:
+      body += 'if (enter.failed())\n'
+      body += '  return %s;\n' % value
+      body += 'return %s;' % invocation
   return body
 
 
 def DefineMember(filenode, node, member, release, include_version, meta):
   """Returns a definition for a member function of an interface.
 
   Args:
     filenode - IDLNode for the file
     node - IDLNode for the interface
     member - IDLNode for the member function
@@ skipping 219 matching lines @@
     print "Golden file for M13-M14 failed."
     failed = 1
   else:
     print "Golden file for M13-M14 passed."
 
   return failed
 
 
 if __name__ == '__main__':
   sys.exit(Main(sys.argv[1:]))