blob: 4e3d1f3cba4ef268a728a1c6b34d0a77bd91857f [file] [log] [blame]
David McMahon49302942010-02-18 16:55:35 -08001# Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
rspangler@google.comd74220d2009-10-09 20:56:14 +00002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5# Common constants for build scripts
6# This must evaluate properly for both /bin/bash and /bin/sh
7
8# All scripts should die on error unless commands are specifically excepted
9# by prefixing with '!' or surrounded by 'set +e' / 'set -e'.
10# TODO: Re-enable this once shflags is less prone to dying.
11#set -e
12
13# The number of jobs to pass to tools that can run in parallel (such as make
14# and dpkg-buildpackage
Greg Spencer798d75f2011-02-01 22:04:49 -080015NUM_JOBS=$(grep -c "^processor" /proc/cpuinfo)
rspangler@google.comd74220d2009-10-09 20:56:14 +000016
Simon Glass142ca062011-02-09 13:39:43 -080017# True if we have the 'pv' utility - also set up COMMON_PV_CAT for convenience
18COMMON_PV_OK=1
19COMMON_PV_CAT=pv
20pv -V >/dev/null 2>&1 || COMMON_PV_OK=0
21if [ $COMMON_PV_OK -eq 0 ]; then
22 COMMON_PV_CAT=cat
23fi
24
Greg Spencer798d75f2011-02-01 22:04:49 -080025# Make sure we have the location and name of the calling script, using
26# the current value if it is already set.
27SCRIPT_LOCATION=${SCRIPT_LOCATION:-$(dirname "$(readlink -f "$0")")}
28SCRIPT_NAME=${SCRIPT_NAME:-$(basename "$0")}
rspangler@google.comd74220d2009-10-09 20:56:14 +000029
Anton Staaf30acb0b2011-01-26 16:00:20 -080030# Detect whether we're inside a chroot or not
31if [ -e /etc/debian_chroot ]
rspangler@google.comd74220d2009-10-09 20:56:14 +000032then
Anton Staaf30acb0b2011-01-26 16:00:20 -080033 INSIDE_CHROOT=1
rspangler@google.comd74220d2009-10-09 20:56:14 +000034else
Anton Staaf30acb0b2011-01-26 16:00:20 -080035 INSIDE_CHROOT=0
rspangler@google.comd74220d2009-10-09 20:56:14 +000036fi
37
Anton Staaf30acb0b2011-01-26 16:00:20 -080038# Construct a list of possible locations for the source tree. This list is
39# based on various environment variables and globals that may have been set
40# by the calling script.
41function get_gclient_root_list() {
42 if [ $INSIDE_CHROOT -eq 1 ]; then
43 echo "/home/${USER}/trunk"
44
45 if [ -n "${SUDO_USER}" ]; then echo "/home/${SUDO_USER}/trunk"; fi
46 fi
47
48 if [ -n "${COMMON_SH}" ]; then echo "$(dirname "$COMMON_SH")/../.."; fi
49 if [ -n "${BASH_SOURCE}" ]; then echo "$(dirname "$BASH_SOURCE")/../.."; fi
50}
51
52# Based on the list of possible source locations we set GCLIENT_ROOT if it is
53# not already defined by looking for a src directory in each seach path
54# location. If we do not find a valid looking root we error out.
55function get_gclient_root() {
56 if [ -n "${GCLIENT_ROOT}" ]; then
57 return
58 fi
59
60 for path in $(get_gclient_root_list); do
61 if [ -d "${path}/src" ]; then
62 GCLIENT_ROOT=${path}
63 break
64 fi
65 done
66
67 if [ -z "${GCLIENT_ROOT}" ]; then
68 # Using dash or sh, we don't know where we are. $0 refers to the calling
69 # script, not ourselves, so that doesn't help us.
70 echo "Unable to determine location for common.sh. If you are sourcing"
71 echo "common.sh from a script run via dash or sh, you must do it in the"
72 echo "following way:"
73 echo ' COMMON_SH="$(dirname "$0")/../../scripts/common.sh"'
74 echo ' . "$COMMON_SH"'
75 echo "where the first line is the relative path from your script to"
76 echo "common.sh."
77 exit 1
78 fi
79}
80
81# Find root of source tree
82get_gclient_root
83
rspangler@google.comd74220d2009-10-09 20:56:14 +000084# Canonicalize the directories for the root dir and the calling script.
85# readlink is part of coreutils and should be present even in a bare chroot.
tedbo4f44d9e2010-01-08 17:26:11 -080086# This is better than just using
rspangler@google.comd74220d2009-10-09 20:56:14 +000087# FOO = "$(cd $FOO ; pwd)"
tedbo4f44d9e2010-01-08 17:26:11 -080088# since that leaves symbolic links intact.
rspangler@google.comd74220d2009-10-09 20:56:14 +000089# Note that 'realpath' is equivalent to 'readlink -f'.
Mike Frysingera1a06ab2011-08-10 11:40:30 -040090SCRIPT_LOCATION=$(readlink -f "$SCRIPT_LOCATION")
91GCLIENT_ROOT=$(readlink -f "$GCLIENT_ROOT")
rspangler@google.comd74220d2009-10-09 20:56:14 +000092
93# Other directories should always be pathed down from GCLIENT_ROOT.
94SRC_ROOT="$GCLIENT_ROOT/src"
95SRC_INTERNAL="$GCLIENT_ROOT/src-internal"
96SCRIPTS_DIR="$SRC_ROOT/scripts"
97
98# Load developer's custom settings. Default location is in scripts dir,
99# since that's available both inside and outside the chroot. By convention,
100# settings from this file are variables starting with 'CHROMEOS_'
101CHROMEOS_DEV_SETTINGS="${CHROMEOS_DEV_SETTINGS:-$SCRIPTS_DIR/.chromeos_dev}"
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400102if [ -f "$CHROMEOS_DEV_SETTINGS" ]; then
rspangler@google.comd74220d2009-10-09 20:56:14 +0000103 # Turn on exit-on-error during custom settings processing
Greg Spencer798d75f2011-02-01 22:04:49 -0800104 SAVE_OPTS=$(set +o)
rspangler@google.comd74220d2009-10-09 20:56:14 +0000105 set -e
106
107 # Read settings
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400108 . "$CHROMEOS_DEV_SETTINGS"
rspangler@google.comd74220d2009-10-09 20:56:14 +0000109
110 # Restore previous state of exit-on-error
111 eval "$SAVE_OPTS"
112fi
113
114# Load shflags
Zdenek Behan07d24222011-11-02 00:46:25 +0000115# NOTE: This code snippet is in particular used by the au-generator (which
116# stores shflags in ./lib/shflags/) and should not be touched.
117if [ -f "${SCRIPTS_DIR}/lib/shflags/shflags" ]; then
118 . "${SCRIPTS_DIR}/lib/shflags/shflags"
119else
120 . ./lib/shflags/shflags || die "Couldn't find shflags"
121fi
rspangler@google.comd74220d2009-10-09 20:56:14 +0000122
Bill Richardson10d27c22010-01-20 13:38:50 -0800123# Our local mirror
124DEFAULT_CHROMEOS_SERVER=${CHROMEOS_SERVER:-"http://build.chromium.org/mirror"}
rspangler@google.comd74220d2009-10-09 20:56:14 +0000125
Bill Richardson10d27c22010-01-20 13:38:50 -0800126# Upstream mirrors and build suites come in 2 flavors
127# DEV - development chroot, used to build the chromeos image
128# IMG - bootable image, to run on actual hardware
rspangler@google.comd74220d2009-10-09 20:56:14 +0000129
Bill Richardson10d27c22010-01-20 13:38:50 -0800130DEFAULT_DEV_MIRROR=${CHROMEOS_DEV_MIRROR:-"${DEFAULT_CHROMEOS_SERVER}/ubuntu"}
131DEFAULT_DEV_SUITE=${CHROMEOS_DEV_SUITE:-"karmic"}
132
133DEFAULT_IMG_MIRROR=${CHROMEOS_IMG_MIRROR:-"${DEFAULT_CHROMEOS_SERVER}/ubuntu"}
134DEFAULT_IMG_SUITE=${CHROMEOS_IMG_SUITE:-"karmic"}
rspangler@google.comd74220d2009-10-09 20:56:14 +0000135
136# Default location for chroot
137DEFAULT_CHROOT_DIR=${CHROMEOS_CHROOT_DIR:-"$GCLIENT_ROOT/chroot"}
138
139# All output files from build should go under $DEFAULT_BUILD_ROOT, so that
140# they don't pollute the source directory.
141DEFAULT_BUILD_ROOT=${CHROMEOS_BUILD_ROOT:-"$SRC_ROOT/build"}
142
David McMahon49302942010-02-18 16:55:35 -0800143# Set up a global ALL_BOARDS value
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400144if [ -d "$SRC_ROOT/overlays" ]; then
145 ALL_BOARDS=$(cd "$SRC_ROOT/overlays"; \
146 ls -1d overlay-* 2>&- | sed 's,overlay-,,g')
David Rochberg3b910702010-12-02 10:45:21 -0500147fi
David McMahon49302942010-02-18 16:55:35 -0800148# Strip CR
149ALL_BOARDS=$(echo $ALL_BOARDS)
150# Set a default BOARD
151#DEFAULT_BOARD=x86-generic # or...
152DEFAULT_BOARD=$(echo $ALL_BOARDS | awk '{print $NF}')
153
David Jamesff072012010-11-30 13:22:05 -0800154# Enable --fast by default.
Greg Spencer798d75f2011-02-01 22:04:49 -0800155DEFAULT_FAST=${FLAGS_TRUE}
David James03668642010-07-28 17:08:29 -0700156
Chris Sosab0f57322011-10-25 03:07:23 +0000157# Directory to store built images. Should be set by sourcing script when used.
158BUILD_DIR=
Simon Glass142ca062011-02-09 13:39:43 -0800159
160# Standard filenames
Chris Sosab0f57322011-10-25 03:07:23 +0000161CHROMEOS_BASE_IMAGE_NAME="chromiumos_base_image.bin"
Simon Glass142ca062011-02-09 13:39:43 -0800162CHROMEOS_IMAGE_NAME="chromiumos_image.bin"
Chris Sosab0f57322011-10-25 03:07:23 +0000163CHROMEOS_DEVELOPER_IMAGE_NAME="chromiumos_image.bin"
Simon Glass142ca062011-02-09 13:39:43 -0800164CHROMEOS_TEST_IMAGE_NAME="chromiumos_test_image.bin"
Chris Sosab885b802011-02-16 15:33:11 -0800165CHROMEOS_FACTORY_TEST_IMAGE_NAME="chromiumos_factory_image.bin"
Chris Sosab0f57322011-10-25 03:07:23 +0000166CHROMEOS_FACTORY_INSTALL_SHIM_NAME="factory_install_shim.bin"
Simon Glass142ca062011-02-09 13:39:43 -0800167
rspangler@google.comd74220d2009-10-09 20:56:14 +0000168# Directory locations inside the dev chroot
169CHROOT_TRUNK_DIR="/home/$USER/trunk"
170
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700171# Install make for portage ebuilds. Used by build_image and gmergefs.
Chris Masoned11ce172010-11-09 14:22:08 -0800172# TODO: Is /usr/local/autotest-chrome still used by anyone?
Hung-Te Lind32c59f2012-01-19 19:54:01 +0800173COMMON_INSTALL_MASK="
Daniel Erate82f07c2010-12-21 13:39:22 -0800174 *.a
175 *.la
176 /etc/init.d
177 /etc/runlevels
178 /lib/rc
179 /usr/bin/Xnest
180 /usr/bin/Xvfb
181 /usr/include
182 /usr/lib/debug
183 /usr/lib/gcc
184 /usr/lib/gtk-2.0/include
185 /usr/lib/pkgconfig
Daniel Erate82f07c2010-12-21 13:39:22 -0800186 /usr/local/autotest-chrome
187 /usr/man
188 /usr/share/aclocal
189 /usr/share/doc
190 /usr/share/gettext
191 /usr/share/gtk-2.0
192 /usr/share/gtk-doc
193 /usr/share/info
194 /usr/share/man
195 /usr/share/openrc
196 /usr/share/pkgconfig
197 /usr/share/readline
Chris Wolfed13775f2011-07-26 16:34:38 -0400198 /usr/src
Daniel Erate82f07c2010-12-21 13:39:22 -0800199 "
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700200
Hung-Te Lind32c59f2012-01-19 19:54:01 +0800201# Mask for base, dev, and test images (build_image, build_image --test)
202DEFAULT_INSTALL_MASK="
203 $COMMON_INSTALL_MASK
204 /usr/local/autotest
205 "
206
207# Mask for factory test image (build_image --factory)
208FACTORY_TEST_INSTALL_MASK="
209 $COMMON_INSTALL_MASK
210 */.svn
211 */CVS
212 /usr/local/autotest/[^c]*
213 /usr/local/autotest/conmux
214 /usr/local/autotest/client/deps/chrome_test
215 /usr/local/autotest/client/deps/piglit
216 /usr/local/autotest/client/deps/pyauto_dep
217 /usr/local/autotest/client/deps/realtimecomm_*
218 /usr/local/autotest/client/site_tests/desktopui_PageCyclerTests
219 /usr/local/autotest/client/site_tests/graphics_WebGLConformance
220 /usr/local/autotest/client/site_tests/platform_ToolchainOptions
221 /usr/local/autotest/client/site_tests/realtimecomm_GTalk*
222 "
223
224# Mask for factory install shim (build_image --factory_install)
225FACTORY_SHIM_INSTALL_MASK="
226 $DEFAULT_INSTALL_MASK
227 /opt/[^g]*
Daniel Erate82f07c2010-12-21 13:39:22 -0800228 /opt/google/chrome
229 /opt/google/o3d
230 /opt/google/talkplugin
Daniel Erate82f07c2010-12-21 13:39:22 -0800231 /usr/lib/dri
232 /usr/lib/python2.6/test
Daniel Erate82f07c2010-12-21 13:39:22 -0800233 /usr/local/autotest-pkgs
234 /usr/share/X11
235 /usr/share/chewing
236 /usr/share/fonts
237 /usr/share/ibus-pinyin
238 /usr/share/libhangul
239 /usr/share/locale
240 /usr/share/m17n
241 /usr/share/mime
242 /usr/share/sounds
243 /usr/share/tts
244 /usr/share/zoneinfo
245 "
Tom Wai-Hong Tamf87a3672010-05-17 16:06:33 +0800246
Gaurav Shah786ceeb2011-07-26 15:03:37 -0700247# Determine and set up variables needed for fancy color output (if supported).
Gaurav Shah786ceeb2011-07-26 15:03:37 -0700248V_BOLD_RED=
249V_BOLD_GREEN=
250V_BOLD_YELLOW=
Mike Frysingerc2474ae2011-08-11 18:27:44 -0400251V_REVERSE=
252V_VIDOFF=
Gaurav Shah786ceeb2011-07-26 15:03:37 -0700253
254if tput colors >/dev/null 2>&1; then
Mike Frysingerc2474ae2011-08-11 18:27:44 -0400255 # order matters: we want VIDOFF last so that when we trace with `set -x`,
256 # our terminal doesn't bleed colors as bash dumps the values of vars.
Gaurav Shah786ceeb2011-07-26 15:03:37 -0700257 V_BOLD_RED="$(tput bold; tput setaf 1)"
258 V_BOLD_GREEN="$(tput bold; tput setaf 2)"
259 V_BOLD_YELLOW="$(tput bold; tput setaf 3)"
Mike Frysingerc2474ae2011-08-11 18:27:44 -0400260 V_REVERSE="$(tput rev)"
261 V_VIDOFF="$(tput sgr0)"
Gaurav Shah786ceeb2011-07-26 15:03:37 -0700262fi
David McMahonb7eb3a22010-02-09 14:07:40 -0800263
rspangler@google.comd74220d2009-10-09 20:56:14 +0000264# -----------------------------------------------------------------------------
265# Functions
266
Chris Sosaacada732010-02-23 15:20:03 -0800267function setup_board_warning {
tedbo373c3902010-04-12 10:52:40 -0700268 echo
269 echo "$V_REVERSE================= WARNING ======================$V_VIDOFF"
Chris Sosaacada732010-02-23 15:20:03 -0800270 echo
271 echo "*** No default board detected in " \
272 "$GCLIENT_ROOT/src/scripts/.default_board"
273 echo "*** Either run setup_board with default flag set"
274 echo "*** or echo |board_name| > $GCLIENT_ROOT/src/scripts/.default_board"
275 echo
276}
277
278
279# Sets the default board variable for calling script
280function get_default_board {
tedbo373c3902010-04-12 10:52:40 -0700281 DEFAULT_BOARD=
282
Chris Sosaacada732010-02-23 15:20:03 -0800283 if [ -f "$GCLIENT_ROOT/src/scripts/.default_board" ] ; then
Greg Spencer798d75f2011-02-01 22:04:49 -0800284 DEFAULT_BOARD=$(cat "$GCLIENT_ROOT/src/scripts/.default_board")
Mike Frysingerbc36d042011-12-19 16:01:09 -0500285 # Check for user typos like whitespace.
286 if [[ -n ${DEFAULT_BOARD//[a-zA-Z0-9-_]} ]] ; then
287 die ".default_board: invalid name detected; please fix:" \
288 "'${DEFAULT_BOARD}'"
289 fi
Chris Sosaacada732010-02-23 15:20:03 -0800290 fi
291}
292
293
Don Garrett640a0582010-05-04 16:54:28 -0700294# Enter a chroot and restart the current script if needed
295function restart_in_chroot_if_needed {
David Rochberg3b910702010-12-02 10:45:21 -0500296 # NB: Pass in ARGV: restart_in_chroot_if_needed "$@"
Greg Spencer798d75f2011-02-01 22:04:49 -0800297 if [ $INSIDE_CHROOT -ne 1 ]; then
Chris Sosafd2cdec2011-03-24 16:06:59 -0700298 # Get inside_chroot path for script.
299 local chroot_path="$(reinterpret_path_for_chroot "$0")"
Zdenek Behan2811c162011-08-13 00:47:38 +0200300 exec $GCLIENT_ROOT/chromite/bin/cros_sdk -- "$chroot_path" "$@"
Don Garrett640a0582010-05-04 16:54:28 -0700301 fi
302}
303
rspangler@google.comd74220d2009-10-09 20:56:14 +0000304# Fail unless we're inside the chroot. This guards against messing up your
305# workstation.
306function assert_inside_chroot {
Greg Spencer798d75f2011-02-01 22:04:49 -0800307 if [ $INSIDE_CHROOT -ne 1 ]; then
rspangler@google.comd74220d2009-10-09 20:56:14 +0000308 echo "This script must be run inside the chroot. Run this first:"
Zdenek Behan2811c162011-08-13 00:47:38 +0200309 echo " cros_sdk"
rspangler@google.comd74220d2009-10-09 20:56:14 +0000310 exit 1
311 fi
312}
313
314# Fail if we're inside the chroot. This guards against creating or entering
315# nested chroots, among other potential problems.
316function assert_outside_chroot {
Greg Spencer798d75f2011-02-01 22:04:49 -0800317 if [ $INSIDE_CHROOT -ne 0 ]; then
rspangler@google.comd74220d2009-10-09 20:56:14 +0000318 echo "This script must be run outside the chroot."
319 exit 1
320 fi
321}
322
derat@google.com86dcc8e2009-11-21 19:49:49 +0000323function assert_not_root_user {
Greg Spencer798d75f2011-02-01 22:04:49 -0800324 if [ $(id -u) = 0 ]; then
derat@google.com86dcc8e2009-11-21 19:49:49 +0000325 echo "This script must be run as a non-root user."
326 exit 1
327 fi
328}
329
Luigi Semenzato1f82e122010-03-23 12:43:08 -0700330# Check that all arguments are flags; that is, there are no remaining arguments
331# after parsing from shflags. Allow (with a warning) a single empty-string
332# argument.
333#
334# TODO: fix buildbot so that it doesn't pass the empty-string parameter,
335# then change this function.
336#
337# Usage: check_flags_only_and_allow_null_arg "$@" && set --
338function check_flags_only_and_allow_null_arg {
339 do_shift=1
340 if [[ $# == 1 && -z "$@" ]]; then
341 echo "$0: warning: ignoring null argument" >&2
342 shift
343 do_shift=0
344 fi
345 if [[ $# -gt 0 ]]; then
346 echo "error: invalid arguments: \"$@\"" >&2
347 flags_help
348 exit 1
349 fi
350 return $do_shift
351}
352
Chris Sosad9798522010-06-16 14:29:50 -0700353function info {
Chris Sosa131eaf52011-10-27 16:15:53 -0700354 echo -e >&2 "${V_BOLD_GREEN}INFO ${CROS_LOG_PREFIX:-""}: $@${V_VIDOFF}"
Will Drewry55b42c92010-10-20 15:44:11 -0500355}
Chris Sosad9798522010-06-16 14:29:50 -0700356
robotboy2ea03ac2010-02-11 15:30:55 -0800357function warn {
Chris Sosa131eaf52011-10-27 16:15:53 -0700358 echo -e >&2 "${V_BOLD_YELLOW}WARNING ${CROS_LOG_PREFIX:-""}: $@${V_VIDOFF}"
robotboy2ea03ac2010-02-11 15:30:55 -0800359}
360
361function error {
Chris Sosa131eaf52011-10-27 16:15:53 -0700362 echo -e >&2 "${V_BOLD_RED}ERROR ${CROS_LOG_PREFIX:-""}: $@${V_VIDOFF}"
robotboy2ea03ac2010-02-11 15:30:55 -0800363}
David James546747b2010-03-23 15:19:43 -0700364
365function die {
Chris Sosa131eaf52011-10-27 16:15:53 -0700366 error "$@"
David James546747b2010-03-23 15:19:43 -0700367 exit 1
368}
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700369
370# Retry an emerge command according to $FLAGS_retries
David Jamesfd7403a2010-07-09 12:00:17 -0700371# The $EMERGE_JOBS flags will only be added the first time the command is run
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700372function eretry () {
Greg Spencer798d75f2011-02-01 22:04:49 -0800373 local i
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700374 for i in $(seq $FLAGS_retries); do
Greg Spencer798d75f2011-02-01 22:04:49 -0800375 echo "Retrying $@"
376 "$@" $EMERGE_JOBS && return 0
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700377 done
Greg Spencer798d75f2011-02-01 22:04:49 -0800378 "$@" && return 0
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700379 return 1
380}
381
382# Removes single quotes around parameter
383# Arguments:
384# $1 - string which optionally has surrounding quotes
385# Returns:
386# None, but prints the string without quotes.
387function remove_quotes() {
388 echo "$1" | sed -e "s/^'//; s/'$//"
389}
tedbo373c3902010-04-12 10:52:40 -0700390
391# Writes stdin to the given file name as root using sudo in overwrite mode.
392#
393# $1 - The output file name.
394function sudo_clobber() {
395 sudo tee "$1" > /dev/null
396}
397
398# Writes stdin to the given file name as root using sudo in append mode.
399#
400# $1 - The output file name.
401function sudo_append() {
402 sudo tee -a "$1" > /dev/null
403}
robotboy98912212010-04-12 14:08:14 -0700404
Mike Frysinger286b5922011-09-28 11:59:53 -0400405# Execute multiple commands in a single sudo. Generally will speed things
406# up by avoiding multiple calls to `sudo`. If any commands fail, we will
407# call die with the failing command. We can handle a max of ~100 commands,
408# but hopefully no one will ever try that many at once.
409#
410# $@ - The commands to execute, one per arg.
411function sudo_multi() {
412 local i cmds
413
414 # Construct the shell code to execute. It'll be of the form:
415 # ... && ( ( command ) || exit <command index> ) && ...
416 # This way we know which command exited. The exit status of
417 # the underlying command is lost, but we never cared about it
418 # in the first place (other than it is non zero), so oh well.
419 for (( i = 1; i <= $#; ++i )); do
420 cmds+=" && ( ( ${!i} ) || exit $(( i + 10 )) )"
421 done
422
423 # Execute our constructed shell code.
424 sudo -- sh -c ":${cmds[*]}" && i=0 || i=$?
425
426 # See if this failed, and if so, print out the failing command.
427 if [[ $i -gt 10 ]]; then
428 : $(( i -= 10 ))
429 die "sudo_multi failed: ${!i}"
430 elif [[ $i -ne 0 ]]; then
431 die "sudo_multi failed for unknown reason $i"
432 fi
433}
434
Mike Frysinger1aa61242011-09-15 17:46:44 -0400435# Locate all mounts below a specified directory.
436#
437# $1 - The root tree.
438function sub_mounts() {
439 # Assume that `mount` outputs a list of mount points in the order
440 # that things were mounted (since it always has and hopefully always
441 # will). As such, we have to unmount in reverse order to cleanly
442 # unmount submounts (think /dev/pts and /dev).
443 mount | \
Mike Frysingerb2e897b2011-09-22 17:04:35 -0400444 awk -v path="$1" -v len="${#1}" \
445 '(substr($3, 1, len) == path) { print $3 }' | \
Mike Frysinger1aa61242011-09-15 17:46:44 -0400446 tac
447}
448
robotboy98912212010-04-12 14:08:14 -0700449# Unmounts a directory, if the unmount fails, warn, and then lazily unmount.
450#
451# $1 - The path to unmount.
Mike Frysinger1aa61242011-09-15 17:46:44 -0400452function safe_umount_tree {
453 local mounts=$(sub_mounts "$1")
robotboy98912212010-04-12 14:08:14 -0700454
Mike Frysingere8aec372011-09-21 00:03:22 -0400455 # Hmm, this shouldn't normally happen, but anything is possible.
456 if [ -z "${mounts}" ] ; then
457 return 0
458 fi
459
Mike Frysinger1aa61242011-09-15 17:46:44 -0400460 # First try to unmount in one shot to speed things up.
461 if sudo umount -d ${mounts}; then
462 return 0
463 fi
robotboy98912212010-04-12 14:08:14 -0700464
Mike Frysinger1aa61242011-09-15 17:46:44 -0400465 # Well that didn't work, so lazy unmount remaining ones.
466 mounts=$(sub_mounts "$1")
467 warn "Failed to unmount ${mounts}"
468 warn "Doing a lazy unmount"
469 if ! sudo umount -d -l ${mounts}; then
470 mounts=$(sub_mounts "$1")
471 die "Failed to lazily unmount ${mounts}"
robotboy98912212010-04-12 14:08:14 -0700472 fi
473}
Chris Sosa702618f2010-05-14 12:52:32 -0700474
Chris Sosad4455022010-05-20 10:14:06 -0700475# Fixes symlinks that are incorrectly prefixed with the build root ${1}
476# rather than the real running root '/'.
477# TODO(sosa) - Merge setup - cleanup below with this method.
478fix_broken_symlinks() {
479 local build_root="${1}"
480 local symlinks=$(find "${build_root}/usr/local" -lname "${build_root}/*")
Greg Spencer798d75f2011-02-01 22:04:49 -0800481 local symlink
Chris Sosad4455022010-05-20 10:14:06 -0700482 for symlink in ${symlinks}; do
483 echo "Fixing ${symlink}"
484 local target=$(ls -l "${symlink}" | cut -f 2 -d '>')
485 # Trim spaces from target (bashism).
486 target=${target/ /}
487 # Make new target (removes rootfs prefix).
488 new_target=$(echo ${target} | sed "s#${build_root}##")
489
490 echo "Fixing symlink ${symlink}"
491 sudo unlink "${symlink}"
492 sudo ln -sf "${new_target}" "${symlink}"
493 done
494}
495
Chris Sosa702618f2010-05-14 12:52:32 -0700496# Sets up symlinks for the developer root. It is necessary to symlink
497# usr and local since the developer root is mounted at /usr/local and
498# applications expect to be installed under /usr/local/bin, etc.
499# This avoids packages installing into /usr/local/usr/local/bin.
500# ${1} specifies the symlink target for the developer root.
501# ${2} specifies the symlink target for the var directory.
502# ${3} specifies the location of the stateful partition.
503setup_symlinks_on_root() {
504 # Give args better names.
505 local dev_image_target=${1}
506 local var_target=${2}
507 local dev_image_root="${3}/dev_image"
508
509 # If our var target is actually the standard var, we are cleaning up the
510 # symlinks (could also check for /usr/local for the dev_image_target).
511 if [ ${var_target} = "/var" ]; then
512 echo "Cleaning up /usr/local symlinks for ${dev_image_root}"
513 else
514 echo "Setting up symlinks for /usr/local for ${dev_image_root}"
515 fi
516
517 # Set up symlinks that should point to ${dev_image_target}.
Greg Spencer798d75f2011-02-01 22:04:49 -0800518 local path
Chris Sosa702618f2010-05-14 12:52:32 -0700519 for path in usr local; do
520 if [ -h "${dev_image_root}/${path}" ]; then
521 sudo unlink "${dev_image_root}/${path}"
522 elif [ -e "${dev_image_root}/${path}" ]; then
523 die "${dev_image_root}/${path} should be a symlink if exists"
524 fi
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400525 sudo ln -s "${dev_image_target}" "${dev_image_root}/${path}"
Chris Sosa702618f2010-05-14 12:52:32 -0700526 done
527
528 # Setup var symlink.
529 if [ -h "${dev_image_root}/var" ]; then
530 sudo unlink "${dev_image_root}/var"
531 elif [ -e "${dev_image_root}/var" ]; then
532 die "${dev_image_root}/var should be a symlink if it exists"
533 fi
534
535 sudo ln -s "${var_target}" "${dev_image_root}/var"
536}
Nick Sandersd2509272010-06-16 03:50:04 -0700537
Will Drewry55b42c92010-10-20 15:44:11 -0500538# These two helpers clobber the ro compat value in our root filesystem.
539#
540# When the system is built with --enable_rootfs_verification, bit-precise
541# integrity checking is performed. That precision poses a usability issue on
542# systems that automount partitions with recognizable filesystems, such as
543# ext2/3/4. When the filesystem is mounted 'rw', ext2 metadata will be
544# automatically updated even if no other writes are performed to the
545# filesystem. In addition, ext2+ does not support a "read-only" flag for a
546# given filesystem. That said, forward and backward compatibility of
547# filesystem features are supported by tracking if a new feature breaks r/w or
548# just write compatibility. We abuse the read-only compatibility flag[1] in
549# the filesystem header by setting the high order byte (le) to FF. This tells
550# the kernel that features R24-R31 are all enabled. Since those features are
551# undefined on all ext-based filesystem, all standard kernels will refuse to
552# mount the filesystem as read-write -- only read-only[2].
553#
554# [1] 32-bit flag we are modifying:
555# http://git.chromium.org/cgi-bin/gitweb.cgi?p=kernel.git;a=blob;f=include/linux/ext2_fs.h#l417
556# [2] Mount behavior is enforced here:
557# http://git.chromium.org/cgi-bin/gitweb.cgi?p=kernel.git;a=blob;f=fs/ext2/super.c#l857
558#
559# N.B., if the high order feature bits are used in the future, we will need to
560# revisit this technique.
561disable_rw_mount() {
562 local rootfs="$1"
563 local offset="${2-0}" # in bytes
Will Drewry9b7cb512010-10-20 18:11:24 -0500564 local ro_compat_offset=$((0x464 + 3)) # Set 'highest' byte
565 printf '\377' |
Will Drewry55b42c92010-10-20 15:44:11 -0500566 sudo dd of="$rootfs" seek=$((offset + ro_compat_offset)) \
567 conv=notrunc count=1 bs=1
568}
569
570enable_rw_mount() {
571 local rootfs="$1"
572 local offset="${2-0}"
Will Drewry9b7cb512010-10-20 18:11:24 -0500573 local ro_compat_offset=$((0x464 + 3)) # Set 'highest' byte
574 printf '\000' |
Will Drewry55b42c92010-10-20 15:44:11 -0500575 sudo dd of="$rootfs" seek=$((offset + ro_compat_offset)) \
576 conv=notrunc count=1 bs=1
577}
578
Nick Sandersd2509272010-06-16 03:50:04 -0700579# Get current timestamp. Assumes common.sh runs at startup.
580start_time=$(date +%s)
581
582# Print time elsapsed since start_time.
583print_time_elapsed() {
Greg Spencer798d75f2011-02-01 22:04:49 -0800584 local end_time=$(date +%s)
585 local elapsed_seconds=$(($end_time - $start_time))
586 local minutes=$(($elapsed_seconds / 60))
587 local seconds=$(($elapsed_seconds % 60))
Olof Johansson6d491382010-08-09 16:05:50 -0500588 echo "Elapsed time: ${minutes}m${seconds}s"
Nick Sandersd2509272010-06-16 03:50:04 -0700589}
Doug Anderson0c9e88d2010-10-19 14:49:39 -0700590
Anton Staaf9bcd8412011-01-24 15:27:14 -0800591# The board and variant command line options can be used in a number of ways
592# to specify the board and variant. The board can encode both pieces of
593# information separated by underscores. Or the variant can be passed using
594# the separate variant option. This function extracts the canonical board and
595# variant information and provides it in the BOARD, VARIANT and BOARD_VARIANT
596# variables.
597get_board_and_variant() {
598 local flags_board="${1}"
599 local flags_variant="${2}"
600
601 BOARD=$(echo "$flags_board" | cut -d '_' -f 1)
602 VARIANT=${flags_variant:-$(echo "$flags_board" | cut -s -d '_' -f 2)}
603
604 if [ -n "$VARIANT" ]; then
605 BOARD_VARIANT="${BOARD}_${VARIANT}"
606 else
607 BOARD_VARIANT="${BOARD}"
608 fi
609}
Simon Glass142ca062011-02-09 13:39:43 -0800610
611# This function converts a chromiumos image into a test image, either
612# in place or by copying to a new test image filename first. It honors
613# the following flags (see mod_image_for_test.sh)
614#
615# --factory
616# --factory_install
617# --force_copy
618#
619# On entry, pass the directory containing the image, and the image filename
620# On exit, it puts the pathname of the resulting test image into
621# CHROMEOS_RETURN_VAL
622# (yes this is ugly, but perhaps less ugly than the alternatives)
623#
624# Usage:
625# SRC_IMAGE=$(prepare_test_image "directory" "imagefile")
626prepare_test_image() {
627 # If we're asked to modify the image for test, then let's make a copy and
628 # modify that instead.
629 # Check for manufacturing image.
630 local args
631
632 if [ ${FLAGS_factory} -eq ${FLAGS_TRUE} ]; then
633 args="--factory"
634 fi
635
636 # Check for install shim.
637 if [ ${FLAGS_factory_install} -eq ${FLAGS_TRUE} ]; then
638 args="--factory_install"
639 fi
640
641 # Check for forcing copy of image
642 if [ ${FLAGS_force_copy} -eq ${FLAGS_TRUE} ]; then
643 args="${args} --force_copy"
644 fi
645
646 # Modify the image for test, creating a new test image
647 "${SCRIPTS_DIR}/mod_image_for_test.sh" --board=${FLAGS_board} \
648 --image="$1/$2" --noinplace ${args}
649
650 # From now on we use the just-created test image
Simon Glass6e448ae2011-03-03 11:20:54 -0800651 if [ ${FLAGS_factory} -eq ${FLAGS_TRUE} ]; then
652 CHROMEOS_RETURN_VAL="$1/${CHROMEOS_FACTORY_TEST_IMAGE_NAME}"
653 else
654 CHROMEOS_RETURN_VAL="$1/${CHROMEOS_TEST_IMAGE_NAME}"
655 fi
Simon Glass142ca062011-02-09 13:39:43 -0800656}
Anton Staaf6f5262d2011-03-02 09:35:54 -0800657
658# Check that the specified file exists. If the file path is empty or the file
659# doesn't exist on the filesystem generate useful error messages. Otherwise
660# show the user the name and path of the file that will be used. The padding
661# parameter can be used to tabulate multiple name:path pairs. For example:
662#
663# check_for_file "really long name" "...:" "file.foo"
664# check_for_file "short name" ".........:" "another.bar"
665#
666# Results in the following output:
667#
668# Using really long name...: file.foo
669# Using short name.........: another.bar
670#
671# If tabulation is not required then passing "" for padding generates the
672# output "Using <name> <path>"
673check_for_file() {
674 local name=$1
675 local padding=$2
676 local path=$3
677
678 if [ -z "${path}" ]; then
679 die "No ${name} file specified."
680 fi
681
682 if [ ! -e "${path}" ]; then
683 die "No ${name} file found at: ${path}"
684 else
685 info "Using ${name}${padding} ${path}"
686 fi
687}
688
689# Check that the specified tool exists. If it does not exist in the PATH
690# generate a useful error message indicating how to install the ebuild
691# that contains the required tool.
692check_for_tool() {
693 local tool=$1
694 local ebuild=$2
695
696 if ! which "${tool}" >/dev/null ; then
697 error "The ${tool} utility was not found in your path. Run the following"
698 error "command in your chroot to install it: sudo -E emerge ${ebuild}"
699 exit 1
700 fi
701}
Chris Sosafd2cdec2011-03-24 16:06:59 -0700702
703# Reinterprets path from outside the chroot for use inside.
704# Returns "" if "" given.
705# $1 - The path to reinterpret.
706function reinterpret_path_for_chroot() {
707 if [ $INSIDE_CHROOT -ne 1 ]; then
708 if [ -z "${1}" ]; then
709 echo ""
710 else
711 local path_abs_path=$(readlink -f "${1}")
712 local gclient_root_abs_path=$(readlink -f "${GCLIENT_ROOT}")
713
714 # Strip the repository root from the path.
715 local relative_path=$(echo ${path_abs_path} \
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400716 | sed "s:${gclient_root_abs_path}/::")
Chris Sosafd2cdec2011-03-24 16:06:59 -0700717
718 if [ "${relative_path}" = "${path_abs_path}" ]; then
719 die "Error reinterpreting path. Path ${1} is not within source tree."
720 fi
721
722 # Prepend the chroot repository path.
723 echo "/home/${USER}/trunk/${relative_path}"
724 fi
725 else
726 # Path is already inside the chroot :).
727 echo "${1}"
728 fi
729}
Gabe Black83d8b822011-08-01 17:50:09 -0700730
David James0ea96e42011-08-03 11:53:50 -0700731function emerge_custom_kernel() {
732 local install_root="$1"
733 local root=${FLAGS_build_root}/${FLAGS_board}
734 local tmp_pkgdir=${root}/custom-packages
735
736 # Clean up any leftover state in custom directories.
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400737 sudo rm -rf "${tmp_pkgdir}"
David James0ea96e42011-08-03 11:53:50 -0700738
739 # Update chromeos-initramfs to contain the latest binaries from the build
740 # tree. This is basically just packaging up already-built binaries from
741 # $root. We are careful not to muck with the existing prebuilts so that
742 # prebuilts can be uploaded in parallel.
743 # TODO(davidjames): Implement ABI deps so that chromeos-initramfs will be
744 # rebuilt automatically when its dependencies change.
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400745 sudo -E PKGDIR="${tmp_pkgdir}" $EMERGE_BOARD_CMD -1 \
David James0ea96e42011-08-03 11:53:50 -0700746 chromeos-base/chromeos-initramfs || die "Cannot emerge chromeos-initramfs"
747
748 # Verify all dependencies of the kernel are installed. This should be a
749 # no-op, but it's good to check in case a developer didn't run
750 # build_packages.
751 local kernel=$(portageq-${FLAGS_board} expand_virtual ${root} virtual/kernel)
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400752 sudo -E PKGDIR="${tmp_pkgdir}" $EMERGE_BOARD_CMD --onlydeps \
David James0ea96e42011-08-03 11:53:50 -0700753 ${kernel} || die "Cannot emerge kernel dependencies"
754
755 # Build the kernel. This uses the standard root so that we can pick up the
756 # initramfs from there. But we don't actually install the kernel to the
757 # standard root, because that'll muck up the kernel debug symbols there,
758 # which we want to upload in parallel.
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400759 sudo -E PKGDIR="${tmp_pkgdir}" $EMERGE_BOARD_CMD --buildpkgonly \
David James0ea96e42011-08-03 11:53:50 -0700760 ${kernel} || die "Cannot emerge kernel"
761
762 # Install the custom kernel to the provided install root.
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400763 sudo -E PKGDIR="${tmp_pkgdir}" $EMERGE_BOARD_CMD --usepkgonly \
David James0ea96e42011-08-03 11:53:50 -0700764 --root=${install_root} ${kernel} || die "Cannot emerge kernel to root"
765}