blob: 39c1bfc4d7f922c4a660b1b4d086aa1006866a5e [file] [log] [blame]
Sean Daguee263c822014-12-05 14:25:28 -05001#!/bin/bash
2#
Dean Troyerdff49a22014-01-30 15:37:40 -06003# functions-common - Common functions used by DevStack components
4#
5# The canonical copy of this file is maintained in the DevStack repo.
6# All modifications should be made there and then sync'ed to other repos
7# as required.
8#
9# This file is sorted alphabetically within the function groups.
10#
11# - Config Functions
12# - Control Functions
13# - Distro Functions
14# - Git Functions
15# - OpenStack Functions
16# - Package Functions
17# - Process Functions
Dean Troyerdff49a22014-01-30 15:37:40 -060018# - Service Functions
Masayuki Igawaf6368d32014-02-20 13:31:26 +090019# - System Functions
Dean Troyerdff49a22014-01-30 15:37:40 -060020#
21# The following variables are assumed to be defined by certain functions:
22#
23# - ``ENABLED_SERVICES``
24# - ``ERROR_ON_CLONE``
25# - ``FILES``
26# - ``OFFLINE``
Dean Troyerdff49a22014-01-30 15:37:40 -060027# - ``RECLONE``
Masayuki Igawad20f6322014-02-28 09:22:37 +090028# - ``REQUIREMENTS_DIR``
29# - ``STACK_USER``
Dean Troyerdff49a22014-01-30 15:37:40 -060030# - ``TRACK_DEPENDS``
Masayuki Igawad20f6322014-02-28 09:22:37 +090031# - ``UNDO_REQUIREMENTS``
Dean Troyerdff49a22014-01-30 15:37:40 -060032# - ``http_proxy``, ``https_proxy``, ``no_proxy``
Dean Troyer3324f192014-09-18 09:26:39 -050033#
Dean Troyerdff49a22014-01-30 15:37:40 -060034
35# Save trace setting
36XTRACE=$(set +o | grep xtrace)
37set +o xtrace
38
Sean Daguecc524062014-10-01 09:06:43 -040039# Global Config Variables
40declare -A GITREPO
41declare -A GITBRANCH
42declare -A GITDIR
43
Sean Dague53753292014-12-04 19:38:15 -050044TRACK_DEPENDS=${TRACK_DEPENDS:-False}
45
Dean Troyer68162342015-05-13 15:41:03 -050046# Save these variables to .stackenv
47STACK_ENV_VARS="BASE_SQL_CONN DATA_DIR DEST ENABLED_SERVICES HOST_IP \
48 KEYSTONE_AUTH_PROTOCOL KEYSTONE_AUTH_URI KEYSTONE_SERVICE_URI \
Brian Haley180f5eb2015-06-16 13:14:31 -040049 LOGFILE OS_CACERT SERVICE_HOST SERVICE_PROTOCOL STACK_USER TLS_IP \
50 HOST_IPV6"
Dean Troyer68162342015-05-13 15:41:03 -050051
52
53# Saves significant environment variables to .stackenv for later use
54# Refers to a lot of globals, only TOP_DIR and STACK_ENV_VARS are required to
55# function, the rest are simply saved and do not cause problems if they are undefined.
56# save_stackenv [tag]
57function save_stackenv {
58 local tag=${1:-""}
59 # Save some values we generated for later use
60 time_stamp=$(date "+$TIMESTAMP_FORMAT")
61 echo "# $time_stamp $tag" >$TOP_DIR/.stackenv
62 for i in $STACK_ENV_VARS; do
63 echo $i=${!i} >>$TOP_DIR/.stackenv
64 done
65}
Dean Troyerdff49a22014-01-30 15:37:40 -060066
67# Normalize config values to True or False
68# Accepts as False: 0 no No NO false False FALSE
69# Accepts as True: 1 yes Yes YES true True TRUE
70# VAR=$(trueorfalse default-value test-value)
Ian Wienandaee18c72014-02-21 15:35:08 +110071function trueorfalse {
Sean Dague45917cc2014-02-24 16:09:14 -050072 local xtrace=$(set +o | grep xtrace)
73 set +o xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -060074
Mahito OGURA98f59aa2015-05-11 18:02:34 +090075 local default=$1
76 local testval=${!2:-}
77
78 case "$testval" in
79 "1" | [yY]es | "YES" | [tT]rue | "TRUE" ) echo "True" ;;
80 "0" | [nN]o | "NO" | [fF]alse | "FALSE" ) echo "False" ;;
81 * ) echo "$default" ;;
82 esac
83
Sean Dague45917cc2014-02-24 16:09:14 -050084 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -060085}
86
Attila Fazekas1bd79592015-02-24 14:06:56 +010087function isset {
88 [[ -v "$1" ]]
89}
Dean Troyerdff49a22014-01-30 15:37:40 -060090
Dean Troyer68162342015-05-13 15:41:03 -050091
Dean Troyerdff49a22014-01-30 15:37:40 -060092# Control Functions
93# =================
94
95# Prints backtrace info
96# filename:lineno:function
97# backtrace level
98function backtrace {
99 local level=$1
100 local deep=$((${#BASH_SOURCE[@]} - 1))
101 echo "[Call Trace]"
102 while [ $level -le $deep ]; do
103 echo "${BASH_SOURCE[$deep]}:${BASH_LINENO[$deep-1]}:${FUNCNAME[$deep-1]}"
104 deep=$((deep - 1))
105 done
106}
107
108# Prints line number and "message" then exits
109# die $LINENO "message"
Ian Wienandaee18c72014-02-21 15:35:08 +1100110function die {
Dean Troyerdff49a22014-01-30 15:37:40 -0600111 local exitcode=$?
112 set +o xtrace
113 local line=$1; shift
114 if [ $exitcode == 0 ]; then
115 exitcode=1
116 fi
117 backtrace 2
118 err $line "$*"
Dean Troyera25a6f62014-02-24 16:03:41 -0600119 # Give buffers a second to flush
120 sleep 1
Dean Troyerdff49a22014-01-30 15:37:40 -0600121 exit $exitcode
122}
123
124# Checks an environment variable is not set or has length 0 OR if the
125# exit code is non-zero and prints "message" and exits
126# NOTE: env-var is the variable name without a '$'
127# die_if_not_set $LINENO env-var "message"
Ian Wienandaee18c72014-02-21 15:35:08 +1100128function die_if_not_set {
Dean Troyerdff49a22014-01-30 15:37:40 -0600129 local exitcode=$?
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500130 local xtrace=$(set +o | grep xtrace)
Dean Troyerdff49a22014-01-30 15:37:40 -0600131 set +o xtrace
132 local line=$1; shift
133 local evar=$1; shift
134 if ! is_set $evar || [ $exitcode != 0 ]; then
135 die $line "$*"
136 fi
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500137 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -0600138}
139
140# Prints line number and "message" in error format
141# err $LINENO "message"
Ian Wienandaee18c72014-02-21 15:35:08 +1100142function err {
Dean Troyerdff49a22014-01-30 15:37:40 -0600143 local exitcode=$?
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500144 local xtrace=$(set +o | grep xtrace)
Dean Troyerdff49a22014-01-30 15:37:40 -0600145 set +o xtrace
146 local msg="[ERROR] ${BASH_SOURCE[2]}:$1 $2"
147 echo $msg 1>&2;
Dean Troyerdde41d02014-12-09 17:47:57 -0600148 if [[ -n ${LOGDIR} ]]; then
149 echo $msg >> "${LOGDIR}/error.log"
Dean Troyerdff49a22014-01-30 15:37:40 -0600150 fi
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500151 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -0600152 return $exitcode
153}
154
155# Checks an environment variable is not set or has length 0 OR if the
156# exit code is non-zero and prints "message"
157# NOTE: env-var is the variable name without a '$'
158# err_if_not_set $LINENO env-var "message"
Ian Wienandaee18c72014-02-21 15:35:08 +1100159function err_if_not_set {
Dean Troyerdff49a22014-01-30 15:37:40 -0600160 local exitcode=$?
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500161 local xtrace=$(set +o | grep xtrace)
Dean Troyerdff49a22014-01-30 15:37:40 -0600162 set +o xtrace
163 local line=$1; shift
164 local evar=$1; shift
165 if ! is_set $evar || [ $exitcode != 0 ]; then
166 err $line "$*"
167 fi
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500168 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -0600169 return $exitcode
170}
171
172# Exit after outputting a message about the distribution not being supported.
173# exit_distro_not_supported [optional-string-telling-what-is-missing]
174function exit_distro_not_supported {
175 if [[ -z "$DISTRO" ]]; then
176 GetDistro
177 fi
178
179 if [ $# -gt 0 ]; then
180 die $LINENO "Support for $DISTRO is incomplete: no support for $@"
181 else
182 die $LINENO "Support for $DISTRO is incomplete."
183 fi
184}
185
186# Test if the named environment variable is set and not zero length
187# is_set env-var
Ian Wienandaee18c72014-02-21 15:35:08 +1100188function is_set {
Dean Troyerdff49a22014-01-30 15:37:40 -0600189 local var=\$"$1"
190 eval "[ -n \"$var\" ]" # For ex.: sh -c "[ -n \"$var\" ]" would be better, but several exercises depends on this
191}
192
193# Prints line number and "message" in warning format
194# warn $LINENO "message"
Ian Wienandaee18c72014-02-21 15:35:08 +1100195function warn {
Dean Troyerdff49a22014-01-30 15:37:40 -0600196 local exitcode=$?
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500197 local xtrace=$(set +o | grep xtrace)
Dean Troyerdff49a22014-01-30 15:37:40 -0600198 set +o xtrace
199 local msg="[WARNING] ${BASH_SOURCE[2]}:$1 $2"
Sean Daguee4af9292015-04-28 08:57:57 -0400200 echo $msg
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500201 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -0600202 return $exitcode
203}
204
205
206# Distro Functions
207# ================
208
209# Determine OS Vendor, Release and Update
210# Tested with OS/X, Ubuntu, RedHat, CentOS, Fedora
211# Returns results in global variables:
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500212# ``os_VENDOR`` - vendor name: ``Ubuntu``, ``Fedora``, etc
213# ``os_RELEASE`` - major release: ``14.04`` (Ubuntu), ``20`` (Fedora)
214# ``os_UPDATE`` - update: ex. the ``5`` in ``RHEL6.5``
215# ``os_PACKAGE`` - package type: ``deb`` or ``rpm``
216# ``os_CODENAME`` - vendor's codename for release: ``snow leopard``, ``trusty``
Sean Dague53753292014-12-04 19:38:15 -0500217os_VENDOR=""
218os_RELEASE=""
219os_UPDATE=""
220os_PACKAGE=""
221os_CODENAME=""
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500222
Dean Troyerdff49a22014-01-30 15:37:40 -0600223# GetOSVersion
Ian Wienandaee18c72014-02-21 15:35:08 +1100224function GetOSVersion {
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500225
Dean Troyerdff49a22014-01-30 15:37:40 -0600226 # Figure out which vendor we are
227 if [[ -x "`which sw_vers 2>/dev/null`" ]]; then
228 # OS/X
229 os_VENDOR=`sw_vers -productName`
230 os_RELEASE=`sw_vers -productVersion`
231 os_UPDATE=${os_RELEASE##*.}
232 os_RELEASE=${os_RELEASE%.*}
233 os_PACKAGE=""
234 if [[ "$os_RELEASE" =~ "10.7" ]]; then
235 os_CODENAME="lion"
236 elif [[ "$os_RELEASE" =~ "10.6" ]]; then
237 os_CODENAME="snow leopard"
238 elif [[ "$os_RELEASE" =~ "10.5" ]]; then
239 os_CODENAME="leopard"
240 elif [[ "$os_RELEASE" =~ "10.4" ]]; then
241 os_CODENAME="tiger"
242 elif [[ "$os_RELEASE" =~ "10.3" ]]; then
243 os_CODENAME="panther"
244 else
245 os_CODENAME=""
246 fi
247 elif [[ -x $(which lsb_release 2>/dev/null) ]]; then
248 os_VENDOR=$(lsb_release -i -s)
249 os_RELEASE=$(lsb_release -r -s)
250 os_UPDATE=""
251 os_PACKAGE="rpm"
252 if [[ "Debian,Ubuntu,LinuxMint" =~ $os_VENDOR ]]; then
253 os_PACKAGE="deb"
254 elif [[ "SUSE LINUX" =~ $os_VENDOR ]]; then
255 lsb_release -d -s | grep -q openSUSE
256 if [[ $? -eq 0 ]]; then
257 os_VENDOR="openSUSE"
258 fi
259 elif [[ $os_VENDOR == "openSUSE project" ]]; then
260 os_VENDOR="openSUSE"
261 elif [[ $os_VENDOR =~ Red.*Hat ]]; then
262 os_VENDOR="Red Hat"
263 fi
264 os_CODENAME=$(lsb_release -c -s)
265 elif [[ -r /etc/redhat-release ]]; then
266 # Red Hat Enterprise Linux Server release 5.5 (Tikanga)
267 # Red Hat Enterprise Linux Server release 7.0 Beta (Maipo)
268 # CentOS release 5.5 (Final)
269 # CentOS Linux release 6.0 (Final)
270 # Fedora release 16 (Verne)
271 # XenServer release 6.2.0-70446c (xenenterprise)
Wiekus Beukesec47bc12015-03-19 08:20:38 -0700272 # Oracle Linux release 7
Dean Troyerdff49a22014-01-30 15:37:40 -0600273 os_CODENAME=""
274 for r in "Red Hat" CentOS Fedora XenServer; do
275 os_VENDOR=$r
276 if [[ -n "`grep \"$r\" /etc/redhat-release`" ]]; then
277 ver=`sed -e 's/^.* \([0-9].*\) (\(.*\)).*$/\1\|\2/' /etc/redhat-release`
278 os_CODENAME=${ver#*|}
279 os_RELEASE=${ver%|*}
280 os_UPDATE=${os_RELEASE##*.}
281 os_RELEASE=${os_RELEASE%.*}
282 break
283 fi
284 os_VENDOR=""
285 done
Wiekus Beukesec47bc12015-03-19 08:20:38 -0700286 if [ "$os_VENDOR" = "Red Hat" ] && [[ -r /etc/oracle-release ]]; then
287 os_VENDOR=OracleLinux
288 fi
Dean Troyerdff49a22014-01-30 15:37:40 -0600289 os_PACKAGE="rpm"
290 elif [[ -r /etc/SuSE-release ]]; then
291 for r in openSUSE "SUSE Linux"; do
292 if [[ "$r" = "SUSE Linux" ]]; then
293 os_VENDOR="SUSE LINUX"
294 else
295 os_VENDOR=$r
296 fi
297
298 if [[ -n "`grep \"$r\" /etc/SuSE-release`" ]]; then
299 os_CODENAME=`grep "CODENAME = " /etc/SuSE-release | sed 's:.* = ::g'`
300 os_RELEASE=`grep "VERSION = " /etc/SuSE-release | sed 's:.* = ::g'`
301 os_UPDATE=`grep "PATCHLEVEL = " /etc/SuSE-release | sed 's:.* = ::g'`
302 break
303 fi
304 os_VENDOR=""
305 done
306 os_PACKAGE="rpm"
307 # If lsb_release is not installed, we should be able to detect Debian OS
308 elif [[ -f /etc/debian_version ]] && [[ $(cat /proc/version) =~ "Debian" ]]; then
309 os_VENDOR="Debian"
310 os_PACKAGE="deb"
311 os_CODENAME=$(awk '/VERSION=/' /etc/os-release | sed 's/VERSION=//' | sed -r 's/\"|\(|\)//g' | awk '{print $2}')
312 os_RELEASE=$(awk '/VERSION_ID=/' /etc/os-release | sed 's/VERSION_ID=//' | sed 's/\"//g')
313 fi
314 export os_VENDOR os_RELEASE os_UPDATE os_PACKAGE os_CODENAME
315}
316
317# Translate the OS version values into common nomenclature
318# Sets global ``DISTRO`` from the ``os_*`` values
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500319declare DISTRO
320
Ian Wienandaee18c72014-02-21 15:35:08 +1100321function GetDistro {
Dean Troyerdff49a22014-01-30 15:37:40 -0600322 GetOSVersion
323 if [[ "$os_VENDOR" =~ (Ubuntu) || "$os_VENDOR" =~ (Debian) ]]; then
324 # 'Everyone' refers to Ubuntu / Debian releases by the code name adjective
325 DISTRO=$os_CODENAME
326 elif [[ "$os_VENDOR" =~ (Fedora) ]]; then
327 # For Fedora, just use 'f' and the release
328 DISTRO="f$os_RELEASE"
329 elif [[ "$os_VENDOR" =~ (openSUSE) ]]; then
330 DISTRO="opensuse-$os_RELEASE"
331 elif [[ "$os_VENDOR" =~ (SUSE LINUX) ]]; then
332 # For SLE, also use the service pack
333 if [[ -z "$os_UPDATE" ]]; then
334 DISTRO="sle${os_RELEASE}"
335 else
336 DISTRO="sle${os_RELEASE}sp${os_UPDATE}"
337 fi
anju Tiwari6c639c92014-07-15 18:11:54 +0530338 elif [[ "$os_VENDOR" =~ (Red Hat) || \
339 "$os_VENDOR" =~ (CentOS) || \
Wiekus Beukesec47bc12015-03-19 08:20:38 -0700340 "$os_VENDOR" =~ (OracleLinux) ]]; then
Dean Troyerdff49a22014-01-30 15:37:40 -0600341 # Drop the . release as we assume it's compatible
342 DISTRO="rhel${os_RELEASE::1}"
343 elif [[ "$os_VENDOR" =~ (XenServer) ]]; then
344 DISTRO="xs$os_RELEASE"
345 else
346 # Catch-all for now is Vendor + Release + Update
347 DISTRO="$os_VENDOR-$os_RELEASE.$os_UPDATE"
348 fi
349 export DISTRO
350}
351
352# Utility function for checking machine architecture
353# is_arch arch-type
354function is_arch {
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500355 [[ "$(uname -m)" == "$1" ]]
Dean Troyerdff49a22014-01-30 15:37:40 -0600356}
357
Wiekus Beukesec47bc12015-03-19 08:20:38 -0700358# Determine if current distribution is an Oracle distribution
359# is_oraclelinux
360function is_oraclelinux {
361 if [[ -z "$os_VENDOR" ]]; then
362 GetOSVersion
363 fi
364
365 [ "$os_VENDOR" = "OracleLinux" ]
366}
367
368
Dean Troyerdff49a22014-01-30 15:37:40 -0600369# Determine if current distribution is a Fedora-based distribution
370# (Fedora, RHEL, CentOS, etc).
371# is_fedora
372function is_fedora {
373 if [[ -z "$os_VENDOR" ]]; then
374 GetOSVersion
375 fi
376
anju Tiwari6c639c92014-07-15 18:11:54 +0530377 [ "$os_VENDOR" = "Fedora" ] || [ "$os_VENDOR" = "Red Hat" ] || \
Wiekus Beukesec47bc12015-03-19 08:20:38 -0700378 [ "$os_VENDOR" = "CentOS" ] || [ "$os_VENDOR" = "OracleLinux" ]
Dean Troyerdff49a22014-01-30 15:37:40 -0600379}
380
381
382# Determine if current distribution is a SUSE-based distribution
383# (openSUSE, SLE).
384# is_suse
385function is_suse {
386 if [[ -z "$os_VENDOR" ]]; then
387 GetOSVersion
388 fi
389
390 [ "$os_VENDOR" = "openSUSE" ] || [ "$os_VENDOR" = "SUSE LINUX" ]
391}
392
393
394# Determine if current distribution is an Ubuntu-based distribution
395# It will also detect non-Ubuntu but Debian-based distros
396# is_ubuntu
397function is_ubuntu {
398 if [[ -z "$os_PACKAGE" ]]; then
399 GetOSVersion
400 fi
401 [ "$os_PACKAGE" = "deb" ]
402}
403
404
405# Git Functions
406# =============
407
Dean Troyerabc7b1d2014-02-12 12:09:22 -0600408# Returns openstack release name for a given branch name
409# ``get_release_name_from_branch branch-name``
Ian Wienandaee18c72014-02-21 15:35:08 +1100410function get_release_name_from_branch {
Dean Troyerabc7b1d2014-02-12 12:09:22 -0600411 local branch=$1
Adam Gandelman8f385722014-10-14 15:50:18 -0700412 if [[ $branch =~ "stable/" || $branch =~ "proposed/" ]]; then
Dean Troyerabc7b1d2014-02-12 12:09:22 -0600413 echo ${branch#*/}
414 else
415 echo "master"
416 fi
417}
418
Dean Troyerdff49a22014-01-30 15:37:40 -0600419# git clone only if directory doesn't exist already. Since ``DEST`` might not
420# be owned by the installation user, we create the directory and change the
421# ownership to the proper user.
Dean Troyer50cda692014-07-25 11:57:20 -0500422# Set global ``RECLONE=yes`` to simulate a clone when dest-dir exists
423# Set global ``ERROR_ON_CLONE=True`` to abort execution with an error if the git repo
Dean Troyerdff49a22014-01-30 15:37:40 -0600424# does not exist (default is False, meaning the repo will be cloned).
Sean Dague53753292014-12-04 19:38:15 -0500425# Uses globals ``ERROR_ON_CLONE``, ``OFFLINE``, ``RECLONE``
Dean Troyerdff49a22014-01-30 15:37:40 -0600426# git_clone remote dest-dir branch
427function git_clone {
Dean Troyer50cda692014-07-25 11:57:20 -0500428 local git_remote=$1
429 local git_dest=$2
430 local git_ref=$3
431 local orig_dir=$(pwd)
Jamie Lennox51f0de52014-10-20 16:32:34 +0200432 local git_clone_flags=""
Dean Troyer50cda692014-07-25 11:57:20 -0500433
Sean Dague53753292014-12-04 19:38:15 -0500434 RECLONE=$(trueorfalse False RECLONE)
Kevin Benton59d52f32015-01-17 11:29:12 -0800435 if [[ "${GIT_DEPTH}" -gt 0 ]]; then
Jamie Lennox51f0de52014-10-20 16:32:34 +0200436 git_clone_flags="$git_clone_flags --depth $GIT_DEPTH"
437 fi
438
Dean Troyerdff49a22014-01-30 15:37:40 -0600439 if [[ "$OFFLINE" = "True" ]]; then
440 echo "Running in offline mode, clones already exist"
441 # print out the results so we know what change was used in the logs
Dean Troyer50cda692014-07-25 11:57:20 -0500442 cd $git_dest
Dean Troyerdff49a22014-01-30 15:37:40 -0600443 git show --oneline | head -1
Sean Dague64bd0162014-03-12 13:04:22 -0400444 cd $orig_dir
Dean Troyerdff49a22014-01-30 15:37:40 -0600445 return
446 fi
447
Dean Troyer50cda692014-07-25 11:57:20 -0500448 if echo $git_ref | egrep -q "^refs"; then
Dean Troyerdff49a22014-01-30 15:37:40 -0600449 # If our branch name is a gerrit style refs/changes/...
Dean Troyer50cda692014-07-25 11:57:20 -0500450 if [[ ! -d $git_dest ]]; then
Dean Troyerdff49a22014-01-30 15:37:40 -0600451 [[ "$ERROR_ON_CLONE" = "True" ]] && \
452 die $LINENO "Cloning not allowed in this configuration"
Jamie Lennox51f0de52014-10-20 16:32:34 +0200453 git_timed clone $git_clone_flags $git_remote $git_dest
Dean Troyerdff49a22014-01-30 15:37:40 -0600454 fi
Dean Troyer50cda692014-07-25 11:57:20 -0500455 cd $git_dest
456 git_timed fetch $git_remote $git_ref && git checkout FETCH_HEAD
Dean Troyerdff49a22014-01-30 15:37:40 -0600457 else
458 # do a full clone only if the directory doesn't exist
Dean Troyer50cda692014-07-25 11:57:20 -0500459 if [[ ! -d $git_dest ]]; then
Dean Troyerdff49a22014-01-30 15:37:40 -0600460 [[ "$ERROR_ON_CLONE" = "True" ]] && \
461 die $LINENO "Cloning not allowed in this configuration"
Jamie Lennox51f0de52014-10-20 16:32:34 +0200462 git_timed clone $git_clone_flags $git_remote $git_dest
Dean Troyer50cda692014-07-25 11:57:20 -0500463 cd $git_dest
Dean Troyerdff49a22014-01-30 15:37:40 -0600464 # This checkout syntax works for both branches and tags
Dean Troyer50cda692014-07-25 11:57:20 -0500465 git checkout $git_ref
Dean Troyerdff49a22014-01-30 15:37:40 -0600466 elif [[ "$RECLONE" = "True" ]]; then
467 # if it does exist then simulate what clone does if asked to RECLONE
Dean Troyer50cda692014-07-25 11:57:20 -0500468 cd $git_dest
Dean Troyerdff49a22014-01-30 15:37:40 -0600469 # set the url to pull from and fetch
Dean Troyer50cda692014-07-25 11:57:20 -0500470 git remote set-url origin $git_remote
Ian Wienandd53ad0b2014-02-20 13:55:13 +1100471 git_timed fetch origin
Dean Troyerdff49a22014-01-30 15:37:40 -0600472 # remove the existing ignored files (like pyc) as they cause breakage
473 # (due to the py files having older timestamps than our pyc, so python
474 # thinks the pyc files are correct using them)
Dean Troyer50cda692014-07-25 11:57:20 -0500475 find $git_dest -name '*.pyc' -delete
Dean Troyerdff49a22014-01-30 15:37:40 -0600476
Dean Troyer50cda692014-07-25 11:57:20 -0500477 # handle git_ref accordingly to type (tag, branch)
478 if [[ -n "`git show-ref refs/tags/$git_ref`" ]]; then
479 git_update_tag $git_ref
480 elif [[ -n "`git show-ref refs/heads/$git_ref`" ]]; then
481 git_update_branch $git_ref
482 elif [[ -n "`git show-ref refs/remotes/origin/$git_ref`" ]]; then
483 git_update_remote_branch $git_ref
Dean Troyerdff49a22014-01-30 15:37:40 -0600484 else
Dean Troyer50cda692014-07-25 11:57:20 -0500485 die $LINENO "$git_ref is neither branch nor tag"
Dean Troyerdff49a22014-01-30 15:37:40 -0600486 fi
487
488 fi
489 fi
490
491 # print out the results so we know what change was used in the logs
Dean Troyer50cda692014-07-25 11:57:20 -0500492 cd $git_dest
Dean Troyerdff49a22014-01-30 15:37:40 -0600493 git show --oneline | head -1
Sean Dague64bd0162014-03-12 13:04:22 -0400494 cd $orig_dir
Dean Troyerdff49a22014-01-30 15:37:40 -0600495}
496
Sean Daguecc524062014-10-01 09:06:43 -0400497# A variation on git clone that lets us specify a project by it's
498# actual name, like oslo.config. This is exceptionally useful in the
499# library installation case
500function git_clone_by_name {
501 local name=$1
502 local repo=${GITREPO[$name]}
503 local dir=${GITDIR[$name]}
504 local branch=${GITBRANCH[$name]}
505 git_clone $repo $dir $branch
506}
507
508
Ian Wienandd53ad0b2014-02-20 13:55:13 +1100509# git can sometimes get itself infinitely stuck with transient network
510# errors or other issues with the remote end. This wraps git in a
511# timeout/retry loop and is intended to watch over non-local git
512# processes that might hang. GIT_TIMEOUT, if set, is passed directly
513# to timeout(1); otherwise the default value of 0 maintains the status
514# quo of waiting forever.
515# usage: git_timed <git-command>
Ian Wienandaee18c72014-02-21 15:35:08 +1100516function git_timed {
Ian Wienandd53ad0b2014-02-20 13:55:13 +1100517 local count=0
518 local timeout=0
519
520 if [[ -n "${GIT_TIMEOUT}" ]]; then
521 timeout=${GIT_TIMEOUT}
522 fi
523
524 until timeout -s SIGINT ${timeout} git "$@"; do
525 # 124 is timeout(1)'s special return code when it reached the
526 # timeout; otherwise assume fatal failure
527 if [[ $? -ne 124 ]]; then
528 die $LINENO "git call failed: [git $@]"
529 fi
530
531 count=$(($count + 1))
Sean Daguee4af9292015-04-28 08:57:57 -0400532 warn $LINENO "timeout ${count} for git call: [git $@]"
Ian Wienandd53ad0b2014-02-20 13:55:13 +1100533 if [ $count -eq 3 ]; then
534 die $LINENO "Maximum of 3 git retries reached"
535 fi
536 sleep 5
537 done
538}
539
Dean Troyerdff49a22014-01-30 15:37:40 -0600540# git update using reference as a branch.
541# git_update_branch ref
Ian Wienandaee18c72014-02-21 15:35:08 +1100542function git_update_branch {
Dean Troyer50cda692014-07-25 11:57:20 -0500543 local git_branch=$1
Dean Troyerdff49a22014-01-30 15:37:40 -0600544
Dean Troyer50cda692014-07-25 11:57:20 -0500545 git checkout -f origin/$git_branch
Dean Troyerdff49a22014-01-30 15:37:40 -0600546 # a local branch might not exist
Dean Troyer50cda692014-07-25 11:57:20 -0500547 git branch -D $git_branch || true
548 git checkout -b $git_branch
Dean Troyerdff49a22014-01-30 15:37:40 -0600549}
550
551# git update using reference as a branch.
552# git_update_remote_branch ref
Ian Wienandaee18c72014-02-21 15:35:08 +1100553function git_update_remote_branch {
Dean Troyer50cda692014-07-25 11:57:20 -0500554 local git_branch=$1
Dean Troyerdff49a22014-01-30 15:37:40 -0600555
Dean Troyer50cda692014-07-25 11:57:20 -0500556 git checkout -b $git_branch -t origin/$git_branch
Dean Troyerdff49a22014-01-30 15:37:40 -0600557}
558
559# git update using reference as a tag. Be careful editing source at that repo
560# as working copy will be in a detached mode
561# git_update_tag ref
Ian Wienandaee18c72014-02-21 15:35:08 +1100562function git_update_tag {
Dean Troyer50cda692014-07-25 11:57:20 -0500563 local git_tag=$1
Dean Troyerdff49a22014-01-30 15:37:40 -0600564
Dean Troyer50cda692014-07-25 11:57:20 -0500565 git tag -d $git_tag
Dean Troyerdff49a22014-01-30 15:37:40 -0600566 # fetching given tag only
Dean Troyer50cda692014-07-25 11:57:20 -0500567 git_timed fetch origin tag $git_tag
568 git checkout -f $git_tag
Dean Troyerdff49a22014-01-30 15:37:40 -0600569}
570
571
572# OpenStack Functions
573# ===================
574
575# Get the default value for HOST_IP
576# get_default_host_ip fixed_range floating_range host_ip_iface host_ip
Ian Wienandaee18c72014-02-21 15:35:08 +1100577function get_default_host_ip {
Dean Troyerdff49a22014-01-30 15:37:40 -0600578 local fixed_range=$1
579 local floating_range=$2
580 local host_ip_iface=$3
581 local host_ip=$4
Brian Haley180f5eb2015-06-16 13:14:31 -0400582 local af=$5
Dean Troyerdff49a22014-01-30 15:37:40 -0600583
Dean Troyerdff49a22014-01-30 15:37:40 -0600584 # Search for an IP unless an explicit is set by ``HOST_IP`` environment variable
585 if [ -z "$host_ip" -o "$host_ip" == "dhcp" ]; then
586 host_ip=""
Andreas Scheuringa3430272015-03-09 16:55:32 +0100587 # Find the interface used for the default route
Brian Haley180f5eb2015-06-16 13:14:31 -0400588 host_ip_iface=${host_ip_iface:-$(ip -f $af route | awk '/default/ {print $5}' | head -1)}
589 local host_ips=$(LC_ALL=C ip -f $af addr show ${host_ip_iface} | awk /$af'/ {split($2,parts,"/"); print parts[1]}')
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500590 local ip
591 for ip in $host_ips; do
Dean Troyerdff49a22014-01-30 15:37:40 -0600592 # Attempt to filter out IP addresses that are part of the fixed and
593 # floating range. Note that this method only works if the ``netaddr``
594 # python library is installed. If it is not installed, an error
595 # will be printed and the first IP from the interface will be used.
596 # If that is not correct set ``HOST_IP`` in ``localrc`` to the correct
597 # address.
Brian Haley180f5eb2015-06-16 13:14:31 -0400598 if [[ "$af" == "inet6" ]]; then
599 host_ip=$ip
600 break;
601 fi
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500602 if ! (address_in_net $ip $fixed_range || address_in_net $ip $floating_range); then
603 host_ip=$ip
Dean Troyerdff49a22014-01-30 15:37:40 -0600604 break;
605 fi
606 done
607 fi
608 echo $host_ip
609}
610
Attila Fazekasf71b5002014-05-28 09:52:22 +0200611# Generates hex string from ``size`` byte of pseudo random data
612# generate_hex_string size
613function generate_hex_string {
614 local size=$1
615 hexdump -n "$size" -v -e '/1 "%02x"' /dev/urandom
616}
617
Dean Troyerdff49a22014-01-30 15:37:40 -0600618# Grab a numbered field from python prettytable output
619# Fields are numbered starting with 1
620# Reverse syntax is supported: -1 is the last field, -2 is second to last, etc.
621# get_field field-number
Ian Wienandaee18c72014-02-21 15:35:08 +1100622function get_field {
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500623 local data field
Dean Troyerdff49a22014-01-30 15:37:40 -0600624 while read data; do
625 if [ "$1" -lt 0 ]; then
626 field="(\$(NF$1))"
627 else
628 field="\$$(($1 + 1))"
629 fi
630 echo "$data" | awk -F'[ \t]*\\|[ \t]*' "{print $field}"
631 done
632}
633
yuntongjinf26deea2015-02-28 10:50:34 +0800634# install default policy
635# copy over a default policy.json and policy.d for projects
636function install_default_policy {
637 local project=$1
638 local project_uc=$(echo $1|tr a-z A-Z)
639 local conf_dir="${project_uc}_CONF_DIR"
640 # eval conf dir to get the variable
641 conf_dir="${!conf_dir}"
642 local project_dir="${project_uc}_DIR"
643 # eval project dir to get the variable
644 project_dir="${!project_dir}"
645 local sample_conf_dir="${project_dir}/etc/${project}"
646 local sample_policy_dir="${project_dir}/etc/${project}/policy.d"
647
648 # first copy any policy.json
649 cp -p $sample_conf_dir/policy.json $conf_dir
650 # then optionally copy over policy.d
651 if [[ -d $sample_policy_dir ]]; then
652 cp -r $sample_policy_dir $conf_dir/policy.d
653 fi
654}
655
Dean Troyerdff49a22014-01-30 15:37:40 -0600656# Add a policy to a policy.json file
657# Do nothing if the policy already exists
658# ``policy_add policy_file policy_name policy_permissions``
Ian Wienandaee18c72014-02-21 15:35:08 +1100659function policy_add {
Dean Troyerdff49a22014-01-30 15:37:40 -0600660 local policy_file=$1
661 local policy_name=$2
662 local policy_perm=$3
663
664 if grep -q ${policy_name} ${policy_file}; then
665 echo "Policy ${policy_name} already exists in ${policy_file}"
666 return
667 fi
668
669 # Add a terminating comma to policy lines without one
670 # Remove the closing '}' and all lines following to the end-of-file
671 local tmpfile=$(mktemp)
672 uniq ${policy_file} | sed -e '
673 s/]$/],/
674 /^[}]/,$d
675 ' > ${tmpfile}
676
677 # Append policy and closing brace
678 echo " \"${policy_name}\": ${policy_perm}" >>${tmpfile}
679 echo "}" >>${tmpfile}
680
681 mv ${tmpfile} ${policy_file}
682}
683
Alistair Coles24779f62014-10-15 18:57:59 +0100684# Gets or creates a domain
685# Usage: get_or_create_domain <name> <description>
686function get_or_create_domain {
Steve Martinellib74e01c2014-12-18 01:35:35 -0500687 local os_url="$KEYSTONE_SERVICE_URI_V3"
Alistair Coles24779f62014-10-15 18:57:59 +0100688 # Gets domain id
689 local domain_id=$(
690 # Gets domain id
691 openstack --os-token=$OS_TOKEN --os-url=$os_url \
692 --os-identity-api-version=3 domain show $1 \
693 -f value -c id 2>/dev/null ||
694 # Creates new domain
695 openstack --os-token=$OS_TOKEN --os-url=$os_url \
696 --os-identity-api-version=3 domain create $1 \
697 --description "$2" \
698 -f value -c id
699 )
700 echo $domain_id
701}
702
Steve Martinellib74e01c2014-12-18 01:35:35 -0500703# Gets or creates group
Jamie Lennox9d7e7762015-05-29 01:08:53 +0000704# Usage: get_or_create_group <groupname> <domain> [<description>]
Steve Martinellib74e01c2014-12-18 01:35:35 -0500705function get_or_create_group {
Steve Martinellib74e01c2014-12-18 01:35:35 -0500706 local desc="${3:-}"
707 local os_url="$KEYSTONE_SERVICE_URI_V3"
708 # Gets group id
709 local group_id=$(
710 # Creates new group with --or-show
711 openstack --os-token=$OS_TOKEN --os-url=$os_url \
712 --os-identity-api-version=3 group create $1 \
Jamie Lennox9d7e7762015-05-29 01:08:53 +0000713 --domain $2 --description "$desc" --or-show \
Steve Martinellib74e01c2014-12-18 01:35:35 -0500714 -f value -c id
715 )
716 echo $group_id
717}
718
Bartosz Górski0abde392014-02-28 14:15:19 +0100719# Gets or creates user
Jamie Lennox9d7e7762015-05-29 01:08:53 +0000720# Usage: get_or_create_user <username> <password> <domain> [<email>]
Bartosz Górski0abde392014-02-28 14:15:19 +0100721function get_or_create_user {
Jamie Lennox9d7e7762015-05-29 01:08:53 +0000722 if [[ ! -z "$4" ]]; then
723 local email="--email=$4"
Gael Chamoulaud6dd8a8b2014-07-22 01:12:12 +0200724 else
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500725 local email=""
Gael Chamoulaud6dd8a8b2014-07-22 01:12:12 +0200726 fi
Bartosz Górski0abde392014-02-28 14:15:19 +0100727 # Gets user id
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500728 local user_id=$(
Steve Martinelli245daa22014-11-14 02:17:22 -0500729 # Creates new user with --or-show
Jamie Lennox9d7e7762015-05-29 01:08:53 +0000730 openstack user create \
Bartosz Górski0abde392014-02-28 14:15:19 +0100731 $1 \
732 --password "$2" \
Jamie Lennox9d7e7762015-05-29 01:08:53 +0000733 --os-url=$KEYSTONE_SERVICE_URI_V3 \
734 --os-identity-api-version=3 \
735 --domain=$3 \
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500736 $email \
Steve Martinelli245daa22014-11-14 02:17:22 -0500737 --or-show \
Bartosz Górski0abde392014-02-28 14:15:19 +0100738 -f value -c id
739 )
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500740 echo $user_id
Bartosz Górski0abde392014-02-28 14:15:19 +0100741}
742
743# Gets or creates project
Jamie Lennoxb632c9e2015-05-28 23:36:15 +0000744# Usage: get_or_create_project <name> <domain>
Bartosz Górski0abde392014-02-28 14:15:19 +0100745function get_or_create_project {
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500746 local project_id=$(
Steve Martinelli245daa22014-11-14 02:17:22 -0500747 # Creates new project with --or-show
Jamie Lennoxb632c9e2015-05-28 23:36:15 +0000748 openstack --os-url=$KEYSTONE_SERVICE_URI_V3 \
749 --os-identity-api-version=3 \
750 project create $1 \
751 --domain=$2 \
752 --or-show -f value -c id
Bartosz Górski0abde392014-02-28 14:15:19 +0100753 )
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500754 echo $project_id
Bartosz Górski0abde392014-02-28 14:15:19 +0100755}
756
757# Gets or creates role
758# Usage: get_or_create_role <name>
759function get_or_create_role {
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500760 local role_id=$(
Steve Martinelli245daa22014-11-14 02:17:22 -0500761 # Creates role with --or-show
762 openstack role create $1 --or-show -f value -c id
Bartosz Górski0abde392014-02-28 14:15:19 +0100763 )
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500764 echo $role_id
Bartosz Górski0abde392014-02-28 14:15:19 +0100765}
766
Jamie Lennox9b215db2015-02-10 18:19:57 +1100767# Gets or adds user role to project
768# Usage: get_or_add_user_project_role <role> <user> <project>
769function get_or_add_user_project_role {
Bartosz Górski0abde392014-02-28 14:15:19 +0100770 # Gets user role id
Steve Martinelli5541a612015-01-19 15:58:49 -0500771 local user_role_id=$(openstack role list \
772 --user $2 \
Bartosz Górski0abde392014-02-28 14:15:19 +0100773 --project $3 \
774 --column "ID" \
775 --column "Name" \
776 | grep " $1 " | get_field 1)
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500777 if [[ -z "$user_role_id" ]]; then
Bartosz Górski0abde392014-02-28 14:15:19 +0100778 # Adds role to user
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500779 user_role_id=$(openstack role add \
Bartosz Górski0abde392014-02-28 14:15:19 +0100780 $1 \
781 --user $2 \
782 --project $3 \
783 | grep " id " | get_field 2)
784 fi
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500785 echo $user_role_id
Bartosz Górski0abde392014-02-28 14:15:19 +0100786}
787
Steve Martinelli4599fd12015-03-12 21:30:58 -0400788# Gets or adds group role to project
789# Usage: get_or_add_group_project_role <role> <group> <project>
790function get_or_add_group_project_role {
791 # Gets group role id
792 local group_role_id=$(openstack role list \
793 --group $2 \
794 --project $3 \
795 --column "ID" \
796 --column "Name" \
797 | grep " $1 " | get_field 1)
798 if [[ -z "$group_role_id" ]]; then
799 # Adds role to group
800 group_role_id=$(openstack role add \
801 $1 \
802 --group $2 \
803 --project $3 \
804 | grep " id " | get_field 2)
805 fi
806 echo $group_role_id
807}
808
Bartosz Górski0abde392014-02-28 14:15:19 +0100809# Gets or creates service
810# Usage: get_or_create_service <name> <type> <description>
811function get_or_create_service {
812 # Gets service id
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500813 local service_id=$(
Bartosz Górski0abde392014-02-28 14:15:19 +0100814 # Gets service id
815 openstack service show $1 -f value -c id 2>/dev/null ||
816 # Creates new service if not exists
817 openstack service create \
Steve Martinelli789af5c2015-01-19 16:11:44 -0500818 $2 \
819 --name $1 \
Bartosz Górski0abde392014-02-28 14:15:19 +0100820 --description="$3" \
821 -f value -c id
822 )
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500823 echo $service_id
Bartosz Górski0abde392014-02-28 14:15:19 +0100824}
825
826# Gets or creates endpoint
827# Usage: get_or_create_endpoint <service> <region> <publicurl> <adminurl> <internalurl>
828function get_or_create_endpoint {
829 # Gets endpoint id
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500830 local endpoint_id=$(openstack endpoint list \
Bartosz Górski0abde392014-02-28 14:15:19 +0100831 --column "ID" \
832 --column "Region" \
833 --column "Service Name" \
834 | grep " $2 " \
835 | grep " $1 " | get_field 1)
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500836 if [[ -z "$endpoint_id" ]]; then
Bartosz Górski0abde392014-02-28 14:15:19 +0100837 # Creates new endpoint
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500838 endpoint_id=$(openstack endpoint create \
Bartosz Górski0abde392014-02-28 14:15:19 +0100839 $1 \
840 --region $2 \
841 --publicurl $3 \
842 --adminurl $4 \
843 --internalurl $5 \
844 | grep " id " | get_field 2)
845 fi
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500846 echo $endpoint_id
Bartosz Górski0abde392014-02-28 14:15:19 +0100847}
Dean Troyerdff49a22014-01-30 15:37:40 -0600848
Dean Troyerd5dfa4c2014-07-25 11:13:11 -0500849
Dean Troyerdff49a22014-01-30 15:37:40 -0600850# Package Functions
851# =================
852
853# _get_package_dir
Ian Wienandaee18c72014-02-21 15:35:08 +1100854function _get_package_dir {
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800855 local base_dir=$1
Dean Troyerdff49a22014-01-30 15:37:40 -0600856 local pkg_dir
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800857
858 if [[ -z "$base_dir" ]]; then
859 base_dir=$FILES
860 fi
Dean Troyerdff49a22014-01-30 15:37:40 -0600861 if is_ubuntu; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800862 pkg_dir=$base_dir/debs
Dean Troyerdff49a22014-01-30 15:37:40 -0600863 elif is_fedora; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800864 pkg_dir=$base_dir/rpms
Dean Troyerdff49a22014-01-30 15:37:40 -0600865 elif is_suse; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800866 pkg_dir=$base_dir/rpms-suse
Dean Troyerdff49a22014-01-30 15:37:40 -0600867 else
868 exit_distro_not_supported "list of packages"
869 fi
870 echo "$pkg_dir"
871}
872
873# Wrapper for ``apt-get`` to set cache and proxy environment variables
874# Uses globals ``OFFLINE``, ``*_proxy``
875# apt_get operation package [package ...]
Ian Wienandaee18c72014-02-21 15:35:08 +1100876function apt_get {
Sean Dague45917cc2014-02-24 16:09:14 -0500877 local xtrace=$(set +o | grep xtrace)
878 set +o xtrace
879
Dean Troyerdff49a22014-01-30 15:37:40 -0600880 [[ "$OFFLINE" = "True" || -z "$@" ]] && return
881 local sudo="sudo"
882 [[ "$(id -u)" = "0" ]] && sudo="env"
Sean Dague45917cc2014-02-24 16:09:14 -0500883
884 $xtrace
Sean Dague53753292014-12-04 19:38:15 -0500885
Dean Troyerdff49a22014-01-30 15:37:40 -0600886 $sudo DEBIAN_FRONTEND=noninteractive \
Sean Dague53753292014-12-04 19:38:15 -0500887 http_proxy=${http_proxy:-} https_proxy=${https_proxy:-} \
888 no_proxy=${no_proxy:-} \
Dean Troyerdff49a22014-01-30 15:37:40 -0600889 apt-get --option "Dpkg::Options::=--force-confold" --assume-yes "$@"
890}
891
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800892function _parse_package_files {
893 local files_to_parse=$@
Dean Troyerdff49a22014-01-30 15:37:40 -0600894
Dean Troyerdff49a22014-01-30 15:37:40 -0600895 if [[ -z "$DISTRO" ]]; then
896 GetDistro
897 fi
Dean Troyerdff49a22014-01-30 15:37:40 -0600898
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800899 for fname in ${files_to_parse}; do
Dean Troyerdff49a22014-01-30 15:37:40 -0600900 local OIFS line package distros distro
901 [[ -e $fname ]] || continue
902
903 OIFS=$IFS
904 IFS=$'\n'
905 for line in $(<${fname}); do
906 if [[ $line =~ "NOPRIME" ]]; then
907 continue
908 fi
909
910 # Assume we want this package
911 package=${line%#*}
912 inst_pkg=1
913
914 # Look for # dist:xxx in comment
915 if [[ $line =~ (.*)#.*dist:([^ ]*) ]]; then
916 # We are using BASH regexp matching feature.
917 package=${BASH_REMATCH[1]}
918 distros=${BASH_REMATCH[2]}
919 # In bash ${VAR,,} will lowecase VAR
920 # Look for a match in the distro list
921 if [[ ! ${distros,,} =~ ${DISTRO,,} ]]; then
922 # If no match then skip this package
923 inst_pkg=0
924 fi
925 fi
926
Dean Troyerdff49a22014-01-30 15:37:40 -0600927 if [[ $inst_pkg = 1 ]]; then
928 echo $package
929 fi
930 done
931 IFS=$OIFS
932 done
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800933}
934
935# get_packages() collects a list of package names of any type from the
936# prerequisite files in ``files/{debs|rpms}``. The list is intended
937# to be passed to a package installer such as apt or yum.
938#
939# Only packages required for the services in 1st argument will be
940# included. Two bits of metadata are recognized in the prerequisite files:
941#
942# - ``# NOPRIME`` defers installation to be performed later in `stack.sh`
943# - ``# dist:DISTRO`` or ``dist:DISTRO1,DISTRO2`` limits the selection
944# of the package to the distros listed. The distro names are case insensitive.
945function get_packages {
946 local xtrace=$(set +o | grep xtrace)
947 set +o xtrace
948 local services=$@
949 local package_dir=$(_get_package_dir)
950 local file_to_parse=""
951 local service=""
952
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800953 if [[ -z "$package_dir" ]]; then
954 echo "No package directory supplied"
955 return 1
956 fi
957 for service in ${services//,/ }; do
958 # Allow individual services to specify dependencies
959 if [[ -e ${package_dir}/${service} ]]; then
960 file_to_parse="${file_to_parse} ${package_dir}/${service}"
961 fi
962 # NOTE(sdague) n-api needs glance for now because that's where
963 # glance client is
964 if [[ $service == n-api ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700965 if [[ ! $file_to_parse =~ $package_dir/nova ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800966 file_to_parse="${file_to_parse} ${package_dir}/nova"
967 fi
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700968 if [[ ! $file_to_parse =~ $package_dir/glance ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800969 file_to_parse="${file_to_parse} ${package_dir}/glance"
970 fi
971 elif [[ $service == c-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700972 if [[ ! $file_to_parse =~ $package_dir/cinder ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800973 file_to_parse="${file_to_parse} ${package_dir}/cinder"
974 fi
975 elif [[ $service == ceilometer-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700976 if [[ ! $file_to_parse =~ $package_dir/ceilometer ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800977 file_to_parse="${file_to_parse} ${package_dir}/ceilometer"
978 fi
979 elif [[ $service == s-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700980 if [[ ! $file_to_parse =~ $package_dir/swift ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800981 file_to_parse="${file_to_parse} ${package_dir}/swift"
982 fi
983 elif [[ $service == n-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700984 if [[ ! $file_to_parse =~ $package_dir/nova ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800985 file_to_parse="${file_to_parse} ${package_dir}/nova"
986 fi
987 elif [[ $service == g-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700988 if [[ ! $file_to_parse =~ $package_dir/glance ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800989 file_to_parse="${file_to_parse} ${package_dir}/glance"
990 fi
991 elif [[ $service == key* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700992 if [[ ! $file_to_parse =~ $package_dir/keystone ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800993 file_to_parse="${file_to_parse} ${package_dir}/keystone"
994 fi
995 elif [[ $service == q-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -0700996 if [[ ! $file_to_parse =~ $package_dir/neutron ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -0800997 file_to_parse="${file_to_parse} ${package_dir}/neutron"
998 fi
999 elif [[ $service == ir-* ]]; then
Ryan Hsu6f3f3102015-03-19 16:26:45 -07001000 if [[ ! $file_to_parse =~ $package_dir/ironic ]]; then
Adam Gandelman7ca90cd2015-03-04 17:25:07 -08001001 file_to_parse="${file_to_parse} ${package_dir}/ironic"
1002 fi
1003 fi
1004 done
1005 echo "$(_parse_package_files $file_to_parse)"
1006 $xtrace
1007}
1008
1009# get_plugin_packages() collects a list of package names of any type from a
1010# plugin's prerequisite files in ``$PLUGIN/devstack/files/{debs|rpms}``. The
1011# list is intended to be passed to a package installer such as apt or yum.
1012#
1013# Only packages required for enabled and collected plugins will included.
1014#
Dean Troyerdc97cb72015-03-28 08:20:50 -05001015# The same metadata used in the main DevStack prerequisite files may be used
Adam Gandelman7ca90cd2015-03-04 17:25:07 -08001016# in these prerequisite files, see get_packages() for more info.
1017function get_plugin_packages {
1018 local xtrace=$(set +o | grep xtrace)
1019 set +o xtrace
1020 local files_to_parse=""
1021 local package_dir=""
1022 for plugin in ${DEVSTACK_PLUGINS//,/ }; do
1023 local package_dir="$(_get_package_dir ${GITDIR[$plugin]}/devstack/files)"
1024 files_to_parse+="$package_dir/$plugin"
1025 done
1026 echo "$(_parse_package_files $files_to_parse)"
Sean Dague45917cc2014-02-24 16:09:14 -05001027 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -06001028}
1029
1030# Distro-agnostic package installer
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001031# Uses globals ``NO_UPDATE_REPOS``, ``REPOS_UPDATED``, ``RETRY_UPDATE``
Dean Troyerdff49a22014-01-30 15:37:40 -06001032# install_package package [package ...]
Monty Taylor5cc6d2c2014-06-06 08:45:16 -04001033function update_package_repo {
Sean Dague53753292014-12-04 19:38:15 -05001034 NO_UPDATE_REPOS=${NO_UPDATE_REPOS:-False}
1035 REPOS_UPDATED=${REPOS_UPDATED:-False}
1036 RETRY_UPDATE=${RETRY_UPDATE:-False}
1037
Paul Linchpiner9e179742014-07-13 22:23:00 -07001038 if [[ "$NO_UPDATE_REPOS" = "True" ]]; then
Monty Taylor5cc6d2c2014-06-06 08:45:16 -04001039 return 0
1040 fi
Dean Troyerdff49a22014-01-30 15:37:40 -06001041
Monty Taylor5cc6d2c2014-06-06 08:45:16 -04001042 if is_ubuntu; then
1043 local xtrace=$(set +o | grep xtrace)
1044 set +o xtrace
1045 if [[ "$REPOS_UPDATED" != "True" || "$RETRY_UPDATE" = "True" ]]; then
1046 # if there are transient errors pulling the updates, that's fine.
1047 # It may be secondary repositories that we don't really care about.
1048 apt_get update || /bin/true
1049 REPOS_UPDATED=True
1050 fi
Sean Dague45917cc2014-02-24 16:09:14 -05001051 $xtrace
Monty Taylor5cc6d2c2014-06-06 08:45:16 -04001052 fi
1053}
1054
1055function real_install_package {
1056 if is_ubuntu; then
Dean Troyerdff49a22014-01-30 15:37:40 -06001057 apt_get install "$@"
1058 elif is_fedora; then
1059 yum_install "$@"
1060 elif is_suse; then
1061 zypper_install "$@"
1062 else
1063 exit_distro_not_supported "installing packages"
1064 fi
1065}
1066
Monty Taylor5cc6d2c2014-06-06 08:45:16 -04001067# Distro-agnostic package installer
1068# install_package package [package ...]
1069function install_package {
1070 update_package_repo
1071 real_install_package $@ || RETRY_UPDATE=True update_package_repo && real_install_package $@
1072}
1073
Dean Troyerdff49a22014-01-30 15:37:40 -06001074# Distro-agnostic function to tell if a package is installed
1075# is_package_installed package [package ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001076function is_package_installed {
Dean Troyerdff49a22014-01-30 15:37:40 -06001077 if [[ -z "$@" ]]; then
1078 return 1
1079 fi
1080
1081 if [[ -z "$os_PACKAGE" ]]; then
1082 GetOSVersion
1083 fi
1084
1085 if [[ "$os_PACKAGE" = "deb" ]]; then
1086 dpkg -s "$@" > /dev/null 2> /dev/null
1087 elif [[ "$os_PACKAGE" = "rpm" ]]; then
1088 rpm --quiet -q "$@"
1089 else
1090 exit_distro_not_supported "finding if a package is installed"
1091 fi
1092}
1093
1094# Distro-agnostic package uninstaller
1095# uninstall_package package [package ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001096function uninstall_package {
Dean Troyerdff49a22014-01-30 15:37:40 -06001097 if is_ubuntu; then
1098 apt_get purge "$@"
1099 elif is_fedora; then
Ian Wienand36298ee2015-02-04 10:29:31 +11001100 sudo ${YUM:-yum} remove -y "$@" ||:
Dean Troyerdff49a22014-01-30 15:37:40 -06001101 elif is_suse; then
1102 sudo zypper rm "$@"
1103 else
1104 exit_distro_not_supported "uninstalling packages"
1105 fi
1106}
1107
1108# Wrapper for ``yum`` to set proxy environment variables
Daniel P. Berrange63d25d92014-12-09 15:21:22 +00001109# Uses globals ``OFFLINE``, ``*_proxy``, ``YUM``
Dean Troyerdff49a22014-01-30 15:37:40 -06001110# yum_install package [package ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001111function yum_install {
Dean Troyerdff49a22014-01-30 15:37:40 -06001112 [[ "$OFFLINE" = "True" ]] && return
1113 local sudo="sudo"
1114 [[ "$(id -u)" = "0" ]] && sudo="env"
Ian Wienandb27f16d2014-02-28 14:29:02 +11001115
1116 # The manual check for missing packages is because yum -y assumes
1117 # missing packages are OK. See
1118 # https://bugzilla.redhat.com/show_bug.cgi?id=965567
Ian Wienandfdf00f22015-03-13 11:50:02 +11001119 $sudo http_proxy="${http_proxy:-}" https_proxy="${https_proxy:-}" \
1120 no_proxy="${no_proxy:-}" \
Ian Wienand36298ee2015-02-04 10:29:31 +11001121 ${YUM:-yum} install -y "$@" 2>&1 | \
Ian Wienandb27f16d2014-02-28 14:29:02 +11001122 awk '
1123 BEGIN { fail=0 }
1124 /No package/ { fail=1 }
1125 { print }
1126 END { exit fail }' || \
1127 die $LINENO "Missing packages detected"
1128
1129 # also ensure we catch a yum failure
1130 if [[ ${PIPESTATUS[0]} != 0 ]]; then
Ian Wienand36298ee2015-02-04 10:29:31 +11001131 die $LINENO "${YUM:-yum} install failure"
Ian Wienandb27f16d2014-02-28 14:29:02 +11001132 fi
Dean Troyerdff49a22014-01-30 15:37:40 -06001133}
1134
1135# zypper wrapper to set arguments correctly
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001136# Uses globals ``OFFLINE``, ``*_proxy``
Dean Troyerdff49a22014-01-30 15:37:40 -06001137# zypper_install package [package ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001138function zypper_install {
Dean Troyerdff49a22014-01-30 15:37:40 -06001139 [[ "$OFFLINE" = "True" ]] && return
1140 local sudo="sudo"
1141 [[ "$(id -u)" = "0" ]] && sudo="env"
Ian Wienandfdf00f22015-03-13 11:50:02 +11001142 $sudo http_proxy="${http_proxy:-}" https_proxy="${https_proxy:-}" \
1143 no_proxy="${no_proxy:-}" \
Dean Troyerdff49a22014-01-30 15:37:40 -06001144 zypper --non-interactive install --auto-agree-with-licenses "$@"
1145}
1146
1147
1148# Process Functions
1149# =================
1150
1151# _run_process() is designed to be backgrounded by run_process() to simulate a
1152# fork. It includes the dirty work of closing extra filehandles and preparing log
1153# files to produce the same logs as screen_it(). The log filename is derived
Dean Troyerdde41d02014-12-09 17:47:57 -06001154# from the service name.
1155# Uses globals ``CURRENT_LOG_TIME``, ``LOGDIR``, ``SCREEN_LOGDIR``, ``SCREEN_NAME``, ``SERVICE_DIR``
Chris Dent2f27a0e2014-09-09 13:46:02 +01001156# If an optional group is provided sg will be used to set the group of
1157# the command.
1158# _run_process service "command-line" [group]
Ian Wienandaee18c72014-02-21 15:35:08 +11001159function _run_process {
Sean Dague6e137ab2015-04-29 08:22:24 -04001160 # disable tracing through the exec redirects, it's just confusing in the logs.
1161 xtrace=$(set +o | grep xtrace)
1162 set +o xtrace
1163
Dean Troyerdff49a22014-01-30 15:37:40 -06001164 local service=$1
1165 local command="$2"
Chris Dent2f27a0e2014-09-09 13:46:02 +01001166 local group=$3
Dean Troyerdff49a22014-01-30 15:37:40 -06001167
1168 # Undo logging redirections and close the extra descriptors
1169 exec 1>&3
1170 exec 2>&3
1171 exec 3>&-
1172 exec 6>&-
1173
Dean Troyerdde41d02014-12-09 17:47:57 -06001174 local real_logfile="${LOGDIR}/${service}.log.${CURRENT_LOG_TIME}"
1175 if [[ -n ${LOGDIR} ]]; then
1176 exec 1>&"$real_logfile" 2>&1
1177 ln -sf "$real_logfile" ${LOGDIR}/${service}.log
1178 if [[ -n ${SCREEN_LOGDIR} ]]; then
1179 # Drop the backward-compat symlink
1180 ln -sf "$real_logfile" ${SCREEN_LOGDIR}/screen-${service}.log
1181 fi
Dean Troyerdff49a22014-01-30 15:37:40 -06001182
1183 # TODO(dtroyer): Hack to get stdout from the Python interpreter for the logs.
1184 export PYTHONUNBUFFERED=1
1185 fi
1186
Sean Dague6e137ab2015-04-29 08:22:24 -04001187 # reenable xtrace before we do *real* work
1188 $xtrace
1189
Dean Troyer3159a822014-08-27 14:13:58 -05001190 # Run under ``setsid`` to force the process to become a session and group leader.
1191 # The pid saved can be used with pkill -g to get the entire process group.
Chris Dent2f27a0e2014-09-09 13:46:02 +01001192 if [[ -n "$group" ]]; then
1193 setsid sg $group "$command" & echo $! >$SERVICE_DIR/$SCREEN_NAME/$service.pid
1194 else
1195 setsid $command & echo $! >$SERVICE_DIR/$SCREEN_NAME/$service.pid
1196 fi
Dean Troyer3159a822014-08-27 14:13:58 -05001197
1198 # Just silently exit this process
1199 exit 0
Dean Troyerdff49a22014-01-30 15:37:40 -06001200}
1201
1202# Helper to remove the ``*.failure`` files under ``$SERVICE_DIR/$SCREEN_NAME``.
1203# This is used for ``service_check`` when all the ``screen_it`` are called finished
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001204# Uses globals ``SCREEN_NAME``, ``SERVICE_DIR``
Dean Troyerdff49a22014-01-30 15:37:40 -06001205# init_service_check
Ian Wienandaee18c72014-02-21 15:35:08 +11001206function init_service_check {
Dean Troyerdff49a22014-01-30 15:37:40 -06001207 SCREEN_NAME=${SCREEN_NAME:-stack}
1208 SERVICE_DIR=${SERVICE_DIR:-${DEST}/status}
1209
1210 if [[ ! -d "$SERVICE_DIR/$SCREEN_NAME" ]]; then
1211 mkdir -p "$SERVICE_DIR/$SCREEN_NAME"
1212 fi
1213
1214 rm -f "$SERVICE_DIR/$SCREEN_NAME"/*.failure
1215}
1216
1217# Find out if a process exists by partial name.
1218# is_running name
Ian Wienandaee18c72014-02-21 15:35:08 +11001219function is_running {
Dean Troyerdff49a22014-01-30 15:37:40 -06001220 local name=$1
1221 ps auxw | grep -v grep | grep ${name} > /dev/null
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001222 local exitcode=$?
Dean Troyerdff49a22014-01-30 15:37:40 -06001223 # some times I really hate bash reverse binary logic
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001224 return $exitcode
Dean Troyerdff49a22014-01-30 15:37:40 -06001225}
1226
Dean Troyer3159a822014-08-27 14:13:58 -05001227# Run a single service under screen or directly
1228# If the command includes shell metachatacters (;<>*) it must be run using a shell
Chris Dent2f27a0e2014-09-09 13:46:02 +01001229# If an optional group is provided sg will be used to run the
1230# command as that group.
1231# run_process service "command-line" [group]
Ian Wienandaee18c72014-02-21 15:35:08 +11001232function run_process {
Dean Troyerdff49a22014-01-30 15:37:40 -06001233 local service=$1
1234 local command="$2"
Chris Dent2f27a0e2014-09-09 13:46:02 +01001235 local group=$3
Dean Troyerdff49a22014-01-30 15:37:40 -06001236
Dean Troyer3159a822014-08-27 14:13:58 -05001237 if is_service_enabled $service; then
1238 if [[ "$USE_SCREEN" = "True" ]]; then
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001239 screen_process "$service" "$command" "$group"
Dean Troyer3159a822014-08-27 14:13:58 -05001240 else
1241 # Spawn directly without screen
Chris Dent2f27a0e2014-09-09 13:46:02 +01001242 _run_process "$service" "$command" "$group" &
Dean Troyer3159a822014-08-27 14:13:58 -05001243 fi
1244 fi
Dean Troyerdff49a22014-01-30 15:37:40 -06001245}
1246
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001247# Helper to launch a process in a named screen
Dean Troyerdde41d02014-12-09 17:47:57 -06001248# Uses globals ``CURRENT_LOG_TIME``, ```LOGDIR``, ``SCREEN_LOGDIR``, `SCREEN_NAME``,
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001249# ``SERVICE_DIR``, ``USE_SCREEN``
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001250# screen_process name "command-line" [group]
Chris Dent2f27a0e2014-09-09 13:46:02 +01001251# Run a command in a shell in a screen window, if an optional group
1252# is provided, use sg to set the group of the command.
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001253function screen_process {
1254 local name=$1
Dean Troyer3159a822014-08-27 14:13:58 -05001255 local command="$2"
Chris Dent2f27a0e2014-09-09 13:46:02 +01001256 local group=$3
Dean Troyer3159a822014-08-27 14:13:58 -05001257
Sean Dagueea22a4f2014-06-27 15:21:41 -04001258 SCREEN_NAME=${SCREEN_NAME:-stack}
1259 SERVICE_DIR=${SERVICE_DIR:-${DEST}/status}
Sean Dague53753292014-12-04 19:38:15 -05001260 USE_SCREEN=$(trueorfalse True USE_SCREEN)
Dean Troyerdff49a22014-01-30 15:37:40 -06001261
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001262 screen -S $SCREEN_NAME -X screen -t $name
Dean Troyerdff49a22014-01-30 15:37:40 -06001263
Dean Troyerdde41d02014-12-09 17:47:57 -06001264 local real_logfile="${LOGDIR}/${name}.log.${CURRENT_LOG_TIME}"
1265 echo "LOGDIR: $LOGDIR"
1266 echo "SCREEN_LOGDIR: $SCREEN_LOGDIR"
1267 echo "log: $real_logfile"
1268 if [[ -n ${LOGDIR} ]]; then
1269 screen -S $SCREEN_NAME -p $name -X logfile "$real_logfile"
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001270 screen -S $SCREEN_NAME -p $name -X log on
Dean Troyerdde41d02014-12-09 17:47:57 -06001271 ln -sf "$real_logfile" ${LOGDIR}/${name}.log
1272 if [[ -n ${SCREEN_LOGDIR} ]]; then
1273 # Drop the backward-compat symlink
1274 ln -sf "$real_logfile" ${SCREEN_LOGDIR}/screen-${1}.log
1275 fi
Dean Troyerdff49a22014-01-30 15:37:40 -06001276 fi
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001277
1278 # sleep to allow bash to be ready to be send the command - we are
1279 # creating a new window in screen and then sends characters, so if
Sean Dague4d7ee092015-04-07 10:40:49 -04001280 # bash isn't running by the time we send the command, nothing
1281 # happens. This sleep was added originally to handle gate runs
1282 # where we needed this to be at least 3 seconds to pass
1283 # consistently on slow clouds. Now this is configurable so that we
1284 # can determine a reasonable value for the local case which should
1285 # be much smaller.
1286 sleep ${SCREEN_SLEEP:-3}
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001287
1288 NL=`echo -ne '\015'`
1289 # This fun command does the following:
1290 # - the passed server command is backgrounded
1291 # - the pid of the background process is saved in the usual place
1292 # - the server process is brought back to the foreground
1293 # - if the server process exits prematurely the fg command errors
1294 # and a message is written to stdout and the process failure file
1295 #
1296 # The pid saved can be used in stop_process() as a process group
1297 # id to kill off all child processes
1298 if [[ -n "$group" ]]; then
1299 command="sg $group '$command'"
1300 fi
Ian Wienandb28b2702015-04-16 08:43:43 +10001301
1302 # Append the process to the screen rc file
1303 screen_rc "$name" "$command"
1304
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001305 screen -S $SCREEN_NAME -p $name -X stuff "$command & echo \$! >$SERVICE_DIR/$SCREEN_NAME/${name}.pid; fg || echo \"$name failed to start\" | tee \"$SERVICE_DIR/$SCREEN_NAME/${name}.failure\"$NL"
Dean Troyerdff49a22014-01-30 15:37:40 -06001306}
1307
1308# Screen rc file builder
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001309# Uses globals ``SCREEN_NAME``, ``SCREENRC``
Dean Troyerdff49a22014-01-30 15:37:40 -06001310# screen_rc service "command-line"
1311function screen_rc {
1312 SCREEN_NAME=${SCREEN_NAME:-stack}
1313 SCREENRC=$TOP_DIR/$SCREEN_NAME-screenrc
1314 if [[ ! -e $SCREENRC ]]; then
1315 # Name the screen session
1316 echo "sessionname $SCREEN_NAME" > $SCREENRC
1317 # Set a reasonable statusbar
1318 echo "hardstatus alwayslastline '$SCREEN_HARDSTATUS'" >> $SCREENRC
1319 # Some distributions override PROMPT_COMMAND for the screen terminal type - turn that off
1320 echo "setenv PROMPT_COMMAND /bin/true" >> $SCREENRC
1321 echo "screen -t shell bash" >> $SCREENRC
1322 fi
1323 # If this service doesn't already exist in the screenrc file
1324 if ! grep $1 $SCREENRC 2>&1 > /dev/null; then
1325 NL=`echo -ne '\015'`
1326 echo "screen -t $1 bash" >> $SCREENRC
1327 echo "stuff \"$2$NL\"" >> $SCREENRC
1328
Dean Troyerdde41d02014-12-09 17:47:57 -06001329 if [[ -n ${LOGDIR} ]]; then
1330 echo "logfile ${LOGDIR}/${1}.log.${CURRENT_LOG_TIME}" >>$SCREENRC
Dean Troyerdff49a22014-01-30 15:37:40 -06001331 echo "log on" >>$SCREENRC
1332 fi
1333 fi
1334}
1335
1336# Stop a service in screen
1337# If a PID is available use it, kill the whole process group via TERM
1338# If screen is being used kill the screen window; this will catch processes
1339# that did not leave a PID behind
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001340# Uses globals ``SCREEN_NAME``, ``SERVICE_DIR``, ``USE_SCREEN``
Chris Dent2f27a0e2014-09-09 13:46:02 +01001341# screen_stop_service service
Dean Troyer3159a822014-08-27 14:13:58 -05001342function screen_stop_service {
1343 local service=$1
1344
Dean Troyerdff49a22014-01-30 15:37:40 -06001345 SCREEN_NAME=${SCREEN_NAME:-stack}
1346 SERVICE_DIR=${SERVICE_DIR:-${DEST}/status}
Sean Dague53753292014-12-04 19:38:15 -05001347 USE_SCREEN=$(trueorfalse True USE_SCREEN)
Dean Troyerdff49a22014-01-30 15:37:40 -06001348
Dean Troyer3159a822014-08-27 14:13:58 -05001349 if is_service_enabled $service; then
1350 # Clean up the screen window
Attila Fazekasf750a6f2015-07-01 12:17:35 +02001351 screen -S $SCREEN_NAME -p $service -X kill || true
Dean Troyer3159a822014-08-27 14:13:58 -05001352 fi
1353}
1354
1355# Stop a service process
1356# If a PID is available use it, kill the whole process group via TERM
1357# If screen is being used kill the screen window; this will catch processes
1358# that did not leave a PID behind
1359# Uses globals ``SERVICE_DIR``, ``USE_SCREEN``
1360# stop_process service
1361function stop_process {
1362 local service=$1
1363
1364 SERVICE_DIR=${SERVICE_DIR:-${DEST}/status}
Sean Dague53753292014-12-04 19:38:15 -05001365 USE_SCREEN=$(trueorfalse True USE_SCREEN)
Dean Troyer3159a822014-08-27 14:13:58 -05001366
1367 if is_service_enabled $service; then
Dean Troyerdff49a22014-01-30 15:37:40 -06001368 # Kill via pid if we have one available
Dean Troyer3159a822014-08-27 14:13:58 -05001369 if [[ -r $SERVICE_DIR/$SCREEN_NAME/$service.pid ]]; then
1370 pkill -g $(cat $SERVICE_DIR/$SCREEN_NAME/$service.pid)
1371 rm $SERVICE_DIR/$SCREEN_NAME/$service.pid
Dean Troyerdff49a22014-01-30 15:37:40 -06001372 fi
1373 if [[ "$USE_SCREEN" = "True" ]]; then
1374 # Clean up the screen window
Dean Troyer3159a822014-08-27 14:13:58 -05001375 screen_stop_service $service
Dean Troyerdff49a22014-01-30 15:37:40 -06001376 fi
1377 fi
1378}
1379
1380# Helper to get the status of each running service
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001381# Uses globals ``SCREEN_NAME``, ``SERVICE_DIR``
Dean Troyerdff49a22014-01-30 15:37:40 -06001382# service_check
Ian Wienandaee18c72014-02-21 15:35:08 +11001383function service_check {
Dean Troyerdff49a22014-01-30 15:37:40 -06001384 local service
1385 local failures
1386 SCREEN_NAME=${SCREEN_NAME:-stack}
1387 SERVICE_DIR=${SERVICE_DIR:-${DEST}/status}
1388
1389
1390 if [[ ! -d "$SERVICE_DIR/$SCREEN_NAME" ]]; then
1391 echo "No service status directory found"
1392 return
1393 fi
1394
1395 # Check if there is any falure flag file under $SERVICE_DIR/$SCREEN_NAME
Sean Dague09bd7c82014-02-03 08:35:26 +09001396 # make this -o errexit safe
1397 failures=`ls "$SERVICE_DIR/$SCREEN_NAME"/*.failure 2>/dev/null || /bin/true`
Dean Troyerdff49a22014-01-30 15:37:40 -06001398
1399 for service in $failures; do
1400 service=`basename $service`
1401 service=${service%.failure}
1402 echo "Error: Service $service is not running"
1403 done
1404
1405 if [ -n "$failures" ]; then
Sean Dague12379222014-02-27 17:16:46 -05001406 die $LINENO "More details about the above errors can be found with screen, with ./rejoin-stack.sh"
Dean Troyerdff49a22014-01-30 15:37:40 -06001407 fi
1408}
1409
Chris Dent2f27a0e2014-09-09 13:46:02 +01001410# Tail a log file in a screen if USE_SCREEN is true.
1411function tail_log {
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001412 local name=$1
Chris Dent2f27a0e2014-09-09 13:46:02 +01001413 local logfile=$2
1414
Sean Dague53753292014-12-04 19:38:15 -05001415 USE_SCREEN=$(trueorfalse True USE_SCREEN)
Chris Dent2f27a0e2014-09-09 13:46:02 +01001416 if [[ "$USE_SCREEN" = "True" ]]; then
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001417 screen_process "$name" "sudo tail -f $logfile"
Chris Dent2f27a0e2014-09-09 13:46:02 +01001418 fi
1419}
1420
Dean Troyerdff49a22014-01-30 15:37:40 -06001421
Dean Troyer3159a822014-08-27 14:13:58 -05001422# Deprecated Functions
1423# --------------------
1424
1425# _old_run_process() is designed to be backgrounded by old_run_process() to simulate a
1426# fork. It includes the dirty work of closing extra filehandles and preparing log
1427# files to produce the same logs as screen_it(). The log filename is derived
1428# from the service name and global-and-now-misnamed ``SCREEN_LOGDIR``
1429# Uses globals ``CURRENT_LOG_TIME``, ``SCREEN_LOGDIR``, ``SCREEN_NAME``, ``SERVICE_DIR``
1430# _old_run_process service "command-line"
1431function _old_run_process {
1432 local service=$1
1433 local command="$2"
1434
1435 # Undo logging redirections and close the extra descriptors
1436 exec 1>&3
1437 exec 2>&3
1438 exec 3>&-
1439 exec 6>&-
1440
1441 if [[ -n ${SCREEN_LOGDIR} ]]; then
Dean Troyerad5cc982014-12-10 16:35:32 -06001442 exec 1>&${SCREEN_LOGDIR}/screen-${1}.log.${CURRENT_LOG_TIME} 2>&1
1443 ln -sf ${SCREEN_LOGDIR}/screen-${1}.log.${CURRENT_LOG_TIME} ${SCREEN_LOGDIR}/screen-${1}.log
Dean Troyer3159a822014-08-27 14:13:58 -05001444
1445 # TODO(dtroyer): Hack to get stdout from the Python interpreter for the logs.
1446 export PYTHONUNBUFFERED=1
1447 fi
1448
1449 exec /bin/bash -c "$command"
1450 die "$service exec failure: $command"
1451}
1452
1453# old_run_process() launches a child process that closes all file descriptors and
1454# then exec's the passed in command. This is meant to duplicate the semantics
1455# of screen_it() without screen. PIDs are written to
1456# ``$SERVICE_DIR/$SCREEN_NAME/$service.pid`` by the spawned child process.
1457# old_run_process service "command-line"
1458function old_run_process {
1459 local service=$1
1460 local command="$2"
1461
1462 # Spawn the child process
1463 _old_run_process "$service" "$command" &
1464 echo $!
1465}
1466
1467# Compatibility for existing start_XXXX() functions
1468# Uses global ``USE_SCREEN``
1469# screen_it service "command-line"
1470function screen_it {
1471 if is_service_enabled $1; then
1472 # Append the service to the screen rc file
1473 screen_rc "$1" "$2"
1474
1475 if [[ "$USE_SCREEN" = "True" ]]; then
Adam Gandelman8543a0f2014-10-16 17:42:33 -07001476 screen_process "$1" "$2"
Dean Troyer3159a822014-08-27 14:13:58 -05001477 else
1478 # Spawn directly without screen
1479 old_run_process "$1" "$2" >$SERVICE_DIR/$SCREEN_NAME/$1.pid
1480 fi
1481 fi
1482}
1483
1484# Compatibility for existing stop_XXXX() functions
1485# Stop a service in screen
1486# If a PID is available use it, kill the whole process group via TERM
1487# If screen is being used kill the screen window; this will catch processes
1488# that did not leave a PID behind
1489# screen_stop service
1490function screen_stop {
1491 # Clean up the screen window
1492 stop_process $1
1493}
1494
1495
Sean Dague2c65e712014-12-18 09:44:56 -05001496# Plugin Functions
1497# =================
1498
1499DEVSTACK_PLUGINS=${DEVSTACK_PLUGINS:-""}
1500
1501# enable_plugin <name> <url> [branch]
1502#
1503# ``name`` is an arbitrary name - (aka: glusterfs, nova-docker, zaqar)
1504# ``url`` is a git url
1505# ``branch`` is a gitref. If it's not set, defaults to master
1506function enable_plugin {
1507 local name=$1
1508 local url=$2
1509 local branch=${3:-master}
1510 DEVSTACK_PLUGINS+=",$name"
1511 GITREPO[$name]=$url
1512 GITDIR[$name]=$DEST/$name
1513 GITBRANCH[$name]=$branch
1514}
1515
1516# fetch_plugins
1517#
1518# clones all plugins
1519function fetch_plugins {
1520 local plugins="${DEVSTACK_PLUGINS}"
1521 local plugin
1522
1523 # short circuit if nothing to do
1524 if [[ -z $plugins ]]; then
1525 return
1526 fi
1527
Dean Troyerdc97cb72015-03-28 08:20:50 -05001528 echo "Fetching DevStack plugins"
Sean Dague2c65e712014-12-18 09:44:56 -05001529 for plugin in ${plugins//,/ }; do
1530 git_clone_by_name $plugin
1531 done
1532}
1533
1534# load_plugin_settings
1535#
1536# Load settings from plugins in the order that they were registered
1537function load_plugin_settings {
1538 local plugins="${DEVSTACK_PLUGINS}"
1539 local plugin
1540
1541 # short circuit if nothing to do
1542 if [[ -z $plugins ]]; then
1543 return
1544 fi
1545
1546 echo "Loading plugin settings"
1547 for plugin in ${plugins//,/ }; do
1548 local dir=${GITDIR[$plugin]}
1549 # source any known settings
1550 if [[ -f $dir/devstack/settings ]]; then
1551 source $dir/devstack/settings
1552 fi
1553 done
1554}
1555
Sean Dague6e275e12015-03-26 05:54:28 -04001556# plugin_override_defaults
1557#
1558# Run an extremely early setting phase for plugins that allows default
1559# overriding of services.
1560function plugin_override_defaults {
1561 local plugins="${DEVSTACK_PLUGINS}"
1562 local plugin
1563
1564 # short circuit if nothing to do
1565 if [[ -z $plugins ]]; then
1566 return
1567 fi
1568
1569 echo "Overriding Configuration Defaults"
1570 for plugin in ${plugins//,/ }; do
1571 local dir=${GITDIR[$plugin]}
1572 # source any overrides
1573 if [[ -f $dir/devstack/override-defaults ]]; then
1574 # be really verbose that an override is happening, as it
1575 # may not be obvious if things fail later.
1576 echo "$plugin has overriden the following defaults"
1577 cat $dir/devstack/override-defaults
1578 source $dir/devstack/override-defaults
1579 fi
1580 done
1581}
1582
Sean Dague2c65e712014-12-18 09:44:56 -05001583# run_plugins
1584#
1585# Run the devstack/plugin.sh in all the plugin directories. These are
1586# run in registration order.
1587function run_plugins {
1588 local mode=$1
1589 local phase=$2
Bharat Kumar Kobagana441ff072015-01-08 12:26:26 +05301590
1591 local plugins="${DEVSTACK_PLUGINS}"
1592 local plugin
Sean Dague2c65e712014-12-18 09:44:56 -05001593 for plugin in ${plugins//,/ }; do
1594 local dir=${GITDIR[$plugin]}
1595 if [[ -f $dir/devstack/plugin.sh ]]; then
1596 source $dir/devstack/plugin.sh $mode $phase
1597 fi
1598 done
1599}
1600
1601function run_phase {
1602 local mode=$1
1603 local phase=$2
1604 if [[ -d $TOP_DIR/extras.d ]]; then
1605 for i in $TOP_DIR/extras.d/*.sh; do
1606 [[ -r $i ]] && source $i $mode $phase
1607 done
1608 fi
1609 # the source phase corresponds to settings loading in plugins
1610 if [[ "$mode" == "source" ]]; then
1611 load_plugin_settings
Sean Dague6e275e12015-03-26 05:54:28 -04001612 elif [[ "$mode" == "override_defaults" ]]; then
1613 plugin_override_defaults
Sean Dague2c65e712014-12-18 09:44:56 -05001614 else
1615 run_plugins $mode $phase
1616 fi
1617}
1618
Dean Troyerdff49a22014-01-30 15:37:40 -06001619
1620# Service Functions
1621# =================
1622
1623# remove extra commas from the input string (i.e. ``ENABLED_SERVICES``)
1624# _cleanup_service_list service-list
Ian Wienandaee18c72014-02-21 15:35:08 +11001625function _cleanup_service_list {
Dean Troyerdff49a22014-01-30 15:37:40 -06001626 echo "$1" | sed -e '
1627 s/,,/,/g;
1628 s/^,//;
1629 s/,$//
1630 '
1631}
1632
1633# disable_all_services() removes all current services
1634# from ``ENABLED_SERVICES`` to reset the configuration
1635# before a minimal installation
1636# Uses global ``ENABLED_SERVICES``
1637# disable_all_services
Ian Wienandaee18c72014-02-21 15:35:08 +11001638function disable_all_services {
Dean Troyerdff49a22014-01-30 15:37:40 -06001639 ENABLED_SERVICES=""
1640}
1641
1642# Remove all services starting with '-'. For example, to install all default
1643# services except rabbit (rabbit) set in ``localrc``:
1644# ENABLED_SERVICES+=",-rabbit"
1645# Uses global ``ENABLED_SERVICES``
1646# disable_negated_services
Ian Wienandaee18c72014-02-21 15:35:08 +11001647function disable_negated_services {
Ian Wienand2796a822015-04-15 08:59:04 +10001648 local to_remove=""
1649 local remaining=""
Dean Troyerdff49a22014-01-30 15:37:40 -06001650 local service
Ian Wienand2796a822015-04-15 08:59:04 +10001651
1652 # build up list of services that should be removed; i.e. they
1653 # begin with "-"
1654 for service in ${ENABLED_SERVICES//,/ }; do
Dean Troyerdff49a22014-01-30 15:37:40 -06001655 if [[ ${service} == -* ]]; then
Ian Wienand2796a822015-04-15 08:59:04 +10001656 to_remove+=",${service#-}"
1657 else
1658 remaining+=",${service}"
Dean Troyerdff49a22014-01-30 15:37:40 -06001659 fi
1660 done
Ian Wienand2796a822015-04-15 08:59:04 +10001661
1662 # go through the service list. if this service appears in the "to
1663 # be removed" list, drop it
fumihiko kakuma8606c982015-04-13 09:55:06 +09001664 ENABLED_SERVICES=$(remove_disabled_services "$remaining" "$to_remove")
Dean Troyerdff49a22014-01-30 15:37:40 -06001665}
1666
1667# disable_service() removes the services passed as argument to the
1668# ``ENABLED_SERVICES`` list, if they are present.
1669#
1670# For example:
1671# disable_service rabbit
1672#
1673# This function does not know about the special cases
1674# for nova, glance, and neutron built into is_service_enabled().
1675# Uses global ``ENABLED_SERVICES``
1676# disable_service service [service ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001677function disable_service {
Dean Troyerdff49a22014-01-30 15:37:40 -06001678 local tmpsvcs=",${ENABLED_SERVICES},"
1679 local service
1680 for service in $@; do
1681 if is_service_enabled $service; then
1682 tmpsvcs=${tmpsvcs//,$service,/,}
1683 fi
1684 done
1685 ENABLED_SERVICES=$(_cleanup_service_list "$tmpsvcs")
1686}
1687
1688# enable_service() adds the services passed as argument to the
1689# ``ENABLED_SERVICES`` list, if they are not already present.
1690#
1691# For example:
Sean Dague37eca482015-06-16 07:19:22 -04001692# enable_service q-svc
Dean Troyerdff49a22014-01-30 15:37:40 -06001693#
1694# This function does not know about the special cases
1695# for nova, glance, and neutron built into is_service_enabled().
1696# Uses global ``ENABLED_SERVICES``
1697# enable_service service [service ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001698function enable_service {
Dean Troyerdff49a22014-01-30 15:37:40 -06001699 local tmpsvcs="${ENABLED_SERVICES}"
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001700 local service
Dean Troyerdff49a22014-01-30 15:37:40 -06001701 for service in $@; do
1702 if ! is_service_enabled $service; then
1703 tmpsvcs+=",$service"
1704 fi
1705 done
1706 ENABLED_SERVICES=$(_cleanup_service_list "$tmpsvcs")
1707 disable_negated_services
1708}
1709
1710# is_service_enabled() checks if the service(s) specified as arguments are
1711# enabled by the user in ``ENABLED_SERVICES``.
1712#
1713# Multiple services specified as arguments are ``OR``'ed together; the test
1714# is a short-circuit boolean, i.e it returns on the first match.
1715#
1716# There are special cases for some 'catch-all' services::
1717# **nova** returns true if any service enabled start with **n-**
1718# **cinder** returns true if any service enabled start with **c-**
1719# **ceilometer** returns true if any service enabled start with **ceilometer**
1720# **glance** returns true if any service enabled start with **g-**
1721# **neutron** returns true if any service enabled start with **q-**
1722# **swift** returns true if any service enabled start with **s-**
1723# **trove** returns true if any service enabled start with **tr-**
1724# For backward compatibility if we have **swift** in ENABLED_SERVICES all the
1725# **s-** services will be enabled. This will be deprecated in the future.
1726#
1727# Cells within nova is enabled if **n-cell** is in ``ENABLED_SERVICES``.
1728# We also need to make sure to treat **n-cell-region** and **n-cell-child**
1729# as enabled in this case.
1730#
1731# Uses global ``ENABLED_SERVICES``
1732# is_service_enabled service [service ...]
Ian Wienandaee18c72014-02-21 15:35:08 +11001733function is_service_enabled {
Sean Dague45917cc2014-02-24 16:09:14 -05001734 local xtrace=$(set +o | grep xtrace)
1735 set +o xtrace
1736 local enabled=1
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001737 local services=$@
1738 local service
Dean Troyerdff49a22014-01-30 15:37:40 -06001739 for service in ${services}; do
Sean Dague45917cc2014-02-24 16:09:14 -05001740 [[ ,${ENABLED_SERVICES}, =~ ,${service}, ]] && enabled=0
Dean Troyerdff49a22014-01-30 15:37:40 -06001741
1742 # Look for top-level 'enabled' function for this service
1743 if type is_${service}_enabled >/dev/null 2>&1; then
1744 # A function exists for this service, use it
1745 is_${service}_enabled
Sean Dague45917cc2014-02-24 16:09:14 -05001746 enabled=$?
Dean Troyerdff49a22014-01-30 15:37:40 -06001747 fi
1748
1749 # TODO(dtroyer): Remove these legacy special-cases after the is_XXX_enabled()
1750 # are implemented
1751
Sean Dague45917cc2014-02-24 16:09:14 -05001752 [[ ${service} == n-cell-* && ${ENABLED_SERVICES} =~ "n-cell" ]] && enabled=0
Chris Dent2f27a0e2014-09-09 13:46:02 +01001753 [[ ${service} == n-cpu-* && ${ENABLED_SERVICES} =~ "n-cpu" ]] && enabled=0
Sean Dague45917cc2014-02-24 16:09:14 -05001754 [[ ${service} == "nova" && ${ENABLED_SERVICES} =~ "n-" ]] && enabled=0
Sean Dague45917cc2014-02-24 16:09:14 -05001755 [[ ${service} == "ceilometer" && ${ENABLED_SERVICES} =~ "ceilometer-" ]] && enabled=0
1756 [[ ${service} == "glance" && ${ENABLED_SERVICES} =~ "g-" ]] && enabled=0
1757 [[ ${service} == "ironic" && ${ENABLED_SERVICES} =~ "ir-" ]] && enabled=0
1758 [[ ${service} == "neutron" && ${ENABLED_SERVICES} =~ "q-" ]] && enabled=0
1759 [[ ${service} == "trove" && ${ENABLED_SERVICES} =~ "tr-" ]] && enabled=0
1760 [[ ${service} == "swift" && ${ENABLED_SERVICES} =~ "s-" ]] && enabled=0
1761 [[ ${service} == s-* && ${ENABLED_SERVICES} =~ "swift" ]] && enabled=0
Dean Troyerdff49a22014-01-30 15:37:40 -06001762 done
Sean Dague45917cc2014-02-24 16:09:14 -05001763 $xtrace
1764 return $enabled
Dean Troyerdff49a22014-01-30 15:37:40 -06001765}
1766
fumihiko kakuma8606c982015-04-13 09:55:06 +09001767# remove specified list from the input string
1768# remove_disabled_services service-list remove-list
1769function remove_disabled_services {
1770 local service_list=$1
1771 local remove_list=$2
1772 local service
1773 local enabled=""
1774
1775 for service in ${service_list//,/ }; do
1776 local remove
1777 local add=1
1778 for remove in ${remove_list//,/ }; do
1779 if [[ ${remove} == ${service} ]]; then
1780 add=0
1781 break
1782 fi
1783 done
1784 if [[ $add == 1 ]]; then
1785 enabled="${enabled},$service"
1786 fi
1787 done
1788 _cleanup_service_list "$enabled"
1789}
1790
Dean Troyerdff49a22014-01-30 15:37:40 -06001791# Toggle enable/disable_service for services that must run exclusive of each other
1792# $1 The name of a variable containing a space-separated list of services
1793# $2 The name of a variable in which to store the enabled service's name
1794# $3 The name of the service to enable
1795function use_exclusive_service {
1796 local options=${!1}
1797 local selection=$3
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001798 local out=$2
Dean Troyerdff49a22014-01-30 15:37:40 -06001799 [ -z $selection ] || [[ ! "$options" =~ "$selection" ]] && return 1
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001800 local opt
Dean Troyerdff49a22014-01-30 15:37:40 -06001801 for opt in $options;do
1802 [[ "$opt" = "$selection" ]] && enable_service $opt || disable_service $opt
1803 done
1804 eval "$out=$selection"
1805 return 0
1806}
1807
1808
Masayuki Igawaf6368d32014-02-20 13:31:26 +09001809# System Functions
1810# ================
Dean Troyerdff49a22014-01-30 15:37:40 -06001811
1812# Only run the command if the target file (the last arg) is not on an
1813# NFS filesystem.
Ian Wienandaee18c72014-02-21 15:35:08 +11001814function _safe_permission_operation {
Sean Dague45917cc2014-02-24 16:09:14 -05001815 local xtrace=$(set +o | grep xtrace)
1816 set +o xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -06001817 local args=( $@ )
1818 local last
1819 local sudo_cmd
1820 local dir_to_check
1821
1822 let last="${#args[*]} - 1"
1823
Dean Troyerd5dfa4c2014-07-25 11:13:11 -05001824 local dir_to_check=${args[$last]}
Dean Troyerdff49a22014-01-30 15:37:40 -06001825 if [ ! -d "$dir_to_check" ]; then
1826 dir_to_check=`dirname "$dir_to_check"`
1827 fi
1828
1829 if is_nfs_directory "$dir_to_check" ; then
Sean Dague45917cc2014-02-24 16:09:14 -05001830 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -06001831 return 0
1832 fi
1833
1834 if [[ $TRACK_DEPENDS = True ]]; then
1835 sudo_cmd="env"
1836 else
1837 sudo_cmd="sudo"
1838 fi
1839
Sean Dague45917cc2014-02-24 16:09:14 -05001840 $xtrace
Dean Troyerdff49a22014-01-30 15:37:40 -06001841 $sudo_cmd $@
1842}
1843
1844# Exit 0 if address is in network or 1 if address is not in network
1845# ip-range is in CIDR notation: 1.2.3.4/20
1846# address_in_net ip-address ip-range
Ian Wienandaee18c72014-02-21 15:35:08 +11001847function address_in_net {
Dean Troyerdff49a22014-01-30 15:37:40 -06001848 local ip=$1
1849 local range=$2
1850 local masklen=${range#*/}
1851 local network=$(maskip ${range%/*} $(cidr2netmask $masklen))
1852 local subnet=$(maskip $ip $(cidr2netmask $masklen))
1853 [[ $network == $subnet ]]
1854}
1855
1856# Add a user to a group.
1857# add_user_to_group user group
Ian Wienandaee18c72014-02-21 15:35:08 +11001858function add_user_to_group {
Dean Troyerdff49a22014-01-30 15:37:40 -06001859 local user=$1
1860 local group=$2
1861
Thomas Bechtolda8580852015-05-31 00:04:33 +02001862 sudo usermod -a -G "$group" "$user"
Dean Troyerdff49a22014-01-30 15:37:40 -06001863}
1864
1865# Convert CIDR notation to a IPv4 netmask
1866# cidr2netmask cidr-bits
Ian Wienandaee18c72014-02-21 15:35:08 +11001867function cidr2netmask {
Dean Troyerdff49a22014-01-30 15:37:40 -06001868 local maskpat="255 255 255 255"
1869 local maskdgt="254 252 248 240 224 192 128"
1870 set -- ${maskpat:0:$(( ($1 / 8) * 4 ))}${maskdgt:$(( (7 - ($1 % 8)) * 4 )):3}
1871 echo ${1-0}.${2-0}.${3-0}.${4-0}
1872}
1873
1874# Gracefully cp only if source file/dir exists
1875# cp_it source destination
1876function cp_it {
1877 if [ -e $1 ] || [ -d $1 ]; then
1878 cp -pRL $1 $2
1879 fi
1880}
1881
1882# HTTP and HTTPS proxy servers are supported via the usual environment variables [1]
1883# ``http_proxy``, ``https_proxy`` and ``no_proxy``. They can be set in
1884# ``localrc`` or on the command line if necessary::
1885#
1886# [1] http://www.w3.org/Daemon/User/Proxies/ProxyClients.html
1887#
1888# http_proxy=http://proxy.example.com:3128/ no_proxy=repo.example.net ./stack.sh
1889
Ian Wienandaee18c72014-02-21 15:35:08 +11001890function export_proxy_variables {
Sean Dague53753292014-12-04 19:38:15 -05001891 if isset http_proxy ; then
Dean Troyerdff49a22014-01-30 15:37:40 -06001892 export http_proxy=$http_proxy
1893 fi
Sean Dague53753292014-12-04 19:38:15 -05001894 if isset https_proxy ; then
Dean Troyerdff49a22014-01-30 15:37:40 -06001895 export https_proxy=$https_proxy
1896 fi
Sean Dague53753292014-12-04 19:38:15 -05001897 if isset no_proxy ; then
Dean Troyerdff49a22014-01-30 15:37:40 -06001898 export no_proxy=$no_proxy
1899 fi
1900}
1901
1902# Returns true if the directory is on a filesystem mounted via NFS.
Ian Wienandaee18c72014-02-21 15:35:08 +11001903function is_nfs_directory {
Dean Troyerdff49a22014-01-30 15:37:40 -06001904 local mount_type=`stat -f -L -c %T $1`
1905 test "$mount_type" == "nfs"
1906}
1907
1908# Return the network portion of the given IP address using netmask
1909# netmask is in the traditional dotted-quad format
1910# maskip ip-address netmask
Ian Wienandaee18c72014-02-21 15:35:08 +11001911function maskip {
Dean Troyerdff49a22014-01-30 15:37:40 -06001912 local ip=$1
1913 local mask=$2
1914 local l="${ip%.*}"; local r="${ip#*.}"; local n="${mask%.*}"; local m="${mask#*.}"
1915 local subnet=$((${ip%%.*}&${mask%%.*})).$((${r%%.*}&${m%%.*})).$((${l##*.}&${n##*.})).$((${ip##*.}&${mask##*.}))
1916 echo $subnet
1917}
1918
Chris Dent3a2c86a2015-05-12 13:41:25 +00001919# Return the current python as "python<major>.<minor>"
1920function python_version {
1921 local python_version=$(python -c 'import sys; print("%s.%s" % sys.version_info[0:2])')
1922 echo "python${python_version}"
1923}
1924
Dean Troyerdff49a22014-01-30 15:37:40 -06001925# Service wrapper to restart services
1926# restart_service service-name
Ian Wienandaee18c72014-02-21 15:35:08 +11001927function restart_service {
Dean Troyerdff49a22014-01-30 15:37:40 -06001928 if is_ubuntu; then
1929 sudo /usr/sbin/service $1 restart
1930 else
1931 sudo /sbin/service $1 restart
1932 fi
1933}
1934
1935# Only change permissions of a file or directory if it is not on an
1936# NFS filesystem.
Ian Wienandaee18c72014-02-21 15:35:08 +11001937function safe_chmod {
Dean Troyerdff49a22014-01-30 15:37:40 -06001938 _safe_permission_operation chmod $@
1939}
1940
1941# Only change ownership of a file or directory if it is not on an NFS
1942# filesystem.
Ian Wienandaee18c72014-02-21 15:35:08 +11001943function safe_chown {
Dean Troyerdff49a22014-01-30 15:37:40 -06001944 _safe_permission_operation chown $@
1945}
1946
1947# Service wrapper to start services
1948# start_service service-name
Ian Wienandaee18c72014-02-21 15:35:08 +11001949function start_service {
Dean Troyerdff49a22014-01-30 15:37:40 -06001950 if is_ubuntu; then
1951 sudo /usr/sbin/service $1 start
1952 else
1953 sudo /sbin/service $1 start
1954 fi
1955}
1956
1957# Service wrapper to stop services
1958# stop_service service-name
Ian Wienandaee18c72014-02-21 15:35:08 +11001959function stop_service {
Dean Troyerdff49a22014-01-30 15:37:40 -06001960 if is_ubuntu; then
1961 sudo /usr/sbin/service $1 stop
1962 else
1963 sudo /sbin/service $1 stop
1964 fi
1965}
1966
Sean Dague442e4e92015-06-24 13:24:02 -04001967# Test with a finite retry loop.
1968#
1969function test_with_retry {
1970 local testcmd=$1
1971 local failmsg=$2
1972 local until=${3:-10}
1973 local sleep=${4:-0.5}
1974
1975 if ! timeout $until sh -c "while ! $testcmd; do sleep $sleep; done"; then
1976 die $LINENO "$failmsg"
1977 fi
1978}
1979
Dean Troyerdff49a22014-01-30 15:37:40 -06001980
1981# Restore xtrace
1982$XTRACE
1983
1984# Local variables:
1985# mode: shell-script
1986# End: