blob: 4cde5a6534c86ee7a3b20c0ad74dba62041717d1 [file] [log] [blame]
Mike Frysingerd6925b52012-07-16 16:11:00 -04001# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Brian Harring3fec5a82012-03-01 05:57:03 -08002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5"""Main builder code for Chromium OS.
6
7Used by Chromium OS buildbot configuration for all Chromium OS builds including
8full and pre-flight-queue builds.
9"""
10
Mike Frysinger383367e2014-09-16 15:06:17 -040011from __future__ import print_function
12
Brian Harring3fec5a82012-03-01 05:57:03 -080013import distutils.version
14import glob
Aviv Keshet669eb5e2014-06-23 08:53:01 -070015import json
Mike Frysinger05c5faf2015-02-04 21:46:46 -050016import mock
Mike Frysingerb0b0caa2015-11-07 01:05:18 -050017import optparse # pylint: disable=deprecated-module
Brian Harring3fec5a82012-03-01 05:57:03 -080018import os
Aviv Keshetcf9c2722014-02-25 15:15:10 -080019import pickle
Brian Harring3fec5a82012-03-01 05:57:03 -080020import sys
21
Mike Frysingere4d68c22015-02-04 21:26:24 -050022from chromite.cbuildbot import builders
Don Garrett88b8d782014-05-13 17:30:55 -070023from chromite.cbuildbot import cbuildbot_run
Don Garrett4af20982015-05-29 19:02:23 -070024from chromite.cbuildbot import config_lib
Don Garrett88b8d782014-05-13 17:30:55 -070025from chromite.cbuildbot import constants
26from chromite.cbuildbot import manifest_version
27from chromite.cbuildbot import remote_try
28from chromite.cbuildbot import repository
29from chromite.cbuildbot import tee
Aviv Keshet420de512015-05-18 14:28:48 -070030from chromite.cbuildbot import topology
Prathmesh Prabhud51d7502014-12-21 01:42:55 -080031from chromite.cbuildbot import tree_status
Don Garrett88b8d782014-05-13 17:30:55 -070032from chromite.cbuildbot import trybot_patch_pool
Ningning Xiaad483542016-05-24 12:27:21 -070033from chromite.cbuildbot import failures_lib
Don Garrett88b8d782014-05-13 17:30:55 -070034from chromite.cbuildbot.stages import completion_stages
Aviv Keshet2982af52014-08-13 16:07:57 -070035from chromite.lib import cidb
Brian Harringc92a7012012-02-29 10:11:34 -080036from chromite.lib import cgroups
Brian Harringa184efa2012-03-04 11:51:25 -080037from chromite.lib import cleanup
Brian Harringb6cf9142012-09-01 20:43:17 -070038from chromite.lib import commandline
Brian Harring1b8c4c82012-05-29 23:03:04 -070039from chromite.lib import cros_build_lib
Ralph Nathan91874ca2015-03-19 13:29:41 -070040from chromite.lib import cros_logging as logging
David James97d95872012-11-16 15:09:56 -080041from chromite.lib import git
Gabe Blackde694a32015-02-19 15:11:11 -080042from chromite.lib import graphite
Stefan Zagerd49d9ff2014-08-15 21:33:37 -070043from chromite.lib import gob_util
Brian Harringaf019fb2012-05-10 15:06:13 -070044from chromite.lib import osutils
David James6450a0a2012-12-04 07:59:53 -080045from chromite.lib import parallel
Don Garrettb4318362014-10-03 15:49:36 -070046from chromite.lib import retry_stats
Brian Harring3fec5a82012-03-01 05:57:03 -080047from chromite.lib import sudo
David James3432acd2013-11-27 10:02:18 -080048from chromite.lib import timeout_util
Paul Hobbsfcf10342015-12-29 15:52:31 -080049from chromite.lib import ts_mon_config
Brian Harring3fec5a82012-03-01 05:57:03 -080050
Ryan Cuiadd49122012-03-21 22:19:58 -070051
Brian Harring3fec5a82012-03-01 05:57:03 -080052_DEFAULT_LOG_DIR = 'cbuildbot_logs'
53_BUILDBOT_LOG_FILE = 'cbuildbot.log'
54_DEFAULT_EXT_BUILDROOT = 'trybot'
55_DEFAULT_INT_BUILDROOT = 'trybot-internal'
Brian Harring351ce442012-03-09 16:38:14 -080056_BUILDBOT_REQUIRED_BINARIES = ('pbzip2',)
Ryan Cui1c13a252012-10-16 15:00:16 -070057_API_VERSION_ATTR = 'api_version'
Brian Harring3fec5a82012-03-01 05:57:03 -080058
59
Don Garrett4af20982015-05-29 19:02:23 -070060def _PrintValidConfigs(site_config, display_all=False):
Brian Harring3fec5a82012-03-01 05:57:03 -080061 """Print a list of valid buildbot configs.
62
Mike Frysinger02e1e072013-11-10 22:11:34 -050063 Args:
Don Garrett4af20982015-05-29 19:02:23 -070064 site_config: config_lib.SiteConfig containing all config info.
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070065 display_all: Print all configs. Otherwise, prints only configs with
66 trybot_list=True.
Brian Harring3fec5a82012-03-01 05:57:03 -080067 """
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070068 def _GetSortKey(config_name):
Don Garrett4af20982015-05-29 19:02:23 -070069 config_dict = site_config[config_name]
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070070 return (not config_dict['trybot_list'], config_dict['description'],
71 config_name)
72
Brian Harring3fec5a82012-03-01 05:57:03 -080073 COLUMN_WIDTH = 45
Mike Frysinger13b08882015-05-28 04:36:10 -040074 if not display_all:
75 print('Note: This is the common list; for all configs, use --all.')
Mike Frysinger383367e2014-09-16 15:06:17 -040076 print('config'.ljust(COLUMN_WIDTH), 'description')
77 print('------'.ljust(COLUMN_WIDTH), '-----------')
Don Garrett4af20982015-05-29 19:02:23 -070078 config_names = site_config.keys()
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070079 config_names.sort(key=_GetSortKey)
Brian Harring3fec5a82012-03-01 05:57:03 -080080 for name in config_names:
Don Garrett4af20982015-05-29 19:02:23 -070081 if display_all or site_config[name]['trybot_list']:
82 desc = site_config[name].get('description')
Ryan Cui4f6cf7e2012-04-18 16:12:27 -070083 desc = desc if desc else ''
Mike Frysinger383367e2014-09-16 15:06:17 -040084 print(name.ljust(COLUMN_WIDTH), desc)
Brian Harring3fec5a82012-03-01 05:57:03 -080085
86
Brian Harring3fec5a82012-03-01 05:57:03 -080087def _ConfirmBuildRoot(buildroot):
88 """Confirm with user the inferred buildroot, and mark it as confirmed."""
Ralph Nathan446aee92015-03-23 14:44:56 -070089 logging.warning('Using default directory %s as buildroot', buildroot)
Brian Harring521e7242012-11-01 16:57:42 -070090 if not cros_build_lib.BooleanPrompt(default=False):
91 print('Please specify a different buildroot via the --buildroot option.')
Brian Harring3fec5a82012-03-01 05:57:03 -080092 sys.exit(0)
93
94 if not os.path.exists(buildroot):
95 os.mkdir(buildroot)
96
97 repository.CreateTrybotMarker(buildroot)
98
99
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700100def _ConfirmRemoteBuildbotRun():
101 """Confirm user wants to run with --buildbot --remote."""
Ralph Nathan446aee92015-03-23 14:44:56 -0700102 logging.warning(
David Jameseecba232014-06-11 11:35:11 -0700103 'You are about to launch a PRODUCTION job! This is *NOT* a '
104 'trybot run! Are you sure?')
Brian Harring521e7242012-11-01 16:57:42 -0700105 if not cros_build_lib.BooleanPrompt(default=False):
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700106 print('Please specify --pass-through="--debug".')
107 sys.exit(0)
108
109
Ryan Cui5ba7e152012-05-10 14:36:52 -0700110def _DetermineDefaultBuildRoot(sourceroot, internal_build):
Brian Harring3fec5a82012-03-01 05:57:03 -0800111 """Default buildroot to be under the directory that contains current checkout.
112
Mike Frysinger02e1e072013-11-10 22:11:34 -0500113 Args:
Brian Harring3fec5a82012-03-01 05:57:03 -0800114 internal_build: Whether the build is an internal build
Ryan Cui5ba7e152012-05-10 14:36:52 -0700115 sourceroot: Use specified sourceroot.
Brian Harring3fec5a82012-03-01 05:57:03 -0800116 """
Ryan Cui5ba7e152012-05-10 14:36:52 -0700117 if not repository.IsARepoRoot(sourceroot):
Brian Harring1b8c4c82012-05-29 23:03:04 -0700118 cros_build_lib.Die(
119 'Could not find root of local checkout at %s. Please specify '
120 'using the --sourceroot option.' % sourceroot)
Brian Harring3fec5a82012-03-01 05:57:03 -0800121
122 # Place trybot buildroot under the directory containing current checkout.
Ryan Cui5ba7e152012-05-10 14:36:52 -0700123 top_level = os.path.dirname(os.path.realpath(sourceroot))
Brian Harring3fec5a82012-03-01 05:57:03 -0800124 if internal_build:
125 buildroot = os.path.join(top_level, _DEFAULT_INT_BUILDROOT)
126 else:
127 buildroot = os.path.join(top_level, _DEFAULT_EXT_BUILDROOT)
128
129 return buildroot
130
131
132def _BackupPreviousLog(log_file, backup_limit=25):
133 """Rename previous log.
134
135 Args:
136 log_file: The absolute path to the previous log.
Aviv Keshet9e4236b2013-12-13 13:07:50 -0800137 backup_limit: Maximum number of old logs to keep.
Brian Harring3fec5a82012-03-01 05:57:03 -0800138 """
139 if os.path.exists(log_file):
140 old_logs = sorted(glob.glob(log_file + '.*'),
141 key=distutils.version.LooseVersion)
142
143 if len(old_logs) >= backup_limit:
144 os.remove(old_logs[0])
145
146 last = 0
147 if old_logs:
148 last = int(old_logs.pop().rpartition('.')[2])
149
150 os.rename(log_file, log_file + '.' + str(last + 1))
151
Ryan Cui5616a512012-08-17 13:39:36 -0700152
Gaurav Shah298aa372014-01-31 09:27:24 -0800153def _IsDistributedBuilder(options, chrome_rev, build_config):
154 """Determines whether the builder should be a DistributedBuilder.
155
156 Args:
157 options: options passed on the commandline.
158 chrome_rev: Chrome revision to build.
159 build_config: Builder configuration dictionary.
160
161 Returns:
162 True if the builder should be a distributed_builder
163 """
Don Garrett0bc85672015-07-23 19:46:00 +0000164 if build_config['pre_cq']:
Gaurav Shah298aa372014-01-31 09:27:24 -0800165 return True
166 elif not options.buildbot:
167 return False
168 elif chrome_rev in (constants.CHROME_REV_TOT,
169 constants.CHROME_REV_LOCAL,
170 constants.CHROME_REV_SPEC):
171 # We don't do distributed logic to TOT Chrome PFQ's, nor local
172 # chrome roots (e.g. chrome try bots)
173 # TODO(davidjames): Update any builders that rely on this logic to use
174 # manifest_version=False instead.
175 return False
176 elif build_config['manifest_version']:
177 return True
178
179 return False
180
181
Don Garretta52a5b02015-06-02 14:52:57 -0700182def _RunBuildStagesWrapper(options, site_config, build_config):
Brian Harring3fec5a82012-03-01 05:57:03 -0800183 """Helper function that wraps RunBuildStages()."""
Ralph Nathan03047282015-03-23 11:09:32 -0700184 logging.info('cbuildbot was executed with args %s' %
185 cros_build_lib.CmdToStr(sys.argv))
Brian Harring3fec5a82012-03-01 05:57:03 -0800186
David Jamesa0a664e2013-02-13 09:52:01 -0800187 chrome_rev = build_config['chrome_rev']
188 if options.chrome_rev:
189 chrome_rev = options.chrome_rev
190 if chrome_rev == constants.CHROME_REV_TOT:
Stefan Zagerd49d9ff2014-08-15 21:33:37 -0700191 options.chrome_version = gob_util.GetTipOfTrunkRevision(
192 constants.CHROMIUM_GOB_URL)
David Jamesa0a664e2013-02-13 09:52:01 -0800193 options.chrome_rev = constants.CHROME_REV_SPEC
194
David James4a404a52013-02-19 13:07:59 -0800195 # If it's likely we'll need to build Chrome, fetch the source.
Mike Frysingerf4d9ab82013-03-05 19:38:05 -0500196 if build_config['sync_chrome'] is None:
David Jameseecba232014-06-11 11:35:11 -0700197 options.managed_chrome = (
198 chrome_rev != constants.CHROME_REV_LOCAL and
Mike Frysingerf4d9ab82013-03-05 19:38:05 -0500199 (not build_config['usepkg_build_packages'] or chrome_rev or
David James3cce4642013-05-10 17:50:23 -0700200 build_config['profile'] or options.rietveld_patches))
Mike Frysingerf4d9ab82013-03-05 19:38:05 -0500201 else:
202 options.managed_chrome = build_config['sync_chrome']
David James2333c182013-02-13 16:16:15 -0800203
204 if options.managed_chrome:
205 # Tell Chrome to fetch the source locally.
Matt Tennant628ffdd2013-11-27 14:44:39 -0800206 internal = constants.USE_CHROME_INTERNAL in build_config['useflags']
David James2333c182013-02-13 16:16:15 -0800207 chrome_src = 'chrome-src-internal' if internal else 'chrome-src'
208 options.chrome_root = os.path.join(options.cache_dir, 'distfiles', 'target',
209 chrome_src)
David James9e27e662013-02-14 13:42:43 -0800210 elif options.rietveld_patches:
David James4a404a52013-02-19 13:07:59 -0800211 cros_build_lib.Die('This builder does not support Rietveld patches.')
David James2333c182013-02-13 16:16:15 -0800212
Aviv Keshet669eb5e2014-06-23 08:53:01 -0700213 metadata_dump_dict = {}
214 if options.metadata_dump:
215 with open(options.metadata_dump, 'r') as metadata_file:
216 metadata_dump_dict = json.loads(metadata_file.read())
217
Matt Tennant95a42ad2013-12-27 15:38:36 -0800218 # We are done munging options values, so freeze options object now to avoid
219 # further abuse of it.
220 # TODO(mtennant): one by one identify each options value override and see if
221 # it can be handled another way. Try to push this freeze closer and closer
222 # to the start of the script (e.g. in or after _PostParseCheck).
223 options.Freeze()
224
Matt Tennant0940c382014-01-21 20:43:55 -0800225 with parallel.Manager() as manager:
Don Garretta52a5b02015-06-02 14:52:57 -0700226 builder_run = cbuildbot_run.BuilderRun(
227 options, site_config, build_config, manager)
Aviv Keshet669eb5e2014-06-23 08:53:01 -0700228 if metadata_dump_dict:
229 builder_run.attrs.metadata.UpdateWithDict(metadata_dump_dict)
Mike Frysingere4d68c22015-02-04 21:26:24 -0500230
Mike Frysinger05c5faf2015-02-04 21:46:46 -0500231 if builder_run.config.builder_class_name is None:
Don Garrett56e6ed32015-06-23 16:52:20 -0700232 # TODO: This should get relocated to chromeos_config.
Mike Frysingere4d68c22015-02-04 21:26:24 -0500233 if _IsDistributedBuilder(options, chrome_rev, build_config):
Mike Frysinger05c5faf2015-02-04 21:46:46 -0500234 builder_cls_name = 'simple_builders.DistributedBuilder'
Mike Frysingere4d68c22015-02-04 21:26:24 -0500235 else:
Mike Frysinger05c5faf2015-02-04 21:46:46 -0500236 builder_cls_name = 'simple_builders.SimpleBuilder'
237 builder_cls = builders.GetBuilderClass(builder_cls_name)
238 builder = builder_cls(builder_run)
Matt Tennant0940c382014-01-21 20:43:55 -0800239 else:
Mike Frysinger05c5faf2015-02-04 21:46:46 -0500240 builder = builders.Builder(builder_run)
Mike Frysingere4d68c22015-02-04 21:26:24 -0500241
Matt Tennant0940c382014-01-21 20:43:55 -0800242 if not builder.Run():
243 sys.exit(1)
Brian Harring3fec5a82012-03-01 05:57:03 -0800244
245
246# Parser related functions
Ryan Cui5ba7e152012-05-10 14:36:52 -0700247def _CheckLocalPatches(sourceroot, local_patches):
Brian Harring3fec5a82012-03-01 05:57:03 -0800248 """Do an early quick check of the passed-in patches.
249
250 If the branch of a project is not specified we append the current branch the
251 project is on.
Ryan Cui5ba7e152012-05-10 14:36:52 -0700252
David Jamese3b06062013-11-09 18:52:02 -0800253 TODO(davidjames): The project:branch format isn't unique, so this means that
254 we can't differentiate what directory the user intended to apply patches to.
255 We should references by directory instead.
256
Ryan Cui5ba7e152012-05-10 14:36:52 -0700257 Args:
258 sourceroot: The checkout where patches are coming from.
Aviv Keshet9e4236b2013-12-13 13:07:50 -0800259 local_patches: List of patches to check in project:branch format.
David Jamese3b06062013-11-09 18:52:02 -0800260
261 Returns:
262 A list of patches that have been verified, in project:branch format.
Brian Harring3fec5a82012-03-01 05:57:03 -0800263 """
Ryan Cuicedd8a52012-03-22 02:28:35 -0700264 verified_patches = []
David James97d95872012-11-16 15:09:56 -0800265 manifest = git.ManifestCheckout.Cached(sourceroot)
Ryan Cuicedd8a52012-03-22 02:28:35 -0700266 for patch in local_patches:
David Jamese3b06062013-11-09 18:52:02 -0800267 project, _, branch = patch.partition(':')
268
Gaurav Shah7afb0562013-12-26 15:05:39 -0800269 checkouts = manifest.FindCheckouts(project, only_patchable=True)
David Jamese3b06062013-11-09 18:52:02 -0800270 if not checkouts:
271 cros_build_lib.Die('Project %s does not exist.' % (project,))
272 if len(checkouts) > 1:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700273 cros_build_lib.Die(
David Jamese3b06062013-11-09 18:52:02 -0800274 'We do not yet support local patching for projects that are checked '
275 'out to multiple directories. Try uploading your patch to gerrit '
276 'and referencing it via the -g option instead.'
277 )
Brian Harring3fec5a82012-03-01 05:57:03 -0800278
David Jamese3b06062013-11-09 18:52:02 -0800279 ok = False
280 for checkout in checkouts:
281 project_dir = checkout.GetPath(absolute=True)
Brian Harring3fec5a82012-03-01 05:57:03 -0800282
David Jamese3b06062013-11-09 18:52:02 -0800283 # If no branch was specified, we use the project's current branch.
Brian Harring3fec5a82012-03-01 05:57:03 -0800284 if not branch:
David Jamese3b06062013-11-09 18:52:02 -0800285 local_branch = git.GetCurrentBranch(project_dir)
286 else:
287 local_branch = branch
Brian Harring3fec5a82012-03-01 05:57:03 -0800288
David Jamesf1a07612014-04-28 17:48:52 -0700289 if local_branch and git.DoesCommitExistInRepo(project_dir, local_branch):
David Jamese3b06062013-11-09 18:52:02 -0800290 verified_patches.append('%s:%s' % (project, local_branch))
291 ok = True
292
293 if not ok:
294 if branch:
295 cros_build_lib.Die('Project %s does not have branch %s'
David Jameseecba232014-06-11 11:35:11 -0700296 % (project, branch))
David Jamese3b06062013-11-09 18:52:02 -0800297 else:
298 cros_build_lib.Die('Project %s is not on a branch!' % (project,))
Brian Harring3fec5a82012-03-01 05:57:03 -0800299
Ryan Cuicedd8a52012-03-22 02:28:35 -0700300 return verified_patches
Brian Harring3fec5a82012-03-01 05:57:03 -0800301
302
Brian Harring3fec5a82012-03-01 05:57:03 -0800303def _CheckChromeVersionOption(_option, _opt_str, value, parser):
304 """Upgrade other options based on chrome_version being passed."""
305 value = value.strip()
306
307 if parser.values.chrome_rev is None and value:
308 parser.values.chrome_rev = constants.CHROME_REV_SPEC
309
310 parser.values.chrome_version = value
311
312
313def _CheckChromeRootOption(_option, _opt_str, value, parser):
314 """Validate and convert chrome_root to full-path form."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800315 if parser.values.chrome_rev is None:
316 parser.values.chrome_rev = constants.CHROME_REV_LOCAL
317
Ryan Cui5ba7e152012-05-10 14:36:52 -0700318 parser.values.chrome_root = value
Brian Harring3fec5a82012-03-01 05:57:03 -0800319
320
David Jamesac8c2a72013-02-13 18:44:33 -0800321def FindCacheDir(_parser, _options):
Brian Harringae0a5322012-09-15 01:46:51 -0700322 return None
323
324
Ryan Cui5ba7e152012-05-10 14:36:52 -0700325class CustomGroup(optparse.OptionGroup):
Aviv Keshet9e4236b2013-12-13 13:07:50 -0800326 """Custom option group which supports arguments passed-through to trybot."""
David Jameseecba232014-06-11 11:35:11 -0700327
Ryan Cui5ba7e152012-05-10 14:36:52 -0700328 def add_remote_option(self, *args, **kwargs):
329 """For arguments that are passed-through to remote trybot."""
330 return optparse.OptionGroup.add_option(self, *args,
331 remote_pass_through=True,
332 **kwargs)
333
334
Ryan Cui1c13a252012-10-16 15:00:16 -0700335class CustomOption(commandline.FilteringOption):
336 """Subclass FilteringOption class to implement pass-through and api."""
Ryan Cui5ba7e152012-05-10 14:36:52 -0700337
Ryan Cui1c13a252012-10-16 15:00:16 -0700338 ACTIONS = commandline.FilteringOption.ACTIONS + ('extend',)
339 STORE_ACTIONS = commandline.FilteringOption.STORE_ACTIONS + ('extend',)
340 TYPED_ACTIONS = commandline.FilteringOption.TYPED_ACTIONS + ('extend',)
341 ALWAYS_TYPED_ACTIONS = (commandline.FilteringOption.ALWAYS_TYPED_ACTIONS +
342 ('extend',))
Ryan Cui79319ab2012-05-21 12:59:18 -0700343
Ryan Cui5ba7e152012-05-10 14:36:52 -0700344 def __init__(self, *args, **kwargs):
345 # The remote_pass_through argument specifies whether we should directly
346 # pass the argument (with its value) onto the remote trybot.
347 self.pass_through = kwargs.pop('remote_pass_through', False)
Ryan Cui1c13a252012-10-16 15:00:16 -0700348 self.api_version = int(kwargs.pop('api', '0'))
349 commandline.FilteringOption.__init__(self, *args, **kwargs)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700350
351 def take_action(self, action, dest, opt, value, values, parser):
Ryan Cui79319ab2012-05-21 12:59:18 -0700352 if action == 'extend':
Mike Frysingerd6925b52012-07-16 16:11:00 -0400353 # If there is extra spaces between each argument, we get '' which later
354 # code barfs on, so skip those. e.g. We see this with the forms:
355 # cbuildbot -p 'proj:branch ' ...
356 # cbuildbot -p ' proj:branch' ...
357 # cbuildbot -p 'proj:branch proj2:branch' ...
358 lvalue = value.split()
Ryan Cui79319ab2012-05-21 12:59:18 -0700359 values.ensure_value(dest, []).extend(lvalue)
Ryan Cui79319ab2012-05-21 12:59:18 -0700360
Ryan Cui1c13a252012-10-16 15:00:16 -0700361 commandline.FilteringOption.take_action(
362 self, action, dest, opt, value, values, parser)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700363
364
Ryan Cui1c13a252012-10-16 15:00:16 -0700365class CustomParser(commandline.FilteringParser):
Aviv Keshet9e4236b2013-12-13 13:07:50 -0800366 """Custom option parser which supports arguments passed-trhough to trybot"""
Matt Tennante8179042013-10-01 15:47:32 -0700367
Brian Harringb6cf9142012-09-01 20:43:17 -0700368 DEFAULT_OPTION_CLASS = CustomOption
369
370 def add_remote_option(self, *args, **kwargs):
371 """For arguments that are passed-through to remote trybot."""
Ryan Cui1c13a252012-10-16 15:00:16 -0700372 return self.add_option(*args, remote_pass_through=True, **kwargs)
Brian Harringb6cf9142012-09-01 20:43:17 -0700373
374
Brian Harring3fec5a82012-03-01 05:57:03 -0800375def _CreateParser():
Ryan Cui16d9e1f2012-05-11 10:50:18 -0700376 """Generate and return the parser with all the options."""
Brian Harring3fec5a82012-03-01 05:57:03 -0800377 # Parse options
David Jameseecba232014-06-11 11:35:11 -0700378 usage = 'usage: %prog [options] buildbot_config [buildbot_config ...]'
Brian Harringae0a5322012-09-15 01:46:51 -0700379 parser = CustomParser(usage=usage, caching=FindCacheDir)
Brian Harring3fec5a82012-03-01 05:57:03 -0800380
381 # Main options
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400382 parser.add_option('-l', '--list', action='store_true', dest='list',
383 default=False,
384 help='List the suggested trybot configs to use (see --all)')
Brian Harring3fec5a82012-03-01 05:57:03 -0800385 parser.add_option('-a', '--all', action='store_true', dest='print_all',
386 default=False,
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400387 help='List all of the buildbot configs available w/--list')
388
389 parser.add_option('--local', default=False, action='store_true',
David Jameseecba232014-06-11 11:35:11 -0700390 help=('Specifies that this tryjob should be run locally. '
391 'Implies --debug.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400392 parser.add_option('--remote', default=False, action='store_true',
Ryan Cui88b901c2013-06-21 11:35:30 -0700393 help='Specifies that this tryjob should be run remotely.')
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400394
Ryan Cuie1e4e662012-05-21 16:39:46 -0700395 parser.add_remote_option('-b', '--branch',
David Jameseecba232014-06-11 11:35:11 -0700396 help=('The manifest branch to test. The branch to '
397 'check the buildroot out to.'))
Ryan Cui5ba7e152012-05-10 14:36:52 -0700398 parser.add_option('-r', '--buildroot', dest='buildroot', type='path',
David Jameseecba232014-06-11 11:35:11 -0700399 help=('Root directory where source is checked out to, and '
400 'where the build occurs. For external build configs, '
401 "defaults to 'trybot' directory at top level of your "
402 'repo-managed checkout.'))
Prathmesh Prabhu867c1172015-06-02 17:57:59 -0700403 parser.add_option('--bootstrap-dir', type='path', default=None,
404 help='Bootstrapping cbuildbot may involve checking out '
405 'multiple copies of chromite. All these checkouts '
406 'will be contained in the directory specified here. '
407 'Default:%s' % osutils.GetGlobalTempDir())
David Riley486b2612016-02-22 19:59:26 -0800408 parser.add_remote_option('--android_rev', default=None, type='choice',
409 choices=constants.VALID_ANDROID_REVISIONS,
410 help=('Revision of Android to use, of type [%s]'
411 % '|'.join(constants.VALID_ANDROID_REVISIONS)))
412 parser.add_remote_option('--chrome_rev', default=None, type='choice',
413 choices=constants.VALID_CHROME_REVISIONS,
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700414 help=('Revision of Chrome to use, of type [%s]'
415 % '|'.join(constants.VALID_CHROME_REVISIONS)))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700416 parser.add_remote_option('--profile', default=None, type='string',
417 action='store', dest='profile',
418 help='Name of profile to sub-specify board variant.')
Don Garrettb85658c2015-06-30 19:07:22 -0700419 parser.add_option('-c', '--config_repo',
420 help=('Cloneable path to the git repository containing '
421 'the site configuration to use.'))
Brian Harring3fec5a82012-03-01 05:57:03 -0800422
Ryan Cuif4f84be2012-07-09 18:50:41 -0700423 #
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400424 # Patch selection options.
425 #
426
427 group = CustomGroup(
428 parser,
429 'Patch Options')
430
431 group.add_remote_option('-g', '--gerrit-patches', action='extend',
432 default=[], type='string',
433 metavar="'Id1 *int_Id2...IdN'",
David Jameseecba232014-06-11 11:35:11 -0700434 help=('Space-separated list of short-form Gerrit '
435 "Change-Id's or change numbers to patch. "
436 "Please prepend '*' to internal Change-Id's"))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400437 group.add_remote_option('-G', '--rietveld-patches', action='extend',
438 default=[], type='string',
439 metavar="'id1[:subdir1]...idN[:subdirN]'",
David Jameseecba232014-06-11 11:35:11 -0700440 help=('Space-separated list of short-form Rietveld '
441 'issue numbers to patch. If no subdir is '
442 'specified, the src directory is used.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400443 group.add_option('-p', '--local-patches', action='extend', default=[],
444 metavar="'<project1>[:<branch1>]...<projectN>[:<branchN>]'",
David Jameseecba232014-06-11 11:35:11 -0700445 help=('Space-separated list of project branches with '
446 'patches to apply. Projects are specified by name. '
447 'If no branch is specified the current branch of the '
448 'project will be used.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400449
450 parser.add_option_group(group)
451
452 #
453 # Remote trybot options.
454 #
455
456 group = CustomGroup(
457 parser,
458 'Remote Trybot Options (--remote)')
459
Paul Hobbs5cd050d2015-06-30 16:52:28 -0700460 group.add_option('--use-buildbucket', default=False, action='store_true',
461 help=('Use buildbucket instead of git to request'
462 'the tryjob(s).'))
463
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400464 group.add_remote_option('--hwtest', dest='hwtest', action='store_true',
David Jameseecba232014-06-11 11:35:11 -0700465 default=False,
466 help='Run the HWTest stage (tests on real hardware)')
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400467 group.add_option('--remote-description', default=None,
David Jameseecba232014-06-11 11:35:11 -0700468 help=('Attach an optional description to a --remote run '
469 'to make it easier to identify the results when it '
470 'finishes'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400471 group.add_option('--slaves', action='extend', default=[],
David Jameseecba232014-06-11 11:35:11 -0700472 help=('Specify specific remote tryslaves to run on (e.g. '
473 'build149-m2); if the bot is busy, it will be queued'))
Don Garrett4bb21682014-03-03 16:16:23 -0800474 group.add_remote_option('--channel', dest='channels', action='extend',
475 default=[],
David Jameseecba232014-06-11 11:35:11 -0700476 help=('Specify a channel for a payloads trybot. Can '
477 'be specified multiple times. No valid for '
478 'non-payloads configs.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400479 group.add_option('--test-tryjob', action='store_true',
480 default=False,
David Jameseecba232014-06-11 11:35:11 -0700481 help=('Submit a tryjob to the test repository. Will not '
482 'show up on the production trybot waterfall.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400483
484 parser.add_option_group(group)
485
486 #
Ryan Cui88b901c2013-06-21 11:35:30 -0700487 # Branch creation options.
488 #
489
490 group = CustomGroup(
491 parser,
492 'Branch Creation Options (used with branch-util)')
493
494 group.add_remote_option('--branch-name',
495 help='The branch to create or delete.')
496 group.add_remote_option('--delete-branch', default=False, action='store_true',
497 help='Delete the branch specified in --branch-name.')
498 group.add_remote_option('--rename-to', type='string',
499 help='Rename a branch to the specified name.')
500 group.add_remote_option('--force-create', default=False, action='store_true',
501 help='Overwrites an existing branch.')
Prathmesh Prabhue5f4d472015-05-07 16:52:10 -0700502 group.add_remote_option('--skip-remote-push', default=False,
503 action='store_true',
504 help='Do not actually push to remote git repos. '
505 'Used for end-to-end testing branching.')
Ryan Cui88b901c2013-06-21 11:35:30 -0700506
507 parser.add_option_group(group)
508
509 #
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400510 # Advanced options.
Ryan Cuif4f84be2012-07-09 18:50:41 -0700511 #
512
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700513 group = CustomGroup(
Brian Harring3fec5a82012-03-01 05:57:03 -0800514 parser,
515 'Advanced Options',
516 'Caution: use these options at your own risk.')
517
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400518 group.add_remote_option('--bootstrap-args', action='append', default=[],
David Jameseecba232014-06-11 11:35:11 -0700519 help=('Args passed directly to the bootstrap re-exec '
520 'to skip verification by the bootstrap code'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700521 group.add_remote_option('--buildbot', dest='buildbot', action='store_true',
522 default=False, help='This is running on a buildbot')
Don Garrett0af6ba52016-05-27 14:53:31 -0700523 parser.add_remote_option('--repo-cache', type='path',
524 help='Directory from which to copy a repo checkout '
525 'if our build root is empty, to avoid '
Don Garrettbb79be92016-09-27 11:14:07 -0700526 'excessive GoB load with a fresh sync.'
527 'Deprecated in favor of --git-cache-dir.')
Prathmesh Prabhu51d774f2015-07-17 15:11:49 -0700528 group.add_remote_option('--no-buildbot-tags', action='store_false',
529 dest='enable_buildbot_tags', default=True,
530 help='Suppress buildbot specific tags from log '
531 'output. This is used to hide recursive '
532 'cbuilbot runs on the waterfall.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700533 group.add_remote_option('--buildnumber', help='build number', type='int',
534 default=0)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700535 group.add_option('--chrome_root', default=None, type='path',
536 action='callback', callback=_CheckChromeRootOption,
537 dest='chrome_root', help='Local checkout of Chrome to use.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700538 group.add_remote_option('--chrome_version', default=None, type='string',
539 action='callback', dest='chrome_version',
540 callback=_CheckChromeVersionOption,
David Jameseecba232014-06-11 11:35:11 -0700541 help=('Used with SPEC logic to force a particular '
Stefan Zagerd49d9ff2014-08-15 21:33:37 -0700542 'git revision of chrome rather than the '
David Jameseecba232014-06-11 11:35:11 -0700543 'latest.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700544 group.add_remote_option('--clobber', action='store_true', dest='clobber',
545 default=False,
546 help='Clears an old checkout before syncing')
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400547 group.add_remote_option('--latest-toolchain', action='store_true',
548 default=False,
549 help='Use the latest toolchain.')
Ryan Cui5ba7e152012-05-10 14:36:52 -0700550 parser.add_option('--log_dir', dest='log_dir', type='path',
Brian Harring3fec5a82012-03-01 05:57:03 -0800551 help=('Directory where logs are stored.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700552 group.add_remote_option('--maxarchives', dest='max_archive_builds',
553 default=3, type='int',
David Jameseecba232014-06-11 11:35:11 -0700554 help='Change the local saved build count limit.')
Ryan Cuibbd3d4b2012-08-17 12:20:37 -0700555 parser.add_remote_option('--manifest-repo-url',
556 help=('Overrides the default manifest repo url.'))
David James565bc9a2013-04-08 14:54:45 -0700557 group.add_remote_option('--compilecheck', action='store_true', default=False,
558 help='Only verify compilation and unit tests.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700559 group.add_remote_option('--noarchive', action='store_false', dest='archive',
560 default=True, help="Don't run archive stage.")
Ryan Cuif7f24692012-05-18 16:35:33 -0700561 group.add_remote_option('--nobootstrap', action='store_false',
562 dest='bootstrap', default=True,
David Jameseecba232014-06-11 11:35:11 -0700563 help=("Don't checkout and run from a standalone "
564 'chromite repo.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700565 group.add_remote_option('--nobuild', action='store_false', dest='build',
566 default=True,
567 help="Don't actually build (for cbuildbot dev)")
568 group.add_remote_option('--noclean', action='store_false', dest='clean',
569 default=True, help="Don't clean the buildroot")
Ryan Cuif7f24692012-05-18 16:35:33 -0700570 group.add_remote_option('--nocgroups', action='store_false', dest='cgroups',
571 default=True,
572 help='Disable cbuildbots usage of cgroups.')
Ryan Cui3ea98e02013-08-07 16:01:48 -0700573 group.add_remote_option('--nochromesdk', action='store_false',
574 dest='chrome_sdk', default=True,
David Jameseecba232014-06-11 11:35:11 -0700575 help=("Don't run the ChromeSDK stage which builds "
576 'Chrome outside of the chroot.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700577 group.add_remote_option('--noprebuilts', action='store_false',
578 dest='prebuilts', default=True,
579 help="Don't upload prebuilts.")
Ryan Cui88b901c2013-06-21 11:35:30 -0700580 group.add_remote_option('--nopatch', action='store_false',
581 dest='postsync_patch', default=True,
582 help=("Don't run PatchChanges stage. This does not "
David Jameseecba232014-06-11 11:35:11 -0700583 'disable patching in of chromite patches '
584 'during BootstrapStage.'))
Don Garrett82c0ae82014-02-03 18:25:11 -0800585 group.add_remote_option('--nopaygen', action='store_false',
586 dest='paygen', default=True,
587 help="Don't generate payloads.")
Ryan Cui88b901c2013-06-21 11:35:30 -0700588 group.add_remote_option('--noreexec', action='store_false',
589 dest='postsync_reexec', default=True,
590 help="Don't reexec into the buildroot after syncing.")
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400591 group.add_remote_option('--nosdk', action='store_true',
592 default=False,
593 help='Re-create the SDK from scratch.')
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700594 group.add_remote_option('--nosync', action='store_false', dest='sync',
595 default=True, help="Don't sync before building.")
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700596 group.add_remote_option('--notests', action='store_false', dest='tests',
597 default=True,
David Jameseecba232014-06-11 11:35:11 -0700598 help=('Override values from buildconfig and run no '
599 'tests.'))
Nam T. Nguyenc93f1342014-07-11 14:40:54 -0700600 group.add_remote_option('--noimagetests', action='store_false',
601 dest='image_test', default=True,
602 help=('Override values from buildconfig and run no '
603 'image tests.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700604 group.add_remote_option('--nouprev', action='store_false', dest='uprev',
605 default=True,
David Jameseecba232014-06-11 11:35:11 -0700606 help=('Override values from buildconfig and never '
607 'uprev.'))
David Jamesdac7a912013-11-18 11:14:44 -0800608 group.add_option('--reference-repo', action='store', default=None,
609 dest='reference_repo',
David Jameseecba232014-06-11 11:35:11 -0700610 help=('Reuse git data stored in an existing repo '
611 'checkout. This can drastically reduce the network '
612 'time spent setting up the trybot checkout. By '
613 "default, if this option isn't given but cbuildbot "
614 'is invoked from a repo checkout, cbuildbot will '
615 'use the repo root.'))
Ryan Cuicedd8a52012-03-22 02:28:35 -0700616 group.add_option('--resume', action='store_true', default=False,
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700617 help='Skip stages already successfully completed.')
618 group.add_remote_option('--timeout', action='store', type='int', default=0,
David Jameseecba232014-06-11 11:35:11 -0700619 help=('Specify the maximum amount of time this job '
620 'can run for, at which point the build will be '
621 'aborted. If set to zero, then there is no '
622 'timeout.'))
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700623 group.add_remote_option('--version', dest='force_version', default=None,
David Jameseecba232014-06-11 11:35:11 -0700624 help=('Used with manifest logic. Forces use of this '
625 'version rather than create or get latest. '
626 'Examples: 4815.0.0-rc1, 4815.1.2'))
Ningning Xia16a4aa42016-09-19 15:30:10 -0700627 group.add_remote_option('--git-cache-dir', dest='git_cache_dir', default=None,
Don Garrettbb79be92016-09-27 11:14:07 -0700628 api=constants.REEXEC_API_GIT_CACHE_DIR,
Ningning Xia16a4aa42016-09-19 15:30:10 -0700629 help=('Specify the cache directory to store the '
630 'project caches populated by the git-cache '
631 'tool. Bootstrap the projects based on the git '
632 'cache files instead of fetching them directly '
633 'from the GoB servers.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400634
635 parser.add_option_group(group)
636
637 #
638 # Internal options.
639 #
640
641 group = CustomGroup(
642 parser,
Mike Frysinger34db8692013-11-11 14:54:08 -0500643 'Internal Chromium OS Build Team Options',
644 'Caution: these are for meant for the Chromium OS build team only')
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400645
646 group.add_remote_option('--archive-base', type='gs_path',
David Jameseecba232014-06-11 11:35:11 -0700647 help=('Base GS URL (gs://<bucket_name>/<path>) to '
648 'upload archive artifacts to'))
649 group.add_remote_option(
650 '--cq-gerrit-query', dest='cq_gerrit_override', default=None,
651 help=('If given, this gerrit query will be used to find what patches to '
652 "test, rather than the normal 'CommitQueue>=1 AND Verified=1 AND "
653 "CodeReview=2' query it defaults to. Use with care- note "
654 'additionally this setting only has an effect if the buildbot '
655 "target is a cq target, and we're in buildbot mode."))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400656 group.add_option('--pass-through', dest='pass_through_args', action='append',
657 type='string', default=[])
658 group.add_option('--reexec-api-version', dest='output_api_version',
Ryan Cuif4f84be2012-07-09 18:50:41 -0700659 action='store_true', default=False,
David Jameseecba232014-06-11 11:35:11 -0700660 help=('Used for handling forwards/backwards compatibility '
661 'with --resume and --bootstrap'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400662 group.add_option('--remote-trybot', dest='remote_trybot',
663 action='store_true', default=False,
664 help='Indicates this is running on a remote trybot machine')
Ningning Xia20aa3a82016-07-18 11:16:18 -0700665 group.add_option('--buildbucket-id', dest='buildbucket_id', default=None,
666 help=('The unique ID in buildbucket of current build '
667 'generated by buildbucket.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400668 group.add_remote_option('--remote-patches', action='extend', default=[],
David Jameseecba232014-06-11 11:35:11 -0700669 help=('Patches uploaded by the trybot client when '
670 'run using the -p option'))
Brian Harringf611e6e2012-07-17 18:47:44 -0700671 # Note the default here needs to be hardcoded to 3; that is the last version
672 # that lacked this functionality.
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400673 group.add_option('--remote-version', default=3, type=int, action='store',
David Jameseecba232014-06-11 11:35:11 -0700674 help=('Used for compatibility checks w/tryjobs running in '
675 'older chromite instances'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400676 group.add_option('--sourceroot', type='path', default=constants.SOURCE_ROOT)
677 group.add_remote_option('--test-bootstrap', action='store_true',
678 default=False,
David Jameseecba232014-06-11 11:35:11 -0700679 help=('Causes cbuildbot to bootstrap itself twice, '
680 'in the sequence A->B->C: A(unpatched) patches '
681 'and bootstraps B; B patches and bootstraps C'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400682 group.add_remote_option('--validation_pool', default=None,
David Jameseecba232014-06-11 11:35:11 -0700683 help=('Path to a pickled validation pool. Intended '
684 'for use only with the commit queue.'))
Aviv Keshet007a24a2014-06-23 08:10:55 -0700685 group.add_remote_option('--metadata_dump', default=None,
Aviv Keshet669eb5e2014-06-23 08:53:01 -0700686 help=('Path to a json dumped metadata file. This '
687 'will be used as the initial metadata.'))
Aviv Keshetacf4cfb2014-07-30 12:31:22 -0700688 group.add_remote_option('--master-build-id', default=None, type=int,
689 api=constants.REEXEC_API_MASTER_BUILD_ID,
690 help=('cidb build id of the master build to this '
691 'slave build.'))
Aviv Kesheta0159be2013-12-12 13:56:28 -0800692 group.add_remote_option('--mock-tree-status', dest='mock_tree_status',
693 default=None, action='store',
David Jameseecba232014-06-11 11:35:11 -0700694 help=('Override the tree status value that would be '
695 'returned from the the actual tree. Example '
696 'values: open, closed, throttled. When used '
697 'in conjunction with --debug, the tree status '
698 'will not be ignored as it usually is in a '
699 '--debug run.'))
700 group.add_remote_option(
701 '--mock-slave-status', dest='mock_slave_status', default=None,
702 action='store', metavar='MOCK_SLAVE_STATUS_PICKLE_FILE',
703 help=('Override the result of the _FetchSlaveStatuses method of '
704 'MasterSlaveSyncCompletionStage, by specifying a file with a '
705 'pickle of the result to be returned.'))
Mike Frysinger81af6ef2013-04-03 02:24:09 -0400706
707 parser.add_option_group(group)
Ryan Cuif4f84be2012-07-09 18:50:41 -0700708
709 #
Brian Harring3fec5a82012-03-01 05:57:03 -0800710 # Debug options
Ryan Cuif4f84be2012-07-09 18:50:41 -0700711 #
Brian Harringfec89fe2012-09-23 07:30:54 -0700712 # Temporary hack; in place till --dry-run replaces --debug.
713 # pylint: disable=W0212
Brian Harring009db502012-10-10 02:21:37 -0700714 group = parser.debug_group
Brian Harringfec89fe2012-09-23 07:30:54 -0700715 debug = [x for x in group.option_list if x._long_opts == ['--debug']][0]
David Jameseecba232014-06-11 11:35:11 -0700716 debug.help += ' Currently functions as --dry-run in addition.'
Brian Harringfec89fe2012-09-23 07:30:54 -0700717 debug.pass_through = True
Brian Harring3fec5a82012-03-01 05:57:03 -0800718 group.add_option('--notee', action='store_false', dest='tee', default=True,
David Jameseecba232014-06-11 11:35:11 -0700719 help=('Disable logging and internal tee process. Primarily '
720 'used for debugging cbuildbot itself.'))
Brian Harring3fec5a82012-03-01 05:57:03 -0800721 return parser
722
723
Ryan Cui85867972012-02-23 18:21:49 -0800724def _FinishParsing(options, args):
725 """Perform some parsing tasks that need to take place after optparse.
726
727 This function needs to be easily testable! Keep it free of
728 environment-dependent code. Put more detailed usage validation in
729 _PostParseCheck().
Brian Harring3fec5a82012-03-01 05:57:03 -0800730
731 Args:
Matt Tennant759e2352013-09-27 15:14:44 -0700732 options: The options object returned by optparse
733 args: The args object returned by optparse
Brian Harring3fec5a82012-03-01 05:57:03 -0800734 """
Ryan Cui41023d92012-11-13 19:59:50 -0800735 # Populate options.pass_through_args.
736 accepted, _ = commandline.FilteringParser.FilterArgs(
737 options.parsed_args, lambda x: x.opt_inst.pass_through)
738 options.pass_through_args.extend(accepted)
Brian Harring07039b52012-05-13 17:56:47 -0700739
Brian Harring3fec5a82012-03-01 05:57:03 -0800740 if options.chrome_root:
741 if options.chrome_rev != constants.CHROME_REV_LOCAL:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700742 cros_build_lib.Die('Chrome rev must be %s if chrome_root is set.' %
743 constants.CHROME_REV_LOCAL)
David Jamesa0a664e2013-02-13 09:52:01 -0800744 elif options.chrome_rev == constants.CHROME_REV_LOCAL:
745 cros_build_lib.Die('Chrome root must be set if chrome_rev is %s.' %
746 constants.CHROME_REV_LOCAL)
Brian Harring3fec5a82012-03-01 05:57:03 -0800747
748 if options.chrome_version:
749 if options.chrome_rev != constants.CHROME_REV_SPEC:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700750 cros_build_lib.Die('Chrome rev must be %s if chrome_version is set.' %
751 constants.CHROME_REV_SPEC)
David Jamesa0a664e2013-02-13 09:52:01 -0800752 elif options.chrome_rev == constants.CHROME_REV_SPEC:
753 cros_build_lib.Die(
754 'Chrome rev must not be %s if chrome_version is not set.'
755 % constants.CHROME_REV_SPEC)
Brian Harring3fec5a82012-03-01 05:57:03 -0800756
David James9e27e662013-02-14 13:42:43 -0800757 patches = bool(options.gerrit_patches or options.local_patches or
758 options.rietveld_patches)
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700759 if options.remote:
760 if options.local:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700761 cros_build_lib.Die('Cannot specify both --remote and --local')
Ryan Cui54da0702012-04-19 18:38:08 -0700762
Don Garrett5af1d262014-05-16 15:49:37 -0700763 # options.channels is a convenient way to detect payloads builds.
Prathmesh Prabhu0bc7f122015-07-25 17:09:26 -0700764 if (not options.list and not options.buildbot and not options.channels and
765 not patches):
Gaurav Shahed0399e2014-02-03 13:36:22 -0800766 prompt = ('No patches were provided; are you sure you want to just '
767 'run a remote build of %s?' % (
768 options.branch if options.branch else 'ToT'))
769 if not cros_build_lib.BooleanPrompt(prompt=prompt, default=False):
Brian Harring7cc4b932012-11-01 16:58:55 -0700770 cros_build_lib.Die('Must provide patches when running with --remote.')
Brian Harring3fec5a82012-03-01 05:57:03 -0800771
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700772 # --debug needs to be explicitly passed through for remote invocations.
773 release_mode_with_patches = (options.buildbot and patches and
774 '--debug' not in options.pass_through_args)
775 else:
776 if len(args) > 1:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700777 cros_build_lib.Die('Multiple configs not supported if not running with '
Brian Harringf1aad832012-07-18 10:46:39 -0700778 '--remote. Got %r', args)
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700779
Ryan Cui79319ab2012-05-21 12:59:18 -0700780 if options.slaves:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700781 cros_build_lib.Die('Cannot use --slaves if not running with --remote.')
Ryan Cui79319ab2012-05-21 12:59:18 -0700782
Ryan Cuieaa9efd2012-04-25 17:56:45 -0700783 release_mode_with_patches = (options.buildbot and patches and
784 not options.debug)
785
David James5734ea32012-08-15 20:23:49 -0700786 # When running in release mode, make sure we are running with checked-in code.
787 # We want checked-in cbuildbot/scripts to prevent errors, and we want to build
788 # a release image with checked-in code for CrOS packages.
789 if release_mode_with_patches:
790 cros_build_lib.Die(
791 'Cannot provide patches when running with --buildbot!')
792
Ryan Cuiba41ad32012-03-08 17:15:29 -0800793 if options.buildbot and options.remote_trybot:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700794 cros_build_lib.Die(
795 '--buildbot and --remote-trybot cannot be used together.')
Ryan Cuiba41ad32012-03-08 17:15:29 -0800796
Ryan Cui85867972012-02-23 18:21:49 -0800797 # Record whether --debug was set explicitly vs. it was inferred.
798 options.debug_forced = False
799 if options.debug:
800 options.debug_forced = True
Ryan Cui88b901c2013-06-21 11:35:30 -0700801 if not options.debug:
Ryan Cui16ca5812012-03-08 20:34:27 -0800802 # We don't set debug by default for
803 # 1. --buildbot invocations.
804 # 2. --remote invocations, because it needs to push changes to the tryjob
805 # repo.
806 options.debug = not options.buildbot and not options.remote
Brian Harring3fec5a82012-03-01 05:57:03 -0800807
Ryan Cui1c13a252012-10-16 15:00:16 -0700808 # Record the configs targeted.
809 options.build_targets = args[:]
810
Ryan Cui88b901c2013-06-21 11:35:30 -0700811 if constants.BRANCH_UTIL_CONFIG in options.build_targets:
Ryan Cui7bd8db62013-08-08 16:29:51 -0700812 if options.remote:
813 cros_build_lib.Die(
Matt Tennanta130ea32013-12-19 09:38:39 -0800814 'Running %s as a remote tryjob is not yet supported.',
815 constants.BRANCH_UTIL_CONFIG)
Ryan Cui88b901c2013-06-21 11:35:30 -0700816 if len(options.build_targets) > 1:
817 cros_build_lib.Die(
Matt Tennanta130ea32013-12-19 09:38:39 -0800818 'Cannot run %s with any other configs.',
819 constants.BRANCH_UTIL_CONFIG)
Ryan Cui88b901c2013-06-21 11:35:30 -0700820 if not options.branch_name:
821 cros_build_lib.Die(
Matt Tennanta130ea32013-12-19 09:38:39 -0800822 'Must specify --branch-name with the %s config.',
823 constants.BRANCH_UTIL_CONFIG)
824 if options.branch and options.branch != options.branch_name:
Ryan Cui88b901c2013-06-21 11:35:30 -0700825 cros_build_lib.Die(
Matt Tennanta130ea32013-12-19 09:38:39 -0800826 'If --branch is specified with the %s config, it must'
827 ' have the same value as --branch-name.',
828 constants.BRANCH_UTIL_CONFIG)
829
830 exclusive_opts = {'--version': options.force_version,
831 '--delete-branch': options.delete_branch,
David Jameseecba232014-06-11 11:35:11 -0700832 '--rename-to': options.rename_to}
Matt Tennanta130ea32013-12-19 09:38:39 -0800833 if 1 != sum(1 for x in exclusive_opts.values() if x):
834 cros_build_lib.Die('When using the %s config, you must'
835 ' specifiy one and only one of the following'
836 ' options: %s.', constants.BRANCH_UTIL_CONFIG,
837 ', '.join(exclusive_opts.keys()))
838
839 # When deleting or renaming a branch, the --branch and --nobootstrap
840 # options are implied.
841 if options.delete_branch or options.rename_to:
842 if not options.branch:
Ralph Nathan03047282015-03-23 11:09:32 -0700843 logging.info('Automatically enabling sync to branch %s for this %s '
844 'flow.', options.branch_name,
845 constants.BRANCH_UTIL_CONFIG)
Matt Tennanta130ea32013-12-19 09:38:39 -0800846 options.branch = options.branch_name
847 if options.bootstrap:
Ralph Nathan03047282015-03-23 11:09:32 -0700848 logging.info('Automatically disabling bootstrap step for this %s flow.',
849 constants.BRANCH_UTIL_CONFIG)
Matt Tennanta130ea32013-12-19 09:38:39 -0800850 options.bootstrap = False
851
Ryan Cui88b901c2013-06-21 11:35:30 -0700852 elif any([options.delete_branch, options.rename_to, options.branch_name]):
853 cros_build_lib.Die(
854 'Cannot specify --delete-branch, --rename-to or --branch-name when not '
855 'running the %s config', constants.BRANCH_UTIL_CONFIG)
856
Brian Harring3fec5a82012-03-01 05:57:03 -0800857
Brian Harring1d7ba942012-04-24 06:37:18 -0700858# pylint: disable=W0613
Don Garrett4af20982015-05-29 19:02:23 -0700859def _PostParseCheck(parser, options, args, site_config):
Ryan Cui85867972012-02-23 18:21:49 -0800860 """Perform some usage validation after we've parsed the arguments
Brian Harring3fec5a82012-03-01 05:57:03 -0800861
Ryan Cui85867972012-02-23 18:21:49 -0800862 Args:
Aviv Keshet9e4236b2013-12-13 13:07:50 -0800863 parser: Option parser that was used to parse arguments.
864 options: The options returned by optparse.
865 args: The args returned by optparse.
Don Garrett4af20982015-05-29 19:02:23 -0700866 site_config: config_lib.SiteConfig containing all config info.
Ryan Cui85867972012-02-23 18:21:49 -0800867 """
Don Garrett0a873e02015-06-30 17:55:10 -0700868 if not args:
869 parser.error('Invalid usage: no configuration targets provided.'
870 'Use -h to see usage. Use -l to list supported configs.')
871
Ryan Cuie1e4e662012-05-21 16:39:46 -0700872 if not options.branch:
David James97d95872012-11-16 15:09:56 -0800873 options.branch = git.GetChromiteTrackingBranch()
Ryan Cuie1e4e662012-05-21 16:39:46 -0700874
Brian Harringae0a5322012-09-15 01:46:51 -0700875 if not repository.IsARepoRoot(options.sourceroot):
876 if options.local_patches:
877 raise Exception('Could not find repo checkout at %s!'
878 % options.sourceroot)
879
David Jamesac8c2a72013-02-13 18:44:33 -0800880 # Because the default cache dir depends on other options, FindCacheDir
881 # always returns None, and we setup the default here.
Brian Harringae0a5322012-09-15 01:46:51 -0700882 if options.cache_dir is None:
883 # Note, options.sourceroot is set regardless of the path
884 # actually existing.
David Jamesac8c2a72013-02-13 18:44:33 -0800885 if options.buildroot is not None:
Brian Harringae0a5322012-09-15 01:46:51 -0700886 options.cache_dir = os.path.join(options.buildroot, '.cache')
David Jamesac8c2a72013-02-13 18:44:33 -0800887 elif os.path.exists(options.sourceroot):
888 options.cache_dir = os.path.join(options.sourceroot, '.cache')
Brian Harringae0a5322012-09-15 01:46:51 -0700889 else:
890 options.cache_dir = parser.FindCacheDir(parser, options)
891 options.cache_dir = os.path.abspath(options.cache_dir)
Brian Harring8c1d7b12012-10-04 17:36:32 -0700892 parser.ConfigureCacheDir(options.cache_dir)
Brian Harringae0a5322012-09-15 01:46:51 -0700893
Yu-Ju Hong2c066762013-10-28 14:05:08 -0700894 osutils.SafeMakedirsNonRoot(options.cache_dir)
Ryan Cui5ba7e152012-05-10 14:36:52 -0700895
Brian Harring609dc4e2012-05-07 02:17:44 -0700896 if options.local_patches:
Brian Harring1d7ba942012-04-24 06:37:18 -0700897 options.local_patches = _CheckLocalPatches(
Brian Harring609dc4e2012-05-07 02:17:44 -0700898 options.sourceroot, options.local_patches)
Brian Harring1d7ba942012-04-24 06:37:18 -0700899
900 default = os.environ.get('CBUILDBOT_DEFAULT_MODE')
901 if (default and not any([options.local, options.buildbot,
902 options.remote, options.remote_trybot])):
Ralph Nathan03047282015-03-23 11:09:32 -0700903 logging.info('CBUILDBOT_DEFAULT_MODE=%s env var detected, using it.'
904 % default)
Brian Harring1d7ba942012-04-24 06:37:18 -0700905 default = default.lower()
906 if default == 'local':
907 options.local = True
908 elif default == 'remote':
909 options.remote = True
910 elif default == 'buildbot':
911 options.buildbot = True
912 else:
Brian Harring1b8c4c82012-05-29 23:03:04 -0700913 cros_build_lib.Die("CBUILDBOT_DEFAULT_MODE value %s isn't supported. "
914 % default)
Ryan Cui85867972012-02-23 18:21:49 -0800915
Matt Tennant763497d2014-01-17 16:45:54 -0800916 # Ensure that all args are legitimate config targets.
Chris Sosa55cdc942014-04-16 13:08:37 -0700917 invalid_targets = []
Matt Tennant763497d2014-01-17 16:45:54 -0800918 for arg in args:
Don Garrett4af20982015-05-29 19:02:23 -0700919 if arg not in site_config:
Chris Sosa55cdc942014-04-16 13:08:37 -0700920 invalid_targets.append(arg)
Ralph Nathan59900422015-03-24 10:41:17 -0700921 logging.error('No such configuraton target: "%s".', arg)
Don Garrett4bb21682014-03-03 16:16:23 -0800922 continue
923
Don Garrett4af20982015-05-29 19:02:23 -0700924 build_config = site_config[arg]
925
Don Garrett5af1d262014-05-16 15:49:37 -0700926 is_payloads_build = build_config.build_type == constants.PAYLOADS_TYPE
927
928 if options.channels and not is_payloads_build:
Don Garrett4bb21682014-03-03 16:16:23 -0800929 cros_build_lib.Die('--channel must only be used with a payload config,'
930 ' not target (%s).' % arg)
Matt Tennant763497d2014-01-17 16:45:54 -0800931
Don Garrett5af1d262014-05-16 15:49:37 -0700932 if not options.channels and is_payloads_build:
933 cros_build_lib.Die('payload configs (%s) require --channel to do anything'
934 ' useful.' % arg)
935
Matt Tennant2c192032014-01-16 13:49:28 -0800936 # The --version option is not compatible with an external target unless the
937 # --buildbot option is specified. More correctly, only "paladin versions"
938 # will work with external targets, and those are only used with --buildbot.
939 # If --buildbot is specified, then user should know what they are doing and
940 # only specify a version that will work. See crbug.com/311648.
Don Garrett4bb21682014-03-03 16:16:23 -0800941 if (options.force_version and
942 not (options.buildbot or build_config.internal)):
943 cros_build_lib.Die('Cannot specify --version without --buildbot for an'
944 ' external target (%s).' % arg)
Matt Tennant2c192032014-01-16 13:49:28 -0800945
Chris Sosa55cdc942014-04-16 13:08:37 -0700946 if invalid_targets:
947 cros_build_lib.Die('One or more invalid configuration targets specified. '
948 'You can check the available configs by running '
949 '`cbuildbot --list --all`')
Matt Tennant763497d2014-01-17 16:45:54 -0800950
Ryan Cui85867972012-02-23 18:21:49 -0800951
Don Garrett0a873e02015-06-30 17:55:10 -0700952def _ParseCommandLine(parser, argv):
Ryan Cui85867972012-02-23 18:21:49 -0800953 """Completely parse the commandline arguments"""
Brian Harring3fec5a82012-03-01 05:57:03 -0800954 (options, args) = parser.parse_args(argv)
Brian Harring37e559b2012-05-22 20:47:32 -0700955
Matt Tennant763497d2014-01-17 16:45:54 -0800956 # Strip out null arguments.
957 # TODO(rcui): Remove when buildbot is fixed
958 args = [arg for arg in args if arg]
959
Brian Harring37e559b2012-05-22 20:47:32 -0700960 if options.output_api_version:
Mike Frysinger383367e2014-09-16 15:06:17 -0400961 print(constants.REEXEC_API_VERSION)
Brian Harring37e559b2012-05-22 20:47:32 -0700962 sys.exit(0)
963
Ryan Cui85867972012-02-23 18:21:49 -0800964 _FinishParsing(options, args)
965 return options, args
966
967
Aviv Keshet420de512015-05-18 14:28:48 -0700968_ENVIRONMENT_PROD = 'prod'
969_ENVIRONMENT_DEBUG = 'debug'
970_ENVIRONMENT_STANDALONE = 'standalone'
971
972
973def _GetRunEnvironment(options, build_config):
974 """Determine whether this is a prod/debug/standalone run."""
975 # TODO(akeshet): This is a temporary workaround to make sure that the cidb
976 # is not used on waterfalls that the db schema does not support (in particular
977 # the chromeos.chrome waterfall).
978 # See crbug.com/406940
979 waterfall = os.environ.get('BUILDBOT_MASTERNAME', '')
980 if not waterfall in constants.CIDB_KNOWN_WATERFALLS:
981 return _ENVIRONMENT_STANDALONE
982
983 # TODO(akeshet): Clean up this code once we have better defined flags to
984 # specify on-or-off waterfall and on-or-off production runs of cbuildbot.
985 # See crbug.com/331417
986
987 # --buildbot runs should use the production services, unless the --debug flag
988 # is also present.
989 if options.buildbot:
990 if options.debug:
991 return _ENVIRONMENT_DEBUG
992 else:
993 return _ENVIRONMENT_PROD
994
995 # --remote-trybot runs should use the debug services, with the exception of
996 # pre-cq builds, which should use the production services.
997 if options.remote_trybot:
998 if build_config['pre_cq']:
999 return _ENVIRONMENT_PROD
1000 else:
1001 return _ENVIRONMENT_DEBUG
1002
1003 # If neither --buildbot nor --remote-trybot flag was used, don't use external
1004 # services.
1005 return _ENVIRONMENT_STANDALONE
1006
1007
Gabe Blackde694a32015-02-19 15:11:11 -08001008def _SetupConnections(options, build_config):
1009 """Set up CIDB and graphite connections using the appropriate Setup call.
Aviv Keshet2982af52014-08-13 16:07:57 -07001010
1011 Args:
1012 options: Command line options structure.
Aviv Keshet64133022014-08-25 15:50:52 -07001013 build_config: Config object for this build.
Aviv Keshet2982af52014-08-13 16:07:57 -07001014 """
Aviv Keshet420de512015-05-18 14:28:48 -07001015 # Outline:
1016 # 1) Based on options and build_config, decide whether we are a production
1017 # run, debug run, or standalone run.
1018 # 2) Set up cidb instance accordingly.
1019 # 3) Update topology info from cidb, so that any other service set up can use
1020 # topology.
1021 # 4) Set up any other services.
1022 run_type = _GetRunEnvironment(options, build_config)
1023
1024 if run_type == _ENVIRONMENT_PROD:
1025 cidb.CIDBConnectionFactory.SetupProdCidb()
Paul Hobbsd5a0f812016-07-26 16:10:31 -07001026 context = ts_mon_config.SetupTsMonGlobalState('cbuildbot', indirect=True)
Aviv Keshet420de512015-05-18 14:28:48 -07001027 elif run_type == _ENVIRONMENT_DEBUG:
1028 cidb.CIDBConnectionFactory.SetupDebugCidb()
Paul Hobbsd5a0f812016-07-26 16:10:31 -07001029 context = ts_mon_config.TrivialContextManager()
Aviv Keshet420de512015-05-18 14:28:48 -07001030 else:
Aviv Keshet62d1a0e2014-08-22 21:16:13 -07001031 cidb.CIDBConnectionFactory.SetupNoCidb()
Paul Hobbsd5a0f812016-07-26 16:10:31 -07001032 context = ts_mon_config.TrivialContextManager()
Aviv Keshet62d1a0e2014-08-22 21:16:13 -07001033
Aviv Keshet420de512015-05-18 14:28:48 -07001034 db = cidb.CIDBConnectionFactory.GetCIDBConnectionForBuilder()
1035 topology.FetchTopologyFromCIDB(db)
Aviv Keshet64133022014-08-25 15:50:52 -07001036
Aviv Keshet420de512015-05-18 14:28:48 -07001037 if run_type == _ENVIRONMENT_PROD:
1038 graphite.ESMetadataFactory.SetupProd()
1039 graphite.StatsFactory.SetupProd()
1040 elif run_type == _ENVIRONMENT_DEBUG:
1041 graphite.ESMetadataFactory.SetupReadOnly()
1042 graphite.StatsFactory.SetupDebug()
1043 else:
1044 graphite.ESMetadataFactory.SetupReadOnly()
1045 graphite.StatsFactory.SetupMock()
Aviv Keshet2982af52014-08-13 16:07:57 -07001046
Paul Hobbsd5a0f812016-07-26 16:10:31 -07001047 return context
1048
Aviv Keshet2982af52014-08-13 16:07:57 -07001049
Don Garrettbfb33532015-07-07 16:56:43 -07001050def _FetchInitialBootstrapConfigRepo(repo_url, branch_name):
1051 """Fetch the TOT site config repo, if necessary to start bootstrap."""
1052
1053 # If we are part of a repo checkout, the manifest stages control things.
1054 if git.FindRepoDir(constants.SOURCE_ROOT):
1055 return
1056
1057 # We must be part of a bootstrap chromite checkout, probably on a buildbot.
1058
1059 # If the config directory already exists, we have started the bootstrap
1060 # process. Assume the boostrap stage did the right thing, and leave the config
1061 # directory alone.
1062 if os.path.exists(constants.SITE_CONFIG_DIR):
1063 return
1064
1065 # We are part of a clean chromite checkout (buildbot always cleans chromite
1066 # before launching us), so create the initial site config checkout.
1067 logging.info('Fetching Config Repo: %s', repo_url)
1068 git.Clone(constants.SITE_CONFIG_DIR, repo_url)
1069
1070 if branch_name:
1071 git.RunGit(constants.SITE_CONFIG_DIR, ['checkout', branch_name])
1072
Don Garrett05fa8102015-09-10 14:48:05 -07001073 # Clear the cached SiteConfig, if there was one.
1074 config_lib.ClearConfigCache()
Don Garrettbfb33532015-07-07 16:56:43 -07001075
Matt Tennant759e2352013-09-27 15:14:44 -07001076# TODO(build): This function is too damn long.
Ryan Cui85867972012-02-23 18:21:49 -08001077def main(argv):
David James59a0a2b2013-03-22 14:04:44 -07001078 # Turn on strict sudo checks.
1079 cros_build_lib.STRICT_SUDO = True
1080
Ryan Cui85867972012-02-23 18:21:49 -08001081 # Set umask to 022 so files created by buildbot are readable.
Mike Frysinger60ec1012013-10-21 00:11:10 -04001082 os.umask(0o22)
Ryan Cui85867972012-02-23 18:21:49 -08001083
Ryan Cui85867972012-02-23 18:21:49 -08001084 parser = _CreateParser()
Don Garrett0a873e02015-06-30 17:55:10 -07001085 options, args = _ParseCommandLine(parser, argv)
1086
Don Garrettbfb33532015-07-07 16:56:43 -07001087 if options.buildbot and options.config_repo:
1088 _FetchInitialBootstrapConfigRepo(options.config_repo, options.branch)
Don Garrettde81cc72015-07-07 13:23:28 -07001089
1090 if options.config_repo:
Don Garrettbfb33532015-07-07 16:56:43 -07001091 # Ensure expected config file is present.
1092 if not os.path.exists(constants.SITE_CONFIG_FILE):
1093 cros_build_lib.Die('Unabled to find: %s', constants.SITE_CONFIG_FILE)
Don Garrettde81cc72015-07-07 13:23:28 -07001094
Don Garrettb85658c2015-06-30 19:07:22 -07001095 # Fetch our site_config now, because we need it to do anything else.
Don Garrettde81cc72015-07-07 13:23:28 -07001096 site_config = config_lib.GetConfig()
Don Garrettb85658c2015-06-30 19:07:22 -07001097
Don Garrett0a873e02015-06-30 17:55:10 -07001098 if options.list:
1099 _PrintValidConfigs(site_config, options.print_all)
1100 sys.exit(0)
Brian Harring3fec5a82012-03-01 05:57:03 -08001101
Don Garrett4af20982015-05-29 19:02:23 -07001102 _PostParseCheck(parser, options, args, site_config)
Brian Harring3fec5a82012-03-01 05:57:03 -08001103
Mike Frysinger8fd67dc2012-12-03 23:51:18 -05001104 cros_build_lib.AssertOutsideChroot()
Zdenek Behan98ec2fb2012-08-31 17:12:18 +02001105
Prathmesh Prabhu51d774f2015-07-17 15:11:49 -07001106 if options.enable_buildbot_tags:
1107 logging.EnableBuildbotMarkers()
Brian Harring3fec5a82012-03-01 05:57:03 -08001108 if options.remote:
Ralph Nathan23a12212015-03-25 10:27:54 -07001109 logging.getLogger().setLevel(logging.WARNING)
Ryan Cui16ca5812012-03-08 20:34:27 -08001110
Brian Harring3fec5a82012-03-01 05:57:03 -08001111 # Verify configs are valid.
Dan Shi0bdb7132013-07-30 16:22:12 -07001112 # If hwtest flag is enabled, post a warning that HWTest step may fail if the
1113 # specified board is not a released platform or it is a generic overlay.
Brian Harring3fec5a82012-03-01 05:57:03 -08001114 for bot in args:
Don Garrett4af20982015-05-29 19:02:23 -07001115 build_config = site_config[bot]
Aviv Kesheta96a2a92013-02-05 17:21:51 -08001116 if options.hwtest:
Ralph Nathan446aee92015-03-23 14:44:56 -07001117 logging.warning(
Dan Shi0bdb7132013-07-30 16:22:12 -07001118 'If %s is not a released platform or it is a generic overlay, '
1119 'the HWTest step will most likely not run; please ask the lab '
1120 'team for help if this is unexpected.' % build_config['boards'])
Brian Harring3fec5a82012-03-01 05:57:03 -08001121
1122 # Verify gerrit patches are valid.
Mike Frysinger383367e2014-09-16 15:06:17 -04001123 print('Verifying patches...')
Mike Frysingerb80d0192015-02-04 22:08:59 -05001124 patch_pool = trybot_patch_pool.TrybotPatchPool.FromOptions(
1125 gerrit_patches=options.gerrit_patches,
1126 local_patches=options.local_patches,
1127 sourceroot=options.sourceroot,
1128 remote_patches=options.remote_patches)
Ryan Cui16d9e1f2012-05-11 10:50:18 -07001129
Ryan Cuieaa9efd2012-04-25 17:56:45 -07001130 # --debug need to be explicitly passed through for remote invocations.
1131 if options.buildbot and '--debug' not in options.pass_through_args:
1132 _ConfirmRemoteBuildbotRun()
1133
Mike Frysinger383367e2014-09-16 15:06:17 -04001134 print('Submitting tryjob...')
Paul Hobbsd5a0f812016-07-26 16:10:31 -07001135 with _SetupConnections(options, build_config):
1136 tryjob = remote_try.RemoteTryJob(options, args, patch_pool.local_patches)
1137 tryjob.Submit(testjob=options.test_tryjob, dryrun=False)
Mike Frysinger383367e2014-09-16 15:06:17 -04001138 print('Tryjob submitted!')
1139 print(('Go to %s to view the status of your job.'
1140 % tryjob.GetTrybotWaterfallLink()))
Brian Harring3fec5a82012-03-01 05:57:03 -08001141 sys.exit(0)
Matt Tennant759e2352013-09-27 15:14:44 -07001142
Ryan Cui54da0702012-04-19 18:38:08 -07001143 elif (not options.buildbot and not options.remote_trybot
1144 and not options.resume and not options.local):
Mike Frysinger5672a2a2014-05-31 22:29:57 -04001145 cros_build_lib.Die('Please use --remote or --local to run trybots')
Brian Harring3fec5a82012-03-01 05:57:03 -08001146
Ningning Xiac691e432016-08-11 14:52:59 -07001147 elif options.buildbot and not options.debug:
1148 if not cros_build_lib.HostIsCIBuilder():
1149 # Cannot run --buildbot if both --debug and --remote aren't specified.
1150 cros_build_lib.Die('This host isn\'t a continuous-integration builder.')
1151
Matt Tennant759e2352013-09-27 15:14:44 -07001152 # Only one config arg is allowed in this mode, which was confirmed earlier.
Ryan Cui8be16062012-04-24 12:05:26 -07001153 bot_id = args[-1]
Don Garrett4af20982015-05-29 19:02:23 -07001154 build_config = site_config[bot_id]
Brian Harring3fec5a82012-03-01 05:57:03 -08001155
Don Garrettbbd7b552014-05-16 13:15:21 -07001156 # TODO: Re-enable this block when reference_repo support handles this
1157 # properly. (see chromium:330775)
1158 # if options.reference_repo is None:
1159 # repo_path = os.path.join(options.sourceroot, '.repo')
1160 # # If we're being run from a repo checkout, reuse the repo's git pool to
1161 # # cut down on sync time.
1162 # if os.path.exists(repo_path):
1163 # options.reference_repo = options.sourceroot
1164
1165 if options.reference_repo:
David Jamesdac7a912013-11-18 11:14:44 -08001166 if not os.path.exists(options.reference_repo):
1167 parser.error('Reference path %s does not exist'
1168 % (options.reference_repo,))
1169 elif not os.path.exists(os.path.join(options.reference_repo, '.repo')):
1170 parser.error('Reference path %s does not look to be the base of a '
1171 'repo checkout; no .repo exists in the root.'
1172 % (options.reference_repo,))
1173
Brian Harringf11bf682012-05-14 15:53:43 -07001174 if (options.buildbot or options.remote_trybot) and not options.resume:
Brian Harring470f6112012-03-02 11:47:10 -08001175 if not options.cgroups:
Ryan Cuid4a24212012-04-04 18:08:12 -07001176 parser.error('Options --buildbot/--remote-trybot and --nocgroups cannot '
1177 'be used together. Cgroup support is required for '
1178 'buildbot/remote-trybot mode.')
Mike Frysingera78a56e2012-11-20 06:02:30 -05001179 if not cgroups.Cgroup.IsSupported():
Ryan Cuid4a24212012-04-04 18:08:12 -07001180 parser.error('Option --buildbot/--remote-trybot was given, but this '
1181 'system does not support cgroups. Failing.')
Brian Harring3fec5a82012-03-01 05:57:03 -08001182
David Jamesaad5cc72012-10-26 15:03:13 -07001183 missing = osutils.FindMissingBinaries(_BUILDBOT_REQUIRED_BINARIES)
Brian Harring351ce442012-03-09 16:38:14 -08001184 if missing:
David Jameseecba232014-06-11 11:35:11 -07001185 parser.error('Option --buildbot/--remote-trybot requires the following '
Ryan Cuid4a24212012-04-04 18:08:12 -07001186 "binaries which couldn't be found in $PATH: %s"
Brian Harring351ce442012-03-09 16:38:14 -08001187 % (', '.join(missing)))
1188
David Jamesdac7a912013-11-18 11:14:44 -08001189 if options.reference_repo:
1190 options.reference_repo = os.path.abspath(options.reference_repo)
1191
Brian Harring3fec5a82012-03-01 05:57:03 -08001192 if not options.buildroot:
1193 if options.buildbot:
Gaurav Shah59abcb52014-12-09 15:27:11 -08001194 parser.error('Please specify a buildroot with the --buildroot option.')
Matt Tennantd55b1f42012-04-13 14:15:01 -07001195
Ryan Cui5ba7e152012-05-10 14:36:52 -07001196 options.buildroot = _DetermineDefaultBuildRoot(options.sourceroot,
1197 build_config['internal'])
Brian Harring470f6112012-03-02 11:47:10 -08001198 # We use a marker file in the buildroot to indicate the user has
1199 # consented to using this directory.
1200 if not os.path.exists(repository.GetTrybotMarkerPath(options.buildroot)):
1201 _ConfirmBuildRoot(options.buildroot)
Brian Harring3fec5a82012-03-01 05:57:03 -08001202
1203 # Sanity check of buildroot- specifically that it's not pointing into the
1204 # midst of an existing repo since git-repo doesn't support nesting.
Brian Harring3fec5a82012-03-01 05:57:03 -08001205 if (not repository.IsARepoRoot(options.buildroot) and
David James13a69c92013-05-09 10:37:42 -07001206 git.FindRepoDir(options.buildroot)):
Brian Harring3fec5a82012-03-01 05:57:03 -08001207 parser.error('Configured buildroot %s points into a repository checkout, '
1208 'rather than the root of it. This is not supported.'
1209 % options.buildroot)
1210
Chris Sosab5ea3b42012-10-25 15:25:20 -07001211 if not options.log_dir:
1212 options.log_dir = os.path.join(options.buildroot, _DEFAULT_LOG_DIR)
1213
Brian Harringd166aaf2012-05-14 18:31:53 -07001214 log_file = None
1215 if options.tee:
Chris Sosab5ea3b42012-10-25 15:25:20 -07001216 log_file = os.path.join(options.log_dir, _BUILDBOT_LOG_FILE)
1217 osutils.SafeMakedirs(options.log_dir)
Brian Harringd166aaf2012-05-14 18:31:53 -07001218 _BackupPreviousLog(log_file)
1219
Brian Harring1b8c4c82012-05-29 23:03:04 -07001220 with cros_build_lib.ContextManagerStack() as stack:
Ningning Xia96876d52016-03-31 10:26:39 -07001221 options.preserve_paths = set()
David Jamescebc7272013-07-17 16:45:05 -07001222 if log_file is not None:
1223 # We don't want the critical section to try to clean up the tee process,
1224 # so we run Tee (forked off) outside of it. This prevents a deadlock
1225 # because the Tee process only exits when its pipe is closed, and the
1226 # critical section accidentally holds on to that file handle.
1227 stack.Add(tee.Tee, log_file)
1228 options.preserve_paths.add(_DEFAULT_LOG_DIR)
1229
Brian Harringc2d09d92012-05-13 22:03:15 -07001230 critical_section = stack.Add(cleanup.EnforcedCleanupSection)
1231 stack.Add(sudo.SudoKeepAlive)
Brian Harringd166aaf2012-05-14 18:31:53 -07001232
Brian Harringc2d09d92012-05-13 22:03:15 -07001233 if not options.resume:
Brian Harring2bf55e12012-05-13 21:31:55 -07001234 # If we're in resume mode, use our parents tempdir rather than
1235 # nesting another layer.
David James4bc13702013-03-26 08:08:04 -07001236 stack.Add(osutils.TempDir, prefix='cbuildbot-tmp', set_global=True)
David Jameseecba232014-06-11 11:35:11 -07001237 logging.debug('Cbuildbot tempdir is %r.', os.environ.get('TMP'))
Brian Harringd166aaf2012-05-14 18:31:53 -07001238
Brian Harringc2d09d92012-05-13 22:03:15 -07001239 if options.cgroups:
1240 stack.Add(cgroups.SimpleContainChildren, 'cbuildbot')
Brian Harringa184efa2012-03-04 11:51:25 -08001241
Brian Harringc2d09d92012-05-13 22:03:15 -07001242 # Mark everything between EnforcedCleanupSection and here as having to
1243 # be rolled back via the contextmanager cleanup handlers. This
1244 # ensures that sudo bits cannot outlive cbuildbot, that anything
1245 # cgroups would kill gets killed, etc.
David Jamesfb3aac92013-10-16 13:26:52 -07001246 stack.Add(critical_section.ForkWatchdog)
Brian Harringd166aaf2012-05-14 18:31:53 -07001247
Brian Harringc2d09d92012-05-13 22:03:15 -07001248 if not options.buildbot:
Don Garrett6747eba2015-06-25 16:00:16 -07001249 build_config = config_lib.OverrideConfigForTrybot(
Don Garrett4bb21682014-03-03 16:16:23 -08001250 build_config, options)
Brian Harringc2d09d92012-05-13 22:03:15 -07001251
Aviv Kesheta0159be2013-12-12 13:56:28 -08001252 if options.mock_tree_status is not None:
Prathmesh Prabhud51d7502014-12-21 01:42:55 -08001253 stack.Add(mock.patch.object, tree_status, '_GetStatus',
Aviv Kesheta0159be2013-12-12 13:56:28 -08001254 return_value=options.mock_tree_status)
1255
Aviv Keshetcf9c2722014-02-25 15:15:10 -08001256 if options.mock_slave_status is not None:
1257 with open(options.mock_slave_status, 'r') as f:
Aviv Keshet4e750022014-03-07 16:50:34 -08001258 mock_statuses = pickle.load(f)
1259 for key, value in mock_statuses.iteritems():
1260 mock_statuses[key] = manifest_version.BuilderStatus(**value)
Yu-Ju Hongd0fda382014-05-09 15:28:24 -07001261 stack.Add(mock.patch.object,
1262 completion_stages.MasterSlaveSyncCompletionStage,
1263 '_FetchSlaveStatuses',
1264 return_value=mock_statuses)
Aviv Keshetcf9c2722014-02-25 15:15:10 -08001265
Paul Hobbsd5a0f812016-07-26 16:10:31 -07001266 stack.Add(_SetupConnections, options, build_config)
Don Garrettb4318362014-10-03 15:49:36 -07001267 retry_stats.SetupStats()
Aviv Keshet2982af52014-08-13 16:07:57 -07001268
Aviv Keshet446f07f2016-03-08 11:32:31 -08001269 timeout_display_message = None
Prathmesh Prabhu80e05df2014-12-11 15:20:33 -08001270 # For master-slave builds: Update slave's timeout using master's published
1271 # deadline.
1272 if options.buildbot and options.master_build_id is not None:
1273 slave_timeout = None
1274 if cidb.CIDBConnectionFactory.IsCIDBSetup():
1275 cidb_handle = cidb.CIDBConnectionFactory.GetCIDBConnectionForBuilder()
1276 if cidb_handle:
1277 slave_timeout = cidb_handle.GetTimeToDeadline(options.master_build_id)
1278
1279 if slave_timeout is not None:
1280 # Cut me some slack. We artificially add a a small time here to the
1281 # slave_timeout because '0' is handled specially, and because we don't
1282 # want to timeout while trying to set things up.
1283 slave_timeout = slave_timeout + 20
1284 if options.timeout == 0 or slave_timeout < options.timeout:
1285 logging.info('Updating slave build timeout to %d seconds enforced '
Ralph Nathan03047282015-03-23 11:09:32 -07001286 'by the master', slave_timeout)
Prathmesh Prabhu80e05df2014-12-11 15:20:33 -08001287 options.timeout = slave_timeout
Aviv Keshet84017582016-05-18 16:59:59 -07001288 timeout_display_message = (
1289 'This build has reached the timeout deadline set by the master. '
1290 'Either this stage or a previous one took too long (see stage '
1291 'timing historical summary in ReportStage) or the build failed '
1292 'to start on time.')
Prathmesh Prabhu80e05df2014-12-11 15:20:33 -08001293 else:
1294 logging.warning('Could not get master deadline for master-slave build. '
1295 'Can not set slave timeout.')
1296
1297 if options.timeout > 0:
Aviv Keshet446f07f2016-03-08 11:32:31 -08001298 stack.Add(timeout_util.FatalTimeout, options.timeout,
1299 timeout_display_message)
Ningning Xiaad483542016-05-24 12:27:21 -07001300 try:
1301 _RunBuildStagesWrapper(options, site_config, build_config)
1302 except failures_lib.ExitEarlyException as ex:
1303 # This build finished successfully. Do not re-raise ExitEarlyException.
1304 logging.info('One stage exited early: %s', ex)