| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 1 | # functions - Common functions used by DevStack components | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 2 | # | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 3 | # The following variables are assumed to be defined by certain functions: | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 4 | # ``ENABLED_SERVICES`` | 
 | 5 | # ``EROR_ON_CLONE`` | 
 | 6 | # ``FILES`` | 
 | 7 | # ``GLANCE_HOSTPORT`` | 
 | 8 | # ``OFFLINE`` | 
 | 9 | # ``PIP_DOWNLOAD_CACHE`` | 
| Maru Newby | 3a87edd | 2012-10-25 23:01:06 +0000 | [diff] [blame] | 10 | # ``PIP_USE_MIRRORS`` | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 11 | # ``RECLONE`` | 
 | 12 | # ``TRACK_DEPENDS`` | 
 | 13 | # ``http_proxy``, ``https_proxy``, ``no_proxy`` | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 14 |  | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 15 |  | 
| Dean Troyer | 27e3269 | 2012-03-16 16:16:56 -0500 | [diff] [blame] | 16 | # Save trace setting | 
 | 17 | XTRACE=$(set +o | grep xtrace) | 
 | 18 | set +o xtrace | 
 | 19 |  | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 20 |  | 
| Dean Troyer | d4f69b2 | 2013-07-24 12:24:43 -0500 | [diff] [blame] | 21 | # Convert CIDR notation to a IPv4 netmask | 
 | 22 | # cidr2netmask cidr-bits | 
 | 23 | function cidr2netmask() { | 
 | 24 |     local maskpat="255 255 255 255" | 
 | 25 |     local maskdgt="254 252 248 240 224 192 128" | 
 | 26 |     set -- ${maskpat:0:$(( ($1 / 8) * 4 ))}${maskdgt:$(( (7 - ($1 % 8)) * 4 )):3} | 
 | 27 |     echo ${1-0}.${2-0}.${3-0}.${4-0} | 
 | 28 | } | 
 | 29 |  | 
 | 30 |  | 
 | 31 | # Return the network portion of the given IP address using netmask | 
 | 32 | # netmask is in the traditional dotted-quad format | 
 | 33 | # maskip ip-address netmask | 
 | 34 | function maskip() { | 
 | 35 |     local ip=$1 | 
 | 36 |     local mask=$2 | 
 | 37 |     local l="${ip%.*}"; local r="${ip#*.}"; local n="${mask%.*}"; local m="${mask#*.}" | 
 | 38 |     local subnet=$((${ip%%.*}&${mask%%.*})).$((${r%%.*}&${m%%.*})).$((${l##*.}&${n##*.})).$((${ip##*.}&${mask##*.})) | 
 | 39 |     echo $subnet | 
 | 40 | } | 
 | 41 |  | 
 | 42 |  | 
 | 43 | # Exit 0 if address is in network or 1 if address is not in network | 
 | 44 | # ip-range is in CIDR notation: 1.2.3.4/20 | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 45 | # address_in_net ip-address ip-range | 
| Vishvananda Ishaya | c9ad14b | 2012-07-03 20:29:01 +0000 | [diff] [blame] | 46 | function address_in_net() { | 
| Dean Troyer | d4f69b2 | 2013-07-24 12:24:43 -0500 | [diff] [blame] | 47 |     local ip=$1 | 
 | 48 |     local range=$2 | 
 | 49 |     local masklen=${range#*/} | 
 | 50 |     local network=$(maskip ${range%/*} $(cidr2netmask $masklen)) | 
 | 51 |     local subnet=$(maskip $ip $(cidr2netmask $masklen)) | 
 | 52 |     [[ $network == $subnet ]] | 
| Vishvananda Ishaya | c9ad14b | 2012-07-03 20:29:01 +0000 | [diff] [blame] | 53 | } | 
 | 54 |  | 
 | 55 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 56 | # Wrapper for ``apt-get`` to set cache and proxy environment variables | 
 | 57 | # Uses globals ``OFFLINE``, ``*_proxy` | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 58 | # apt_get operation package [package ...] | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 59 | function apt_get() { | 
| Dean Troyer | d0b21e2 | 2012-03-07 14:52:25 -0600 | [diff] [blame] | 60 |     [[ "$OFFLINE" = "True" || -z "$@" ]] && return | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 61 |     local sudo="sudo" | 
 | 62 |     [[ "$(id -u)" = "0" ]] && sudo="env" | 
 | 63 |     $sudo DEBIAN_FRONTEND=noninteractive \ | 
 | 64 |         http_proxy=$http_proxy https_proxy=$https_proxy \ | 
| Osamu Habuka | 7abe4f2 | 2012-07-25 12:39:32 +0900 | [diff] [blame] | 65 |         no_proxy=$no_proxy \ | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 66 |         apt-get --option "Dpkg::Options::=--force-confold" --assume-yes "$@" | 
 | 67 | } | 
 | 68 |  | 
 | 69 |  | 
 | 70 | # Gracefully cp only if source file/dir exists | 
 | 71 | # cp_it source destination | 
 | 72 | function cp_it { | 
 | 73 |     if [ -e $1 ] || [ -d $1 ]; then | 
 | 74 |         cp -pRL $1 $2 | 
 | 75 |     fi | 
 | 76 | } | 
 | 77 |  | 
 | 78 |  | 
| Kui Shi | 5e28a3e | 2013-08-02 17:26:28 +0800 | [diff] [blame] | 79 | # Prints backtrace info | 
 | 80 | # filename:lineno:function | 
 | 81 | function backtrace { | 
 | 82 |     local level=$1 | 
 | 83 |     local deep=$((${#BASH_SOURCE[@]} - 1)) | 
 | 84 |     echo "[Call Trace]" | 
 | 85 |     while [ $level -le $deep ]; do | 
 | 86 |         echo "${BASH_SOURCE[$deep]}:${BASH_LINENO[$deep-1]}:${FUNCNAME[$deep-1]}" | 
 | 87 |         deep=$((deep - 1)) | 
 | 88 |     done | 
 | 89 | } | 
 | 90 |  | 
 | 91 |  | 
| Dean Troyer | ac93efb | 2013-03-13 14:30:54 -0500 | [diff] [blame] | 92 | # Prints line number and "message" then exits | 
 | 93 | # die $LINENO "message" | 
| Dean Troyer | 27e3269 | 2012-03-16 16:16:56 -0500 | [diff] [blame] | 94 | function die() { | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 95 |     local exitcode=$? | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 96 |     set +o xtrace | 
 | 97 |     local line=$1; shift | 
| Nachi Ueno | 07115eb | 2013-02-26 12:38:18 -0800 | [diff] [blame] | 98 |     if [ $exitcode == 0 ]; then | 
 | 99 |         exitcode=1 | 
 | 100 |     fi | 
| Kui Shi | 5e28a3e | 2013-08-02 17:26:28 +0800 | [diff] [blame] | 101 |     backtrace 2 | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 102 |     err $line "$*" | 
| Dean Troyer | 27e3269 | 2012-03-16 16:16:56 -0500 | [diff] [blame] | 103 |     exit $exitcode | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 104 | } | 
 | 105 |  | 
 | 106 |  | 
 | 107 | # Checks an environment variable is not set or has length 0 OR if the | 
 | 108 | # exit code is non-zero and prints "message" and exits | 
 | 109 | # NOTE: env-var is the variable name without a '$' | 
| Dean Troyer | ac93efb | 2013-03-13 14:30:54 -0500 | [diff] [blame] | 110 | # die_if_not_set $LINENO env-var "message" | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 111 | function die_if_not_set() { | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 112 |     local exitcode=$? | 
 | 113 |     FXTRACE=$(set +o | grep xtrace) | 
 | 114 |     set +o xtrace | 
 | 115 |     local line=$1; shift | 
 | 116 |     local evar=$1; shift | 
 | 117 |     if ! is_set $evar || [ $exitcode != 0 ]; then | 
 | 118 |         die $line "$*" | 
 | 119 |     fi | 
 | 120 |     $FXTRACE | 
 | 121 | } | 
 | 122 |  | 
 | 123 |  | 
 | 124 | # Prints line number and "message" in error format | 
 | 125 | # err $LINENO "message" | 
 | 126 | function err() { | 
 | 127 |     local exitcode=$? | 
 | 128 |     errXTRACE=$(set +o | grep xtrace) | 
 | 129 |     set +o xtrace | 
| Kui Shi | 17df077 | 2013-08-02 17:55:41 +0800 | [diff] [blame] | 130 |     local msg="[ERROR] ${BASH_SOURCE[2]}:$1 $2" | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 131 |     echo $msg 1>&2; | 
 | 132 |     if [[ -n ${SCREEN_LOGDIR} ]]; then | 
 | 133 |         echo $msg >> "${SCREEN_LOGDIR}/error.log" | 
 | 134 |     fi | 
 | 135 |     $errXTRACE | 
 | 136 |     return $exitcode | 
 | 137 | } | 
 | 138 |  | 
 | 139 |  | 
 | 140 | # Checks an environment variable is not set or has length 0 OR if the | 
 | 141 | # exit code is non-zero and prints "message" | 
 | 142 | # NOTE: env-var is the variable name without a '$' | 
 | 143 | # err_if_not_set $LINENO env-var "message" | 
 | 144 | function err_if_not_set() { | 
 | 145 |     local exitcode=$? | 
 | 146 |     errinsXTRACE=$(set +o | grep xtrace) | 
 | 147 |     set +o xtrace | 
 | 148 |     local line=$1; shift | 
 | 149 |     local evar=$1; shift | 
 | 150 |     if ! is_set $evar || [ $exitcode != 0 ]; then | 
 | 151 |         err $line "$*" | 
 | 152 |     fi | 
 | 153 |     $errinsXTRACE | 
 | 154 |     return $exitcode | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 155 | } | 
 | 156 |  | 
 | 157 |  | 
| Dean Troyer | 48352ee | 2012-12-12 12:50:38 -0600 | [diff] [blame] | 158 | # HTTP and HTTPS proxy servers are supported via the usual environment variables [1] | 
 | 159 | # ``http_proxy``, ``https_proxy`` and ``no_proxy``. They can be set in | 
 | 160 | # ``localrc`` or on the command line if necessary:: | 
 | 161 | # | 
 | 162 | # [1] http://www.w3.org/Daemon/User/Proxies/ProxyClients.html | 
 | 163 | # | 
 | 164 | #     http_proxy=http://proxy.example.com:3128/ no_proxy=repo.example.net ./stack.sh | 
 | 165 |  | 
 | 166 | function export_proxy_variables() { | 
 | 167 |     if [[ -n "$http_proxy" ]]; then | 
 | 168 |         export http_proxy=$http_proxy | 
 | 169 |     fi | 
 | 170 |     if [[ -n "$https_proxy" ]]; then | 
 | 171 |         export https_proxy=$https_proxy | 
 | 172 |     fi | 
 | 173 |     if [[ -n "$no_proxy" ]]; then | 
 | 174 |         export no_proxy=$no_proxy | 
 | 175 |     fi | 
 | 176 | } | 
 | 177 |  | 
 | 178 |  | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 179 | # Grab a numbered field from python prettytable output | 
 | 180 | # Fields are numbered starting with 1 | 
 | 181 | # Reverse syntax is supported: -1 is the last field, -2 is second to last, etc. | 
 | 182 | # get_field field-number | 
 | 183 | function get_field() { | 
 | 184 |     while read data; do | 
 | 185 |         if [ "$1" -lt 0 ]; then | 
 | 186 |             field="(\$(NF$1))" | 
 | 187 |         else | 
 | 188 |             field="\$$(($1 + 1))" | 
 | 189 |         fi | 
 | 190 |         echo "$data" | awk -F'[ \t]*\\|[ \t]*' "{print $field}" | 
 | 191 |     done | 
 | 192 | } | 
 | 193 |  | 
 | 194 |  | 
| Dean Troyer | c892bde | 2013-03-13 14:06:13 -0500 | [diff] [blame] | 195 | # Get the default value for HOST_IP | 
 | 196 | # get_default_host_ip fixed_range floating_range host_ip_iface host_ip | 
 | 197 | function get_default_host_ip() { | 
 | 198 |     local fixed_range=$1 | 
 | 199 |     local floating_range=$2 | 
 | 200 |     local host_ip_iface=$3 | 
 | 201 |     local host_ip=$4 | 
 | 202 |  | 
 | 203 |     # Find the interface used for the default route | 
 | 204 |     host_ip_iface=${host_ip_iface:-$(ip route | sed -n '/^default/{ s/.*dev \(\w\+\)\s\+.*/\1/; p; }' | head -1)} | 
 | 205 |     # Search for an IP unless an explicit is set by ``HOST_IP`` environment variable | 
 | 206 |     if [ -z "$host_ip" -o "$host_ip" == "dhcp" ]; then | 
 | 207 |         host_ip="" | 
 | 208 |         host_ips=`LC_ALL=C ip -f inet addr show ${host_ip_iface} | awk '/inet/ {split($2,parts,"/");  print parts[1]}'` | 
 | 209 |         for IP in $host_ips; do | 
 | 210 |             # Attempt to filter out IP addresses that are part of the fixed and | 
 | 211 |             # floating range. Note that this method only works if the ``netaddr`` | 
 | 212 |             # python library is installed. If it is not installed, an error | 
 | 213 |             # will be printed and the first IP from the interface will be used. | 
 | 214 |             # If that is not correct set ``HOST_IP`` in ``localrc`` to the correct | 
 | 215 |             # address. | 
 | 216 |             if ! (address_in_net $IP $fixed_range || address_in_net $IP $floating_range); then | 
 | 217 |                 host_ip=$IP | 
 | 218 |                 break; | 
 | 219 |             fi | 
 | 220 |         done | 
 | 221 |     fi | 
 | 222 |     echo $host_ip | 
 | 223 | } | 
 | 224 |  | 
 | 225 |  | 
| Isaku Yamahata | 8c43809 | 2013-02-12 22:30:56 +0900 | [diff] [blame] | 226 | function _get_package_dir() { | 
 | 227 |     local pkg_dir | 
 | 228 |     if is_ubuntu; then | 
 | 229 |         pkg_dir=$FILES/apts | 
 | 230 |     elif is_fedora; then | 
 | 231 |         pkg_dir=$FILES/rpms | 
 | 232 |     elif is_suse; then | 
 | 233 |         pkg_dir=$FILES/rpms-suse | 
 | 234 |     else | 
 | 235 |         exit_distro_not_supported "list of packages" | 
 | 236 |     fi | 
 | 237 |     echo "$pkg_dir" | 
 | 238 | } | 
 | 239 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 240 |  | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 241 | # get_packages() collects a list of package names of any type from the | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 242 | # prerequisite files in ``files/{apts|rpms}``.  The list is intended | 
 | 243 | # to be passed to a package installer such as apt or yum. | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 244 | # | 
| Isaku Yamahata | 8c43809 | 2013-02-12 22:30:56 +0900 | [diff] [blame] | 245 | # Only packages required for the services in 1st argument will be | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 246 | # included.  Two bits of metadata are recognized in the prerequisite files: | 
 | 247 | # - ``# NOPRIME`` defers installation to be performed later in stack.sh | 
 | 248 | # - ``# dist:DISTRO`` or ``dist:DISTRO1,DISTRO2`` limits the selection | 
 | 249 | #   of the package to the distros listed.  The distro names are case insensitive. | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 250 | function get_packages() { | 
| Isaku Yamahata | 8c43809 | 2013-02-12 22:30:56 +0900 | [diff] [blame] | 251 |     local services=$1 | 
 | 252 |     local package_dir=$(_get_package_dir) | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 253 |     local file_to_parse | 
 | 254 |     local service | 
 | 255 |  | 
 | 256 |     if [[ -z "$package_dir" ]]; then | 
 | 257 |         echo "No package directory supplied" | 
 | 258 |         return 1 | 
 | 259 |     fi | 
 | 260 |     if [[ -z "$DISTRO" ]]; then | 
| Vincent Untz | 855c587 | 2012-10-04 13:36:46 +0200 | [diff] [blame] | 261 |         GetDistro | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 262 |     fi | 
| Isaku Yamahata | 8c43809 | 2013-02-12 22:30:56 +0900 | [diff] [blame] | 263 |     for service in general ${services//,/ }; do | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 264 |         # Allow individual services to specify dependencies | 
 | 265 |         if [[ -e ${package_dir}/${service} ]]; then | 
 | 266 |             file_to_parse="${file_to_parse} $service" | 
 | 267 |         fi | 
 | 268 |         # NOTE(sdague) n-api needs glance for now because that's where | 
 | 269 |         # glance client is | 
 | 270 |         if [[ $service == n-api ]]; then | 
 | 271 |             if [[ ! $file_to_parse =~ nova ]]; then | 
 | 272 |                 file_to_parse="${file_to_parse} nova" | 
 | 273 |             fi | 
 | 274 |             if [[ ! $file_to_parse =~ glance ]]; then | 
 | 275 |                 file_to_parse="${file_to_parse} glance" | 
 | 276 |             fi | 
 | 277 |         elif [[ $service == c-* ]]; then | 
 | 278 |             if [[ ! $file_to_parse =~ cinder ]]; then | 
 | 279 |                 file_to_parse="${file_to_parse} cinder" | 
 | 280 |             fi | 
| John H. Tran | 9336164 | 2012-07-26 11:22:05 -0700 | [diff] [blame] | 281 |         elif [[ $service == ceilometer-* ]]; then | 
 | 282 |             if [[ ! $file_to_parse =~ ceilometer ]]; then | 
 | 283 |                 file_to_parse="${file_to_parse} ceilometer" | 
 | 284 |             fi | 
| Chmouel Boudjnah | 0c3a558 | 2013-03-06 10:58:33 +0100 | [diff] [blame] | 285 |         elif [[ $service == s-* ]]; then | 
 | 286 |             if [[ ! $file_to_parse =~ swift ]]; then | 
 | 287 |                 file_to_parse="${file_to_parse} swift" | 
 | 288 |             fi | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 289 |         elif [[ $service == n-* ]]; then | 
 | 290 |             if [[ ! $file_to_parse =~ nova ]]; then | 
 | 291 |                 file_to_parse="${file_to_parse} nova" | 
 | 292 |             fi | 
 | 293 |         elif [[ $service == g-* ]]; then | 
 | 294 |             if [[ ! $file_to_parse =~ glance ]]; then | 
 | 295 |                 file_to_parse="${file_to_parse} glance" | 
 | 296 |             fi | 
 | 297 |         elif [[ $service == key* ]]; then | 
 | 298 |             if [[ ! $file_to_parse =~ keystone ]]; then | 
 | 299 |                 file_to_parse="${file_to_parse} keystone" | 
 | 300 |             fi | 
| Robert Collins | 0a9954f | 2012-11-20 11:34:25 +1300 | [diff] [blame] | 301 |         elif [[ $service == q-* ]]; then | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 302 |             if [[ ! $file_to_parse =~ neutron ]]; then | 
 | 303 |                 file_to_parse="${file_to_parse} neutron" | 
| Robert Collins | 0a9954f | 2012-11-20 11:34:25 +1300 | [diff] [blame] | 304 |             fi | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 305 |         fi | 
 | 306 |     done | 
 | 307 |  | 
 | 308 |     for file in ${file_to_parse}; do | 
 | 309 |         local fname=${package_dir}/${file} | 
 | 310 |         local OIFS line package distros distro | 
 | 311 |         [[ -e $fname ]] || continue | 
 | 312 |  | 
 | 313 |         OIFS=$IFS | 
 | 314 |         IFS=$'\n' | 
 | 315 |         for line in $(<${fname}); do | 
 | 316 |             if [[ $line =~ "NOPRIME" ]]; then | 
 | 317 |                 continue | 
 | 318 |             fi | 
 | 319 |  | 
| Christian Berendt | 71d5630 | 2013-07-22 11:37:42 +0200 | [diff] [blame] | 320 |             # Assume we want this package | 
 | 321 |             package=${line%#*} | 
 | 322 |             inst_pkg=1 | 
 | 323 |  | 
 | 324 |             # Look for # dist:xxx in comment | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 325 |             if [[ $line =~ (.*)#.*dist:([^ ]*) ]]; then | 
 | 326 |                 # We are using BASH regexp matching feature. | 
 | 327 |                 package=${BASH_REMATCH[1]} | 
 | 328 |                 distros=${BASH_REMATCH[2]} | 
 | 329 |                 # In bash ${VAR,,} will lowecase VAR | 
| Christian Berendt | 71d5630 | 2013-07-22 11:37:42 +0200 | [diff] [blame] | 330 |                 # Look for a match in the distro list | 
 | 331 |                 if [[ ! ${distros,,} =~ ${DISTRO,,} ]]; then | 
 | 332 |                     # If no match then skip this package | 
 | 333 |                     inst_pkg=0 | 
 | 334 |                 fi | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 335 |             fi | 
 | 336 |  | 
| Christian Berendt | 71d5630 | 2013-07-22 11:37:42 +0200 | [diff] [blame] | 337 |             # Look for # testonly in comment | 
 | 338 |             if [[ $line =~ (.*)#.*testonly.* ]]; then | 
 | 339 |                 package=${BASH_REMATCH[1]} | 
 | 340 |                 # Are we installing test packages? (test for the default value) | 
 | 341 |                 if [[ $INSTALL_TESTONLY_PACKAGES = "False" ]]; then | 
 | 342 |                     # If not installing test packages the skip this package | 
 | 343 |                     inst_pkg=0 | 
 | 344 |                 fi | 
 | 345 |             fi | 
 | 346 |  | 
 | 347 |             if [[ $inst_pkg = 1 ]]; then | 
 | 348 |                 echo $package | 
 | 349 |             fi | 
| Dean Troyer | 7e27051 | 2012-06-14 15:23:24 -0500 | [diff] [blame] | 350 |         done | 
 | 351 |         IFS=$OIFS | 
 | 352 |     done | 
 | 353 | } | 
 | 354 |  | 
 | 355 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 356 | # Determine OS Vendor, Release and Update | 
 | 357 | # Tested with OS/X, Ubuntu, RedHat, CentOS, Fedora | 
 | 358 | # Returns results in global variables: | 
 | 359 | # os_VENDOR - vendor name | 
 | 360 | # os_RELEASE - release | 
 | 361 | # os_UPDATE - update | 
 | 362 | # os_PACKAGE - package type | 
 | 363 | # os_CODENAME - vendor's codename for release | 
 | 364 | # GetOSVersion | 
 | 365 | GetOSVersion() { | 
 | 366 |     # Figure out which vendor we are | 
 | 367 |     if [[ -n "`which sw_vers 2>/dev/null`" ]]; then | 
 | 368 |         # OS/X | 
 | 369 |         os_VENDOR=`sw_vers -productName` | 
 | 370 |         os_RELEASE=`sw_vers -productVersion` | 
 | 371 |         os_UPDATE=${os_RELEASE##*.} | 
 | 372 |         os_RELEASE=${os_RELEASE%.*} | 
 | 373 |         os_PACKAGE="" | 
 | 374 |         if [[ "$os_RELEASE" =~ "10.7" ]]; then | 
 | 375 |             os_CODENAME="lion" | 
 | 376 |         elif [[ "$os_RELEASE" =~ "10.6" ]]; then | 
 | 377 |             os_CODENAME="snow leopard" | 
 | 378 |         elif [[ "$os_RELEASE" =~ "10.5" ]]; then | 
 | 379 |             os_CODENAME="leopard" | 
 | 380 |         elif [[ "$os_RELEASE" =~ "10.4" ]]; then | 
 | 381 |             os_CODENAME="tiger" | 
 | 382 |         elif [[ "$os_RELEASE" =~ "10.3" ]]; then | 
 | 383 |             os_CODENAME="panther" | 
 | 384 |         else | 
 | 385 |             os_CODENAME="" | 
 | 386 |         fi | 
 | 387 |     elif [[ -x $(which lsb_release 2>/dev/null) ]]; then | 
 | 388 |         os_VENDOR=$(lsb_release -i -s) | 
 | 389 |         os_RELEASE=$(lsb_release -r -s) | 
 | 390 |         os_UPDATE="" | 
| Attila Fazekas | af988fd | 2013-01-13 14:20:47 +0100 | [diff] [blame] | 391 |         os_PACKAGE="rpm" | 
| Derek Morton | 4a8496e | 2013-04-08 23:46:08 -0500 | [diff] [blame] | 392 |         if [[ "Debian,Ubuntu,LinuxMint" =~ $os_VENDOR ]]; then | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 393 |             os_PACKAGE="deb" | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 394 |         elif [[ "SUSE LINUX" =~ $os_VENDOR ]]; then | 
 | 395 |             lsb_release -d -s | grep -q openSUSE | 
 | 396 |             if [[ $? -eq 0 ]]; then | 
 | 397 |                 os_VENDOR="openSUSE" | 
 | 398 |             fi | 
| Vincent Untz | cd1fe98 | 2013-03-12 18:04:29 +0100 | [diff] [blame] | 399 |         elif [[ $os_VENDOR == "openSUSE project" ]]; then | 
 | 400 |             os_VENDOR="openSUSE" | 
| Attila Fazekas | af988fd | 2013-01-13 14:20:47 +0100 | [diff] [blame] | 401 |         elif [[ $os_VENDOR =~ Red.*Hat ]]; then | 
 | 402 |             os_VENDOR="Red Hat" | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 403 |         fi | 
 | 404 |         os_CODENAME=$(lsb_release -c -s) | 
 | 405 |     elif [[ -r /etc/redhat-release ]]; then | 
 | 406 |         # Red Hat Enterprise Linux Server release 5.5 (Tikanga) | 
 | 407 |         # CentOS release 5.5 (Final) | 
 | 408 |         # CentOS Linux release 6.0 (Final) | 
 | 409 |         # Fedora release 16 (Verne) | 
| Bob Ball | 4669122 | 2013-08-12 17:28:50 +0100 | [diff] [blame] | 410 |         # XenServer release 6.2.0-70446c (xenenterprise) | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 411 |         os_CODENAME="" | 
| Bob Ball | 4669122 | 2013-08-12 17:28:50 +0100 | [diff] [blame] | 412 |         for r in "Red Hat" CentOS Fedora XenServer; do | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 413 |             os_VENDOR=$r | 
 | 414 |             if [[ -n "`grep \"$r\" /etc/redhat-release`" ]]; then | 
 | 415 |                 ver=`sed -e 's/^.* \(.*\) (\(.*\)).*$/\1\|\2/' /etc/redhat-release` | 
 | 416 |                 os_CODENAME=${ver#*|} | 
 | 417 |                 os_RELEASE=${ver%|*} | 
 | 418 |                 os_UPDATE=${os_RELEASE##*.} | 
 | 419 |                 os_RELEASE=${os_RELEASE%.*} | 
 | 420 |                 break | 
 | 421 |             fi | 
 | 422 |             os_VENDOR="" | 
 | 423 |         done | 
 | 424 |         os_PACKAGE="rpm" | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 425 |     elif [[ -r /etc/SuSE-release ]]; then | 
 | 426 |         for r in openSUSE "SUSE Linux"; do | 
 | 427 |             if [[ "$r" = "SUSE Linux" ]]; then | 
 | 428 |                 os_VENDOR="SUSE LINUX" | 
 | 429 |             else | 
 | 430 |                 os_VENDOR=$r | 
 | 431 |             fi | 
 | 432 |  | 
 | 433 |             if [[ -n "`grep \"$r\" /etc/SuSE-release`" ]]; then | 
 | 434 |                 os_CODENAME=`grep "CODENAME = " /etc/SuSE-release | sed 's:.* = ::g'` | 
 | 435 |                 os_RELEASE=`grep "VERSION = " /etc/SuSE-release | sed 's:.* = ::g'` | 
 | 436 |                 os_UPDATE=`grep "PATCHLEVEL = " /etc/SuSE-release | sed 's:.* = ::g'` | 
 | 437 |                 break | 
 | 438 |             fi | 
 | 439 |             os_VENDOR="" | 
 | 440 |         done | 
 | 441 |         os_PACKAGE="rpm" | 
| Émilien Macchi | b2ef890 | 2013-05-04 00:48:20 +0200 | [diff] [blame] | 442 |     # If lsb_release is not installed, we should be able to detect Debian OS | 
 | 443 |     elif [[ -f /etc/debian_version ]] && [[ $(cat /proc/version) =~ "Debian" ]]; then | 
 | 444 |         os_VENDOR="Debian" | 
 | 445 |         os_PACKAGE="deb" | 
 | 446 |         os_CODENAME=$(awk '/VERSION=/' /etc/os-release | sed 's/VERSION=//' | sed -r 's/\"|\(|\)//g' | awk '{print $2}') | 
 | 447 |         os_RELEASE=$(awk '/VERSION_ID=/' /etc/os-release | sed 's/VERSION_ID=//' | sed 's/\"//g') | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 448 |     fi | 
 | 449 |     export os_VENDOR os_RELEASE os_UPDATE os_PACKAGE os_CODENAME | 
 | 450 | } | 
 | 451 |  | 
| Andrew Laski | f900bd7 | 2012-09-05 17:23:14 -0400 | [diff] [blame] | 452 |  | 
| Dean Troyer | a9e0a48 | 2012-07-09 14:07:23 -0500 | [diff] [blame] | 453 | # Translate the OS version values into common nomenclature | 
 | 454 | # Sets ``DISTRO`` from the ``os_*`` values | 
 | 455 | function GetDistro() { | 
 | 456 |     GetOSVersion | 
| Émilien Macchi | b2ef890 | 2013-05-04 00:48:20 +0200 | [diff] [blame] | 457 |     if [[ "$os_VENDOR" =~ (Ubuntu) || "$os_VENDOR" =~ (Debian) ]]; then | 
 | 458 |         # 'Everyone' refers to Ubuntu / Debian releases by the code name adjective | 
| Dean Troyer | a9e0a48 | 2012-07-09 14:07:23 -0500 | [diff] [blame] | 459 |         DISTRO=$os_CODENAME | 
 | 460 |     elif [[ "$os_VENDOR" =~ (Fedora) ]]; then | 
 | 461 |         # For Fedora, just use 'f' and the release | 
 | 462 |         DISTRO="f$os_RELEASE" | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 463 |     elif [[ "$os_VENDOR" =~ (openSUSE) ]]; then | 
 | 464 |         DISTRO="opensuse-$os_RELEASE" | 
 | 465 |     elif [[ "$os_VENDOR" =~ (SUSE LINUX) ]]; then | 
 | 466 |         # For SLE, also use the service pack | 
 | 467 |         if [[ -z "$os_UPDATE" ]]; then | 
 | 468 |             DISTRO="sle${os_RELEASE}" | 
 | 469 |         else | 
 | 470 |             DISTRO="sle${os_RELEASE}sp${os_UPDATE}" | 
 | 471 |         fi | 
| Ian Wienand | d857f4b | 2013-03-20 14:51:06 +1100 | [diff] [blame] | 472 |     elif [[ "$os_VENDOR" =~ (Red Hat) || "$os_VENDOR" =~ (CentOS) ]]; then | 
 | 473 |         # Drop the . release as we assume it's compatible | 
 | 474 |         DISTRO="rhel${os_RELEASE::1}" | 
| Bob Ball | 4669122 | 2013-08-12 17:28:50 +0100 | [diff] [blame] | 475 |     elif [[ "$os_VENDOR" =~ (XenServer) ]]; then | 
 | 476 |         DISTRO="xs$os_RELEASE" | 
| Dean Troyer | a9e0a48 | 2012-07-09 14:07:23 -0500 | [diff] [blame] | 477 |     else | 
 | 478 |         # Catch-all for now is Vendor + Release + Update | 
 | 479 |         DISTRO="$os_VENDOR-$os_RELEASE.$os_UPDATE" | 
 | 480 |     fi | 
 | 481 |     export DISTRO | 
 | 482 | } | 
 | 483 |  | 
 | 484 |  | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 485 | # Determine if current distribution is a Fedora-based distribution | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 486 | # (Fedora, RHEL, CentOS, etc). | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 487 | # is_fedora | 
 | 488 | function is_fedora { | 
 | 489 |     if [[ -z "$os_VENDOR" ]]; then | 
 | 490 |         GetOSVersion | 
 | 491 |     fi | 
 | 492 |  | 
 | 493 |     [ "$os_VENDOR" = "Fedora" ] || [ "$os_VENDOR" = "Red Hat" ] || [ "$os_VENDOR" = "CentOS" ] | 
 | 494 | } | 
 | 495 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 496 |  | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 497 | # Determine if current distribution is a SUSE-based distribution | 
 | 498 | # (openSUSE, SLE). | 
 | 499 | # is_suse | 
 | 500 | function is_suse { | 
 | 501 |     if [[ -z "$os_VENDOR" ]]; then | 
 | 502 |         GetOSVersion | 
 | 503 |     fi | 
 | 504 |  | 
| Steve Baker | 1a7bbd2 | 2012-12-03 17:04:02 +1300 | [diff] [blame] | 505 |     [ "$os_VENDOR" = "openSUSE" ] || [ "$os_VENDOR" = "SUSE LINUX" ] | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 506 | } | 
 | 507 |  | 
 | 508 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 509 | # Determine if current distribution is an Ubuntu-based distribution | 
 | 510 | # It will also detect non-Ubuntu but Debian-based distros | 
 | 511 | # is_ubuntu | 
 | 512 | function is_ubuntu { | 
 | 513 |     if [[ -z "$os_PACKAGE" ]]; then | 
 | 514 |         GetOSVersion | 
 | 515 |     fi | 
 | 516 |     [ "$os_PACKAGE" = "deb" ] | 
 | 517 | } | 
 | 518 |  | 
 | 519 |  | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 520 | # Exit after outputting a message about the distribution not being supported. | 
 | 521 | # exit_distro_not_supported [optional-string-telling-what-is-missing] | 
 | 522 | function exit_distro_not_supported { | 
 | 523 |     if [[ -z "$DISTRO" ]]; then | 
 | 524 |         GetDistro | 
 | 525 |     fi | 
 | 526 |  | 
 | 527 |     if [ $# -gt 0 ]; then | 
| Nachi Ueno | 07115eb | 2013-02-26 12:38:18 -0800 | [diff] [blame] | 528 |         die $LINENO "Support for $DISTRO is incomplete: no support for $@" | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 529 |     else | 
| Nachi Ueno | 07115eb | 2013-02-26 12:38:18 -0800 | [diff] [blame] | 530 |         die $LINENO "Support for $DISTRO is incomplete." | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 531 |     fi | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 532 | } | 
 | 533 |  | 
| Daniel Jones | fa868cb | 2013-06-18 15:28:01 -0500 | [diff] [blame] | 534 | # Utility function for checking machine architecture | 
 | 535 | # is_arch arch-type | 
 | 536 | function is_arch { | 
 | 537 |     ARCH_TYPE=$1 | 
 | 538 |  | 
 | 539 |     [ "($uname -m)" = "$ARCH_TYPE" ] | 
 | 540 | } | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 541 |  | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 542 | # git clone only if directory doesn't exist already.  Since ``DEST`` might not | 
 | 543 | # be owned by the installation user, we create the directory and change the | 
 | 544 | # ownership to the proper user. | 
 | 545 | # Set global RECLONE=yes to simulate a clone when dest-dir exists | 
| James E. Blair | 94cb960 | 2012-06-22 15:28:29 -0700 | [diff] [blame] | 546 | # Set global ERROR_ON_CLONE=True to abort execution with an error if the git repo | 
 | 547 | # does not exist (default is False, meaning the repo will be cloned). | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 548 | # Uses global ``OFFLINE`` | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 549 | # git_clone remote dest-dir branch | 
 | 550 | function git_clone { | 
 | 551 |     [[ "$OFFLINE" = "True" ]] && return | 
 | 552 |  | 
 | 553 |     GIT_REMOTE=$1 | 
 | 554 |     GIT_DEST=$2 | 
| Evgeniy Afonichev | 6a3912d | 2012-07-10 14:02:43 +0300 | [diff] [blame] | 555 |     GIT_REF=$3 | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 556 |  | 
| Evgeniy Afonichev | 6a3912d | 2012-07-10 14:02:43 +0300 | [diff] [blame] | 557 |     if echo $GIT_REF | egrep -q "^refs"; then | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 558 |         # If our branch name is a gerrit style refs/changes/... | 
 | 559 |         if [[ ! -d $GIT_DEST ]]; then | 
| James E. Blair | 94cb960 | 2012-06-22 15:28:29 -0700 | [diff] [blame] | 560 |             [[ "$ERROR_ON_CLONE" = "True" ]] && exit 1 | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 561 |             git clone $GIT_REMOTE $GIT_DEST | 
 | 562 |         fi | 
 | 563 |         cd $GIT_DEST | 
| Evgeniy Afonichev | 6a3912d | 2012-07-10 14:02:43 +0300 | [diff] [blame] | 564 |         git fetch $GIT_REMOTE $GIT_REF && git checkout FETCH_HEAD | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 565 |     else | 
 | 566 |         # do a full clone only if the directory doesn't exist | 
 | 567 |         if [[ ! -d $GIT_DEST ]]; then | 
| James E. Blair | 94cb960 | 2012-06-22 15:28:29 -0700 | [diff] [blame] | 568 |             [[ "$ERROR_ON_CLONE" = "True" ]] && exit 1 | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 569 |             git clone $GIT_REMOTE $GIT_DEST | 
 | 570 |             cd $GIT_DEST | 
 | 571 |             # This checkout syntax works for both branches and tags | 
| Evgeniy Afonichev | 6a3912d | 2012-07-10 14:02:43 +0300 | [diff] [blame] | 572 |             git checkout $GIT_REF | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 573 |         elif [[ "$RECLONE" == "yes" ]]; then | 
 | 574 |             # if it does exist then simulate what clone does if asked to RECLONE | 
 | 575 |             cd $GIT_DEST | 
 | 576 |             # set the url to pull from and fetch | 
 | 577 |             git remote set-url origin $GIT_REMOTE | 
 | 578 |             git fetch origin | 
 | 579 |             # remove the existing ignored files (like pyc) as they cause breakage | 
 | 580 |             # (due to the py files having older timestamps than our pyc, so python | 
 | 581 |             # thinks the pyc files are correct using them) | 
 | 582 |             find $GIT_DEST -name '*.pyc' -delete | 
| Evgeniy Afonichev | 6a3912d | 2012-07-10 14:02:43 +0300 | [diff] [blame] | 583 |  | 
 | 584 |             # handle GIT_REF accordingly to type (tag, branch) | 
 | 585 |             if [[ -n "`git show-ref refs/tags/$GIT_REF`" ]]; then | 
 | 586 |                 git_update_tag $GIT_REF | 
 | 587 |             elif [[ -n "`git show-ref refs/heads/$GIT_REF`" ]]; then | 
 | 588 |                 git_update_branch $GIT_REF | 
| Andrew Laski | f900bd7 | 2012-09-05 17:23:14 -0400 | [diff] [blame] | 589 |             elif [[ -n "`git show-ref refs/remotes/origin/$GIT_REF`" ]]; then | 
 | 590 |                 git_update_remote_branch $GIT_REF | 
| Evgeniy Afonichev | 6a3912d | 2012-07-10 14:02:43 +0300 | [diff] [blame] | 591 |             else | 
 | 592 |                 echo $GIT_REF is neither branch nor tag | 
 | 593 |                 exit 1 | 
 | 594 |             fi | 
 | 595 |  | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 596 |         fi | 
 | 597 |     fi | 
 | 598 | } | 
 | 599 |  | 
 | 600 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 601 | # git update using reference as a branch. | 
 | 602 | # git_update_branch ref | 
 | 603 | function git_update_branch() { | 
 | 604 |  | 
 | 605 |     GIT_BRANCH=$1 | 
 | 606 |  | 
 | 607 |     git checkout -f origin/$GIT_BRANCH | 
 | 608 |     # a local branch might not exist | 
 | 609 |     git branch -D $GIT_BRANCH || true | 
 | 610 |     git checkout -b $GIT_BRANCH | 
 | 611 | } | 
 | 612 |  | 
 | 613 |  | 
 | 614 | # git update using reference as a branch. | 
 | 615 | # git_update_remote_branch ref | 
 | 616 | function git_update_remote_branch() { | 
 | 617 |  | 
 | 618 |     GIT_BRANCH=$1 | 
 | 619 |  | 
 | 620 |     git checkout -b $GIT_BRANCH -t origin/$GIT_BRANCH | 
 | 621 | } | 
 | 622 |  | 
 | 623 |  | 
 | 624 | # git update using reference as a tag. Be careful editing source at that repo | 
 | 625 | # as working copy will be in a detached mode | 
 | 626 | # git_update_tag ref | 
 | 627 | function git_update_tag() { | 
 | 628 |  | 
 | 629 |     GIT_TAG=$1 | 
 | 630 |  | 
 | 631 |     git tag -d $GIT_TAG | 
 | 632 |     # fetching given tag only | 
 | 633 |     git fetch origin tag $GIT_TAG | 
 | 634 |     git checkout -f $GIT_TAG | 
 | 635 | } | 
 | 636 |  | 
 | 637 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 638 | # Comment an option in an INI file | 
| Chmouel Boudjnah | c7214e8 | 2012-06-06 13:56:39 +0200 | [diff] [blame] | 639 | # inicomment config-file section option | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 640 | function inicomment() { | 
 | 641 |     local file=$1 | 
 | 642 |     local section=$2 | 
 | 643 |     local option=$3 | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 644 |     sed -i -e "/^\[$section\]/,/^\[.*\]/ s|^\($option[ \t]*=.*$\)|#\1|" "$file" | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 645 | } | 
 | 646 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 647 |  | 
| Chmouel Boudjnah | c7214e8 | 2012-06-06 13:56:39 +0200 | [diff] [blame] | 648 | # Uncomment an option in an INI file | 
 | 649 | # iniuncomment config-file section option | 
 | 650 | function iniuncomment() { | 
 | 651 |     local file=$1 | 
 | 652 |     local section=$2 | 
 | 653 |     local option=$3 | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 654 |     sed -i -e "/^\[$section\]/,/^\[.*\]/ s|[^ \t]*#[ \t]*\($option[ \t]*=.*$\)|\1|" "$file" | 
| Chmouel Boudjnah | c7214e8 | 2012-06-06 13:56:39 +0200 | [diff] [blame] | 655 | } | 
 | 656 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 657 |  | 
 | 658 | # Get an option from an INI file | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 659 | # iniget config-file section option | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 660 | function iniget() { | 
 | 661 |     local file=$1 | 
 | 662 |     local section=$2 | 
 | 663 |     local option=$3 | 
 | 664 |     local line | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 665 |     line=$(sed -ne "/^\[$section\]/,/^\[.*\]/ { /^$option[ \t]*=/ p; }" "$file") | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 666 |     echo ${line#*=} | 
 | 667 | } | 
 | 668 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 669 |  | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 670 | # Determinate is the given option present in the INI file | 
 | 671 | # ini_has_option config-file section option | 
 | 672 | function ini_has_option() { | 
 | 673 |     local file=$1 | 
 | 674 |     local section=$2 | 
 | 675 |     local option=$3 | 
 | 676 |     local line | 
 | 677 |     line=$(sed -ne "/^\[$section\]/,/^\[.*\]/ { /^$option[ \t]*=/ p; }" "$file") | 
 | 678 |     [ -n "$line" ] | 
 | 679 | } | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 680 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 681 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 682 | # Set an option in an INI file | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 683 | # iniset config-file section option value | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 684 | function iniset() { | 
 | 685 |     local file=$1 | 
 | 686 |     local section=$2 | 
 | 687 |     local option=$3 | 
 | 688 |     local value=$4 | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 689 |     if ! grep -q "^\[$section\]" "$file"; then | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 690 |         # Add section at the end | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 691 |         echo -e "\n[$section]" >>"$file" | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 692 |     fi | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 693 |     if ! ini_has_option "$file" "$section" "$option"; then | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 694 |         # Add it | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 695 |         sed -i -e "/^\[$section\]/ a\\ | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 696 | $option = $value | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 697 | " "$file" | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 698 |     else | 
 | 699 |         # Replace it | 
| Attila Fazekas | 588eb41 | 2012-12-20 10:57:16 +0100 | [diff] [blame] | 700 |         sed -i -e "/^\[$section\]/,/^\[.*\]/ s|^\($option[ \t]*=[ \t]*\).*$|\1$value|" "$file" | 
| Dean Troyer | 09e636e | 2012-03-19 16:31:12 -0500 | [diff] [blame] | 701 |     fi | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 702 | } | 
 | 703 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 704 |  | 
| Lianhao Lu | 239f324 | 2013-03-01 15:54:02 +0800 | [diff] [blame] | 705 | # Get a multiple line option from an INI file | 
 | 706 | # iniget_multiline config-file section option | 
 | 707 | function iniget_multiline() { | 
 | 708 |     local file=$1 | 
 | 709 |     local section=$2 | 
 | 710 |     local option=$3 | 
 | 711 |     local values | 
 | 712 |     values=$(sed -ne "/^\[$section\]/,/^\[.*\]/ { s/^$option[ \t]*=[ \t]*//gp; }" "$file") | 
 | 713 |     echo ${values} | 
 | 714 | } | 
 | 715 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 716 |  | 
| Lianhao Lu | 239f324 | 2013-03-01 15:54:02 +0800 | [diff] [blame] | 717 | # Set a multiple line option in an INI file | 
 | 718 | # iniset_multiline config-file section option value1 value2 valu3 ... | 
 | 719 | function iniset_multiline() { | 
 | 720 |     local file=$1 | 
 | 721 |     local section=$2 | 
 | 722 |     local option=$3 | 
 | 723 |     shift 3 | 
 | 724 |     local values | 
 | 725 |     for v in $@; do | 
 | 726 |         # The later sed command inserts each new value in the line next to | 
 | 727 |         # the section identifier, which causes the values to be inserted in | 
 | 728 |         # the reverse order. Do a reverse here to keep the original order. | 
 | 729 |         values="$v ${values}" | 
 | 730 |     done | 
 | 731 |     if ! grep -q "^\[$section\]" "$file"; then | 
 | 732 |         # Add section at the end | 
 | 733 |         echo -e "\n[$section]" >>"$file" | 
 | 734 |     else | 
 | 735 |         # Remove old values | 
 | 736 |         sed -i -e "/^\[$section\]/,/^\[.*\]/ { /^$option[ \t]*=/ d; }" "$file" | 
 | 737 |     fi | 
 | 738 |     # Add new ones | 
 | 739 |     for v in $values; do | 
 | 740 |         sed -i -e "/^\[$section\]/ a\\ | 
 | 741 | $option = $v | 
 | 742 | " "$file" | 
 | 743 |     done | 
 | 744 | } | 
 | 745 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 746 |  | 
| Lianhao Lu | 239f324 | 2013-03-01 15:54:02 +0800 | [diff] [blame] | 747 | # Append a new option in an ini file without replacing the old value | 
 | 748 | # iniadd config-file section option value1 value2 value3 ... | 
 | 749 | function iniadd() { | 
 | 750 |     local file=$1 | 
 | 751 |     local section=$2 | 
 | 752 |     local option=$3 | 
 | 753 |     shift 3 | 
 | 754 |     local values="$(iniget_multiline $file $section $option) $@" | 
 | 755 |     iniset_multiline $file $section $option $values | 
 | 756 | } | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 757 |  | 
| Dean Troyer | 896eb66 | 2013-04-05 15:02:01 -0500 | [diff] [blame] | 758 | # Find out if a process exists by partial name. | 
 | 759 | # is_running name | 
 | 760 | function is_running() { | 
 | 761 |     local name=$1 | 
 | 762 |     ps auxw | grep -v grep | grep ${name} > /dev/null | 
 | 763 |     RC=$? | 
 | 764 |     # some times I really hate bash reverse binary logic | 
 | 765 |     return $RC | 
 | 766 | } | 
 | 767 |  | 
 | 768 |  | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 769 | # is_service_enabled() checks if the service(s) specified as arguments are | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 770 | # enabled by the user in ``ENABLED_SERVICES``. | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 771 | # | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 772 | # Multiple services specified as arguments are ``OR``'ed together; the test | 
 | 773 | # is a short-circuit boolean, i.e it returns on the first match. | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 774 | # | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 775 | # There are special cases for some 'catch-all' services:: | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 776 | #   **nova** returns true if any service enabled start with **n-** | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 777 | #   **cinder** returns true if any service enabled start with **c-** | 
 | 778 | #   **ceilometer** returns true if any service enabled start with **ceilometer** | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 779 | #   **glance** returns true if any service enabled start with **g-** | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 780 | #   **neutron** returns true if any service enabled start with **q-** | 
| Chmouel Boudjnah | 0c3a558 | 2013-03-06 10:58:33 +0100 | [diff] [blame] | 781 | #   **swift** returns true if any service enabled start with **s-** | 
 | 782 | #   For backward compatibility if we have **swift** in ENABLED_SERVICES all the | 
 | 783 | #   **s-** services will be enabled. This will be deprecated in the future. | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 784 | # | 
| Chris Behrens | c62c2b9 | 2013-07-24 03:56:13 -0700 | [diff] [blame] | 785 | # Cells within nova is enabled if **n-cell** is in ``ENABLED_SERVICES``. | 
 | 786 | # We also need to make sure to treat **n-cell-region** and **n-cell-child** | 
 | 787 | # as enabled in this case. | 
 | 788 | # | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 789 | # Uses global ``ENABLED_SERVICES`` | 
 | 790 | # is_service_enabled service [service ...] | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 791 | function is_service_enabled() { | 
 | 792 |     services=$@ | 
 | 793 |     for service in ${services}; do | 
 | 794 |         [[ ,${ENABLED_SERVICES}, =~ ,${service}, ]] && return 0 | 
| Chris Behrens | c62c2b9 | 2013-07-24 03:56:13 -0700 | [diff] [blame] | 795 |         [[ ${service} == n-cell-* && ${ENABLED_SERVICES} =~ "n-cell" ]] && return 0 | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 796 |         [[ ${service} == "nova" && ${ENABLED_SERVICES} =~ "n-" ]] && return 0 | 
| Dean Troyer | 67787e6 | 2012-05-02 11:48:15 -0500 | [diff] [blame] | 797 |         [[ ${service} == "cinder" && ${ENABLED_SERVICES} =~ "c-" ]] && return 0 | 
| John H. Tran | 9336164 | 2012-07-26 11:22:05 -0700 | [diff] [blame] | 798 |         [[ ${service} == "ceilometer" && ${ENABLED_SERVICES} =~ "ceilometer-" ]] && return 0 | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 799 |         [[ ${service} == "glance" && ${ENABLED_SERVICES} =~ "g-" ]] && return 0 | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 800 |         [[ ${service} == "neutron" && ${ENABLED_SERVICES} =~ "q-" ]] && return 0 | 
| Chmouel Boudjnah | 0c3a558 | 2013-03-06 10:58:33 +0100 | [diff] [blame] | 801 |         [[ ${service} == "swift" && ${ENABLED_SERVICES} =~ "s-" ]] && return 0 | 
 | 802 |         [[ ${service} == s-* && ${ENABLED_SERVICES} =~ "swift" ]] && return 0 | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 803 |     done | 
 | 804 |     return 1 | 
 | 805 | } | 
 | 806 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 807 |  | 
 | 808 | # remove extra commas from the input string (i.e. ``ENABLED_SERVICES``) | 
 | 809 | # _cleanup_service_list service-list | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 810 | function _cleanup_service_list () { | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 811 |     echo "$1" | sed -e ' | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 812 |         s/,,/,/g; | 
 | 813 |         s/^,//; | 
 | 814 |         s/,$// | 
 | 815 |     ' | 
 | 816 | } | 
 | 817 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 818 |  | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 819 | # enable_service() adds the services passed as argument to the | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 820 | # ``ENABLED_SERVICES`` list, if they are not already present. | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 821 | # | 
 | 822 | # For example: | 
| Joe Gordon | 6fd2811 | 2012-11-13 16:55:41 -0800 | [diff] [blame] | 823 | #   enable_service qpid | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 824 | # | 
 | 825 | # This function does not know about the special cases | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 826 | # for nova, glance, and neutron built into is_service_enabled(). | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 827 | # Uses global ``ENABLED_SERVICES`` | 
 | 828 | # enable_service service [service ...] | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 829 | function enable_service() { | 
 | 830 |     local tmpsvcs="${ENABLED_SERVICES}" | 
 | 831 |     for service in $@; do | 
 | 832 |         if ! is_service_enabled $service; then | 
 | 833 |             tmpsvcs+=",$service" | 
 | 834 |         fi | 
 | 835 |     done | 
 | 836 |     ENABLED_SERVICES=$(_cleanup_service_list "$tmpsvcs") | 
 | 837 |     disable_negated_services | 
 | 838 | } | 
 | 839 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 840 |  | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 841 | # disable_service() removes the services passed as argument to the | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 842 | # ``ENABLED_SERVICES`` list, if they are present. | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 843 | # | 
 | 844 | # For example: | 
| Joe Gordon | 6fd2811 | 2012-11-13 16:55:41 -0800 | [diff] [blame] | 845 | #   disable_service rabbit | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 846 | # | 
 | 847 | # This function does not know about the special cases | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 848 | # for nova, glance, and neutron built into is_service_enabled(). | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 849 | # Uses global ``ENABLED_SERVICES`` | 
 | 850 | # disable_service service [service ...] | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 851 | function disable_service() { | 
 | 852 |     local tmpsvcs=",${ENABLED_SERVICES}," | 
 | 853 |     local service | 
 | 854 |     for service in $@; do | 
 | 855 |         if is_service_enabled $service; then | 
 | 856 |             tmpsvcs=${tmpsvcs//,$service,/,} | 
 | 857 |         fi | 
 | 858 |     done | 
 | 859 |     ENABLED_SERVICES=$(_cleanup_service_list "$tmpsvcs") | 
 | 860 | } | 
 | 861 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 862 |  | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 863 | # disable_all_services() removes all current services | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 864 | # from ``ENABLED_SERVICES`` to reset the configuration | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 865 | # before a minimal installation | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 866 | # Uses global ``ENABLED_SERVICES`` | 
 | 867 | # disable_all_services | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 868 | function disable_all_services() { | 
 | 869 |     ENABLED_SERVICES="" | 
 | 870 | } | 
 | 871 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 872 |  | 
 | 873 | # Remove all services starting with '-'.  For example, to install all default | 
| Joe Gordon | 6fd2811 | 2012-11-13 16:55:41 -0800 | [diff] [blame] | 874 | # services except rabbit (rabbit) set in ``localrc``: | 
 | 875 | # ENABLED_SERVICES+=",-rabbit" | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 876 | # Uses global ``ENABLED_SERVICES`` | 
 | 877 | # disable_negated_services | 
| Doug Hellmann | f04178f | 2012-07-05 17:10:03 -0400 | [diff] [blame] | 878 | function disable_negated_services() { | 
 | 879 |     local tmpsvcs="${ENABLED_SERVICES}" | 
 | 880 |     local service | 
 | 881 |     for service in ${tmpsvcs//,/ }; do | 
 | 882 |         if [[ ${service} == -* ]]; then | 
 | 883 |             tmpsvcs=$(echo ${tmpsvcs}|sed -r "s/(,)?(-)?${service#-}(,)?/,/g") | 
 | 884 |         fi | 
 | 885 |     done | 
 | 886 |     ENABLED_SERVICES=$(_cleanup_service_list "$tmpsvcs") | 
 | 887 | } | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 888 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 889 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 890 | # Distro-agnostic package installer | 
 | 891 | # install_package package [package ...] | 
 | 892 | function install_package() { | 
| Vincent Untz | c18b965 | 2012-12-04 12:36:34 +0100 | [diff] [blame] | 893 |     if is_ubuntu; then | 
| Vincent Untz | c0482e6 | 2012-06-12 11:30:43 +0200 | [diff] [blame] | 894 |         [[ "$NO_UPDATE_REPOS" = "True" ]] || apt_get update | 
 | 895 |         NO_UPDATE_REPOS=True | 
 | 896 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 897 |         apt_get install "$@" | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 898 |     elif is_fedora; then | 
 | 899 |         yum_install "$@" | 
 | 900 |     elif is_suse; then | 
 | 901 |         zypper_install "$@" | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 902 |     else | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 903 |         exit_distro_not_supported "installing packages" | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 904 |     fi | 
 | 905 | } | 
 | 906 |  | 
 | 907 |  | 
| Dean Troyer | 995eb92 | 2013-03-07 16:11:40 -0600 | [diff] [blame] | 908 | # Distro-agnostic package uninstaller | 
 | 909 | # uninstall_package package [package ...] | 
 | 910 | function uninstall_package() { | 
 | 911 |     if is_ubuntu; then | 
 | 912 |         apt_get purge "$@" | 
 | 913 |     elif is_fedora; then | 
| Ian Wienand | 2c678cc | 2013-03-20 13:00:44 +1100 | [diff] [blame] | 914 |         sudo yum remove -y "$@" | 
| Dean Troyer | 995eb92 | 2013-03-07 16:11:40 -0600 | [diff] [blame] | 915 |     elif is_suse; then | 
| Ian Wienand | 2c678cc | 2013-03-20 13:00:44 +1100 | [diff] [blame] | 916 |         sudo rpm -e "$@" | 
| Dean Troyer | 995eb92 | 2013-03-07 16:11:40 -0600 | [diff] [blame] | 917 |     else | 
 | 918 |         exit_distro_not_supported "uninstalling packages" | 
 | 919 |     fi | 
 | 920 | } | 
 | 921 |  | 
 | 922 |  | 
| Vincent Untz | 71ebc6f | 2012-06-12 13:45:15 +0200 | [diff] [blame] | 923 | # Distro-agnostic function to tell if a package is installed | 
 | 924 | # is_package_installed package [package ...] | 
 | 925 | function is_package_installed() { | 
 | 926 |     if [[ -z "$@" ]]; then | 
 | 927 |         return 1 | 
 | 928 |     fi | 
 | 929 |  | 
 | 930 |     if [[ -z "$os_PACKAGE" ]]; then | 
 | 931 |         GetOSVersion | 
 | 932 |     fi | 
| Vincent Untz | c18b965 | 2012-12-04 12:36:34 +0100 | [diff] [blame] | 933 |  | 
| Vincent Untz | 71ebc6f | 2012-06-12 13:45:15 +0200 | [diff] [blame] | 934 |     if [[ "$os_PACKAGE" = "deb" ]]; then | 
| Dean Troyer | 04762cd | 2013-08-27 17:06:14 -0500 | [diff] [blame] | 935 |         dpkg -s "$@" > /dev/null 2> /dev/null | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 936 |     elif [[ "$os_PACKAGE" = "rpm" ]]; then | 
| Vincent Untz | 71ebc6f | 2012-06-12 13:45:15 +0200 | [diff] [blame] | 937 |         rpm --quiet -q "$@" | 
| Vincent Untz | 00011c0 | 2012-12-06 09:56:32 +0100 | [diff] [blame] | 938 |     else | 
 | 939 |         exit_distro_not_supported "finding if a package is installed" | 
| Vincent Untz | 71ebc6f | 2012-06-12 13:45:15 +0200 | [diff] [blame] | 940 |     fi | 
 | 941 | } | 
 | 942 |  | 
 | 943 |  | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 944 | # Test if the named environment variable is set and not zero length | 
 | 945 | # is_set env-var | 
 | 946 | function is_set() { | 
 | 947 |     local var=\$"$1" | 
| Attila Fazekas | 251d3b5 | 2012-12-16 15:05:44 +0100 | [diff] [blame] | 948 |     eval "[ -n \"$var\" ]" # For ex.: sh -c "[ -n \"$var\" ]" would be better, but several exercises depends on this | 
| Dean Troyer | 489bd2a | 2012-03-02 10:44:29 -0600 | [diff] [blame] | 949 | } | 
 | 950 |  | 
 | 951 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 952 | # Wrapper for ``pip install`` to set cache and proxy environment variables | 
| Maru Newby | 3a87edd | 2012-10-25 23:01:06 +0000 | [diff] [blame] | 953 | # Uses globals ``OFFLINE``, ``PIP_DOWNLOAD_CACHE``, ``PIP_USE_MIRRORS``, | 
 | 954 | #   ``TRACK_DEPENDS``, ``*_proxy` | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 955 | # pip_install package [package ...] | 
 | 956 | function pip_install { | 
| Dean Troyer | d0b21e2 | 2012-03-07 14:52:25 -0600 | [diff] [blame] | 957 |     [[ "$OFFLINE" = "True" || -z "$@" ]] && return | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 958 |     if [[ -z "$os_PACKAGE" ]]; then | 
 | 959 |         GetOSVersion | 
 | 960 |     fi | 
| Dean Troyer | cc6b443 | 2013-04-08 15:38:03 -0500 | [diff] [blame] | 961 |     if [[ $TRACK_DEPENDS = True ]]; then | 
| Monty Taylor | 47f0206 | 2012-07-26 11:09:24 -0500 | [diff] [blame] | 962 |         source $DEST/.venv/bin/activate | 
 | 963 |         CMD_PIP=$DEST/.venv/bin/pip | 
 | 964 |         SUDO_PIP="env" | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 965 |     else | 
| Monty Taylor | 47f0206 | 2012-07-26 11:09:24 -0500 | [diff] [blame] | 966 |         SUDO_PIP="sudo" | 
| Vincent Untz | 8ec2722 | 2012-11-29 09:25:31 +0100 | [diff] [blame] | 967 |         CMD_PIP=$(get_pip_command) | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 968 |     fi | 
| Ian Wienand | d67dd87 | 2013-04-11 11:14:36 +1000 | [diff] [blame] | 969 |  | 
| Roman Gorodeckij | 99405a4 | 2013-08-07 09:20:36 -0400 | [diff] [blame] | 970 |     # Mirror option not needed anymore because pypi has CDN available, | 
 | 971 |     # but it's useful in certain circumstances | 
 | 972 |     PIP_USE_MIRRORS=${PIP_USE_MIRRORS:-False} | 
| Maru Newby | 3a87edd | 2012-10-25 23:01:06 +0000 | [diff] [blame] | 973 |     if [[ "$PIP_USE_MIRRORS" != "False" ]]; then | 
 | 974 |         PIP_MIRROR_OPT="--use-mirrors" | 
 | 975 |     fi | 
| Ian Wienand | d67dd87 | 2013-04-11 11:14:36 +1000 | [diff] [blame] | 976 |  | 
| Ian Wienand | 31dcd3e | 2013-07-16 13:36:34 +1000 | [diff] [blame] | 977 |     # pip < 1.4 has a bug where it will use an already existing build | 
 | 978 |     # directory unconditionally.  Say an earlier component installs | 
 | 979 |     # foo v1.1; pip will have built foo's source in | 
 | 980 |     # /tmp/$USER-pip-build.  Even if a later component specifies foo < | 
 | 981 |     # 1.1, the existing extracted build will be used and cause | 
 | 982 |     # confusing errors.  By creating unique build directories we avoid | 
 | 983 |     # this problem. See | 
 | 984 |     #  https://github.com/pypa/pip/issues/709 | 
 | 985 |     local pip_build_tmp=$(mktemp --tmpdir -d pip-build.XXXXX) | 
 | 986 |  | 
| Monty Taylor | 47f0206 | 2012-07-26 11:09:24 -0500 | [diff] [blame] | 987 |     $SUDO_PIP PIP_DOWNLOAD_CACHE=${PIP_DOWNLOAD_CACHE:-/var/cache/pip} \ | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 988 |         HTTP_PROXY=$http_proxy \ | 
 | 989 |         HTTPS_PROXY=$https_proxy \ | 
| Osamu Habuka | 7abe4f2 | 2012-07-25 12:39:32 +0900 | [diff] [blame] | 990 |         NO_PROXY=$no_proxy \ | 
| Ian Wienand | 31dcd3e | 2013-07-16 13:36:34 +1000 | [diff] [blame] | 991 |         $CMD_PIP install --build=${pip_build_tmp} \ | 
 | 992 |         $PIP_MIRROR_OPT $@ \ | 
 | 993 |         && $SUDO_PIP rm -rf ${pip_build_tmp} | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 994 | } | 
 | 995 |  | 
 | 996 |  | 
| Ian Wienand | 31dcd3e | 2013-07-16 13:36:34 +1000 | [diff] [blame] | 997 | # Cleanup anything from /tmp on unstack | 
 | 998 | # clean_tmp | 
 | 999 | function cleanup_tmp { | 
 | 1000 |     local tmp_dir=${TMPDIR:-/tmp} | 
 | 1001 |  | 
 | 1002 |     # see comments in pip_install | 
 | 1003 |     sudo rm -rf ${tmp_dir}/pip-build.* | 
 | 1004 | } | 
 | 1005 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1006 | # Service wrapper to restart services | 
 | 1007 | # restart_service service-name | 
 | 1008 | function restart_service() { | 
| Vincent Untz | c18b965 | 2012-12-04 12:36:34 +0100 | [diff] [blame] | 1009 |     if is_ubuntu; then | 
| Dean Troyer | 5218d45 | 2012-02-04 02:13:23 -0600 | [diff] [blame] | 1010 |         sudo /usr/sbin/service $1 restart | 
 | 1011 |     else | 
 | 1012 |         sudo /sbin/service $1 restart | 
 | 1013 |     fi | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1014 | } | 
 | 1015 |  | 
 | 1016 |  | 
| Dean Troyer | 681f3fd | 2013-02-27 19:00:39 -0600 | [diff] [blame] | 1017 | # _run_process() is designed to be backgrounded by run_process() to simulate a | 
 | 1018 | # fork.  It includes the dirty work of closing extra filehandles and preparing log | 
 | 1019 | # files to produce the same logs as screen_it().  The log filename is derived | 
 | 1020 | # from the service name and global-and-now-misnamed SCREEN_LOGDIR | 
 | 1021 | # _run_process service "command-line" | 
 | 1022 | function _run_process() { | 
 | 1023 |     local service=$1 | 
 | 1024 |     local command="$2" | 
 | 1025 |  | 
 | 1026 |     # Undo logging redirections and close the extra descriptors | 
 | 1027 |     exec 1>&3 | 
 | 1028 |     exec 2>&3 | 
 | 1029 |     exec 3>&- | 
 | 1030 |     exec 6>&- | 
 | 1031 |  | 
 | 1032 |     if [[ -n ${SCREEN_LOGDIR} ]]; then | 
 | 1033 |         exec 1>&${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log 2>&1 | 
 | 1034 |         ln -sf ${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log ${SCREEN_LOGDIR}/screen-${1}.log | 
 | 1035 |  | 
 | 1036 |         # TODO(dtroyer): Hack to get stdout from the Python interpreter for the logs. | 
 | 1037 |         export PYTHONUNBUFFERED=1 | 
 | 1038 |     fi | 
 | 1039 |  | 
 | 1040 |     exec /bin/bash -c "$command" | 
 | 1041 |     die "$service exec failure: $command" | 
 | 1042 | } | 
 | 1043 |  | 
 | 1044 |  | 
 | 1045 | # run_process() launches a child process that closes all file descriptors and | 
 | 1046 | # then exec's the passed in command.  This is meant to duplicate the semantics | 
 | 1047 | # of screen_it() without screen.  PIDs are written to | 
 | 1048 | # $SERVICE_DIR/$SCREEN_NAME/$service.pid | 
 | 1049 | # run_process service "command-line" | 
 | 1050 | function run_process() { | 
 | 1051 |     local service=$1 | 
 | 1052 |     local command="$2" | 
 | 1053 |  | 
 | 1054 |     # Spawn the child process | 
 | 1055 |     _run_process "$service" "$command" & | 
 | 1056 |     echo $! | 
 | 1057 | } | 
 | 1058 |  | 
 | 1059 |  | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1060 | # Helper to launch a service in a named screen | 
 | 1061 | # screen_it service "command-line" | 
 | 1062 | function screen_it { | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1063 |     SCREEN_NAME=${SCREEN_NAME:-stack} | 
| jiajun xu | a941424 | 2012-12-06 16:30:57 +0800 | [diff] [blame] | 1064 |     SERVICE_DIR=${SERVICE_DIR:-${DEST}/status} | 
| Dean Troyer | 681f3fd | 2013-02-27 19:00:39 -0600 | [diff] [blame] | 1065 |     USE_SCREEN=$(trueorfalse True $USE_SCREEN) | 
| jiajun xu | a941424 | 2012-12-06 16:30:57 +0800 | [diff] [blame] | 1066 |  | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1067 |     if is_service_enabled $1; then | 
 | 1068 |         # Append the service to the screen rc file | 
 | 1069 |         screen_rc "$1" "$2" | 
 | 1070 |  | 
| Dean Troyer | 681f3fd | 2013-02-27 19:00:39 -0600 | [diff] [blame] | 1071 |         if [[ "$USE_SCREEN" = "True" ]]; then | 
 | 1072 |             screen -S $SCREEN_NAME -X screen -t $1 | 
| Jeremy Stanley | 25ebbcd | 2013-02-17 15:45:55 +0000 | [diff] [blame] | 1073 |  | 
| Dean Troyer | 681f3fd | 2013-02-27 19:00:39 -0600 | [diff] [blame] | 1074 |             if [[ -n ${SCREEN_LOGDIR} ]]; then | 
 | 1075 |                 screen -S $SCREEN_NAME -p $1 -X logfile ${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log | 
 | 1076 |                 screen -S $SCREEN_NAME -p $1 -X log on | 
 | 1077 |                 ln -sf ${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log ${SCREEN_LOGDIR}/screen-${1}.log | 
 | 1078 |             fi | 
| Jeremy Stanley | 25ebbcd | 2013-02-17 15:45:55 +0000 | [diff] [blame] | 1079 |  | 
| Vishvananda Ishaya | 58e2134 | 2013-02-11 16:48:12 -0800 | [diff] [blame] | 1080 |             # sleep to allow bash to be ready to be send the command - we are | 
 | 1081 |             # creating a new window in screen and then sends characters, so if | 
 | 1082 |             # bash isn't running by the time we send the command, nothing happens | 
 | 1083 |             sleep 1.5 | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1084 |  | 
| Vishvananda Ishaya | 58e2134 | 2013-02-11 16:48:12 -0800 | [diff] [blame] | 1085 |             NL=`echo -ne '\015'` | 
| Clark Boylan | 41815cd | 2013-08-16 14:57:38 -0700 | [diff] [blame] | 1086 |             screen -S $SCREEN_NAME -p $1 -X stuff "$2 || echo \"$1 failed to start\" | tee \"$SERVICE_DIR/$SCREEN_NAME/$1.failure\"$NL" | 
| Vishvananda Ishaya | 58e2134 | 2013-02-11 16:48:12 -0800 | [diff] [blame] | 1087 |         else | 
| Dean Troyer | 681f3fd | 2013-02-27 19:00:39 -0600 | [diff] [blame] | 1088 |             # Spawn directly without screen | 
 | 1089 |             run_process "$1" "$2" >$SERVICE_DIR/$SCREEN_NAME/$service.pid | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1090 |         fi | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1091 |     fi | 
 | 1092 | } | 
 | 1093 |  | 
 | 1094 |  | 
 | 1095 | # Screen rc file builder | 
 | 1096 | # screen_rc service "command-line" | 
 | 1097 | function screen_rc { | 
 | 1098 |     SCREEN_NAME=${SCREEN_NAME:-stack} | 
 | 1099 |     SCREENRC=$TOP_DIR/$SCREEN_NAME-screenrc | 
 | 1100 |     if [[ ! -e $SCREENRC ]]; then | 
 | 1101 |         # Name the screen session | 
 | 1102 |         echo "sessionname $SCREEN_NAME" > $SCREENRC | 
 | 1103 |         # Set a reasonable statusbar | 
 | 1104 |         echo "hardstatus alwayslastline '$SCREEN_HARDSTATUS'" >> $SCREENRC | 
| Steven Dake | 3039657 | 2013-06-30 16:11:54 -0700 | [diff] [blame] | 1105 |         # Some distributions override PROMPT_COMMAND for the screen terminal type - turn that off | 
 | 1106 |         echo "setenv PROMPT_COMMAND /bin/true" >> $SCREENRC | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1107 |         echo "screen -t shell bash" >> $SCREENRC | 
 | 1108 |     fi | 
 | 1109 |     # If this service doesn't already exist in the screenrc file | 
 | 1110 |     if ! grep $1 $SCREENRC 2>&1 > /dev/null; then | 
 | 1111 |         NL=`echo -ne '\015'` | 
 | 1112 |         echo "screen -t $1 bash" >> $SCREENRC | 
 | 1113 |         echo "stuff \"$2$NL\"" >> $SCREENRC | 
 | 1114 |     fi | 
 | 1115 | } | 
 | 1116 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1117 |  | 
| jiajun xu | a941424 | 2012-12-06 16:30:57 +0800 | [diff] [blame] | 1118 | # Helper to remove the *.failure files under $SERVICE_DIR/$SCREEN_NAME | 
 | 1119 | # This is used for service_check when all the screen_it are called finished | 
 | 1120 | # init_service_check | 
 | 1121 | function init_service_check() { | 
 | 1122 |     SCREEN_NAME=${SCREEN_NAME:-stack} | 
 | 1123 |     SERVICE_DIR=${SERVICE_DIR:-${DEST}/status} | 
 | 1124 |  | 
 | 1125 |     if [[ ! -d "$SERVICE_DIR/$SCREEN_NAME" ]]; then | 
 | 1126 |         mkdir -p "$SERVICE_DIR/$SCREEN_NAME" | 
 | 1127 |     fi | 
 | 1128 |  | 
 | 1129 |     rm -f "$SERVICE_DIR/$SCREEN_NAME"/*.failure | 
 | 1130 | } | 
 | 1131 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1132 |  | 
| jiajun xu | a941424 | 2012-12-06 16:30:57 +0800 | [diff] [blame] | 1133 | # Helper to get the status of each running service | 
 | 1134 | # service_check | 
 | 1135 | function service_check() { | 
 | 1136 |     local service | 
 | 1137 |     local failures | 
 | 1138 |     SCREEN_NAME=${SCREEN_NAME:-stack} | 
 | 1139 |     SERVICE_DIR=${SERVICE_DIR:-${DEST}/status} | 
 | 1140 |  | 
 | 1141 |  | 
 | 1142 |     if [[ ! -d "$SERVICE_DIR/$SCREEN_NAME" ]]; then | 
 | 1143 |         echo "No service status directory found" | 
 | 1144 |         return | 
 | 1145 |     fi | 
 | 1146 |  | 
 | 1147 |     # Check if there is any falure flag file under $SERVICE_DIR/$SCREEN_NAME | 
 | 1148 |     failures=`ls "$SERVICE_DIR/$SCREEN_NAME"/*.failure 2>/dev/null` | 
 | 1149 |  | 
 | 1150 |     for service in $failures; do | 
 | 1151 |         service=`basename $service` | 
| Bob Ball | 46287d8 | 2013-07-30 09:43:17 +0100 | [diff] [blame] | 1152 |         service=${service%.failure} | 
| jiajun xu | a941424 | 2012-12-06 16:30:57 +0800 | [diff] [blame] | 1153 |         echo "Error: Service $service is not running" | 
 | 1154 |     done | 
 | 1155 |  | 
 | 1156 |     if [ -n "$failures" ]; then | 
 | 1157 |         echo "More details about the above errors can be found with screen, with ./rejoin-stack.sh" | 
 | 1158 |     fi | 
 | 1159 | } | 
| Dean Troyer | 1573335 | 2012-09-06 11:51:30 -0500 | [diff] [blame] | 1160 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1161 |  | 
| Monty Taylor | 408a4a7 | 2013-08-02 15:43:47 -0400 | [diff] [blame] | 1162 | # ``pip install -e`` the package, which processes the dependencies | 
 | 1163 | # using pip before running `setup.py develop` | 
| Monty Taylor | b5bbaac | 2013-08-06 10:35:02 -0300 | [diff] [blame] | 1164 | # Uses globals ``STACK_USER``, ``TRACK_DEPENDS``, ``REQUIREMENTS_DIR`` | 
| Dean Troyer | bbafb1b | 2012-06-11 16:51:39 -0500 | [diff] [blame] | 1165 | # setup_develop directory | 
 | 1166 | function setup_develop() { | 
| Sean Dague | 6c84463 | 2013-07-31 06:50:14 -0400 | [diff] [blame] | 1167 |     local project_dir=$1 | 
| Dean Troyer | cc6b443 | 2013-04-08 15:38:03 -0500 | [diff] [blame] | 1168 |     if [[ $TRACK_DEPENDS = True ]]; then | 
| Monty Taylor | 47f0206 | 2012-07-26 11:09:24 -0500 | [diff] [blame] | 1169 |         SUDO_CMD="env" | 
 | 1170 |     else | 
 | 1171 |         SUDO_CMD="sudo" | 
 | 1172 |     fi | 
| Sean Dague | 6c84463 | 2013-07-31 06:50:14 -0400 | [diff] [blame] | 1173 |  | 
 | 1174 |     echo "cd $REQUIREMENTS_DIR; $SUDO_CMD python update.py $project_dir" | 
 | 1175 |  | 
| Dean Troyer | 62d1d69 | 2013-08-01 17:40:40 -0500 | [diff] [blame] | 1176 |     # Don't update repo if local changes exist | 
 | 1177 |     if (cd $project_dir && git diff --quiet); then | 
 | 1178 |         (cd $REQUIREMENTS_DIR; \ | 
 | 1179 |             $SUDO_CMD python update.py $project_dir) | 
 | 1180 |     fi | 
| Sean Dague | 6c84463 | 2013-07-31 06:50:14 -0400 | [diff] [blame] | 1181 |  | 
| Monty Taylor | b5bbaac | 2013-08-06 10:35:02 -0300 | [diff] [blame] | 1182 |     pip_install -e $project_dir | 
 | 1183 |     # ensure that further actions can do things like setup.py sdist | 
 | 1184 |     $SUDO_CMD chown -R $STACK_USER $1/*.egg-info | 
| Dean Troyer | bbafb1b | 2012-06-11 16:51:39 -0500 | [diff] [blame] | 1185 | } | 
 | 1186 |  | 
 | 1187 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1188 | # Service wrapper to start services | 
 | 1189 | # start_service service-name | 
 | 1190 | function start_service() { | 
| Vincent Untz | c18b965 | 2012-12-04 12:36:34 +0100 | [diff] [blame] | 1191 |     if is_ubuntu; then | 
| Dean Troyer | 5218d45 | 2012-02-04 02:13:23 -0600 | [diff] [blame] | 1192 |         sudo /usr/sbin/service $1 start | 
 | 1193 |     else | 
 | 1194 |         sudo /sbin/service $1 start | 
 | 1195 |     fi | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1196 | } | 
 | 1197 |  | 
 | 1198 |  | 
 | 1199 | # Service wrapper to stop services | 
 | 1200 | # stop_service service-name | 
 | 1201 | function stop_service() { | 
| Vincent Untz | c18b965 | 2012-12-04 12:36:34 +0100 | [diff] [blame] | 1202 |     if is_ubuntu; then | 
| Dean Troyer | 5218d45 | 2012-02-04 02:13:23 -0600 | [diff] [blame] | 1203 |         sudo /usr/sbin/service $1 stop | 
 | 1204 |     else | 
 | 1205 |         sudo /sbin/service $1 stop | 
 | 1206 |     fi | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 1207 | } | 
 | 1208 |  | 
 | 1209 |  | 
 | 1210 | # Normalize config values to True or False | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 1211 | # Accepts as False: 0 no false False FALSE | 
 | 1212 | # Accepts as True: 1 yes true True TRUE | 
 | 1213 | # VAR=$(trueorfalse default-value test-value) | 
| Dean Troyer | 7f9aa71 | 2012-01-31 12:11:56 -0600 | [diff] [blame] | 1214 | function trueorfalse() { | 
 | 1215 |     local default=$1 | 
 | 1216 |     local testval=$2 | 
 | 1217 |  | 
 | 1218 |     [[ -z "$testval" ]] && { echo "$default"; return; } | 
 | 1219 |     [[ "0 no false False FALSE" =~ "$testval" ]] && { echo "False"; return; } | 
 | 1220 |     [[ "1 yes true True TRUE" =~ "$testval" ]] && { echo "True"; return; } | 
 | 1221 |     echo "$default" | 
 | 1222 | } | 
| Dean Troyer | 27e3269 | 2012-03-16 16:16:56 -0500 | [diff] [blame] | 1223 |  | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1224 |  | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1225 | # Retrieve an image from a URL and upload into Glance | 
 | 1226 | # Uses the following variables: | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 1227 | #   ``FILES`` must be set to the cache dir | 
 | 1228 | #   ``GLANCE_HOSTPORT`` | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1229 | # upload_image image-url glance-token | 
 | 1230 | function upload_image() { | 
 | 1231 |     local image_url=$1 | 
 | 1232 |     local token=$2 | 
 | 1233 |  | 
 | 1234 |     # Create a directory for the downloaded image tarballs. | 
 | 1235 |     mkdir -p $FILES/images | 
 | 1236 |  | 
 | 1237 |     # Downloads the image (uec ami+aki style), then extracts it. | 
 | 1238 |     IMAGE_FNAME=`basename "$image_url"` | 
 | 1239 |     if [[ ! -f $FILES/$IMAGE_FNAME || "$(stat -c "%s" $FILES/$IMAGE_FNAME)" = "0" ]]; then | 
 | 1240 |         wget -c $image_url -O $FILES/$IMAGE_FNAME | 
 | 1241 |         if [[ $? -ne 0 ]]; then | 
 | 1242 |             echo "Not found: $image_url" | 
 | 1243 |             return | 
 | 1244 |         fi | 
 | 1245 |     fi | 
 | 1246 |  | 
 | 1247 |     # OpenVZ-format images are provided as .tar.gz, but not decompressed prior to loading | 
 | 1248 |     if [[ "$image_url" =~ 'openvz' ]]; then | 
 | 1249 |         IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1250 |         IMAGE_NAME="${IMAGE_FNAME%.tar.gz}" | 
 | 1251 |         glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME" --is-public=True --container-format ami --disk-format ami < "${IMAGE}" | 
 | 1252 |         return | 
 | 1253 |     fi | 
 | 1254 |  | 
| Sreeram Yerrapragada | cbaff86 | 2013-07-24 19:49:23 -0700 | [diff] [blame] | 1255 |     # vmdk format images | 
 | 1256 |     if [[ "$image_url" =~ '.vmdk' ]]; then | 
 | 1257 |         IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1258 |         IMAGE_NAME="${IMAGE_FNAME%.vmdk}" | 
| Ryan Hsu | a6273b9 | 2013-09-04 23:51:29 -0700 | [diff] [blame^] | 1259 |  | 
 | 1260 |         # Before we can upload vmdk type images to glance, we need to know it's | 
 | 1261 |         # disk type, storage adapter, and networking adapter. These values are | 
 | 1262 |         # passed to glance as custom properties. We take these values from the | 
 | 1263 |         # vmdk filename, which is expected in the following format: | 
 | 1264 |         # | 
 | 1265 |         #     <name>-<disk type>:<storage adapter>:<network adapter> | 
 | 1266 |         # | 
 | 1267 |         # If the filename does not follow the above format then the vsphere | 
 | 1268 |         # driver will supply default values. | 
 | 1269 |         property_string=`echo "$IMAGE_NAME" | grep -oP '(?<=-)(?!.*-).+:.+:.+$'` | 
 | 1270 |         if [[ ! -z "$property_string" ]]; then | 
 | 1271 |             IFS=':' read -a props <<< "$property_string" | 
 | 1272 |             vmdk_disktype="${props[0]}" | 
 | 1273 |             vmdk_adapter_type="${props[1]}" | 
 | 1274 |             vmdk_net_adapter="${props[2]}" | 
 | 1275 |         fi | 
 | 1276 |  | 
 | 1277 |         glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME" --is-public=True --container-format bare --disk-format vmdk --property vmware-disktype="$vmdk_disktype" --property vmware_adaptertype="$vmdk_adapter_type" --property hw_vif_model="$vmdk_net_adapter" < "${IMAGE}" | 
| Sreeram Yerrapragada | cbaff86 | 2013-07-24 19:49:23 -0700 | [diff] [blame] | 1278 |         return | 
 | 1279 |     fi | 
 | 1280 |  | 
| Mate Lakat | bc2ef92 | 2013-08-15 18:06:59 +0100 | [diff] [blame] | 1281 |     # XenServer-vhd-ovf-format images are provided as .vhd.tgz | 
| Davanum Srinivas | 316ed6c | 2013-02-06 15:29:49 -0500 | [diff] [blame] | 1282 |     # and should not be decompressed prior to loading | 
 | 1283 |     if [[ "$image_url" =~ '.vhd.tgz' ]]; then | 
 | 1284 |         IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1285 |         IMAGE_NAME="${IMAGE_FNAME%.vhd.tgz}" | 
 | 1286 |         glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME" --is-public=True --container-format=ovf --disk-format=vhd < "${IMAGE}" | 
 | 1287 |         return | 
 | 1288 |     fi | 
 | 1289 |  | 
| Mate Lakat | bc2ef92 | 2013-08-15 18:06:59 +0100 | [diff] [blame] | 1290 |     # .xen-raw.tgz suggests a Xen capable raw image inside a tgz. | 
 | 1291 |     # and should not be decompressed prior to loading. | 
 | 1292 |     # Setting metadata, so PV mode is used. | 
 | 1293 |     if [[ "$image_url" =~ '.xen-raw.tgz' ]]; then | 
 | 1294 |         IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1295 |         IMAGE_NAME="${IMAGE_FNAME%.xen-raw.tgz}" | 
 | 1296 |         glance \ | 
 | 1297 |           --os-auth-token $token \ | 
 | 1298 |           --os-image-url http://$GLANCE_HOSTPORT \ | 
 | 1299 |           image-create \ | 
 | 1300 |             --name "$IMAGE_NAME" --is-public=True \ | 
 | 1301 |             --container-format=tgz --disk-format=raw \ | 
 | 1302 |             --property vm_mode=xen < "${IMAGE}" | 
 | 1303 |         return | 
 | 1304 |     fi | 
 | 1305 |  | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1306 |     KERNEL="" | 
 | 1307 |     RAMDISK="" | 
 | 1308 |     DISK_FORMAT="" | 
 | 1309 |     CONTAINER_FORMAT="" | 
 | 1310 |     UNPACK="" | 
 | 1311 |     case "$IMAGE_FNAME" in | 
 | 1312 |         *.tar.gz|*.tgz) | 
 | 1313 |             # Extract ami and aki files | 
 | 1314 |             [ "${IMAGE_FNAME%.tar.gz}" != "$IMAGE_FNAME" ] && | 
 | 1315 |                 IMAGE_NAME="${IMAGE_FNAME%.tar.gz}" || | 
 | 1316 |                 IMAGE_NAME="${IMAGE_FNAME%.tgz}" | 
 | 1317 |             xdir="$FILES/images/$IMAGE_NAME" | 
 | 1318 |             rm -Rf "$xdir"; | 
 | 1319 |             mkdir "$xdir" | 
 | 1320 |             tar -zxf $FILES/$IMAGE_FNAME -C "$xdir" | 
 | 1321 |             KERNEL=$(for f in "$xdir/"*-vmlinuz* "$xdir/"aki-*/image; do | 
 | 1322 |                      [ -f "$f" ] && echo "$f" && break; done; true) | 
 | 1323 |             RAMDISK=$(for f in "$xdir/"*-initrd* "$xdir/"ari-*/image; do | 
 | 1324 |                      [ -f "$f" ] && echo "$f" && break; done; true) | 
 | 1325 |             IMAGE=$(for f in "$xdir/"*.img "$xdir/"ami-*/image; do | 
 | 1326 |                      [ -f "$f" ] && echo "$f" && break; done; true) | 
 | 1327 |             if [[ -z "$IMAGE_NAME" ]]; then | 
 | 1328 |                 IMAGE_NAME=$(basename "$IMAGE" ".img") | 
 | 1329 |             fi | 
 | 1330 |             ;; | 
 | 1331 |         *.img) | 
 | 1332 |             IMAGE="$FILES/$IMAGE_FNAME"; | 
 | 1333 |             IMAGE_NAME=$(basename "$IMAGE" ".img") | 
| Dean Troyer | 636a3ff | 2012-09-14 11:36:07 -0500 | [diff] [blame] | 1334 |             format=$(qemu-img info ${IMAGE} | awk '/^file format/ { print $3; exit }') | 
 | 1335 |             if [[ ",qcow2,raw,vdi,vmdk,vpc," =~ ",$format," ]]; then | 
 | 1336 |                 DISK_FORMAT=$format | 
 | 1337 |             else | 
 | 1338 |                 DISK_FORMAT=raw | 
 | 1339 |             fi | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1340 |             CONTAINER_FORMAT=bare | 
 | 1341 |             ;; | 
 | 1342 |         *.img.gz) | 
 | 1343 |             IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1344 |             IMAGE_NAME=$(basename "$IMAGE" ".img.gz") | 
 | 1345 |             DISK_FORMAT=raw | 
 | 1346 |             CONTAINER_FORMAT=bare | 
 | 1347 |             UNPACK=zcat | 
 | 1348 |             ;; | 
 | 1349 |         *.qcow2) | 
 | 1350 |             IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1351 |             IMAGE_NAME=$(basename "$IMAGE" ".qcow2") | 
 | 1352 |             DISK_FORMAT=qcow2 | 
 | 1353 |             CONTAINER_FORMAT=bare | 
 | 1354 |             ;; | 
| Jonathan Michalon | 0680204 | 2013-03-21 14:29:58 +0100 | [diff] [blame] | 1355 |         *.iso) | 
 | 1356 |             IMAGE="$FILES/${IMAGE_FNAME}" | 
 | 1357 |             IMAGE_NAME=$(basename "$IMAGE" ".iso") | 
 | 1358 |             DISK_FORMAT=iso | 
 | 1359 |             CONTAINER_FORMAT=bare | 
 | 1360 |             ;; | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1361 |         *) echo "Do not know what to do with $IMAGE_FNAME"; false;; | 
 | 1362 |     esac | 
 | 1363 |  | 
 | 1364 |     if [ "$CONTAINER_FORMAT" = "bare" ]; then | 
 | 1365 |         if [ "$UNPACK" = "zcat" ]; then | 
| Christian Berendt | a7a219a | 2013-07-30 18:22:32 +0200 | [diff] [blame] | 1366 |             glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME" --is-public True --container-format=$CONTAINER_FORMAT --disk-format $DISK_FORMAT < <(zcat --force "${IMAGE}") | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1367 |         else | 
| Christian Berendt | a7a219a | 2013-07-30 18:22:32 +0200 | [diff] [blame] | 1368 |             glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME" --is-public True --container-format=$CONTAINER_FORMAT --disk-format $DISK_FORMAT < "${IMAGE}" | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1369 |         fi | 
 | 1370 |     else | 
 | 1371 |         # Use glance client to add the kernel the root filesystem. | 
 | 1372 |         # We parse the results of the first upload to get the glance ID of the | 
 | 1373 |         # kernel for use when uploading the root filesystem. | 
 | 1374 |         KERNEL_ID=""; RAMDISK_ID=""; | 
 | 1375 |         if [ -n "$KERNEL" ]; then | 
| Christian Berendt | a7a219a | 2013-07-30 18:22:32 +0200 | [diff] [blame] | 1376 |             KERNEL_ID=$(glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME-kernel" --is-public True --container-format aki --disk-format aki < "$KERNEL" | grep ' id ' | get_field 2) | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1377 |         fi | 
 | 1378 |         if [ -n "$RAMDISK" ]; then | 
| Christian Berendt | a7a219a | 2013-07-30 18:22:32 +0200 | [diff] [blame] | 1379 |             RAMDISK_ID=$(glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "$IMAGE_NAME-ramdisk" --is-public True --container-format ari --disk-format ari < "$RAMDISK" | grep ' id ' | get_field 2) | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1380 |         fi | 
| Christian Berendt | a7a219a | 2013-07-30 18:22:32 +0200 | [diff] [blame] | 1381 |         glance --os-auth-token $token --os-image-url http://$GLANCE_HOSTPORT image-create --name "${IMAGE_NAME%.img}" --is-public True --container-format ami --disk-format ami ${KERNEL_ID:+--property kernel_id=$KERNEL_ID} ${RAMDISK_ID:+--property ramdisk_id=$RAMDISK_ID} < "${IMAGE}" | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1382 |     fi | 
 | 1383 | } | 
 | 1384 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1385 |  | 
| Dean Troyer | c1b486a | 2012-11-05 14:26:09 -0600 | [diff] [blame] | 1386 | # Set the database backend to use | 
 | 1387 | # When called from stackrc/localrc DATABASE_BACKENDS has not been | 
 | 1388 | # initialized yet, just save the configuration selection and call back later | 
 | 1389 | # to validate it. | 
 | 1390 | #  $1 The name of the database backend to use (mysql, postgresql, ...) | 
 | 1391 | function use_database { | 
 | 1392 |     if [[ -z "$DATABASE_BACKENDS" ]]; then | 
| Dean Troyer | afc29fe | 2013-02-07 15:56:24 -0600 | [diff] [blame] | 1393 |         # No backends registered means this is likely called from ``localrc`` | 
 | 1394 |         # This is now deprecated usage | 
| Dean Troyer | c1b486a | 2012-11-05 14:26:09 -0600 | [diff] [blame] | 1395 |         DATABASE_TYPE=$1 | 
| Bob Ball | 3aa8887 | 2013-02-28 17:39:41 +0000 | [diff] [blame] | 1396 |         DEPRECATED_TEXT="$DEPRECATED_TEXT\nThe database backend needs to be properly set in ENABLED_SERVICES; use_database is deprecated localrc\n" | 
| Attila Fazekas | 251d3b5 | 2012-12-16 15:05:44 +0100 | [diff] [blame] | 1397 |     else | 
| Dean Troyer | afc29fe | 2013-02-07 15:56:24 -0600 | [diff] [blame] | 1398 |         # This should no longer get called...here for posterity | 
| Attila Fazekas | 251d3b5 | 2012-12-16 15:05:44 +0100 | [diff] [blame] | 1399 |         use_exclusive_service DATABASE_BACKENDS DATABASE_TYPE $1 | 
| Dean Troyer | c1b486a | 2012-11-05 14:26:09 -0600 | [diff] [blame] | 1400 |     fi | 
| Dean Troyer | c1b486a | 2012-11-05 14:26:09 -0600 | [diff] [blame] | 1401 | } | 
 | 1402 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1403 |  | 
| Terry Wilson | 428af5a | 2012-11-01 16:12:39 -0400 | [diff] [blame] | 1404 | # Toggle enable/disable_service for services that must run exclusive of each other | 
 | 1405 | #  $1 The name of a variable containing a space-separated list of services | 
 | 1406 | #  $2 The name of a variable in which to store the enabled service's name | 
 | 1407 | #  $3 The name of the service to enable | 
 | 1408 | function use_exclusive_service { | 
 | 1409 |     local options=${!1} | 
 | 1410 |     local selection=$3 | 
 | 1411 |     out=$2 | 
 | 1412 |     [ -z $selection ] || [[ ! "$options" =~ "$selection" ]] && return 1 | 
 | 1413 |     for opt in $options;do | 
 | 1414 |         [[ "$opt" = "$selection" ]] && enable_service $opt || disable_service $opt | 
 | 1415 |     done | 
 | 1416 |     eval "$out=$selection" | 
 | 1417 |     return 0 | 
 | 1418 | } | 
| Dean Troyer | ca0e3d0 | 2012-04-13 15:58:37 -0500 | [diff] [blame] | 1419 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1420 |  | 
| Dean Troyer | 3a3a2ba | 2012-12-11 15:26:24 -0600 | [diff] [blame] | 1421 | # Wait for an HTTP server to start answering requests | 
 | 1422 | # wait_for_service timeout url | 
 | 1423 | function wait_for_service() { | 
 | 1424 |     local timeout=$1 | 
 | 1425 |     local url=$2 | 
 | 1426 |     timeout $timeout sh -c "while ! http_proxy= https_proxy= curl -s $url >/dev/null; do sleep 1; done" | 
 | 1427 | } | 
 | 1428 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1429 |  | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 1430 | # Wrapper for ``yum`` to set proxy environment variables | 
 | 1431 | # Uses globals ``OFFLINE``, ``*_proxy` | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1432 | # yum_install package [package ...] | 
 | 1433 | function yum_install() { | 
 | 1434 |     [[ "$OFFLINE" = "True" ]] && return | 
 | 1435 |     local sudo="sudo" | 
 | 1436 |     [[ "$(id -u)" = "0" ]] && sudo="env" | 
 | 1437 |     $sudo http_proxy=$http_proxy https_proxy=$https_proxy \ | 
| Osamu Habuka | 7abe4f2 | 2012-07-25 12:39:32 +0900 | [diff] [blame] | 1438 |         no_proxy=$no_proxy \ | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1439 |         yum install -y "$@" | 
 | 1440 | } | 
 | 1441 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1442 |  | 
 | 1443 | # zypper wrapper to set arguments correctly | 
 | 1444 | # zypper_install package [package ...] | 
 | 1445 | function zypper_install() { | 
 | 1446 |     [[ "$OFFLINE" = "True" ]] && return | 
 | 1447 |     local sudo="sudo" | 
 | 1448 |     [[ "$(id -u)" = "0" ]] && sudo="env" | 
 | 1449 |     $sudo http_proxy=$http_proxy https_proxy=$https_proxy \ | 
 | 1450 |         zypper --non-interactive install --auto-agree-with-licenses "$@" | 
 | 1451 | } | 
 | 1452 |  | 
 | 1453 |  | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1454 | # ping check | 
 | 1455 | # Uses globals ``ENABLED_SERVICES`` | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1456 | # ping_check from-net ip boot-timeout expected | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1457 | function ping_check() { | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 1458 |     if is_service_enabled neutron; then | 
 | 1459 |         _ping_check_neutron  "$1" $2 $3 $4 | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1460 |         return | 
 | 1461 |     fi | 
 | 1462 |     _ping_check_novanet "$1" $2 $3 $4 | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1463 | } | 
 | 1464 |  | 
 | 1465 | # ping check for nova | 
 | 1466 | # Uses globals ``MULTI_HOST``, ``PRIVATE_NETWORK`` | 
 | 1467 | function _ping_check_novanet() { | 
 | 1468 |     local from_net=$1 | 
 | 1469 |     local ip=$2 | 
 | 1470 |     local boot_timeout=$3 | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1471 |     local expected=${4:-"True"} | 
 | 1472 |     local check_command="" | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1473 |     MULTI_HOST=`trueorfalse False $MULTI_HOST` | 
 | 1474 |     if [[ "$MULTI_HOST" = "True" && "$from_net" = "$PRIVATE_NETWORK_NAME" ]]; then | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1475 |         return | 
 | 1476 |     fi | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1477 |     if [[ "$expected" = "True" ]]; then | 
 | 1478 |         check_command="while ! ping -c1 -w1 $ip; do sleep 1; done" | 
 | 1479 |     else | 
 | 1480 |         check_command="while ping -c1 -w1 $ip; do sleep 1; done" | 
 | 1481 |     fi | 
 | 1482 |     if ! timeout $boot_timeout sh -c "$check_command"; then | 
 | 1483 |         if [[ "$expected" = "True" ]]; then | 
| Nachi Ueno | 07115eb | 2013-02-26 12:38:18 -0800 | [diff] [blame] | 1484 |             die $LINENO "[Fail] Couldn't ping server" | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1485 |         else | 
| Nachi Ueno | 07115eb | 2013-02-26 12:38:18 -0800 | [diff] [blame] | 1486 |             die $LINENO "[Fail] Could ping server" | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1487 |         fi | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1488 |         exit 1 | 
 | 1489 |     fi | 
 | 1490 | } | 
 | 1491 |  | 
| Nachi Ueno | 6769b16 | 2013-08-12 18:18:56 -0700 | [diff] [blame] | 1492 | # Get ip of instance | 
 | 1493 | function get_instance_ip(){ | 
 | 1494 |     local vm_id=$1 | 
 | 1495 |     local network_name=$2 | 
 | 1496 |     local nova_result="$(nova show $vm_id)" | 
 | 1497 |     local ip=$(echo "$nova_result" | grep "$network_name" | get_field 2) | 
 | 1498 |     if [[ $ip = "" ]];then | 
 | 1499 |         echo "$nova_result" | 
 | 1500 |         die $LINENO "[Fail] Coudn't get ipaddress of VM" | 
 | 1501 |         exit 1 | 
 | 1502 |     fi | 
 | 1503 |     echo $ip | 
 | 1504 | } | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1505 |  | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1506 | # ssh check | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1507 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1508 | # ssh_check net-name key-file floating-ip default-user active-timeout | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1509 | function ssh_check() { | 
| Mark McClain | b05c876 | 2013-07-06 23:29:39 -0400 | [diff] [blame] | 1510 |     if is_service_enabled neutron; then | 
 | 1511 |         _ssh_check_neutron  "$1" $2 $3 $4 $5 | 
| Nachi Ueno | 5db5bfa | 2012-10-29 11:25:29 -0700 | [diff] [blame] | 1512 |         return | 
 | 1513 |     fi | 
 | 1514 |     _ssh_check_novanet "$1" $2 $3 $4 $5 | 
 | 1515 | } | 
 | 1516 |  | 
 | 1517 | function _ssh_check_novanet() { | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1518 |     local NET_NAME=$1 | 
 | 1519 |     local KEY_FILE=$2 | 
 | 1520 |     local FLOATING_IP=$3 | 
 | 1521 |     local DEFAULT_INSTANCE_USER=$4 | 
 | 1522 |     local ACTIVE_TIMEOUT=$5 | 
| Dean Troyer | 6931c13 | 2012-11-07 16:51:21 -0600 | [diff] [blame] | 1523 |     local probe_cmd="" | 
| Dean Troyer | cc6b443 | 2013-04-08 15:38:03 -0500 | [diff] [blame] | 1524 |     if ! timeout $ACTIVE_TIMEOUT sh -c "while ! ssh -o StrictHostKeyChecking=no -i $KEY_FILE ${DEFAULT_INSTANCE_USER}@$FLOATING_IP echo success; do sleep 1; done"; then | 
| Nachi Ueno | 07115eb | 2013-02-26 12:38:18 -0800 | [diff] [blame] | 1525 |         die $LINENO "server didn't become ssh-able!" | 
| Nachi Ueno | fda946e | 2012-10-24 17:26:02 -0700 | [diff] [blame] | 1526 |     fi | 
 | 1527 | } | 
| Dean Troyer | 13dc5cc | 2012-03-27 14:50:45 -0500 | [diff] [blame] | 1528 |  | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 1529 |  | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 1530 | # Add a user to a group. | 
 | 1531 | # add_user_to_group user group | 
 | 1532 | function add_user_to_group() { | 
 | 1533 |     local user=$1 | 
 | 1534 |     local group=$2 | 
 | 1535 |  | 
 | 1536 |     if [[ -z "$os_VENDOR" ]]; then | 
 | 1537 |         GetOSVersion | 
 | 1538 |     fi | 
 | 1539 |  | 
 | 1540 |     # SLE11 and openSUSE 12.2 don't have the usual usermod | 
 | 1541 |     if ! is_suse || [[ "$os_VENDOR" = "openSUSE" && "$os_RELEASE" != "12.2" ]]; then | 
 | 1542 |         sudo usermod -a -G "$group" "$user" | 
 | 1543 |     else | 
 | 1544 |         sudo usermod -A "$group" "$user" | 
 | 1545 |     fi | 
 | 1546 | } | 
 | 1547 |  | 
 | 1548 |  | 
| Jakub Ruzicka | 4196d55 | 2013-01-30 15:35:54 +0100 | [diff] [blame] | 1549 | # Get the path to the direcotry where python executables are installed. | 
 | 1550 | # get_python_exec_prefix | 
 | 1551 | function get_python_exec_prefix() { | 
| Martin Vidner | 4f9b33d | 2013-06-27 13:11:22 +0000 | [diff] [blame] | 1552 |     if is_fedora || is_suse; then | 
| Jakub Ruzicka | 4196d55 | 2013-01-30 15:35:54 +0100 | [diff] [blame] | 1553 |         echo "/usr/bin" | 
 | 1554 |     else | 
 | 1555 |         echo "/usr/local/bin" | 
 | 1556 |     fi | 
 | 1557 | } | 
 | 1558 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1559 |  | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 1560 | # Get the location of the $module-rootwrap executables, where module is cinder | 
 | 1561 | # or nova. | 
 | 1562 | # get_rootwrap_location module | 
 | 1563 | function get_rootwrap_location() { | 
 | 1564 |     local module=$1 | 
 | 1565 |  | 
| Jakub Ruzicka | 4196d55 | 2013-01-30 15:35:54 +0100 | [diff] [blame] | 1566 |     echo "$(get_python_exec_prefix)/$module-rootwrap" | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 1567 | } | 
 | 1568 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1569 |  | 
| Vincent Untz | 8ec2722 | 2012-11-29 09:25:31 +0100 | [diff] [blame] | 1570 | # Get the path to the pip command. | 
 | 1571 | # get_pip_command | 
 | 1572 | function get_pip_command() { | 
| Dean Troyer | d2cfcaa | 2013-08-01 14:17:27 -0500 | [diff] [blame] | 1573 |     which pip || which pip-python | 
| Ian Wienand | 535a814 | 2013-05-15 09:25:27 +1000 | [diff] [blame] | 1574 |  | 
 | 1575 |     if [ $? -ne 0 ]; then | 
 | 1576 |         die $LINENO "Unable to find pip; cannot continue" | 
 | 1577 |     fi | 
| Vincent Untz | 8ec2722 | 2012-11-29 09:25:31 +0100 | [diff] [blame] | 1578 | } | 
| Vincent Untz | 856a11e | 2012-11-21 16:04:12 +0100 | [diff] [blame] | 1579 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1580 |  | 
| Ian Wienand | 0488edd | 2013-04-11 12:04:36 +1000 | [diff] [blame] | 1581 | # Path permissions sanity check | 
 | 1582 | # check_path_perm_sanity path | 
 | 1583 | function check_path_perm_sanity() { | 
 | 1584 |     # Ensure no element of the path has 0700 permissions, which is very | 
 | 1585 |     # likely to cause issues for daemons.  Inspired by default 0700 | 
 | 1586 |     # homedir permissions on RHEL and common practice of making DEST in | 
 | 1587 |     # the stack user's homedir. | 
 | 1588 |  | 
 | 1589 |     local real_path=$(readlink -f $1) | 
 | 1590 |     local rebuilt_path="" | 
 | 1591 |     for i in $(echo ${real_path} | tr "/" " "); do | 
 | 1592 |         rebuilt_path=$rebuilt_path"/"$i | 
 | 1593 |  | 
 | 1594 |         if [[ $(stat -c '%a' ${rebuilt_path}) = 700 ]]; then | 
 | 1595 |             echo "*** DEST path element" | 
 | 1596 |             echo "***    ${rebuilt_path}" | 
 | 1597 |             echo "*** appears to have 0700 permissions." | 
 | 1598 |             echo "*** This is very likely to cause fatal issues for devstack daemons." | 
 | 1599 |  | 
 | 1600 |             if [[ -n "$SKIP_PATH_SANITY" ]]; then | 
 | 1601 |                 return | 
 | 1602 |             else | 
 | 1603 |                 echo "*** Set SKIP_PATH_SANITY to skip this check" | 
 | 1604 |                 die $LINENO "Invalid path permissions" | 
 | 1605 |             fi | 
 | 1606 |         fi | 
 | 1607 |     done | 
 | 1608 | } | 
 | 1609 |  | 
| Dean Troyer | 1a6d449 | 2013-06-03 16:47:36 -0500 | [diff] [blame] | 1610 |  | 
| Kyle Mestery | 51a3f1f | 2013-06-13 11:47:56 +0000 | [diff] [blame] | 1611 | # This function recursively compares versions, and is not meant to be | 
 | 1612 | # called by anything other than vercmp_numbers below. This function does | 
 | 1613 | # not work with alphabetic versions. | 
 | 1614 | # | 
 | 1615 | # _vercmp_r sep ver1 ver2 | 
 | 1616 | function _vercmp_r { | 
 | 1617 |   typeset sep | 
 | 1618 |   typeset -a ver1=() ver2=() | 
 | 1619 |   sep=$1; shift | 
 | 1620 |   ver1=("${@:1:sep}") | 
 | 1621 |   ver2=("${@:sep+1}") | 
 | 1622 |  | 
 | 1623 |   if ((ver1 > ver2)); then | 
 | 1624 |     echo 1; return 0 | 
 | 1625 |   elif ((ver2 > ver1)); then | 
 | 1626 |     echo -1; return 0 | 
 | 1627 |   fi | 
 | 1628 |  | 
 | 1629 |   if ((sep <= 1)); then | 
 | 1630 |     echo 0; return 0 | 
 | 1631 |   fi | 
 | 1632 |  | 
 | 1633 |   _vercmp_r $((sep-1)) "${ver1[@]:1}" "${ver2[@]:1}" | 
 | 1634 | } | 
 | 1635 |  | 
 | 1636 |  | 
 | 1637 | # This function compares two versions and is meant to be called by | 
 | 1638 | # external callers. Please note the function assumes non-alphabetic | 
 | 1639 | # versions. For example, this will work: | 
 | 1640 | # | 
 | 1641 | #   vercmp_numbers 1.10 1.4 | 
 | 1642 | # | 
 | 1643 | # The above will return "1", as 1.10 is greater than 1.4. | 
 | 1644 | # | 
 | 1645 | #   vercmp_numbers 5.2 6.4 | 
 | 1646 | # | 
 | 1647 | # The above will return "-1", as 5.2 is less than 6.4. | 
 | 1648 | # | 
 | 1649 | #   vercmp_numbers 4.0 4.0 | 
 | 1650 | # | 
 | 1651 | # The above will return "0", as the versions are equal. | 
 | 1652 | # | 
 | 1653 | # vercmp_numbers ver1 ver2 | 
 | 1654 | vercmp_numbers() { | 
 | 1655 |   typeset v1=$1 v2=$2 sep | 
 | 1656 |   typeset -a ver1 ver2 | 
 | 1657 |  | 
 | 1658 |   IFS=. read -ra ver1 <<< "$v1" | 
 | 1659 |   IFS=. read -ra ver2 <<< "$v2" | 
 | 1660 |  | 
 | 1661 |   _vercmp_r "${#ver1[@]}" "${ver1[@]}" "${ver2[@]}" | 
 | 1662 | } | 
 | 1663 |  | 
 | 1664 |  | 
| Dean Troyer | 533e14d | 2013-08-30 15:11:22 -0500 | [diff] [blame] | 1665 | # ``policy_add policy_file policy_name policy_permissions`` | 
 | 1666 | # | 
 | 1667 | # Add a policy to a policy.json file | 
 | 1668 | # Do nothing if the policy already exists | 
 | 1669 |  | 
 | 1670 | function policy_add() { | 
 | 1671 |     local policy_file=$1 | 
 | 1672 |     local policy_name=$2 | 
 | 1673 |     local policy_perm=$3 | 
 | 1674 |  | 
 | 1675 |     if grep -q ${policy_name} ${policy_file}; then | 
 | 1676 |         echo "Policy ${policy_name} already exists in ${policy_file}" | 
 | 1677 |         return | 
 | 1678 |     fi | 
 | 1679 |  | 
 | 1680 |     # Add a terminating comma to policy lines without one | 
 | 1681 |     # Remove the closing '}' and all lines following to the end-of-file | 
 | 1682 |     local tmpfile=$(mktemp) | 
 | 1683 |     uniq ${policy_file} | sed -e ' | 
 | 1684 |         s/]$/],/ | 
 | 1685 |         /^[}]/,$d | 
 | 1686 |     ' > ${tmpfile} | 
 | 1687 |  | 
 | 1688 |     # Append policy and closing brace | 
 | 1689 |     echo "    \"${policy_name}\": ${policy_perm}" >>${tmpfile} | 
 | 1690 |     echo "}" >>${tmpfile} | 
 | 1691 |  | 
 | 1692 |     mv ${tmpfile} ${policy_file} | 
 | 1693 | } | 
 | 1694 |  | 
 | 1695 |  | 
| Dean Troyer | 27e3269 | 2012-03-16 16:16:56 -0500 | [diff] [blame] | 1696 | # Restore xtrace | 
| Chmouel Boudjnah | 408b009 | 2012-03-15 23:21:55 +0000 | [diff] [blame] | 1697 | $XTRACE | 
| Dean Troyer | 4a43b7b | 2012-08-28 17:43:40 -0500 | [diff] [blame] | 1698 |  | 
 | 1699 |  | 
 | 1700 | # Local variables: | 
| Sean Dague | 584d90e | 2013-03-29 14:34:53 -0400 | [diff] [blame] | 1701 | # mode: shell-script | 
| Andrew Laski | f900bd7 | 2012-09-05 17:23:14 -0400 | [diff] [blame] | 1702 | # End: |