| Index: build/android/pylib/utils/isolator.py
|
| diff --git a/build/android/pylib/utils/isolator.py b/build/android/pylib/utils/isolator.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f8177e007395c93769b21014315fd6beb7824b6b
|
| --- /dev/null
|
| +++ b/build/android/pylib/utils/isolator.py
|
| @@ -0,0 +1,192 @@
|
| +# 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.
|
| +
|
| +import fnmatch
|
| +import glob
|
| +import os
|
| +import shutil
|
| +import sys
|
| +import tempfile
|
| +
|
| +from devil.utils import cmd_helper
|
| +from pylib import constants
|
| +from pylib.constants import host_paths
|
| +
|
| +
|
| +_ISOLATE_SCRIPT = os.path.join(
|
| + host_paths.DIR_SOURCE_ROOT, 'tools', 'swarming_client', 'isolate.py')
|
| +
|
| +
|
| +def DefaultPathVariables():
|
| + return {
|
| + 'DEPTH': host_paths.DIR_SOURCE_ROOT,
|
| + 'PRODUCT_DIR': constants.GetOutDirectory(),
|
| + }
|
| +
|
| +
|
| +def DefaultConfigVariables():
|
| + # Note: This list must match the --config-vars in build/isolate.gypi
|
| + return {
|
| + 'CONFIGURATION_NAME': constants.GetBuildType(),
|
| + 'OS': 'android',
|
| + 'asan': '0',
|
| + 'branding': 'Chromium',
|
| + 'chromeos': '0',
|
| + 'component': 'static_library',
|
| + 'enable_pepper_cdms': '0',
|
| + 'enable_plugins': '0',
|
| + 'fastbuild': '0',
|
| + 'icu_use_data_file_flag': '1',
|
| + 'kasko': '0',
|
| + 'lsan': '0',
|
| + 'msan': '0',
|
| + # TODO(maruel): This may not always be true.
|
| + 'target_arch': 'arm',
|
| + 'tsan': '0',
|
| + 'use_custom_libcxx': '0',
|
| + 'use_instrumented_libraries': '0',
|
| + 'use_prebuilt_instrumented_libraries': '0',
|
| + 'use_ozone': '0',
|
| + 'use_x11': '0',
|
| + 'v8_use_external_startup_data': '1',
|
| + 'msvs_version': '0',
|
| + }
|
| +
|
| +
|
| +def IsIsolateEmpty(isolate_path):
|
| + """Returns whether there are no files in the .isolate."""
|
| + with open(isolate_path) as f:
|
| + return "'files': []" in f.read()
|
| +
|
| +
|
| +class Isolator(object):
|
| + """Manages calls to isolate.py for the android test runner scripts."""
|
| +
|
| + def __init__(self):
|
| + self._isolate_deps_dir = tempfile.mkdtemp()
|
| +
|
| + @property
|
| + def isolate_deps_dir(self):
|
| + return self._isolate_deps_dir
|
| +
|
| + def Clear(self):
|
| + """Deletes the isolate dependency directory."""
|
| + if os.path.exists(self._isolate_deps_dir):
|
| + shutil.rmtree(self._isolate_deps_dir)
|
| +
|
| + def Remap(self, isolate_abs_path, isolated_abs_path,
|
| + path_variables=None, config_variables=None):
|
| + """Remaps data dependencies into |self._isolate_deps_dir|.
|
| +
|
| + Args:
|
| + isolate_abs_path: The absolute path to the .isolate file, which specifies
|
| + data dependencies in the source tree.
|
| + isolated_abs_path: The absolute path to the .isolated file, which is
|
| + generated by isolate.py and specifies data dependencies in
|
| + |self._isolate_deps_dir| and their digests.
|
| + path_variables: A dict containing everything that should be passed
|
| + as a |--path-variable| to the isolate script. Defaults to the return
|
| + value of |DefaultPathVariables()|.
|
| + config_variables: A dict containing everything that should be passed
|
| + as a |--config-variable| to the isolate script. Defaults to the return
|
| + value of |DefaultConfigVariables()|.
|
| + Raises:
|
| + Exception if the isolate command fails for some reason.
|
| + """
|
| + if not path_variables:
|
| + path_variables = DefaultPathVariables()
|
| + if not config_variables:
|
| + config_variables = DefaultConfigVariables()
|
| +
|
| + isolate_cmd = [
|
| + sys.executable, _ISOLATE_SCRIPT, 'remap',
|
| + '--isolate', isolate_abs_path,
|
| + '--isolated', isolated_abs_path,
|
| + '--outdir', self._isolate_deps_dir,
|
| + ]
|
| + for k, v in path_variables.iteritems():
|
| + isolate_cmd.extend(['--path-variable', k, v])
|
| + for k, v in config_variables.iteritems():
|
| + isolate_cmd.extend(['--config-variable', k, v])
|
| +
|
| + exit_code, _ = cmd_helper.GetCmdStatusAndOutput(isolate_cmd)
|
| + if exit_code:
|
| + raise Exception('isolate command failed: %s' % ' '.join(isolate_cmd))
|
| +
|
| + def VerifyHardlinks(self):
|
| + """Checks |isolate_deps_dir| for a hardlink.
|
| +
|
| + Returns:
|
| + True if a hardlink is found.
|
| + False if nothing is found.
|
| + Raises:
|
| + Exception if a non-hardlink is found.
|
| + """
|
| + for root, _, filenames in os.walk(self._isolate_deps_dir):
|
| + if filenames:
|
| + linked_file = os.path.join(root, filenames[0])
|
| + orig_file = os.path.join(
|
| + self._isolate_deps_dir,
|
| + os.path.relpath(linked_file, self._isolate_deps_dir))
|
| + if os.stat(linked_file).st_ino == os.stat(orig_file).st_ino:
|
| + return True
|
| + else:
|
| + raise Exception('isolate remap command did not use hardlinks.')
|
| + return False
|
| +
|
| + def PurgeExcluded(self, deps_exclusion_list):
|
| + """Deletes anything on |deps_exclusion_list| from |self._isolate_deps_dir|.
|
| +
|
| + Args:
|
| + deps_exclusion_list: A list of globs to exclude from the isolate
|
| + dependency directory.
|
| + """
|
| + excluded_paths = (
|
| + x for y in deps_exclusion_list
|
| + for x in glob.glob(
|
| + os.path.abspath(os.path.join(self._isolate_deps_dir, y))))
|
| + for p in excluded_paths:
|
| + if os.path.isdir(p):
|
| + shutil.rmtree(p)
|
| + else:
|
| + os.remove(p)
|
| +
|
| + @classmethod
|
| + def _DestructiveMerge(cls, src, dest):
|
| + if os.path.exists(dest) and os.path.isdir(dest):
|
| + for p in os.listdir(src):
|
| + cls._DestructiveMerge(os.path.join(src, p), os.path.join(dest, p))
|
| + os.rmdir(src)
|
| + else:
|
| + shutil.move(src, dest)
|
| +
|
| +
|
| + def MoveOutputDeps(self):
|
| + """Moves files from the output directory to the top level of
|
| + |self._isolate_deps_dir|.
|
| +
|
| + Moves pak files from the output directory to to <isolate_deps_dir>/paks
|
| + Moves files from the product directory to <isolate_deps_dir>
|
| + """
|
| + # On Android, all pak files need to be in the top-level 'paks' directory.
|
| + paks_dir = os.path.join(self._isolate_deps_dir, 'paks')
|
| + os.mkdir(paks_dir)
|
| +
|
| + deps_out_dir = os.path.join(
|
| + self._isolate_deps_dir,
|
| + os.path.relpath(os.path.join(constants.GetOutDirectory(), os.pardir),
|
| + host_paths.DIR_SOURCE_ROOT))
|
| + for root, _, filenames in os.walk(deps_out_dir):
|
| + for filename in fnmatch.filter(filenames, '*.pak'):
|
| + shutil.move(os.path.join(root, filename), paks_dir)
|
| +
|
| + # Move everything in PRODUCT_DIR to top level.
|
| + deps_product_dir = os.path.join(
|
| + deps_out_dir, os.path.basename(constants.GetOutDirectory()))
|
| + if os.path.isdir(deps_product_dir):
|
| + for p in os.listdir(deps_product_dir):
|
| + Isolator._DestructiveMerge(os.path.join(deps_product_dir, p),
|
| + os.path.join(self._isolate_deps_dir, p))
|
| + os.rmdir(deps_product_dir)
|
| + os.rmdir(deps_out_dir)
|
|
|