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