blob: 2a4b18eb2f732e1582e8b53218c238f111e328fe [file] [log] [blame]
Mads Ager418d1ca2017-05-22 09:35:49 +02001# 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
7import hashlib
Søren Gjesse6e5e5842019-09-03 08:48:30 +02008import json
Mads Ager418d1ca2017-05-22 09:35:49 +02009import os
Tamas Kenez82efeb52017-06-12 13:56:22 +020010import re
Mads Ager418d1ca2017-05-22 09:35:49 +020011import shutil
12import subprocess
13import sys
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +020014import tarfile
Mads Ager418d1ca2017-05-22 09:35:49 +020015import tempfile
Morten Krogh-Jespersen16e925d2019-01-25 14:40:38 +010016import zipfile
Mads Ager418d1ca2017-05-22 09:35:49 +020017
Ian Zerny37097652019-04-11 13:13:27 +020018import defines
19import jdk
20
Christoffer Quist Adamsen17879c12019-01-22 16:13:54 +010021ANDROID_JAR_DIR = 'third_party/android_jar/lib-v{api}'
22ANDROID_JAR = os.path.join(ANDROID_JAR_DIR, 'android.jar')
Ian Zerny3f54e222019-02-12 10:51:17 +010023TOOLS_DIR = defines.TOOLS_DIR
24REPO_ROOT = defines.REPO_ROOT
25THIRD_PARTY = defines.THIRD_PARTY
Morten Krogh-Jespersen220e5702019-02-27 12:57:01 +010026ANDROID_SDK = os.path.join(THIRD_PARTY, 'android_sdk')
Tamas Kenezfc34cd82017-07-13 12:43:57 +020027MEMORY_USE_TMP_FILE = 'memory_use.tmp'
Tamas Kenez02bff032017-07-18 12:13:58 +020028DEX_SEGMENTS_RESULT_PATTERN = re.compile('- ([^:]+): ([0-9]+)')
Mads Ager12a56bc2017-11-27 11:51:25 +010029BUILD = os.path.join(REPO_ROOT, 'build')
Ian Zerny0f5fc732018-11-15 14:34:41 +010030BUILD_DEPS_DIR = os.path.join(BUILD, 'deps')
31BUILD_MAIN_DIR = os.path.join(BUILD, 'classes', 'main')
Morten Krogh-Jespersen51db2b02020-11-11 12:49:26 +010032BUILD_JAVA_MAIN_DIR = os.path.join(BUILD, 'classes', 'java', 'main')
Ian Zerny0f5fc732018-11-15 14:34:41 +010033BUILD_TEST_DIR = os.path.join(BUILD, 'classes', 'test')
Mads Ager12a56bc2017-11-27 11:51:25 +010034LIBS = os.path.join(BUILD, 'libs')
35GENERATED_LICENSE_DIR = os.path.join(BUILD, 'generatedLicense')
Mads Agera4911eb2017-11-22 13:19:36 +010036SRC_ROOT = os.path.join(REPO_ROOT, 'src', 'main', 'java')
Søren Gjesse6e5e5842019-09-03 08:48:30 +020037TEST_ROOT = os.path.join(REPO_ROOT, 'src', 'test', 'java')
Ian Zerny59dfa4c2019-10-25 10:34:36 +020038REPO_SOURCE = 'https://r8.googlesource.com/r8'
Søren Gjessedc9d8a22017-10-12 12:40:59 +020039
40D8 = 'd8'
41R8 = 'r8'
Tamas Kenez03ab76f2018-12-07 14:33:25 +010042R8LIB = 'r8lib'
Morten Krogh-Jespersene28db462019-01-09 13:32:15 +010043R8LIB_NO_DEPS = 'r8LibNoDeps'
Morten Krogh-Jespersen98ee89a2021-10-25 20:59:02 +020044R8RETRACE = 'R8Retrace'
45R8RETRACE_NO_DEPS = 'R8RetraceNoDeps'
Mads Agerb10c07f2017-11-27 13:25:52 +010046R8_SRC = 'sourceJar'
Søren Gjesse17fc67d2019-12-04 14:50:17 +010047LIBRARY_DESUGAR_CONVERSIONS = 'buildLibraryDesugarConversions'
Ian Zerny161ff742022-01-20 12:39:40 +010048R8_TESTS_TARGET = 'TestJar'
49R8_TESTS_DEPS_TARGET = 'RepackageTestDeps'
50R8LIB_TESTS_TARGET = 'configureTestForR8Lib'
51R8LIB_TESTS_DEPS_TARGET = R8_TESTS_DEPS_TARGET
Søren Gjessedc9d8a22017-10-12 12:40:59 +020052
Morten Krogh-Jespersen51db2b02020-11-11 12:49:26 +010053ALL_DEPS_JAR = os.path.join(LIBS, 'deps_all.jar')
Rico Wind74fab302017-10-02 07:25:33 +020054D8_JAR = os.path.join(LIBS, 'd8.jar')
55R8_JAR = os.path.join(LIBS, 'r8.jar')
Ian Zerny161ff742022-01-20 12:39:40 +010056R8_WITH_RELOCATED_DEPS_JAR = os.path.join(LIBS, 'r8_with_relocated_deps.jar')
Tamas Kenez03ab76f2018-12-07 14:33:25 +010057R8LIB_JAR = os.path.join(LIBS, 'r8lib.jar')
Ian Zerny5ffa58f2020-02-26 08:37:14 +010058R8LIB_MAP = os.path.join(LIBS, 'r8lib.jar.map')
Mads Agerb10c07f2017-11-27 13:25:52 +010059R8_SRC_JAR = os.path.join(LIBS, 'r8-src.jar')
Tamas Kenez03ab76f2018-12-07 14:33:25 +010060R8LIB_EXCLUDE_DEPS_JAR = os.path.join(LIBS, 'r8lib-exclude-deps.jar')
Tamas Kenez180be092018-12-05 15:23:06 +010061R8_FULL_EXCLUDE_DEPS_JAR = os.path.join(LIBS, 'r8-full-exclude-deps.jar')
Morten Krogh-Jespersen98ee89a2021-10-25 20:59:02 +020062R8RETRACE_JAR = os.path.join(LIBS, 'r8retrace.jar')
63R8RETRACE_EXCLUDE_DEPS_JAR = os.path.join(LIBS, 'r8retrace-exclude-deps.jar')
Ian Zerny161ff742022-01-20 12:39:40 +010064R8_TESTS_JAR = os.path.join(LIBS, 'r8tests.jar')
65R8LIB_TESTS_JAR = os.path.join(LIBS, 'r8libtestdeps-cf.jar')
66R8_TESTS_DEPS_JAR = os.path.join(LIBS, 'test_deps_all.jar')
67R8LIB_TESTS_DEPS_JAR = R8_TESTS_DEPS_JAR
Mads Ager12a56bc2017-11-27 11:51:25 +010068MAVEN_ZIP = os.path.join(LIBS, 'r8.zip')
Rico Wind8fc8bfa2019-03-22 09:57:36 +010069MAVEN_ZIP_LIB = os.path.join(LIBS, 'r8lib.zip')
Søren Gjesse17fc67d2019-12-04 14:50:17 +010070LIBRARY_DESUGAR_CONVERSIONS_ZIP = os.path.join(LIBS, 'library_desugar_conversions.zip')
71
Søren Gjesse6e5e5842019-09-03 08:48:30 +020072DESUGAR_CONFIGURATION = os.path.join(
Søren Gjesse927a92e2019-12-04 15:18:06 +010073 'src', 'library_desugar', 'desugar_jdk_libs.json')
Søren Gjesseee086b22020-10-30 11:46:39 +010074DESUGAR_IMPLEMENTATION = os.path.join(
Søren Gjesse3dc207b2021-02-15 09:45:30 +010075 'third_party', 'openjdk', 'desugar_jdk_libs', 'desugar_jdk_libs.jar')
Søren Gjesse705a3b12022-03-17 11:37:30 +010076DESUGAR_CONFIGURATION_JDK11_LEGACY = os.path.join(
77 'src', 'library_desugar', 'jdk11', 'desugar_jdk_libs_legacy.json')
78DESUGAR_IMPLEMENTATION_JDK11 = os.path.join(
79 'third_party', 'openjdk', 'desugar_jdk_libs_11', 'desugar_jdk_libs.jar')
Søren Gjesse6e5e5842019-09-03 08:48:30 +020080DESUGAR_CONFIGURATION_MAVEN_ZIP = os.path.join(
81 LIBS, 'desugar_jdk_libs_configuration.zip')
Søren Gjesse705a3b12022-03-17 11:37:30 +010082DESUGAR_CONFIGURATION_LEGACY_JDK11_MAVEN_ZIP = os.path.join(
83 LIBS, 'desugar_jdk_libs_configuration_legacy_jdk11.zip')
Mads Ager12a56bc2017-11-27 11:51:25 +010084GENERATED_LICENSE = os.path.join(GENERATED_LICENSE_DIR, 'LICENSE')
Mathias Rav3fb4a3a2018-05-29 15:41:36 +020085RT_JAR = os.path.join(REPO_ROOT, 'third_party/openjdk/openjdk-rt-1.8/rt.jar')
Mathias Ravb46dc002018-06-06 09:37:11 +020086R8LIB_KEEP_RULES = os.path.join(REPO_ROOT, 'src/main/keep.txt')
Morten Krogh-Jespersen480784d2019-02-05 08:10:46 +010087CF_SEGMENTS_TOOL = os.path.join(THIRD_PARTY, 'cf_segments')
Morten Krogh-Jespersen38c7ca02019-02-04 10:39:57 +010088PINNED_R8_JAR = os.path.join(REPO_ROOT, 'third_party/r8/r8.jar')
89PINNED_PGR8_JAR = os.path.join(REPO_ROOT, 'third_party/r8/r8-pg6.0.1.jar')
Ian Zernyfbb1f7a2019-05-02 14:34:13 +020090SAMPLE_LIBRARIES_SHA_FILE = os.path.join(
91 THIRD_PARTY, 'sample_libraries.tar.gz.sha1')
Morten Krogh-Jespersen45d7a7b2020-11-02 08:31:09 +010092OPENSOURCE_DUMPS_DIR = os.path.join(THIRD_PARTY, 'opensource-apps')
Morten Krogh-Jespersen86222742021-03-02 11:13:33 +010093INTERNAL_DUMPS_DIR = os.path.join(THIRD_PARTY, 'internal-apps')
Søren Gjesse1c115b52019-08-14 12:43:57 +020094BAZEL_SHA_FILE = os.path.join(THIRD_PARTY, 'bazel.tar.gz.sha1')
95BAZEL_TOOL = os.path.join(THIRD_PARTY, 'bazel')
Søren Gjesse699f6362019-10-09 14:56:33 +020096JAVA8_SHA_FILE = os.path.join(THIRD_PARTY, 'openjdk', 'jdk8', 'linux-x86.tar.gz.sha1')
Søren Gjesseef195772021-03-11 16:04:42 +010097JAVA11_SHA_FILE = os.path.join(THIRD_PARTY, 'openjdk', 'jdk-11', 'linux.tar.gz.sha1')
Christoffer Quist Adamsen1ca046c2021-02-21 11:25:16 +010098IGNORE_WARNINGS_RULES = os.path.join(REPO_ROOT, 'src', 'test', 'ignorewarnings.rules')
Morten Krogh-Jespersen38c7ca02019-02-04 10:39:57 +010099
Morten Krogh-Jespersen220e5702019-02-27 12:57:01 +0100100ANDROID_HOME_ENVIROMENT_NAME = "ANDROID_HOME"
101ANDROID_TOOLS_VERSION_ENVIRONMENT_NAME = "ANDROID_TOOLS_VERSION"
Morten Krogh-Jespersenc8efedd2019-01-28 11:36:17 +0100102USER_HOME = os.path.expanduser('~')
Morten Krogh-Jespersen220e5702019-02-27 12:57:01 +0100103
Morten Krogh-Jespersen0981b722019-10-09 10:00:33 +0200104R8_TEST_RESULTS_BUCKET = 'r8-test-results'
105
106def archive_file(name, gs_dir, src_file):
107 gs_file = '%s/%s' % (gs_dir, name)
108 upload_file_to_cloud_storage(src_file, gs_file, public_read=False)
109
110def archive_value(name, gs_dir, value):
111 with TempDir() as temp:
112 tempfile = os.path.join(temp, name);
113 with open(tempfile, 'w') as f:
114 f.write(str(value))
115 archive_file(name, gs_dir, tempfile)
116
Christoffer Quist Adamsen4d38d032021-04-20 12:31:31 +0200117def find_cloud_storage_file_from_options(name, options, orElse=None):
118 # Import archive on-demand since archive depends on utils.
119 from archive import GetUploadDestination
120 hash_or_version = find_hash_or_version_from_options(options)
121 if not hash_or_version:
122 return orElse
123 is_hash = options.commit_hash is not None
124 download_path = GetUploadDestination(hash_or_version, name, is_hash)
125 if file_exists_on_cloud_storage(download_path):
126 out = tempfile.NamedTemporaryFile().name
127 download_file_from_cloud_storage(download_path, out)
128 return out
129 else:
130 raise Exception('Could not find file {} from hash/version: {}.'
131 .format(name, hash_or_version))
132
133def find_r8_jar_from_options(options):
134 return find_cloud_storage_file_from_options('r8.jar', options)
135
136def find_r8_lib_jar_from_options(options):
137 return find_cloud_storage_file_from_options('r8lib.jar', options)
138
139def find_hash_or_version_from_options(options):
140 if options.tag:
141 return find_hash_or_version_from_tag(options.tag)
142 else:
143 return options.commit_hash or options.version
144
145def find_hash_or_version_from_tag(tag_or_hash):
Rico Windfd186372022-02-28 08:55:48 +0100146 info = subprocess.check_output([
Christoffer Quist Adamsen4d38d032021-04-20 12:31:31 +0200147 'git',
148 'show',
149 tag_or_hash,
150 '-s',
Rico Windfd186372022-02-28 08:55:48 +0100151 '--format=oneline']).decode('utf-8').splitlines()[-1].split()
Christoffer Quist Adamsen4d38d032021-04-20 12:31:31 +0200152 # The info should be on the following form [hash,"Version",version]
153 if len(info) == 3 and len(info[0]) == 40 and info[1] == "Version":
154 return info[2]
155 return None
156
Morten Krogh-Jespersen220e5702019-02-27 12:57:01 +0100157def getAndroidHome():
158 return os.environ.get(
159 ANDROID_HOME_ENVIROMENT_NAME, os.path.join(USER_HOME, 'Android', 'Sdk'))
160
161def getAndroidBuildTools():
162 version = os.environ.get(ANDROID_TOOLS_VERSION_ENVIRONMENT_NAME, '28.0.3')
163 return os.path.join(getAndroidHome(), 'build-tools', version)
Morten Krogh-Jespersenc8efedd2019-01-28 11:36:17 +0100164
Christoffer Quist Adamsen5c9ded12021-01-14 14:29:37 +0100165def is_python3():
166 return sys.version_info.major == 3
167
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100168def Print(s, quiet=False):
169 if quiet:
170 return
171 print(s)
172
173def Warn(message):
174 CRED = '\033[91m'
175 CEND = '\033[0m'
176 print(CRED + message + CEND)
177
178def PrintCmd(cmd, env=None, quiet=False):
179 if quiet:
180 return
181 if type(cmd) is list:
182 cmd = ' '.join(cmd)
183 if env:
184 env = ' '.join(['{}=\"{}\"'.format(x, y) for x, y in env.iteritems()])
185 print('Running: {} {}'.format(env, cmd))
186 else:
187 print('Running: {}'.format(cmd))
Mads Ager418d1ca2017-05-22 09:35:49 +0200188 # I know this will hit os on windows eventually if we don't do this.
189 sys.stdout.flush()
190
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100191class ProgressLogger(object):
192 CLEAR_LINE = '\033[K'
193 UP = '\033[F'
194
195 def __init__(self, quiet=False):
196 self._count = 0
197 self._has_printed = False
198 self._quiet = quiet
199
200 def log(self, text):
201 if self._quiet:
202 if self._has_printed:
203 sys.stdout.write(ProgressLogger.UP + ProgressLogger.CLEAR_LINE)
204 if len(text) > 140:
205 text = text[0:140] + '...'
206 print(text)
207 self._has_printed = True
208
209 def done(self):
210 if self._quiet and self._has_printed:
211 sys.stdout.write(ProgressLogger.UP + ProgressLogger.CLEAR_LINE)
212 print('')
213 sys.stdout.write(ProgressLogger.UP)
214
Morten Krogh-Jespersen7cdd3a72019-03-13 14:58:25 +0100215def RunCmd(cmd, env_vars=None, quiet=False, fail=True, logging=True):
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100216 PrintCmd(cmd, env=env_vars, quiet=quiet)
217 env = os.environ.copy()
218 if env_vars:
219 env.update(env_vars)
220 process = subprocess.Popen(
221 cmd, env=env, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
222 stdout = []
Morten Krogh-Jespersen7cdd3a72019-03-13 14:58:25 +0100223 logger = ProgressLogger(quiet=quiet) if logging else None
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100224 failed = False
225 while True:
Rico Wind744ba752021-01-22 06:24:49 +0100226 line = process.stdout.readline().decode('utf-8')
227 if line != '':
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100228 stripped = line.rstrip()
229 stdout.append(stripped)
Morten Krogh-Jespersen7cdd3a72019-03-13 14:58:25 +0100230 if logger:
231 logger.log(stripped)
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100232 # TODO(christofferqa): r8 should fail with non-zero exit code.
Morten Krogh-Jespersen121c47b2019-01-25 09:57:21 +0100233 if ('AssertionError:' in stripped
234 or 'CompilationError:' in stripped
235 or 'CompilationFailedException:' in stripped
Morten Krogh-Jespersen5d02a6b2019-10-29 14:48:56 +0100236 or 'Compilation failed' in stripped
237 or 'FAILURE:' in stripped
238 or 'org.gradle.api.ProjectConfigurationException' in stripped
239 or 'BUILD FAILED' in stripped):
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100240 failed = True
241 else:
Morten Krogh-Jespersen7cdd3a72019-03-13 14:58:25 +0100242 if logger:
243 logger.done()
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100244 exit_code = process.poll()
245 if exit_code or failed:
246 for line in stdout:
247 Warn(line)
Christoffer Quist Adamsen7cf4c562019-03-07 10:57:33 +0100248 if fail:
249 raise subprocess.CalledProcessError(
250 exit_code or -1, cmd, output='\n'.join(stdout))
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100251 return stdout
252
Christoffer Quist Adamsen7cf4c562019-03-07 10:57:33 +0100253def RunGradlew(
254 args, clean=True, stacktrace=True, use_daemon=False, env_vars=None,
Morten Krogh-Jespersen7cdd3a72019-03-13 14:58:25 +0100255 quiet=False, fail=True, logging=True):
Christoffer Quist Adamsen7cf4c562019-03-07 10:57:33 +0100256 cmd = ['./gradlew']
257 if clean:
258 assert 'clean' not in args
259 cmd.append('clean')
260 if stacktrace:
261 assert '--stacktrace' not in args
262 cmd.append('--stacktrace')
263 if not use_daemon:
264 assert '--no-daemon' not in args
265 cmd.append('--no-daemon')
266 cmd.extend(args)
Morten Krogh-Jespersen7cdd3a72019-03-13 14:58:25 +0100267 return RunCmd(cmd, env_vars=env_vars, quiet=quiet, fail=fail, logging=logging)
Christoffer Quist Adamsen7cf4c562019-03-07 10:57:33 +0100268
Rico Windf80f5a22017-06-16 09:15:57 +0200269def IsWindows():
Ian Zerny3f54e222019-02-12 10:51:17 +0100270 return defines.IsWindows()
Ian Zerny5fffb0a2019-02-11 13:54:22 +0100271
272def IsLinux():
Ian Zerny3f54e222019-02-12 10:51:17 +0100273 return defines.IsLinux()
Ian Zerny5fffb0a2019-02-11 13:54:22 +0100274
275def IsOsX():
Ian Zerny3f54e222019-02-12 10:51:17 +0100276 return defines.IsOsX()
Ian Zerny5fffb0a2019-02-11 13:54:22 +0100277
278def EnsureDepFromGoogleCloudStorage(dep, tgz, sha1, msg):
279 if not os.path.exists(dep) or os.path.getmtime(tgz) < os.path.getmtime(sha1):
280 DownloadFromGoogleCloudStorage(sha1)
281 # Update the mtime of the tar file to make sure we do not run again unless
282 # there is an update.
283 os.utime(tgz, None)
284 else:
Rico Wind3d369b42021-01-12 10:26:24 +0100285 print('Ensure cloud dependency:', msg, 'present')
Rico Windf80f5a22017-06-16 09:15:57 +0200286
Jean-Marie Henaffe4e36d12018-04-05 10:33:50 +0200287def DownloadFromX20(sha1_file):
288 download_script = os.path.join(REPO_ROOT, 'tools', 'download_from_x20.py')
289 cmd = [download_script, sha1_file]
290 PrintCmd(cmd)
291 subprocess.check_call(cmd)
292
Rico Wind59593922021-03-03 09:12:36 +0100293def DownloadFromGoogleCloudStorage(sha1_file, bucket='r8-deps', auth=False,
294 quiet=False):
Rico Windf80f5a22017-06-16 09:15:57 +0200295 suffix = '.bat' if IsWindows() else ''
296 download_script = 'download_from_google_storage%s' % suffix
Rico Wind533e3ce2019-04-04 10:26:12 +0200297 cmd = [download_script]
298 if not auth:
299 cmd.append('-n')
300 cmd.extend(['-b', bucket, '-u', '-s', sha1_file])
Rico Wind59593922021-03-03 09:12:36 +0100301 if not quiet:
302 PrintCmd(cmd)
303 subprocess.check_call(cmd)
304 else:
305 subprocess.check_output(cmd)
Mads Ager418d1ca2017-05-22 09:35:49 +0200306
307def get_sha1(filename):
308 sha1 = hashlib.sha1()
309 with open(filename, 'rb') as f:
310 while True:
311 chunk = f.read(1024*1024)
312 if not chunk:
313 break
314 sha1.update(chunk)
315 return sha1.hexdigest()
316
Rico Wind1b52acf2021-03-21 12:36:55 +0100317def is_main():
Rico Windfd186372022-02-28 08:55:48 +0100318 remotes = subprocess.check_output(['git', 'branch', '-r', '--contains',
319 'HEAD']).decode('utf-8')
Rico Wind1b52acf2021-03-21 12:36:55 +0100320 return 'origin/main' in remotes
Rico Wind1b09c562019-01-17 08:53:09 +0100321
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200322def get_HEAD_sha1():
Morten Krogh-Jespersene0ce6a32019-02-07 11:44:45 +0100323 return get_HEAD_sha1_for_checkout(REPO_ROOT)
324
325def get_HEAD_sha1_for_checkout(checkout):
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200326 cmd = ['git', 'rev-parse', 'HEAD']
327 PrintCmd(cmd)
Morten Krogh-Jespersene0ce6a32019-02-07 11:44:45 +0100328 with ChangedWorkingDirectory(checkout):
Christoffer Quist Adamsen50930e42021-01-20 11:55:12 +0100329 return subprocess.check_output(cmd).decode('utf-8').strip()
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200330
Tamas Kenez971eec62017-05-24 11:08:40 +0200331def makedirs_if_needed(path):
332 try:
333 os.makedirs(path)
334 except OSError:
335 if not os.path.isdir(path):
336 raise
337
Rico Windef7420c2021-10-14 13:16:01 +0200338def get_gsutil():
339 return 'gsutil.py' if os.name != 'nt' else 'gsutil.py.bat'
340
Rico Wind800fd712018-09-24 11:29:33 +0200341def upload_dir_to_cloud_storage(directory, destination, is_html=False, public_read=True):
Rico Winda94f01c2017-06-27 10:32:34 +0200342 # Upload and make the content encoding right for viewing directly
Rico Windef7420c2021-10-14 13:16:01 +0200343 cmd = [get_gsutil(), '-m', 'cp']
Rico Windd0d88cf2018-02-09 09:46:11 +0100344 if is_html:
345 cmd += ['-z', 'html']
Rico Wind800fd712018-09-24 11:29:33 +0200346 if public_read:
347 cmd += ['-a', 'public-read']
348 cmd += ['-R', directory, destination]
Rico Winda94f01c2017-06-27 10:32:34 +0200349 PrintCmd(cmd)
350 subprocess.check_call(cmd)
351
Rico Wind800fd712018-09-24 11:29:33 +0200352def upload_file_to_cloud_storage(source, destination, public_read=True):
Rico Windef7420c2021-10-14 13:16:01 +0200353 cmd = [get_gsutil(), 'cp']
Rico Wind800fd712018-09-24 11:29:33 +0200354 if public_read:
355 cmd += ['-a', 'public-read']
356 cmd += [source, destination]
Rico Windb4621c12017-08-28 12:48:53 +0200357 PrintCmd(cmd)
358 subprocess.check_call(cmd)
359
Rico Wind139eece2018-09-25 09:42:09 +0200360def delete_file_from_cloud_storage(destination):
Rico Windef7420c2021-10-14 13:16:01 +0200361 cmd = [get_gsutil(), 'rm', destination]
Rico Wind139eece2018-09-25 09:42:09 +0200362 PrintCmd(cmd)
363 subprocess.check_call(cmd)
364
Rico Wind4fd2dda2018-09-26 17:41:45 +0200365def ls_files_on_cloud_storage(destination):
Rico Windef7420c2021-10-14 13:16:01 +0200366 cmd = [get_gsutil(), 'ls', destination]
Rico Wind4fd2dda2018-09-26 17:41:45 +0200367 PrintCmd(cmd)
Rico Windfd186372022-02-28 08:55:48 +0100368 return subprocess.check_output(cmd).decode('utf-8')
Rico Wind4fd2dda2018-09-26 17:41:45 +0200369
Rico Wind139eece2018-09-25 09:42:09 +0200370def cat_file_on_cloud_storage(destination, ignore_errors=False):
Rico Windef7420c2021-10-14 13:16:01 +0200371 cmd = [get_gsutil(), 'cat', destination]
Rico Wind139eece2018-09-25 09:42:09 +0200372 PrintCmd(cmd)
373 try:
Rico Wind8dadb312022-02-28 08:40:20 +0100374 return subprocess.check_output(cmd).decode('utf-8').strip()
Rico Wind139eece2018-09-25 09:42:09 +0200375 except subprocess.CalledProcessError as e:
376 if ignore_errors:
377 return ''
378 else:
379 raise e
380
381def file_exists_on_cloud_storage(destination):
Rico Windef7420c2021-10-14 13:16:01 +0200382 cmd = [get_gsutil(), 'ls', destination]
Rico Wind139eece2018-09-25 09:42:09 +0200383 PrintCmd(cmd)
384 return subprocess.call(cmd) == 0
385
Christoffer Quist Adamsen870fa462020-12-15 10:50:54 +0100386def download_file_from_cloud_storage(source, destination, quiet=False):
Rico Windef7420c2021-10-14 13:16:01 +0200387 cmd = [get_gsutil(), 'cp', source, destination]
Christoffer Quist Adamsen870fa462020-12-15 10:50:54 +0100388 PrintCmd(cmd, quiet=quiet)
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200389 subprocess.check_call(cmd)
390
Morten Krogh-Jespersen220e5702019-02-27 12:57:01 +0100391def create_archive(name, sources=None):
392 if not sources:
393 sources = [name]
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200394 tarname = '%s.tar.gz' % name
395 with tarfile.open(tarname, 'w:gz') as tar:
Morten Krogh-Jespersen54090862019-02-19 11:31:10 +0100396 for source in sources:
397 tar.add(source)
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200398 return tarname
399
400def extract_dir(filename):
401 return filename[0:len(filename) - len('.tar.gz')]
402
403def unpack_archive(filename):
404 dest_dir = extract_dir(filename)
405 if os.path.exists(dest_dir):
Rico Wind3d369b42021-01-12 10:26:24 +0100406 print('Deleting existing dir %s' % dest_dir)
Jean-Marie Henaff7a64eec2018-05-31 15:30:35 +0200407 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-Jespersenec3047b2020-08-18 13:09:06 +0200412def check_gcert():
Ian Zerny86c4cfd2022-01-17 13:43:37 +0100413 status = subprocess.call(['gcertstatus'])
414 if status != 0:
415 subprocess.check_call(['gcert'])
Morten Krogh-Jespersen54090862019-02-19 11:31:10 +0100416
Rico Wind1a29c4f2018-01-25 08:43:08 +0100417# 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.
420def cloud_storage_exists(destination):
Rico Windef7420c2021-10-14 13:16:01 +0200421 cmd = [get_gsutil(), 'ls', destination]
Rico Wind1a29c4f2018-01-25 08:43:08 +0100422 PrintCmd(cmd)
423 exit_code = subprocess.call(cmd)
424 return exit_code == 0
425
Mads Ager418d1ca2017-05-22 09:35:49 +0200426class TempDir(object):
Søren Gjessecbeae782019-05-21 14:14:25 +0200427 def __init__(self, prefix='', delete=True):
Mads Ager418d1ca2017-05-22 09:35:49 +0200428 self._temp_dir = None
429 self._prefix = prefix
Søren Gjessecbeae782019-05-21 14:14:25 +0200430 self._delete = delete
Mads Ager418d1ca2017-05-22 09:35:49 +0200431
432 def __enter__(self):
433 self._temp_dir = tempfile.mkdtemp(self._prefix)
434 return self._temp_dir
435
436 def __exit__(self, *_):
Søren Gjessecbeae782019-05-21 14:14:25 +0200437 if self._delete:
438 shutil.rmtree(self._temp_dir, ignore_errors=True)
Mads Ager418d1ca2017-05-22 09:35:49 +0200439
440class ChangedWorkingDirectory(object):
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100441 def __init__(self, working_directory, quiet=False):
442 self._quiet = quiet
Mads Ager418d1ca2017-05-22 09:35:49 +0200443 self._working_directory = working_directory
444
445 def __enter__(self):
446 self._old_cwd = os.getcwd()
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100447 if not self._quiet:
Rico Wind3d369b42021-01-12 10:26:24 +0100448 print('Enter directory:', self._working_directory)
Mads Ager418d1ca2017-05-22 09:35:49 +0200449 os.chdir(self._working_directory)
450
451 def __exit__(self, *_):
Christoffer Quist Adamsen1de3dde2019-01-24 13:17:46 +0100452 if not self._quiet:
Rico Wind3d369b42021-01-12 10:26:24 +0100453 print('Enter directory:', self._old_cwd)
Mads Ager418d1ca2017-05-22 09:35:49 +0200454 os.chdir(self._old_cwd)
Tamas Kenez82efeb52017-06-12 13:56:22 +0200455
456# Reading Android CTS test_result.xml
457
458class CtsModule(object):
459 def __init__(self, module_name):
460 self.name = module_name
461
462class CtsTestCase(object):
463 def __init__(self, test_case_name):
464 self.name = test_case_name
465
466class 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.
473def 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 Windf80f5a22017-06-16 09:15:57 +0200490 assert outcome in ['fail', 'pass']
Tamas Kenez82efeb52017-06-12 13:56:22 +0200491 yield CtsTest(m.groups()[1], outcome == 'pass')
Tamas Kenezfc34cd82017-07-13 12:43:57 +0200492
493def 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 Kenez02bff032017-07-18 12:13:58 +0200513 return result
514
515# Return a dictionary: {segment_name -> segments_size}
516def getDexSegmentSizes(dex_files):
517 assert len(dex_files) > 0
Ian Zerny3f54e222019-02-12 10:51:17 +0100518 cmd = [jdk.GetJavaExecutable(), '-jar', R8_JAR, 'dexsegments']
Tamas Kenez02bff032017-07-18 12:13:58 +0200519 cmd.extend(dex_files)
520 PrintCmd(cmd)
Morten Krogh-Jespersenf54dd782021-02-09 09:06:08 +0100521 output = subprocess.check_output(cmd).decode('utf-8')
Tamas Kenez02bff032017-07-18 12:13:58 +0200522
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-Jespersen38c7ca02019-02-04 10:39:57 +0100536# Return a dictionary: {segment_name -> segments_size}
537def getCfSegmentSizes(cfFile):
Ian Zerny3f54e222019-02-12 10:51:17 +0100538 cmd = [jdk.GetJavaExecutable(),
Morten Krogh-Jespersen38c7ca02019-02-04 10:39:57 +0100539 '-cp',
Morten Krogh-Jespersen480784d2019-02-05 08:10:46 +0100540 CF_SEGMENTS_TOOL,
Morten Krogh-Jespersen38c7ca02019-02-04 10:39:57 +0100541 'com.android.tools.r8.cf_segments.MeasureLib',
542 cfFile]
543 PrintCmd(cmd)
Rico Windfd186372022-02-28 08:55:48 +0100544 output = subprocess.check_output(cmd).decode('utf-8')
Morten Krogh-Jespersen38c7ca02019-02-04 10:39:57 +0100545
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 Gjesse1c115b52019-08-14 12:43:57 +0200559def get_maven_path(artifact, version):
560 return os.path.join('com', 'android', 'tools', artifact, version)
Rico Windc0b16382018-05-17 13:23:43 +0200561
Morten Krogh-Jespersen38c7ca02019-02-04 10:39:57 +0100562def 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
Tamas Kenez02bff032017-07-18 12:13:58 +0200568def print_dexsegments(prefix, dex_files):
569 for segment_name, size in getDexSegmentSizes(dex_files).items():
570 print('{}-{}(CodeSize): {}'
571 .format(prefix, segment_name, size))
Tamas Kenez2cf47cf2017-07-25 10:22:52 +0200572
Mads Agerbc7b2ce2018-02-05 11:28:47 +0100573# Ensure that we are not benchmarking with a google jvm.
Tamas Kenez2cf47cf2017-07-25 10:22:52 +0200574def check_java_version():
Ian Zerny3f54e222019-02-12 10:51:17 +0100575 cmd= [jdk.GetJavaExecutable(), '-version']
Rico Windfd186372022-02-28 08:55:48 +0100576 output = subprocess.check_output(cmd, stderr = subprocess.STDOUT).decode('utf-8')
577 m = re.search('openjdk version "([^"]*)"', output)
Tamas Kenez2cf47cf2017-07-25 10:22:52 +0200578 if m is None:
579 raise Exception("Can't check java version: no version string in output"
580 " of 'java -version': '{}'".format(output))
581 version = m.groups(0)[0]
Mads Agerbc7b2ce2018-02-05 11:28:47 +0100582 m = re.search('google', version)
583 if m is not None:
584 raise Exception("Do not use google JVM for benchmarking: " + version)
Tamas Kenez0cad51c2017-08-21 14:42:01 +0200585
Christoffer Quist Adamsen17879c12019-01-22 16:13:54 +0100586def get_android_jar_dir(api):
587 return os.path.join(REPO_ROOT, ANDROID_JAR_DIR.format(api=api))
588
Rico Wind9d70f612018-08-31 09:17:43 +0200589def get_android_jar(api):
590 return os.path.join(REPO_ROOT, ANDROID_JAR.format(api=api))
Rico Windda6836e2018-12-07 12:32:03 +0100591
Christoffer Quist Adamsen17879c12019-01-22 16:13:54 +0100592def get_android_optional_jars(api):
593 android_optional_jars_dir = os.path.join(get_android_jar_dir(api), 'optional')
594 android_optional_jars = [
595 os.path.join(android_optional_jars_dir, 'android.test.base.jar'),
596 os.path.join(android_optional_jars_dir, 'android.test.mock.jar'),
597 os.path.join(android_optional_jars_dir, 'android.test.runner.jar'),
598 os.path.join(android_optional_jars_dir, 'org.apache.http.legacy.jar')
599 ]
600 return [
601 android_optional_jar for android_optional_jar in android_optional_jars
602 if os.path.isfile(android_optional_jar)]
603
Rico Windfaaac012019-02-25 11:24:05 +0100604def is_bot():
Rico Winde11f4392019-03-18 07:59:37 +0100605 return 'SWARMING_BOT_ID' in os.environ
Morten Krogh-Jespersen16e925d2019-01-25 14:40:38 +0100606
607def uncompressed_size(path):
608 return sum(z.file_size for z in zipfile.ZipFile(path).infolist())
Morten Krogh-Jespersen0de13732019-03-01 08:56:39 +0100609
610def getR8Version(path):
611 cmd = [jdk.GetJavaExecutable(), '-cp', path, 'com.android.tools.r8.R8',
612 '--version']
Rico Windfd186372022-02-28 08:55:48 +0100613 output = subprocess.check_output(cmd, stderr = subprocess.STDOUT).decode('utf-8')
Jinseong Jeona2394232019-11-26 22:17:55 -0800614 # output is of the form 'R8 <version> (with additional info)'
615 # so we split on '('; clean up tailing spaces; and strip off 'R8 '.
616 return output.split('(')[0].strip()[3:]
Søren Gjesse6e5e5842019-09-03 08:48:30 +0200617
Søren Gjesse705a3b12022-03-17 11:37:30 +0100618def desugar_configuration_version(configuration):
619 with open(configuration, 'r') as f:
Søren Gjesse6e5e5842019-09-03 08:48:30 +0200620 configuration_json = json.loads(f.read())
621 configuration_format_version = \
622 configuration_json.get('configuration_format_version')
Søren Gjesse6e5e5842019-09-03 08:48:30 +0200623 version = configuration_json.get('version')
624 if not version:
625 raise Exception(
Søren Gjesse705a3b12022-03-17 11:37:30 +0100626 'No "version" found in ' + configuration)
627 check_basic_semver_version(version, 'in ' + configuration, allowPrerelease = True)
Søren Gjesse6e5e5842019-09-03 08:48:30 +0200628 return version
Søren Gjesse1e171532019-09-03 09:44:22 +0200629
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100630class SemanticVersion:
Søren Gjesse705a3b12022-03-17 11:37:30 +0100631 def __init__(self, major, minor, patch, prerelease):
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100632 self.major = major
633 self.minor = minor
634 self.patch = patch
Søren Gjesse705a3b12022-03-17 11:37:30 +0100635 self.prerelease = prerelease
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100636 # Build metadata currently not suppported
637
638 def larger_than(self, other):
Søren Gjesse29d108a2022-04-07 10:49:49 +0200639 if self.prerelease or other.prerelease:
Søren Gjesse705a3b12022-03-17 11:37:30 +0100640 raise Exception("Comparison with prerelease not implemented")
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100641 if self.major > other.major:
642 return True
643 if self.major == other.major and self.minor > other.minor:
644 return True
645 if self.patch:
646 return (self.major == other.major
647 and self.minor == other.minor
648 and self.patch > other.patch)
649 else:
650 return False
651
652
Søren Gjesse705a3b12022-03-17 11:37:30 +0100653# Check that the passed string is formatted as a basic semver version (x.y.z or x.y.z-prerelease
654# depending on the value of allowPrerelease).
655# See https://semver.org/. The regexp parts used are not all complient with what is suggested
656# on https://semver.org/#is-there-a-suggested-regular-expression-regex-to-check-a-semver-string.
657def check_basic_semver_version(version, error_context = '', components = 3, allowPrerelease = False):
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100658 regexp = '^'
659 for x in range(components):
660 regexp += '([0-9]+)'
661 if x < components - 1:
662 regexp += '\\.'
Søren Gjesse705a3b12022-03-17 11:37:30 +0100663 if allowPrerelease:
664 # This part is from
665 # https://semver.org/#is-there-a-suggested-regular-expression-regex-to-check-a-semver-string
666 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 Gjesse4e5c6fe2019-11-05 17:17:43 +0100667 regexp += '$'
668 reg = re.compile(regexp)
669 match = reg.match(version)
670 if not match:
Søren Gjesse1e171532019-09-03 09:44:22 +0200671 raise Exception("Invalid version '"
672 + version
673 + "'"
674 + (' ' + error_context) if len(error_context) > 0 else '')
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100675 if components == 2:
Søren Gjesse705a3b12022-03-17 11:37:30 +0100676 return SemanticVersion(int(match.group(1)), int(match.group(2)), None, None)
677 elif components == 3 and not allowPrerelease:
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100678 return SemanticVersion(
Søren Gjesse705a3b12022-03-17 11:37:30 +0100679 int(match.group(1)), int(match.group(2)), int(match.group(3)), None)
680 elif components == 3 and allowPrerelease:
681 return SemanticVersion(
682 int(match.group(1)), int(match.group(2)), int(match.group(3)), match.group('prerelease'))
Søren Gjesse4e5c6fe2019-11-05 17:17:43 +0100683 else:
684 raise Exception('Argument "components" must be 2 or 3')