added gentoo startup style
[util-vserver.git] / util-vserver / scripts / vserver.functions
1 # $Id$  --*- sh -*--
2
3 # Copyright (C) 2003 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 ## Expected env:
19 #  $VSERVER_DIR   ... path to vserver-cfg dir
20 #  $VSERVER_NAME  ... name of vserver
21
22 declare -a NICE_CMD=()
23 declare -a CHBIND_OPTS=()
24 declare -a CAP_OPTS=()
25 declare -a CHCONTEXT_INIT_OPTS=()
26 declare -a CHCONTEXT_FLAG_OPTS=()
27 declare -a CHCONTEXT_OPTS=()
28 declare -a CAPCHROOT_OPTS=()
29 declare -a INTERFACES=()
30
31 declare -a INITCMD_START=()
32 declare -a INITCMD_START_SYNC=()
33 declare -a INITCMD_STOP=()
34 declare -a INITCMD_STOP_SYNC=()
35 declare -a INITCMD_PREPARE=()
36 declare -a INITKILL_SEQ=()
37 declare -a ENTER_SHELL=()
38
39 declare -a OPTS_VCONTEXT_CREATE=()
40 declare -a OPTS_VCONTEXT_MIGRATE=()
41 declare -a OPTS_VATTRIBUTE=()
42 USE_VNAMESPACE=
43
44 INTERFACE_CMDS_IDX=0
45 _HAVE_INTERFACE_OPTIONS=
46 _HAVE_CHBIND_OPTIONS=
47
48 INITSTYLE=sysv
49
50 S_CONTEXT=
51
52 SILENT_OPT=
53
54 : ${VSERVER_NAME:=$(basename "$VSERVER_DIR")}
55
56 if test -e "$VSERVER_DIR"/noisy -o "$OPTION_VERBOSE"; then
57     SILENT_OPT=
58 else
59     SILENT_OPT='--silent'
60 fi
61
62 function _generateChbindOptions
63 {
64     local vdir="$1"
65     local i
66     local bcast=
67
68     test "$_HAVE_INTERFACE_OPTIONS" || _generateInterfaceOptions "$vdir"
69
70     local f=$vdir/interfaces/bcast
71     getFileValue bcast "$f"
72     
73     CHBIND_OPTS=( $SILENT_OPT ${bcast:+--bcast "$bcast"} )
74
75     for i in "${INTERFACES[@]}"; do
76         CHBIND_OPTS=( "${CHBIND_OPTS[@]}" --ip "$i" )
77     done
78
79     _HAVE_CHBIND_OPTIONS=1
80 }
81
82 function _generateNiceCommand
83 {
84     local vdir=$1
85     local nice
86
87     test -r "$vdir/nice" || return 0;
88     read nice <"$vdir"/nice
89
90     NICE_CMD=( $_NICE -$nice )
91 }
92
93 function _generateCCapabilityOptions
94 {
95     local vdir=$1
96     local cap
97     local f="$vdir"/ccapabilities
98     
99     test -e "$f" || return 0
100     while read cap; do
101         case x"$cap" in
102             (x|x\#)     ;;
103             (*)         OPTS_VATTRIBUTE=( "${OPTS_VATTRIBUTE[@]}" --ccap "$cap" );;
104         esac
105     done <"$f"
106 }
107
108 function _generateBCapabilityOptions
109 {
110     local vdir=$1
111     local cap
112     local f="$vdir"/bcapabilities
113     
114     test -e "$f" || return 0
115     while read cap; do
116         case x"$cap" in
117             (x|x\#)     ;;
118             (*)         OPTS_VATTRIBUTE=( "${OPTS_VATTRIBUTE[@]}" --bcap "$cap" );;
119         esac
120     done <"$f"
121 }
122
123 function _generateCapabilityOptions
124 {
125     local vdir=$1
126     local cap
127
128     _generateBCapabilityOptions "$vdir"
129     _generateCCapabilityOptions "$vdir"
130     
131     test -e "$vdir"/capabilities || return 0
132
133     CAP_OPTS=()
134     CAPCHROOT_OPTS=()
135
136     while read cap; do
137         case x"$cap" in
138             (x|x\#)     ;;
139             (!CAP_SYSCHROOT)
140                 CAP_OPTS=( "${CAP_OPTS[@]}" --cap "$cap" )
141                 CAPCHROOT_OPTS=( "${CAPCHROOT_OPTS[@]}" --nochroot )
142                 ;;
143             (*)
144                 CAP_OPTS=( "${CAP_OPTS[@]}" --cap "$cap" )
145                 ;;
146         esac
147     done <"$vdir"/capabilities
148 }
149
150 function getEnterShell
151 {
152     local vdir=$1
153
154     ENTER_SHELL=()
155     
156     getFileValue ENTER_SHELL     "$vdir"/shell
157     
158     test "$ENTER_SHELL" || {
159         local i
160         for i in "/bin/bash -login" "/bin/sh -l" /bin/csh; do
161             set -- $i
162             test -x "$vdir/vdir/$1" || continue
163             ENTER_SHELL=( "$@" )
164             break
165         done
166     }
167 }
168
169 ## Usage: sendKillSequence <ctx> <signal> [<wait> <signal>]*
170 function sendKillSequence
171 {
172     local ctx=$1
173     local wait=
174     shift
175
176     while isCtxRunning "$ctx"; do
177         test -z "$wait" || sleep "$wait"
178         
179         $_VKILL -c "$ctx" ${1:+-s "$1"}
180         test "$2" || break
181         wait="$2"
182         shift 2
183     done
184 }
185
186 function _generateInitOptions
187 {
188     local vdir=$1
189     local cfgdir=$vdir/apps/init
190     local runlevel_start
191     local runlevel_stop
192     local i f
193
194     INITCMD_START=()
195     INITCMD_STOP=()
196     INITCMD_START_SYNC=()
197     INITCMD_STOP_SYNC=()
198     INITCMD_PREPARE=()
199
200     INITKILL_SEQ=( 15 5 9 )
201     CHCONTEXT_INIT_OPTS=()
202
203
204     getFileValue INITSTYLE      "$cfgdir"/style
205     getFileValue runlevel_start "$cfgdir"/runlevel
206     getFileValue runlevel_start "$cfgdir"/runlevel.start
207     getFileValue runlevel_stop  "$cfgdir"/runlevel.stop
208     getFileArray INITKILL_SEQ   "$cfgdir"/killseq
209
210     case x"$INITSTYLE" in
211         (xsysv)
212             test "$runlevel_start" || runlevel_start=3
213             test "$runlevel_stop"  || runlevel_stop=6
214
215             for i in /etc/init.d/rc /etc/rc.d/rc; do
216                 test -x "$vdir/vdir/$i" || continue
217                 INITCMD_START=( "$i" "$runlevel_start" )
218                 INITCMD_STOP=(  "$i" "$runlevel_stop"  )
219             done
220             INITCMD_PREPARE=( $_FAKE_RUNLEVEL "$runlevel_start" /var/run/utmp )
221             ;;
222             
223         (xplain)
224             INITCMD_START=( /sbin/init )
225             INITCMD_STOP=(  /sbin/init )
226             CHCONTEXT_INIT_OPTS=( --disconnect --flag fakeinit )
227             OPTS_VCONTEXT_MIGRATE=( "${OPTS_VCONTEXT_MIGRATE[@]}" --fakeinit --disconnect )
228             test -z "$runlevel_start" || INITCMD_START=( "${INITCMD_START[@]}" "$runlevel_start" )
229             test -z "$runlevel_stop"  || INITCMD_STOP=(  "${INITCMD_STOP[@]}"  "$runlevel_stop"  )
230             ;;
231             
232         (xminit)
233             INITCMD_START=( /sbin/minit-start )
234             INITCMD_STOP=(  /sbin/minit-stop  )
235             CHCONTEXT_INIT_OPTS=( --disconnect --flag fakeinit )
236             INITCMD_START_SYNC=( "$_INITSYNC_MINIT_START" "$vdir" )
237             INITCMD_STOP_SYNC=(  "$_INITSYNC_MINIT_STOP"  "$vdir" )
238             OPTS_VCONTEXT_MIGRATE=( "${OPTS_VCONTEXT_MIGRATE[@]}" --fakeinit --disconnect )
239             test -z "$runlevel_start" || INITCMD_START=( "${INITCMD_START[@]}" "$runlevel_start" )
240             test -z "$runlevel_stop"  || INITCMD_STOP=(  "${INITCMD_STOP[@]}"  "$runlevel_stop"  )
241             ;;
242
243         (xgentoo)
244             INITCMD_START=( /sbin/rc default  )
245             INITCMD_STOP=(  /sbin/rc shutdown )
246             ;;
247
248         (x) ;;
249         (*) echo "Unknown init-style '$INITSTYLE'; aborting" >&2;
250             exit 1;;
251     esac
252
253     getFileArray INITCMD_START      "$cfgdir"/cmd.start
254     getFileArray INITCMD_STOP       "$cfgdir"/cmd.stop
255     getFileArray INITCMD_START_SYNC "$cfgdir"/cmd.start-sync
256     getFileArray INITCMD_STOP_SYNC  "$cfgdir"/cmd.stop-sync
257     getFileArray INITCMD_PREPARE    "$cfgdir"/cmd.prepare
258
259     
260     test -e "$cfgdir"/sync -o "$OPTION_FORCE_SYNC" || {
261         INITCMD_START_SYNC=()
262         INITCMD_STOP_SYNC=()
263     }
264 }
265
266 function _generateFlagOptions
267 {
268     local vdir=$1
269
270     CHCONTEXT_FLAG_OPTS=()
271
272     test -e "$vdir"/flags || return 0
273
274     while read flag; do
275         case x"$flag" in
276             (x|x\#)             ;;
277             (xnamespace)        ;;
278             (xfakeinit)
279                 CHCONTEXT_INIT_OPTS=( --disconnect --flag fakeinit )
280                 OPTS_VCONTEXT_MIGRATE=( "${OPTS_VCONTEXT_MIGRATE[@]}" --fakeinit --disconnect )
281                 ;;
282             (*)
283                 OPTS_VATTRIBUTE=( "${OPTS_VATTRIBUTE[@]}" --flag "$flag" )
284                 CHCONTEXT_FLAG_OPTS=( "${CHCONTEXT_FLAG_OPTS[@]}"
285                                       --flag "$flag" )
286                 ;;
287         esac
288     done <"$vdir"/flags
289
290     isAvoidNamespace "$cfgdir" || {
291         USE_VNAMESPACE=1
292         CHCONTEXT_FLAG_OPTS=( "${CHCONTEXT_FLAG_OPTS[@]}" --flag namespace )
293     }
294 }
295
296 function _generateChcontextOptions
297 {
298     local vdir=$1
299     local ctx hostname domainname
300     local cap_opts
301     local flag
302
303     {
304         read ctx        <"$vdir"/context     || :
305         read hostname   <"$vdir"/hostname    || :
306         read domainname <"$vdir"/domainname  || :
307     } 2>/dev/null
308
309     test -z "$S_CONTEXT" || ctx=$S_CONTEXT
310
311     _generateCapabilityOptions "$vdir"
312     _generateFlagOptions       "$vdir"
313
314     CHCONTEXT_OPTS=( $SILENT_OPT \
315                      "${CHCONTEXT_FLAG_OPTS[@]}" \
316                      "${CAP_OPTS[@]}" \
317                      --secure
318                      ${ctx:+--ctx "$ctx"} \
319                      ${hostname:+--hostname "$hostname"} \
320                      ${domainname:+--domainname "$domainname"} )
321
322     OPTS_VCONTEXT_CREATE=( $SILENT_OPT \
323                            ${ctx:+--xid "$ctx"} )
324     ## put '--secure' at front so that it can be overridden
325     OPTS_VATTRIBUTE=( --secure "${OPTS_VATTRIBUTE[@]}" )
326 }
327
328 function _getInterfaceValue
329 {
330     local val=$1
331     local dflt=$2
332     shift 2
333     
334     local i
335     local tmp
336
337     for i; do
338         read tmp  <"$i/$val" && break || :
339     done 2>/dev/null
340
341     : ${tmp:=$dflt}
342     eval $val=\$tmp
343 }
344
345 ## Usage: _transformMask2Prefix <result-varname> <prefix> <mask>
346 function _transformMask2Prefix
347 {
348     local tmp=$2
349     
350     test "$tmp" || {
351         $_MASK2PREFIX "$3" || tmp=$?
352     }
353
354     eval $1=\$tmp
355     return 0
356 }
357
358 function _addInterfaceCmd
359 {
360     eval INTERFACE_CMDS_${INTERFACE_CMDS_IDX}='( "$@" )'
361     let ++INTERFACE_CMDS_IDX
362 }
363
364 ## Usage: _processSingleInterface <interface-directory>
365 function _processSingleInterface
366 {
367     local iface=$1
368
369     local ip
370     local dev
371     local prefix
372     local mask
373     local bcast
374     local name
375     local scope
376
377     _getInterfaceValue ip     '' "$iface"
378     _getInterfaceValue dev    '' "$iface" "$iface/.."
379     _getInterfaceValue prefix '' "$iface" "$iface/.."
380     _getInterfaceValue mask   '' "$iface" "$iface/.."
381     _getInterfaceValue bcast  '' "$iface" "$iface/.."
382     _getInterfaceValue name   '' "$iface"
383     _getInterfaceValue scope  '' "$iface" "$iface/.."
384
385     test "$ip" || { echo $"Can not read ip for '$iface'"  >&2; return 1; }
386     
387     test ! -e "$iface"/only_ip || {
388         echo $"The 'only_ip' flag for interface '$iface' is deprecated; use 'nodev' instead of" >&2
389         unset dev
390     }
391     test ! -e "$iface"/nodev   || unset dev
392     
393     if test "$dev"; then
394         _transformMask2Prefix prefix "$prefix" "$mask"
395
396         case "$dev" in
397             (*.*)
398                 test -d /proc/net/vlan || {
399                     echo -e $"VLAN device-name used, but vlan subsystem not enabled.\nTry to execute 'modprobe 8021q' before starting the vservers"  >&2
400                     return 1
401                 }
402                 test -f /proc/net/vlan || {
403                     _addInterfaceCmd VCONFIG "${dev/./ }"
404                     _addInterfaceCmd IP_ADDR 127.0.0.1/8 broadcast 127.255.255.255 dev "$dev"
405                     _addInterfaceCmd IP_LINK "$dev"
406                 }
407                 ;;
408         esac
409
410         INTERFACES=( "${INTERFACES[@]}" "$ip${prefix:+/$prefix}" )
411         _addInterfaceCmd IP_ADDR  "$ip${prefix:+/$prefix}" broadcast ${bcast:-+} ${name:+label "$dev:$name"} dev "$dev"
412         #_addInterfaceCmd IP_ROUTE "$ip${prefix:+/$prefix}" dev "$dev"
413         _addInterfaceCmd IP_LINK  "$dev"
414     fi
415 }
416
417 ## Usage: _generateInterfaceOptions <vserver-directory>
418 function _generateInterfaceOptions
419 {
420     local iface
421
422     for iface in "$1/interfaces/"*; do
423         test   -d "$iface"          || continue
424         test ! -e "$iface"/disabled || continue
425     
426         _processSingleInterface "$iface"
427     done
428     _HAVE_INTERFACE_OPTIONS=1
429 }
430
431 function enableInterfaces
432 {
433     local i=0
434     declare -a var
435
436     while test $i -lt $INTERFACE_CMDS_IDX; do
437         eval var='( "${INTERFACE_CMDS_'$i'[@]}" )'
438         local type=${var[0]}
439         unset var[0]
440         
441         case "$type" in
442             VCONFIG)    $_VCONFIG  add "${var[@]}";;
443             IP_ADDR)    $_IP addr  add "${var[@]}";;
444             IP_LINK)    $_IP link  set "${var[@]}" up;;
445             IP_ROUTE)   $_IP route add "${var[@]}";;
446             *)          echo "Unknown interface-command type '$type'" >&2; false;;
447         esac
448
449         let ++i
450     done
451 }
452
453 function disableInterfaces
454 {
455     test "$_HAVE_INTERFACE_OPTIONS" || _generateInterfaceOptions "$1"
456
457     local i=$INTERFACE_CMDS_IDX
458     declare -a var
459
460     while test $i -gt 0; do
461         let --i || :
462
463         eval var='( "${INTERFACE_CMDS_'$i'[@]}" )'
464         local type=${var[0]}
465         unset var[0]
466         
467         case "$type" in
468             VCONFIG)    $_VCONFIG  rem "${var[@]}";;
469             IP_ADDR)    $_IP addr  del "${var[@]}";;
470             IP_LINK)    ;; ## Ignore the link-down command for now
471             IP_ROUTE)   $_IP route del "${var[@]}";;
472             *)          echo "Unknown interface-command type '$type'" >&2; false;;
473         esac
474     done
475 }
476
477 ## Usage: prepareInit <vserver-directory>
478 function prepareInit
479 {
480     pushd "$1/vdir" >/dev/null
481     case "$INITSTYLE" in
482         sysv)
483             { find var/run  ! -type d -print0; \
484               find var/lock ! -type d -print0; } | xargs -0r $_CHROOT_RM
485             ;;
486         minit)
487             ;;
488     esac
489     "${INITCMD_PREPARE[@]}"
490     popd >/dev/null
491 }
492
493 function generateOptions
494 {
495     _generateInterfaceOptions  "$1"
496     test "$_HAVE_CHBIND_OPTIONS" || _generateChbindOptions "$1" 
497     _generateNiceCommand       "$1"
498     _generateInitOptions       "$1"
499     _generateChcontextOptions  "$1"
500 }
501
502 function _mountVserverInternal
503 {
504     local fstab="$1"
505     test -e "$fstab" || return 0
506     shift
507     
508     "$@" $_SECURE_MOUNT -a --chroot "$vdir" --fstab "$fstab"
509 }
510
511 function mountVserver
512 {
513     local cfgdir=$1
514     local ns_opt=$2
515     local vdir=$1/vdir
516     local mtab_src
517
518     test -e "$cfgdir"/fstab -o \
519          -e "$cfgdir"/fstab.local || return 0
520
521     findObject -r mtab_src "$vdir"/apps/init/mtab "$CONFDIR"/.defaults/init/mtab "$PKGLIBDEFAULTDIR"/mtab /dev/null
522     
523     pushd "$vdir" >/dev/null
524     $_CHROOT_CAT /etc/mtab <"$mtab_src"
525     popd >/dev/null
526
527     test "$_HAVE_CHBIND_OPTIONS" || _generateChbindOptions "$cfgdir"
528
529     test -z "$NAMESPACE_CLEANUP" || isAvoidNamespace "$cfgdir" || \
530         $_VNAMESPACE --cleanup
531
532     _mountVserverInternal "$cfgdir"/fstab       $_CHBIND "${CHBIND_OPTS[@]}"
533     _mountVserverInternal "$cfgdir"/fstab.local
534
535     isAvoidNamespace "$cfgdir" || \
536         $_SECURE_MOUNT --rbind -n "$vdir" "/"
537 }
538
539 function _umountVserverInternal
540 {
541     local fstab="$1"
542     test -e "$fstab" || return 0
543     shift
544
545     tac "$fstab" | {
546         is_ok=1
547         while read src dst tmp; do
548             test "$tmp" || continue
549             case x"$src" in
550                 (x\#*)  continue;;
551             esac
552
553         
554             "$@" $_EXEC_CD "$dst" $_UMOUNT -lfn . || is_ok=
555         done
556         test "$is_ok"
557     }
558 }
559
560 function umountVserver
561 {
562     local cfgdir=$1
563     local vdir=$1/vdir
564     local is_ok=1
565
566     isAvoidNamespace "$cfgdir"    || return 0
567     test -e "$cfgdir"/fstab -o \
568          -e "$cfgdir"/fstab.local || return 0
569     test "$_HAVE_CHBIND_OPTIONS"  || _generateChbindOptions "$cfgdir"
570     
571     pushd "$vdir/" >/dev/null || return 1
572         _umountVserverInternal  "$cfgdir"/fstab.local                              || is_ok=
573         _umountVserverInternal  "$cfgdir"/fstab       $_CHBIND "${CHBIND_OPTS[@]}" || is_ok=
574     popd >/dev/null           || return 1
575
576     test "$is_ok"
577 }
578
579 function _sourceWrap
580 {
581     local vdir name flavor start i
582     . "$@"
583 }
584
585 ## Usage: execScriptlets <vserver-cfgdir> <vserver-name> <script-flavor>
586 function execScriptlets
587 {
588     local vdir=$1
589     local name=$2
590     local flavor=$3
591     local start=
592     local i
593
594     for i in "$vdir/scripts/$flavor" "$vdir/scripts/$flavor.d"/*; do
595         test -f "$i" -a -r "$i" || continue
596         test -x "$i" || start=_sourceWrap
597         $start "$i" "$flavor" "$name"
598     done
599 }