Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 1 | #!/usr/bin/python3 |
| 2 | |
| 3 | import argparse |
| 4 | import sys |
| 5 | import subprocess |
| 6 | import re |
| 7 | import os |
| 8 | import os.path |
| 9 | import string |
| 10 | import shutil |
| 11 | import errno |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 12 | import queue |
| 13 | import threading |
Brian Silverman | f2bbe09 | 2014-05-13 16:55:03 -0700 | [diff] [blame] | 14 | import pty |
Brian Silverman | 6bca472 | 2014-05-20 17:02:49 -0700 | [diff] [blame] | 15 | import signal |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 16 | |
| 17 | class TestThread(threading.Thread): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 18 | """Runs 1 test and keeps track of its current state. |
| 19 | |
| 20 | A TestThread is either waiting to start the test, actually running it, done, |
| 21 | running it, or stopped. The first 3 always happen in that order and can |
| 22 | change to stopped at any time. |
| 23 | |
| 24 | It will finish (ie join() will return) once the process has exited, at which |
| 25 | point accessing process to see the status is OK. |
| 26 | |
| 27 | Attributes: |
| 28 | executable: The file path of the executable to run. |
| 29 | env: The environment variables to set. |
| 30 | done_queue: A queue.Queue to place self on once done running the test. |
| 31 | start_semaphore: A threading.Semaphore to wait on before starting. |
| 32 | process_lock: A lock around process. |
| 33 | process: The currently executing test process or None. Synchronized by |
| 34 | process_lock. |
| 35 | stopped: True if we're stopped. |
| 36 | """ |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 37 | def __init__(self, executable, env, done_queue, start_semaphore): |
| 38 | super(TestThread, self).__init__( |
| 39 | name=os.path.split(executable)[1]) |
| 40 | |
| 41 | self.executable = executable |
| 42 | self.env = env |
| 43 | self.done_queue = done_queue |
| 44 | self.start_semaphore = start_semaphore |
| 45 | |
| 46 | self.process_lock = threading.Lock() |
| 47 | self.process = None |
| 48 | self.stopped = False |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 49 | self.returncode = None |
| 50 | self.output = None |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 51 | |
| 52 | def run(self): |
| 53 | with self.start_semaphore: |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 54 | if self.stopped: |
| 55 | return |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 56 | test_output('Starting test %s...' % self.name) |
Brian Silverman | f2bbe09 | 2014-05-13 16:55:03 -0700 | [diff] [blame] | 57 | self.output, subprocess_output = pty.openpty() |
| 58 | try: |
| 59 | with self.process_lock: |
| 60 | self.process = subprocess.Popen(self.executable, |
| 61 | env=self.env, |
| 62 | stderr=subprocess.STDOUT, |
| 63 | stdout=subprocess_output, |
| 64 | stdin=open(os.devnull, 'r')) |
| 65 | finally: |
| 66 | os.close(subprocess_output) |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 67 | self.process.wait() |
| 68 | with self.process_lock: |
| 69 | self.returncode = self.process.returncode |
| 70 | self.process = None |
| 71 | if not self.stopped: |
| 72 | self.done_queue.put(self) |
| 73 | |
| 74 | def terminate_process(self): |
Brian Silverman | bf0e1db | 2014-05-10 22:13:15 -0700 | [diff] [blame] | 75 | """Asks any currently running process to stop.""" |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 76 | with self.process_lock: |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 77 | if not self.process: |
| 78 | return |
Brian Silverman | a582658 | 2014-06-03 19:46:35 -0700 | [diff] [blame] | 79 | try: |
| 80 | self.process.terminate() |
| 81 | except OSError as e: |
| 82 | if e.errno == errno.ESRCH: |
| 83 | # We don't really care if it's already gone. |
| 84 | pass |
| 85 | else: |
| 86 | raise e |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 87 | def kill_process(self): |
Brian Silverman | bf0e1db | 2014-05-10 22:13:15 -0700 | [diff] [blame] | 88 | """Forcibly terminates any running process.""" |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 89 | with self.process_lock: |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 90 | if not self.process: |
| 91 | return |
Brian Silverman | 6bca472 | 2014-05-20 17:02:49 -0700 | [diff] [blame] | 92 | try: |
| 93 | self.process.kill() |
| 94 | except OSError as e: |
| 95 | if e.errno == errno.ESRCH: |
| 96 | # We don't really care if it's already gone. |
| 97 | pass |
| 98 | else: |
| 99 | raise e |
Brian Silverman | bf0e1db | 2014-05-10 22:13:15 -0700 | [diff] [blame] | 100 | def stop(self): |
| 101 | """Changes self to the stopped state.""" |
| 102 | with self.process_lock: |
| 103 | self.stopped = True |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 104 | |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 105 | def aos_path(): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 106 | """Returns: |
| 107 | A relative path to the aos directory. |
| 108 | """ |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 109 | return os.path.join(os.path.dirname(__file__), '..') |
| 110 | |
| 111 | def get_ip(device): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 112 | """Retrieves the IP address for a given device.""" |
| 113 | FILENAME = os.path.normpath(os.path.join(aos_path(), '..', |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 114 | 'output', 'ip_base.txt')) |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 115 | if not os.access(FILENAME, os.R_OK): |
| 116 | os.makedirs(os.path.dirname(FILENAME), exist_ok=True) |
| 117 | with open(FILENAME, 'w') as f: |
| 118 | f.write('10.9.71') |
| 119 | with open(FILENAME, 'r') as f: |
| 120 | base = f.readline() |
| 121 | if device == 'prime': |
| 122 | return base + '.179' |
| 123 | elif device == 'robot': |
| 124 | return base + '.2' |
| 125 | else: |
| 126 | raise Exception('Unknown device %s to get an IP address for.' % device) |
| 127 | |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 128 | def user_output(message): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 129 | """Prints message to the user.""" |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 130 | print('build.py: ' + message, file=sys.stderr) |
| 131 | |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 132 | # A lock to avoid making a mess intermingling test-related messages. |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 133 | test_output_lock = threading.RLock() |
| 134 | def test_output(message): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 135 | """Prints message to the user. Intended for messages related to tests.""" |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 136 | with test_output_lock: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 137 | print('tests: ' + message, file=sys.stdout) |
| 138 | |
| 139 | def call_download_externals(argument): |
| 140 | """Calls download_externals.sh for a given set of externals. |
| 141 | |
| 142 | Args: |
| 143 | argument: The argument to pass to the shell script to tell it what to |
| 144 | download. |
| 145 | """ |
| 146 | subprocess.check_call( |
| 147 | (os.path.join(aos_path(), 'build', 'download_externals.sh'), |
| 148 | argument), |
| 149 | stdin=open(os.devnull, 'r')) |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 150 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 151 | class Processor(object): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 152 | """Represents a processor architecture we can build for.""" |
| 153 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 154 | class UnknownPlatform(Exception): |
| 155 | def __init__(self, message): |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 156 | super(Processor.UnknownPlatform, self).__init__() |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 157 | self.message = message |
| 158 | |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 159 | class Platform(object): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 160 | """Represents a single way to build the code.""" |
| 161 | |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 162 | def outdir(self): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 163 | """Returns: |
| 164 | The path of the directory build outputs get put in to. |
| 165 | """ |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 166 | return os.path.join( |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 167 | aos_path(), '..', 'output', self.outname()) |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 168 | def build_ninja(self): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 169 | """Returns: |
| 170 | The path of the build.ninja file. |
| 171 | """ |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 172 | return os.path.join(self.outdir(), 'build.ninja') |
| 173 | |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 174 | def do_deploy(self, dry_run, command): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 175 | """Helper for subclasses to implement deploy. |
| 176 | |
| 177 | Args: |
| 178 | dry_run: If True, prints the command instead of actually running it. |
| 179 | command: A tuple of command-line arguments. |
| 180 | """ |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 181 | real_command = (('echo',) + command) if dry_run else command |
| 182 | subprocess.check_call(real_command, stdin=open(os.devnull, 'r')) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 183 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 184 | def deploy(self, dry_run): |
| 185 | """Downloads the compiled code to the target computer.""" |
| 186 | raise NotImplementedError('deploy should be overriden') |
| 187 | def outname(self): |
| 188 | """Returns: |
| 189 | The name of the directory the code will be compiled to. |
| 190 | """ |
| 191 | raise NotImplementedError('outname should be overriden') |
| 192 | def os(self): |
| 193 | """Returns: |
| 194 | The name of the operating system this platform is for. |
| 195 | |
| 196 | This will be used as the value of the OS gyp variable. |
| 197 | """ |
| 198 | raise NotImplementedError('os should be overriden') |
| 199 | def gyp_platform(self): |
| 200 | """Returns: |
| 201 | The platform name the .gyp files know. |
| 202 | |
| 203 | This will be used as the value of the PLATFORM gyp variable. |
| 204 | """ |
| 205 | raise NotImplementedError('gyp_platform should be overriden') |
| 206 | def architecture(self): |
| 207 | """Returns: |
| 208 | The processor architecture for this platform. |
| 209 | |
| 210 | This will be used as the value of the ARCHITECTURE gyp variable. |
| 211 | """ |
| 212 | raise NotImplementedError('architecture should be overriden') |
| 213 | def compiler(self): |
| 214 | """Returns: |
| 215 | The compiler used for this platform. |
| 216 | |
| 217 | Everything before the first _ will be used as the value of the |
| 218 | COMPILER gyp variable and the whole thing will be used as the value |
| 219 | of the FULL_COMPILER gyp variable. |
| 220 | """ |
| 221 | raise NotImplementedError('compiler should be overriden') |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 222 | def sanitizer(self): |
| 223 | """Returns: |
| 224 | The sanitizer used on this platform. |
| 225 | |
| 226 | This will be used as the value of the SANITIZER gyp variable. |
| 227 | |
| 228 | "none" if there isn't one. |
| 229 | """ |
| 230 | raise NotImplementedError('sanitizer should be overriden') |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 231 | def debug(self): |
| 232 | """Returns: |
| 233 | Whether or not this platform compiles with debugging information. |
| 234 | |
| 235 | The DEBUG gyp variable will be set to "yes" or "no" based on this. |
| 236 | """ |
| 237 | raise NotImplementedError('debug should be overriden') |
| 238 | def build_env(self): |
| 239 | """Returns: |
| 240 | A map of environment variables to set while building this platform. |
| 241 | """ |
| 242 | raise NotImplementedError('build_env should be overriden') |
Brian Silverman | bd380fd | 2014-05-13 16:55:24 -0700 | [diff] [blame] | 243 | def priority(self): |
| 244 | """Returns: |
| 245 | A relative priority for this platform relative to other ones. |
| 246 | |
| 247 | Higher priority platforms will get built, tested, etc first. Generally, |
| 248 | platforms which give higher-quality compiler errors etc should come first. |
| 249 | """ |
| 250 | return 0 |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 251 | |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 252 | def check_installed(self, platforms, is_deploy): |
| 253 | """Makes sure that everything necessary to build platforms are installed.""" |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 254 | raise NotImplementedError('check_installed should be overriden') |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 255 | def parse_platforms(self, platforms_string): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 256 | """Args: |
| 257 | string: A user-supplied string saying which platforms to select. |
| 258 | |
| 259 | Returns: |
| 260 | A tuple of Platform objects. |
| 261 | |
| 262 | Raises: |
| 263 | Processor.UnknownPlatform: Parsing string didn't work out. |
| 264 | """ |
| 265 | raise NotImplementedError('parse_platforms should be overriden') |
| 266 | def extra_gyp_flags(self): |
| 267 | """Returns: |
| 268 | A tuple of extra flags to pass to gyp (if any). |
| 269 | """ |
| 270 | return () |
| 271 | def modify_ninja_file(self, ninja_file): |
| 272 | """Modifies a freshly generated ninja file as necessary. |
| 273 | |
| 274 | Args: |
| 275 | ninja_file: Path to the file to modify. |
| 276 | """ |
| 277 | pass |
| 278 | def download_externals(self, platforms): |
| 279 | """Calls download_externals as appropriate to build platforms. |
| 280 | |
| 281 | Args: |
| 282 | platforms: A list of platforms to download external libraries for. |
| 283 | """ |
| 284 | raise NotImplementedError('download_externals should be overriden') |
| 285 | |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 286 | def do_check_installed(self, other_packages): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 287 | """Helper for subclasses to implement check_installed. |
| 288 | |
| 289 | Args: |
| 290 | other_packages: A tuple of platform-specific packages to check for.""" |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 291 | all_packages = other_packages |
| 292 | # Necessary to build stuff. |
| 293 | all_packages += ('ccache', 'make') |
| 294 | # Necessary to download stuff to build. |
| 295 | all_packages += ('wget', 'git', 'subversion', 'patch', 'unzip', 'bzip2') |
| 296 | # Necessary to build externals stuff. |
| 297 | all_packages += ('python', 'gcc', 'g++') |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 298 | try: |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 299 | # TODO(brians): Check versions too. |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 300 | result = subprocess.check_output( |
| 301 | ('dpkg-query', '--show') + all_packages, |
| 302 | stdin=open(os.devnull, 'r'), |
| 303 | stderr=subprocess.STDOUT) |
| 304 | except subprocess.CalledProcessError as e: |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 305 | output = e.output.decode('utf-8').rstrip() |
| 306 | not_found = [] |
| 307 | for line in output.splitlines(True): |
| 308 | match = re.match(r'dpkg-query: no packages found matching (.*)', |
| 309 | line) |
| 310 | if match: |
| 311 | not_found.append(match.group(1)) |
| 312 | user_output('Some packages not installed: %s.' % ', '.join(not_found)) |
| 313 | user_output('Try something like `sudo apt-get install %s`.' % |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 314 | ' '.join(not_found)) |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 315 | exit(1) |
| 316 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 317 | class CRIOProcessor(Processor): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 318 | """A Processor subclass for building cRIO code.""" |
| 319 | |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 320 | class Platform(Processor.Platform): |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 321 | def __init__(self, debug, wind_base): |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 322 | super(CRIOProcessor.Platform, self).__init__() |
| 323 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 324 | self.__debug = debug |
| 325 | self.__wind_base = wind_base |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 326 | |
| 327 | def __repr__(self): |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 328 | return 'CRIOProcessor.Platform(debug=%s)' % self.debug() |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 329 | def __str__(self): |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 330 | return 'crio%s' % ('-debug' if self.debug() else '') |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 331 | |
| 332 | def outname(self): |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 333 | return 'crio-debug' if self.debug() else 'crio' |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 334 | def os(self): |
| 335 | return 'vxworks' |
| 336 | def gyp_platform(self): |
| 337 | return 'crio' |
| 338 | def architecture(self): |
| 339 | return 'ppc' |
| 340 | def compiler(self): |
| 341 | return 'gcc' |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 342 | def sanitizer(self): |
| 343 | return 'none' |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 344 | def debug(self): |
| 345 | return self.__debug |
| 346 | def wind_base(self): |
| 347 | return self.__wind_base |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 348 | |
Brian Silverman | e48c09a | 2014-04-30 18:04:58 -0700 | [diff] [blame] | 349 | # TODO(brians): test this |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 350 | def deploy(self, dry_run): |
| 351 | self.do_deploy(dry_run, |
| 352 | ('ncftpput', get_ip('robot'), '/', |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 353 | os.path.join(self.outdir(), 'lib', |
| 354 | 'FRC_UserProgram.out'))) |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 355 | |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 356 | def build_env(self): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 357 | return {'WIND_BASE': self.wind_base()} |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 358 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 359 | def __init__(self): |
| 360 | super(CRIOProcessor, self).__init__() |
| 361 | |
| 362 | if 'WIND_BASE' in os.environ: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 363 | self.__wind_base = os.environ['WIND_BASE'] |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 364 | else: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 365 | self.__wind_base = '/usr/local/powerpc-wrs-vxworks/wind_base' |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 366 | |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 367 | def parse_platforms(self, platforms_string): |
| 368 | if platforms_string is None or platforms_string == 'crio': |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 369 | return (CRIOProcessor.Platform(False, self.wind_base()),) |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 370 | elif string == 'crio-debug' or string == 'debug': |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 371 | return (CRIOProcessor.Platform(True, self.wind_base()),) |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 372 | else: |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 373 | raise Processor.UnknownPlatform( |
| 374 | 'Unknown cRIO platform "%s".' % platforms_string) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 375 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 376 | def wind_base(self): |
| 377 | return self.__wind_base |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 378 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 379 | def extra_gyp_flags(self): |
| 380 | return ('-DWIND_BASE=%s' % self.wind_base(),) |
| 381 | |
| 382 | def modify_ninja_file(self, ninja_file): |
| 383 | subprocess.check_call( |
| 384 | ('sed', '-i', |
| 385 | 's/nm -gD/nm/g', ninja_file), |
| 386 | stdin=open(os.devnull, 'r')) |
| 387 | |
| 388 | def download_externals(self, _): |
| 389 | call_download_externals('crio') |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 390 | |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 391 | def check_installed(self, platforms, is_deploy): |
| 392 | packages = ('powerpc-wrs-vxworks', 'tcl') |
| 393 | if is_deploy: |
| 394 | packages += ('ncftp',) |
| 395 | self.do_check_installed(packages) |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 396 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 397 | class PrimeProcessor(Processor): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 398 | """A Processor subclass for building prime code.""" |
| 399 | |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 400 | class Platform(Processor.Platform): |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 401 | def __init__(self, architecture, compiler, debug, sanitizer): |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 402 | super(PrimeProcessor.Platform, self).__init__() |
| 403 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 404 | self.__architecture = architecture |
| 405 | self.__compiler = compiler |
| 406 | self.__debug = debug |
| 407 | self.__sanitizer = sanitizer |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 408 | |
| 409 | def __repr__(self): |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 410 | return 'PrimeProcessor.Platform(architecture=%s, compiler=%s, debug=%s' \ |
| 411 | ', sanitizer=%s)' \ |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 412 | % (self.architecture(), self.compiler(), self.debug(), |
| 413 | self.sanitizer()) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 414 | def __str__(self): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 415 | return '%s-%s%s-%s' % (self.architecture(), self.compiler(), |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 416 | '-debug' if self.debug() else '', self.sanitizer()) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 417 | |
| 418 | def os(self): |
| 419 | return 'linux' |
| 420 | def gyp_platform(self): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 421 | return '%s-%s-%s' % (self.os(), self.architecture(), self.compiler()) |
| 422 | def architecture(self): |
| 423 | return self.__architecture |
| 424 | def compiler(self): |
| 425 | return self.__compiler |
| 426 | def sanitizer(self): |
| 427 | return self.__sanitizer |
| 428 | def debug(self): |
| 429 | return self.__debug |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 430 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 431 | def outname(self): |
| 432 | return str(self) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 433 | |
Brian Silverman | bd380fd | 2014-05-13 16:55:24 -0700 | [diff] [blame] | 434 | def priority(self): |
| 435 | r = 0 |
| 436 | if self.compiler() == 'gcc': |
| 437 | r -= 100 |
| 438 | elif self.compiler() == 'clang': |
| 439 | r += 100 |
| 440 | if self.sanitizer() != 'none': |
| 441 | r -= 50 |
| 442 | elif self.debug(): |
| 443 | r -= 10 |
| 444 | if self.architecture() == 'amd64': |
| 445 | r += 5 |
| 446 | return r |
| 447 | |
Brian Silverman | e48c09a | 2014-04-30 18:04:58 -0700 | [diff] [blame] | 448 | # TODO(brians): test this |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 449 | def deploy(self, dry_run): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 450 | # Downloads code to the prime in a way that avoids clashing too badly with |
| 451 | # starter (like the naive download everything one at a time). |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 452 | SUM = 'md5sum' |
| 453 | TARGET_DIR = '/home/driver/robot_code/bin' |
| 454 | TEMP_DIR = '/tmp/aos_downloader' |
| 455 | TARGET = 'driver@' + get_ip('prime') |
| 456 | |
| 457 | from_dir = os.path.join(self.outdir(), 'outputs') |
| 458 | sums = subprocess.check_output((SUM,) + tuple(os.listdir(from_dir)), |
| 459 | stdin=open(os.devnull, 'r'), |
| 460 | cwd=from_dir) |
| 461 | to_download = subprocess.check_output( |
| 462 | ('ssh', TARGET, |
| 463 | """rm -rf {TMPDIR} && mkdir {TMPDIR} && cd {TO_DIR} |
| 464 | && echo '{SUMS}' | {SUM} --check --quiet |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 465 | |& grep -F FAILED | sed 's/^\\(.*\\): FAILED.*"'$'"/\\1/g'""". |
| 466 | format(TMPDIR=TEMP_DIR, TO_DIR=TARGET_DIR, SUMS=sums, SUM=SUM))) |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 467 | if not to_download: |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 468 | user_output("Nothing to download") |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 469 | return |
| 470 | self.do_deploy( |
| 471 | dry_run, |
| 472 | ('scp', '-o', 'Compression yes') + to_download |
| 473 | + (('%s:%s' % (TARGET, TEMP_DIR)),)) |
| 474 | if not dry_run: |
| 475 | subprocess.check_call( |
| 476 | ('ssh', TARGET, |
| 477 | """mv {TMPDIR}/* {TO_DIR} |
| 478 | && echo 'Done moving new executables into place' |
| 479 | && ionice -c 3 bash -c 'sync && sync && sync'""".format( |
| 480 | TMPDIR=TEMP_DIR, TO_DIR=TARGET_DIR))) |
| 481 | |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 482 | def build_env(self): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 483 | OTHER_SYSROOT = '/opt/clang-3.5/' |
| 484 | SYMBOLIZER_PATH = OTHER_SYSROOT + 'bin/llvm-symbolizer' |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 485 | r = {} |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 486 | if self.compiler() == 'clang' or self.compiler() == 'gcc_4.8': |
| 487 | r['LD_LIBRARY_PATH'] = OTHER_SYSROOT + 'lib64' |
| 488 | if self.sanitizer() == 'address': |
| 489 | r['ASAN_SYMBOLIZER_PATH'] = SYMBOLIZER_PATH |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 490 | r['ASAN_OPTIONS'] = \ |
Brian Silverman | 407ca82 | 2014-06-05 18:36:41 -0700 | [diff] [blame^] | 491 | 'detect_leaks=1:check_initialization_order=1:strict_init_order=1' \ |
| 492 | ':detect_stack_use_after_return=1:detect_odr_violation=2' |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 493 | elif self.sanitizer() == 'memory': |
| 494 | r['MSAN_SYMBOLIZER_PATH'] = SYMBOLIZER_PATH |
| 495 | elif self.sanitizer() == 'thread': |
| 496 | r['TSAN_OPTIONS'] = 'external_symbolizer_path=' + SYMBOLIZER_PATH |
Brian Silverman | d3fac73 | 2014-05-03 16:03:46 -0700 | [diff] [blame] | 497 | |
| 498 | r['CCACHE_COMPRESS'] = 'yes' |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 499 | r['CCACHE_DIR'] = os.path.abspath(os.path.join(aos_path(), '..', 'output', |
| 500 | 'ccache_dir')) |
Brian Silverman | d3fac73 | 2014-05-03 16:03:46 -0700 | [diff] [blame] | 501 | r['CCACHE_HASHDIR'] = 'yes' |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 502 | if self.compiler() == 'clang': |
Brian Silverman | d3fac73 | 2014-05-03 16:03:46 -0700 | [diff] [blame] | 503 | # clang doesn't like being run directly on the preprocessed files. |
| 504 | r['CCACHE_CPP2'] = 'yes' |
| 505 | # Without this, ccache slows down because of the generated header files. |
| 506 | # The race condition that this opens up isn't a problem because the build |
| 507 | # system finishes modifying header files before compiling anything that |
| 508 | # uses them. |
| 509 | r['CCACHE_SLOPPINESS'] = 'include_file_mtime' |
Brian Silverman | d3fac73 | 2014-05-03 16:03:46 -0700 | [diff] [blame] | 510 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 511 | if self.architecture() == 'amd64': |
Brian Silverman | d3fac73 | 2014-05-03 16:03:46 -0700 | [diff] [blame] | 512 | r['PATH'] = os.path.join(aos_path(), 'build', 'bin-ld.gold') + \ |
| 513 | ':' + os.environ['PATH'] |
| 514 | |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 515 | return r |
| 516 | |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 517 | ARCHITECTURES = ('arm', 'amd64') |
| 518 | COMPILERS = ('clang', 'gcc', 'gcc_4.8') |
| 519 | SANITIZERS = ('address', 'undefined', 'integer', 'memory', 'thread', 'none') |
| 520 | SANITIZER_TEST_WARNINGS = { |
| 521 | 'memory': (True, |
| 522 | """We don't have all of the libraries instrumented which leads to lots of false |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 523 | errors with msan (especially stdlibc++). |
| 524 | TODO(brians): Figure out a way to deal with it."""), |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 525 | } |
Brian Silverman | a5301e3 | 2014-05-03 10:51:49 -0700 | [diff] [blame] | 526 | PIE_SANITIZERS = ('memory', 'thread') |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 527 | |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 528 | def __init__(self, is_test, is_deploy): |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 529 | super(PrimeProcessor, self).__init__() |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 530 | |
| 531 | platforms = [] |
| 532 | for architecture in PrimeProcessor.ARCHITECTURES: |
| 533 | for compiler in PrimeProcessor.COMPILERS: |
| 534 | for debug in [True, False]: |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 535 | if architecture == 'arm' and compiler == 'gcc_4.8': |
| 536 | # We don't have a compiler to use here. |
| 537 | continue |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 538 | platforms.append( |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 539 | PrimeProcessor.Platform(architecture, compiler, debug, 'none')) |
| 540 | for sanitizer in PrimeProcessor.SANITIZERS: |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 541 | for compiler in ('gcc_4.8', 'clang'): |
| 542 | if compiler == 'gcc_4.8' and (sanitizer == 'undefined' or |
| 543 | sanitizer == 'integer' or |
| 544 | sanitizer == 'memory'): |
| 545 | # GCC 4.8 doesn't support these sanitizers. |
| 546 | continue |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 547 | if sanitizer == 'none': |
| 548 | # We already added sanitizer == 'none' above. |
| 549 | continue |
| 550 | platforms.append( |
| 551 | PrimeProcessor.Platform('amd64', compiler, True, sanitizer)) |
| 552 | self.__platforms = frozenset(platforms) |
| 553 | |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 554 | if is_test: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 555 | default_platforms = self.select_platforms(architecture='amd64', |
| 556 | debug=True) |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 557 | for sanitizer, warning in PrimeProcessor.SANITIZER_TEST_WARNINGS.items(): |
| 558 | if warning[0]: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 559 | default_platforms -= self.select_platforms(sanitizer=sanitizer) |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 560 | elif is_deploy: |
| 561 | # TODO(brians): Switch to deploying the code built with clang. |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 562 | default_platforms = self.select_platforms(architecture='arm', |
| 563 | compiler='gcc', |
| 564 | debug=False) |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 565 | else: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 566 | default_platforms = self.select_platforms(debug=False) |
| 567 | self.__default_platforms = frozenset(default_platforms) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 568 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 569 | def platforms(self): |
| 570 | return self.__platforms |
| 571 | def default_platforms(self): |
| 572 | return self.__default_platforms |
| 573 | |
| 574 | def download_externals(self, platforms): |
| 575 | to_download = set() |
| 576 | for architecture in PrimeProcessor.ARCHITECTURES: |
| 577 | for sanitizer in PrimeProcessor.PIE_SANITIZERS: |
| 578 | if platforms & self.select_platforms(architecture=architecture, |
| 579 | sanitizer=sanitizer): |
| 580 | to_download.add(architecture + '-fPIE') |
| 581 | if platforms & self.select_platforms(architecture=architecture, |
| 582 | sanitizer='none'): |
| 583 | to_download.add(architecture) |
| 584 | for download_target in to_download: |
| 585 | call_download_externals(download_target) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 586 | |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 587 | def parse_platforms(self, platform_string): |
| 588 | if platform_string is None: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 589 | return self.default_platforms() |
| 590 | r = self.default_platforms() |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 591 | for part in platform_string.split(','): |
Brian Silverman | 1867aae | 2014-05-05 17:16:34 -0700 | [diff] [blame] | 592 | if part == 'all': |
| 593 | r = self.platforms() |
| 594 | elif part[0] == '+': |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 595 | r = r | self.select_platforms_string(part[1:]) |
| 596 | elif part[0] == '-': |
| 597 | r = r - self.select_platforms_string(part[1:]) |
| 598 | elif part[0] == '=': |
| 599 | r = self.select_platforms_string(part[1:]) |
| 600 | else: |
Brian Silverman | 7cd5ad4 | 2014-04-27 08:11:30 -0500 | [diff] [blame] | 601 | selected = self.select_platforms_string(part) |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 602 | r = r - (self.platforms() - selected) |
Brian Silverman | 7cd5ad4 | 2014-04-27 08:11:30 -0500 | [diff] [blame] | 603 | if not r: |
| 604 | r = selected |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 605 | return r |
| 606 | |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 607 | def select_platforms(self, architecture=None, compiler=None, debug=None, |
| 608 | sanitizer=None): |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 609 | r = [] |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 610 | for platform in self.platforms(): |
| 611 | if architecture is None or platform.architecture() == architecture: |
| 612 | if compiler is None or platform.compiler() == compiler: |
| 613 | if debug is None or platform.debug() == debug: |
| 614 | if sanitizer is None or platform.sanitizer() == sanitizer: |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 615 | r.append(platform) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 616 | return set(r) |
| 617 | |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 618 | def select_platforms_string(self, platforms_string): |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 619 | architecture, compiler, debug, sanitizer = None, None, None, None |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 620 | for part in platforms_string.split('-'): |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 621 | if part in PrimeProcessor.ARCHITECTURES: |
| 622 | architecture = part |
| 623 | elif part in PrimeProcessor.COMPILERS: |
| 624 | compiler = part |
| 625 | elif part in ['debug', 'dbg']: |
| 626 | debug = True |
| 627 | elif part in ['release', 'nodebug', 'ndb']: |
| 628 | debug = False |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 629 | elif part in PrimeProcessor.SANITIZERS: |
| 630 | sanitizer = part |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 631 | else: |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 632 | raise Processor.UnknownPlatform( |
| 633 | 'Unknown platform string component "%s".' % part) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 634 | return self.select_platforms( |
| 635 | architecture=architecture, |
| 636 | compiler=compiler, |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 637 | debug=debug, |
| 638 | sanitizer=sanitizer) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 639 | |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 640 | def check_installed(self, platforms, is_deploy): |
| 641 | packages = set(('lzip', 'm4', 'realpath')) |
| 642 | packages.add('ruby') |
| 643 | # clang-format from here gets used for all versions. |
| 644 | packages.add('clang-3.5') |
| 645 | packages.add('arm-eabi-gcc') |
| 646 | for platform in platforms: |
| 647 | if platform.architecture() == 'arm': |
| 648 | packages.add('gcc-4.7-arm-linux-gnueabihf') |
| 649 | packages.add('g++-4.7-arm-linux-gnueabihf') |
| 650 | if platform.compiler() == 'clang' or platform.compiler() == 'gcc_4.8': |
| 651 | packages.add('clang-3.5') |
Brian Silverman | 1867aae | 2014-05-05 17:16:34 -0700 | [diff] [blame] | 652 | if platform.compiler() == 'gcc_4.8': |
| 653 | packages.add('libcloog-isl3:amd64') |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 654 | if is_deploy: |
| 655 | packages.add('openssh-client') |
| 656 | if platform.compiler() == 'gcc' and platform.architecture() == 'amd64': |
| 657 | packages.add('gcc-4.7') |
| 658 | packages.add('g++-4.7') |
| 659 | |
| 660 | self.do_check_installed(tuple(packages)) |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 661 | |
Brian Silverman | 6bca472 | 2014-05-20 17:02:49 -0700 | [diff] [blame] | 662 | def strsignal(num): |
| 663 | # It ends up with SIGIOT instead otherwise, which is weird. |
| 664 | if num == signal.SIGABRT: |
| 665 | return 'SIGABRT' |
| 666 | # SIGCLD is a weird way to spell it. |
| 667 | if num == signal.SIGCHLD: |
| 668 | return 'SIGCHLD' |
| 669 | |
| 670 | SIGNALS_TO_NAMES = dict((getattr(signal, n), n) |
| 671 | for n in dir(signal) if n.startswith('SIG') |
| 672 | and '_' not in n) |
| 673 | return SIGNALS_TO_NAMES.get(num, 'Unknown signal %d' % num) |
| 674 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 675 | def main(): |
| 676 | class TryParsingAgain(Exception): |
| 677 | pass |
| 678 | |
| 679 | class TryAgainArgumentParser(argparse.ArgumentParser): |
| 680 | def __init__(self, **kwargs): |
| 681 | super(TryAgainArgumentParser, self).__init__(**kwargs) |
| 682 | |
| 683 | def error(self, message): |
| 684 | raise TryParsingAgain |
| 685 | |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 686 | def set_up_parser(parser, args): |
| 687 | def add_build_args(parser): |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 688 | parser.add_argument( |
| 689 | 'target', |
| 690 | help='target to build', |
| 691 | nargs='*') |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 692 | parser.add_argument( |
| 693 | '--jobs', '-j', |
| 694 | help='number of things to do at once', |
| 695 | type=int) |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 696 | def add_common_args(parser): |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 697 | parser.add_argument( |
| 698 | 'platforms', |
| 699 | help='platform(s) to act on', |
| 700 | nargs='?') |
| 701 | |
| 702 | parser.add_argument('--processor', required=True, help='prime or crio') |
| 703 | parser.add_argument('--main_gyp', required=True, help='main .gyp file') |
| 704 | subparsers = parser.add_subparsers(dest='action_name') |
| 705 | |
| 706 | build_parser = subparsers.add_parser( |
| 707 | 'build', |
| 708 | help='build the code (default)') |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 709 | add_common_args(build_parser) |
| 710 | add_build_args(build_parser) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 711 | |
| 712 | clean_parser = subparsers.add_parser( |
| 713 | 'clean', |
| 714 | help='remove all output directories') |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 715 | add_common_args(clean_parser) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 716 | |
| 717 | deploy_parser = subparsers.add_parser( |
| 718 | 'deploy', |
| 719 | help='build and download the code') |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 720 | add_common_args(deploy_parser) |
| 721 | add_build_args(deploy_parser) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 722 | deploy_parser.add_argument( |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 723 | '-n', '--dry-run', |
| 724 | help="don't actually download anything", |
| 725 | action='store_true') |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 726 | |
Brian Silverman | e48c09a | 2014-04-30 18:04:58 -0700 | [diff] [blame] | 727 | tests_parser = subparsers.add_parser( |
| 728 | 'tests', |
| 729 | help='run tests') |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 730 | add_common_args(tests_parser) |
| 731 | add_build_args(tests_parser) |
Brian Silverman | e48c09a | 2014-04-30 18:04:58 -0700 | [diff] [blame] | 732 | |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 733 | return parser.parse_args(args) |
| 734 | |
| 735 | try: |
| 736 | parser = TryAgainArgumentParser() |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 737 | args = set_up_parser(parser, sys.argv[1:]) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 738 | except TryParsingAgain: |
| 739 | parser = argparse.ArgumentParser() |
| 740 | REQUIRED_ARGS_END = 5 |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 741 | args = set_up_parser(parser, sys.argv[1:REQUIRED_ARGS_END] + ['build'] + |
| 742 | sys.argv[(REQUIRED_ARGS_END):]) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 743 | |
| 744 | if args.processor == 'crio': |
| 745 | processor = CRIOProcessor() |
| 746 | elif args.processor == 'prime': |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 747 | processor = PrimeProcessor(args.action_name == 'tests', |
| 748 | args.action_name == 'deploy') |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 749 | else: |
| 750 | parser.exit(status=1, message='Unknown processor "%s".' % args.processor) |
| 751 | |
| 752 | if 'target' in args: |
| 753 | targets = args.target[:] |
| 754 | else: |
| 755 | targets = [] |
| 756 | unknown_platform_error = None |
| 757 | try: |
| 758 | platforms = processor.parse_platforms(args.platforms) |
| 759 | except Processor.UnknownPlatform as e: |
| 760 | unknown_platform_error = e.message |
| 761 | targets.append(args.platforms) |
| 762 | platforms = processor.parse_platforms(None) |
| 763 | if not platforms: |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 764 | user_output("No platforms selected!") |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 765 | exit(1) |
| 766 | |
Brian Silverman | 9b7a684 | 2014-05-05 16:19:11 -0700 | [diff] [blame] | 767 | processor.check_installed(platforms, args.action_name == 'deploy') |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 768 | processor.download_externals(platforms) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 769 | |
| 770 | class ToolsConfig(object): |
| 771 | def __init__(self): |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 772 | self.variables = {'AOS': aos_path()} |
| 773 | with open(os.path.join(aos_path(), 'build', 'tools_config'), 'r') as f: |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 774 | for line in f: |
| 775 | if line[0] == '#': |
| 776 | pass |
| 777 | elif line.isspace(): |
| 778 | pass |
| 779 | else: |
| 780 | new_name, new_value = line.rstrip().split('=') |
| 781 | for name, value in self.variables.items(): |
| 782 | new_value = new_value.replace('${%s}' % name, value) |
| 783 | self.variables[new_name] = new_value |
| 784 | def __getitem__(self, key): |
| 785 | return self.variables[key] |
| 786 | |
| 787 | tools_config = ToolsConfig() |
| 788 | |
| 789 | def handle_clean_error(function, path, excinfo): |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 790 | _, _ = function, path |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 791 | if issubclass(OSError, excinfo[0]): |
| 792 | if excinfo[1].errno == errno.ENOENT: |
| 793 | # Who cares if the file we're deleting isn't there? |
| 794 | return |
| 795 | raise excinfo[1] |
| 796 | |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 797 | def need_to_run_gyp(platform): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 798 | """Determines if we need to run gyp again or not. |
| 799 | |
| 800 | The generated build files are supposed to re-run gyp again themselves, but |
| 801 | that doesn't work (or at least it used to not) and we sometimes want to |
| 802 | modify the results anyways. |
| 803 | |
| 804 | Args: |
| 805 | platform: The platform to check for. |
| 806 | """ |
Brian Silverman | f0d3c78 | 2014-05-02 23:56:32 -0700 | [diff] [blame] | 807 | if not os.path.exists(platform.build_ninja()): |
| 808 | return True |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 809 | if os.path.getmtime(__file__) > os.path.getmtime(platform.build_ninja()): |
| 810 | return True |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 811 | dirs = os.listdir(os.path.join(aos_path(), '..')) |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 812 | # Looking through these folders takes a long time and isn't useful. |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 813 | if dirs.count('output'): |
| 814 | dirs.remove('output') |
| 815 | if dirs.count('.git'): |
| 816 | dirs.remove('.git') |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 817 | return not not subprocess.check_output( |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 818 | ('find',) + tuple(os.path.join(aos_path(), '..', d) for d in dirs) |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 819 | + ('-newer', platform.build_ninja(), |
| 820 | '(', '-name', '*.gyp', '-or', '-name', '*.gypi', ')'), |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 821 | stdin=open(os.devnull, 'r')) |
| 822 | |
| 823 | def env(platform): |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 824 | """Makes sure we pass through important environmental variables. |
| 825 | |
| 826 | Returns: |
| 827 | An environment suitable for passing to subprocess.Popen and friends. |
| 828 | """ |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 829 | build_env = dict(platform.build_env()) |
Brian Silverman | d3fac73 | 2014-05-03 16:03:46 -0700 | [diff] [blame] | 830 | if not 'TERM' in build_env: |
| 831 | build_env['TERM'] = os.environ['TERM'] |
| 832 | if not 'PATH' in build_env: |
| 833 | build_env['PATH'] = os.environ['PATH'] |
Brian Silverman | a4aff56 | 2014-05-02 17:43:50 -0700 | [diff] [blame] | 834 | return build_env |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 835 | |
Brian Silverman | bd380fd | 2014-05-13 16:55:24 -0700 | [diff] [blame] | 836 | sorted_platforms = sorted(platforms, |
| 837 | key=lambda platform: -platform.priority()) |
| 838 | |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 839 | to_build = [] |
Brian Silverman | bd380fd | 2014-05-13 16:55:24 -0700 | [diff] [blame] | 840 | for platform in sorted_platforms: |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 841 | to_build.append(str(platform)) |
| 842 | if len(to_build) > 1: |
| 843 | to_build[-1] = 'and ' + to_build[-1] |
| 844 | user_output('Building %s...' % ', '.join(to_build)) |
| 845 | |
| 846 | if args.action_name == 'tests': |
| 847 | for sanitizer, warning in PrimeProcessor.SANITIZER_TEST_WARNINGS.items(): |
| 848 | warned_about = platforms & processor.select_platforms(sanitizer=sanitizer) |
| 849 | if warned_about: |
| 850 | user_output(warning[1]) |
| 851 | if warning[0]: |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 852 | # TODO(brians): Add a --force flag or something to override this? |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 853 | user_output('Refusing to run tests for sanitizer %s.' % sanitizer) |
| 854 | exit(1) |
| 855 | |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 856 | num = 1 |
Brian Silverman | bd380fd | 2014-05-13 16:55:24 -0700 | [diff] [blame] | 857 | for platform in sorted_platforms: |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 858 | user_output('Building %s (%d/%d)...' % (platform, num, len(platforms))) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 859 | if args.action_name == 'clean': |
| 860 | shutil.rmtree(platform.outdir(), onerror=handle_clean_error) |
| 861 | else: |
| 862 | if need_to_run_gyp(platform): |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 863 | user_output('Running gyp...') |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 864 | gyp = subprocess.Popen( |
| 865 | (tools_config['GYP'], |
| 866 | '--check', |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 867 | '--depth=%s' % os.path.join(aos_path(), '..'), |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 868 | '--no-circular-check', |
| 869 | '-f', 'ninja', |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 870 | '-I%s' % os.path.join(aos_path(), 'build', 'aos.gypi'), |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 871 | '-I/dev/stdin', '-Goutput_dir=output', |
| 872 | '-DOS=%s' % platform.os(), |
| 873 | '-DPLATFORM=%s' % platform.gyp_platform(), |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 874 | '-DARCHITECTURE=%s' % platform.architecture(), |
| 875 | '-DCOMPILER=%s' % platform.compiler().split('_')[0], |
| 876 | '-DFULL_COMPILER=%s' % platform.compiler(), |
| 877 | '-DDEBUG=%s' % ('yes' if platform.debug() else 'no'), |
| 878 | '-DSANITIZER=%s' % platform.sanitizer(), |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 879 | '-DSANITIZER_FPIE=%s' % |
| 880 | ('-fPIE' if platform.sanitizer() in PrimeProcessor.PIE_SANITIZERS |
| 881 | else '')) + |
Brian Silverman | b3d5054 | 2014-04-23 14:28:55 -0500 | [diff] [blame] | 882 | processor.extra_gyp_flags() + (args.main_gyp,), |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 883 | stdin=subprocess.PIPE) |
| 884 | gyp.communicate((""" |
| 885 | { |
| 886 | 'target_defaults': { |
| 887 | 'configurations': { |
| 888 | '%s': {} |
| 889 | } |
| 890 | } |
| 891 | }""" % platform.outname()).encode()) |
| 892 | if gyp.returncode: |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 893 | user_output("Running gyp failed!") |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 894 | exit(1) |
Brian Silverman | e6bada6 | 2014-05-04 16:16:54 -0700 | [diff] [blame] | 895 | processor.modify_ninja_file(platform.build_ninja()) |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 896 | user_output('Done running gyp') |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 897 | else: |
Brian Silverman | 47cd6f6 | 2014-05-03 10:35:52 -0700 | [diff] [blame] | 898 | user_output("Not running gyp") |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 899 | |
| 900 | try: |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 901 | call = (tools_config['NINJA'], |
| 902 | '-C', platform.outdir()) + tuple(targets) |
| 903 | if args.jobs: |
| 904 | call += ('-j', str(args.jobs)) |
| 905 | subprocess.check_call(call, |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 906 | stdin=open(os.devnull, 'r'), |
| 907 | env=env(platform)) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 908 | except subprocess.CalledProcessError as e: |
| 909 | if unknown_platform_error is not None: |
Brian Silverman | a9b1e5c | 2014-04-30 18:08:04 -0700 | [diff] [blame] | 910 | user_output(unknown_platform_error) |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 911 | raise e |
| 912 | |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 913 | if args.action_name == 'deploy': |
| 914 | platform.deploy(args.dry_run) |
Brian Silverman | e48c09a | 2014-04-30 18:04:58 -0700 | [diff] [blame] | 915 | elif args.action_name == 'tests': |
| 916 | dirname = os.path.join(platform.outdir(), 'tests') |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 917 | done_queue = queue.Queue() |
| 918 | running = [] |
| 919 | if args.jobs: |
| 920 | number_jobs = args.jobs |
| 921 | else: |
| 922 | number_jobs = os.sysconf('SC_NPROCESSORS_ONLN') + 2 |
| 923 | test_start_semaphore = threading.Semaphore(number_jobs) |
| 924 | if targets: |
| 925 | to_run = [] |
| 926 | for target in targets: |
| 927 | if target.endswith('_test'): |
| 928 | to_run.append(target) |
| 929 | else: |
| 930 | to_run = os.listdir(dirname) |
| 931 | for f in to_run: |
| 932 | thread = TestThread(os.path.join(dirname, f), env(platform), done_queue, |
| 933 | test_start_semaphore) |
| 934 | running.append(thread) |
| 935 | thread.start() |
| 936 | try: |
| 937 | while running: |
| 938 | done = done_queue.get() |
| 939 | running.remove(done) |
| 940 | with test_output_lock: |
| 941 | test_output('Output from test %s:' % done.name) |
Brian Silverman | f2bbe09 | 2014-05-13 16:55:03 -0700 | [diff] [blame] | 942 | with os.fdopen(done.output) as output_file: |
| 943 | try: |
| 944 | for line in output_file: |
| 945 | if not sys.stdout.isatty(): |
| 946 | # Remove color escape codes. |
| 947 | line = re.sub(r'\x1B\[[0-9;]*[a-zA-Z]', '', line) |
| 948 | sys.stdout.write(line) |
| 949 | except IOError as e: |
| 950 | # We want to just ignore EIOs from reading the master pty because that just |
| 951 | # means we hit the end. |
| 952 | if e.errno != errno.EIO: |
| 953 | raise e |
| 954 | if not done.returncode: |
| 955 | test_output('Test %s succeeded' % done.name) |
| 956 | else: |
Brian Silverman | 6bca472 | 2014-05-20 17:02:49 -0700 | [diff] [blame] | 957 | if done.returncode < 0: |
| 958 | sig = -done.returncode |
| 959 | test_output('Test %s was killed by signal %d (%s)' % \ |
| 960 | (done.name, sig, strsignal(sig))) |
| 961 | elif done.returncode != 1: |
| 962 | test_output('Test %s exited with %d' % \ |
| 963 | (done.name, done.returncode)) |
| 964 | else: |
| 965 | test_output('Test %s failed' % done.name) |
| 966 | user_output('Aborting because of test failure for %s.' % \ |
| 967 | platform) |
Brian Silverman | f2bbe09 | 2014-05-13 16:55:03 -0700 | [diff] [blame] | 968 | exit(1) |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 969 | finally: |
| 970 | if running: |
| 971 | test_output('Killing other tests...') |
Brian Silverman | bf0e1db | 2014-05-10 22:13:15 -0700 | [diff] [blame] | 972 | # Stop all of them before killing processes because otherwise stopping some of |
| 973 | # them tends to let other ones that are waiting to start go. |
| 974 | for thread in running: |
Brian Silverman | a582658 | 2014-06-03 19:46:35 -0700 | [diff] [blame] | 975 | test_output('\tKilling %s' % thread.name) |
Brian Silverman | bf0e1db | 2014-05-10 22:13:15 -0700 | [diff] [blame] | 976 | thread.stop() |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 977 | for thread in running: |
| 978 | thread.terminate_process() |
| 979 | to_remove = [] |
| 980 | for thread in running: |
| 981 | thread.join(5) |
| 982 | if not thread.is_alive(): |
Brian Silverman | 452aaec | 2014-05-05 16:52:18 -0700 | [diff] [blame] | 983 | to_remove.append(thread) |
| 984 | for thread in to_remove: |
| 985 | running.remove(thread) |
Brian Silverman | c3740c3 | 2014-05-04 12:42:47 -0700 | [diff] [blame] | 986 | for thread in running: |
| 987 | test_output( |
| 988 | 'Test %s did not terminate. Killing it.' % thread.name) |
| 989 | thread.kill_process() |
| 990 | thread.join() |
| 991 | test_output('Done killing other tests') |
Brian Silverman | be6cfe2 | 2014-04-27 08:06:27 -0500 | [diff] [blame] | 992 | |
Brian Silverman | c2d8e5a | 2014-05-01 18:33:12 -0700 | [diff] [blame] | 993 | user_output('Done building %s (%d/%d)' % (platform, num, len(platforms))) |
| 994 | num += 1 |
Brian Silverman | a29ebf9 | 2014-04-23 13:08:49 -0500 | [diff] [blame] | 995 | |
| 996 | if __name__ == '__main__': |
| 997 | main() |