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