lib/functions

   1 #!/bin/sh - keep it for file(1) to get bourne shell script result
   2 # functions     This file contains functions to be used by most or all
   3 #               shell scripts in the /etc/rc.d/init.d directory.
   4 #
   5 # $Id$
   6 #
   7 # Author:       Miquel van Smoorenburg, <miquels@drinkel.nl.mugnet.org>
   8 # Hacked by:    Greg Galloway and Marc Ewing
   9 # Modified for PLD Linux by:
  10 #               Marek Obuchowicz <elephant@pld-linux.org>
  11 #               Arkadiusz Miśkiewicz <misiek@pld-linux.org>
  12 #               Michał Kochanowicz <mkochano@pld-linux.org>
  13 #               Łukasz Pawelczyk <havner@pld-linux.org>
  14
  15 # First set up a default search path.
  16 export PATH="/sbin:/usr/sbin:/bin:/usr/bin"
  17
  18 # Set defaults
  19 if [ -z "$COLUMNS" -o -z "$LINES" ]; then
  20         _setterm() {
  21                 set -- $(stty size 2>/dev/null)
  22                 LINES=${LINES:-$1}
  23                 COLUMNS=${COLUMNS:-$2}
  24         }
  25         _setterm
  26         unset _setterm
  27 fi
  28 [ -z "$LINES" ] || [ "$LINES" -le 0 ] && LINES=40
  29 [ -z "$COLUMNS" ] || [ "$COLUMNS" -le 0 ] && COLUMNS=80
  30 export LINES COLUMNS
  31 INIT_COL=$((COLUMNS - 13))
  32
  33 # Set colors
  34 RED=1
  35 GREEN=2
  36 YELLOW=3
  37 BLUE=4
  38 MAGENTA=5
  39 CYAN=6
  40 WHITE=7
  41 NORMAL=15
  42 # Bold definition (second parameter to termput setaf)
  43 BOLD=1
  44 NOBOLD=0
  45 # Default colors
  46 CBRACKETS="$CYAN"       # brackets [ ] color
  47 CDONE="$GREEN"          # DONE and WORK color
  48 CBUSY="$MAGENTA"        # BUSY color
  49 CFAIL="$RED"            # FAIL and DIED color
  50 CPOWEREDBY="$CYAN"      # "Powered by" color
  51 CPLD="$GREEN"           # "PLD Linux Distribution" color
  52 CI="$RED"               # Capital I color (press I to enter interactive startup)
  53 CRESMAN="$GREEN"        # "Resource Manager" color
  54 CHARS=""                # Characters displayed on the beginning of show line
  55 CCHARS="$NORMAL"        # Color of these characters (look at /etc/sysconfig/init-colors.gentoo example)
  56
  57 # save from env, set by /sbin/service
  58 env_upstart=$USE_UPSTART
  59
  60 # Source configuration if available - may override default values
  61 [ -r /etc/sysconfig/init-colors ] && . /etc/sysconfig/init-colors
  62 [ -r /etc/sysconfig/system ] && . /etc/sysconfig/system
  63 [ -r /etc/sysconfig/bootsplash ] && . /etc/sysconfig/bootsplash
  64
  65 [ "$env_upstart" ] && USE_UPSTART=$env_upstart
  66
  67 if [ -z "$VSERVER" -o "$VSERVER" = "detect" ]; then
  68         {
  69                 while read _f _ctx; do
  70                         [ "$_f" = "VxID:" -o "$_f" = "s_context:" ] && break
  71                 done </proc/self/status
  72         } 2>/dev/null
  73         if [ -z "$_ctx" -o "$_ctx" = "0" ]; then
  74                 VSERVER=no
  75         else
  76                 VSERVER=yes
  77         fi
  78         unset _f _ctx
  79 fi
  80
  81 # VSERVER_ISOLATION_NET = isolation only inside of vserver guests
  82 if [ -z "$VSERVER_ISOLATION_NET" -o "$VSERVER_ISOLATION_NET" = "detect" ]; then
  83         VSERVER_ISOLATION_NET=no
  84         if [ "$VSERVER" = "yes" ]; then
  85                 if [ -f /proc/self/nsproxy ]; then
  86                         {
  87                                 while read _t _data; do
  88                                         [ "$_t" = "net:" ] && break
  89                                 done < /proc/self/nsproxy
  90                         } 2> /dev/null
  91                 else
  92                         # assume old kernel mode
  93                         VSERVER_ISOLATION_NET=yes
  94                 fi
  95                 if [ "${_data##*\(}" = "I)" ]; then
  96                         VSERVER_ISOLATION_NET=yes
  97                 fi
  98                 unset _f _data
  99         fi
 100 fi
 101
 102 # we need to know in functions if we were called from a terminal
 103 if [ -z "$ISATTY" ]; then
 104         [ -t ] && ISATTY=yes || ISATTY=no
 105         export ISATTY
 106 fi
 107
 108 is_yes() {
 109         # Test syntax
 110         if [ $# = 0 ]; then
 111                 msg_usage " is_yes {value}"
 112                 return 2
 113         fi
 114
 115         # Check value
 116         case "$1" in
 117         yes|Yes|YES|true|True|TRUE|on|On|ON|Y|y|1)
 118                 # true returns zero
 119                 return 0
 120                 ;;
 121         *)
 122                 # false returns one
 123                 return 1
 124                 ;;
 125         esac
 126 }
 127
 128 is_no() {
 129         # Test syntax
 130         if [ $# = 0 ]; then
 131                 msg_usage " is_no {value}"
 132                 return 2
 133         fi
 134
 135         case "$1" in
 136         no|No|NO|false|False|FALSE|off|Off|OFF|N|n|0)
 137                 # true returns zero
 138                 return 0
 139                 ;;
 140         *)
 141                 # false returns one
 142                 return 1
 143                 ;;
 144         esac
 145 }
 146
 147 # checks if file is empty
 148 # empty lines and lines beginning with hash are ignored
 149 is_empty_file() {
 150         [ -s "$1" ] || return 0
 151         grep -vqE "^(#|[[:blank:]]*$)" "$1" && return 1 || return 0
 152 }
 153
 154 # returns OK if $1 contains $2
 155 strstr() {
 156         local a=$2
 157         [ "${1#*$a*}" = "$1" ] && return 1
 158         return 0
 159 }
 160
 161 if is_yes "$FASTRC" || is_yes "$IN_SHUTDOWN"; then
 162         RC_LOGGING=no
 163 fi
 164
 165 if is_no "$RC_LOGGING"; then
 166         initlog() {
 167                 RESULT=0
 168                 while [ "$1" != "${1##-}" ]; do
 169                         case $1 in
 170                         -c)
 171                                 shift
 172                                 $1
 173                                 RESULT=$?
 174                                 break
 175                                 ;;
 176                         *)
 177                                 shift
 178                                 ;;
 179                         esac
 180                 done
 181                 return $RESULT
 182         }
 183 fi
 184
 185 kernelver() {
 186         local _x _y _z v v1 old_IFS ver
 187         {
 188                 read _x _y v _z
 189                 old_IFS=$IFS
 190                 # strip _* or -* from versions like: "2.6.25_vanilla-1", "2.6.25-1"
 191                 IFS='_-'
 192                 set -- $v
 193                 v1=${1}
 194                 IFS='.'
 195                 set -- $v1
 196                 IFS=$old_IFS
 197
 198                 ver=${3}
 199                 while [ ${#ver} -lt 3 ]; do ver="0$ver"; done
 200                 ver="$2$ver"
 201                 while [ ${#ver} -lt 6 ]; do ver="0$ver"; done
 202                 ver="$1$ver"
 203                 while [ ${#ver} -lt 9 ]; do ver="0$ver"; done
 204                 echo $ver
 205         } < /proc/version
 206 }
 207
 208 kernelverser() {
 209         local _x _y _z v v1 old_IFS ver
 210         {
 211                 read _x _y v _z
 212                 old_IFS=$IFS
 213                 # strip _* or -* from versions like: "2.6.25_vanilla-1", "2.6.25-1"
 214                 IFS='_-'
 215                 set -- $v
 216                 v1=${1}
 217                 IFS='.'
 218                 set -- $v1
 219                 IFS=$old_IFS
 220                 ver=$2
 221                 while [ ${#ver} -lt 3 ]; do ver="0$ver"; done
 222                 ver="$1$ver"
 223                 while [ ${#ver} -lt 6 ]; do ver="0$ver"; done
 224                 echo $ver
 225         } </proc/version
 226 }
 227
 228 kernelvermser() {
 229         local _x _y _z v v1 old_IFS ver
 230         {
 231                 read _x _y v _z
 232                 old_IFS=$IFS
 233                 # strip _* or -* from versions like: "2.6.25_vanilla-1", "2.6.25-1"
 234                 IFS='_-'
 235                 set -- $v
 236                 v1=${1}
 237                 IFS='.'
 238                 set -- $v1
 239                 IFS=$old_IFS
 240                 ver="$1"
 241                 while [ ${#ver} -lt 3 ]; do ver="0$ver"; done
 242                 echo $ver
 243         } </proc/version
 244 }
 245
 246 # Colors workaround
 247 termput() {
 248         is_yes "$ISATTY" || return
 249
 250         if is_yes "$FASTRC" || is_no "$TPUT"; then
 251                 case "$1" in
 252                 hpa)
 253                         echo -ne "\033[$(($2+1))G"
 254                         ;;
 255                 cuu*)
 256                         echo -ne "\033[${2}A"
 257                         ;;
 258                 el)
 259                         echo -ne "\033[0K"
 260                         ;;
 261                 setaf)
 262                         local ISBOLD
 263                         if [ -n "$3" ]; then
 264                                 ISBOLD="$3"
 265                         else
 266                                 ISBOLD="$NOBOLD";
 267                         fi
 268                         is_yes "$COLOR_INIT" && echo -ne "\033[${ISBOLD};3${2}m"
 269                         ;;
 270                 op)
 271                         termput setaf $NORMAL
 272                         ;;
 273                 esac
 274         else
 275                 case "$1" in
 276                 hpa | cuu* | el)
 277                         tput "$@"
 278                         ;;
 279                 setaf)
 280                         if [ "$3" = "1" ]; then tput bold; else tput sgr0; fi
 281                         is_yes "$COLOR_INIT" && tput setaf "$2"
 282                         ;;
 283                 op)
 284                         termput setaf $NORMAL
 285                         ;;
 286                 esac
 287         fi
 288 }
 289
 290 if [ ! -x /bin/printf ]; then
 291         # printf equivalent
 292         # FIXME: buggy when single or double quotes in message!
 293         printf() {
 294                 local text m
 295                 text="$1"
 296                 shift
 297                 if [ $# -gt 0 ]; then
 298                         m="$1"
 299                         shift
 300                         while [ $# -gt 0 ]; do
 301                                 m="$m\",\"$1"
 302                                 shift
 303                         done
 304                 fi
 305                 awk "BEGIN {printf \"$text\", \"$m\"; }"
 306         }
 307 fi
 308
 309 # National language support function
 310 nls() {
 311         local msg_echo nls_domain text message
 312         msg_echo='\n'
 313         nls_domain="$NLS_DOMAIN"
 314         while [ "$1" != "${1##-}" ]; do
 315                 case "$1" in
 316                 --nls-domain)
 317                         shift
 318                         nls_domain="$1"
 319                         shift
 320                         ;;
 321                 -n)
 322                         msg_echo=''
 323                         shift
 324                         ;;
 325                 esac
 326         done
 327         message="$1"
 328         shift
 329
 330         # empty message, so we return --misiek
 331         if [ -z "$message" ]; then
 332                 echo -en "$msg_echo"
 333                 return
 334         fi
 335
 336         if is_yes "$GETTEXT"; then
 337                 message=$(TEXTDOMAINDIR="/etc/sysconfig/locale" gettext -e --domain="${nls_domain:-rc-scripts}" "$message")
 338         fi
 339
 340         printf "$message" "$@"
 341         echo -en "$msg_echo"
 342 }
 343
 344 rc_splash() {
 345         local action="$1"
 346
 347         if ! is_no "$BOOT_SPLASH" && ! is_yes "$VSERVER"; then
 348                 [ -x /bin/splash ] && /bin/splash "$action"
 349         fi
 350
 351         : $((progress++))
 352 }
 353
 354 msg_network_down() {
 355         nls "ERROR: Networking is down. %s can't be run." "$1" >&2
 356 }
 357
 358 msg_starting() {
 359         show "Starting %s service" "$1"
 360 }
 361
 362 msg_already_running() {
 363         nls "%s service is already running." "$1"
 364 }
 365
 366 msg_stopping() {
 367         show "Stopping %s service" "$1"
 368 }
 369
 370 msg_not_running() {
 371         nls "%s service is not running." "$1"
 372 }
 373
 374 msg_reloading() {
 375         show "Reloading %s service" "$1"
 376 }
 377
 378 msg_usage() {
 379         nls "Usage: %s" "$*"
 380 }
 381
 382 # Some functions to handle PLD Linux-style messages
 383 show() {
 384         local text len
 385
 386         if is_no "$FASTRC" && is_yes "$GETTEXT"; then
 387                 text=$(nls -n "$@")
 388         else
 389                 text=$(printf "$@")
 390         fi
 391         len=${#text}
 392         while [ $((len++)) -lt $INIT_COL ]; do
 393                 text="$text."
 394         done
 395         if [ -n "$CHARS" ]; then
 396                 termput setaf $CCHARS
 397                 echo -n "$CHARS"
 398                 termput op
 399         fi
 400         echo -n "$text"
 401 }
 402
 403 deltext() {
 404         termput hpa $INIT_COL
 405 }
 406
 407 # Displays message in square brackests ("[ DONE ]"). Takes two arguments.
 408 # First is the text to display, second is color number to use (argument to
 409 # tput setaf). If second argument is not given, default (2, green) will be
 410 # used).
 411 progress() {
 412         local COLOR
 413         if [ -n "$2" ]; then
 414                 COLOR="$2"
 415         else
 416                 COLOR="$CDONE"
 417         fi
 418         deltext
 419         echo -n "$(termput setaf $CBRACKETS)[$(termput setaf $COLOR) $(nls --nls-domain rc-scripts "$1") $(termput setaf $CBRACKETS)]$(termput op)"
 420 }
 421
 422 busy() {
 423         echo -n "$_busy"
 424 }
 425
 426 ok() {
 427         echo "$_ok"
 428 }
 429
 430 started() {
 431         echo "$_started"
 432 }
 433
 434 fail() {
 435         echo "$_fail"
 436         return 1
 437 }
 438
 439 died() {
 440         echo "$_died"
 441         return 1
 442 }
 443
 444 # Check if $pid (could be plural) are running
 445 checkpid() {
 446         while [ "$1" ]; do
 447                 [ -d "/proc/$1" ] && return 0
 448                 shift
 449         done
 450         return 1
 451 }
 452
 453 # - outside chroot get only those processes, which are outside chroot.
 454 # - inside chroot get only those processes, which are inside chroot.
 455 # - don't filter out pids which do not have corresponding running processes (process died etc)
 456 # (note: some processes like named are chrooted but run outside chroot)
 457 # - do nothing inside vserver
 458 filter_chroot() {
 459         if is_yes "$VSERVER"; then
 460                 echo $@
 461                 return
 462         fi
 463         if [ $# -lt 1 -o ! -d /proc/1 ]; then
 464                 echo $@
 465                 return
 466         fi
 467         local root_dir good_pids="" good_add_pid
 468         for root_pid in $@; do
 469                 root_dir=$(resolvesymlink /proc/${root_pid}/root)
 470                 if [ -n "$root_dir" ]; then
 471                         good_add_pid=1
 472                         if [ -n "${SYSTEM_CHROOTS}" ]; then
 473                                 for r_dir in ${SYSTEM_CHROOTS}; do
 474                                         echo "$root_dir" | grep -q "^${r_dir}" && good_add_pid=0
 475                                 done
 476                         fi
 477                         [ "$good_add_pid" -eq 1 ] && good_pids="$good_pids $root_pid"
 478                 elif [ ! -d "/proc/$root_pid" ]; then
 479                         good_pids="$good_pids $root_pid"
 480                 fi
 481         done
 482         echo $good_pids
 483 }
 484
 485 # Usage:
 486 # run_cmd Message command_to_run
 487 # run_cmd -a Message command_to_run
 488 # run_cmd --user "username" "Message" command_to_run
 489 run_cmd() {
 490         local force_err=0 exit_code=0 errors user
 491         while [ $# -gt 0 ]; do
 492                 case "$1" in
 493                 -a)
 494                         force_err=1
 495                         ;;
 496                 --user)
 497                         shift
 498                         user=$1
 499                         ;;
 500                 *)
 501                         break
 502                 esac
 503                 shift
 504         done
 505
 506         local message=$1; shift
 507         show "$message"; busy
 508
 509         if errors=$(
 510                 cd /
 511                 export HOME=/tmp TMPDIR=/tmp
 512                 if is_no "$RC_LOGGING"; then
 513                         ${user:+setuidgid -s $user} "$@" 2>&1
 514                 else
 515                         ${user:+setuidgid -s $user} initlog -c "$*" 2>&1
 516                 fi
 517                 ); then
 518                 ok
 519                 log_success "$1 $message"
 520         else
 521                 fail
 522                 log_failed "$1 $message"
 523                 exit_code=1
 524         fi
 525         [ -n "$errors" ] && [ $exit_code -eq 1 -o $force_err -eq 1 ] && echo "$errors"
 526         return $exit_code
 527 }
 528
 529 _daemon_set_ulimits() {
 530         local opt val ksh=${KSH_VERSION:+1}
 531         set -- ${SERVICE_LIMITS:-$DEFAULT_SERVICE_LIMITS}
 532         while [ $# -gt 0 ]; do
 533                 opt=$1
 534                 val=$2
 535                 if [ "$ksh" ]; then
 536                         case "$opt" in
 537                         -Hu)
 538                                 opt=-Hp
 539                         ;;
 540                         -Su)
 541                                 opt=-Sp
 542                         ;;
 543                         -u)
 544                                 opt=-p
 545                         ;;
 546                         esac
 547                 fi
 548                 ulimit $opt $val
 549                 shift 2
 550         done
 551 }
 552
 553 # A function to start a program (now it's useful on read-only filesystem too)
 554 daemon() {
 555         local errors="" prog="" end="" waitname="" waittime=""
 556         local exit_code=0
 557         local nice=$SERVICE_RUN_NICE_LEVEL
 558         local fork user closefds redirfds pidfile makepid chdir=/
 559
 560         # NOTE: if you wonder how the shellish (by syntax) $prog works in ssd mode,
 561         # then the answer is: it totally ignores $prog and uses "$@" itself.
 562
 563         while [ $# -gt 0 ]; do
 564                 case $1 in
 565                 '')
 566                 msg_usage " daemon [--check] [--user user] [--fork] [--chdir directory] [--closefds] [--redirfds] [--waitforname procname] [--waitfortime seconds] [--pidfile file] [--makepid] [+/-nicelevel] {program} <program args>"
 567                         return 2
 568                         ;;
 569                 --check)
 570                         # for compatibility with redhat/mandrake
 571                         nls "warning: --check option is ignored!"
 572                         shift
 573                         ;;
 574                 --user)
 575                         shift
 576                         [ "$1" != "root" ] && prog="/bin/su $1 -s /bin/sh -c \""
 577                         user=$1
 578                         ;;
 579                 --fork)
 580                         fork=1
 581                         prog="/usr/bin/setsid sh -c \""
 582                         end='&'
 583                         ;;
 584                 --chdir)
 585                         shift
 586                         chdir=$1
 587                         ;;
 588                 --closefds)
 589                         closefds=1
 590                         ;;
 591                 --redirfds)
 592                         redirfds=1
 593                         ;;
 594                 --waitforname)
 595                         shift
 596                         waitname="$1"
 597                         ;;
 598                 --waitfortime)
 599                         shift
 600                         waittime="$1"
 601                         ;;
 602                 --pidfile)
 603                         shift
 604                         pidfile="$1"
 605                         case "$pidfile" in /*);; *) pidfile="/var/run/$pidfile";; esac
 606                         ;;
 607                 --makepid)
 608                         makepid=1
 609                         ;;
 610                 -*|+*)
 611                         nice=$1
 612                         shift
 613                         break
 614                         ;;
 615                 *)
 616                         break
 617                         ;;
 618                 esac
 619                 shift
 620         done
 621         # If command to execute ends with quotation mark, add remaining
 622         # arguments and close quotation.
 623         if [ "$prog" != "${prog%\"}" ]; then
 624                 prog="$prog $*$end\""
 625         else
 626                 prog="$prog $*$end"
 627         fi
 628
 629         _daemon_set_ulimits
 630
 631         [ -z "$DEFAULT_SERVICE_UMASK" ] && DEFAULT_SERVICE_UMASK=022
 632         [ -z "$DEFAULT_SERVICE_RUN_NICE_LEVEL" ] && DEFAULT_SERVICE_RUN_NICE_LEVEL=0
 633
 634         # And start it up.
 635         busy
 636         cd $chdir
 637         [ -n "$SERVICE_CPUSET" ] && is_yes "$CPUSETS" && echo $$ > "/dev/cpuset/${SERVICE_CPUSET}/tasks"
 638         if errors=$(
 639                 umask ${SERVICE_UMASK:-$DEFAULT_SERVICE_UMASK};
 640                 export USER=root HOME=/tmp TMPDIR=/tmp
 641
 642                 nice=${nice:-$DEFAULT_SERVICE_RUN_NICE_LEVEL}
 643                 nice=${nice:-0}
 644
 645                 # make nice level absolute, not to be dependant of nice level of shell where service started
 646                 nice=$((nice-$(nice)))
 647
 648                 if [ "$closefds" = 1 ]; then
 649                         exec 1>&-
 650                         exec 2>&-
 651                         exec 0>&-
 652                 elif [ "$redirfds" = 1 ]; then
 653                         exec 1>/dev/null
 654                         exec 2>/dev/null
 655                         exec 0>/dev/null
 656                 else
 657                         exec 2>&1
 658                 fi
 659
 660                 if is_no "$RC_LOGGING"; then
 661                         prog=$1; shift
 662                         if [ ! -x $prog ]; then
 663                                 logger -t rc-scripts -p daemon.debug "daemon: Searching PATH for $prog, consider using full path in initscript"
 664                                 local a o=$IFS
 665                                 IFS=:
 666                                 for a in $PATH; do
 667                                         if [ -x $a/$prog ]; then
 668                                                 prog=$a/$prog
 669                                                 break
 670                                         fi
 671                                 done
 672                                 IFS=$o
 673                         fi
 674                         /sbin/start-stop-daemon -q --start \
 675                                 --nicelevel $nice \
 676                                 ${pidfile:+--pidfile $pidfile} \
 677                                 ${makepid:+--make-pidfile} \
 678                                 ${user:+--chuid $user} \
 679                                 ${chdir:+--chdir "$chdir"} \
 680                                 ${fork:+--background} \
 681                                 ${waitname:+--name $waitname} \
 682                                 ${SERVICE_DROPCAPS:+--dropcap $SERVICE_DROPCAPS} \
 683                                 --exec "$prog" \
 684                                 -- ${1:+"$@"}
 685                 else
 686                         nice -n $nice initlog -c "$prog" 2>&1
 687                 fi
 688                 ); then
 689
 690                 if [ -n "$waitname" -a -n "$waittime" ]; then
 691                         # Save basename.
 692                         base=${waitname##*/}
 693                         # Find pid.
 694                         pid=$(pidofproc "$waitname" "$pidfile")
 695                         [ -z "$pid" ] && pid=$(pidofproc "$base" "$pidfile")
 696                         i=0
 697                         while [ "$i" -lt "$waittime" ]; do
 698                                 i=$((i + 1))
 699                                 checkpid $pid && sleep 1 || break
 700                         done
 701                 fi
 702                 log_success "$1 startup"
 703                 ok
 704         else
 705                 exit_code=1
 706                 fail
 707                 log_failed "$1 startup"
 708                 [ -n "$errors" ] && echo >&2 "$errors"
 709         fi
 710         return $exit_code
 711 }
 712
 713 # A function to stop a program.
 714 killproc() {
 715         local notset killlevel base pid pidfile result
 716         # Test syntax.
 717         if [ $# = 0 ]; then
 718                 msg_usage " killproc [--pidfile PIDFILE] {program} [-SIGNAME]"
 719                 return 2
 720         fi
 721
 722         while [ "$1" != "${1##-}" ]; do
 723                 case $1 in
 724                 --pidfile|-p)
 725                         pidfile="$2"
 726                         case "$pidfile" in /*);; *) pidfile="/var/run/$pidfile";; esac
 727                         shift 2
 728                         ;;
 729                 --waitforname)
 730                         waitname="$2"
 731                         shift 2
 732                         ;;
 733                 --waitfortime)
 734                         waittime="$2"
 735                         shift 2
 736                         ;;
 737                 esac
 738         done
 739
 740         busy
 741
 742         local notset=0
 743         # check for second arg to be kill level
 744         if [ -n "$2" ]; then
 745                 killlevel=$2
 746         else
 747                 notset=1
 748         fi
 749
 750         # experimental start-stop-daemon based killing.
 751         # works only with pidfile
 752         if is_no "$RC_LOGGING" && [ "$pidfile" ]; then
 753                 local sig=${killlevel:--TERM}
 754                 /sbin/start-stop-daemon -q --stop \
 755                         --retry ${sig#-}/10/${sig#-}/60/KILL/10 \
 756                         -s ${sig#-} \
 757                         ${pidfile:+--pidfile $pidfile}
 758                 result=$?
 759                 if [ "$result" -eq 0 ]; then
 760                         ok
 761                 else
 762                         fail
 763                 fi
 764                 return $result
 765         fi
 766
 767
 768         # Save basename.
 769         base=${1##*/}
 770
 771         # Find pid.
 772         pid=$(pidofproc "$1" "$pidfile")
 773         [ -z "$pid" ] && pid=$(pidofproc "$base" "$pidfile")
 774
 775         # Kill it.
 776         if [ -n "$pid" -a "$pid" != "$$" ] && checkpid $pid 2>&1; then
 777                 if [ "$notset" = "1" ]; then
 778                         if checkpid $pid 2>&1; then
 779                                 # TERM first, then KILL if not dead
 780                                 kill -TERM $pid
 781                                 usleep 100000
 782                                 if checkpid $pid && sleep 1 &&
 783                                         checkpid $pid && sleep 3 &&
 784                                         checkpid $pid; then
 785                                         # XXX: SIGKILL is sent already on 4th second!
 786                                         # HARMFUL for example to mysqld (which is already workarounded)
 787                                         kill -KILL $pid
 788                                         usleep 100000
 789                                 fi
 790                         fi
 791                         checkpid $pid
 792                         result=$?
 793                         if [ "$result" -eq 0 ]; then
 794                                 fail
 795                                 log_failed "$1 shutdown"
 796                         else
 797                                 ok
 798                                 log_success "$1 shutdown"
 799                         fi
 800                         result=$(( ! $result ))
 801                 else
 802                         # use specified level only
 803                         if checkpid $pid > /dev/null 2>&1; then
 804                                 kill $killlevel $pid
 805                                 result=$?
 806                                 if [ "$result" -eq 0 ]; then
 807                                         ok
 808                                         log_success "$1 got $killlevel"
 809                                 else
 810                                         result=7
 811                                         fail
 812                                         log_failed "$1 didn't get $killlevel"
 813                                 fi
 814                         else
 815                                 result=7
 816                                 died
 817                                 log_failed "$1 shutdown"
 818                         fi
 819                 fi
 820         else
 821                 died
 822                 log_failed "$1 shutdown"
 823                 result=7
 824         fi
 825
 826         if [ -n "$waitname" -a -n "$waittime" ]; then
 827                 # Save basename.
 828                 base=${waitname##*/}
 829                 # Find pid.
 830                 pid=$(pidofproc "$waitname" "$pidfile")
 831                 [ -z "$pid" ] && pid=$(pidofproc "$base" "$pidfile")
 832                 i=0
 833                 while [ "$i" -lt "$waittime" ]; do
 834                         i=$(( i + 1 ))
 835                         checkpid $pid && sleep 1 || break
 836                 done
 837         fi
 838
 839         # Remove pid file if any.
 840         if [ "$notset" = "1" ]; then
 841                 rm -f /var/run/${base}.pid
 842         fi
 843
 844         return $result
 845 }
 846
 847 # A function to find the pid of a program.
 848 pidofproc() {
 849         local pid pidfile base=${1##*/}
 850         pidfile="$base.pid"
 851         [ -n "$2" ] && pidfile="$2"
 852
 853         # Test syntax.
 854         if [ $# = 0 ]; then
 855                 msg_usage " pidofproc {program}"
 856                 return 2
 857         fi
 858
 859         # First try pidfile or "/var/run/*.pid"
 860         case "$pidfile" in
 861                 /*)pidfile="${pidfile}";;
 862                 *) pidfile="/var/run/$pidfile";;
 863         esac
 864         if [ -f "${pidfile}" ]; then
 865                 local p pid=""
 866                 for p in $(< "${pidfile}"); do
 867                         [ -z "$(echo "$p" | awk '{gsub(/[0-9]/,"");print;}')" ] && pid="$pid $p"
 868                 done
 869         fi
 870
 871         # Next try "pidof"
 872         [ -z "$pid" ] && pidof -o $$ -o $PPID -o %PPID -x "$1"
 873         pid=$(filter_chroot "$pid")
 874         echo $pid
 875 }
 876
 877 # status [--pidfile PIDFILE] {subsys} [{daemon}]"
 878 status() {
 879         local pid subsys daemon cpuset_msg pidfile
 880         if [ "$1" = "--pidfile" -o "$1" = "-p" ]; then
 881                 pidfile=$2
 882                 case "$pidfile" in /*);; *) pidfile="/var/run/$pidfile";; esac
 883                 shift 2
 884         fi
 885
 886         subsys=$1
 887         daemon=${2:-$subsys}
 888
 889         # Test syntax.
 890         if [ $# = 0 ]; then
 891                 msg_usage " status [--pidfile PIDFILE] {subsys} [{daemon}]"
 892                 return 2
 893         fi
 894
 895         # if pidfile specified, pid must be there
 896         if [ "$pidfile" ]; then
 897                 [ -f "$pidfile" ] && read pid < $pidfile
 898                 # filter_chroot does not filter out dead pids, so this extra check, see t/status-pidfile.sh
 899                 if [ ! -d "/proc/$pid" ]; then
 900                         pid=
 901                 fi
 902         else
 903                 pid=$(pidof -o $$ -o $PPID -o %PPID -x $daemon)
 904         fi
 905         pid=$(filter_chroot "$pid")
 906
 907         if [ "$pid" ]; then
 908                 cpuset_msg="..."
 909                 if [ -n "$SERVICE_CPUSET" ] && is_yes "$CPUSETS"; then
 910                         if grep -q "$pid" "/dev/cpuset/${SERVICE_CPUSET}/tasks"; then
 911                                 cpuset_msg=$(nls " in cpuset %s..." "$SERVICE_CPUSET")
 912                         else
 913                                 cpuset_msg=$(nls " outside of configured cpuset %s..." "$SERVICE_CPUSET")
 914                         fi
 915                 fi
 916                 nls "%s (pid %s) is running%s" "$daemon" "$pid" "$cpuset_msg"
 917                 return 0
 918         fi
 919
 920         # Next try "/var/run/*.pid" files; if pidfile is not set
 921         local base=${daemon##*/}
 922         if [ -z "$pidfile" -a -f /var/run/${base}.pid ]; then
 923                 read pid < /var/run/${base}.pid
 924                 pid=$(filter_chroot "$pid")
 925                 if [ "$pid" ]; then
 926                         nls "%s dead but pid file (%s) exists" "$subsys" /var/run/${base}.pid
 927                         return 1
 928                 fi
 929         fi
 930
 931         # See if /var/lock/subsys/$subsys exists
 932         if [ -f /var/lock/subsys/$subsys ]; then
 933                 nls "daemon %s dead but subsys (%s) locked" "$daemon" "$subsys"
 934                 return 2
 935         fi
 936         nls "%s is stopped" "$subsys"
 937         return 3
 938 }
 939
 940 # Confirm whether we really want to run this service
 941 confirm() {
 942         local answer
 943         nls -n "Start service %s (Y)es/(N)o/(C)ontinue? [Y] " "$1"
 944         read answer
 945         case $answer in
 946         y|Y|t|T|j|J|"")
 947                 return 0
 948                 ;;
 949         c|C|k|K|w|W)
 950                 return 2
 951                 ;;
 952         n|N)
 953                 return 1
 954                 ;;
 955         *)
 956                 confirm $1
 957                 return $?
 958                 ;;
 959         esac
 960 }
 961
 962 # module is needed (ie. is requested, is available and isn't loaded already)
 963 is_module() {
 964         # module name without .o at end
 965         if ! lsmod | grep -q "$1"; then
 966                 if ls -1R /lib/modules/$(uname -r)/ 2> /dev/null | grep -q "^${1}.\(\|k\)o\(\|.gz\)"; then
 967                         # true
 968                         return 0
 969                 fi
 970         fi
 971         # false
 972         return 1
 973 }
 974
 975 _modprobe() {
 976         local parsed single die args foo result
 977         parsed=no
 978         while is_no "$parsed"; do
 979                 case "$1" in
 980                 "single")
 981                         single=yes
 982                         shift
 983                         ;;
 984                 "die")
 985                         die=yes
 986                         shift
 987                         ;;
 988                 -*)
 989                         args="$args $1"
 990                         shift
 991                         ;;
 992                 *)
 993                         parsed=yes
 994                         ;;
 995                 esac
 996         done
 997         if is_yes "${single}"; then
 998                 foo="$@"
 999                 show "Loading %s kernel module(s)" "$foo"
1000                 busy
1001         fi
1002         if [ -x /sbin/modprobe ]; then
1003                 /sbin/modprobe -s $args "$@"
1004                 result=$?
1005         else
1006                 deltext; fail
1007                 result=1
1008         fi
1009         if is_yes "${single}"; then
1010                 deltext
1011                 if [ $result = "0" ]; then
1012                         is_yes "$single" && ok
1013                 else
1014                         fail
1015                         if is_yes "$die"; then
1016                                 nls "Could not load %s kernel module(s)" "$@"
1017                                 exit 1
1018                         fi
1019                 fi
1020         fi
1021 }
1022
1023 if is_no "$RC_LOGGING"; then
1024         log_success() {
1025                 :
1026         }
1027
1028         log_failed() {
1029                 :
1030         }
1031 else
1032         log_success() {
1033                 initlog -n $0 -s "$1 $2" -e 1
1034         }
1035
1036         log_failed() {
1037                 initlog -n $0 -s "$1 $2" -e 2
1038         }
1039 fi
1040
1041 # Check if any flavor of portmapper is running
1042 check_portmapper() {
1043         if [ -x /usr/sbin/rpcinfo ]; then
1044                 if /usr/sbin/rpcinfo -p localhost >/dev/null 2>/dev/null; then
1045                         return 0
1046                 else
1047                         return 1
1048                 fi
1049         elif [ -z "$(pidof portmap)" -a -z "$(pidof rpcbind)" ]; then
1050                 return 1
1051         fi
1052         return 0
1053 }
1054
1055 # is_fsmounted fstype mntpoint
1056 # Check if filesystem fstype is mounted on mntpoint
1057 is_fsmounted() {
1058         local fstype=$1
1059         local mntpoint=$2
1060
1061         [ -n "$fstype" -a -n "$mntpoint" ] || return 1
1062
1063         if [ -r /proc/mounts ]; then
1064                 grep -qE "[[:blank:]]$mntpoint[[:blank:]]+$fstype[[:blank:]]" /proc/mounts
1065                 return $?
1066         else
1067                 if [ "$(stat -L -f -c %T $mntpoint 2>/dev/null)" = "$fstype" ]; then
1068                         return 0
1069                 else
1070                         return 1
1071                 fi
1072         fi
1073 }
1074
1075 # __umount_loop awk_program fstab_file first_msg retry_msg umount_args
1076 # awk_program should process fstab_file and return a list of fstab-encoded
1077 # paths; it doesn't have to handle comments in fstab_file.
1078 __umount_loop() {
1079         local remaining sig=
1080         local retry=3 count
1081
1082         remaining=$(LC_ALL=C awk "/^#/ {next} $1" "$2" | sort -r)
1083         while [ -n "$remaining" -a "$retry" -gt 0 ]; do
1084                 if [ "$retry" -eq 3 ]; then
1085                         run_cmd "$3" fstab-decode umount $5 $remaining
1086                 else
1087                         run_cmd "$4" fstab-decode umount $5 $remaining
1088                 fi
1089                 count=4
1090                 remaining=$(LC_ALL=C awk "/^#/ {next} $1" "$2" | sort -r)
1091                 while [ "$count" -gt 0 ]; do
1092                         [ -z "$remaining" ] && break
1093                         count=$(($count-1))
1094                         usleep 500000
1095                         remaining=$(LC_ALL=C awk "/^#/ {next} $1" "$2" | sort -r)
1096                 done
1097                 [ -z "$remaining" ] && break
1098                 fstab-decode /bin/fuser -k -m $sig $remaining >/dev/null
1099                 sleep 3
1100                 retry=$(($retry -1))
1101                 sig=-9
1102         done
1103 }
1104
1105 # Similar to __umount loop above, specialized for loopback devices
1106 __umount_loopback_loop() {
1107         local remaining devremaining sig=
1108         local retry=3
1109
1110         remaining=$(awk '$1 ~ /^\/dev\/loop/ && $2 != "/" {print $2}' /proc/mounts)
1111         devremaining=$(awk '$1 ~ /^\/dev\/loop/ && $2 != "/" {print $1}' /proc/mounts)
1112         while [ -n "$remaining" -a "$retry" -gt 0 ]; do
1113                 if [ "$retry" -eq 3 ]; then
1114                         run_cmd "Unmounting loopback filesystems: " \
1115                                 fstab-decode umount $remaining
1116                 else
1117                         run_cmd "Unmounting loopback filesystems (retry):" \
1118                                 fstab-decode umount $remaining
1119                 fi
1120                 for dev in $devremaining ; do
1121                         losetup $dev > /dev/null 2>&1 && \
1122                                 run_cmd "Detaching loopback device $dev: " \
1123                                 losetup -d $dev
1124                 done
1125                 remaining=$(awk '$1 ~ /^\/dev\/loop/ && $2 != "/" {print $2}' /proc/mounts)
1126                 devremaining=$(awk '$1 ~ /^\/dev\/loop/ && $2 != "/" {print $1}' /proc/mounts)
1127                 [ -z "$remaining" ] && break
1128                 fstab-decode /bin/fuser -k -m $sig $remaining >/dev/null
1129                 sleep 3
1130                 retry=$(($retry -1))
1131                 sig=-9
1132         done
1133 }
1134
1135 rc_cache_init() {
1136         # If we have cachefile, use it.
1137         # If we don't, create memory variables and try to save silently,
1138         local cachefile='/var/cache/rc-scripts/msg.cache'
1139
1140         local term
1141         if is_yes "$ISATTY"; then
1142                 term=$TERM
1143         else
1144                 term=dumb
1145         fi
1146
1147         # We create $check variable which is used to invalidate the cache.
1148         # The $check contains user locale and terminal.
1149         local check="$term.$LC_MESSAGES.$INIT_COL"
1150
1151         if [ -f "$cachefile" -a "$cachefile" -nt /etc/sysconfig/system -a "$cachefile" -nt /etc/sysconfig/init-colors ]; then
1152                 if . "$cachefile" 2>/dev/null; then
1153                         if [ "$check" = "$_check" ]; then
1154                                 return
1155                         fi
1156                 fi
1157         fi
1158
1159         # primitive caching
1160         _busy=$(progress "BUSY" "$CBUSY")
1161         _ok=$(progress "DONE")
1162         _started=$(progress "WORK")
1163         _fail=$(progress "FAIL" "$CFAIL")
1164         _died=$(progress "DIED" "$CFAIL")
1165
1166         # we don't use heredoc, as ksh attempts to create tempfile then
1167         (> "$cachefile" ) 2>/dev/null || return
1168         echo "_busy='$_busy';" >> "$cachefile"
1169         echo "_ok='$_ok';" >> "$cachefile"
1170         echo "_started='$_started';" >> "$cachefile"
1171         echo "_fail='$_fail';" >> "$cachefile"
1172         echo "_died='$_died';" >> "$cachefile"
1173         echo "_check='$check';" >> "$cachefile"
1174 }
1175
1176 rc_gettext_init() {
1177         if [ -z "$GETTEXT" ]; then
1178                 if [ -x /bin/gettext -o -x /usr/bin/gettext ]; then
1179                         GETTEXT=yes
1180                 else
1181                         GETTEXT=no
1182                 fi
1183         fi
1184
1185         if [ -z "$TPUT" ]; then
1186                 if [ -d /usr/share/terminfo ] && [ -x /usr/bin/tput -o -x /bin/tput ]; then
1187                         TPUT=yes
1188                         # check if we are on proper terminal
1189                         tput longname >/dev/null 2>&1 || TPUT=no
1190                 else
1191                         TPUT=no
1192                 fi
1193         fi
1194 }
1195
1196 use_upstart () {
1197         # True when upstart-event-based boot should be used
1198         is_yes "$USE_UPSTART" && return 0
1199         is_no "$USE_UPSTART" && return 1
1200         if [ ! -x /sbin/initctl ] ; then
1201                 USE_UPSTART="no"
1202                 return 1
1203         fi
1204         local cmdline=$(cat /proc/cmdline 2>/dev/null)
1205         if strstr "$cmdline" "pld.no-upstart" ; then
1206                 USE_UPSTART="no"
1207                 return 1
1208         else
1209                 USE_UPSTART="yes"
1210                 return 0
1211         fi
1212 }
1213
1214 emit () {
1215         # emit upstart signal
1216         # only when 'upstart' boot is enabled
1217         use_upstart || return 0
1218         /sbin/initctl emit "$@"
1219 }
1220
1221 is_upstart_task() {
1222         # Return 0 if the given service is an upstart task.
1223         grep -q '^task' "/etc/init/$1.conf"
1224 }
1225 is_upstart_running() {
1226         # Return 0 if the given service is running via upstart
1227         initctl status "$1" 2>/dev/null | grep -q running
1228 }
1229 upstart_start() {
1230         local service=$1
1231         if is_upstart_running "${service}"; then
1232                 msg_already_running "$service"
1233                 return 0
1234         fi
1235
1236         msg_starting "${service}"
1237         if errors=$(/sbin/initctl start ${service} 2>&1) ; then
1238                 ok
1239                 return 0
1240         else
1241                 fail
1242                 echo "$errors" >&2
1243                 return 1
1244         fi
1245 }
1246 upstart_stop() {
1247         local service=$1
1248         if ! is_upstart_running "${service}"; then
1249                 msg_not_running "$name"
1250                 return 0
1251         fi
1252         msg_stopping "${service}"
1253         if errors=$(/sbin/initctl stop ${service}) ; then
1254                 ok
1255                 return 0
1256         else
1257                 fail
1258                 echo "$errors" >&2
1259                 return 1
1260         fi
1261 }
1262 upstart_reload() {
1263         local service=$1
1264         if ! is_upstart_running "${service}" && ! is_upstart_task "${service}" ; then
1265                 return 0
1266         fi
1267         msg_reloading "${service}"
1268         if errors=$(/sbin/initctl reload ${service}) ; then
1269                 ok
1270                 return 0
1271         else
1272                 fail
1273                 echo "$errors" >&2
1274                 return 1
1275         fi
1276 }
1277 upstart_status() {
1278         # get service status
1279         # should be compliant with
1280         # http://refspecs.freestandards.org/LSB_3.1.1/LSB-Core-generic/LSB-Core-generic/iniscrptact.html
1281         local service=$1
1282         local status
1283         if is_upstart_task "${service}" ; then
1284                 # we probably should have a way to handle task status
1285                 return 0
1286         fi
1287         if ! status=$(/sbin/initctl status "${service}") ; then
1288                 # program or service status is not known
1289                 return 4
1290         fi
1291         if strstr "$status" "running" ; then
1292                 # program is running or service is OK
1293                 echo "$status"
1294                 return 0
1295         else
1296                 # program is not running
1297                 echo "$status"
1298                 return 3
1299         fi
1300         # TODO: other statuses
1301 }
1302
1303 _upstart_controlled() {
1304         # If the service is to be handled by upstart
1305         # execute the start/stop/etc. commands the upstart way
1306         if ! use_upstart; then
1307                 return 0
1308         fi
1309         local script=$1
1310         shift
1311         local command=$1
1312         [ $# -gt 0 ] && shift
1313         local name=$(basename "$script")
1314         if [ ! -f /etc/init/${name}.conf ] ; then
1315                 return 0
1316         fi
1317         local commands
1318         local extra_commands
1319         local has_configtest
1320         if [ "$1" = "--except" ] ; then
1321                 shift
1322                 commands="$*"
1323                 for cmd in $commands ; do
1324                         if [ "$command" = "$cmd" ] ; then
1325                                 return 0
1326                         fi
1327                         case "$cmd" in
1328                                 start|stop|status|reload|restart|try-restart|force-reload)
1329                                         ;;
1330                                 configtest)
1331                                         has_configtest=yes
1332                                         extra_commands="|$cmd"
1333                                         ;;
1334                                 *)
1335                                         extra_commands="|$cmd"
1336                                         ;;
1337                         esac
1338                 done
1339         elif [ -n "$*" ] ; then
1340                 commands="$*"
1341                 local cmd
1342                 local found=0
1343                 # is there a better way
1344                 for cmd in $commands ; do
1345                         if [ "$command" = "$cmd" ] ; then
1346                                 found=1
1347                                 break;
1348                         fi
1349                 done
1350                 if [ $found = 0 ] ; then
1351                         # let the script handle it
1352                         return 0
1353                 fi
1354         fi
1355         case "$command" in
1356                 start)
1357                         upstart_start $name
1358                         exit $?
1359                         ;;
1360                 stop)
1361                         upstart_stop $name
1362                         exit $?
1363                         ;;
1364                 status)
1365                         upstart_status $name
1366                         exit $?
1367                         ;;
1368                 restart)
1369                         if is_yes "$has_configtest" ; then
1370                                 "$script" configtest || exit 1
1371                         fi
1372                         upstart_stop $name
1373                         upstart_start $name
1374                         exit $?
1375                         ;;
1376                 try-restart)
1377                         if ! is_upstart_running "$name" ; then
1378                                 exit 0
1379                         fi
1380                         if is_yes "$has_configtest" ; then
1381                                 "$script" configtest || exit 1
1382                         fi
1383                         upstart_stop $name
1384                         upstart_start $name
1385                         exit $?
1386                         ;;
1387                 reload)
1388                         if is_yes "$has_configtest" ; then
1389                                 "$script" configtest || exit 1
1390                         fi
1391                         if is_upstart_task "$name" ; then
1392                                 nls "$command not implemented for $name"
1393                                 exit 3
1394                         else
1395                                 upstart_reload "$name"
1396                                 exit $?
1397                         fi
1398                         ;;
1399                 force-reload)
1400                         if is_yes "$has_configtest" ; then
1401                                 "$script" configtest || exit 1
1402                         fi
1403                         if is_upstart_task "$name" ; then
1404                                 upstart_stop "$name"
1405                                 upstart_start "$name"
1406                                 exit $?
1407                         else
1408                                 upstart_reload "$name"
1409                                 exit $?
1410                         fi
1411                         ;;
1412                 *)
1413                         msg_usage "$0 {start|stop|restart|reload|force-reload|status$extra_commands}"
1414                         exit 3
1415                         ;;
1416         esac
1417         return 1 # should not happen
1418 }
1419
1420 # Usage:
1421 #       somewhere at the begining of init script:
1422 #     upstart_controlled
1423 #               - to pass implement all upstart commands via initctl
1424 #                 start, stop, status, restart, reload and force_reload
1425 #                 are implemented
1426 #     upstart_controlled command...
1427 #               - to pass handle only specific commands the upstart way
1428 #                 and leave the rest to the script
1429 #
1430 alias upstart_controlled='_upstart_controlled $0 "$@"'
1431
1432 rc_gettext_init
1433 rc_cache_init
1434
1435 #/*
1436 # * Local variables:
1437 # * mode: sh
1438 # * indent-tabs-mode: notnil
1439 # * End:
1440 # *
1441 # */