blob: aacdb8657cfdcdc51e30e3fc64ccb1bb4bbc7476 [file] [log] [blame]
Brian Harring3fec5a82012-03-01 05:57:03 -08001#!/usr/bin/python
2
Mike Frysingerd6925b52012-07-16 16:11:00 -04003# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Brian Harring3fec5a82012-03-01 05:57:03 -08004# Use of this source code is governed by a BSD-style license that can be
5# found in the LICENSE file.
6
7"""Main builder code for Chromium OS.
8
9Used by Chromium OS buildbot configuration for all Chromium OS builds including
10full and pre-flight-queue builds.
11"""
12
13import distutils.version
Mike Frysinger6903c762012-12-04 01:57:16 -050014import errno
Brian Harring3fec5a82012-03-01 05:57:03 -080015import glob
Chris Sosa4f6ffaf2012-05-01 17:05:44 -070016import logging
Brian Harring3fec5a82012-03-01 05:57:03 -080017import optparse
18import os
19import pprint
20import sys
Ryan Cui54da0702012-04-19 18:38:08 -070021import time
Brian Harring3fec5a82012-03-01 05:57:03 -080022
23from chromite.buildbot import builderstage as bs
Brian Harring3fec5a82012-03-01 05:57:03 -080024from chromite.buildbot import cbuildbot_config
25from chromite.buildbot import cbuildbot_stages as stages
26from chromite.buildbot import cbuildbot_results as results_lib
Brian Harring3fec5a82012-03-01 05:57:03 -080027from chromite.buildbot import constants
Brian Harring3fec5a82012-03-01 05:57:03 -080028from chromite.buildbot import remote_try
29from chromite.buildbot import repository
30from chromite.buildbot import tee
Ryan Cui16d9e1f2012-05-11 10:50:18 -070031from chromite.buildbot import trybot_patch_pool
Brian Harring3fec5a82012-03-01 05:57:03 -080032
Brian Harringc92a7012012-02-29 10:11:34 -080033from chromite.lib import cgroups
Brian Harringa184efa2012-03-04 11:51:25 -080034from chromite.lib import cleanup
Brian Harringb6cf9142012-09-01 20:43:17 -070035from chromite.lib import commandline
Brian Harring1b8c4c82012-05-29 23:03:04 -070036from chromite.lib import cros_build_lib
Brian Harring511055e2012-10-10 02:58:59 -070037from chromite.lib import gerrit
David James97d95872012-11-16 15:09:56 -080038from chromite.lib import git
Brian Harringaf019fb2012-05-10 15:06:13 -070039from chromite.lib import osutils
Brian Harring511055e2012-10-10 02:58:59 -070040from chromite.lib import patch as cros_patch
David James6450a0a2012-12-04 07:59:53 -080041from chromite.lib import parallel
Brian Harring3fec5a82012-03-01 05:57:03 -080042from chromite.lib import sudo
43
Ryan Cuiadd49122012-03-21 22:19:58 -070044
Brian Harring1b8c4c82012-05-29 23:03:04 -070045cros_build_lib.STRICT_SUDO = True
Brian Harring3fec5a82012-03-01 05:57:03 -080046
47_DEFAULT_LOG_DIR = 'cbuildbot_logs'
48_BUILDBOT_LOG_FILE = 'cbuildbot.log'
49_DEFAULT_EXT_BUILDROOT = 'trybot'
50_DEFAULT_INT_BUILDROOT = 'trybot-internal'
Brian Harring3fec5a82012-03-01 05:57:03 -080051_DISTRIBUTED_TYPES = [constants.COMMIT_QUEUE_TYPE, constants.PFQ_TYPE,
52 constants.CANARY_TYPE, constants.CHROME_PFQ_TYPE,
53 constants.PALADIN_TYPE]
Brian Harring351ce442012-03-09 16:38:14 -080054_BUILDBOT_REQUIRED_BINARIES = ('pbzip2',)
Ryan Cui1c13a252012-10-16 15:00:16 -070055_API_VERSION_ATTR = 'api_version'
Brian Harring3fec5a82012-03-01 05:57:03 -080056
57
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070058def _PrintValidConfigs(display_all=False):
Brian Harring3fec5a82012-03-01 05:57:03 -080059 """Print a list of valid buildbot configs.
60
61 Arguments:
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070062 display_all: Print all configs. Otherwise, prints only configs with
63 trybot_list=True.
Brian Harring3fec5a82012-03-01 05:57:03 -080064 """
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070065 def _GetSortKey(config_name):
66 config_dict = cbuildbot_config.config[config_name]
67 return (not config_dict['trybot_list'], config_dict['description'],
68 config_name)
69
Brian Harring3fec5a82012-03-01 05:57:03 -080070 COLUMN_WIDTH = 45
71 print 'config'.ljust(COLUMN_WIDTH), 'description'
72 print '------'.ljust(COLUMN_WIDTH), '-----------'
73 config_names = cbuildbot_config.config.keys()
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070074 config_names.sort(key=_GetSortKey)
Brian Harring3fec5a82012-03-01 05:57:03 -080075 for name in config_names:
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070076 if display_all or cbuildbot_config.config[name]['trybot_list']:
77 desc = cbuildbot_config.config[name].get('description')
78 desc = desc if desc else ''
Brian Harring3fec5a82012-03-01 05:57:03 -080079 print name.ljust(COLUMN_WIDTH), desc
80
81
82def _GetConfig(config_name):
83 """Gets the configuration for the build"""
84 if not cbuildbot_config.config.has_key(config_name):
85 print 'Non-existent configuration %s specified.' % config_name
86 print 'Please specify one of:'
87 _PrintValidConfigs()
88 sys.exit(1)
89
90 result = cbuildbot_config.config[config_name]
91
92 return result
93
94
Ryan Cuie1e4e662012-05-21 16:39:46 -070095def AcquirePoolFromOptions(options):
Ryan Cui16d9e1f2012-05-11 10:50:18 -070096 """Generate patch objects from passed in options.
Brian Harring3fec5a82012-03-01 05:57:03 -080097
98 Args:
Ryan Cui16d9e1f2012-05-11 10:50:18 -070099 options: The options object generated by optparse.
Brian Harring3fec5a82012-03-01 05:57:03 -0800100
Ryan Cuif7f24692012-05-18 16:35:33 -0700101 Returns:
102 trybot_patch_pool.TrybotPatchPool object.
103
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700104 Raises:
Brian Harring511055e2012-10-10 02:58:59 -0700105 gerrit.GerritException, cros_patch.PatchException
Brian Harring3fec5a82012-03-01 05:57:03 -0800106 """
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700107 gerrit_patches = []
108 local_patches = []
109 remote_patches = []
Brian Harring3fec5a82012-03-01 05:57:03 -0800110
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700111 if options.gerrit_patches:
Brian Harring511055e2012-10-10 02:58:59 -0700112 gerrit_patches = gerrit.GetGerritPatchInfo(
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700113 options.gerrit_patches)
114 for patch in gerrit_patches:
115 if patch.IsAlreadyMerged():
Brian Harring1b8c4c82012-05-29 23:03:04 -0700116 cros_build_lib.Warning('Patch %s has already been merged.' % str(patch))
Brian Harring3fec5a82012-03-01 05:57:03 -0800117
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700118 if options.local_patches:
David James97d95872012-11-16 15:09:56 -0800119 manifest = git.ManifestCheckout.Cached(options.sourceroot)
Brian Harring609dc4e2012-05-07 02:17:44 -0700120 local_patches = cros_patch.PrepareLocalPatches(manifest,
121 options.local_patches)
Brian Harring3fec5a82012-03-01 05:57:03 -0800122
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700123 if options.remote_patches:
124 remote_patches = cros_patch.PrepareRemotePatches(
125 options.remote_patches)
Brian Harring3fec5a82012-03-01 05:57:03 -0800126
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700127 return trybot_patch_pool.TrybotPatchPool(gerrit_patches, local_patches,
128 remote_patches)
Brian Harring3fec5a82012-03-01 05:57:03 -0800129
130
Brian Harring3fec5a82012-03-01 05:57:03 -0800131class Builder(object):
132 """Parent class for all builder types.
133
134 This class functions as a parent class for various build types. It's intended
135 use is builder_instance.Run().
136
137 Vars:
Brian Harring3fec5a82012-03-01 05:57:03 -0800138 build_config: The configuration dictionary from cbuildbot_config.
139 options: The options provided from optparse in main().
Ryan Cui5616a512012-08-17 13:39:36 -0700140 archive_urls: Where our artifacts for this builder will be archived.
Brian Harring3fec5a82012-03-01 05:57:03 -0800141 release_tag: The associated "chrome os version" of this build.
Brian Harring3fec5a82012-03-01 05:57:03 -0800142 """
143
Ryan Cuie1e4e662012-05-21 16:39:46 -0700144 def __init__(self, options, build_config):
Brian Harring3fec5a82012-03-01 05:57:03 -0800145 """Initializes instance variables. Must be called by all subclasses."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800146 self.build_config = build_config
147 self.options = options
148
149 # TODO, Remove here and in config after bug chromium-os:14649 is fixed.
150 if self.build_config['chromeos_official']:
151 os.environ['CHROMEOS_OFFICIAL'] = '1'
152
David James58e0c092012-03-04 20:31:12 -0800153 self.archive_stages = {}
Brian Harring3fec5a82012-03-01 05:57:03 -0800154 self.archive_urls = {}
155 self.release_tag = None
Brian Harring76d1bf62012-06-01 13:52:48 -0700156 self.patch_pool = trybot_patch_pool.TrybotPatchPool()
Brian Harring3fec5a82012-03-01 05:57:03 -0800157
Ryan Cuie1e4e662012-05-21 16:39:46 -0700158 bs.BuilderStage.SetManifestBranch(self.options.branch)
Ryan Cuif7f24692012-05-18 16:35:33 -0700159
Brian Harring3fec5a82012-03-01 05:57:03 -0800160 def Initialize(self):
161 """Runs through the initialization steps of an actual build."""
Ryan Cuif7f24692012-05-18 16:35:33 -0700162 if self.options.resume:
163 results_lib.LoadCheckpoint(self.options.buildroot)
Brian Harring3fec5a82012-03-01 05:57:03 -0800164
Brian Harring3fec5a82012-03-01 05:57:03 -0800165 self._RunStage(stages.CleanUpStage)
166
167 def _GetStageInstance(self, stage, *args, **kwargs):
168 """Helper function to get an instance given the args.
169
David James944a48e2012-03-07 12:19:03 -0800170 Useful as almost all stages just take in options and build_config.
Brian Harring3fec5a82012-03-01 05:57:03 -0800171 """
David James944a48e2012-03-07 12:19:03 -0800172 config = kwargs.pop('config', self.build_config)
173 return stage(self.options, config, *args, **kwargs)
Brian Harring3fec5a82012-03-01 05:57:03 -0800174
175 def _SetReleaseTag(self):
176 """Sets the release tag from the manifest_manager.
177
178 Must be run after sync stage as syncing enables us to have a release tag.
179 """
180 # Extract version we have decided to build into self.release_tag.
181 manifest_manager = stages.ManifestVersionedSyncStage.manifest_manager
182 if manifest_manager:
183 self.release_tag = manifest_manager.current_version
184
185 def _RunStage(self, stage, *args, **kwargs):
186 """Wrapper to run a stage."""
187 stage_instance = self._GetStageInstance(stage, *args, **kwargs)
188 return stage_instance.Run()
189
190 def GetSyncInstance(self):
191 """Returns an instance of a SyncStage that should be run.
192
193 Subclasses must override this method.
194 """
195 raise NotImplementedError()
196
197 def RunStages(self):
198 """Subclasses must override this method. Runs the appropriate code."""
199 raise NotImplementedError()
200
Brian Harring3fec5a82012-03-01 05:57:03 -0800201 def _ShouldReExecuteInBuildRoot(self):
202 """Returns True if this build should be re-executed in the buildroot."""
203 abs_buildroot = os.path.abspath(self.options.buildroot)
204 return not os.path.abspath(__file__).startswith(abs_buildroot)
205
206 def _ReExecuteInBuildroot(self, sync_instance):
207 """Reexecutes self in buildroot and returns True if build succeeds.
208
209 This allows the buildbot code to test itself when changes are patched for
210 buildbot-related code. This is a no-op if the buildroot == buildroot
211 of the running chromite checkout.
212
213 Args:
214 sync_instance: Instance of the sync stage that was run to sync.
215
216 Returns:
217 True if the Build succeeded.
218 """
Brian Harring3fec5a82012-03-01 05:57:03 -0800219 if not self.options.resume:
Ryan Cuif7f24692012-05-18 16:35:33 -0700220 results_lib.WriteCheckpoint(self.options.buildroot)
Brian Harring3fec5a82012-03-01 05:57:03 -0800221
Ryan Cui1c13a252012-10-16 15:00:16 -0700222 args = stages.BootstrapStage.FilterArgsForTargetCbuildbot(
223 self.options.buildroot, constants.PATH_TO_CBUILDBOT, self.options)
Brian Harring37e559b2012-05-22 20:47:32 -0700224
Brian Harring3fec5a82012-03-01 05:57:03 -0800225 # Re-write paths to use absolute paths.
226 # Suppress any timeout options given from the commandline in the
227 # invoked cbuildbot; our timeout will enforce it instead.
Ryan Cui1c13a252012-10-16 15:00:16 -0700228 args += ['--resume', '--timeout', '0', '--notee', '--nocgroups',
229 '--buildroot', os.path.abspath(self.options.buildroot)]
Brian Harringae0a5322012-09-15 01:46:51 -0700230
Brian Harring3fec5a82012-03-01 05:57:03 -0800231 if self.options.chrome_root:
Ryan Cui1c13a252012-10-16 15:00:16 -0700232 args += ['--chrome_root',
233 os.path.abspath(self.options.chrome_root)]
Brian Harring3fec5a82012-03-01 05:57:03 -0800234
235 if stages.ManifestVersionedSyncStage.manifest_manager:
236 ver = stages.ManifestVersionedSyncStage.manifest_manager.current_version
Ryan Cui1c13a252012-10-16 15:00:16 -0700237 args += ['--version', ver]
Brian Harring3fec5a82012-03-01 05:57:03 -0800238
239 if isinstance(sync_instance, stages.CommitQueueSyncStage):
240 vp_file = sync_instance.SaveValidationPool()
Ryan Cui1c13a252012-10-16 15:00:16 -0700241 args += ['--validation_pool', vp_file]
Brian Harring3fec5a82012-03-01 05:57:03 -0800242
243 # Re-run the command in the buildroot.
244 # Finally, be generous and give the invoked cbuildbot 30s to shutdown
245 # when something occurs. It should exit quicker, but the sigterm may
246 # hit while the system is particularly busy.
Brian Harring1b8c4c82012-05-29 23:03:04 -0700247 return_obj = cros_build_lib.RunCommand(
Ryan Cui1c13a252012-10-16 15:00:16 -0700248 args, cwd=self.options.buildroot, error_code_ok=True, kill_timeout=30)
Brian Harring3fec5a82012-03-01 05:57:03 -0800249 return return_obj.returncode == 0
250
Ryan Cuif7f24692012-05-18 16:35:33 -0700251 def _InitializeTrybotPatchPool(self):
252 """Generate patch pool from patches specified on the command line.
253
254 Do this only if we need to patch changes later on.
255 """
256 changes_stage = stages.PatchChangesStage.StageNamePrefix()
257 check_func = results_lib.Results.PreviouslyCompletedRecord
258 if not check_func(changes_stage) or self.options.bootstrap:
Ryan Cuie1e4e662012-05-21 16:39:46 -0700259 self.patch_pool = AcquirePoolFromOptions(self.options)
Ryan Cuif7f24692012-05-18 16:35:33 -0700260
261 def _GetBootstrapStage(self):
262 """Constructs and returns the BootStrapStage object.
263
264 We return None when there are no chromite patches to test, and
265 --test-bootstrap wasn't passed in.
266 """
267 stage = None
268 chromite_pool = self.patch_pool.Filter(project=constants.CHROMITE_PROJECT)
Ryan Cui5616a512012-08-17 13:39:36 -0700269 manifest_pool = self.patch_pool.FilterManifest()
David James97d95872012-11-16 15:09:56 -0800270 chromite_branch = git.GetChromiteTrackingBranch()
Ryan Cui5616a512012-08-17 13:39:36 -0700271 if (chromite_pool or manifest_pool or self.options.test_bootstrap
Ryan Cuie1e4e662012-05-21 16:39:46 -0700272 or chromite_branch != self.options.branch):
Ryan Cuif7f24692012-05-18 16:35:33 -0700273 stage = stages.BootstrapStage(self.options, self.build_config,
Ryan Cui5616a512012-08-17 13:39:36 -0700274 chromite_pool, manifest_pool)
Ryan Cuif7f24692012-05-18 16:35:33 -0700275 return stage
276
Brian Harring3fec5a82012-03-01 05:57:03 -0800277 def Run(self):
Ryan Cuif7f24692012-05-18 16:35:33 -0700278 """Main runner for this builder class. Runs build and prints summary.
279
280 Returns:
281 Whether the build succeeded.
282 """
283 self._InitializeTrybotPatchPool()
284
285 if self.options.bootstrap:
286 bootstrap_stage = self._GetBootstrapStage()
287 if bootstrap_stage:
288 # BootstrapStage blocks on re-execution of cbuildbot.
289 bootstrap_stage.Run()
290 return bootstrap_stage.returncode == 0
291
Brian Harring3fec5a82012-03-01 05:57:03 -0800292 print_report = True
David James3d4d3502012-04-09 15:12:06 -0700293 exception_thrown = False
Brian Harring3fec5a82012-03-01 05:57:03 -0800294 success = True
295 try:
296 self.Initialize()
297 sync_instance = self.GetSyncInstance()
298 sync_instance.Run()
299 self._SetReleaseTag()
300
Ryan Cui967f7cc2012-08-17 13:01:12 -0700301 # Filter out patches to manifest, since PatchChangesStage can't handle
302 # them. Manifest patches are patched in the BootstrapStage.
303 non_manifest_patches = self.patch_pool.FilterManifest(negate=True)
304 if non_manifest_patches:
305 self._RunStage(stages.PatchChangesStage, non_manifest_patches)
Brian Harring3fec5a82012-03-01 05:57:03 -0800306
307 if self._ShouldReExecuteInBuildRoot():
308 print_report = False
309 success = self._ReExecuteInBuildroot(sync_instance)
310 else:
311 self.RunStages()
David James7fbf2d42012-07-14 18:23:49 -0700312 except results_lib.StepFailure:
313 # StepFailure exceptions are already recorded in the report, so there
314 # is no need to print these tracebacks twice.
315 exception_thrown = True
316 if not print_report:
317 raise
David James3d4d3502012-04-09 15:12:06 -0700318 except Exception:
319 exception_thrown = True
320 raise
Brian Harring3fec5a82012-03-01 05:57:03 -0800321 finally:
322 if print_report:
Ryan Cuif7f24692012-05-18 16:35:33 -0700323 results_lib.WriteCheckpoint(self.options.buildroot)
Brian Harring3fec5a82012-03-01 05:57:03 -0800324 print '\n\n\n@@@BUILD_STEP Report@@@\n'
325 results_lib.Results.Report(sys.stdout, self.archive_urls,
326 self.release_tag)
327 success = results_lib.Results.BuildSucceededSoFar()
David James3d4d3502012-04-09 15:12:06 -0700328 if exception_thrown and success:
329 success = False
David Jamesbb20ac82012-07-18 10:59:16 -0700330 cros_build_lib.PrintBuildbotStepWarnings()
331 print """\
David James3d4d3502012-04-09 15:12:06 -0700332Exception thrown, but all stages marked successful. This is an internal error,
333because the stage that threw the exception should be marked as failing."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800334
335 return success
336
337
338class SimpleBuilder(Builder):
339 """Builder that performs basic vetting operations."""
340
341 def GetSyncInstance(self):
342 """Sync to lkgm or TOT as necessary.
343
344 Returns: the instance of the sync stage that was run.
345 """
Chris Sosa52a81b62012-11-14 06:12:54 -0800346 if self.build_config['use_lkgm']:
Brian Harring3fec5a82012-03-01 05:57:03 -0800347 sync_stage = self._GetStageInstance(stages.LKGMSyncStage)
Chris Sosa52a81b62012-11-14 06:12:54 -0800348 elif self.build_config['use_chrome_lkgm']:
349 sync_stage = self._GetStageInstance(stages.ChromeLKGMSyncStage)
Brian Harring3fec5a82012-03-01 05:57:03 -0800350 else:
351 sync_stage = self._GetStageInstance(stages.SyncStage)
352
353 return sync_stage
354
David James58e0c092012-03-04 20:31:12 -0800355 def _RunBackgroundStagesForBoard(self, board):
356 """Run background board-specific stages for the specified board."""
David James58e0c092012-03-04 20:31:12 -0800357 archive_stage = self.archive_stages[board]
David James944a48e2012-03-07 12:19:03 -0800358 configs = self.build_config['board_specific_configs']
359 config = configs.get(board, self.build_config)
360 stage_list = [[stages.VMTestStage, board, archive_stage],
Mike Frysinger5e20ec42012-09-28 23:39:56 -0400361 [stages.SignerTestStage, board, archive_stage],
David James944a48e2012-03-07 12:19:03 -0800362 [stages.UnitTestStage, board],
Chris Sosa6ed77e52012-10-22 12:57:31 -0700363 [stages.UploadPrebuiltsStage, board, archive_stage],
364 [stages.DevInstallerPrebuiltsStage, board, archive_stage]]
Brian Harring3fec5a82012-03-01 05:57:03 -0800365
David James58e0c092012-03-04 20:31:12 -0800366 # We can not run hw tests without archiving the payloads.
367 if self.options.archive:
David James944a48e2012-03-07 12:19:03 -0800368 for suite in config['hw_tests']:
Peter Mayo85e94372012-08-17 12:42:14 -0400369 stage_list.append([stages.HWTestStage, board, archive_stage, suite])
Chris Sosab50dc932012-03-01 14:00:58 -0800370
Chris Sosa817b1f92012-07-19 15:00:23 -0700371 for suite in config['async_hw_tests']:
372 stage_list.append([stages.ASyncHWTestStage, board, archive_stage,
373 suite])
374
David James944a48e2012-03-07 12:19:03 -0800375 steps = [self._GetStageInstance(*x, config=config).Run for x in stage_list]
David James6450a0a2012-12-04 07:59:53 -0800376 parallel.RunParallelSteps(steps + [archive_stage.Run])
Brian Harring3fec5a82012-03-01 05:57:03 -0800377
378 def RunStages(self):
379 """Runs through build process."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800380 # TODO(sosa): Split these out into classes.
Brian Harring3fec5a82012-03-01 05:57:03 -0800381 if self.build_config['build_type'] == constants.CHROOT_BUILDER_TYPE:
Mike Frysingerfddaeb52012-11-20 11:17:31 -0500382 self._RunStage(stages.UprevStage, boards=[], enter_chroot=False)
Mike Frysinger0fe33322012-10-31 22:02:49 -0400383 self._RunStage(stages.BuildBoardStage, [constants.CHROOT_BUILDER_BOARD])
Zdenek Behan62a57792012-08-31 15:09:08 +0200384 self._RunStage(stages.SDKPackageStage)
Brian Harring3fec5a82012-03-01 05:57:03 -0800385 self._RunStage(stages.SDKTestStage)
386 self._RunStage(stages.UploadPrebuiltsStage,
Chris Sosa6a5dceb2012-05-14 13:48:56 -0700387 constants.CHROOT_BUILDER_BOARD, None)
Brian Harring3fec5a82012-03-01 05:57:03 -0800388 elif self.build_config['build_type'] == constants.REFRESH_PACKAGES_TYPE:
Mike Frysinger0fe33322012-10-31 22:02:49 -0400389 self._RunStage(stages.BuildBoardStage)
Brian Harring3fec5a82012-03-01 05:57:03 -0800390 self._RunStage(stages.RefreshPackageStatusStage)
391 else:
Mike Frysinger0fe33322012-10-31 22:02:49 -0400392 self._RunStage(stages.BuildBoardStage)
Brian Harring3fec5a82012-03-01 05:57:03 -0800393 self._RunStage(stages.UprevStage)
Brian Harring3fec5a82012-03-01 05:57:03 -0800394
David James944a48e2012-03-07 12:19:03 -0800395 configs = self.build_config['board_specific_configs']
David James58e0c092012-03-04 20:31:12 -0800396 for board in self.build_config['boards']:
David James944a48e2012-03-07 12:19:03 -0800397 config = configs.get(board, self.build_config)
398 archive_stage = self._GetStageInstance(stages.ArchiveStage, board,
399 config=config)
David James58e0c092012-03-04 20:31:12 -0800400 self.archive_stages[board] = archive_stage
401
David James944a48e2012-03-07 12:19:03 -0800402 # Set up a process pool to run test/archive stages in the background.
403 # This process runs task(board) for each board added to the queue.
David James58e0c092012-03-04 20:31:12 -0800404 task = self._RunBackgroundStagesForBoard
Brian Harring9fdd23d2012-12-07 12:09:08 -0800405 with parallel.BackgroundTaskRunner(task) as queue:
David James944a48e2012-03-07 12:19:03 -0800406 for board in self.build_config['boards']:
David James58e0c092012-03-04 20:31:12 -0800407 # Run BuildTarget in the foreground.
David James944a48e2012-03-07 12:19:03 -0800408 archive_stage = self.archive_stages[board]
409 config = configs.get(board, self.build_config)
410 self._RunStage(stages.BuildTargetStage, board, archive_stage,
Chris Sosa1a87b3e2012-04-12 13:20:42 -0700411 self.release_tag, config=config)
David James58e0c092012-03-04 20:31:12 -0800412 self.archive_urls[board] = archive_stage.GetDownloadUrl()
413
David James944a48e2012-03-07 12:19:03 -0800414 # Kick off task(board) in the background.
David James58e0c092012-03-04 20:31:12 -0800415 queue.put([board])
416
Brian Harring3fec5a82012-03-01 05:57:03 -0800417
418class DistributedBuilder(SimpleBuilder):
419 """Build class that has special logic to handle distributed builds.
420
421 These builds sync using git/manifest logic in manifest_versions. In general
422 they use a non-distributed builder code for the bulk of the work.
423 """
Ryan Cuif7f24692012-05-18 16:35:33 -0700424 def __init__(self, *args, **kwargs):
Brian Harring3fec5a82012-03-01 05:57:03 -0800425 """Initializes a buildbot builder.
426
427 Extra variables:
428 completion_stage_class: Stage used to complete a build. Set in the Sync
429 stage.
430 """
Ryan Cuif7f24692012-05-18 16:35:33 -0700431 super(DistributedBuilder, self).__init__(*args, **kwargs)
Brian Harring3fec5a82012-03-01 05:57:03 -0800432 self.completion_stage_class = None
433
434 def GetSyncInstance(self):
435 """Syncs the tree using one of the distributed sync logic paths.
436
437 Returns: the instance of the sync stage that was run.
438 """
439 # Determine sync class to use. CQ overrides PFQ bits so should check it
440 # first.
441 if cbuildbot_config.IsCQType(self.build_config['build_type']):
442 sync_stage = self._GetStageInstance(stages.CommitQueueSyncStage)
443 self.completion_stage_class = stages.CommitQueueCompletionStage
444 elif cbuildbot_config.IsPFQType(self.build_config['build_type']):
445 sync_stage = self._GetStageInstance(stages.LKGMCandidateSyncStage)
446 self.completion_stage_class = stages.LKGMCandidateSyncCompletionStage
447 else:
448 sync_stage = self._GetStageInstance(stages.ManifestVersionedSyncStage)
449 self.completion_stage_class = stages.ManifestVersionedSyncCompletionStage
450
451 return sync_stage
452
453 def Publish(self, was_build_successful):
454 """Completes build by publishing any required information."""
455 completion_stage = self._GetStageInstance(self.completion_stage_class,
456 was_build_successful)
457 completion_stage.Run()
458 name = completion_stage.name
459 if not results_lib.Results.WasStageSuccessful(name):
460 should_publish_changes = False
461 else:
462 should_publish_changes = (self.build_config['master'] and
463 was_build_successful)
464
465 if should_publish_changes:
466 self._RunStage(stages.PublishUprevChangesStage)
467
468 def RunStages(self):
469 """Runs simple builder logic and publishes information to overlays."""
470 was_build_successful = False
471 try:
David Jamesf55709e2012-03-13 09:10:15 -0700472 super(DistributedBuilder, self).RunStages()
473 was_build_successful = results_lib.Results.BuildSucceededSoFar()
Brian Harring3fec5a82012-03-01 05:57:03 -0800474 except SystemExit as ex:
475 # If a stage calls sys.exit(0), it's exiting with success, so that means
476 # we should mark ourselves as successful.
477 if ex.code == 0:
478 was_build_successful = True
479 raise
480 finally:
481 self.Publish(was_build_successful)
482
Brian Harring3fec5a82012-03-01 05:57:03 -0800483
484def _ConfirmBuildRoot(buildroot):
485 """Confirm with user the inferred buildroot, and mark it as confirmed."""
Brian Harring521e7242012-11-01 16:57:42 -0700486 cros_build_lib.Warning('Using default directory %s as buildroot', buildroot)
487 if not cros_build_lib.BooleanPrompt(default=False):
488 print('Please specify a different buildroot via the --buildroot option.')
Brian Harring3fec5a82012-03-01 05:57:03 -0800489 sys.exit(0)
490
491 if not os.path.exists(buildroot):
492 os.mkdir(buildroot)
493
494 repository.CreateTrybotMarker(buildroot)
495
496
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700497def _ConfirmRemoteBuildbotRun():
498 """Confirm user wants to run with --buildbot --remote."""
Brian Harring521e7242012-11-01 16:57:42 -0700499 cros_build_lib.Warning(
500 'You are about to launch a PRODUCTION job! This is *NOT* a '
501 'trybot run! Are you sure?')
502 if not cros_build_lib.BooleanPrompt(default=False):
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700503 print('Please specify --pass-through="--debug".')
504 sys.exit(0)
505
506
Ryan Cui5ba7e152012-05-10 14:36:52 -0700507def _DetermineDefaultBuildRoot(sourceroot, internal_build):
Brian Harring3fec5a82012-03-01 05:57:03 -0800508 """Default buildroot to be under the directory that contains current checkout.
509
510 Arguments:
511 internal_build: Whether the build is an internal build
Ryan Cui5ba7e152012-05-10 14:36:52 -0700512 sourceroot: Use specified sourceroot.
Brian Harring3fec5a82012-03-01 05:57:03 -0800513 """
Ryan Cui5ba7e152012-05-10 14:36:52 -0700514 if not repository.IsARepoRoot(sourceroot):
Brian Harring1b8c4c82012-05-29 23:03:04 -0700515 cros_build_lib.Die(
516 'Could not find root of local checkout at %s. Please specify '
517 'using the --sourceroot option.' % sourceroot)
Brian Harring3fec5a82012-03-01 05:57:03 -0800518
519 # Place trybot buildroot under the directory containing current checkout.
Ryan Cui5ba7e152012-05-10 14:36:52 -0700520 top_level = os.path.dirname(os.path.realpath(sourceroot))
Brian Harring3fec5a82012-03-01 05:57:03 -0800521 if internal_build:
522 buildroot = os.path.join(top_level, _DEFAULT_INT_BUILDROOT)
523 else:
524 buildroot = os.path.join(top_level, _DEFAULT_EXT_BUILDROOT)
525
526 return buildroot
527
528
Mike Frysinger6903c762012-12-04 01:57:16 -0500529def _DisableYamaHardLinkChecks():
530 """Disable Yama kernel hardlink security checks.
531
532 The security module disables hardlinking to files you do not have
533 write access to which causes some of our build scripts problems.
534 Disable it so we don't have to worry about it.
535 """
536 PROC_PATH = '/proc/sys/kernel/yama/protected_nonaccess_hardlinks'
537 SYSCTL_PATH = PROC_PATH[len('/proc/sys/'):].replace('/', '.')
538
539 # Yama not available in this system -- nothing to do.
540 if not os.path.exists(PROC_PATH):
541 return
542
543 # Already disabled -- nothing to do.
544 if osutils.ReadFile(PROC_PATH).strip() == '0':
545 return
546
547 # Create a hardlink in a tempdir and see if we get back EPERM.
548 with osutils.TempDirContextManager() as tempdir:
549 try:
550 os.link('/bin/sh', os.path.join(tempdir, 'sh'))
551 except OSError as e:
552 if e.errno == errno.EPERM:
553 cros_build_lib.Warning('Disabling Yama hardlink security')
554 cros_build_lib.SudoRunCommand(['sysctl', '%s=0' % SYSCTL_PATH])
555
556
Brian Harring3fec5a82012-03-01 05:57:03 -0800557def _BackupPreviousLog(log_file, backup_limit=25):
558 """Rename previous log.
559
560 Args:
561 log_file: The absolute path to the previous log.
562 """
563 if os.path.exists(log_file):
564 old_logs = sorted(glob.glob(log_file + '.*'),
565 key=distutils.version.LooseVersion)
566
567 if len(old_logs) >= backup_limit:
568 os.remove(old_logs[0])
569
570 last = 0
571 if old_logs:
572 last = int(old_logs.pop().rpartition('.')[2])
573
574 os.rename(log_file, log_file + '.' + str(last + 1))
575
Ryan Cui5616a512012-08-17 13:39:36 -0700576
David James944a48e2012-03-07 12:19:03 -0800577def _RunBuildStagesWrapper(options, build_config):
Brian Harring3fec5a82012-03-01 05:57:03 -0800578 """Helper function that wraps RunBuildStages()."""
579 def IsDistributedBuilder():
580 """Determines whether the build_config should be a DistributedBuilder."""
581 if not options.buildbot:
582 return False
583 elif build_config['build_type'] in _DISTRIBUTED_TYPES:
584 chrome_rev = build_config['chrome_rev']
Chris Sosa52a81b62012-11-14 06:12:54 -0800585 if options.chrome_rev:
586 chrome_rev = options.chrome_rev
587
Brian Harring3fec5a82012-03-01 05:57:03 -0800588 # We don't do distributed logic to TOT Chrome PFQ's, nor local
589 # chrome roots (e.g. chrome try bots)
590 if chrome_rev not in [constants.CHROME_REV_TOT,
591 constants.CHROME_REV_LOCAL,
592 constants.CHROME_REV_SPEC]:
593 return True
594
595 return False
596
Brian Harring1b8c4c82012-05-29 23:03:04 -0700597 cros_build_lib.Info("cbuildbot executed with args %s"
598 % ' '.join(map(repr, sys.argv)))
Brian Harring3fec5a82012-03-01 05:57:03 -0800599
Ryan Cuif7f24692012-05-18 16:35:33 -0700600 target = DistributedBuilder if IsDistributedBuilder() else SimpleBuilder
Ryan Cuie1e4e662012-05-21 16:39:46 -0700601 buildbot = target(options, build_config)
Brian Harringd166aaf2012-05-14 18:31:53 -0700602 if not buildbot.Run():
603 sys.exit(1)
Brian Harring3fec5a82012-03-01 05:57:03 -0800604
605
606# Parser related functions
Ryan Cui5ba7e152012-05-10 14:36:52 -0700607def _CheckLocalPatches(sourceroot, local_patches):
Brian Harring3fec5a82012-03-01 05:57:03 -0800608 """Do an early quick check of the passed-in patches.
609
610 If the branch of a project is not specified we append the current branch the
611 project is on.
Ryan Cui5ba7e152012-05-10 14:36:52 -0700612
613 Args:
614 sourceroot: The checkout where patches are coming from.
Brian Harring3fec5a82012-03-01 05:57:03 -0800615 """
Ryan Cuicedd8a52012-03-22 02:28:35 -0700616 verified_patches = []
David James97d95872012-11-16 15:09:56 -0800617 manifest = git.ManifestCheckout.Cached(sourceroot)
Ryan Cuicedd8a52012-03-22 02:28:35 -0700618 for patch in local_patches:
Brian Harring3fec5a82012-03-01 05:57:03 -0800619 components = patch.split(':')
620 if len(components) > 2:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700621 cros_build_lib.Die(
622 'Specify local patches in project[:branch] format. Got %s' % patch)
Brian Harring3fec5a82012-03-01 05:57:03 -0800623
624 # validate project
625 project = components[0]
Brian Harring3fec5a82012-03-01 05:57:03 -0800626
Brian Harring609dc4e2012-05-07 02:17:44 -0700627 try:
628 project_dir = manifest.GetProjectPath(project, True)
629 except KeyError:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700630 cros_build_lib.Die('Project %s does not exist.' % project)
Brian Harring3fec5a82012-03-01 05:57:03 -0800631
632 # If no branch was specified, we use the project's current branch.
633 if len(components) == 1:
David James97d95872012-11-16 15:09:56 -0800634 branch = git.GetCurrentBranch(project_dir)
Brian Harring3fec5a82012-03-01 05:57:03 -0800635 if not branch:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700636 cros_build_lib.Die('Project %s is not on a branch!' % project)
Brian Harring3fec5a82012-03-01 05:57:03 -0800637 else:
638 branch = components[1]
David James97d95872012-11-16 15:09:56 -0800639 if not git.DoesLocalBranchExist(project_dir, branch):
Brian Harring1b8c4c82012-05-29 23:03:04 -0700640 cros_build_lib.Die('Project %s does not have branch %s'
641 % (project, branch))
Brian Harring3fec5a82012-03-01 05:57:03 -0800642
Brian Harring609dc4e2012-05-07 02:17:44 -0700643 verified_patches.append('%s:%s' % (project, branch))
Brian Harring3fec5a82012-03-01 05:57:03 -0800644
Ryan Cuicedd8a52012-03-22 02:28:35 -0700645 return verified_patches
Brian Harring3fec5a82012-03-01 05:57:03 -0800646
647
Brian Harring3fec5a82012-03-01 05:57:03 -0800648def _CheckChromeVersionOption(_option, _opt_str, value, parser):
649 """Upgrade other options based on chrome_version being passed."""
650 value = value.strip()
651
652 if parser.values.chrome_rev is None and value:
653 parser.values.chrome_rev = constants.CHROME_REV_SPEC
654
655 parser.values.chrome_version = value
656
657
658def _CheckChromeRootOption(_option, _opt_str, value, parser):
659 """Validate and convert chrome_root to full-path form."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800660 if parser.values.chrome_rev is None:
661 parser.values.chrome_rev = constants.CHROME_REV_LOCAL
662
Ryan Cui5ba7e152012-05-10 14:36:52 -0700663 parser.values.chrome_root = value
Brian Harring3fec5a82012-03-01 05:57:03 -0800664
665
666def _CheckChromeRevOption(_option, _opt_str, value, parser):
667 """Validate the chrome_rev option."""
668 value = value.strip()
669 if value not in constants.VALID_CHROME_REVISIONS:
670 raise optparse.OptionValueError('Invalid chrome rev specified')
671
672 parser.values.chrome_rev = value
673
674
Brian Harringfec89fe2012-09-23 07:30:54 -0700675def _CheckGerritChromeOption(_option, _opt_str, _value, parser):
Ryan Cuif37608e2012-07-10 14:28:48 -0700676 """Validate the chrome_rev option."""
677 if parser.values.chrome_rev is None:
678 parser.values.chrome_rev = constants.CHROME_REV_TOT
679
680 parser.values.gerrit_chrome = True
681
682
Brian Harringae0a5322012-09-15 01:46:51 -0700683def FindCacheDir(parser, options):
684 if constants.SHARED_CACHE_ENVVAR in os.environ:
685 return commandline.OptionParser.FindCacheDir(parser, options)
686 return None
687
688
Ryan Cui5ba7e152012-05-10 14:36:52 -0700689class CustomGroup(optparse.OptionGroup):
690 def add_remote_option(self, *args, **kwargs):
691 """For arguments that are passed-through to remote trybot."""
692 return optparse.OptionGroup.add_option(self, *args,
693 remote_pass_through=True,
694 **kwargs)
695
696
Ryan Cui1c13a252012-10-16 15:00:16 -0700697class CustomOption(commandline.FilteringOption):
698 """Subclass FilteringOption class to implement pass-through and api."""
Ryan Cui5ba7e152012-05-10 14:36:52 -0700699
Ryan Cui1c13a252012-10-16 15:00:16 -0700700 ACTIONS = commandline.FilteringOption.ACTIONS + ('extend',)
701 STORE_ACTIONS = commandline.FilteringOption.STORE_ACTIONS + ('extend',)
702 TYPED_ACTIONS = commandline.FilteringOption.TYPED_ACTIONS + ('extend',)
703 ALWAYS_TYPED_ACTIONS = (commandline.FilteringOption.ALWAYS_TYPED_ACTIONS +
704 ('extend',))
Ryan Cui79319ab2012-05-21 12:59:18 -0700705
Ryan Cui5ba7e152012-05-10 14:36:52 -0700706 def __init__(self, *args, **kwargs):
707 # The remote_pass_through argument specifies whether we should directly
708 # pass the argument (with its value) onto the remote trybot.
709 self.pass_through = kwargs.pop('remote_pass_through', False)
Ryan Cui1c13a252012-10-16 15:00:16 -0700710 self.api_version = int(kwargs.pop('api', '0'))
711 commandline.FilteringOption.__init__(self, *args, **kwargs)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700712
713 def take_action(self, action, dest, opt, value, values, parser):
Ryan Cui79319ab2012-05-21 12:59:18 -0700714 if action == 'extend':
Mike Frysingerd6925b52012-07-16 16:11:00 -0400715 # If there is extra spaces between each argument, we get '' which later
716 # code barfs on, so skip those. e.g. We see this with the forms:
717 # cbuildbot -p 'proj:branch ' ...
718 # cbuildbot -p ' proj:branch' ...
719 # cbuildbot -p 'proj:branch proj2:branch' ...
720 lvalue = value.split()
Ryan Cui79319ab2012-05-21 12:59:18 -0700721 values.ensure_value(dest, []).extend(lvalue)
Ryan Cui79319ab2012-05-21 12:59:18 -0700722
Ryan Cui1c13a252012-10-16 15:00:16 -0700723 commandline.FilteringOption.take_action(
724 self, action, dest, opt, value, values, parser)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700725
726
Ryan Cui1c13a252012-10-16 15:00:16 -0700727class CustomParser(commandline.FilteringParser):
Brian Harringb6cf9142012-09-01 20:43:17 -0700728
729 DEFAULT_OPTION_CLASS = CustomOption
730
731 def add_remote_option(self, *args, **kwargs):
732 """For arguments that are passed-through to remote trybot."""
Ryan Cui1c13a252012-10-16 15:00:16 -0700733 return self.add_option(*args, remote_pass_through=True, **kwargs)
Brian Harringb6cf9142012-09-01 20:43:17 -0700734
735
Brian Harring3fec5a82012-03-01 05:57:03 -0800736def _CreateParser():
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700737 """Generate and return the parser with all the options."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800738 # Parse options
739 usage = "usage: %prog [options] buildbot_config"
Brian Harringae0a5322012-09-15 01:46:51 -0700740 parser = CustomParser(usage=usage, caching=FindCacheDir)
Brian Harring3fec5a82012-03-01 05:57:03 -0800741
742 # Main options
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700743 # The remote_pass_through parameter to add_option is implemented by the
744 # CustomOption class. See CustomOption for more information.
Brian Harring3fec5a82012-03-01 05:57:03 -0800745 parser.add_option('-a', '--all', action='store_true', dest='print_all',
746 default=False,
747 help=('List all of the buildbot configs available. Use '
748 'with the --list option'))
Ryan Cuie1e4e662012-05-21 16:39:46 -0700749 parser.add_remote_option('-b', '--branch',
750 help='The manifest branch to test. The branch to '
751 'check the buildroot out to.')
Ryan Cui5ba7e152012-05-10 14:36:52 -0700752 parser.add_option('-r', '--buildroot', dest='buildroot', type='path',
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700753 help='Root directory where source is checked out to, and '
754 'where the build occurs. For external build configs, '
755 "defaults to 'trybot' directory at top level of your "
756 'repo-managed checkout.')
757 parser.add_remote_option('--chrome_rev', default=None, type='string',
758 action='callback', dest='chrome_rev',
759 callback=_CheckChromeRevOption,
760 help=('Revision of Chrome to use, of type [%s]'
761 % '|'.join(constants.VALID_CHROME_REVISIONS)))
Ryan Cui79319ab2012-05-21 12:59:18 -0700762 parser.add_remote_option('-g', '--gerrit-patches', action='extend',
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700763 default=[], type='string',
764 metavar="'Id1 *int_Id2...IdN'",
765 help=("Space-separated list of short-form Gerrit "
766 "Change-Id's or change numbers to patch. "
767 "Please prepend '*' to internal Change-Id's"))
Brian Harring3fec5a82012-03-01 05:57:03 -0800768 parser.add_option('-l', '--list', action='store_true', dest='list',
769 default=False,
770 help=('List the suggested trybot configs to use. Use '
771 '--all to list all of the available configs.'))
Ryan Cui54da0702012-04-19 18:38:08 -0700772 parser.add_option('--local', default=False, action='store_true',
773 help=('Specifies that this tryjob should be run locally.'))
Ryan Cui79319ab2012-05-21 12:59:18 -0700774 parser.add_option('-p', '--local-patches', action='extend', default=[],
Brian Harring3fec5a82012-03-01 05:57:03 -0800775 metavar="'<project1>[:<branch1>]...<projectN>[:<branchN>]'",
776 help=('Space-separated list of project branches with '
777 'patches to apply. Projects are specified by name. '
778 'If no branch is specified the current branch of the '
779 'project will be used.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700780 parser.add_remote_option('--profile', default=None, type='string',
781 action='store', dest='profile',
782 help='Name of profile to sub-specify board variant.')
Brian Harring3fec5a82012-03-01 05:57:03 -0800783 parser.add_option('--remote', default=False, action='store_true',
Brian Harring3fec5a82012-03-01 05:57:03 -0800784 help=('Specifies that this tryjob should be run remotely.'))
Brian Harring219a2b82012-07-18 15:30:12 -0700785 parser.add_option('--remote-description', default=None,
786 help=('Attach an optional description to a --remote run '
787 'to make it easier to identify the results when it '
788 'finishes.'))
Brian Harring3fec5a82012-03-01 05:57:03 -0800789
Ryan Cuif4f84be2012-07-09 18:50:41 -0700790 #
Brian Harring3fec5a82012-03-01 05:57:03 -0800791 # Advanced options
Ryan Cuif4f84be2012-07-09 18:50:41 -0700792 #
793
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700794 group = CustomGroup(
Brian Harring3fec5a82012-03-01 05:57:03 -0800795 parser,
796 'Advanced Options',
797 'Caution: use these options at your own risk.')
798
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700799 group.add_remote_option('--buildbot', dest='buildbot', action='store_true',
800 default=False, help='This is running on a buildbot')
801 group.add_remote_option('--buildnumber', help='build number', type='int',
802 default=0)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700803 group.add_option('--chrome_root', default=None, type='path',
804 action='callback', callback=_CheckChromeRootOption,
805 dest='chrome_root', help='Local checkout of Chrome to use.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700806 group.add_remote_option('--chrome_version', default=None, type='string',
807 action='callback', dest='chrome_version',
808 callback=_CheckChromeVersionOption,
809 help='Used with SPEC logic to force a particular SVN '
810 'revision of chrome rather than the latest.')
811 group.add_remote_option('--clobber', action='store_true', dest='clobber',
812 default=False,
813 help='Clears an old checkout before syncing')
Yu-Ju Hong52134292012-06-28 12:50:42 -0700814 group.add_remote_option('--hwtest', dest='hwtest', action='store_true',
815 default=False,
816 help='This adds HW test for remote trybot')
Ryan Cui5ba7e152012-05-10 14:36:52 -0700817 parser.add_option('--log_dir', dest='log_dir', type='path',
Brian Harring3fec5a82012-03-01 05:57:03 -0800818 help=('Directory where logs are stored.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700819 group.add_remote_option('--maxarchives', dest='max_archive_builds',
820 default=3, type='int',
821 help="Change the local saved build count limit.")
Ryan Cuibbd3d4b2012-08-17 12:20:37 -0700822 parser.add_remote_option('--manifest-repo-url',
823 help=('Overrides the default manifest repo url.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700824 group.add_remote_option('--noarchive', action='store_false', dest='archive',
825 default=True, help="Don't run archive stage.")
Ryan Cuif7f24692012-05-18 16:35:33 -0700826 group.add_remote_option('--nobootstrap', action='store_false',
827 dest='bootstrap', default=True,
828 help="Don't checkout and run from a standalone "
829 "chromite repo.")
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700830 group.add_remote_option('--nobuild', action='store_false', dest='build',
831 default=True,
832 help="Don't actually build (for cbuildbot dev)")
833 group.add_remote_option('--noclean', action='store_false', dest='clean',
834 default=True, help="Don't clean the buildroot")
Ryan Cuif7f24692012-05-18 16:35:33 -0700835 group.add_remote_option('--nocgroups', action='store_false', dest='cgroups',
836 default=True,
837 help='Disable cbuildbots usage of cgroups.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700838 group.add_remote_option('--noprebuilts', action='store_false',
839 dest='prebuilts', default=True,
840 help="Don't upload prebuilts.")
841 group.add_remote_option('--nosync', action='store_false', dest='sync',
842 default=True, help="Don't sync before building.")
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700843 group.add_remote_option('--notests', action='store_false', dest='tests',
844 default=True,
845 help='Override values from buildconfig and run no '
846 'tests.')
847 group.add_remote_option('--nouprev', action='store_false', dest='uprev',
848 default=True,
849 help='Override values from buildconfig and never '
850 'uprev.')
Brian Harring3fec5a82012-03-01 05:57:03 -0800851 group.add_option('--reference-repo', action='store', default=None,
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700852 dest='reference_repo',
853 help='Reuse git data stored in an existing repo '
854 'checkout. This can drastically reduce the network '
855 'time spent setting up the trybot checkout. By '
856 "default, if this option isn't given but cbuildbot "
857 'is invoked from a repo checkout, cbuildbot will '
858 'use the repo root.')
Ryan Cuicedd8a52012-03-22 02:28:35 -0700859 group.add_option('--resume', action='store_true', default=False,
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700860 help='Skip stages already successfully completed.')
861 group.add_remote_option('--timeout', action='store', type='int', default=0,
862 help='Specify the maximum amount of time this job '
863 'can run for, at which point the build will be '
864 'aborted. If set to zero, then there is no '
865 'timeout.')
Ryan Cui39bdbbf2012-02-29 16:15:39 -0800866 group.add_option('--test-tryjob', action='store_true',
867 default=False,
868 help='Submit a tryjob to the test repository. Will not '
869 'show up on the production trybot waterfall.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700870 group.add_remote_option('--validation_pool', default=None,
871 help='Path to a pickled validation pool. Intended '
872 'for use only with the commit queue.')
873 group.add_remote_option('--version', dest='force_version', default=None,
874 help='Used with manifest logic. Forces use of this '
875 'version rather than create or get latest.')
Brian Harringab8fb5c2012-07-18 11:28:22 -0700876 group.add_remote_option('--cq-gerrit-query', dest='cq_gerrit_override',
877 default=None,
878 help=
879 "If given, this gerrit query will be used to find what patches to test, "
Brian Harringc5cabfe2012-12-19 01:04:44 -0800880 "rather than the normal 'CommitQueue=1 AND Verified=1 AND CodeReview=2' "
Brian Harringab8fb5c2012-07-18 11:28:22 -0700881 "query it defaults to. Use with care- note additionally this setting "
882 "only has an effect if the buildbot target is a cq target, and we're "
883 "in buildbot mode.")
Brian Harring3fec5a82012-03-01 05:57:03 -0800884
885 parser.add_option_group(group)
886
Ryan Cuif4f84be2012-07-09 18:50:41 -0700887 #
888 # Hidden options.
889 #
890
891 # The base GS URL (gs://<bucket_name>/<path>) to archive artifacts to.
892 parser.add_remote_option('--archive-base', type='gs_path',
893 help=optparse.SUPPRESS_HELP)
894 # bootstrap-args are not verified by the bootstrap code. It gets passed
895 # direcly to the bootstrap re-execution.
896 parser.add_remote_option('--bootstrap-args', action='append',
897 default=[], help=optparse.SUPPRESS_HELP)
Ryan Cuif37608e2012-07-10 14:28:48 -0700898 # Specify to use Gerrit Source for building Chrome. Implies
899 # --chrome_rev=CHROME_REV_TOT.
900 parser.add_option('--gerrit-chrome', action='callback', default=False,
901 callback=_CheckGerritChromeOption, dest='gerrit_chrome',
902 help=optparse.SUPPRESS_HELP)
Ryan Cuif4f84be2012-07-09 18:50:41 -0700903 parser.add_option('--pass-through', dest='pass_through_args', action='append',
904 type='string', default=[], help=optparse.SUPPRESS_HELP)
905 # Used for handling forwards/backwards compatibility for --resume and
906 # --bootstrap.
907 parser.add_option('--reexec-api-version', dest='output_api_version',
908 action='store_true', default=False,
909 help=optparse.SUPPRESS_HELP)
910 # Indicates this is running on a remote trybot machine.
911 parser.add_option('--remote-trybot', dest='remote_trybot',
912 action='store_true', default=False,
913 help=optparse.SUPPRESS_HELP)
914 # Patches uploaded by trybot client when run using the -p option.
915 parser.add_remote_option('--remote-patches', action='extend', default=[],
916 help=optparse.SUPPRESS_HELP)
917 # Specify specific remote tryslaves to run on.
918 parser.add_option('--slaves', action='extend', default=[],
919 help=optparse.SUPPRESS_HELP)
920 parser.add_option('--sourceroot', type='path', default=constants.SOURCE_ROOT,
921 help=optparse.SUPPRESS_HELP)
922 # Causes cbuildbot to bootstrap itself twice, in the sequence A->B->C.
923 # A(unpatched) patches and bootstraps B. B patches and bootstraps C.
924 parser.add_remote_option('--test-bootstrap', action='store_true',
925 default=False, help=optparse.SUPPRESS_HELP)
Brian Harringf611e6e2012-07-17 18:47:44 -0700926 # Note the default here needs to be hardcoded to 3; that is the last version
927 # that lacked this functionality.
928 # This is used so that cbuildbot when processing tryjobs from
929 # older chromite instances, we can use it for handling compatibility.
930 parser.add_option('--remote-version', default=3, type=int, action='store',
931 help=optparse.SUPPRESS_HELP)
Ryan Cuif4f84be2012-07-09 18:50:41 -0700932
933 #
Brian Harring3fec5a82012-03-01 05:57:03 -0800934 # Debug options
Ryan Cuif4f84be2012-07-09 18:50:41 -0700935 #
Brian Harringfec89fe2012-09-23 07:30:54 -0700936 # Temporary hack; in place till --dry-run replaces --debug.
937 # pylint: disable=W0212
Brian Harring009db502012-10-10 02:21:37 -0700938 group = parser.debug_group
Brian Harringfec89fe2012-09-23 07:30:54 -0700939 debug = [x for x in group.option_list if x._long_opts == ['--debug']][0]
940 debug.help += " Currently functions as --dry-run in addition."
941 debug.pass_through = True
Brian Harring3fec5a82012-03-01 05:57:03 -0800942 group.add_option('--dump_config', action='store_true', dest='dump_config',
943 default=False,
944 help='Dump out build config options, and exit.')
945 group.add_option('--notee', action='store_false', dest='tee', default=True,
946 help="Disable logging and internal tee process. Primarily "
947 "used for debugging cbuildbot itself.")
Brian Harring3fec5a82012-03-01 05:57:03 -0800948 return parser
949
950
Ryan Cui85867972012-02-23 18:21:49 -0800951def _FinishParsing(options, args):
952 """Perform some parsing tasks that need to take place after optparse.
953
954 This function needs to be easily testable! Keep it free of
955 environment-dependent code. Put more detailed usage validation in
956 _PostParseCheck().
Brian Harring3fec5a82012-03-01 05:57:03 -0800957
958 Args:
Ryan Cui85867972012-02-23 18:21:49 -0800959 options, args: The options/args object returned by optparse
Brian Harring3fec5a82012-03-01 05:57:03 -0800960 """
Ryan Cui41023d92012-11-13 19:59:50 -0800961 # Populate options.pass_through_args.
962 accepted, _ = commandline.FilteringParser.FilterArgs(
963 options.parsed_args, lambda x: x.opt_inst.pass_through)
964 options.pass_through_args.extend(accepted)
Brian Harring07039b52012-05-13 17:56:47 -0700965
Brian Harring3fec5a82012-03-01 05:57:03 -0800966 if options.chrome_root:
967 if options.chrome_rev != constants.CHROME_REV_LOCAL:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700968 cros_build_lib.Die('Chrome rev must be %s if chrome_root is set.' %
969 constants.CHROME_REV_LOCAL)
Brian Harring3fec5a82012-03-01 05:57:03 -0800970 else:
971 if options.chrome_rev == constants.CHROME_REV_LOCAL:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700972 cros_build_lib.Die('Chrome root must be set if chrome_rev is %s.' %
973 constants.CHROME_REV_LOCAL)
Brian Harring3fec5a82012-03-01 05:57:03 -0800974
975 if options.chrome_version:
976 if options.chrome_rev != constants.CHROME_REV_SPEC:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700977 cros_build_lib.Die('Chrome rev must be %s if chrome_version is set.' %
978 constants.CHROME_REV_SPEC)
Brian Harring3fec5a82012-03-01 05:57:03 -0800979 else:
980 if options.chrome_rev == constants.CHROME_REV_SPEC:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700981 cros_build_lib.Die(
982 'Chrome rev must not be %s if chrome_version is not set.'
983 % constants.CHROME_REV_SPEC)
Brian Harring3fec5a82012-03-01 05:57:03 -0800984
Ryan Cuif37608e2012-07-10 14:28:48 -0700985 if options.gerrit_chrome:
986 if options.remote_trybot or options.remote:
987 cros_build_lib.Die('Cannot use --gerrit-chrome with remote trybots!')
988 elif options.chrome_rev != constants.CHROME_REV_TOT:
989 cros_build_lib.Die('Chrome rev must be %s if chrome_root is set.' %
990 constants.CHROME_REV_TOT)
991
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700992 patches = bool(options.gerrit_patches or options.local_patches)
993 if options.remote:
994 if options.local:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700995 cros_build_lib.Die('Cannot specify both --remote and --local')
Ryan Cui54da0702012-04-19 18:38:08 -0700996
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700997 if not options.buildbot and not patches:
Brian Harring7cc4b932012-11-01 16:58:55 -0700998 if not cros_build_lib.BooleanPrompt(
999 prompt="No patches were provided; are you sure you want to just "
1000 "run a remote build of ToT?", default=False):
1001 cros_build_lib.Die('Must provide patches when running with --remote.')
Brian Harring3fec5a82012-03-01 05:57:03 -08001002
Ryan Cuieaa9efd2012-04-25 17:56:45 -07001003 # --debug needs to be explicitly passed through for remote invocations.
1004 release_mode_with_patches = (options.buildbot and patches and
1005 '--debug' not in options.pass_through_args)
1006 else:
1007 if len(args) > 1:
Brian Harring1b8c4c82012-05-29 23:03:04 -07001008 cros_build_lib.Die('Multiple configs not supported if not running with '
Brian Harringf1aad832012-07-18 10:46:39 -07001009 '--remote. Got %r', args)
Ryan Cuieaa9efd2012-04-25 17:56:45 -07001010
Ryan Cui79319ab2012-05-21 12:59:18 -07001011 if options.slaves:
Brian Harring1b8c4c82012-05-29 23:03:04 -07001012 cros_build_lib.Die('Cannot use --slaves if not running with --remote.')
Ryan Cui79319ab2012-05-21 12:59:18 -07001013
Ryan Cuieaa9efd2012-04-25 17:56:45 -07001014 release_mode_with_patches = (options.buildbot and patches and
1015 not options.debug)
1016
David James5734ea32012-08-15 20:23:49 -07001017 # When running in release mode, make sure we are running with checked-in code.
1018 # We want checked-in cbuildbot/scripts to prevent errors, and we want to build
1019 # a release image with checked-in code for CrOS packages.
1020 if release_mode_with_patches:
1021 cros_build_lib.Die(
1022 'Cannot provide patches when running with --buildbot!')
1023
Ryan Cuiba41ad32012-03-08 17:15:29 -08001024 if options.buildbot and options.remote_trybot:
Brian Harring1b8c4c82012-05-29 23:03:04 -07001025 cros_build_lib.Die(
1026 '--buildbot and --remote-trybot cannot be used together.')
Ryan Cuiba41ad32012-03-08 17:15:29 -08001027
Ryan Cui85867972012-02-23 18:21:49 -08001028 # Record whether --debug was set explicitly vs. it was inferred.
1029 options.debug_forced = False
1030 if options.debug:
1031 options.debug_forced = True
1032 else:
Ryan Cui16ca5812012-03-08 20:34:27 -08001033 # We don't set debug by default for
1034 # 1. --buildbot invocations.
1035 # 2. --remote invocations, because it needs to push changes to the tryjob
1036 # repo.
1037 options.debug = not options.buildbot and not options.remote
Brian Harring3fec5a82012-03-01 05:57:03 -08001038
Ryan Cui1c13a252012-10-16 15:00:16 -07001039 # Record the configs targeted.
1040 options.build_targets = args[:]
1041
Brian Harring3fec5a82012-03-01 05:57:03 -08001042
Brian Harring1d7ba942012-04-24 06:37:18 -07001043# pylint: disable=W0613
Brian Harringae0a5322012-09-15 01:46:51 -07001044def _PostParseCheck(parser, options, args):
Ryan Cui85867972012-02-23 18:21:49 -08001045 """Perform some usage validation after we've parsed the arguments
Brian Harring3fec5a82012-03-01 05:57:03 -08001046
Ryan Cui85867972012-02-23 18:21:49 -08001047 Args:
1048 options/args: The options/args object returned by optparse
1049 """
Ryan Cuie1e4e662012-05-21 16:39:46 -07001050 if not options.branch:
David James97d95872012-11-16 15:09:56 -08001051 options.branch = git.GetChromiteTrackingBranch()
Ryan Cuie1e4e662012-05-21 16:39:46 -07001052
Brian Harringae0a5322012-09-15 01:46:51 -07001053 if not repository.IsARepoRoot(options.sourceroot):
1054 if options.local_patches:
1055 raise Exception('Could not find repo checkout at %s!'
1056 % options.sourceroot)
1057
1058 # Ensure we have a workable cachedir from this point forward.
1059 if options.cache_dir is None:
1060 # Note, options.sourceroot is set regardless of the path
1061 # actually existing.
1062 if os.path.exists(options.sourceroot):
1063 options.cache_dir = os.path.join(options.sourceroot, '.cache')
1064 elif options.buildroot is not None:
1065 options.cache_dir = os.path.join(options.buildroot, '.cache')
1066 else:
1067 options.cache_dir = parser.FindCacheDir(parser, options)
1068 options.cache_dir = os.path.abspath(options.cache_dir)
Brian Harring8c1d7b12012-10-04 17:36:32 -07001069 parser.ConfigureCacheDir(options.cache_dir)
Brian Harringae0a5322012-09-15 01:46:51 -07001070
1071 osutils.SafeMakedirs(options.cache_dir)
Ryan Cui5ba7e152012-05-10 14:36:52 -07001072
Brian Harring609dc4e2012-05-07 02:17:44 -07001073 if options.local_patches:
Brian Harring1d7ba942012-04-24 06:37:18 -07001074 options.local_patches = _CheckLocalPatches(
Brian Harring609dc4e2012-05-07 02:17:44 -07001075 options.sourceroot, options.local_patches)
Brian Harring1d7ba942012-04-24 06:37:18 -07001076
1077 default = os.environ.get('CBUILDBOT_DEFAULT_MODE')
1078 if (default and not any([options.local, options.buildbot,
1079 options.remote, options.remote_trybot])):
Brian Harring1b8c4c82012-05-29 23:03:04 -07001080 cros_build_lib.Info("CBUILDBOT_DEFAULT_MODE=%s env var detected, using it."
1081 % default)
Brian Harring1d7ba942012-04-24 06:37:18 -07001082 default = default.lower()
1083 if default == 'local':
1084 options.local = True
1085 elif default == 'remote':
1086 options.remote = True
1087 elif default == 'buildbot':
1088 options.buildbot = True
1089 else:
Brian Harring1b8c4c82012-05-29 23:03:04 -07001090 cros_build_lib.Die("CBUILDBOT_DEFAULT_MODE value %s isn't supported. "
1091 % default)
Ryan Cui85867972012-02-23 18:21:49 -08001092
1093
1094def _ParseCommandLine(parser, argv):
1095 """Completely parse the commandline arguments"""
Brian Harring3fec5a82012-03-01 05:57:03 -08001096 (options, args) = parser.parse_args(argv)
Brian Harring37e559b2012-05-22 20:47:32 -07001097
1098 if options.output_api_version:
Ryan Cui724f1d32012-10-15 18:10:22 -07001099 print constants.REEXEC_API_VERSION
Brian Harring37e559b2012-05-22 20:47:32 -07001100 sys.exit(0)
1101
Ryan Cui54da0702012-04-19 18:38:08 -07001102 if options.list:
1103 _PrintValidConfigs(options.print_all)
1104 sys.exit(0)
1105
Ryan Cui8be16062012-04-24 12:05:26 -07001106 # Strip out null arguments.
1107 # TODO(rcui): Remove when buildbot is fixed
1108 args = [arg for arg in args if arg]
1109 if not args:
1110 parser.error('Invalid usage. Use -h to see usage. Use -l to list '
1111 'supported configs.')
1112
Ryan Cui85867972012-02-23 18:21:49 -08001113 _FinishParsing(options, args)
1114 return options, args
1115
1116
1117def main(argv):
1118 # Set umask to 022 so files created by buildbot are readable.
1119 os.umask(022)
1120
Ryan Cui85867972012-02-23 18:21:49 -08001121 parser = _CreateParser()
1122 (options, args) = _ParseCommandLine(parser, argv)
Brian Harring3fec5a82012-03-01 05:57:03 -08001123
Brian Harringae0a5322012-09-15 01:46:51 -07001124 _PostParseCheck(parser, options, args)
Brian Harring3fec5a82012-03-01 05:57:03 -08001125
Mike Frysinger8fd67dc2012-12-03 23:51:18 -05001126 cros_build_lib.AssertOutsideChroot()
Zdenek Behan98ec2fb2012-08-31 17:12:18 +02001127
Brian Harring3fec5a82012-03-01 05:57:03 -08001128 if options.remote:
Brian Harring1b8c4c82012-05-29 23:03:04 -07001129 cros_build_lib.logger.setLevel(logging.WARNING)
Ryan Cui16ca5812012-03-08 20:34:27 -08001130
Brian Harring3fec5a82012-03-01 05:57:03 -08001131 # Verify configs are valid.
1132 for bot in args:
1133 _GetConfig(bot)
1134
1135 # Verify gerrit patches are valid.
Ryan Cui16ca5812012-03-08 20:34:27 -08001136 print 'Verifying patches...'
Ryan Cuie1e4e662012-05-21 16:39:46 -07001137 patch_pool = AcquirePoolFromOptions(options)
Ryan Cui16d9e1f2012-05-11 10:50:18 -07001138
Ryan Cuieaa9efd2012-04-25 17:56:45 -07001139 # --debug need to be explicitly passed through for remote invocations.
1140 if options.buildbot and '--debug' not in options.pass_through_args:
1141 _ConfirmRemoteBuildbotRun()
1142
Ryan Cui16ca5812012-03-08 20:34:27 -08001143 print 'Submitting tryjob...'
Ryan Cui16d9e1f2012-05-11 10:50:18 -07001144 tryjob = remote_try.RemoteTryJob(options, args, patch_pool.local_patches)
Ryan Cuia25d8eb2012-07-11 14:54:27 -07001145 tryjob.Submit(testjob=options.test_tryjob, dryrun=False)
Ryan Cui16ca5812012-03-08 20:34:27 -08001146 print 'Tryjob submitted!'
1147 print ('Go to %s to view the status of your job.'
Ryan Cui4906e1c2012-04-03 20:09:34 -07001148 % tryjob.GetTrybotWaterfallLink())
Brian Harring3fec5a82012-03-01 05:57:03 -08001149 sys.exit(0)
Ryan Cui54da0702012-04-19 18:38:08 -07001150 elif (not options.buildbot and not options.remote_trybot
1151 and not options.resume and not options.local):
Brian Harring1b8c4c82012-05-29 23:03:04 -07001152 cros_build_lib.Warning(
1153 'Running in LOCAL TRYBOT mode! Use --remote to submit REMOTE '
1154 'tryjobs. Use --local to suppress this message.')
1155 cros_build_lib.Warning(
Ryan Cui51591352012-07-09 15:15:53 -07001156 'In the future, --local will be required to run the local '
Brian Harring1b8c4c82012-05-29 23:03:04 -07001157 'trybot.')
Ryan Cui54da0702012-04-19 18:38:08 -07001158 time.sleep(5)
Brian Harring3fec5a82012-03-01 05:57:03 -08001159
Ryan Cui8be16062012-04-24 12:05:26 -07001160 # Only expecting one config
1161 bot_id = args[-1]
1162 build_config = _GetConfig(bot_id)
Brian Harring3fec5a82012-03-01 05:57:03 -08001163
1164 if options.reference_repo is None:
Ryan Cui5ba7e152012-05-10 14:36:52 -07001165 repo_path = os.path.join(options.sourceroot, '.repo')
Brian Harring3fec5a82012-03-01 05:57:03 -08001166 # If we're being run from a repo checkout, reuse the repo's git pool to
1167 # cut down on sync time.
1168 if os.path.exists(repo_path):
Ryan Cui5ba7e152012-05-10 14:36:52 -07001169 options.reference_repo = options.sourceroot
Brian Harring3fec5a82012-03-01 05:57:03 -08001170 elif options.reference_repo:
1171 if not os.path.exists(options.reference_repo):
1172 parser.error('Reference path %s does not exist'
1173 % (options.reference_repo,))
1174 elif not os.path.exists(os.path.join(options.reference_repo, '.repo')):
1175 parser.error('Reference path %s does not look to be the base of a '
1176 'repo checkout; no .repo exists in the root.'
1177 % (options.reference_repo,))
Ryan Cuid4a24212012-04-04 18:08:12 -07001178
Brian Harringf11bf682012-05-14 15:53:43 -07001179 if (options.buildbot or options.remote_trybot) and not options.resume:
Brian Harring470f6112012-03-02 11:47:10 -08001180 if not options.cgroups:
Ryan Cuid4a24212012-04-04 18:08:12 -07001181 parser.error('Options --buildbot/--remote-trybot and --nocgroups cannot '
1182 'be used together. Cgroup support is required for '
1183 'buildbot/remote-trybot mode.')
Mike Frysingera78a56e2012-11-20 06:02:30 -05001184 if not cgroups.Cgroup.IsSupported():
Ryan Cuid4a24212012-04-04 18:08:12 -07001185 parser.error('Option --buildbot/--remote-trybot was given, but this '
1186 'system does not support cgroups. Failing.')
Brian Harring3fec5a82012-03-01 05:57:03 -08001187
David Jamesaad5cc72012-10-26 15:03:13 -07001188 missing = osutils.FindMissingBinaries(_BUILDBOT_REQUIRED_BINARIES)
Brian Harring351ce442012-03-09 16:38:14 -08001189 if missing:
Ryan Cuid4a24212012-04-04 18:08:12 -07001190 parser.error("Option --buildbot/--remote-trybot requires the following "
1191 "binaries which couldn't be found in $PATH: %s"
Brian Harring351ce442012-03-09 16:38:14 -08001192 % (', '.join(missing)))
1193
Brian Harring3fec5a82012-03-01 05:57:03 -08001194 if options.reference_repo:
1195 options.reference_repo = os.path.abspath(options.reference_repo)
1196
1197 if options.dump_config:
1198 # This works, but option ordering is bad...
1199 print 'Configuration %s:' % bot_id
1200 pretty_printer = pprint.PrettyPrinter(indent=2)
1201 pretty_printer.pprint(build_config)
1202 sys.exit(0)
1203
1204 if not options.buildroot:
1205 if options.buildbot:
Mike Frysinger6903c762012-12-04 01:57:16 -05001206 parser.error('Please specify a buildroot with the --buildbot option.')
Matt Tennantd55b1f42012-04-13 14:15:01 -07001207
Ryan Cui5ba7e152012-05-10 14:36:52 -07001208 options.buildroot = _DetermineDefaultBuildRoot(options.sourceroot,
1209 build_config['internal'])
Brian Harring470f6112012-03-02 11:47:10 -08001210 # We use a marker file in the buildroot to indicate the user has
1211 # consented to using this directory.
1212 if not os.path.exists(repository.GetTrybotMarkerPath(options.buildroot)):
1213 _ConfirmBuildRoot(options.buildroot)
Brian Harring3fec5a82012-03-01 05:57:03 -08001214
1215 # Sanity check of buildroot- specifically that it's not pointing into the
1216 # midst of an existing repo since git-repo doesn't support nesting.
Brian Harring3fec5a82012-03-01 05:57:03 -08001217 if (not repository.IsARepoRoot(options.buildroot) and
David James6b80dc62012-02-29 15:34:40 -08001218 repository.InARepoRepository(options.buildroot)):
Brian Harring3fec5a82012-03-01 05:57:03 -08001219 parser.error('Configured buildroot %s points into a repository checkout, '
1220 'rather than the root of it. This is not supported.'
1221 % options.buildroot)
1222
Chris Sosab5ea3b42012-10-25 15:25:20 -07001223 if not options.log_dir:
1224 options.log_dir = os.path.join(options.buildroot, _DEFAULT_LOG_DIR)
1225
Brian Harringd166aaf2012-05-14 18:31:53 -07001226 log_file = None
1227 if options.tee:
Chris Sosab5ea3b42012-10-25 15:25:20 -07001228 log_file = os.path.join(options.log_dir, _BUILDBOT_LOG_FILE)
1229 osutils.SafeMakedirs(options.log_dir)
Brian Harringd166aaf2012-05-14 18:31:53 -07001230 _BackupPreviousLog(log_file)
1231
Brian Harring1b8c4c82012-05-29 23:03:04 -07001232 with cros_build_lib.ContextManagerStack() as stack:
Brian Harringc2d09d92012-05-13 22:03:15 -07001233 critical_section = stack.Add(cleanup.EnforcedCleanupSection)
1234 stack.Add(sudo.SudoKeepAlive)
Brian Harringd166aaf2012-05-14 18:31:53 -07001235
Brian Harringc2d09d92012-05-13 22:03:15 -07001236 if not options.resume:
Brian Harring2bf55e12012-05-13 21:31:55 -07001237 # If we're in resume mode, use our parents tempdir rather than
1238 # nesting another layer.
Ryan Cui504db722013-01-22 11:48:01 -08001239 stack.Add(osutils.TempDirContextManager, prefix='cbuildbot-tmp')
Brian Harringc2d09d92012-05-13 22:03:15 -07001240 logging.debug("Cbuildbot tempdir is %r.", os.environ.get('TMP'))
Brian Harringd166aaf2012-05-14 18:31:53 -07001241
Brian Harringead69102012-07-31 15:54:07 -07001242 # TODO(ferringb): update this once https://gerrit.chromium.org/gerrit/25359
1243 # is landed- it's sensitive to the manifest-versions cache path.
Brian Harringae0a5322012-09-15 01:46:51 -07001244 options.preserve_paths = set(['manifest-versions', '.cache',
Brian Harringead69102012-07-31 15:54:07 -07001245 'manifest-versions-internal'])
Brian Harringd166aaf2012-05-14 18:31:53 -07001246 if log_file is not None:
1247 stack.Add(tee.Tee, log_file)
Brian Harring2d8f9ff2012-06-30 15:58:28 -07001248 options.preserve_paths.add(_DEFAULT_LOG_DIR)
Brian Harringd166aaf2012-05-14 18:31:53 -07001249
Brian Harringc2d09d92012-05-13 22:03:15 -07001250 if options.cgroups:
1251 stack.Add(cgroups.SimpleContainChildren, 'cbuildbot')
Brian Harringa184efa2012-03-04 11:51:25 -08001252
Brian Harringc2d09d92012-05-13 22:03:15 -07001253 # Mark everything between EnforcedCleanupSection and here as having to
1254 # be rolled back via the contextmanager cleanup handlers. This
1255 # ensures that sudo bits cannot outlive cbuildbot, that anything
1256 # cgroups would kill gets killed, etc.
1257 critical_section.ForkWatchdog()
Brian Harringd166aaf2012-05-14 18:31:53 -07001258
Brian Harringc2d09d92012-05-13 22:03:15 -07001259 if options.timeout > 0:
Brian Harring1b8c4c82012-05-29 23:03:04 -07001260 stack.Add(cros_build_lib.Timeout, options.timeout)
Brian Harringa184efa2012-03-04 11:51:25 -08001261
Brian Harringc2d09d92012-05-13 22:03:15 -07001262 if not options.buildbot:
1263 build_config = cbuildbot_config.OverrideConfigForTrybot(
1264 build_config,
1265 options.remote_trybot)
1266
Mike Frysinger6903c762012-12-04 01:57:16 -05001267 if options.buildbot or options.remote_trybot:
1268 _DisableYamaHardLinkChecks()
1269
Brian Harringc2d09d92012-05-13 22:03:15 -07001270 _RunBuildStagesWrapper(options, build_config)