blob: f2e0009fcd7bf712729497b897e6498bae4a3bed [file] [log] [blame]
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001#!/bin/bash
Chris Sosa2b736002012-02-12 16:16:08 -08002# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
rspangler@google.comd74220d2009-10-09 20:56:14 +00003# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
rspangler@google.comd74220d2009-10-09 20:56:14 +00006# All scripts should die on error unless commands are specifically excepted
7# by prefixing with '!' or surrounded by 'set +e' / 'set -e'.
rspangler@google.comd74220d2009-10-09 20:56:14 +00008
9# The number of jobs to pass to tools that can run in parallel (such as make
10# and dpkg-buildpackage
Alex Deymo135faa12015-01-21 11:20:13 -080011if [[ -z ${NUM_JOBS:-} ]]; then
Brian Harring28bb01f2012-05-09 15:26:09 -070012 NUM_JOBS=$(grep -c "^processor" /proc/cpuinfo)
13fi
14# Ensure that any sub scripts we invoke get the max proc count.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050015export NUM_JOBS
rspangler@google.comd74220d2009-10-09 20:56:14 +000016
Gilad Arnold8f861c02013-01-09 09:19:51 -080017# Returns the pv command if it's available, otherwise plain-old cat. Note that
18# this function echoes the command, rather than running it, so it can be used
19# as an argument to other commands (like sudo).
20pv_cat_cmd() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050021 if type -P pv >&/dev/null; then
Gilad Arnold8f861c02013-01-09 09:19:51 -080022 # Limit pv's output to 80 columns, for readability.
23 local term_cols=$(stty size 2>/dev/null | cut -d' ' -f2)
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050024 if [[ ${term_cols:-0} -gt 80 ]]; then
Seigo Nonaka89b7b592013-05-21 17:30:18 +090025 echo pv -w 80 -B 4m
Gilad Arnold8f861c02013-01-09 09:19:51 -080026 else
Seigo Nonaka89b7b592013-05-21 17:30:18 +090027 echo pv -B 4m
Gilad Arnold8f861c02013-01-09 09:19:51 -080028 fi
29 else
30 echo cat
31 fi
32}
Simon Glass142ca062011-02-09 13:39:43 -080033
Greg Spencer798d75f2011-02-01 22:04:49 -080034# Make sure we have the location and name of the calling script, using
35# the current value if it is already set.
Alex Deymo7ba56be2015-02-12 14:28:06 -080036: ${SCRIPT_LOCATION:=$(dirname "$(readlink -f -- "$0")")}
37: ${SCRIPT_NAME:=$(basename -- "$0")}
rspangler@google.comd74220d2009-10-09 20:56:14 +000038
Anton Staaf30acb0b2011-01-26 16:00:20 -080039# Detect whether we're inside a chroot or not
David James47e0e8a2015-03-11 15:24:10 -070040CHROOT_VERSION_FILE=/etc/cros_chroot_version
41if [[ -e ${CHROOT_VERSION_FILE} ]]; then
Anton Staaf30acb0b2011-01-26 16:00:20 -080042 INSIDE_CHROOT=1
rspangler@google.comd74220d2009-10-09 20:56:14 +000043else
Anton Staaf30acb0b2011-01-26 16:00:20 -080044 INSIDE_CHROOT=0
rspangler@google.comd74220d2009-10-09 20:56:14 +000045fi
46
Mike Frysinger669b28b2012-02-07 18:01:00 -050047# Determine and set up variables needed for fancy color output (if supported).
48V_BOLD_RED=
49V_BOLD_GREEN=
50V_BOLD_YELLOW=
51V_REVERSE=
52V_VIDOFF=
53
Steve Fungff6345d2014-02-04 16:14:55 -080054if [[ -t 1 ]] && tput colors >&/dev/null; then
Mike Frysinger669b28b2012-02-07 18:01:00 -050055 # order matters: we want VIDOFF last so that when we trace with `set -x`,
56 # our terminal doesn't bleed colors as bash dumps the values of vars.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050057 V_BOLD_RED=$(tput bold; tput setaf 1)
58 V_BOLD_GREEN=$(tput bold; tput setaf 2)
59 V_BOLD_YELLOW=$(tput bold; tput setaf 3)
60 V_REVERSE=$(tput rev)
61 V_VIDOFF=$(tput sgr0)
Mike Frysinger669b28b2012-02-07 18:01:00 -050062fi
63
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050064# Turn on bash debug support if available for backtraces.
65shopt -s extdebug 2>/dev/null
Brian Harring7f175a52012-03-02 05:37:00 -080066
Marc Herbertff9fa312015-01-16 17:18:53 -080067# Output a backtrace. Optional parameter allows hiding the last
68# frame(s) so functions like "die()" can hide their additional
69# frame(s) if they wish.
70dump_trace() {
71 # Default = 0 hidden frames: show everything except dump_trace
72 # frame itself.
73 local hidden_frames=${1:-0}
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050074 local j n p func src line args
75 p=${#BASH_ARGV[@]}
Marc Herbertff9fa312015-01-16 17:18:53 -080076
Marc Herbertf51cf8f2015-01-20 13:56:00 -080077 error "$(date)"
Marc Herbertee1fcbb2015-05-28 10:37:46 -070078 error "$(ps f -o pgid,ppid,pid,etime,cputime,%cpu,command)"
Marc Herbertf51cf8f2015-01-20 13:56:00 -080079
Marc Herbertff9fa312015-01-16 17:18:53 -080080 # Frame 0 is ourselves so it's always suppressed / does not count.
81 for (( n = ${#FUNCNAME[@]}; n > hidden_frames; --n )); do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050082 func=${FUNCNAME[${n} - 1]}
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050083 line=${BASH_LINENO[${n} - 1]}
84 args=
85 if [[ -z ${BASH_ARGC[${n} -1]} ]]; then
86 args='(args unknown, no debug available)'
Brian Harring7f175a52012-03-02 05:37:00 -080087 else
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050088 for (( j = 0; j < ${BASH_ARGC[${n} -1]}; ++j )); do
89 args="${args:+${args} }'${BASH_ARGV[$(( p - j - 1 ))]}'"
90 done
91 ! (( p -= ${BASH_ARGC[${n} - 1]} ))
Brian Harring7f175a52012-03-02 05:37:00 -080092 fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050093 if [[ ${n} == ${#FUNCNAME[@]} ]]; then
Marc Herbertee1fcbb2015-05-28 10:37:46 -070094 error "Arguments of $$: ${0##/*} ${args}"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -050095 error "Backtrace: (most recent call is last)"
96 else
Marc Herbert2e8c5902015-01-15 20:17:31 -080097 src=${BASH_SOURCE[${n}]##*/}
Marc Herbertfa6cbe72015-01-20 14:09:09 -080098 curr_func=${FUNCNAME[${n}]}
99 error "$(printf ' %s:%s:%s(), called: %s %s ' \
100 "${src}" "${line}" "${curr_func}" "${func}" "${args}")"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500101 fi
102 done
103}
Brian Harring7f175a52012-03-02 05:37:00 -0800104
Mike Frysinger669b28b2012-02-07 18:01:00 -0500105# Declare these asap so that code below can safely assume they exist.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400106_message() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500107 local prefix=$1
Brian Harring7f175a52012-03-02 05:37:00 -0800108 shift
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500109 if [[ $# -eq 0 ]]; then
110 echo -e "${prefix}${CROS_LOG_PREFIX:-""}:${V_VIDOFF}" >&2
Brian Harring7f175a52012-03-02 05:37:00 -0800111 return
112 fi
113 (
114 # Handle newlines in the message, prefixing each chunk correctly.
115 # Do this in a subshell to avoid having to track IFS/set -f state.
116 IFS="
117"
118 set +f
119 set -- $*
120 IFS=' '
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500121 if [[ $# -eq 0 ]]; then
Brian Harring7f175a52012-03-02 05:37:00 -0800122 # Empty line was requested.
123 set -- ''
124 fi
125 for line in "$@"; do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500126 echo -e "${prefix}${CROS_LOG_PREFIX:-}: ${line}${V_VIDOFF}" >&2
Brian Harring7f175a52012-03-02 05:37:00 -0800127 done
128 )
129}
130
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400131info() {
Brian Harring7f175a52012-03-02 05:37:00 -0800132 _message "${V_BOLD_GREEN}INFO " "$*"
Mike Frysinger669b28b2012-02-07 18:01:00 -0500133}
134
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400135warn() {
Brian Harring7f175a52012-03-02 05:37:00 -0800136 _message "${V_BOLD_YELLOW}WARNING " "$*"
Mike Frysinger669b28b2012-02-07 18:01:00 -0500137}
138
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400139error() {
Brian Harring7f175a52012-03-02 05:37:00 -0800140 _message "${V_BOLD_RED}ERROR " "$*"
Mike Frysinger669b28b2012-02-07 18:01:00 -0500141}
142
Brian Harring7f175a52012-03-02 05:37:00 -0800143
144# For all die functions, they must explicitly force set +eu;
Alex Deymof9235952015-01-15 11:15:09 -0800145# no reason to have them cause their own crash if we're in the middle
Brian Harring7f175a52012-03-02 05:37:00 -0800146# of reporting an error condition then exiting.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400147die_err_trap() {
Mike Frysinger9ebc8ce2015-04-06 13:15:01 -0400148 local result=${1:-$?}
149 local command=${2:-${BASH_COMMAND:-command unknown}}
Brian Harring7f175a52012-03-02 05:37:00 -0800150 set +e +u
151
Mike Frysinger9ebc8ce2015-04-06 13:15:01 -0400152 if [[ ${result} == "0" ]]; then
153 # Let callers simplify by setting us as an EXIT trap handler.
154 return 0
155 fi
156
Brian Harring7f175a52012-03-02 05:37:00 -0800157 # Per the message, bash misreports 127 as 1 during err trap sometimes.
158 # Note this fact to ensure users don't place too much faith in the
159 # exit code in that case.
160 set -- "Command '${command}' exited with nonzero code: ${result}"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500161 if [[ ${result} -eq 1 ]] && [[ -z $(type -t ${command}) ]]; then
162 set -- "$@" \
163 '(Note bash sometimes misreports "command not found" as exit code 1 '\
Brian Harring7f175a52012-03-02 05:37:00 -0800164'instead of 127)'
Brian Harring7f175a52012-03-02 05:37:00 -0800165 fi
Marc Herbertff9fa312015-01-16 17:18:53 -0800166 dump_trace 1
Brian Harring7f175a52012-03-02 05:37:00 -0800167 error
168 error "Command failed:"
169 DIE_PREFIX=' '
170 die_notrace "$@"
171}
172
173# Exit this script due to a failure, outputting a backtrace in the process.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400174die() {
Brian Harring7f175a52012-03-02 05:37:00 -0800175 set +e +u
Marc Herbertff9fa312015-01-16 17:18:53 -0800176 dump_trace 1
Brian Harring7f175a52012-03-02 05:37:00 -0800177 error
178 error "Error was:"
179 DIE_PREFIX=' '
180 die_notrace "$@"
181}
182
183# Exit this script w/out a backtrace.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400184die_notrace() {
Brian Harring7f175a52012-03-02 05:37:00 -0800185 set +e +u
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500186 if [[ $# -eq 0 ]]; then
Brian Harring7f175a52012-03-02 05:37:00 -0800187 set -- '(no error message given)'
188 fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500189 local line
Brian Harring7f175a52012-03-02 05:37:00 -0800190 for line in "$@"; do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500191 error "${DIE_PREFIX}${line}"
Brian Harring7f175a52012-03-02 05:37:00 -0800192 done
Mike Frysinger669b28b2012-02-07 18:01:00 -0500193 exit 1
194}
195
David Jamesbf13ea82013-05-08 14:17:25 -0700196# Check for a single string in a list of space-separated strings.
197# e.g. has "foo" "foo bar baz" is true, but has "f" "foo bar baz" is not.
198has() { [[ " ${*:2} " == *" $1 "* ]]; }
199
Brian Harring9086a3f2013-01-14 01:43:16 -0800200# Directory locations inside the dev chroot; try the new default,
201# falling back to user specific paths if the upgrade has yet to
202# happen.
203_user="${USER}"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500204[[ ${USER} == "root" ]] && _user="${SUDO_USER}"
Brian Harring9086a3f2013-01-14 01:43:16 -0800205_CHROOT_TRUNK_DIRS=( "/home/${_user}/trunk" /mnt/host/source )
206_DEPOT_TOOLS_DIRS=( "/home/${_user}/depot_tools" /mnt/host/depot_tools )
Don Garrett63b47d32015-03-09 16:18:58 -0700207_WORKSPACE_DIRS=( "/home/${_user}/workspace" /mnt/host/workspace )
Brian Harring9086a3f2013-01-14 01:43:16 -0800208unset _user
209
210_process_mount_pt() {
211 # Given 4 arguments; the root path, the variable to set,
212 # the old location, and the new; finally, forcing the upgrade is doable
213 # via if a 5th arg is provided.
214 # This will then try to migrate the old to new if we can do so right now
215 # (else leaving symlinks in place w/in the new), and will set $1 to the
216 # new location.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500217 local base=${1:-/} var=$2 old=$3 new=$4 force=${5:-false}
218 local _sudo=$([[ ${USER} != "root" ]] && echo sudo)
219 local val=${new}
Mike Frysinger2d9a97d2015-02-25 01:38:56 -0500220 if ${force} || [[ -L ${base}/${new} ]] || [[ ! -e ${base}/${new} ]]; then
Brian Harring9086a3f2013-01-14 01:43:16 -0800221 # Ok, it's either a symlink or this is the first run. Upgrade if we can-
222 # specifically, if we're outside the chroot and we can rmdir the old.
223 # If we cannot rmdir the old, that's due to a mount being bound to that
224 # point (even if we can't see it, it's there)- thus fallback to adding
225 # compat links.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500226 if ${force} || ( [[ ${INSIDE_CHROOT} -eq 0 ]] && \
227 ${_sudo} rmdir "${base}/${old}" 2>/dev/null ); then
Brian Harring9086a3f2013-01-14 01:43:16 -0800228 ${_sudo} rm -f "${base}/${new}" || :
229 ${_sudo} mkdir -p "${base}/${new}" "$(dirname "${base}/${old}" )"
230 ${_sudo} ln -s "${new}" "${base}/${old}"
231 else
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500232 if [[ ! -L ${base}/${new} ]]; then
Brian Harring9086a3f2013-01-14 01:43:16 -0800233 # We can't do the upgrade right now; install compatibility links.
234 ${_sudo} mkdir -p "$(dirname "${base}/${new}")" "${base}/${old}"
235 ${_sudo} ln -s "${old}" "${base}/${new}"
236 fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500237 val=${old}
Brian Harring9086a3f2013-01-14 01:43:16 -0800238 fi
239 fi
240 eval "${var}=\"${val}\""
241}
242
243set_chroot_trunk_dir() {
244 # This takes two optional arguments; the first being the path to the chroot
245 # base; this is only used by enter_chroot. The second argument is whether
246 # or not to force the new pathways; this is only used by make_chroot. Passing
247 # a non-null value for $2 forces the new paths.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500248 if [[ ${INSIDE_CHROOT} -eq 0 ]] && [[ -z ${1-} ]]; then
Brian Harring9086a3f2013-01-14 01:43:16 -0800249 # Can't do the upgrade, thus skip trying to do so.
250 CHROOT_TRUNK_DIR="${_CHROOT_TRUNK_DIRS[1]}"
251 DEPOT_TOOLS_DIR="${_DEPOT_TOOLS_DIRS[1]}"
Don Garrett63b47d32015-03-09 16:18:58 -0700252 WORKSPACE_DIR="${_WORKSPACE_DIRS[1]}"
Brian Harring9086a3f2013-01-14 01:43:16 -0800253 return
254 fi
Alex Deymo76277de2015-01-22 11:02:13 -0800255 _process_mount_pt "${1:-}" CHROOT_TRUNK_DIR "${_CHROOT_TRUNK_DIRS[@]}" \
256 ${2:+true}
257 _process_mount_pt "${1:-}" DEPOT_TOOLS_DIR "${_DEPOT_TOOLS_DIRS[@]}" \
258 ${2:+true}
Don Garrett63b47d32015-03-09 16:18:58 -0700259 _process_mount_pt "${1:-}" WORKSPACE_DIR "${_WORKSPACE_DIRS[@]}" \
260 ${2:+true}
Brian Harring9086a3f2013-01-14 01:43:16 -0800261}
262
263set_chroot_trunk_dir
264
Anton Staaf30acb0b2011-01-26 16:00:20 -0800265# Construct a list of possible locations for the source tree. This list is
266# based on various environment variables and globals that may have been set
267# by the calling script.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400268get_gclient_root_list() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500269 if [[ ${INSIDE_CHROOT} -eq 1 ]]; then
Brian Harring2499bfb2012-12-18 16:23:31 -0800270 echo "${CHROOT_TRUNK_DIR}"
Anton Staaf30acb0b2011-01-26 16:00:20 -0800271 fi
272
Alex Deymo135faa12015-01-21 11:20:13 -0800273 if [[ -n ${COMMON_SH:-} ]]; then echo "$(dirname "${COMMON_SH}")/../.."; fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500274 if [[ -n ${BASH_SOURCE} ]]; then echo "$(dirname "${BASH_SOURCE}")/../.."; fi
Anton Staaf30acb0b2011-01-26 16:00:20 -0800275}
276
277# Based on the list of possible source locations we set GCLIENT_ROOT if it is
278# not already defined by looking for a src directory in each seach path
279# location. If we do not find a valid looking root we error out.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400280get_gclient_root() {
Alex Deymo135faa12015-01-21 11:20:13 -0800281 if [[ -n ${GCLIENT_ROOT:-} ]]; then
Anton Staaf30acb0b2011-01-26 16:00:20 -0800282 return
283 fi
284
285 for path in $(get_gclient_root_list); do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500286 if [[ -d ${path}/src ]]; then
Anton Staaf30acb0b2011-01-26 16:00:20 -0800287 GCLIENT_ROOT=${path}
288 break
289 fi
290 done
291
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500292 if [[ -z ${GCLIENT_ROOT} ]]; then
Anton Staaf30acb0b2011-01-26 16:00:20 -0800293 # Using dash or sh, we don't know where we are. $0 refers to the calling
294 # script, not ourselves, so that doesn't help us.
295 echo "Unable to determine location for common.sh. If you are sourcing"
296 echo "common.sh from a script run via dash or sh, you must do it in the"
297 echo "following way:"
298 echo ' COMMON_SH="$(dirname "$0")/../../scripts/common.sh"'
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500299 echo ' . "${COMMON_SH}"'
Anton Staaf30acb0b2011-01-26 16:00:20 -0800300 echo "where the first line is the relative path from your script to"
301 echo "common.sh."
302 exit 1
303 fi
304}
305
David James76764882012-10-24 19:46:29 -0700306# Populate the ENVIRONMENT_WHITELIST array.
307load_environment_whitelist() {
308 set -f
309 ENVIRONMENT_WHITELIST=(
310 $("${GCLIENT_ROOT}/chromite/scripts/cros_env_whitelist")
311 )
312 set +f
313}
314
Anton Staaf30acb0b2011-01-26 16:00:20 -0800315# Find root of source tree
316get_gclient_root
317
rspangler@google.comd74220d2009-10-09 20:56:14 +0000318# Canonicalize the directories for the root dir and the calling script.
319# readlink is part of coreutils and should be present even in a bare chroot.
tedbo4f44d9e2010-01-08 17:26:11 -0800320# This is better than just using
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500321# FOO="$(cd ${FOO} ; pwd)"
tedbo4f44d9e2010-01-08 17:26:11 -0800322# since that leaves symbolic links intact.
rspangler@google.comd74220d2009-10-09 20:56:14 +0000323# Note that 'realpath' is equivalent to 'readlink -f'.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500324SCRIPT_LOCATION=$(readlink -f "${SCRIPT_LOCATION}")
325GCLIENT_ROOT=$(readlink -f "${GCLIENT_ROOT}")
rspangler@google.comd74220d2009-10-09 20:56:14 +0000326
327# Other directories should always be pathed down from GCLIENT_ROOT.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500328SRC_ROOT="${GCLIENT_ROOT}/src"
329SRC_INTERNAL="${GCLIENT_ROOT}/src-internal"
330SCRIPTS_DIR="${SRC_ROOT}/scripts"
331BUILD_LIBRARY_DIR="${SCRIPTS_DIR}/build_library"
Mike Frysinger66accd22017-09-13 03:50:30 -0400332CHROMITE_BIN="${GCLIENT_ROOT}/chromite/bin"
Alex Klein999443c2018-10-17 12:02:05 -0600333IMAGES_DIR="${GCLIENT_ROOT}/src/build/images"
rspangler@google.comd74220d2009-10-09 20:56:14 +0000334
335# Load developer's custom settings. Default location is in scripts dir,
336# since that's available both inside and outside the chroot. By convention,
337# settings from this file are variables starting with 'CHROMEOS_'
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500338: ${CHROMEOS_DEV_SETTINGS:=${SCRIPTS_DIR}/.chromeos_dev}
339if [[ -f ${CHROMEOS_DEV_SETTINGS} ]]; then
rspangler@google.comd74220d2009-10-09 20:56:14 +0000340 # Turn on exit-on-error during custom settings processing
Greg Spencer798d75f2011-02-01 22:04:49 -0800341 SAVE_OPTS=$(set +o)
Brian Harring7f175a52012-03-02 05:37:00 -0800342 switch_to_strict_mode
rspangler@google.comd74220d2009-10-09 20:56:14 +0000343
344 # Read settings
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500345 . "${CHROMEOS_DEV_SETTINGS}"
rspangler@google.comd74220d2009-10-09 20:56:14 +0000346
347 # Restore previous state of exit-on-error
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500348 eval "${SAVE_OPTS}"
rspangler@google.comd74220d2009-10-09 20:56:14 +0000349fi
350
351# Load shflags
Zdenek Behan07d24222011-11-02 00:46:25 +0000352# NOTE: This code snippet is in particular used by the au-generator (which
353# stores shflags in ./lib/shflags/) and should not be touched.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500354if [[ -f ${SCRIPTS_DIR}/lib/shflags/shflags ]]; then
Mike Frysinger77c674b2012-02-07 18:05:07 -0500355 . "${SCRIPTS_DIR}/lib/shflags/shflags" || die "Couldn't find shflags"
Zdenek Behan07d24222011-11-02 00:46:25 +0000356else
357 . ./lib/shflags/shflags || die "Couldn't find shflags"
358fi
rspangler@google.comd74220d2009-10-09 20:56:14 +0000359
Bill Richardson10d27c22010-01-20 13:38:50 -0800360# Our local mirror
361DEFAULT_CHROMEOS_SERVER=${CHROMEOS_SERVER:-"http://build.chromium.org/mirror"}
rspangler@google.comd74220d2009-10-09 20:56:14 +0000362
Bill Richardson10d27c22010-01-20 13:38:50 -0800363# Upstream mirrors and build suites come in 2 flavors
364# DEV - development chroot, used to build the chromeos image
365# IMG - bootable image, to run on actual hardware
rspangler@google.comd74220d2009-10-09 20:56:14 +0000366
Bill Richardson10d27c22010-01-20 13:38:50 -0800367DEFAULT_DEV_MIRROR=${CHROMEOS_DEV_MIRROR:-"${DEFAULT_CHROMEOS_SERVER}/ubuntu"}
368DEFAULT_DEV_SUITE=${CHROMEOS_DEV_SUITE:-"karmic"}
369
370DEFAULT_IMG_MIRROR=${CHROMEOS_IMG_MIRROR:-"${DEFAULT_CHROMEOS_SERVER}/ubuntu"}
371DEFAULT_IMG_SUITE=${CHROMEOS_IMG_SUITE:-"karmic"}
rspangler@google.comd74220d2009-10-09 20:56:14 +0000372
373# Default location for chroot
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500374DEFAULT_CHROOT_DIR=${CHROMEOS_CHROOT_DIR:-"${GCLIENT_ROOT}/chroot"}
rspangler@google.comd74220d2009-10-09 20:56:14 +0000375
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500376# All output files from build should go under ${DEFAULT_BUILD_ROOT}, so that
rspangler@google.comd74220d2009-10-09 20:56:14 +0000377# they don't pollute the source directory.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500378DEFAULT_BUILD_ROOT=${CHROMEOS_BUILD_ROOT:-"${SRC_ROOT}/build"}
rspangler@google.comd74220d2009-10-09 20:56:14 +0000379
Chris Ching4bc95a12016-11-22 13:44:13 -0700380# Default location for event files
381DEFAULT_EVENT_DIR=${DEFAULT_EVENT_DIR:-"${DEFAULT_BUILD_ROOT}/events"}
382
383# Default event file. Format is YYYYDD.HHMM.json
384DEFAULT_EVENT_FILE=${DEFAULT_EVENT_FILE:-"${DEFAULT_EVENT_DIR}/$(date +%Y%m%d.%H%M.).json"}
385
David McMahon49302942010-02-18 16:55:35 -0800386# Set up a global ALL_BOARDS value
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500387if [[ -d ${SRC_ROOT}/overlays ]]; then
388 ALL_BOARDS=$(cd "${SRC_ROOT}/overlays"; \
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400389 ls -1d overlay-* 2>&- | sed 's,overlay-,,g')
David Rochberg3b910702010-12-02 10:45:21 -0500390fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500391# Normalize whitespace.
392ALL_BOARDS=$(echo ${ALL_BOARDS})
Mike Frysingerc17a4932012-03-12 17:07:40 -0400393
394# Sets the default board variable for calling script.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500395if [[ -f ${GCLIENT_ROOT}/src/scripts/.default_board ]]; then
396 DEFAULT_BOARD=$(<"${GCLIENT_ROOT}/src/scripts/.default_board")
Mike Frysingerc17a4932012-03-12 17:07:40 -0400397 # Check for user typos like whitespace.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500398 if [[ -n ${DEFAULT_BOARD//[a-zA-Z0-9-_]} ]]; then
Mike Frysingerc17a4932012-03-12 17:07:40 -0400399 die ".default_board: invalid name detected; please fix:" \
400 "'${DEFAULT_BOARD}'"
401 fi
402fi
403# Stub to get people to upgrade.
404get_default_board() {
405 warn "please upgrade your script, and make sure to run build_packages"
406}
David McMahon49302942010-02-18 16:55:35 -0800407
David Jamesff072012010-11-30 13:22:05 -0800408# Enable --fast by default.
Greg Spencer798d75f2011-02-01 22:04:49 -0800409DEFAULT_FAST=${FLAGS_TRUE}
David James03668642010-07-28 17:08:29 -0700410
Chris Sosab0f57322011-10-25 03:07:23 +0000411# Directory to store built images. Should be set by sourcing script when used.
412BUILD_DIR=
Simon Glass142ca062011-02-09 13:39:43 -0800413
Mike Frysinger34808e52017-01-11 21:14:08 -0500414# Path to the verified boot directory where we get signing related keys/scripts.
415VBOOT_DIR="${CHROOT_TRUNK_DIR}/src/platform/vboot_reference"
416VBOOT_TESTKEYS_DIR="${VBOOT_DIR}/tests/testkeys"
Mike Frysinger40cea032017-03-12 19:48:20 -0400417# We load these from the chroot rather than directly from the vboot source repo
418# so we work correctly even in a minilayout.
419VBOOT_DEVKEYS_DIR="/usr/share/vboot/devkeys"
420VBOOT_SIGNING_DIR="/usr/share/vboot/bin"
Mike Frysinger34808e52017-01-11 21:14:08 -0500421
Simon Glass142ca062011-02-09 13:39:43 -0800422# Standard filenames
Chris Sosab0f57322011-10-25 03:07:23 +0000423CHROMEOS_BASE_IMAGE_NAME="chromiumos_base_image.bin"
Simon Glass142ca062011-02-09 13:39:43 -0800424CHROMEOS_IMAGE_NAME="chromiumos_image.bin"
Chris Sosab0f57322011-10-25 03:07:23 +0000425CHROMEOS_DEVELOPER_IMAGE_NAME="chromiumos_image.bin"
Gilad Arnold08366272012-02-08 10:46:26 -0800426CHROMEOS_RECOVERY_IMAGE_NAME="recovery_image.bin"
Simon Glass142ca062011-02-09 13:39:43 -0800427CHROMEOS_TEST_IMAGE_NAME="chromiumos_test_image.bin"
Chris Sosab0f57322011-10-25 03:07:23 +0000428CHROMEOS_FACTORY_INSTALL_SHIM_NAME="factory_install_shim.bin"
Bertrand SIMONNETb17cd572015-04-01 10:27:37 -0700429SYSROOT_SETTINGS_FILE="/var/cache/edb/chromeos"
Simon Glass142ca062011-02-09 13:39:43 -0800430
Rahul Chaudhry306f6882015-05-13 15:14:13 -0700431# Install mask for portage ebuilds. Used by build_image and gmergefs.
Chris Masoned11ce172010-11-09 14:22:08 -0800432# TODO: Is /usr/local/autotest-chrome still used by anyone?
Hung-Te Lind32c59f2012-01-19 19:54:01 +0800433COMMON_INSTALL_MASK="
Daniel Erate82f07c2010-12-21 13:39:22 -0800434 *.a
Alex Deymo1856c4f2014-09-24 08:15:34 -0700435 *.c
436 *.cc
Rahul Chaudhry306f6882015-05-13 15:14:13 -0700437 *.go
Daniel Erate82f07c2010-12-21 13:39:22 -0800438 *.la
Rene Bolldorf32228222012-07-21 14:18:22 -0400439 *.h
Brian Norrisef0880d2016-05-19 12:39:56 -0700440 *.hh
Rene Bolldorf32228222012-07-21 14:18:22 -0400441 *.hpp
Kevin Cernekee6ab57a92015-06-25 15:43:06 -0700442 *.h++
Nam T. Nguyenaa15f142015-05-15 14:27:50 -0700443 *.hxx
Mike Frysinger97c6ac92014-06-16 00:06:37 -0400444 */.keep*
Daniel Erate82f07c2010-12-21 13:39:22 -0800445 /etc/init.d
446 /etc/runlevels
Qijiang Fan6834cba2018-04-04 16:41:56 +0900447 /etc/selinux/intermediates
Hung-Te Lin76272be2012-08-07 19:10:09 +0800448 /firmware
Mike Frysinger785f09d2013-09-25 22:11:21 -0400449 /lib/modules/*/vdso
Daniel Erate82f07c2010-12-21 13:39:22 -0800450 /lib/rc
Tomasz Figa435f0e52017-01-06 14:28:06 +0900451 /opt/google/containers/android/vendor/lib*/pkgconfig
Benjamin Gordonffcb2b12017-10-24 13:02:46 -0600452 /opt/google/containers/android/build
Nam T. Nguyenf1de1a72015-05-27 08:37:46 -0700453 /usr/bin/*-config
Daniel Erate82f07c2010-12-21 13:39:22 -0800454 /usr/bin/Xnest
455 /usr/bin/Xvfb
Manoj Gupta019843b2017-08-17 13:28:18 -0700456 /usr/include/c++
Mike Frysinger10c56472013-03-01 00:58:10 -0500457 /usr/include/nspr/*
Yunlian Jianged7cd2d2018-10-27 13:07:02 -0700458 /usr/include/rpcsvc/*.x
Mike Frysinger10c56472013-03-01 00:58:10 -0500459 /usr/include/X11/*
Daniel Erate82f07c2010-12-21 13:39:22 -0800460 /usr/lib/debug
Rahul Chaudhry306f6882015-05-13 15:14:13 -0700461 /usr/lib/gopath
Mike Frysinger27b5edb2012-08-08 18:52:06 -0400462 /usr/lib*/pkgconfig
Daniel Erate82f07c2010-12-21 13:39:22 -0800463 /usr/local/autotest-chrome
464 /usr/man
465 /usr/share/aclocal
Brian Norris0f3c4e22016-06-21 12:23:18 -0700466 /usr/share/cups/drv
Daniel Erate82f07c2010-12-21 13:39:22 -0800467 /usr/share/doc
468 /usr/share/gettext
469 /usr/share/gtk-2.0
470 /usr/share/gtk-doc
471 /usr/share/info
472 /usr/share/man
Brian Norris0f3c4e22016-06-21 12:23:18 -0700473 /usr/share/ppd
Daniel Erate82f07c2010-12-21 13:39:22 -0800474 /usr/share/openrc
475 /usr/share/pkgconfig
Ryan Harrisonb8bd5942013-02-02 13:39:43 -0500476 /usr/share/profiling
Daniel Erate82f07c2010-12-21 13:39:22 -0800477 /usr/share/readline
Chris Wolfed13775f2011-07-26 16:34:38 -0400478 /usr/src
Daniel Erate82f07c2010-12-21 13:39:22 -0800479 "
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700480
Hung-Te Lind32c59f2012-01-19 19:54:01 +0800481# Mask for base, dev, and test images (build_image, build_image --test)
482DEFAULT_INSTALL_MASK="
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500483 ${COMMON_INSTALL_MASK}
Don Garrettac1cd0d2013-07-18 18:04:06 -0700484 /boot/config-*
485 /boot/System.map-*
Aviv Kesheta1838ba2013-07-23 10:58:23 -0700486 /usr/local/build/autotest
Olof Johansson1222b2e2012-08-08 15:27:35 -0700487 /lib/modules/*/build
488 /lib/modules/*/source
Kees Cook74f163b2012-12-18 15:46:51 -0800489 test_*.ko
Hung-Te Lind32c59f2012-01-19 19:54:01 +0800490 "
491
Chris Sosac9422fa2012-03-05 15:58:07 -0800492# Mask for factory install shim (build_image factory_install)
Hung-Te Lind32c59f2012-01-19 19:54:01 +0800493FACTORY_SHIM_INSTALL_MASK="
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500494 ${DEFAULT_INSTALL_MASK}
Cheng-Han Yangc924c2c2018-11-02 17:02:29 +0800495 /opt/google/chrome
496 /opt/google/containers
Vic Yang982556a2012-12-11 11:41:54 +0800497 /usr/lib64/dri
Hung-Te Lin7f721ec2017-07-18 15:14:48 +0800498 /usr/lib/dri
Daniel Erate82f07c2010-12-21 13:39:22 -0800499 /usr/share/X11
Hung-Te Lin7f721ec2017-07-18 15:14:48 +0800500 /usr/share/chromeos-assets/[^i]*
501 /usr/share/chromeos-assets/i[^m]*
Daniel Erate82f07c2010-12-21 13:39:22 -0800502 /usr/share/fonts
Daniel Erate82f07c2010-12-21 13:39:22 -0800503 /usr/share/locale
Daniel Erate82f07c2010-12-21 13:39:22 -0800504 /usr/share/mime
Vic Yang982556a2012-12-11 11:41:54 +0800505 /usr/share/oem
Daniel Erate82f07c2010-12-21 13:39:22 -0800506 /usr/share/sounds
507 /usr/share/tts
508 /usr/share/zoneinfo
509 "
Tom Wai-Hong Tamf87a3672010-05-17 16:06:33 +0800510
Sabin Floarese790ea72016-11-21 14:56:41 +0200511# Mask for images without systemd.
512SYSTEMD_INSTALL_MASK="
513 /lib/systemd
514 /usr/lib/systemd
515"
516
rspangler@google.comd74220d2009-10-09 20:56:14 +0000517# -----------------------------------------------------------------------------
518# Functions
519
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400520setup_board_warning() {
tedbo373c3902010-04-12 10:52:40 -0700521 echo
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500522 echo "${V_REVERSE}================ WARNING =====================${V_VIDOFF}"
Chris Sosaacada732010-02-23 15:20:03 -0800523 echo
524 echo "*** No default board detected in " \
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500525 "${GCLIENT_ROOT}/src/scripts/.default_board"
Chris Sosaacada732010-02-23 15:20:03 -0800526 echo "*** Either run setup_board with default flag set"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500527 echo "*** or echo |board_name| > ${GCLIENT_ROOT}/src/scripts/.default_board"
Chris Sosaacada732010-02-23 15:20:03 -0800528 echo
529}
530
David James22dc2ba2012-11-29 15:46:58 -0800531is_nfs() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500532 [[ $(stat -f -L -c %T "$1") == "nfs" ]]
David James22dc2ba2012-11-29 15:46:58 -0800533}
534
535warn_if_nfs() {
536 if is_nfs "$1"; then
Marc MERLINfcd84ef2013-03-07 14:34:43 -0800537 warn "$1 is on NFS. This is untested. You can send patches if it's broken."
David James22dc2ba2012-11-29 15:46:58 -0800538 fi
539}
540
Don Garrett640a0582010-05-04 16:54:28 -0700541# Enter a chroot and restart the current script if needed
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400542restart_in_chroot_if_needed() {
David Rochberg3b910702010-12-02 10:45:21 -0500543 # NB: Pass in ARGV: restart_in_chroot_if_needed "$@"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500544 if [[ ${INSIDE_CHROOT} -ne 1 ]]; then
Chris Sosafd2cdec2011-03-24 16:06:59 -0700545 # Get inside_chroot path for script.
546 local chroot_path="$(reinterpret_path_for_chroot "$0")"
Mike Frysinger66accd22017-09-13 03:50:30 -0400547 exec "${CHROMITE_BIN}/cros_sdk" -- "${chroot_path}" "$@"
Don Garrett640a0582010-05-04 16:54:28 -0700548 fi
549}
550
rspangler@google.comd74220d2009-10-09 20:56:14 +0000551# Fail unless we're inside the chroot. This guards against messing up your
552# workstation.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400553assert_inside_chroot() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500554 if [[ ${INSIDE_CHROOT} -ne 1 ]]; then
rspangler@google.comd74220d2009-10-09 20:56:14 +0000555 echo "This script must be run inside the chroot. Run this first:"
Zdenek Behan2811c162011-08-13 00:47:38 +0200556 echo " cros_sdk"
rspangler@google.comd74220d2009-10-09 20:56:14 +0000557 exit 1
558 fi
559}
560
561# Fail if we're inside the chroot. This guards against creating or entering
562# nested chroots, among other potential problems.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400563assert_outside_chroot() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500564 if [[ ${INSIDE_CHROOT} -ne 0 ]]; then
rspangler@google.comd74220d2009-10-09 20:56:14 +0000565 echo "This script must be run outside the chroot."
566 exit 1
567 fi
568}
569
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400570assert_not_root_user() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500571 if [[ ${UID:-$(id -u)} == 0 ]]; then
derat@google.com86dcc8e2009-11-21 19:49:49 +0000572 echo "This script must be run as a non-root user."
573 exit 1
574 fi
575}
576
David James76764882012-10-24 19:46:29 -0700577assert_root_user() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500578 if [[ ${UID:-$(id -u)} != 0 ]] || [[ ${SUDO_USER:-root} == "root" ]]; then
David James76764882012-10-24 19:46:29 -0700579 die_notrace "This script must be run using sudo from a non-root user."
580 fi
581}
582
Luigi Semenzato1f82e122010-03-23 12:43:08 -0700583# Check that all arguments are flags; that is, there are no remaining arguments
584# after parsing from shflags. Allow (with a warning) a single empty-string
585# argument.
586#
587# TODO: fix buildbot so that it doesn't pass the empty-string parameter,
588# then change this function.
589#
590# Usage: check_flags_only_and_allow_null_arg "$@" && set --
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400591check_flags_only_and_allow_null_arg() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500592 local do_shift=1
593 if [[ $# -eq 1 ]] && [[ -z $1 ]]; then
Luigi Semenzato1f82e122010-03-23 12:43:08 -0700594 echo "$0: warning: ignoring null argument" >&2
595 shift
596 do_shift=0
597 fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500598 if [[ $# -gt 0 ]]; then
599 echo "error: invalid arguments: \"$*\"" >&2
Luigi Semenzato1f82e122010-03-23 12:43:08 -0700600 flags_help
601 exit 1
602 fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500603 return ${do_shift}
Luigi Semenzato1f82e122010-03-23 12:43:08 -0700604}
605
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700606# Removes single quotes around parameter
607# Arguments:
608# $1 - string which optionally has surrounding quotes
609# Returns:
610# None, but prints the string without quotes.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400611remove_quotes() {
Chris Sosaaa1a7fd2010-04-02 14:06:29 -0700612 echo "$1" | sed -e "s/^'//; s/'$//"
613}
tedbo373c3902010-04-12 10:52:40 -0700614
615# Writes stdin to the given file name as root using sudo in overwrite mode.
616#
617# $1 - The output file name.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400618sudo_clobber() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500619 sudo tee "$1" >/dev/null
tedbo373c3902010-04-12 10:52:40 -0700620}
621
622# Writes stdin to the given file name as root using sudo in append mode.
623#
624# $1 - The output file name.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400625sudo_append() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500626 sudo tee -a "$1" >/dev/null
tedbo373c3902010-04-12 10:52:40 -0700627}
robotboy98912212010-04-12 14:08:14 -0700628
Mike Frysinger286b5922011-09-28 11:59:53 -0400629# Execute multiple commands in a single sudo. Generally will speed things
630# up by avoiding multiple calls to `sudo`. If any commands fail, we will
631# call die with the failing command. We can handle a max of ~100 commands,
632# but hopefully no one will ever try that many at once.
633#
634# $@ - The commands to execute, one per arg.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400635sudo_multi() {
Mike Frysinger286b5922011-09-28 11:59:53 -0400636 local i cmds
637
638 # Construct the shell code to execute. It'll be of the form:
639 # ... && ( ( command ) || exit <command index> ) && ...
640 # This way we know which command exited. The exit status of
641 # the underlying command is lost, but we never cared about it
642 # in the first place (other than it is non zero), so oh well.
643 for (( i = 1; i <= $#; ++i )); do
644 cmds+=" && ( ( ${!i} ) || exit $(( i + 10 )) )"
645 done
646
647 # Execute our constructed shell code.
648 sudo -- sh -c ":${cmds[*]}" && i=0 || i=$?
649
650 # See if this failed, and if so, print out the failing command.
651 if [[ $i -gt 10 ]]; then
652 : $(( i -= 10 ))
653 die "sudo_multi failed: ${!i}"
654 elif [[ $i -ne 0 ]]; then
655 die "sudo_multi failed for unknown reason $i"
656 fi
657}
658
Chris Masonebbccc242014-02-08 16:23:53 -0800659# Clears out stale shadow-utils locks in the given target root.
660sudo_clear_shadow_locks() {
661 info "Clearing shadow utils lockfiles under $1"
662 sudo rm -f "$1/etc/"{passwd,group,shadow,gshadow}.lock*
663}
664
David James76764882012-10-24 19:46:29 -0700665# Writes stdin to the given file name as the sudo user in overwrite mode.
666#
667# $@ - The output file names.
668user_clobber() {
669 install -m644 -o ${SUDO_UID} -g ${SUDO_GID} /dev/stdin "$@"
670}
671
David James22dc2ba2012-11-29 15:46:58 -0800672# Copies the specified file owned by the user to the specified location.
673# If the copy fails as root (e.g. due to root_squash and NFS), retry the copy
674# with the user's account before failing.
675user_cp() {
676 cp -p "$@" 2>/dev/null || sudo -u ${SUDO_USER} -- cp -p "$@"
677}
678
David James76764882012-10-24 19:46:29 -0700679# Appends stdin to the given file name as the sudo user.
680#
681# $1 - The output file name.
682user_append() {
683 cat >> "$1"
684 chown ${SUDO_UID}:${SUDO_GID} "$1"
685}
686
687# Create the specified directory, along with parents, as the sudo user.
688#
689# $@ - The directories to create.
690user_mkdir() {
691 install -o ${SUDO_UID} -g ${SUDO_GID} -d "$@"
692}
693
694# Create the specified symlink as the sudo user.
695#
696# $1 - Link target
697# $2 - Link name
698user_symlink() {
699 ln -sfT "$1" "$2"
700 chown -h ${SUDO_UID}:${SUDO_GID} "$2"
701}
702
Mike Frysinger1aa61242011-09-15 17:46:44 -0400703# Locate all mounts below a specified directory.
704#
705# $1 - The root tree.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400706sub_mounts() {
Mike Frysinger1aa61242011-09-15 17:46:44 -0400707 # Assume that `mount` outputs a list of mount points in the order
708 # that things were mounted (since it always has and hopefully always
709 # will). As such, we have to unmount in reverse order to cleanly
710 # unmount submounts (think /dev/pts and /dev).
Alex Deymo7ba56be2015-02-12 14:28:06 -0800711 awk -v path="$1" -v len="${#1}" \
Benjamin Gordon1d5afed2017-06-14 16:35:32 -0600712 '(substr($2, 1, len+1) == path ||
713 substr($2, 1, len+1) == (path "/")) { print $2 }' /proc/mounts | \
Zdenek Behan52970f42012-08-30 22:36:40 +0200714 tac | \
715 sed -e 's/\\040(deleted)$//'
716 # Hack(zbehan): If a bind mount's source is mysteriously removed,
717 # we'd end up with an orphaned mount with the above string in its name.
718 # It can only be seen through /proc/mounts and will stick around even
719 # when it should be gone already. crosbug.com/31250
Mike Frysinger1aa61242011-09-15 17:46:44 -0400720}
721
robotboy98912212010-04-12 14:08:14 -0700722# Unmounts a directory, if the unmount fails, warn, and then lazily unmount.
723#
724# $1 - The path to unmount.
Mike Frysinger6b1abb22012-05-11 13:44:06 -0400725safe_umount_tree() {
Alex Deymo7ba56be2015-02-12 14:28:06 -0800726 local mount_point="$1"
robotboy98912212010-04-12 14:08:14 -0700727
Alex Deymo7ba56be2015-02-12 14:28:06 -0800728 local mounts=( $(sub_mounts "${mount_point}") )
729
730 # Silently return if the mount_point was already unmounted.
731 if [[ ${#mounts[@]} -eq 0 ]]; then
Mike Frysingere8aec372011-09-21 00:03:22 -0400732 return 0
733 fi
734
Mike Frysinger1aa61242011-09-15 17:46:44 -0400735 # First try to unmount in one shot to speed things up.
Chirantan Ekbote30823e92016-06-29 11:39:03 -0700736 if LC_ALL=C safe_umount -d "${mounts[@]}"; then
737 return 0
David Jamesd9b67982012-10-26 09:46:50 -0700738 fi
739
740 # Well that didn't work, so lazy unmount remaining ones.
Alex Deymo7ba56be2015-02-12 14:28:06 -0800741 warn "Failed to unmount ${mounts[@]}, these are the processes using the" \
742 "mount points."
743 sudo fuser -vm "${mount_point}" || true
744
Mike Frysinger1aa61242011-09-15 17:46:44 -0400745 warn "Doing a lazy unmount"
Alex Deymo7ba56be2015-02-12 14:28:06 -0800746 if ! safe_umount -d -l "${mounts[@]}"; then
747 mounts=( $(sub_mounts "${mount_point}") )
748 die "Failed to lazily unmount ${mounts[@]}"
robotboy98912212010-04-12 14:08:14 -0700749 fi
750}
Chris Sosa702618f2010-05-14 12:52:32 -0700751
Brian Harringece65e02012-08-30 13:42:21 -0700752
David James76764882012-10-24 19:46:29 -0700753# Run umount as root.
Brian Harringece65e02012-08-30 13:42:21 -0700754safe_umount() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500755 $([[ ${UID:-$(id -u)} != 0 ]] && echo sudo) umount "$@"
Brian Harringece65e02012-08-30 13:42:21 -0700756}
757
Gabe Black0a68b202014-10-08 22:09:10 -0700758# Run a command with sudo in a way that still preferentially uses files
759# from au-generator.zip, but still finds things in /sbin and /usr/sbin when
760# not using au-generator.zip.
761au_generator_sudo() {
762 # When searching for env, the unmodified path is used and env is potentially
763 # found somewhere out in the system. env itself sees the modified PATH
764 # and will find the command where we're telling it to. Running the command
765 # directly without env will escape our constructed PATH.
766 sudo -E PATH="${PATH}:/sbin:/usr/sbin" env "$@"
767}
768
Gabe Blackfbdb1d52014-09-22 23:43:35 -0700769# Setup a loopback device for a file and scan for partitions, with retries.
770#
771# $1 - The file to back the new loopback device.
772# $2-$N - Additional arguments to pass to losetup.
773loopback_partscan() {
774 local lb_dev image="$1"
775 shift
Gabe Black01ba4de2015-03-11 16:34:56 -0700776 lb_dev=$(au_generator_sudo losetup --show -f "$@" "${image}")
777 # Ignore problems deleting existing partitions. There shouldn't be any
778 # which will upset partx, but that's actually ok.
779 au_generator_sudo partx -d "${lb_dev}" 2>/dev/null || true
780 au_generator_sudo partx -a "${lb_dev}"
Gabe Blackfbdb1d52014-09-22 23:43:35 -0700781
Gabe Black7bb6c5a2015-03-16 20:23:06 -0700782 echo "${lb_dev}"
Gabe Blackfbdb1d52014-09-22 23:43:35 -0700783}
784
785# Detach a loopback device set up earlier.
786#
Gabe Black01ba4de2015-03-11 16:34:56 -0700787# $1 - The loop device to detach.
788# $2-$N - Additional arguments to pass to losetup.
Gabe Blackfbdb1d52014-09-22 23:43:35 -0700789loopback_detach() {
Mike Frysingeraf64ed52015-04-06 13:20:39 -0400790 # Retry the deletes before we detach. crbug.com/469259
791 local i
792 for (( i = 0; i < 10; i++ )); do
793 if au_generator_sudo partx -d "$1"; then
794 break
795 fi
796 warn "Sleeping & retrying ..."
797 sync
798 sleep 1
799 done
Gabe Black0a68b202014-10-08 22:09:10 -0700800 au_generator_sudo losetup --detach "$@"
Gabe Blackfbdb1d52014-09-22 23:43:35 -0700801}
802
Gabe Black067b3542014-09-23 01:15:14 -0700803# Get the size of a regular file or a block device.
804#
805# $1 - The regular file or block device to get the size of.
806bd_safe_size() {
807 local file="$1"
808 if [[ -b "${file}" ]]; then
809 sudo blockdev --getsize64 "${file}"
810 else
811 stat -c%s "${file}"
812 fi
813}
814
Matt Tennant298f61a2012-06-25 21:54:33 -0700815get_git_id() {
816 git var GIT_COMMITTER_IDENT | sed -e 's/^.*<\(\S\+\)>.*$/\1/'
817}
818
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500819# Fixes symlinks that are incorrectly prefixed with the build root $1
Chris Sosad4455022010-05-20 10:14:06 -0700820# rather than the real running root '/'.
821# TODO(sosa) - Merge setup - cleanup below with this method.
822fix_broken_symlinks() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500823 local build_root=$1
Chris Sosad4455022010-05-20 10:14:06 -0700824 local symlinks=$(find "${build_root}/usr/local" -lname "${build_root}/*")
Greg Spencer798d75f2011-02-01 22:04:49 -0800825 local symlink
Chris Sosad4455022010-05-20 10:14:06 -0700826 for symlink in ${symlinks}; do
827 echo "Fixing ${symlink}"
828 local target=$(ls -l "${symlink}" | cut -f 2 -d '>')
829 # Trim spaces from target (bashism).
830 target=${target/ /}
831 # Make new target (removes rootfs prefix).
832 new_target=$(echo ${target} | sed "s#${build_root}##")
833
834 echo "Fixing symlink ${symlink}"
835 sudo unlink "${symlink}"
836 sudo ln -sf "${new_target}" "${symlink}"
837 done
838}
839
Chris Sosa702618f2010-05-14 12:52:32 -0700840# Sets up symlinks for the developer root. It is necessary to symlink
841# usr and local since the developer root is mounted at /usr/local and
842# applications expect to be installed under /usr/local/bin, etc.
843# This avoids packages installing into /usr/local/usr/local/bin.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500844# $1 specifies the symlink target for the developer root.
845# $2 specifies the symlink target for the var directory.
846# $3 specifies the location of the stateful partition.
Chris Sosa702618f2010-05-14 12:52:32 -0700847setup_symlinks_on_root() {
848 # Give args better names.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500849 local dev_image_target=$1
850 local var_target=$2
851 local dev_image_root="$3/dev_image"
Chris Sosa702618f2010-05-14 12:52:32 -0700852
853 # If our var target is actually the standard var, we are cleaning up the
854 # symlinks (could also check for /usr/local for the dev_image_target).
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500855 if [[ ${var_target} == "/var" ]]; then
Chris Sosa702618f2010-05-14 12:52:32 -0700856 echo "Cleaning up /usr/local symlinks for ${dev_image_root}"
857 else
858 echo "Setting up symlinks for /usr/local for ${dev_image_root}"
859 fi
860
861 # Set up symlinks that should point to ${dev_image_target}.
Greg Spencer798d75f2011-02-01 22:04:49 -0800862 local path
Chris Sosa702618f2010-05-14 12:52:32 -0700863 for path in usr local; do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500864 if [[ -h ${dev_image_root}/${path} ]]; then
Chris Sosa702618f2010-05-14 12:52:32 -0700865 sudo unlink "${dev_image_root}/${path}"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500866 elif [[ -e ${dev_image_root}/${path} ]]; then
Chris Sosa702618f2010-05-14 12:52:32 -0700867 die "${dev_image_root}/${path} should be a symlink if exists"
868 fi
Mike Frysingera1a06ab2011-08-10 11:40:30 -0400869 sudo ln -s "${dev_image_target}" "${dev_image_root}/${path}"
Chris Sosa702618f2010-05-14 12:52:32 -0700870 done
871
872 # Setup var symlink.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500873 if [[ -h ${dev_image_root}/var ]]; then
Chris Sosa702618f2010-05-14 12:52:32 -0700874 sudo unlink "${dev_image_root}/var"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500875 elif [[ -e ${dev_image_root}/var ]]; then
Chris Sosa702618f2010-05-14 12:52:32 -0700876 die "${dev_image_root}/var should be a symlink if it exists"
877 fi
878
879 sudo ln -s "${var_target}" "${dev_image_root}/var"
880}
Nick Sandersd2509272010-06-16 03:50:04 -0700881
Will Drewry55b42c92010-10-20 15:44:11 -0500882# These two helpers clobber the ro compat value in our root filesystem.
883#
884# When the system is built with --enable_rootfs_verification, bit-precise
885# integrity checking is performed. That precision poses a usability issue on
886# systems that automount partitions with recognizable filesystems, such as
887# ext2/3/4. When the filesystem is mounted 'rw', ext2 metadata will be
888# automatically updated even if no other writes are performed to the
889# filesystem. In addition, ext2+ does not support a "read-only" flag for a
890# given filesystem. That said, forward and backward compatibility of
891# filesystem features are supported by tracking if a new feature breaks r/w or
892# just write compatibility. We abuse the read-only compatibility flag[1] in
893# the filesystem header by setting the high order byte (le) to FF. This tells
894# the kernel that features R24-R31 are all enabled. Since those features are
895# undefined on all ext-based filesystem, all standard kernels will refuse to
896# mount the filesystem as read-write -- only read-only[2].
897#
898# [1] 32-bit flag we are modifying:
Aaron Gable5dde8482013-09-27 18:35:57 -0700899# https://chromium.googlesource.com/chromiumos/third_party/kernel.git/+/master/include/linux/ext2_fs.h#l417
Will Drewry55b42c92010-10-20 15:44:11 -0500900# [2] Mount behavior is enforced here:
Aaron Gable5dde8482013-09-27 18:35:57 -0700901# https://chromium.googlesource.com/chromiumos/third_party/kernel.git/+/master/ext2/super.c#l857
Will Drewry55b42c92010-10-20 15:44:11 -0500902#
903# N.B., if the high order feature bits are used in the future, we will need to
904# revisit this technique.
905disable_rw_mount() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500906 local rootfs=$1
Will Drewry55b42c92010-10-20 15:44:11 -0500907 local offset="${2-0}" # in bytes
Will Drewry9b7cb512010-10-20 18:11:24 -0500908 local ro_compat_offset=$((0x464 + 3)) # Set 'highest' byte
Alex Deymof6a527d2015-01-12 18:30:58 -0800909 is_ext_filesystem "${rootfs}" "${offset}" || return 0
Alex Deymo41226312015-03-09 15:35:24 -0700910 is_ext2_rw_mount_enabled "${rootfs}" "${offset}" || return 0
911
912 make_block_device_rw "${rootfs}"
Will Drewry9b7cb512010-10-20 18:11:24 -0500913 printf '\377' |
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500914 sudo dd of="${rootfs}" seek=$((offset + ro_compat_offset)) \
Alex Deymo41226312015-03-09 15:35:24 -0700915 conv=notrunc count=1 bs=1 status=none
Will Drewry55b42c92010-10-20 15:44:11 -0500916}
917
918enable_rw_mount() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500919 local rootfs=$1
Will Drewry55b42c92010-10-20 15:44:11 -0500920 local offset="${2-0}"
Will Drewry9b7cb512010-10-20 18:11:24 -0500921 local ro_compat_offset=$((0x464 + 3)) # Set 'highest' byte
Alex Deymof6a527d2015-01-12 18:30:58 -0800922 is_ext_filesystem "${rootfs}" "${offset}" || return 0
Alex Deymo41226312015-03-09 15:35:24 -0700923 is_ext2_rw_mount_enabled "${rootfs}" "${offset}" && return 0
924
925 make_block_device_rw "${rootfs}"
Will Drewry9b7cb512010-10-20 18:11:24 -0500926 printf '\000' |
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500927 sudo dd of="${rootfs}" seek=$((offset + ro_compat_offset)) \
Alex Deymo41226312015-03-09 15:35:24 -0700928 conv=notrunc count=1 bs=1 status=none
929}
930
931is_ext2_rw_mount_enabled() {
932 local rootfs=$1
933 local offset="${2-0}"
934 local ro_compat_offset=$((0x464 + 3)) # Get 'highest' byte
935 local ro_compat_flag=$(sudo dd if="${rootfs}" \
936 skip=$((offset + ro_compat_offset)) bs=1 count=1 status=none \
937 2>/dev/null | hexdump -e '1 "%.2x"')
938 test "${ro_compat_flag}" = "00"
Will Drewry55b42c92010-10-20 15:44:11 -0500939}
940
Alex Deymof9235952015-01-15 11:15:09 -0800941# Returns whether the passed rootfs is an extended filesystem by checking the
Alex Deymof6a527d2015-01-12 18:30:58 -0800942# ext2 s_magic field in the superblock.
943is_ext_filesystem() {
944 local rootfs=$1
945 local offset="${2-0}"
946 local ext_magic_offset=$((0x400 + 56))
Alex Deymo41226312015-03-09 15:35:24 -0700947 local ext_magic=$(sudo dd if="${rootfs}" \
948 skip=$((offset + ext_magic_offset)) bs=1 count=2 2>/dev/null |
949 hexdump -e '1/2 "%.4x"')
Alex Deymof6a527d2015-01-12 18:30:58 -0800950 test "${ext_magic}" = "ef53"
951}
952
Alex Deymo41226312015-03-09 15:35:24 -0700953# If the passed argument is a block device, ensure it is writtable and make it
954# writtable if not.
955make_block_device_rw() {
956 local block_dev="$1"
957 [[ -b "${block_dev}" ]] || return 0
958 if [[ $(sudo blockdev --getro "${block_dev}") == "1" ]]; then
959 sudo blockdev --setrw "${block_dev}"
960 fi
961}
962
Nick Sandersd2509272010-06-16 03:50:04 -0700963# Get current timestamp. Assumes common.sh runs at startup.
964start_time=$(date +%s)
965
Matt Tennant298f61a2012-06-25 21:54:33 -0700966# Get time elapsed since start_time in seconds.
967get_elapsed_seconds() {
Greg Spencer798d75f2011-02-01 22:04:49 -0800968 local end_time=$(date +%s)
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500969 local elapsed_seconds=$(( end_time - start_time ))
Matt Tennant298f61a2012-06-25 21:54:33 -0700970 echo ${elapsed_seconds}
971}
972
973# Print time elapsed since start_time.
974print_time_elapsed() {
975 # Optional first arg to specify elapsed_seconds. If not given, will
976 # recalculate elapsed time to now. Optional second arg to specify
977 # command name associated with elapsed time.
978 local elapsed_seconds=${1:-$(get_elapsed_seconds)}
979 local cmd_base=${2:-}
980
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500981 local minutes=$(( elapsed_seconds / 60 ))
982 local seconds=$(( elapsed_seconds % 60 ))
Matt Tennant298f61a2012-06-25 21:54:33 -0700983
Mike Frysingerbaae8eb2013-01-15 01:34:26 -0500984 if [[ -n ${cmd_base} ]]; then
Matt Tennant298f61a2012-06-25 21:54:33 -0700985 info "Elapsed time (${cmd_base}): ${minutes}m${seconds}s"
986 else
987 info "Elapsed time: ${minutes}m${seconds}s"
988 fi
989}
990
991# Associative array for filling in extra command-specific stats before
992# calling command_completed.
993declare -A EXTRA_COMMAND_STATS
994
995# Save original command line.
996command_line_arr=( "$0" "$@" )
997
998command_completed() {
999 # Call print_elapsed_time regardless.
1000 local run_time=$(get_elapsed_seconds)
1001 local cmd_base=$(basename "${command_line_arr[0]}")
1002 print_time_elapsed ${run_time} ${cmd_base}
1003
Alex Deymof9235952015-01-15 11:15:09 -08001004 # Prepare command stats in an associative array. Additional command-specific
Matt Tennant298f61a2012-06-25 21:54:33 -07001005 # stats can be added through EXTRA_COMMAND_STATS associative array.
1006 declare -A stats
1007 stats=(
1008 [cmd_line]=${command_line_arr[*]}
1009 [cmd_base]=${cmd_base}
1010 [cmd_args]=${command_line_arr[*]:1}
1011 [run_time]=${run_time}
1012 [username]=$(get_git_id)
1013 [board]=${FLAGS_board}
1014 [host]=$(hostname -f)
1015 [cpu_count]=$(grep -c processor /proc/cpuinfo)
1016 [cpu_type]=$(uname -p)
1017 )
1018
1019 local attr
1020 for attr in "${!EXTRA_COMMAND_STATS[@]}"; do
1021 stats[${attr}]=${EXTRA_COMMAND_STATS[${attr}]}
1022 done
1023
1024 # Prepare temporary file for stats.
1025 local tmpfile=$(mktemp -t tmp.stats.XXXXXX)
1026 trap "rm -f '${tmpfile}'" EXIT
1027
1028 # Write stats out to temporary file.
1029 echo "Chromium OS Build Command Stats - Version 1" > "${tmpfile}"
1030 for attr in "${!stats[@]}"; do
1031 echo "${attr} ${stats[${attr}]}"
1032 done >> "${tmpfile}"
1033
Matt Tennant298f61a2012-06-25 21:54:33 -07001034 rm "${tmpfile}"
1035 trap - EXIT
Nick Sandersd2509272010-06-16 03:50:04 -07001036}
Doug Anderson0c9e88d2010-10-19 14:49:39 -07001037
Anton Staaf9bcd8412011-01-24 15:27:14 -08001038# The board and variant command line options can be used in a number of ways
1039# to specify the board and variant. The board can encode both pieces of
1040# information separated by underscores. Or the variant can be passed using
1041# the separate variant option. This function extracts the canonical board and
1042# variant information and provides it in the BOARD, VARIANT and BOARD_VARIANT
1043# variables.
1044get_board_and_variant() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001045 local flags_board=$1
1046 local flags_variant=$2
Anton Staaf9bcd8412011-01-24 15:27:14 -08001047
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001048 BOARD=$(echo "${flags_board}" | cut -d '_' -f 1)
1049 VARIANT=${flags_variant:-$(echo "${flags_board}" | cut -s -d '_' -f 2)}
Anton Staaf9bcd8412011-01-24 15:27:14 -08001050
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001051 BOARD_VARIANT=${BOARD}
1052 if [[ -n ${VARIANT} ]]; then
1053 BOARD_VARIANT+="_${VARIANT}"
Anton Staaf9bcd8412011-01-24 15:27:14 -08001054 fi
1055}
Simon Glass142ca062011-02-09 13:39:43 -08001056
Bertrand SIMONNET48b6e782015-03-26 16:04:07 -07001057# Load a setting from the sysroot's standard configuration file,
1058# etc/make.conf.board_setup.
1059# $1 - Path to the sysroot.
1060# $2 - Variable to get.
1061get_sysroot_config() {
1062 local sysroot=$1
1063 local variable=$2
1064 local config_file="${sysroot%/}/etc/make.conf.board_setup"
1065
Bertrand SIMONNETb17cd572015-04-01 10:27:37 -07001066 get_variable "${config_file}" "${variable}"
1067}
1068
1069# Load a single variable from a bash file.
1070# $1 - Path to the file.
1071# $2 - Variable to get.
1072get_variable() {
1073 local filepath=$1
1074 local variable=$2
1075 local lockfile="${filepath}.lock"
1076
1077 if [[ -e "${filepath}" ]]; then
1078 userowned_file "${lockfile}"
Bertrand SIMONNET48b6e782015-03-26 16:04:07 -07001079 (
Bertrand SIMONNETb17cd572015-04-01 10:27:37 -07001080 flock 201
1081 . "${filepath}"
Bertrand SIMONNET48b6e782015-03-26 16:04:07 -07001082 if [[ "${!variable+set}" == "set" ]]; then
1083 echo "${!variable}"
Bertrand SIMONNET48b6e782015-03-26 16:04:07 -07001084 fi
Bertrand SIMONNETb17cd572015-04-01 10:27:37 -07001085 ) 201>"${lockfile}"
1086 fi
1087}
1088
1089# Set a single variable in a KEY=VALUE file.
1090# Note: the file is assumed to be owned by root.
1091# $1 - Path to the file.
1092# $2 - Variable to set.
1093# $3 - Value to set.
1094set_variable() {
1095 local filepath=$1
1096 local variable=$2
1097 local value=$3
1098 local lockfile="${filepath}.lock"
1099
1100 userowned_file "${lockfile}"
1101 (
1102 flock 201
1103 sudo touch "${filepath}"
1104 sudo sed -i -e "/^${variable}=/d" "${filepath}"
1105 printf '\n%s="%s"\n' "${variable}" "${value}" | sudo_append "${filepath}"
1106 ) 201>"${lockfile}"
1107}
1108
1109# Creates a user owned file.
1110# $1 - Path to the file.
1111userowned_file() {
1112 local filepath=$1
1113
1114 if [[ ! -w "${filepath}" ]]; then
1115 cmds=(
1116 "mkdir -p '$(dirname "${filepath}")'"
1117 "touch '${filepath}'"
1118 "chown ${USER} '${filepath}'"
Bertrand SIMONNET48b6e782015-03-26 16:04:07 -07001119 )
Bertrand SIMONNETb17cd572015-04-01 10:27:37 -07001120 sudo_multi "${cmds[@]}"
Bertrand SIMONNET48b6e782015-03-26 16:04:07 -07001121 fi
1122}
1123
Albert Chaulk5d190f72013-07-12 11:42:18 -07001124# Load configuration files that allow board-specific overrides of default
1125# functionality to be specified in overlays.
Steve Fungd3b1f5f2015-03-29 21:47:24 -07001126# $1 - File to load.
Albert Chaulk5d190f72013-07-12 11:42:18 -07001127load_board_specific_script() {
Steve Fung88897cc2015-03-25 13:38:38 -07001128 local file=$1 overlay
1129 [[ $# -ne 1 ]] && die "load_board_specific_script requires exactly 1 param"
1130 for overlay in ${BOARD_OVERLAY}; do
Albert Chaulk5d190f72013-07-12 11:42:18 -07001131 local setup_sh="${overlay}/scripts/${file}"
1132 if [[ -e ${setup_sh} ]]; then
1133 source "${setup_sh}"
1134 fi
1135 done
1136}
1137
Anton Staaf6f5262d2011-03-02 09:35:54 -08001138# Check that the specified file exists. If the file path is empty or the file
1139# doesn't exist on the filesystem generate useful error messages. Otherwise
1140# show the user the name and path of the file that will be used. The padding
1141# parameter can be used to tabulate multiple name:path pairs. For example:
1142#
1143# check_for_file "really long name" "...:" "file.foo"
1144# check_for_file "short name" ".........:" "another.bar"
1145#
1146# Results in the following output:
1147#
1148# Using really long name...: file.foo
1149# Using short name.........: another.bar
1150#
1151# If tabulation is not required then passing "" for padding generates the
1152# output "Using <name> <path>"
1153check_for_file() {
1154 local name=$1
1155 local padding=$2
1156 local path=$3
1157
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001158 if [[ -z ${path} ]]; then
Anton Staaf6f5262d2011-03-02 09:35:54 -08001159 die "No ${name} file specified."
1160 fi
1161
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001162 if [[ ! -e ${path} ]]; then
Anton Staaf6f5262d2011-03-02 09:35:54 -08001163 die "No ${name} file found at: ${path}"
1164 else
1165 info "Using ${name}${padding} ${path}"
1166 fi
1167}
1168
1169# Check that the specified tool exists. If it does not exist in the PATH
1170# generate a useful error message indicating how to install the ebuild
1171# that contains the required tool.
1172check_for_tool() {
1173 local tool=$1
1174 local ebuild=$2
1175
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001176 if ! which "${tool}" >/dev/null; then
Anton Staaf6f5262d2011-03-02 09:35:54 -08001177 error "The ${tool} utility was not found in your path. Run the following"
1178 error "command in your chroot to install it: sudo -E emerge ${ebuild}"
1179 exit 1
1180 fi
1181}
Chris Sosafd2cdec2011-03-24 16:06:59 -07001182
1183# Reinterprets path from outside the chroot for use inside.
1184# Returns "" if "" given.
1185# $1 - The path to reinterpret.
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001186reinterpret_path_for_chroot() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001187 if [[ ${INSIDE_CHROOT} -ne 1 ]]; then
1188 if [[ -z $1 ]]; then
Chris Sosafd2cdec2011-03-24 16:06:59 -07001189 echo ""
1190 else
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001191 local path_abs_path=$(readlink -f "$1")
Chris Sosafd2cdec2011-03-24 16:06:59 -07001192 local gclient_root_abs_path=$(readlink -f "${GCLIENT_ROOT}")
1193
1194 # Strip the repository root from the path.
1195 local relative_path=$(echo ${path_abs_path} \
Mike Frysingera1a06ab2011-08-10 11:40:30 -04001196 | sed "s:${gclient_root_abs_path}/::")
Chris Sosafd2cdec2011-03-24 16:06:59 -07001197
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001198 if [[ ${relative_path} == "${path_abs_path}" ]]; then
1199 die "Error reinterpreting path. Path $1 is not within source tree."
Chris Sosafd2cdec2011-03-24 16:06:59 -07001200 fi
1201
1202 # Prepend the chroot repository path.
Chris Sosaed2ff4b2013-08-30 12:58:41 -07001203 echo "/mnt/host/source/${relative_path}"
Chris Sosafd2cdec2011-03-24 16:06:59 -07001204 fi
1205 else
1206 # Path is already inside the chroot :).
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001207 echo "$1"
Chris Sosafd2cdec2011-03-24 16:06:59 -07001208 fi
1209}
Gabe Black83d8b822011-08-01 17:50:09 -07001210
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001211emerge_custom_kernel() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001212 local install_root=$1
David Jamesdee866c2012-03-15 14:53:19 -07001213 local root=/build/${FLAGS_board}
David James0ea96e42011-08-03 11:53:50 -07001214 local tmp_pkgdir=${root}/custom-packages
1215
1216 # Clean up any leftover state in custom directories.
Mike Frysingera1a06ab2011-08-10 11:40:30 -04001217 sudo rm -rf "${tmp_pkgdir}"
David James0ea96e42011-08-03 11:53:50 -07001218
Aviv Keshet14373912014-06-27 14:34:06 +00001219 # Update chromeos-initramfs to contain the latest binaries from the build
1220 # tree. This is basically just packaging up already-built binaries from
1221 # ${root}. We are careful not to muck with the existing prebuilts so that
1222 # prebuilts can be uploaded in parallel.
1223 # TODO(davidjames): Implement ABI deps so that chromeos-initramfs will be
1224 # rebuilt automatically when its dependencies change.
1225 sudo -E PKGDIR="${tmp_pkgdir}" ${EMERGE_BOARD_CMD} -1 \
1226 chromeos-base/chromeos-initramfs || die "Cannot emerge chromeos-initramfs"
1227
David James0ea96e42011-08-03 11:53:50 -07001228 # Verify all dependencies of the kernel are installed. This should be a
1229 # no-op, but it's good to check in case a developer didn't run
Mike Frysinger0957a182012-03-21 23:17:14 -04001230 # build_packages. We need the expand_virtual call to workaround a bug
1231 # in portage where it only installs the virtual pkg.
1232 local kernel=$(portageq-${FLAGS_board} expand_virtual ${root} \
1233 virtual/linux-sources)
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001234 sudo -E PKGDIR="${tmp_pkgdir}" ${EMERGE_BOARD_CMD} --onlydeps \
David James0ea96e42011-08-03 11:53:50 -07001235 ${kernel} || die "Cannot emerge kernel dependencies"
1236
1237 # Build the kernel. This uses the standard root so that we can pick up the
1238 # initramfs from there. But we don't actually install the kernel to the
1239 # standard root, because that'll muck up the kernel debug symbols there,
1240 # which we want to upload in parallel.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001241 sudo -E PKGDIR="${tmp_pkgdir}" ${EMERGE_BOARD_CMD} --buildpkgonly \
David James0ea96e42011-08-03 11:53:50 -07001242 ${kernel} || die "Cannot emerge kernel"
1243
1244 # Install the custom kernel to the provided install root.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001245 sudo -E PKGDIR="${tmp_pkgdir}" ${EMERGE_BOARD_CMD} --usepkgonly \
David James0ea96e42011-08-03 11:53:50 -07001246 --root=${install_root} ${kernel} || die "Cannot emerge kernel to root"
1247}
Brian Harringfeb04f72012-02-03 21:22:50 -08001248
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001249enable_strict_sudo() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001250 if [[ -z ${CROS_SUDO_KEEP_ALIVE} ]]; then
Brian Harringfeb04f72012-02-03 21:22:50 -08001251 echo "$0 was somehow invoked in a way that the sudo keep alive could"
1252 echo "not be found. Failing due to this. See crosbug.com/18393."
1253 exit 126
1254 fi
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001255 sudo() {
1256 $(type -P sudo) -n "$@"
Brian Harringfeb04f72012-02-03 21:22:50 -08001257 }
1258}
Gilad Arnold207a7c72012-02-09 10:19:16 -08001259
Chris Wolfe21a27b72012-02-27 13:00:51 -05001260# Checks that stdin and stderr are both terminals.
1261# If so, we assume that there is a live user we can interact with.
1262# This check can be overridden by setting the CROS_NO_PROMPT environment
1263# variable to a non-empty value.
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001264is_interactive() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001265 [[ -z ${CROS_NO_PROMPT} && -t 0 && -t 2 ]]
Chris Wolfe21a27b72012-02-27 13:00:51 -05001266}
1267
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001268assert_interactive() {
Chris Wolfe21a27b72012-02-27 13:00:51 -05001269 if ! is_interactive; then
1270 die "Script ${0##*/} tried to get user input on a non-interactive terminal."
1271 fi
1272}
1273
Gilad Arnold207a7c72012-02-09 10:19:16 -08001274# Selection menu with a default option: this is similar to bash's select
1275# built-in, only that in case of an empty selection it'll return the default
1276# choice. Like select, it uses PS3 as the prompt.
1277#
1278# $1: name of variable to be assigned the selected value; it better not be of
1279# the form choose_foo to avoid conflict with local variables.
1280# $2: default value to return in case of an empty user entry.
1281# $3: value to return in case of an invalid choice.
1282# $...: options for selection.
1283#
1284# Usage example:
1285#
1286# PS3="Select one [1]: "
1287# choose reply "foo" "ERROR" "foo" "bar" "foobar"
1288#
1289# This will present the following menu and prompt:
1290#
1291# 1) foo
1292# 2) bar
1293# 3) foobar
1294# Select one [1]:
1295#
1296# The return value will be stored in a variable named 'reply'. If the input is
1297# 1, 2 or 3, the return value will be "foo", "bar" or "foobar", respectively.
1298# If it is empty (i.e. the user clicked Enter) it will be "foo". Anything else
1299# will return "ERROR".
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001300choose() {
Gilad Arnold207a7c72012-02-09 10:19:16 -08001301 typeset -i choose_i=1
1302
1303 # Retrieve output variable name and default return value.
1304 local choose_reply=$1
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001305 local choose_default=$2
1306 local choose_invalid=$3
Gilad Arnold207a7c72012-02-09 10:19:16 -08001307 shift 3
1308
1309 # Select a return value
1310 unset REPLY
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001311 if [[ $# -gt 0 ]]; then
Chris Wolfe21a27b72012-02-27 13:00:51 -05001312 assert_interactive
1313
Gilad Arnold207a7c72012-02-09 10:19:16 -08001314 # Actual options provided, present a menu and prompt for a choice.
1315 local choose_opt
1316 for choose_opt in "$@"; do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001317 echo "${choose_i}) ${choose_opt}" >&2
1318 : $(( ++choose_i ))
Gilad Arnold207a7c72012-02-09 10:19:16 -08001319 done
1320 read -p "$PS3"
1321 fi
1322 # Filter out strings containing non-digits.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001323 if [[ ${REPLY} != "${REPLY%%[!0-9]*}" ]]; then
Gilad Arnold207a7c72012-02-09 10:19:16 -08001324 REPLY=0
1325 fi
1326 choose_i="${REPLY}"
1327
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001328 if [[ ${choose_i} -ge 1 && ${choose_i} -le $# ]]; then
Gilad Arnold207a7c72012-02-09 10:19:16 -08001329 # Valid choice, return the corresponding value.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001330 eval ${choose_reply}=\""${!choose_i}"\"
1331 elif [[ -z ${REPLY} ]]; then
Gilad Arnold207a7c72012-02-09 10:19:16 -08001332 # Empty choice, return default value.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001333 eval ${choose_reply}=\""${choose_default}"\"
Gilad Arnold207a7c72012-02-09 10:19:16 -08001334 else
1335 # Invalid choice, return corresponding value.
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001336 eval ${choose_reply}=\""${choose_invalid}\""
Gilad Arnold207a7c72012-02-09 10:19:16 -08001337 fi
1338}
David James855afb72012-03-14 20:04:59 -07001339
Stephen Barberf9e93a32017-05-17 15:41:57 -07001340# Display a prompt that asks the user to choose yes or no.
1341# $1 - The prompt to be displayed to the user, with " [y/N]: " appended.
1342#
1343# Usage example:
1344#
1345# prompt_yesno "Would you like a cup of tea?"
1346#
1347# The function will return 0 for yes and 1 for no, appropriate for using
1348# in an if statement or loop.
1349prompt_yesno() {
1350 local prompt=$1
1351 local reply
1352
1353 assert_interactive
1354 read -p "${prompt} [y/N]: " reply
1355
1356 # Be strict. No is the default.
1357 if [[ "${reply}" != y && "${reply}" != Y ]]; then
1358 return 1
1359 fi
1360
1361 return 0
1362}
1363
David James855afb72012-03-14 20:04:59 -07001364# Display --help if requested. This is used to hide options from help
1365# that are not intended for developer use.
1366#
1367# How to use:
1368# 1) Declare the options that you want to appear in help.
1369# 2) Call this function.
1370# 3) Declare the options that you don't want to appear in help.
1371#
1372# See build_packages for example usage.
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001373show_help_if_requested() {
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001374 local opt
David James855afb72012-03-14 20:04:59 -07001375 for opt in "$@"; do
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001376 if [[ ${opt} == "-h" || ${opt} == "--help" ]]; then
David James855afb72012-03-14 20:04:59 -07001377 flags_help
1378 exit 0
1379 fi
1380 done
1381}
Brian Harring7f175a52012-03-02 05:37:00 -08001382
Mike Frysinger6b1abb22012-05-11 13:44:06 -04001383switch_to_strict_mode() {
Brian Harring7f175a52012-03-02 05:37:00 -08001384 # Set up strict execution mode; note that the trap
1385 # must follow switch_to_strict_mode, else it will have no effect.
1386 set -e
Mike Frysinger9ebc8ce2015-04-06 13:15:01 -04001387 trap 'die_err_trap' ERR
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001388 if [[ $# -ne 0 ]]; then
Brian Harring7f175a52012-03-02 05:37:00 -08001389 set "$@"
1390 fi
1391}
1392
1393# TODO: Re-enable this once shflags is set -e safe.
1394#switch_to_strict_mode
Brian Harringf264b822012-09-01 01:39:03 -07001395
Kris Rambish44047912012-09-14 16:56:11 -07001396okboat() {
1397 # http://www.chris.com/ascii/index.php?art=transportation/nautical
1398 echo -e "${V_BOLD_GREEN}"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001399 cat <<BOAT
Kris Rambish44047912012-09-14 16:56:11 -07001400 . o ..
1401 o . o o.o
1402 ...oo_
1403 _[__\___
1404 __|_o_o_o_o\__
1405 OK \' ' ' ' ' ' /
1406 ^^^^^^^^^^^^^^^^^^^^
1407BOAT
1408 echo -e "${V_VIDOFF}"
1409}
1410
1411failboat() {
1412 echo -e "${V_BOLD_RED}"
Mike Frysingerbaae8eb2013-01-15 01:34:26 -05001413 cat <<BOAT
Kris Rambish44047912012-09-14 16:56:11 -07001414 '
1415 ' )
1416 ) (
1417 ( .') __/\
1418 (. /o/` \
1419 __/o/` \
1420 FAIL / /o/` /
1421 ^^^^^^^^^^^^^^^^^^^^
1422BOAT
1423 echo -e "${V_VIDOFF}"
1424 die "$* failed"
1425}