Remove the run file if the build fails.
[util-vserver.git] / scripts / functions
1 # $Id$  --*- sh -*--
2
3 # Copyright (C) 2003,2004,2005,2006 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
4 #  
5 # This program is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; version 2 of the License.
8 #  
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 # GNU General Public License for more details.
13 #  
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17
18 _VS_LOGFILE=
19 _VS_ERRFILE=
20
21 _VS_NEWLINE='
22 '
23 declare -r _VS_NEWLINE=${_VS_NEWLINE:0-1}
24 declare -r VS_ALLVSERVERS_ARGS=all,marked,unmarked,stopped,running
25
26 function findObject
27 {
28     local _fo_mod=$1
29     local _fo_var=$2
30     local _fo_file=
31     local _fo_i=X
32     shift 2
33     
34     for _fo_i; do
35         test -n "$_fo_i"         || continue
36         test ! $_fo_mod "$_fo_i" || { _fo_file=$_fo_i; break; }
37     done
38
39     test -z "$_fo_i" -o -n "$_fo_file" || {
40         echo "Can not find file for '$_fo_var'; aborting"
41         exit 1
42     } >&2
43
44     eval "$_fo_var=\"$_fo_file\""
45 }
46
47 function findFile
48 {
49     findObject -f "$@"
50 }
51
52 function findDir
53 {
54     findObject -d "$@"
55 }
56
57 function findAndCopy
58 {
59     local dst=$1
60     test ! -s "$dst"         || return 0
61     
62     local tmp
63     shift
64     findFile tmp "$@"
65
66     test -n "$tmp" -a -s "$tmp" || return 0
67     $_CP -af "$tmp" "$dst"
68 }
69
70 ## Usage: isRegularFile <filename> [<mod>]
71 function isRegularFile
72 {
73     test ${2:--f} "$1" || return 1
74
75     case $1 in
76         (*.rpmsave|*.rpmnew|*.rpmorig|*.cfsaved*|*.~*~) return 1;;
77     esac
78
79     return 0
80 }
81
82 function getPhysicalDir
83 {
84     ( set -P && cd "$1" && pwd )
85 }
86
87 ## Usage:: isDirectoryEmpty <dir> [<ignore-lost+found>]
88 function isDirectoryEmpty
89 {
90     (
91         shopt -s dotglob  || *
92         shopt -s nullglob || *
93
94         ignore_lostfound=1
95         test -z "$2" -o "$2" = 0 || ignore_lostfound=
96         
97         for i in "$1"/*; do
98             case "$i" in
99                 ($1/lost+found)
100                     test "$ignore_lostfound" || exit 1
101                     ;;
102                 (*)     echo "$i"; exit 1
103             esac
104         done
105
106         exit 0
107     )
108 }
109
110 ## Usage: logging <message>
111 function logging
112 {
113     if test -n "$_VS_LOGFILE"; then
114         echo "$@" >>"$_VS_LOGFILE"
115     else
116         echo "$@"
117     fi
118 }
119
120 ## Usage: warning <message>
121 function warning
122 {
123     if test -n "$_VS_ERRFILE"; then
124         echo "$@" >>"$_VS_ERRFILE"
125     else
126         echo "$@" >&2
127     fi
128 }
129
130 ## Usage: panic <message>
131 function panic
132 {
133     if test -n "$_VS_ERRFILE"; then
134         echo "$@" >>"$_VS_ERRFILE"
135     else
136         echo "$@" >&2
137     fi
138
139     exit 1
140 }
141
142 ## Usage: execute <message>
143 function execute
144 {
145     test -z "${DEBUG_EXEC:-}"       || echo "$@"
146     test "${DEBUG_EXEC:-}" = noexec || exec "$@"
147     exit 0
148 }
149
150
151 ## Usage: spawn <message>
152 function spawn
153 {
154     test -z "${DEBUG_EXEC:-}"       || echo  "$@"
155     test "${DEBUG_EXEC:-}" = noexec || "$@"
156 }
157
158 ## Usage: isNumber <arg>
159 function isNumber
160 {
161     local tmp
162     let tmp=$1+0 2>/dev/null || test -z "${1//0/}" -a -n "$1" || return 1
163     return 0
164 }
165
166 ## Usage: hasSubstring <haystack> <needle>+
167 function hasSubstring
168 {
169     local pat=$1
170     local i
171     
172     shift
173
174     for i; do
175         test x"${pat/*$i*/$i}" = x"$i" || continue
176         return 0
177     done
178
179     return 1
180 }
181
182 ## Usage: colorize <style> <command>
183 function colorize
184 {
185     local       style=$1
186     shift
187     
188     if ! $_TTY -s; then
189         "$@"
190     else
191         local   cfile
192         findFile cfile "$__CONFDIR"/.defaults/styles/"$style" ''
193         if test -n "$cfile"; then
194           $_CAT "$cfile"
195         else
196           case "$style" in
197             (bold)      echo -ne "\e[1m";;
198             (emph)      echo -ne "\e[34m";;
199             (info)      echo -ne "\e[0;34m";;
200             (warn*)     echo -ne "\e[1;31m";;
201             (error)     echo -ne "\e[1;33;41m";;
202             (*)         ;;
203           esac
204         fi
205             
206         ( "$@" )
207         echo -ne "\e[m"
208     fi
209 }
210
211 function colpanic
212 {
213     if test -n "$_VS_ERRFILE"; then
214         echo "$@" >>"$_VS_ERRFILE"
215     else
216         colorize error echo -n "$@" >&2
217     fi
218     echo
219
220     exit 1
221 }
222
223 function colwarning
224 {
225     colorize warning warning "$@"
226 }
227
228 function colinfo
229 {
230     colorize info    echo "$@"
231 }
232
233
234 ## Usage: xtermTitle <title>
235 function xtermTitle
236 {
237     $_TTY -s || return 0
238     echo -ne "\e]0;$@\007"
239 }
240
241 _VS_LOCKS=''
242 ## Usage: lock <lockfile> [<timeout>]
243 function lock
244 {
245     local tmp=$($_MKTEMP vserver-lock.XXXXXX)
246     $_RM -f $tmp
247     $_MKFIFO -m600 $tmp
248
249     $_LOCKFILE "$1" $tmp $2 &
250     $_GREP -q true $tmp 2>/dev/null || return 1
251     
252     _VS_LOCKS="$! $_VS_LOCKS"
253 }
254
255 ## Usage: unlock [<num>]
256 function unlock
257 {
258     local num=$1
259     local i
260
261     set -- $_VS_LOCKS
262     while test "$#" -gt 0; do
263         kill -HUP "$1" >/dev/null || :
264         shift
265         test "$num" != 1 || break
266         test -z "$num"   || let --num
267     done
268     _VS_LOCKS="$@"
269 }
270
271 function get_init_cwd
272 {
273     if test -n "$INIT_CWD"; then
274         echo "$INIT_CWD"
275     else
276         pwd
277     fi
278 }
279 function set_init_cwd
280 {
281     INIT_CWD="`pwd`"
282     export INIT_CWD
283 }
284
285 function _setVserverDir
286 {
287     local vserver="$1"
288     case "$vserver" in
289         ./*) VSERVER_DIR="`get_init_cwd`/$vserver";;
290         /*)  VSERVER_DIR="$vserver"               ;;
291         *)   VSERVER_DIR="$__CONFDIR/$vserver"    ;;
292     esac
293 }
294
295 function _setVserverName
296 {
297     if test -e "$VSERVER_DIR"/name; then
298         read VSERVER_NAME <"$VSERVER_DIR"/name
299     else
300         VSERVER_NAME=$(basename "$VSERVER_DIR")
301     fi
302 }
303
304 function _pkgMountBindDir()
305 {
306     test "$1" != "$2" || return 0
307
308     $_MOUNT -n --bind "$1" "$2"
309 }
310
311 function _pkgSetVarsBase
312 {
313     case "$vserver" in
314         ./*|/*)
315             if test -d "$vserver/vdir"; then
316                 BASEDIR=$vserver
317                 VDIR=$(getPhysicalDir "$vserver/vdir")
318                 
319                 PKGDIR=$BASEDIR/apps/pkgmgmt
320                 test -d "$PKGDIR" || {
321                     echo "Can not find configuration-directory for package-managment tools"
322                     exit 1
323                 } >&2
324                 findDir EXECDIR      $PKGDIR/execdir     /
325             else
326                 VDIR=$(getPhysicalDir "$vserver")
327                 PKGDIR=
328             fi
329             ;;
330         *)
331             BASEDIR=$__CONFDIR/$vserver
332             test -d "$BASEDIR" || {
333                 echo "Can not find configuration-directory"
334                 exit 1
335             } >&2
336             
337             VDIR=$BASEDIR/vdir
338             test -d "$VDIR"   || VDIR=$__DEFAULT_VSERVERDIR/$vserver
339             VDIR=$(getPhysicalDir "$VDIR")
340             
341             PKGDIR=$BASEDIR/apps/pkgmgmt
342             test -d "$PKGDIR" || {
343                 echo "Can not find configuration-directory for package-managment tools"
344                 exit 1
345             } >&2
346
347             findDir EXECDIR      $PKGDIR/execdir     /
348
349             ;;
350     esac
351
352     if test -z "$WORKAROUND_106057"; then
353         _rpmdb_mntpoint=/dev
354     else
355         _rpmdb_mntpoint=/.rpmdb
356     fi
357 }
358
359 function _pkgSetVarsRPM
360 {
361     if test -n "$PKGDIR"; then
362         findDir RPMETCDIR    $PKGDIR/rpmetc      $PKGDIR/base/rpm/etc       /etc/rpm
363         findDir RPMSTATEDIR  $PKGDIR/rpmstate    $PKGDIR/base/rpm/state
364
365         findDir RPMLIBDIR    $PKGDIR/rpmlib      /
366
367     else
368         findDir RPMETCDIR    "$VDIR"/etc/rpm     /etc/rpm
369         findDir RPMSTATEDIR  "$VDIR"/var/lib/rpm
370         RPMLIBDIR=/
371     fi
372     
373     RPMSTATEDIR=$(getPhysicalDir "$RPMSTATEDIR")
374     RPMETCDIR=$(getPhysicalDir "$RPMETCDIR")
375 }
376
377 function _pkgSetVarsApt
378 {
379     if test -n "$PKGDIR"; then
380         findDir APTETCDIR    $PKGDIR/aptetc      $PKGDIR/base/apt/etc       /etc/apt
381         findDir APTSTATEDIR  $PKGDIR/aptstate    $PKGDIR/base/apt/state
382         findDir APTCACHEDIR  $PKGDIR/aptcache    $PKGDIR/base/apt/cache
383         findDir APTARCHIVDIR $PKGDIR/aptarchives $PKGDIR/base/apt/archives  /var/cache/apt/archives
384     else
385         findDir APTETCDIR    "$VDIR"/etc/apt            /etc/apt
386         findDir APTSTATEDIR  "$VDIR"/var/state/apt
387         findDir APTCACHEDIR  "$VDIR"/var/cache/apt
388         findDir APTARCHIVDIR "$VDIR"/var/cache/apt/archives /var/cache/apt/archives
389     fi
390
391     findFile APT_CONFIG "$APTETCDIR"/apt.conf ""
392     test -z "$APT_CONFIG" || export APT_CONFIG
393 }
394
395 function _pkgSetVarsYum
396 {
397     if test -n "$PKGDIR"; then
398         findDir YUMETCDIR    $PKGDIR/yumetc      $PKGDIR/base/yum/etc       /etc
399         findDir YUMCACHEDIR  $PKGDIR/yumcache    $PKGDIR/base/yum/cache
400     else
401         findDir YUMETCDIR    "$VDIR"/etc         /etc
402         findDir YUMCACHEDIR  "$VDIR"/var/cache/yum
403     fi
404 }
405
406
407 function _pkgMountBase
408 {
409     :
410 }
411
412 function _pkgMountApt
413 {
414     :
415 }
416
417 function _pkgMountYum
418 {
419     :
420 }
421
422 function _pkgMountRPM
423 {
424     _pkgMountBindDir "$RPMETCDIR" /etc/rpm
425     test "$RPMLIBDIR" = "/" || _pkgMountBindDir "$RPMLIBDIR" /usr/lib/rpm
426
427     pushd "$VDIR" >/dev/null
428
429     $_SECURE_MOUNT --chroot -n --bind "$RPMSTATEDIR" "$_rpmdb_mntpoint"
430     test -z "$WORKAROUND_106057" || mount -n --bind "$RPMSTATEDIR" "$_rpmdb_mntpoint"
431
432     test -e "$VDIR"/proc/self/status || \
433         $_SECURE_MOUNT --chroot -n -t proc none /proc
434
435     popd >/dev/null
436 }
437
438 function _pkgSetEnvBase
439 {
440     test "$EXECDIR" = "/" || {
441         PATH=$EXECDIR:$PATH
442         LD_LIBRARY_PATH=$EXECDIR${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
443     }
444
445     export PATH LD_LIBRARY_PATH
446 }
447
448 function _pkgSetEnvApt
449 {
450     :
451 }
452
453 function _pkgSetEnvYum
454 {
455     :
456 }
457
458 function _pkgSetEnvRPM
459 {
460     CUR_VSERVER=$vserver
461     RPM_FAKE_NAMESPACE_MOUNTS=$_rpmdb_mntpoint
462     RPM_BINARY=$_VRPM_PRELOAD
463
464     export CUR_VSERVER RPM_FAKE_NAMESPACE_MOUNTS RPM_BINARY
465 }
466
467 function pkgInit
468 {
469     local i
470     local vserver=$1
471     shift
472     
473     _pkgSetVarsBase
474     for i; do
475         case "$i" in
476             rpm)        _pkgSetVarsRPM;;
477             apt)        _pkgSetVarsApt;;
478             yum)        _pkgSetVarsYum;;
479             *)          echo "Unknown packaging flavor" >&2; exit 1;;
480         esac
481     done
482
483     _pkgMountBase
484     for i; do
485         case "$i" in
486             rpm)        _pkgMountRPM;;
487             apt)        _pkgMountApt;;
488             yum)        _pkgMountYum;;
489         esac
490     done
491
492     _pkgSetEnvBase
493     for i; do
494         case "$i" in
495             rpm)        _pkgSetEnvRPM;;
496             apt)        _pkgSetEnvApt;;
497             yum)        _pkgSetEnvYum;;
498         esac
499     done
500
501     _PKG_FLAVORS="$@"
502     _PKG_VSERVER=$vserver
503 }
504
505 function isAvoidNamespace
506 {
507     local cfgdir
508
509     $_VSERVER_INFO - FEATURE namespace   || return 0
510     cfgdir=$($_VSERVER_INFO "$1" CFGDIR) || return 0
511     test ! -e "$cfgdir"/namespace        || return 1
512     test -e "$__CONFDIR"/.defaults/nonamespace -o \
513          -e "$cfgdir"/nonamespace
514 }
515
516 function isNamespaceCleanup
517 {
518     local cfgdir
519
520     $_VSERVER_INFO - FEATURE namespace   || return 1
521     cfgdir=$($_VSERVER_INFO "$1" CFGDIR) || return 1
522     isAvoidNamespace "$1"                && return 1
523     test -e "$cfgdir"/namespace-cleanup  && return 0
524     test -e "$cfgdir"/nonamespace-cleanup -o \
525          -e "$__CONFDIR"/.defaults/nonamespace-cleanup && return 1
526     return 0
527 }
528
529 ## Usage: getAllVservers <var> [<KIND>*]
530 function getAllVservers
531 {
532     local _ga_i
533     declare -a _ga_tmp=()
534
535     for _ga_i in $__CONFDIR/*; do
536         isRegularFile "$_ga_i" -d   || continue
537
538         test ! -e "$_ga_i"/disabled || continue
539         test   -d "$_ga_i"/vdir     || continue
540
541         local _ga_doadd=1
542         local _ga_markfile=$_ga_i/apps/init/mark
543         
544         case ${2:-ALL} in
545             (MARKED)    test   -s "$_ga_markfile" || _ga_doadd=;;
546             (UNMARKED)  test ! -s "$_ga_markfile" || _ga_doadd=;;
547             (STOPPED)   ! $_VSERVER "$_ga_i" running &>/dev/null || _ga_doadd=;;
548             (RUNNING)     $_VSERVER "$_ga_i" running &>/dev/null || _ga_doadd=;;
549             (ALL)       ;;
550             (*)         panic $"Unknown vserver tagging '$2'";;
551         esac
552
553         test -z "$_ga_doadd" || _ga_tmp=( "${_ga_tmp[@]}" "${_ga_i##$__CONFDIR/}")
554     done
555
556     eval $1='( "${_ga_tmp[@]}" )'
557 }
558
559 ## Usage: getAllVserversByArg <var> <arg>
560 function getAllVserversByArg
561 {
562     local _gav_mark=
563     
564     case $2 in
565         (--all)         _gav_mark=ALL;;
566         (--marked)      _gav_mark=MARKED;;
567         (--unmarked)    _gav_mark=UNMARKED;;
568         (--stopped)     _gav_mark=STOPPED;;
569         (--running)     _gav_mark=RUNNING;;
570         (*)             return 1;;
571     esac
572
573     getAllVservers "$1" "$_gav_mark"
574 }
575
576 ## Usage: _getProcNumberCount <ctx> <var>
577 function _getProcNumberCount
578 {
579     local _gp_var=$2
580     local _gp_procnr_cnt=0
581
582     # Use /proc/virtual from kernel 2.6 when possible
583     if test -d "/proc/virtual"; then
584         set -- $($_GREP '^Tasks:' "/proc/virtual/$1/status" 2>/dev/null)
585         _gp_procnr_cnt=$2
586     else
587         _gp_procnr_cnt=$($_VPS ax | $_AWK '{print $2}' | $_GREP -x "$1" | $_WC -l )
588     fi
589
590     let _gp_procnr_cnt=_gp_procnr_cnt+0
591     eval $_gp_var=\$_gp_procnr_cnt
592 }
593
594 ## Usage: getVserverCtx <vdir> <result-varname> [<procnumber-varname> [<do-cleanup>]]
595 ## Returns: 0 iff vserver is running
596 function getVserverStatus
597 {
598     test -r "$1"/run || return 1
599
600     local _gvs_ctx
601     read _gvs_ctx <"$1"/run
602     eval "$2"=\$_gvs_ctx
603
604     test "$1" = "$($_VUNAME -g --xid "$_gvs_ctx" context)" || return 1
605
606     test -n "$3"     || return 0
607     local _gvs_tmp
608     _getProcNumberCount "$_gvs_ctx" _gvs_tmp
609     eval "$3"=\$_gvs_tmp
610
611     if test "$_gvs_tmp" = 0; then
612         local runfile=$($_READLINK "$1/run")
613         test -z "$4" || $_RM -f "$runfile"
614         return 1
615     fi
616
617     return 0
618 }
619
620 ## Usage: isCtxRunning <ctx>
621 function isCtxRunning
622 {
623     local _tmp
624     _getProcNumberCount "$1" _tmp
625     test $_tmp -gt 0
626 }
627
628 ## Usage: isVserverRunning <vdir> [<ctx-varname>]
629 function isVserverRunning
630 {
631     local _ivr_ctx _ivr_procnum
632
633     getVserverStatus "$1" _ivr_ctx _ivr_procnum 1 || return 1
634     test "$_ivr_procnum" != 0                     || return 1
635     test -z "$2" || eval "$2"=\$_ivr_ctx
636     return 0
637 }
638
639 ## Called as 'getFileValue <varname> <filename>+'
640 function getFileValue
641 {
642     local _gfv_var=$1
643     local _gfv_file
644     shift
645
646     findFile _gfv_file "$@" ''
647     test -n "$_gfv_file" -a -r "$_gfv_file" || return 0
648     eval read "$_gfv_var" <"$_gfv_file"
649 }
650
651 ## Called as 'getFileArray <varname> <filename>+'
652 function getFileArray
653 {
654     local _gfa_var=$1
655     local _gfa_file
656     shift
657
658     findFile _gfa_file "$@" ''
659     test -n "$_gfa_file" -a -r "$_gfa_file" || return 0
660     local IFS=$_VS_NEWLINE
661     eval "$_gfa_var"='( $(< "$_gfa_file") )'
662 }
663
664 function checkComponents
665 {
666     local       i
667     local       msg=$1
668     local       x_failed=
669
670     shift
671     
672     for i; do
673         local failed=
674         case "$i" in
675             (core)      test -x "$_CHBIND"           || failed=1;;
676             (build)     test -x "$_VSERVER_BUILD"    || failed=1;;
677             (sysv)      test -x "$__INITRDDIR/vserver"    || failed=1;;
678             (devel)     test -d "$__INCLUDEDIR/vserver.h" || failed=1;;
679             (*)         echo "Unknown component '$i'" >&2
680                         return false
681                         ;;
682         esac
683
684         test -z "$failed" || {
685             echo "$msg: $i"
686             x_failed=1
687         } >&2
688     done
689
690     test -z "$x_failed"
691 }
692
693 ## Usage: isKernelAPI <ver> [<cmp-modifier>]
694 function isKernelAPI
695 {
696     local api
697     api=$($_VSERVER_INFO - APIVER) || api=0
698     test $[ $api ] -${2:-ge} $[ $1 ]
699 }
700
701 ## Usage: callInNamespace <vserver> <command> <args>*
702 function callInNamespace
703 {
704     local ctx=
705     
706     isAvoidNamespace "$1" || \
707     ctx=$( $_VSERVER_INFO "$1" CONTEXT f ) || ctx=
708
709     shift
710     if test -n "$ctx"; then
711         $_VNAMESPACE --enter "$ctx" -- "$@"
712     else
713         "$@"
714     fi
715 }
716
717 ## Usage: setDefaultTTY <vdir> [<fallback-tty>]
718 function setDefaultTTY
719 {
720     local cfgdir ttyname
721
722     cfgdir=$($_VSERVER_INFO "$1" APPDIR init) || cfgdir=
723     findObject -e ttyname \
724         ${cfgdir:+"$cfgdir"/tty} \
725         "$__CONFDIR/.defaults/apps/init/tty" \
726         $2 /dev/null
727
728     if test -f "$ttyname"; then
729         exec </dev/null
730     else
731         exec <$ttyname
732     fi
733     exec    >>$ttyname 2>&1
734 }
735
736 ## Usage: killContext <XID> [<SIG>]
737 function killContext
738 {
739     local sig=${2:-9}
740
741     #$_VKILL -s STOP   --xid "$1" 1 &>/dev/null || :
742     $_VKILL -s "$sig" --xid "$1" 1 &>/dev/null || :
743     $_VKILL -s "$sig" --xid "$1"   &>/dev/null || :
744     #$_VKILL -s "$sig" --xid "$1" 1 &>/dev/null || :
745     #$_VKILL -s CONT   --xid "$1" 1 &>/dev/null || :
746 }
747
748 function useVlogin
749 {
750     test ! -e "$__CONFDIR/.defaults/apps/vlogin/disable"
751 }
752
753 ## Usage: pkgmgmt.guessStyle <vserver> <resultvar>
754 function pkgmgmt.guessStyle()
755 {
756     local _pgs_vdir
757     _pgs_vdir=$($_VSERVER_INFO "$1" VDIR) || {
758         echo $"Can not determine vserver-root" >&2
759         return 1
760     }
761     local _pgs_cfgdir=$($_VSERVER_INFO "$1" APPDIR pkgmgmt) || :
762
763     if test -n "$_pgs_cfgdir" -a -e "$_pgs_cfgdir"/style; then
764         read style <"$_pgs_cfgdir"/style
765     elif test -e "$_pgs_vdir"/etc/redhat-release -o -e "$_pgs_vdir"/etc/fedora-release; then
766         style=redhat
767     elif test -e "$_pgs_vdir"/etc/mandrake-release; then
768         style=mandrake
769     elif test -e "$_pgs_vdir"/etc/debian_version; then
770         style=debian
771     elif test -e "$_pgs_vdir"/etc/SuSE-release; then
772         style=suse
773     else
774         echo $"Can not determine packagemanagement style" >&2
775         return 1
776     fi
777
778     eval $2=\$style
779     return 0
780 }
781
782 ## Usage: pkgmgmt.isInternal <vserver>
783 ## returns true iff <vserver> is configured for internal packagemanagement
784 ## A typical application is
785 ## | is_external=
786 ## | pkgmgmt.isInternal "$vserver" || is_external=1
787 function pkgmgmt.isInternal
788 {
789     local cfgdir=$($_VSERVER_INFO "$1" APPDIR pkgmgmt) || :
790
791     test -z "$cfgdir" -o ! -d "$cfgdir" -o -e "$cfgdir"/internal
792 }
793
794 ## Usage: pkgmgmt.isAptAvailable <cfgdir> <vdir> [<is-internal>]
795 function pkgmgmt.isAptAvailable
796 {
797     local cfgdir="$1"
798     local vdir="$2"
799     local is_internal="$3"
800     
801     local have_apt i
802     if test -n "$is_internal"; then
803         have_apt=1
804         test -d "$cfgdir"/base/apt -o -d "$cfgdir"/aptetc || have_apt=
805     else
806         have_apt=
807         for i in /bin /usr/bin /usr/local/bin; do
808             test ! -x "$vdir$i"/apt-get || { have_apt=1; break; }
809         done
810     fi
811
812     test -n "$have_apt" && return 0 || return 1
813 }
814
815 ## Usage: pkgmgmt.isYumAvailable <cfgdir> <vdir> [<is-internal>]
816 function pkgmgmt.isYumAvailable
817 {
818     local cfgdir="$1"
819     local vdir="$2"
820     local is_internal="$3"
821     
822     local have_yum i
823     if test -n "$is_internal"; then
824         have_yum=1
825         test -d "$cfgdir"/base/yum -o -d "$cfgdir"/yumetc || have_yum=
826     else
827         have_yum=
828         for i in /bin /usr/bin /usr/local/bin; do
829             test ! -x "$vdir$i"/yum || { have_yum=1; break; }
830         done
831     fi
832
833     test -n "$have_yum" && return 0 || return 1
834 }
835
836
837 function vshelper.doSanityCheck
838 {
839     local vshelper this_xid i
840     declare -a warnings=()
841     local solution_disable=
842     local solution_sysctl=
843
844     vshelper.isEnabled && vshelper.isEnabled warning || return 0
845     
846     this_xid=$($_VSERVER_INFO - XID) ||
847         panic $"Failed to determine current context; aborting..."
848
849     ## Do nothing in other xid's; the helper will be executed in xid 0 only
850     test "$this_xid" = 0 || return 0
851
852     local proc_file=/proc/sys/kernel/vshelper
853
854     if ! test -r "$proc_file"; then
855         vshelper=
856         warnings=( "${warnings[@]}"
857                    $"File '$proc_file' does not exist but is required for vshelper setup" )
858         solution_disable=1
859     else
860         vshelper=$(cat "$proc_file")
861
862         $_CMP -s "$vshelper" "$_VSHELPER" || {
863             local readable=""
864             test -r "$vshelper" && readable=1
865             warnings=( "${warnings[@]}"
866                        $"The configured vshelper '$vshelper' does not match the 'vshelper'
867   script of the util-vserver package.${readable:+ Maybe you have two versions installed?}"
868                      )
869             solution_disable=1
870             solution_sysctl=1
871         }
872     fi
873
874     test -d "$__VSHELPERSTATEDIR" || {
875         warnings=( "${warnings[@]}"
876                    $"\
877 The vshelper state-directory '$__VSHELPERSTATEDIR' does not exist; since
878 it is created by 'make install', this indicates a serious problem with
879 your util-vserver installation" )
880         solution_disable=1
881     }
882
883     test "${#warnings[@]}" -eq 0 || {
884         warning $"\
885 The following problem(s) were encountered while verifying vshelper
886 functionality:"
887
888         for i in "${warnings[@]}"; do
889             warning "* $i"
890         done
891
892         warning $"\
893         
894 To fix this, you can:"
895
896         test -z "$solution_disable" || warning $"\
897 * disable vshelper entirely by executing
898   | touch \"$__CONFDIR/.defaults/apps/vshelper/disabled\"
899 * disable only this message by executing
900   | touch \"$__CONFDIR/.defaults/apps/vshelper/warning-disabled\""
901
902         test -x "$solution_sysctl" || warning $"\
903 * configure the util-vserver vshelper script, e.g. by adding
904   | kernel.vshelper = $_VSHELPER
905   to /etc/sysctl.conf and rebooting the machine, or by executing
906   | echo \"$_VSHELPER\" >$proc_file"
907
908         warning ""
909
910         return 1
911     }
912
913     return 0
914 }
915
916 ## Usage: vshelper.isEnabled [<style>] [<vserver>]
917 function vshelper.isEnabled
918 {
919     local f=${1:+$1-}disabled
920     test ! -e "$__CONFDIR"/.defaults/apps/vshelper/"$f"     || return 1
921     $_VSERVER_INFO - FEATURE vshelper                       || return 1
922     if test -n "$2"; then
923         local appdir
924         appdir=$($_VSERVER_INFO "$2" APPDIR vshelper)       || return 0
925         test -z "$2" -o ! -e "$appdir/$f"                   || return 1
926     fi
927
928     return 0
929 }
930
931 ## Usage: vshelper.isDebug [<vserver>]
932 function vshelper.isDebug
933 {
934     test ! -e "$__CONFDIR"/.defaults/apps/vshelper/debug    || return 0
935     $_VSERVER_INFO - FEATURE vshelper                       || return 1
936     if test -n "$1"; then
937         local appdir
938         appdir=$($_VSERVER_INFO "$1" APPDIR vshelper)       || return 1
939         test ! -e "$appdir/debug"                           || return 0
940     fi
941
942     return 1
943 }
944
945 function vshelper._getHandlerInternal
946 {
947     local _vghi_var=$1
948     local _vghi_tmp
949     shift
950     shift       ## HACK: see below the note about the 'set -u' mode
951     
952     while test "$#" -ge 2; do
953         local _vghi_mod=$1
954         local _vghi_obj=$2
955         shift 2
956
957         test "$_vghi_mod" "$_vghi_obj" || continue
958         case "$_vghi_mod" in
959             (-x)
960                 eval $_vghi_var=\$_vghi_obj
961                 ;;
962             (-e)
963                 read _vghi_tmp <"$_vghi_obj"
964                 eval $_vghi_var=:\$_vghi_tmp
965                 ;;
966             (*) panic $"Internal error, unexpected modifier '$_vghi_mod'"
967         esac
968         return 0
969     done
970     
971     return 1
972 }
973
974 ## Usage: vshelper.getHandler <result-var> <vserver> <action>
975 function vshelper.getHandler
976 {
977     local _vgh_appdir
978     _vgh_appdir=$($_VSERVER_INFO "$2" APPDIR vshelper) || _vgh_appdir=
979
980     declare -a _vgh_search_list=( X )
981     ## HACK: when we are in 'set -u' mode, empty lists are causing errors
982
983     test -z "$_vgh_appdir" || _vgh_search_list=( "${_vgh_search_list[@]}" -x "$_vgh_appdir/$3" )
984     test -z "$_vgh_appdir" || _vgh_search_list=( "${_vgh_search_list[@]}" -e "$_vgh_appdir/action" )
985     _vgh_search_list=( "${_vgh_search_list[@]}" -x "$__CONFDIR"/.defaults/apps/vshelper/"$3" )
986     _vgh_search_list=( "${_vgh_search_list[@]}" -e "$__CONFDIR"/.defaults/apps/vshelper/action )
987
988     ! vshelper._getHandlerInternal "$1" "${_vgh_search_list[@]}" || return 0
989     eval $1=':restart'
990 }
991
992 ## Usage: vshelper.init <vserver> [<method> <args>*]
993 function vshelper.doInit
994 {
995     vshelper.isEnabled || return 0
996     
997     local xid
998     xid=$($_VSERVER_INFO "$1" CONTEXT false) && test -n "$xid" || {
999         warning $"vshelper.init: can not determine xid of vserver '$vserver'; returned value was '$xid'
1000
1001 This usually means that you're using an init-less init-style, but the
1002 guest isn't configured to start any service. Try enabling a service,
1003 changing the init-style, or making the contexts persistent."
1004         return 1
1005     }
1006
1007     local f="$__VSHELPERSTATEDIR/$xid"
1008     
1009     set -C
1010     $_RM -f "$f"
1011     echo "$1" >"$f"
1012     set +C
1013     
1014     if test -n "$2"; then
1015         shift 1
1016         local i
1017         for i; do
1018             echo "$i" 
1019         done
1020     else
1021         echo "default"
1022     fi >>"$f"
1023
1024     return 0
1025 }
1026
1027 ## Usage: vshelper.doDestroy <vserver> <xid>
1028 function vshelper.doDestroy
1029 {
1030     vshelper.isEnabled || return 0
1031
1032     $_RM -f "$__VSHELPERSTATEDIR/$2"
1033 }
1034
1035 ## Usage: vshelper.initSync <vserver> <pipe-varname> [<method>]
1036 function vshelper.initSync
1037 {
1038     local _vis_tmpdir
1039     _vis_tmpdir=$($_MKTEMPDIR vserver-stop.XXXXXX) || {
1040         warning $"Failed to generate temporary directory for vshelper sync"
1041         return 1
1042     }
1043
1044     local _vis_fifo="$_vis_tmpdir"/pipe
1045     $_MKFIFO -m700 "$_vis_fifo"
1046     vshelper.doInit "$1" "${3:-sync}" "$_vis_fifo"
1047     eval $2=\$_vis_fifo
1048 }
1049
1050 ## Usage: vshelper.getSyncTimeout <vserver> <varname>
1051 function vshelper.getSyncTimeout
1052 {
1053     local _vgst_appdir _vgst_file _vgst_tmp
1054     _vgst_appdir=$($_VSERVER_INFO "$1" APPDIR vshelper) || _vgst_appdir=
1055
1056     findFile _vgst_file ${_vgst_appdir:+"$_vgst_appdir"/sync-timeout} "$__CONFDIR"/.defaults/apps/vshelper/sync-timeout ''
1057     test -n "$_vgst_file" || return 1
1058     read _vgst_tmp <"$_vgst_file"
1059     eval $2=\$_vgst_tmp
1060 }
1061
1062 function vshelper.initStopSync
1063 {
1064         local _iss_sync_dir=$($_MKTEMPDIR vshelper-stop-sync.XXXXXX) || {
1065                 warning $"Failed to generate directory for vshelper sync"
1066                 exit 1
1067         }
1068         $_MKFIFO -m700 "$_iss_sync_dir/pipe"
1069
1070         eval "$1"=\$_iss_sync_dir
1071         VSHELPER_STOP_SYNC="$_iss_sync_dir/pipe"
1072         export VSHELPER_STOP_SYNC
1073 }
1074
1075 function vshelper.waitForStopSync
1076 {
1077         local sync_dir=$1
1078         cat "$sync_dir/pipe" &> /dev/null
1079         rm -fr "$sync_dir"
1080 }
1081
1082 function vshelper.doStopSync
1083 {
1084         test ! -p "$VSHELPER_STOP_SYNC" || echo stopped > "$VSHELPER_STOP_SYNC"
1085 }
1086
1087 function vshelper.isStopSync
1088 {
1089         test -p "$VSHELPER_STOP_SYNC" || return 1
1090         return 0
1091 }
1092
1093 function _rpmFake.getCapFlags
1094 {
1095     local ctx=$1
1096     
1097     if test -n "$ctx" && ! $_VSERVER_INFO - FEATURE migrate; then
1098         set -- $($_CHCONTEXT_COMPAT --xid 1 \
1099             $_SH -c "$_CAT /proc/[0-9]*/status | $_EGREP '^(CapBset|s_context|ctxflags)'" | \
1100             $_GREP -B 1 -A 1 "^s_context: $ctx " | \
1101             $_SED -e '1,3p;d' | $_AWK '{ print $2 }')
1102     else
1103         set --
1104     fi
1105
1106     if test -n "$3"; then
1107         RPM_FAKE_CAP=$[ ~0x$1 ]
1108         RPM_FAKE_FLAGS=$3
1109     else
1110         RPM_FAKE_CAP=$[ ~0xd40c04ff ]
1111         RPM_FAKE_FLAGS=4
1112     fi
1113 }
1114
1115 function rpmFake.init
1116 {
1117     local vdir ctx
1118     
1119     vdir=$($_VSERVER_INFO "$1" VDIR)    || vdir="$1"
1120     ctx=$($_VSERVER_INFO  "$1" CONTEXT) || ctx=
1121
1122     test -d "$vdir" ||
1123         panic $"Can not find chroot environment at '$vdir' for '$1'"
1124
1125     _rpmFake.getCapFlags "$ctx"
1126
1127     RPM_FAKE_CHROOT=$vdir
1128     RPM_FAKE_CTX=$ctx
1129 }
1130
1131 function rpmFake.exec
1132 {
1133     export RPM_FAKE_CHROOT RPM_FAKE_CTX RPM_FAKE_CAP RPM_FAKE_FLAGS
1134     
1135     LD_PRELOAD=$_RPM_FAKE_SO${LD_PRELOAD:+:$LD_PRELOAD} \
1136         exec "$@"
1137 }