Index: third_party/instrumented_libraries/scripts/download_binaries.py |
diff --git a/third_party/instrumented_libraries/scripts/download_binaries.py b/third_party/instrumented_libraries/scripts/download_binaries.py |
index fa339cc76badba3e450cbd2da826db301a64ec54..b7e87b2eda6f79270c8645d7eae50b58a1822ab1 100755 |
--- a/third_party/instrumented_libraries/scripts/download_binaries.py |
+++ b/third_party/instrumented_libraries/scripts/download_binaries.py |
@@ -3,34 +3,69 @@ |
# Use of this source code is governed by a BSD-style license that can be |
# found in the LICENSE file. |
-"""Downloads pre-built sanitizer-instrumented third-party libraries from GCS. |
-This script should only be run from gn. |
-""" |
+"""Downloads pre-built sanitizer-instrumented third-party libraries from GCS.""" |
+import os |
+import re |
import subprocess |
import sys |
+SCRIPT_PATH = os.path.abspath(__file__) |
+BASE_PATH = os.path.normpath(os.path.join(SCRIPT_PATH, *(4 * ['..']))) |
+ |
+# This defaults to the consuming project's base directory, e.g. 'src' in |
+# chromium. |
+BASE_DIR = os.path.basename(BASE_PATH) |
+ |
+ |
+def get_ubuntu_release(): |
+ supported_releases = ['trusty'] |
+ release = subprocess.check_output(['lsb_release', '-cs']).strip() |
+ if release not in supported_releases: |
+ raise Exception("Supported Ubuntu versions: %s", str(supported_releases)) |
+ return release |
+ |
+ |
+def get_configuration(gyp_defines): |
+ if re.search(r'\b(msan)=1', gyp_defines): |
+ if 'msan_track_origins=0' in gyp_defines: |
+ return 'msan-no-origins' |
+ if 'msan_track_origins=2' in gyp_defines: |
+ return 'msan-chained-origins' |
+ if 'msan_track_origins=' not in gyp_defines: |
+ # NB: must be the same as the default value in common.gypi |
+ return 'msan-chained-origins' |
+ raise Exception( |
+ "Prebuilt instrumented libraries not available for your configuration.") |
+ |
+ |
+def get_archive_name(gyp_defines): |
+ return "%s-%s.tgz" % (get_configuration(gyp_defines), get_ubuntu_release()) |
+ |
+ |
def main(args): |
+ gyp_defines = os.environ.get('GYP_DEFINES', '') |
+ if not 'use_prebuilt_instrumented_libraries=1' in gyp_defines: |
+ return 0 |
+ |
if not sys.platform.startswith('linux'): |
- raise Exception("Prebuilt instrumented libraries require Linux.") |
+ raise Exception("'use_prebuilt_instrumented_libraries=1' requires Linux.") |
- sha1file = args[0] |
- tarfile = args[1] |
+ archive_name = get_archive_name(gyp_defines) |
+ sha1file = '%s.sha1' % archive_name |
+ target_directory = os.path.join( |
+ BASE_DIR, 'third_party', 'instrumented_libraries', 'binaries') |
subprocess.check_call([ |
'download_from_google_storage', |
'--no_resume', |
'--no_auth', |
'--bucket', 'chromium-instrumented-libraries', |
- '-s', sha1file, '-o', tarfile]) |
+ '-s', sha1file], cwd=target_directory) |
return 0 |
if __name__ == '__main__': |
- # TODO(thomasanderson): Remove this once all third_party DEPS |
- # entires for this script are removed. |
- if (len(sys.argv) == 1): |
- sys.exit(0) |
sys.exit(main(sys.argv[1:])) |