Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 1 | # Copyright (c) 2016, the R8 project authors. Please see the AUTHORS file |
| 2 | # for details. All rights reserved. Use of this source code is governed by a |
| 3 | # BSD-style license that can be found in the LICENSE file. |
| 4 | |
| 5 | # Different utility functions used accross scripts |
| 6 | |
| 7 | import hashlib |
Clément Béra | 3718ad0 | 2023-09-05 14:12:48 +0200 | [diff] [blame] | 8 | import jdk |
Søren Gjesse | 6e5e584 | 2019-09-03 08:48:30 +0200 | [diff] [blame] | 9 | import json |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 10 | import os |
Tamas Kenez | 82efeb5 | 2017-06-12 13:56:22 +0200 | [diff] [blame] | 11 | import re |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 12 | import shutil |
| 13 | import subprocess |
| 14 | import sys |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 15 | import tarfile |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 16 | import tempfile |
Morten Krogh-Jespersen | 16e925d | 2019-01-25 14:40:38 +0100 | [diff] [blame] | 17 | import zipfile |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 18 | |
Ian Zerny | 3709765 | 2019-04-11 13:13:27 +0200 | [diff] [blame] | 19 | import defines |
Christoffer Quist Adamsen | 65ef298 | 2023-08-24 08:45:39 +0200 | [diff] [blame] | 20 | from thread_utils import print_thread |
Ian Zerny | 3709765 | 2019-04-11 13:13:27 +0200 | [diff] [blame] | 21 | |
Christoffer Quist Adamsen | 17879c1 | 2019-01-22 16:13:54 +0100 | [diff] [blame] | 22 | ANDROID_JAR_DIR = 'third_party/android_jar/lib-v{api}' |
| 23 | ANDROID_JAR = os.path.join(ANDROID_JAR_DIR, 'android.jar') |
Ian Zerny | 3f54e22 | 2019-02-12 10:51:17 +0100 | [diff] [blame] | 24 | TOOLS_DIR = defines.TOOLS_DIR |
| 25 | REPO_ROOT = defines.REPO_ROOT |
| 26 | THIRD_PARTY = defines.THIRD_PARTY |
Christoffer Quist Adamsen | 4d9fc51 | 2022-08-11 19:59:44 +0200 | [diff] [blame] | 27 | BUNDLETOOL_JAR_DIR = os.path.join(THIRD_PARTY, 'bundletool/bundletool-1.11.0') |
| 28 | BUNDLETOOL_JAR = os.path.join(BUNDLETOOL_JAR_DIR, 'bundletool-all-1.11.0.jar') |
Tamas Kenez | fc34cd8 | 2017-07-13 12:43:57 +0200 | [diff] [blame] | 29 | MEMORY_USE_TMP_FILE = 'memory_use.tmp' |
Tamas Kenez | 02bff03 | 2017-07-18 12:13:58 +0200 | [diff] [blame] | 30 | DEX_SEGMENTS_RESULT_PATTERN = re.compile('- ([^:]+): ([0-9]+)') |
Rico Wind | af9e8fb | 2023-10-10 09:25:26 +0200 | [diff] [blame] | 31 | |
Rico Wind | b65d75c | 2023-10-10 11:14:52 +0200 | [diff] [blame] | 32 | DEPENDENCIES_DIR = os.path.join(THIRD_PARTY, 'dependencies') |
| 33 | |
Mads Ager | 12a56bc | 2017-11-27 11:51:25 +0100 | [diff] [blame] | 34 | BUILD = os.path.join(REPO_ROOT, 'build') |
Morten Krogh-Jespersen | 51db2b0 | 2020-11-11 12:49:26 +0100 | [diff] [blame] | 35 | BUILD_JAVA_MAIN_DIR = os.path.join(BUILD, 'classes', 'java', 'main') |
Mads Ager | 12a56bc | 2017-11-27 11:51:25 +0100 | [diff] [blame] | 36 | LIBS = os.path.join(BUILD, 'libs') |
Clément Béra | 3718ad0 | 2023-09-05 14:12:48 +0200 | [diff] [blame] | 37 | CUSTOM_CONVERSION_DIR = os.path.join( |
Clément Béra | db79e5d | 2023-09-06 14:48:35 +0200 | [diff] [blame] | 38 | THIRD_PARTY, 'openjdk', 'custom_conversion') |
Mads Ager | 12a56bc | 2017-11-27 11:51:25 +0100 | [diff] [blame] | 39 | GENERATED_LICENSE_DIR = os.path.join(BUILD, 'generatedLicense') |
Mads Ager | a4911eb | 2017-11-22 13:19:36 +0100 | [diff] [blame] | 40 | SRC_ROOT = os.path.join(REPO_ROOT, 'src', 'main', 'java') |
Ian Zerny | 59dfa4c | 2019-10-25 10:34:36 +0200 | [diff] [blame] | 41 | REPO_SOURCE = 'https://r8.googlesource.com/r8' |
Søren Gjesse | dc9d8a2 | 2017-10-12 12:40:59 +0200 | [diff] [blame] | 42 | |
Rico Wind | 93755fb | 2023-10-09 13:29:29 +0200 | [diff] [blame] | 43 | GRADLE_TASK_CLEAN_TEST = ':test:cleanTest' |
| 44 | GRADLE_TASK_CONSOLIDATED_LICENSE = ':main:consolidatedLicense' |
| 45 | GRADLE_TASK_KEEP_ANNO_JAR = ':keepanno:keepAnnoAnnotationsJar' |
| 46 | GRADLE_TASK_R8 = ':main:r8WithRelocatedDeps' |
Christoffer Quist Adamsen | 31237c3 | 2023-10-13 10:59:53 +0200 | [diff] [blame] | 47 | GRADLE_TASK_R8LIB = ':test:assembleR8LibWithRelocatedDeps' |
| 48 | GRADLE_TASK_R8LIB_NO_DEPS = ':test:assembleR8LibNoDeps' |
| 49 | GRADLE_TASK_RETRACE = ':test:assembleRetraceLibWithRelocatedDeps' |
| 50 | GRADLE_TASK_RETRACE_NO_DEPS = ':test:assembleRetraceLibNoDeps' |
| 51 | GRADLE_TASK_SOURCE_JAR = ':test:packageSources' |
Rico Wind | 93755fb | 2023-10-09 13:29:29 +0200 | [diff] [blame] | 52 | GRADLE_TASK_SWISS_ARMY_KNIFE = ':main:swissArmyKnife' |
| 53 | GRADLE_TASK_TEST = ':test:test' |
Søren Gjesse | d54faae | 2023-10-12 09:07:54 +0200 | [diff] [blame] | 54 | GRADLE_TASK_ALL_TESTS_WITH_APPLY_MAPPING_JAR = ':test:allTestsWithApplyMapping' |
Rico Wind | 833710c | 2023-10-11 15:02:34 +0200 | [diff] [blame] | 55 | GRADLE_TASK_TEST_DEPS_JAR = ':test:allDepsJar' |
| 56 | GRADLE_TASK_TEST_JAR = ':test:allTestsJar' |
Rico Wind | 93755fb | 2023-10-09 13:29:29 +0200 | [diff] [blame] | 57 | |
Søren Gjesse | dc9d8a2 | 2017-10-12 12:40:59 +0200 | [diff] [blame] | 58 | R8 = 'r8' |
Tamas Kenez | 03ab76f | 2018-12-07 14:33:25 +0100 | [diff] [blame] | 59 | R8LIB = 'r8lib' |
Søren Gjesse | dc9d8a2 | 2017-10-12 12:40:59 +0200 | [diff] [blame] | 60 | |
Morten Krogh-Jespersen | 51db2b0 | 2020-11-11 12:49:26 +0100 | [diff] [blame] | 61 | ALL_DEPS_JAR = os.path.join(LIBS, 'deps_all.jar') |
Rico Wind | 74fab30 | 2017-10-02 07:25:33 +0200 | [diff] [blame] | 62 | R8_JAR = os.path.join(LIBS, 'r8.jar') |
Tamas Kenez | 03ab76f | 2018-12-07 14:33:25 +0100 | [diff] [blame] | 63 | R8LIB_JAR = os.path.join(LIBS, 'r8lib.jar') |
Rico Wind | 158ef9f | 2022-05-19 11:08:30 +0200 | [diff] [blame] | 64 | R8LIB_MAP = '%s.map' % R8LIB_JAR |
Mads Ager | b10c07f | 2017-11-27 13:25:52 +0100 | [diff] [blame] | 65 | R8_SRC_JAR = os.path.join(LIBS, 'r8-src.jar') |
Tamas Kenez | 03ab76f | 2018-12-07 14:33:25 +0100 | [diff] [blame] | 66 | R8LIB_EXCLUDE_DEPS_JAR = os.path.join(LIBS, 'r8lib-exclude-deps.jar') |
Tamas Kenez | 180be09 | 2018-12-05 15:23:06 +0100 | [diff] [blame] | 67 | R8_FULL_EXCLUDE_DEPS_JAR = os.path.join(LIBS, 'r8-full-exclude-deps.jar') |
Morten Krogh-Jespersen | 98ee89a | 2021-10-25 20:59:02 +0200 | [diff] [blame] | 68 | R8RETRACE_JAR = os.path.join(LIBS, 'r8retrace.jar') |
| 69 | R8RETRACE_EXCLUDE_DEPS_JAR = os.path.join(LIBS, 'r8retrace-exclude-deps.jar') |
Ian Zerny | 161ff74 | 2022-01-20 12:39:40 +0100 | [diff] [blame] | 70 | R8_TESTS_JAR = os.path.join(LIBS, 'r8tests.jar') |
| 71 | R8LIB_TESTS_JAR = os.path.join(LIBS, 'r8libtestdeps-cf.jar') |
| 72 | R8_TESTS_DEPS_JAR = os.path.join(LIBS, 'test_deps_all.jar') |
| 73 | R8LIB_TESTS_DEPS_JAR = R8_TESTS_DEPS_JAR |
Rico Wind | 8fc8bfa | 2019-03-22 09:57:36 +0100 | [diff] [blame] | 74 | MAVEN_ZIP_LIB = os.path.join(LIBS, 'r8lib.zip') |
Clément Béra | 3718ad0 | 2023-09-05 14:12:48 +0200 | [diff] [blame] | 75 | LIBRARY_DESUGAR_CONVERSIONS_LEGACY_ZIP = os.path.join( |
| 76 | CUSTOM_CONVERSION_DIR, 'library_desugar_conversions_legacy.jar') |
| 77 | LIBRARY_DESUGAR_CONVERSIONS_ZIP = os.path.join( |
| 78 | CUSTOM_CONVERSION_DIR, 'library_desugar_conversions.jar') |
Ian Zerny | f13d18f | 2023-05-24 12:50:37 +0200 | [diff] [blame] | 79 | KEEPANNO_ANNOTATIONS_JAR = os.path.join(LIBS, 'keepanno-annotations.jar') |
Søren Gjesse | 17fc67d | 2019-12-04 14:50:17 +0100 | [diff] [blame] | 80 | |
Søren Gjesse | 6e5e584 | 2019-09-03 08:48:30 +0200 | [diff] [blame] | 81 | DESUGAR_CONFIGURATION = os.path.join( |
Søren Gjesse | 927a92e | 2019-12-04 15:18:06 +0100 | [diff] [blame] | 82 | 'src', 'library_desugar', 'desugar_jdk_libs.json') |
Søren Gjesse | ee086b2 | 2020-10-30 11:46:39 +0100 | [diff] [blame] | 83 | DESUGAR_IMPLEMENTATION = os.path.join( |
Søren Gjesse | 3dc207b | 2021-02-15 09:45:30 +0100 | [diff] [blame] | 84 | 'third_party', 'openjdk', 'desugar_jdk_libs', 'desugar_jdk_libs.jar') |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 85 | DESUGAR_CONFIGURATION_JDK11_LEGACY = os.path.join( |
| 86 | 'src', 'library_desugar', 'jdk11', 'desugar_jdk_libs_legacy.json') |
Søren Gjesse | 2b04769 | 2022-08-19 16:34:38 +0200 | [diff] [blame] | 87 | DESUGAR_CONFIGURATION_JDK11_MINIMAL = os.path.join( |
| 88 | 'src', 'library_desugar', 'jdk11', 'desugar_jdk_libs_minimal.json') |
| 89 | DESUGAR_CONFIGURATION_JDK11 = os.path.join( |
| 90 | 'src', 'library_desugar', 'jdk11', 'desugar_jdk_libs.json') |
| 91 | DESUGAR_CONFIGURATION_JDK11_NIO = os.path.join( |
| 92 | 'src', 'library_desugar', 'jdk11', 'desugar_jdk_libs_nio.json') |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 93 | DESUGAR_IMPLEMENTATION_JDK11 = os.path.join( |
| 94 | 'third_party', 'openjdk', 'desugar_jdk_libs_11', 'desugar_jdk_libs.jar') |
Søren Gjesse | 6e5e584 | 2019-09-03 08:48:30 +0200 | [diff] [blame] | 95 | DESUGAR_CONFIGURATION_MAVEN_ZIP = os.path.join( |
| 96 | LIBS, 'desugar_jdk_libs_configuration.zip') |
Søren Gjesse | e18fa6e | 2022-06-24 15:14:53 +0200 | [diff] [blame] | 97 | DESUGAR_CONFIGURATION_JDK11_LEGACY_MAVEN_ZIP = os.path.join( |
| 98 | LIBS, 'desugar_jdk_libs_configuration_jdk11_legacy.zip') |
Søren Gjesse | 2b04769 | 2022-08-19 16:34:38 +0200 | [diff] [blame] | 99 | DESUGAR_CONFIGURATION_JDK11_MINIMAL_MAVEN_ZIP = os.path.join( |
| 100 | LIBS, 'desugar_jdk_libs_configuration_jdk11_minimal.zip') |
| 101 | DESUGAR_CONFIGURATION_JDK11_MAVEN_ZIP = os.path.join( |
| 102 | LIBS, 'desugar_jdk_libs_configuration_jdk11.zip') |
| 103 | DESUGAR_CONFIGURATION_JDK11_NIO_MAVEN_ZIP = os.path.join( |
| 104 | LIBS, 'desugar_jdk_libs_configuration_jdk11_nio.zip') |
Mads Ager | 12a56bc | 2017-11-27 11:51:25 +0100 | [diff] [blame] | 105 | GENERATED_LICENSE = os.path.join(GENERATED_LICENSE_DIR, 'LICENSE') |
Mathias Rav | 3fb4a3a | 2018-05-29 15:41:36 +0200 | [diff] [blame] | 106 | RT_JAR = os.path.join(REPO_ROOT, 'third_party/openjdk/openjdk-rt-1.8/rt.jar') |
Mathias Rav | b46dc00 | 2018-06-06 09:37:11 +0200 | [diff] [blame] | 107 | R8LIB_KEEP_RULES = os.path.join(REPO_ROOT, 'src/main/keep.txt') |
Morten Krogh-Jespersen | 480784d | 2019-02-05 08:10:46 +0100 | [diff] [blame] | 108 | CF_SEGMENTS_TOOL = os.path.join(THIRD_PARTY, 'cf_segments') |
Morten Krogh-Jespersen | 38c7ca0 | 2019-02-04 10:39:57 +0100 | [diff] [blame] | 109 | PINNED_R8_JAR = os.path.join(REPO_ROOT, 'third_party/r8/r8.jar') |
| 110 | PINNED_PGR8_JAR = os.path.join(REPO_ROOT, 'third_party/r8/r8-pg6.0.1.jar') |
Rico Wind | 8bd693f | 2023-10-13 11:29:32 +0200 | [diff] [blame^] | 111 | |
Morten Krogh-Jespersen | 45d7a7b | 2020-11-02 08:31:09 +0100 | [diff] [blame] | 112 | OPENSOURCE_DUMPS_DIR = os.path.join(THIRD_PARTY, 'opensource-apps') |
Morten Krogh-Jespersen | 8622274 | 2021-03-02 11:13:33 +0100 | [diff] [blame] | 113 | INTERNAL_DUMPS_DIR = os.path.join(THIRD_PARTY, 'internal-apps') |
Søren Gjesse | 1c115b5 | 2019-08-14 12:43:57 +0200 | [diff] [blame] | 114 | BAZEL_SHA_FILE = os.path.join(THIRD_PARTY, 'bazel.tar.gz.sha1') |
| 115 | BAZEL_TOOL = os.path.join(THIRD_PARTY, 'bazel') |
Søren Gjesse | 699f636 | 2019-10-09 14:56:33 +0200 | [diff] [blame] | 116 | JAVA8_SHA_FILE = os.path.join(THIRD_PARTY, 'openjdk', 'jdk8', 'linux-x86.tar.gz.sha1') |
Søren Gjesse | ef19577 | 2021-03-11 16:04:42 +0100 | [diff] [blame] | 117 | JAVA11_SHA_FILE = os.path.join(THIRD_PARTY, 'openjdk', 'jdk-11', 'linux.tar.gz.sha1') |
Søren Gjesse | eddc661 | 2022-09-02 15:38:39 +0200 | [diff] [blame] | 118 | DESUGAR_JDK_LIBS_11_SHA_FILE = os.path.join(THIRD_PARTY, 'openjdk', 'desugar_jdk_libs_11.tar.gz.sha1') |
Christoffer Quist Adamsen | 1ca046c | 2021-02-21 11:25:16 +0100 | [diff] [blame] | 119 | IGNORE_WARNINGS_RULES = os.path.join(REPO_ROOT, 'src', 'test', 'ignorewarnings.rules') |
Morten Krogh-Jespersen | 220e570 | 2019-02-27 12:57:01 +0100 | [diff] [blame] | 120 | ANDROID_HOME_ENVIROMENT_NAME = "ANDROID_HOME" |
| 121 | ANDROID_TOOLS_VERSION_ENVIRONMENT_NAME = "ANDROID_TOOLS_VERSION" |
Morten Krogh-Jespersen | c8efedd | 2019-01-28 11:36:17 +0100 | [diff] [blame] | 122 | USER_HOME = os.path.expanduser('~') |
Morten Krogh-Jespersen | 220e570 | 2019-02-27 12:57:01 +0100 | [diff] [blame] | 123 | |
Morten Krogh-Jespersen | 0981b72 | 2019-10-09 10:00:33 +0200 | [diff] [blame] | 124 | R8_TEST_RESULTS_BUCKET = 'r8-test-results' |
Rico Wind | 635b2de | 2022-04-25 10:35:14 +0200 | [diff] [blame] | 125 | R8_INTERNAL_TEST_RESULTS_BUCKET = 'r8-internal-test-results' |
Morten Krogh-Jespersen | 0981b72 | 2019-10-09 10:00:33 +0200 | [diff] [blame] | 126 | |
| 127 | def archive_file(name, gs_dir, src_file): |
| 128 | gs_file = '%s/%s' % (gs_dir, name) |
Rico Wind | 0342428 | 2022-04-26 15:09:51 +0200 | [diff] [blame] | 129 | upload_file_to_cloud_storage(src_file, gs_file) |
Morten Krogh-Jespersen | 0981b72 | 2019-10-09 10:00:33 +0200 | [diff] [blame] | 130 | |
| 131 | def archive_value(name, gs_dir, value): |
| 132 | with TempDir() as temp: |
Rico Wind | 8bd693f | 2023-10-13 11:29:32 +0200 | [diff] [blame^] | 133 | temparchive = os.path.join(temp, name) |
| 134 | with open(temparchive, 'w') as f: |
Morten Krogh-Jespersen | 0981b72 | 2019-10-09 10:00:33 +0200 | [diff] [blame] | 135 | f.write(str(value)) |
Rico Wind | 8bd693f | 2023-10-13 11:29:32 +0200 | [diff] [blame^] | 136 | archive_file(name, gs_dir, temparchive) |
Morten Krogh-Jespersen | 0981b72 | 2019-10-09 10:00:33 +0200 | [diff] [blame] | 137 | |
Christoffer Quist Adamsen | 4d38d03 | 2021-04-20 12:31:31 +0200 | [diff] [blame] | 138 | def find_cloud_storage_file_from_options(name, options, orElse=None): |
| 139 | # Import archive on-demand since archive depends on utils. |
| 140 | from archive import GetUploadDestination |
| 141 | hash_or_version = find_hash_or_version_from_options(options) |
| 142 | if not hash_or_version: |
| 143 | return orElse |
| 144 | is_hash = options.commit_hash is not None |
| 145 | download_path = GetUploadDestination(hash_or_version, name, is_hash) |
| 146 | if file_exists_on_cloud_storage(download_path): |
| 147 | out = tempfile.NamedTemporaryFile().name |
| 148 | download_file_from_cloud_storage(download_path, out) |
| 149 | return out |
| 150 | else: |
| 151 | raise Exception('Could not find file {} from hash/version: {}.' |
| 152 | .format(name, hash_or_version)) |
| 153 | |
| 154 | def find_r8_jar_from_options(options): |
| 155 | return find_cloud_storage_file_from_options('r8.jar', options) |
| 156 | |
Christoffer Quist Adamsen | 4d38d03 | 2021-04-20 12:31:31 +0200 | [diff] [blame] | 157 | |
| 158 | def find_hash_or_version_from_options(options): |
| 159 | if options.tag: |
| 160 | return find_hash_or_version_from_tag(options.tag) |
| 161 | else: |
| 162 | return options.commit_hash or options.version |
| 163 | |
| 164 | def find_hash_or_version_from_tag(tag_or_hash): |
Rico Wind | fd18637 | 2022-02-28 08:55:48 +0100 | [diff] [blame] | 165 | info = subprocess.check_output([ |
Christoffer Quist Adamsen | 4d38d03 | 2021-04-20 12:31:31 +0200 | [diff] [blame] | 166 | 'git', |
| 167 | 'show', |
| 168 | tag_or_hash, |
| 169 | '-s', |
Rico Wind | fd18637 | 2022-02-28 08:55:48 +0100 | [diff] [blame] | 170 | '--format=oneline']).decode('utf-8').splitlines()[-1].split() |
Christoffer Quist Adamsen | 4d38d03 | 2021-04-20 12:31:31 +0200 | [diff] [blame] | 171 | # The info should be on the following form [hash,"Version",version] |
| 172 | if len(info) == 3 and len(info[0]) == 40 and info[1] == "Version": |
| 173 | return info[2] |
| 174 | return None |
| 175 | |
Morten Krogh-Jespersen | 220e570 | 2019-02-27 12:57:01 +0100 | [diff] [blame] | 176 | def getAndroidHome(): |
| 177 | return os.environ.get( |
| 178 | ANDROID_HOME_ENVIROMENT_NAME, os.path.join(USER_HOME, 'Android', 'Sdk')) |
| 179 | |
| 180 | def getAndroidBuildTools(): |
Christoffer Quist Adamsen | 8c803b4 | 2022-05-31 10:36:17 +0200 | [diff] [blame] | 181 | if ANDROID_TOOLS_VERSION_ENVIRONMENT_NAME in os.environ: |
| 182 | version = os.environ.get(ANDROID_TOOLS_VERSION_ENVIRONMENT_NAME) |
| 183 | build_tools_dir = os.path.join(getAndroidHome(), 'build-tools', version) |
| 184 | assert os.path.exists(build_tools_dir) |
| 185 | return build_tools_dir |
| 186 | else: |
Morten Krogh-Jespersen | 9c1e702 | 2023-02-07 13:56:35 +0100 | [diff] [blame] | 187 | versions = ['33.0.1', '32.0.0'] |
Christoffer Quist Adamsen | 8c803b4 | 2022-05-31 10:36:17 +0200 | [diff] [blame] | 188 | for version in versions: |
| 189 | build_tools_dir = os.path.join(getAndroidHome(), 'build-tools', version) |
| 190 | if os.path.exists(build_tools_dir): |
| 191 | return build_tools_dir |
| 192 | raise Exception('Unable to find Android build-tools') |
Morten Krogh-Jespersen | c8efedd | 2019-01-28 11:36:17 +0100 | [diff] [blame] | 193 | |
Christoffer Quist Adamsen | 5c9ded1 | 2021-01-14 14:29:37 +0100 | [diff] [blame] | 194 | def is_python3(): |
| 195 | return sys.version_info.major == 3 |
| 196 | |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 197 | def Print(s, quiet=False): |
| 198 | if quiet: |
| 199 | return |
| 200 | print(s) |
| 201 | |
| 202 | def Warn(message): |
| 203 | CRED = '\033[91m' |
| 204 | CEND = '\033[0m' |
| 205 | print(CRED + message + CEND) |
| 206 | |
Christoffer Quist Adamsen | 65ef298 | 2023-08-24 08:45:39 +0200 | [diff] [blame] | 207 | def PrintCmd(cmd, env=None, quiet=False, worker_id=None): |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 208 | if quiet: |
| 209 | return |
| 210 | if type(cmd) is list: |
| 211 | cmd = ' '.join(cmd) |
| 212 | if env: |
| 213 | env = ' '.join(['{}=\"{}\"'.format(x, y) for x, y in env.iteritems()]) |
Christoffer Quist Adamsen | 65ef298 | 2023-08-24 08:45:39 +0200 | [diff] [blame] | 214 | print_thread('Running: {} {}'.format(env, cmd), worker_id) |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 215 | else: |
Christoffer Quist Adamsen | 65ef298 | 2023-08-24 08:45:39 +0200 | [diff] [blame] | 216 | print_thread('Running: {}'.format(cmd), worker_id) |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 217 | # I know this will hit os on windows eventually if we don't do this. |
| 218 | sys.stdout.flush() |
| 219 | |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 220 | class ProgressLogger(object): |
| 221 | CLEAR_LINE = '\033[K' |
| 222 | UP = '\033[F' |
| 223 | |
| 224 | def __init__(self, quiet=False): |
| 225 | self._count = 0 |
| 226 | self._has_printed = False |
| 227 | self._quiet = quiet |
| 228 | |
| 229 | def log(self, text): |
Christoffer Quist Adamsen | 7607ebe | 2022-06-28 11:52:46 +0200 | [diff] [blame] | 230 | if len(text.strip()) == 0: |
| 231 | return |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 232 | if self._quiet: |
| 233 | if self._has_printed: |
| 234 | sys.stdout.write(ProgressLogger.UP + ProgressLogger.CLEAR_LINE) |
| 235 | if len(text) > 140: |
| 236 | text = text[0:140] + '...' |
| 237 | print(text) |
| 238 | self._has_printed = True |
| 239 | |
| 240 | def done(self): |
| 241 | if self._quiet and self._has_printed: |
| 242 | sys.stdout.write(ProgressLogger.UP + ProgressLogger.CLEAR_LINE) |
| 243 | print('') |
| 244 | sys.stdout.write(ProgressLogger.UP) |
| 245 | |
Morten Krogh-Jespersen | 7cdd3a7 | 2019-03-13 14:58:25 +0100 | [diff] [blame] | 246 | def RunCmd(cmd, env_vars=None, quiet=False, fail=True, logging=True): |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 247 | PrintCmd(cmd, env=env_vars, quiet=quiet) |
| 248 | env = os.environ.copy() |
| 249 | if env_vars: |
| 250 | env.update(env_vars) |
| 251 | process = subprocess.Popen( |
| 252 | cmd, env=env, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) |
| 253 | stdout = [] |
Morten Krogh-Jespersen | 7cdd3a7 | 2019-03-13 14:58:25 +0100 | [diff] [blame] | 254 | logger = ProgressLogger(quiet=quiet) if logging else None |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 255 | failed = False |
| 256 | while True: |
Rico Wind | 744ba75 | 2021-01-22 06:24:49 +0100 | [diff] [blame] | 257 | line = process.stdout.readline().decode('utf-8') |
| 258 | if line != '': |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 259 | stripped = line.rstrip() |
| 260 | stdout.append(stripped) |
Morten Krogh-Jespersen | 7cdd3a7 | 2019-03-13 14:58:25 +0100 | [diff] [blame] | 261 | if logger: |
| 262 | logger.log(stripped) |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 263 | # TODO(christofferqa): r8 should fail with non-zero exit code. |
Morten Krogh-Jespersen | 121c47b | 2019-01-25 09:57:21 +0100 | [diff] [blame] | 264 | if ('AssertionError:' in stripped |
| 265 | or 'CompilationError:' in stripped |
| 266 | or 'CompilationFailedException:' in stripped |
Morten Krogh-Jespersen | 5d02a6b | 2019-10-29 14:48:56 +0100 | [diff] [blame] | 267 | or 'Compilation failed' in stripped |
| 268 | or 'FAILURE:' in stripped |
| 269 | or 'org.gradle.api.ProjectConfigurationException' in stripped |
| 270 | or 'BUILD FAILED' in stripped): |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 271 | failed = True |
| 272 | else: |
Morten Krogh-Jespersen | 7cdd3a7 | 2019-03-13 14:58:25 +0100 | [diff] [blame] | 273 | if logger: |
| 274 | logger.done() |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 275 | exit_code = process.poll() |
| 276 | if exit_code or failed: |
| 277 | for line in stdout: |
| 278 | Warn(line) |
Christoffer Quist Adamsen | 7cf4c56 | 2019-03-07 10:57:33 +0100 | [diff] [blame] | 279 | if fail: |
| 280 | raise subprocess.CalledProcessError( |
| 281 | exit_code or -1, cmd, output='\n'.join(stdout)) |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 282 | return stdout |
| 283 | |
Rico Wind | f80f5a2 | 2017-06-16 09:15:57 +0200 | [diff] [blame] | 284 | def IsWindows(): |
Ian Zerny | 3f54e22 | 2019-02-12 10:51:17 +0100 | [diff] [blame] | 285 | return defines.IsWindows() |
Ian Zerny | 5fffb0a | 2019-02-11 13:54:22 +0100 | [diff] [blame] | 286 | |
Ian Zerny | 5fffb0a | 2019-02-11 13:54:22 +0100 | [diff] [blame] | 287 | def EnsureDepFromGoogleCloudStorage(dep, tgz, sha1, msg): |
Morten Krogh-Jespersen | 3295c84 | 2023-09-18 13:48:22 +0200 | [diff] [blame] | 288 | if (not os.path.exists(dep) |
| 289 | or not os.path.exists(tgz) |
| 290 | or os.path.getmtime(tgz) < os.path.getmtime(sha1)): |
Ian Zerny | 5fffb0a | 2019-02-11 13:54:22 +0100 | [diff] [blame] | 291 | DownloadFromGoogleCloudStorage(sha1) |
| 292 | # Update the mtime of the tar file to make sure we do not run again unless |
| 293 | # there is an update. |
| 294 | os.utime(tgz, None) |
| 295 | else: |
Rico Wind | 3d369b4 | 2021-01-12 10:26:24 +0100 | [diff] [blame] | 296 | print('Ensure cloud dependency:', msg, 'present') |
Rico Wind | f80f5a2 | 2017-06-16 09:15:57 +0200 | [diff] [blame] | 297 | |
Jean-Marie Henaff | e4e36d1 | 2018-04-05 10:33:50 +0200 | [diff] [blame] | 298 | def DownloadFromX20(sha1_file): |
| 299 | download_script = os.path.join(REPO_ROOT, 'tools', 'download_from_x20.py') |
| 300 | cmd = [download_script, sha1_file] |
| 301 | PrintCmd(cmd) |
| 302 | subprocess.check_call(cmd) |
| 303 | |
Rico Wind | 5959392 | 2021-03-03 09:12:36 +0100 | [diff] [blame] | 304 | def DownloadFromGoogleCloudStorage(sha1_file, bucket='r8-deps', auth=False, |
| 305 | quiet=False): |
Rico Wind | f80f5a2 | 2017-06-16 09:15:57 +0200 | [diff] [blame] | 306 | suffix = '.bat' if IsWindows() else '' |
| 307 | download_script = 'download_from_google_storage%s' % suffix |
Rico Wind | 533e3ce | 2019-04-04 10:26:12 +0200 | [diff] [blame] | 308 | cmd = [download_script] |
| 309 | if not auth: |
| 310 | cmd.append('-n') |
| 311 | cmd.extend(['-b', bucket, '-u', '-s', sha1_file]) |
Rico Wind | 5959392 | 2021-03-03 09:12:36 +0100 | [diff] [blame] | 312 | if not quiet: |
| 313 | PrintCmd(cmd) |
| 314 | subprocess.check_call(cmd) |
| 315 | else: |
| 316 | subprocess.check_output(cmd) |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 317 | |
| 318 | def get_sha1(filename): |
| 319 | sha1 = hashlib.sha1() |
| 320 | with open(filename, 'rb') as f: |
| 321 | while True: |
| 322 | chunk = f.read(1024*1024) |
| 323 | if not chunk: |
| 324 | break |
| 325 | sha1.update(chunk) |
| 326 | return sha1.hexdigest() |
| 327 | |
Ian Zerny | c2de7b7 | 2023-09-06 20:52:16 +0200 | [diff] [blame] | 328 | def get_HEAD_branch(): |
| 329 | result = subprocess.check_output(['git', 'rev-parse', '--abbrev-ref', 'HEAD']).decode('utf-8') |
| 330 | return result.strip() |
| 331 | |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 332 | def get_HEAD_sha1(): |
Morten Krogh-Jespersen | e0ce6a3 | 2019-02-07 11:44:45 +0100 | [diff] [blame] | 333 | return get_HEAD_sha1_for_checkout(REPO_ROOT) |
| 334 | |
Ian Zerny | 9f0345d | 2023-09-07 11:15:00 +0200 | [diff] [blame] | 335 | def get_HEAD_diff_stat(): |
| 336 | return subprocess.check_output(['git', 'diff', '--stat']).decode('utf-8') |
| 337 | |
Morten Krogh-Jespersen | e0ce6a3 | 2019-02-07 11:44:45 +0100 | [diff] [blame] | 338 | def get_HEAD_sha1_for_checkout(checkout): |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 339 | cmd = ['git', 'rev-parse', 'HEAD'] |
| 340 | PrintCmd(cmd) |
Morten Krogh-Jespersen | e0ce6a3 | 2019-02-07 11:44:45 +0100 | [diff] [blame] | 341 | with ChangedWorkingDirectory(checkout): |
Christoffer Quist Adamsen | 50930e4 | 2021-01-20 11:55:12 +0100 | [diff] [blame] | 342 | return subprocess.check_output(cmd).decode('utf-8').strip() |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 343 | |
Tamas Kenez | 971eec6 | 2017-05-24 11:08:40 +0200 | [diff] [blame] | 344 | def makedirs_if_needed(path): |
| 345 | try: |
| 346 | os.makedirs(path) |
| 347 | except OSError: |
| 348 | if not os.path.isdir(path): |
| 349 | raise |
| 350 | |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 351 | def get_gsutil(): |
| 352 | return 'gsutil.py' if os.name != 'nt' else 'gsutil.py.bat' |
| 353 | |
Rico Wind | 0342428 | 2022-04-26 15:09:51 +0200 | [diff] [blame] | 354 | def upload_file_to_cloud_storage(source, destination): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 355 | cmd = [get_gsutil(), 'cp'] |
Rico Wind | 800fd71 | 2018-09-24 11:29:33 +0200 | [diff] [blame] | 356 | cmd += [source, destination] |
Rico Wind | b4621c1 | 2017-08-28 12:48:53 +0200 | [diff] [blame] | 357 | PrintCmd(cmd) |
| 358 | subprocess.check_call(cmd) |
| 359 | |
Rico Wind | 139eece | 2018-09-25 09:42:09 +0200 | [diff] [blame] | 360 | def delete_file_from_cloud_storage(destination): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 361 | cmd = [get_gsutil(), 'rm', destination] |
Rico Wind | 139eece | 2018-09-25 09:42:09 +0200 | [diff] [blame] | 362 | PrintCmd(cmd) |
| 363 | subprocess.check_call(cmd) |
| 364 | |
Rico Wind | 4fd2dda | 2018-09-26 17:41:45 +0200 | [diff] [blame] | 365 | def ls_files_on_cloud_storage(destination): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 366 | cmd = [get_gsutil(), 'ls', destination] |
Rico Wind | 4fd2dda | 2018-09-26 17:41:45 +0200 | [diff] [blame] | 367 | PrintCmd(cmd) |
Rico Wind | fd18637 | 2022-02-28 08:55:48 +0100 | [diff] [blame] | 368 | return subprocess.check_output(cmd).decode('utf-8') |
Rico Wind | 4fd2dda | 2018-09-26 17:41:45 +0200 | [diff] [blame] | 369 | |
Rico Wind | 139eece | 2018-09-25 09:42:09 +0200 | [diff] [blame] | 370 | def cat_file_on_cloud_storage(destination, ignore_errors=False): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 371 | cmd = [get_gsutil(), 'cat', destination] |
Rico Wind | 139eece | 2018-09-25 09:42:09 +0200 | [diff] [blame] | 372 | PrintCmd(cmd) |
| 373 | try: |
Rico Wind | 8dadb31 | 2022-02-28 08:40:20 +0100 | [diff] [blame] | 374 | return subprocess.check_output(cmd).decode('utf-8').strip() |
Rico Wind | 139eece | 2018-09-25 09:42:09 +0200 | [diff] [blame] | 375 | except subprocess.CalledProcessError as e: |
| 376 | if ignore_errors: |
| 377 | return '' |
| 378 | else: |
| 379 | raise e |
| 380 | |
| 381 | def file_exists_on_cloud_storage(destination): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 382 | cmd = [get_gsutil(), 'ls', destination] |
Rico Wind | 139eece | 2018-09-25 09:42:09 +0200 | [diff] [blame] | 383 | PrintCmd(cmd) |
| 384 | return subprocess.call(cmd) == 0 |
| 385 | |
Christoffer Quist Adamsen | 870fa46 | 2020-12-15 10:50:54 +0100 | [diff] [blame] | 386 | def download_file_from_cloud_storage(source, destination, quiet=False): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 387 | cmd = [get_gsutil(), 'cp', source, destination] |
Christoffer Quist Adamsen | 870fa46 | 2020-12-15 10:50:54 +0100 | [diff] [blame] | 388 | PrintCmd(cmd, quiet=quiet) |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 389 | subprocess.check_call(cmd) |
| 390 | |
Morten Krogh-Jespersen | 220e570 | 2019-02-27 12:57:01 +0100 | [diff] [blame] | 391 | def create_archive(name, sources=None): |
| 392 | if not sources: |
| 393 | sources = [name] |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 394 | tarname = '%s.tar.gz' % name |
| 395 | with tarfile.open(tarname, 'w:gz') as tar: |
Morten Krogh-Jespersen | 5409086 | 2019-02-19 11:31:10 +0100 | [diff] [blame] | 396 | for source in sources: |
| 397 | tar.add(source) |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 398 | return tarname |
| 399 | |
| 400 | def extract_dir(filename): |
| 401 | return filename[0:len(filename) - len('.tar.gz')] |
| 402 | |
| 403 | def unpack_archive(filename): |
| 404 | dest_dir = extract_dir(filename) |
| 405 | if os.path.exists(dest_dir): |
Rico Wind | 3d369b4 | 2021-01-12 10:26:24 +0100 | [diff] [blame] | 406 | print('Deleting existing dir %s' % dest_dir) |
Jean-Marie Henaff | 7a64eec | 2018-05-31 15:30:35 +0200 | [diff] [blame] | 407 | shutil.rmtree(dest_dir) |
| 408 | dirname = os.path.dirname(os.path.abspath(filename)) |
| 409 | with tarfile.open(filename, 'r:gz') as tar: |
| 410 | tar.extractall(path=dirname) |
| 411 | |
Morten Krogh-Jespersen | ec3047b | 2020-08-18 13:09:06 +0200 | [diff] [blame] | 412 | def check_gcert(): |
Ian Zerny | 86c4cfd | 2022-01-17 13:43:37 +0100 | [diff] [blame] | 413 | status = subprocess.call(['gcertstatus']) |
| 414 | if status != 0: |
| 415 | subprocess.check_call(['gcert']) |
Morten Krogh-Jespersen | 5409086 | 2019-02-19 11:31:10 +0100 | [diff] [blame] | 416 | |
Rico Wind | 1a29c4f | 2018-01-25 08:43:08 +0100 | [diff] [blame] | 417 | # Note that gcs is eventually consistent with regards to list operations. |
| 418 | # This is not a problem in our case, but don't ever use this method |
| 419 | # for synchronization. |
| 420 | def cloud_storage_exists(destination): |
Rico Wind | ef7420c | 2021-10-14 13:16:01 +0200 | [diff] [blame] | 421 | cmd = [get_gsutil(), 'ls', destination] |
Rico Wind | 1a29c4f | 2018-01-25 08:43:08 +0100 | [diff] [blame] | 422 | PrintCmd(cmd) |
| 423 | exit_code = subprocess.call(cmd) |
| 424 | return exit_code == 0 |
| 425 | |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 426 | class TempDir(object): |
Søren Gjesse | cbeae78 | 2019-05-21 14:14:25 +0200 | [diff] [blame] | 427 | def __init__(self, prefix='', delete=True): |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 428 | self._temp_dir = None |
| 429 | self._prefix = prefix |
Søren Gjesse | cbeae78 | 2019-05-21 14:14:25 +0200 | [diff] [blame] | 430 | self._delete = delete |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 431 | |
| 432 | def __enter__(self): |
| 433 | self._temp_dir = tempfile.mkdtemp(self._prefix) |
| 434 | return self._temp_dir |
| 435 | |
| 436 | def __exit__(self, *_): |
Søren Gjesse | cbeae78 | 2019-05-21 14:14:25 +0200 | [diff] [blame] | 437 | if self._delete: |
| 438 | shutil.rmtree(self._temp_dir, ignore_errors=True) |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 439 | |
| 440 | class ChangedWorkingDirectory(object): |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 441 | def __init__(self, working_directory, quiet=False): |
| 442 | self._quiet = quiet |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 443 | self._working_directory = working_directory |
| 444 | |
| 445 | def __enter__(self): |
| 446 | self._old_cwd = os.getcwd() |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 447 | if not self._quiet: |
Rico Wind | 3d369b4 | 2021-01-12 10:26:24 +0100 | [diff] [blame] | 448 | print('Enter directory:', self._working_directory) |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 449 | os.chdir(self._working_directory) |
| 450 | |
| 451 | def __exit__(self, *_): |
Christoffer Quist Adamsen | 1de3dde | 2019-01-24 13:17:46 +0100 | [diff] [blame] | 452 | if not self._quiet: |
Rico Wind | 3d369b4 | 2021-01-12 10:26:24 +0100 | [diff] [blame] | 453 | print('Enter directory:', self._old_cwd) |
Mads Ager | 418d1ca | 2017-05-22 09:35:49 +0200 | [diff] [blame] | 454 | os.chdir(self._old_cwd) |
Tamas Kenez | 82efeb5 | 2017-06-12 13:56:22 +0200 | [diff] [blame] | 455 | |
| 456 | # Reading Android CTS test_result.xml |
| 457 | |
| 458 | class CtsModule(object): |
| 459 | def __init__(self, module_name): |
| 460 | self.name = module_name |
| 461 | |
| 462 | class CtsTestCase(object): |
| 463 | def __init__(self, test_case_name): |
| 464 | self.name = test_case_name |
| 465 | |
| 466 | class CtsTest(object): |
| 467 | def __init__(self, test_name, outcome): |
| 468 | self.name = test_name |
| 469 | self.outcome = outcome |
| 470 | |
| 471 | # Generator yielding CtsModule, CtsTestCase or CtsTest from |
| 472 | # reading through a CTS test_result.xml file. |
| 473 | def read_cts_test_result(file_xml): |
| 474 | re_module = re.compile('<Module name="([^"]*)"') |
| 475 | re_test_case = re.compile('<TestCase name="([^"]*)"') |
| 476 | re_test = re.compile('<Test result="(pass|fail)" name="([^"]*)"') |
| 477 | with open(file_xml) as f: |
| 478 | for line in f: |
| 479 | m = re_module.search(line) |
| 480 | if m: |
| 481 | yield CtsModule(m.groups()[0]) |
| 482 | continue |
| 483 | m = re_test_case.search(line) |
| 484 | if m: |
| 485 | yield CtsTestCase(m.groups()[0]) |
| 486 | continue |
| 487 | m = re_test.search(line) |
| 488 | if m: |
| 489 | outcome = m.groups()[0] |
Rico Wind | f80f5a2 | 2017-06-16 09:15:57 +0200 | [diff] [blame] | 490 | assert outcome in ['fail', 'pass'] |
Tamas Kenez | 82efeb5 | 2017-06-12 13:56:22 +0200 | [diff] [blame] | 491 | yield CtsTest(m.groups()[1], outcome == 'pass') |
Tamas Kenez | fc34cd8 | 2017-07-13 12:43:57 +0200 | [diff] [blame] | 492 | |
| 493 | def grep_memoryuse(logfile): |
| 494 | re_vmhwm = re.compile('^VmHWM:[ \t]*([0-9]+)[ \t]*([a-zA-Z]*)') |
| 495 | result = None |
| 496 | with open(logfile) as f: |
| 497 | for line in f: |
| 498 | m = re_vmhwm.search(line) |
| 499 | if m: |
| 500 | groups = m.groups() |
| 501 | s = len(groups) |
| 502 | if s >= 1: |
| 503 | result = int(groups[0]) |
| 504 | if s >= 2: |
| 505 | unit = groups[1] |
| 506 | if unit == 'kB': |
| 507 | result *= 1024 |
| 508 | elif unit != '': |
| 509 | raise Exception('Unrecognized unit in memory usage log: {}' |
| 510 | .format(unit)) |
| 511 | if result is None: |
| 512 | raise Exception('No memory usage found in log: {}'.format(logfile)) |
Tamas Kenez | 02bff03 | 2017-07-18 12:13:58 +0200 | [diff] [blame] | 513 | return result |
| 514 | |
| 515 | # Return a dictionary: {segment_name -> segments_size} |
| 516 | def getDexSegmentSizes(dex_files): |
| 517 | assert len(dex_files) > 0 |
Ian Zerny | 3f54e22 | 2019-02-12 10:51:17 +0100 | [diff] [blame] | 518 | cmd = [jdk.GetJavaExecutable(), '-jar', R8_JAR, 'dexsegments'] |
Tamas Kenez | 02bff03 | 2017-07-18 12:13:58 +0200 | [diff] [blame] | 519 | cmd.extend(dex_files) |
| 520 | PrintCmd(cmd) |
Morten Krogh-Jespersen | f54dd78 | 2021-02-09 09:06:08 +0100 | [diff] [blame] | 521 | output = subprocess.check_output(cmd).decode('utf-8') |
Tamas Kenez | 02bff03 | 2017-07-18 12:13:58 +0200 | [diff] [blame] | 522 | |
| 523 | matches = DEX_SEGMENTS_RESULT_PATTERN.findall(output) |
| 524 | |
| 525 | if matches is None or len(matches) == 0: |
| 526 | raise Exception('DexSegments failed to return any output for' \ |
| 527 | ' these files: {}'.format(dex_files)) |
| 528 | |
| 529 | result = {} |
| 530 | |
| 531 | for match in matches: |
| 532 | result[match[0]] = int(match[1]) |
| 533 | |
| 534 | return result |
| 535 | |
Morten Krogh-Jespersen | 38c7ca0 | 2019-02-04 10:39:57 +0100 | [diff] [blame] | 536 | # Return a dictionary: {segment_name -> segments_size} |
| 537 | def getCfSegmentSizes(cfFile): |
Ian Zerny | 3f54e22 | 2019-02-12 10:51:17 +0100 | [diff] [blame] | 538 | cmd = [jdk.GetJavaExecutable(), |
Morten Krogh-Jespersen | 38c7ca0 | 2019-02-04 10:39:57 +0100 | [diff] [blame] | 539 | '-cp', |
Morten Krogh-Jespersen | 480784d | 2019-02-05 08:10:46 +0100 | [diff] [blame] | 540 | CF_SEGMENTS_TOOL, |
Morten Krogh-Jespersen | 38c7ca0 | 2019-02-04 10:39:57 +0100 | [diff] [blame] | 541 | 'com.android.tools.r8.cf_segments.MeasureLib', |
| 542 | cfFile] |
| 543 | PrintCmd(cmd) |
Rico Wind | fd18637 | 2022-02-28 08:55:48 +0100 | [diff] [blame] | 544 | output = subprocess.check_output(cmd).decode('utf-8') |
Morten Krogh-Jespersen | 38c7ca0 | 2019-02-04 10:39:57 +0100 | [diff] [blame] | 545 | |
| 546 | matches = DEX_SEGMENTS_RESULT_PATTERN.findall(output) |
| 547 | |
| 548 | if matches is None or len(matches) == 0: |
| 549 | raise Exception('CfSegments failed to return any output for' \ |
| 550 | ' the file: ' + cfFile) |
| 551 | |
| 552 | result = {} |
| 553 | |
| 554 | for match in matches: |
| 555 | result[match[0]] = int(match[1]) |
| 556 | |
| 557 | return result |
| 558 | |
Søren Gjesse | 1c115b5 | 2019-08-14 12:43:57 +0200 | [diff] [blame] | 559 | def get_maven_path(artifact, version): |
| 560 | return os.path.join('com', 'android', 'tools', artifact, version) |
Rico Wind | c0b1638 | 2018-05-17 13:23:43 +0200 | [diff] [blame] | 561 | |
Morten Krogh-Jespersen | 38c7ca0 | 2019-02-04 10:39:57 +0100 | [diff] [blame] | 562 | def print_cfsegments(prefix, cf_files): |
| 563 | for cf_file in cf_files: |
| 564 | for segment_name, size in getCfSegmentSizes(cf_file).items(): |
| 565 | print('{}-{}(CodeSize): {}' |
| 566 | .format(prefix, segment_name, size)) |
| 567 | |
Christoffer Quist Adamsen | be6661d | 2023-08-24 11:01:12 +0200 | [diff] [blame] | 568 | def print_dexsegments(prefix, dex_files, worker_id=None): |
Tamas Kenez | 02bff03 | 2017-07-18 12:13:58 +0200 | [diff] [blame] | 569 | for segment_name, size in getDexSegmentSizes(dex_files).items(): |
Christoffer Quist Adamsen | be6661d | 2023-08-24 11:01:12 +0200 | [diff] [blame] | 570 | print_thread( |
| 571 | '{}-{}(CodeSize): {}'.format(prefix, segment_name, size), |
| 572 | worker_id) |
Tamas Kenez | 2cf47cf | 2017-07-25 10:22:52 +0200 | [diff] [blame] | 573 | |
Mads Ager | bc7b2ce | 2018-02-05 11:28:47 +0100 | [diff] [blame] | 574 | # Ensure that we are not benchmarking with a google jvm. |
Tamas Kenez | 2cf47cf | 2017-07-25 10:22:52 +0200 | [diff] [blame] | 575 | def check_java_version(): |
Ian Zerny | 3f54e22 | 2019-02-12 10:51:17 +0100 | [diff] [blame] | 576 | cmd= [jdk.GetJavaExecutable(), '-version'] |
Rico Wind | fd18637 | 2022-02-28 08:55:48 +0100 | [diff] [blame] | 577 | output = subprocess.check_output(cmd, stderr = subprocess.STDOUT).decode('utf-8') |
| 578 | m = re.search('openjdk version "([^"]*)"', output) |
Tamas Kenez | 2cf47cf | 2017-07-25 10:22:52 +0200 | [diff] [blame] | 579 | if m is None: |
| 580 | raise Exception("Can't check java version: no version string in output" |
| 581 | " of 'java -version': '{}'".format(output)) |
| 582 | version = m.groups(0)[0] |
Mads Ager | bc7b2ce | 2018-02-05 11:28:47 +0100 | [diff] [blame] | 583 | m = re.search('google', version) |
| 584 | if m is not None: |
| 585 | raise Exception("Do not use google JVM for benchmarking: " + version) |
Tamas Kenez | 0cad51c | 2017-08-21 14:42:01 +0200 | [diff] [blame] | 586 | |
Christoffer Quist Adamsen | 17879c1 | 2019-01-22 16:13:54 +0100 | [diff] [blame] | 587 | def get_android_jar_dir(api): |
| 588 | return os.path.join(REPO_ROOT, ANDROID_JAR_DIR.format(api=api)) |
| 589 | |
Rico Wind | 9d70f61 | 2018-08-31 09:17:43 +0200 | [diff] [blame] | 590 | def get_android_jar(api): |
| 591 | return os.path.join(REPO_ROOT, ANDROID_JAR.format(api=api)) |
Rico Wind | da6836e | 2018-12-07 12:32:03 +0100 | [diff] [blame] | 592 | |
Rico Wind | faaac01 | 2019-02-25 11:24:05 +0100 | [diff] [blame] | 593 | def is_bot(): |
Rico Wind | e11f439 | 2019-03-18 07:59:37 +0100 | [diff] [blame] | 594 | return 'SWARMING_BOT_ID' in os.environ |
Morten Krogh-Jespersen | 16e925d | 2019-01-25 14:40:38 +0100 | [diff] [blame] | 595 | |
| 596 | def uncompressed_size(path): |
| 597 | return sum(z.file_size for z in zipfile.ZipFile(path).infolist()) |
Morten Krogh-Jespersen | 0de1373 | 2019-03-01 08:56:39 +0100 | [diff] [blame] | 598 | |
Søren Gjesse | 2b04769 | 2022-08-19 16:34:38 +0200 | [diff] [blame] | 599 | def desugar_configuration_name_and_version(configuration, is_for_maven): |
| 600 | name = 'desugar_jdk_libs_configuration' |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 601 | with open(configuration, 'r') as f: |
Søren Gjesse | 6e5e584 | 2019-09-03 08:48:30 +0200 | [diff] [blame] | 602 | configuration_json = json.loads(f.read()) |
| 603 | configuration_format_version = \ |
| 604 | configuration_json.get('configuration_format_version') |
Søren Gjesse | 2b04769 | 2022-08-19 16:34:38 +0200 | [diff] [blame] | 605 | if (not configuration_format_version): |
| 606 | raise Exception( |
| 607 | 'No "configuration_format_version" found in ' + configuration) |
| 608 | if (configuration_format_version != 3 |
| 609 | and configuration_format_version != 5 |
| 610 | and configuration_format_version != (200 if is_for_maven else 100)): |
| 611 | raise Exception( |
| 612 | 'Unsupported "configuration_format_version" "%s" found in %s' |
| 613 | % (configuration_format_version, configuration)) |
Søren Gjesse | 6e5e584 | 2019-09-03 08:48:30 +0200 | [diff] [blame] | 614 | version = configuration_json.get('version') |
| 615 | if not version: |
Søren Gjesse | 2b04769 | 2022-08-19 16:34:38 +0200 | [diff] [blame] | 616 | if configuration_format_version == (200 if is_for_maven else 100): |
| 617 | identifier = configuration_json.get('identifier') |
| 618 | if not identifier: |
| 619 | raise Exception( |
| 620 | 'No "identifier" found in ' + configuration) |
| 621 | identifier_split = identifier.split(':') |
| 622 | if (len(identifier_split) != 3): |
| 623 | raise Exception('Invalid "identifier" found in ' + configuration) |
| 624 | if (identifier_split[0] != 'com.tools.android'): |
| 625 | raise Exception('Invalid "identifier" found in ' + configuration) |
| 626 | if not identifier_split[1].startswith('desugar_jdk_libs_configuration'): |
| 627 | raise Exception('Invalid "identifier" found in ' + configuration) |
| 628 | name = identifier_split[1] |
| 629 | version = identifier_split[2] |
| 630 | else: |
| 631 | raise Exception( |
| 632 | 'No "version" found in ' + configuration) |
| 633 | else: |
| 634 | if configuration_format_version == (200 if is_for_maven else 100): |
| 635 | raise Exception( |
| 636 | 'No "version" expected in ' + configuration) |
| 637 | # Disallow prerelease, as older R8 versions cannot parse it causing hard to |
| 638 | # understand errors. |
| 639 | check_basic_semver_version(version, 'in ' + configuration, allowPrerelease = False) |
| 640 | return (name, version) |
Søren Gjesse | 1e17153 | 2019-09-03 09:44:22 +0200 | [diff] [blame] | 641 | |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 642 | class SemanticVersion: |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 643 | def __init__(self, major, minor, patch, prerelease): |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 644 | self.major = major |
| 645 | self.minor = minor |
| 646 | self.patch = patch |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 647 | self.prerelease = prerelease |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 648 | # Build metadata currently not suppported |
| 649 | |
| 650 | def larger_than(self, other): |
Søren Gjesse | 29d108a | 2022-04-07 10:49:49 +0200 | [diff] [blame] | 651 | if self.prerelease or other.prerelease: |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 652 | raise Exception("Comparison with prerelease not implemented") |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 653 | if self.major > other.major: |
| 654 | return True |
| 655 | if self.major == other.major and self.minor > other.minor: |
| 656 | return True |
| 657 | if self.patch: |
| 658 | return (self.major == other.major |
| 659 | and self.minor == other.minor |
| 660 | and self.patch > other.patch) |
| 661 | else: |
| 662 | return False |
| 663 | |
| 664 | |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 665 | # Check that the passed string is formatted as a basic semver version (x.y.z or x.y.z-prerelease |
| 666 | # depending on the value of allowPrerelease). |
| 667 | # See https://semver.org/. The regexp parts used are not all complient with what is suggested |
| 668 | # on https://semver.org/#is-there-a-suggested-regular-expression-regex-to-check-a-semver-string. |
| 669 | def check_basic_semver_version(version, error_context = '', components = 3, allowPrerelease = False): |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 670 | regexp = '^' |
| 671 | for x in range(components): |
| 672 | regexp += '([0-9]+)' |
| 673 | if x < components - 1: |
| 674 | regexp += '\\.' |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 675 | if allowPrerelease: |
| 676 | # This part is from |
| 677 | # https://semver.org/#is-there-a-suggested-regular-expression-regex-to-check-a-semver-string |
| 678 | regexp += r'(?:-(?P<prerelease>(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*)(?:\.(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*))*))?' |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 679 | regexp += '$' |
| 680 | reg = re.compile(regexp) |
| 681 | match = reg.match(version) |
| 682 | if not match: |
Søren Gjesse | 1e17153 | 2019-09-03 09:44:22 +0200 | [diff] [blame] | 683 | raise Exception("Invalid version '" |
| 684 | + version |
| 685 | + "'" |
| 686 | + (' ' + error_context) if len(error_context) > 0 else '') |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 687 | if components == 2: |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 688 | return SemanticVersion(int(match.group(1)), int(match.group(2)), None, None) |
| 689 | elif components == 3 and not allowPrerelease: |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 690 | return SemanticVersion( |
Søren Gjesse | 705a3b1 | 2022-03-17 11:37:30 +0100 | [diff] [blame] | 691 | int(match.group(1)), int(match.group(2)), int(match.group(3)), None) |
| 692 | elif components == 3 and allowPrerelease: |
| 693 | return SemanticVersion( |
| 694 | int(match.group(1)), int(match.group(2)), int(match.group(3)), match.group('prerelease')) |
Søren Gjesse | 4e5c6fe | 2019-11-05 17:17:43 +0100 | [diff] [blame] | 695 | else: |
| 696 | raise Exception('Argument "components" must be 2 or 3') |