Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """Sets environment variables needed to run a chromium unit test.""" | 6 """Sets environment variables needed to run a chromium unit test.""" |
| 7 | 7 |
| 8 import os | 8 import os |
| 9 import stat | 9 import stat |
| 10 import subprocess | 10 import subprocess |
| 11 import sys | 11 import sys |
| 12 | 12 |
| 13 # This is hardcoded to be src/ relative to this script. | 13 # This is hardcoded to be src/ relative to this script. |
| 14 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | 14 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
| 15 | 15 |
| 16 CHROME_SANDBOX_ENV = 'CHROME_DEVEL_SANDBOX' | 16 CHROME_SANDBOX_ENV = 'CHROME_DEVEL_SANDBOX' |
| 17 CHROME_SANDBOX_PATH = '/opt/chromium/chrome_sandbox' | 17 CHROME_SANDBOX_PATH = '/opt/chromium/chrome_sandbox' |
| 18 | 18 |
| 19 | 19 |
| 20 def should_enable_sandbox(cmd, sandbox_path): | 20 def get_sandbox_env(env): |
| 21 """Return a boolean indicating that the current slave is capable of using the | |
| 22 sandbox and should enable it. This should return True iff the slave is a | |
| 23 Linux host with the sandbox file present and configured correctly.""" | |
| 24 if not (sys.platform.startswith('linux') and | |
| 25 os.path.exists(sandbox_path)): | |
| 26 return False | |
| 27 | |
| 28 # Copy the check in tools/build/scripts/slave/runtest.py. | |
| 29 if '--lsan=1' in cmd: | |
| 30 return False | |
| 31 | |
| 32 sandbox_stat = os.stat(sandbox_path) | |
| 33 if ((sandbox_stat.st_mode & stat.S_ISUID) and | |
| 34 (sandbox_stat.st_mode & stat.S_IRUSR) and | |
| 35 (sandbox_stat.st_mode & stat.S_IXUSR) and | |
| 36 (sandbox_stat.st_uid == 0)): | |
| 37 return True | |
| 38 return False | |
| 39 | |
| 40 | |
| 41 def get_sandbox_env(cmd, env, verbose=False): | |
| 42 """Checks enables the sandbox if it is required, otherwise it disables it. | 21 """Checks enables the sandbox if it is required, otherwise it disables it. |
|
jln (very slow on Chromium)
2015/01/27 20:19:48
This comment doesn't seem to match what we wish to
earthdok
2015/01/27 20:44:44
Done.
| |
| 43 Returns the environment flags to set.""" | 22 Returns the environment flags to set.""" |
| 44 extra_env = {} | 23 extra_env = {} |
| 45 chrome_sandbox_path = env.get(CHROME_SANDBOX_ENV, CHROME_SANDBOX_PATH) | 24 chrome_sandbox_path = env.get(CHROME_SANDBOX_ENV, CHROME_SANDBOX_PATH) |
|
jln (very slow on Chromium)
2015/01/27 20:19:48
Could you add a check that CHROME_SANDBOX_ENV is n
earthdok
2015/01/27 20:44:44
Done.
| |
| 46 | 25 extra_env[CHROME_SANDBOX_ENV] = chrome_sandbox_path |
| 47 if should_enable_sandbox(cmd, chrome_sandbox_path): | |
| 48 if verbose: | |
| 49 print 'Enabling sandbox. Setting environment variable:' | |
| 50 print ' %s="%s"' % (CHROME_SANDBOX_ENV, chrome_sandbox_path) | |
| 51 extra_env[CHROME_SANDBOX_ENV] = chrome_sandbox_path | |
| 52 else: | |
| 53 if verbose: | |
| 54 print 'Disabling sandbox. Setting environment variable:' | |
| 55 print ' CHROME_DEVEL_SANDBOX=""' | |
| 56 extra_env['CHROME_DEVEL_SANDBOX'] = '' | |
| 57 | 26 |
| 58 return extra_env | 27 return extra_env |
| 59 | 28 |
| 60 | 29 |
| 61 def trim_cmd(cmd): | 30 def trim_cmd(cmd): |
| 62 """Removes internal flags from cmd since they're just used to communicate from | 31 """Removes internal flags from cmd since they're just used to communicate from |
| 63 the host machine to this script running on the swarm slaves.""" | 32 the host machine to this script running on the swarm slaves.""" |
| 64 internal_flags = frozenset(['--asan=0', '--asan=1', '--lsan=0', '--lsan=1']) | 33 internal_flags = frozenset(['--asan=0', '--asan=1', '--lsan=0', '--lsan=1']) |
| 65 return [i for i in cmd if i not in internal_flags] | 34 return [i for i in cmd if i not in internal_flags] |
| 66 | 35 |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 128 # directory. | 97 # directory. |
| 129 extra_env['DYLD_LIBRARY_PATH'] = str(isolate_output_dir) | 98 extra_env['DYLD_LIBRARY_PATH'] = str(isolate_output_dir) |
| 130 | 99 |
| 131 return extra_env | 100 return extra_env |
| 132 | 101 |
| 133 | 102 |
| 134 def run_executable(cmd, env): | 103 def run_executable(cmd, env): |
| 135 """Runs an executable with: | 104 """Runs an executable with: |
| 136 - environment variable CR_SOURCE_ROOT set to the root directory. | 105 - environment variable CR_SOURCE_ROOT set to the root directory. |
| 137 - environment variable LANGUAGE to en_US.UTF-8. | 106 - environment variable LANGUAGE to en_US.UTF-8. |
| 138 - environment variable CHROME_DEVEL_SANDBOX set if need | 107 - environment variable CHROME_DEVEL_SANDBOX set |
| 139 - Reuses sys.executable automatically. | 108 - Reuses sys.executable automatically. |
| 140 """ | 109 """ |
| 141 extra_env = {} | 110 extra_env = {} |
| 142 # Many tests assume a English interface... | 111 # Many tests assume a English interface... |
| 143 extra_env['LANG'] = 'en_US.UTF-8' | 112 extra_env['LANG'] = 'en_US.UTF-8' |
| 144 # Used by base/base_paths_linux.cc as an override. Just make sure the default | 113 # Used by base/base_paths_linux.cc as an override. Just make sure the default |
| 145 # logic is used. | 114 # logic is used. |
| 146 env.pop('CR_SOURCE_ROOT', None) | 115 env.pop('CR_SOURCE_ROOT', None) |
| 147 extra_env.update(get_sandbox_env(cmd, env)) | 116 extra_env.update(get_sandbox_env(env)) |
| 148 | 117 |
| 149 # Copy logic from tools/build/scripts/slave/runtest.py. | 118 # Copy logic from tools/build/scripts/slave/runtest.py. |
| 150 asan = '--asan=1' in cmd | 119 asan = '--asan=1' in cmd |
| 151 lsan = '--lsan=1' in cmd | 120 lsan = '--lsan=1' in cmd |
| 152 | 121 |
| 153 if asan: | 122 if asan: |
| 154 extra_env.update(get_asan_env(cmd, lsan)) | 123 extra_env.update(get_asan_env(cmd, lsan)) |
| 155 if lsan: | 124 if lsan: |
| 156 cmd.append('--no-sandbox') | 125 cmd.append('--no-sandbox') |
| 157 | 126 |
| (...skipping 27 matching lines...) Expand all Loading... | |
| 185 print >> sys.stderr, 'Failed to start %s' % cmd | 154 print >> sys.stderr, 'Failed to start %s' % cmd |
| 186 raise | 155 raise |
| 187 | 156 |
| 188 | 157 |
| 189 def main(): | 158 def main(): |
| 190 return run_executable(sys.argv[1:], os.environ.copy()) | 159 return run_executable(sys.argv[1:], os.environ.copy()) |
| 191 | 160 |
| 192 | 161 |
| 193 if __name__ == '__main__': | 162 if __name__ == '__main__': |
| 194 sys.exit(main()) | 163 sys.exit(main()) |
| OLD | NEW |