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