Selectively perform JNI registration in render processes on Android.

base/android/jni_generator/jni_generator.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.
 
 """Extracts native methods from a Java file and generates the JNI bindings.
 If you change this, please run and update the tests."""
 
 import collections
 import errno
@@ skipping 614 matching lines @@
       match = re.match(re_constant_field, content)
       if not match:
         continue
       value = re.match(re_constant_field_value, contents[lineno + 2])
       if not value:
         value = re.match(re_constant_field_value, contents[lineno + 3])
       if value:
         self.constant_fields.append(
             ConstantField(name=match.group('name'),
                           value=value.group('value')))
-
     self.inl_header_file_generator = InlHeaderFileGenerator(
-        self.namespace, self.fully_qualified_class, [],
-        self.called_by_natives, self.constant_fields, options)
+        self.namespace, self.fully_qualified_class, [], self.called_by_natives,
+        self.constant_fields, options)
 
   def GetContent(self):
     return self.inl_header_file_generator.GetContent()
 
   @staticmethod
   def CreateFromClass(class_file, options):
     class_name = os.path.splitext(os.path.basename(class_file))[0]
     p = subprocess.Popen(args=[options.javap, '-c', '-verbose',
                                '-s', class_name],
                          cwd=os.path.dirname(class_file),
@@ skipping 13 matching lines @@
       r'//.*?$|/\*.*?\*/|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"',
       re.DOTALL | re.MULTILINE)
 
   def __init__(self, contents, fully_qualified_class, options):
     contents = self._RemoveComments(contents)
     JniParams.SetFullyQualifiedClass(fully_qualified_class)
     JniParams.ExtractImportsAndInnerClasses(contents)
     jni_namespace = ExtractJNINamespace(contents) or options.namespace
     natives = ExtractNatives(contents, options.ptr_type)
     called_by_natives = ExtractCalledByNatives(contents)
+    maindex = JNIFromJavaSource.InMainDex(contents)
     if len(natives) == 0 and len(called_by_natives) == 0:
       raise SyntaxError('Unable to find any JNI methods for %s.' %
                         fully_qualified_class)
     inl_header_file_generator = InlHeaderFileGenerator(
-        jni_namespace, fully_qualified_class, natives, called_by_natives,
-        [], options)
+        jni_namespace, fully_qualified_class, natives, called_by_natives, [],
+        options, maindex)
     self.content = inl_header_file_generator.GetContent()
 
   @classmethod
   def _RemoveComments(cls, contents):
     # We need to support both inline and block comments, and we need to handle
     # strings that contain '//' or '/*'.
     # TODO(bulach): This is a bit hacky. It would be cleaner to use a real Java
     # parser. Maybe we could ditch JNIFromJavaSource and just always use
     # JNIFromJavaP; or maybe we could rewrite this script in Java and use APT.
     # http://code.google.com/p/chromium/issues/detail?id=138941
@@ skipping 10 matching lines @@
   def GetContent(self):
     return self.content
 
   @staticmethod
   def CreateFromFile(java_file_name, options):
     contents = file(java_file_name).read()
     fully_qualified_class = ExtractFullyQualifiedJavaClassName(java_file_name,
                                                                contents)
     return JNIFromJavaSource(contents, fully_qualified_class, options)
 
+  @staticmethod
+  def InMainDex(contents):

[agrieve, 2016/12/07 20:46:19]

nit: odd name. I'd probably just inline the function.

[estevenson, 2016/12/07 23:00:04]

Done.

+    return '@MainDex' in contents
+
 
 class InlHeaderFileGenerator(object):
   """Generates an inline header file for JNI integration."""
 
   def __init__(self, namespace, fully_qualified_class, natives,
-               called_by_natives, constant_fields, options):
+               called_by_natives, constant_fields, options, maindex=False):
     self.namespace = namespace
     self.fully_qualified_class = fully_qualified_class
     self.class_name = self.fully_qualified_class.split('/')[-1]
     self.natives = natives
     self.called_by_natives = called_by_natives
     self.header_guard = fully_qualified_class.replace('/', '_') + '_JNI'
     self.constant_fields = constant_fields
+    self.maindex = maindex
     self.options = options
 
 
   def GetContent(self):
     """Returns the content of the JNI binding file."""
     template = Template("""\
 // Copyright 2014 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.
 
 
 // This file is autogenerated by
 //     ${SCRIPT_NAME}
 // For
 //     ${FULLY_QUALIFIED_CLASS}
 
 #ifndef ${HEADER_GUARD}
 #define ${HEADER_GUARD}
 
 #include <jni.h>
 
 ${INCLUDES}
 
 #include "base/android/jni_int_wrapper.h"
+#include "base/android/jni_utils.h"
 
 // Step 1: forward declarations.
 namespace {
 $CLASS_PATH_DEFINITIONS
 
 }  // namespace
 
 $OPEN_NAMESPACE
 
 $CONSTANT_FIELDS
@@ skipping 102 matching lines @@
 ${NATIVES}
   return true;
 }
 """)
     signature = 'static bool RegisterNativesImpl(JNIEnv* env)'
     early_exit = ''
     if self.options.native_exports_optional:
       early_exit = """\
   if (base::android::IsManualJniRegistrationDisabled()) return true;
 """
+    if not self.maindex:
+      early_exit += """\
+    if (base::android::IsMultidexEnabled(env) &&
+      base::android::GetLibraryProcessType(env) !=
+      base::android::PROCESS_BROWSER) {
+    return true;
+  }
+"""
 
     values = {'REGISTER_NATIVES_SIGNATURE': signature,
               'EARLY_EXIT': early_exit,
               'NATIVES': natives,
              }
 
     return template.substitute(values)
 
   def GetRegisterNativesImplString(self):
     """Returns the shared implementation for RegisterNatives."""
@@ skipping 516 matching lines @@
     root_name = os.path.splitext(os.path.basename(input_file))[0]
     output_file = os.path.join(options.output_dir, root_name) + '_jni.h'
   GenerateJNIHeader(input_file, output_file, options)
 
   if options.depfile:
     build_utils.WriteDepfile(options.depfile, output_file)
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv))