basic 'vserver ... start|stop' function should work now...
[util-vserver.git] / util-vserver / scripts / vserver.functions
1 #! /bin/bash
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
37 INTERFACE_CMDS_IDX=0
38
39 INITSTYLE=sysv
40
41 S_CONTEXT=
42
43 SILENT_OPT=
44 _NEWLINE='
45 '
46 _NEWLINE=${_NEWLINE:0-1}
47
48 : ${VSERVER_NAME:=$(basename "$VSERVER_DIR")}
49
50 if test -e "$VSERVER_DIR"/noisy; then
51     SILENT_OPT=
52 else
53     SILENT_OPT='--silent'
54 fi
55
56 ## Called as '_getFileValue <varname> <filename>'
57 function _getFileValue
58 {
59     test -r "$2" || return 0
60     eval read "$1" <"$2"
61 }
62
63 ## Called as '_getFileArray <varname> <filename>'
64 function _getFileArray
65 {
66     test -r "$2" || return 0
67     
68     local IFS=$_NEWLINE
69     eval "$1"='( $(< "$2") )'
70 }
71
72 function _generateChbindOptions
73 {
74     local vdir="$1"
75     local i
76     local bcast=
77
78     local f=$vdir/interfaces/bcast
79     _getFileValue bcast "$f"
80     
81     CHBIND_OPTS=( $SILENT_OPT ${bcast:+--bcast "$bcast"} )
82
83     for i in "${INTERFACES[@]}"; do
84         CHBIND_OPTS=( "${CHBIND_OPTS[@]}" --ip "$i" )
85     done
86 }
87
88 function _generateNiceCommand
89 {
90     local vdir=$1
91     local nice
92
93     test -r "$vdir/nice" || return 0;
94     read nice <"$vdir"/nice
95
96     NICE_CMD=( $_NICE -$nice )
97 }
98
99 function _generateCapabilityOptions
100 {
101     local vdir=$1
102     local cap
103
104     test -e "$vdir"/capabilities || return 0
105
106     CAP_OPTS=()
107     CAPCHROOT_OPTS=()
108
109     while read cap; do
110         case x"$cap" in
111             x|x#)       ;;
112             !CAP_SYSCHROOT)
113                 CAP_OPTS=( "${CAP_OPTS[@]}" --cap "$cap" )
114                 CAPCHROOT_OPTS=( "${CAPCHROOT_OPTS[@]}" --nochroot )
115                 ;;
116             *)
117                 CAP_OPTS=( "${CAP_OPTS[@]}" --cap "$cap" );;
118         esac
119     done <"$vdir"/capabilities
120 }
121
122
123 function _generateInitOptions
124 {
125     local vdir=$1
126     local cfgdir=$vdir/apps/init
127     local runlevel_start=3
128     local runlevel_stop=6
129     local i f
130
131     INITCMD_START=()
132     INITCMD_STOP=()
133     INITCMD_START_SYNC=()
134     INITCMD_STOP_SYNC=()
135     INITCMD_PREPARE=()
136     
137     CHCONTEXT_INIT_OPTS=()
138     
139     _getFileValue INITSTYLE      "$cfgdir"/style
140     _getFileValue runlevel_start "$cfgdir"/runlevel
141     _getFileValue runlevel_start "$cfgdir"/runlevel.start
142     _getFileValue runlevel_stop  "$cfgdir"/runlevel.stop
143
144     case x"$INITSTYLE" in
145         xsysv)
146             for i in /etc/init.d/rc /etc/rc.d/rc; do
147                 test -x "$vdir/vdir/$i" || continue
148                 INITCMD_START=( "$i" "$runlevel_start" )
149                 INITCMD_STOP=(  "$i" "$runlevel_stop"  )
150             done
151             INITCMD_PREPARE=( $_FAKE_RUNLEVEL "$runlevel_start" /var/run/utmp )
152             ;;
153             
154         xplain)
155             INITCMD_START=( /sbin/init )
156             INITCMD_STOP=(  /sbin/init )
157             CHCONTEXT_INIT_OPTS=( --disconnect --flag fakeinit )
158             test -z "$runlevel_start" || INITCMD_START=( "${INITCMD_START[@]}" "$runlevel_start" )
159             test -z "$runlevel_stop"  || INITCMD_STOP=(  "${INITCMD_STOP[@]}"  "$runlevel_stop"  )
160             ;;
161             
162         xminit)
163             INITCMD_START=( /sbin/minit-start )
164             INITCMD_STOP=(  /sbin/minit-stop  )
165             CHCONTEXT_INIT_OPTS=( --disconnect --flag fakeinit )
166             INITCMD_START_SYNC=( "$_INITSYNC_MINIT_START" "$vdir" )
167             INITCMD_STOP_SYNC=(  "$_INITSYNC_MINIT_STOP"  "$vdir" )
168             test -z "$runlevel_start" || INITCMD_START=( "${INITCMD_START[@]}" "$runlevel_start" )
169             test -z "$runlevel_stop"  || INITCMD_STOP=(  "${INITCMD_STOP[@]}"  "$runlevel_stop"  )
170             ;;
171
172         x)  ;;
173         *)  echo "Unknown init-style '$INITSTYLE'; aborting" >&2;
174             exit 1;;
175     esac
176
177     _getFileArray INITCMD_START      "$cfgdir"/cmd.start
178     _getFileArray INITCMD_STOP       "$cfgdir"/cmd.stop
179     _getFileArray INITCMD_START_SYNC "$cfgdir"/cmd.start-sync
180     _getFileArray INITCMD_STOP_SYNC  "$cfgdir"/cmd.stop-sync
181     _getFileArray INITCMD_PREPARE    "$cfgdir"/cmd.prepare
182     
183     test -e "$cfgdir"/sync -o "$OPTION_FORCE_SYNC" || {
184         INITCMD_START_SYNC=()
185         INITCMD_STOP_SYNC=()
186     }
187 }
188
189 function _generateFlagOptions
190 {
191     local vdir=$1
192
193     CHCONTEXT_FLAG_OPTS=()
194
195     test -e "$vdir"/flags || return 0
196
197     while read flag; do
198         case x"$flag" in
199             x|x#)               ;;
200             dummy)              ;;
201             fakeinit)
202                 CHCONTEXT_FLAG_OPTS=( "${CHCONTEXT_FLAG_OPTS[@]}" 
203                                       --disconnect --flag --fakeinit )
204                 ;;
205             *)
206                 CHCONTEXT_FLAG_OPTS=( "${CHCONTEXT_FLAG_OPTS[@]}"
207                                       --flag "$flag" )
208                 ;;
209         esac
210     done <"$vdir"/flags
211 }
212
213 function _generateChcontextOptions
214 {
215     local vdir=$1
216     local ctx hostname domainname
217     local cap_opts
218     local flag
219
220     {
221         read ctx        <"$vdir"/context
222         read hostname   <"$vdir"/hostname
223         read domainname <"$vdir"/domainname
224     } 2>/dev/null
225
226     test -z "$S_CONTEXT" || ctx=$S_CONTEXT
227
228     _generateCapabilityOptions "$vdir"
229     _generateFlagOptions       "$vdir"
230
231     CHCONTEXT_OPTS=( $SILENT_OPT \
232                      "${CHCONTEXT_FLAG_OPTS[@]}" \
233                      "${CHCONTEXT_INIT_OPTS[@]}" \
234                      "${CAP_OPTS[@]}" \
235                      --secure
236                      ${ctx:+--ctx "$ctx"} \
237                      ${hostname:+--hostname "$hostname"} \
238                      ${domainname:+--domainname "$domainname"} )
239 }
240
241 function _getInterfaceValue
242 {
243     local val=$1
244     local dflt=$2
245     shift 2
246     
247     local i
248     local tmp
249
250     for i; do
251         read tmp  <"$i/$val" && break || :
252     done 2>/dev/null
253
254     : ${tmp:=$dflt}
255     eval $val=\$tmp
256 }
257
258 ## Usage: _transformMask2Prefix <result-varname> <prefix> <mask>
259 function _transformMask2Prefix
260 {
261     local tmp=$2
262     
263     test "$tmp" || {
264         $_MASK2PREFIX "$3" || tmp=$?
265     }
266
267     eval $1=\$tmp
268     return 0
269 }
270
271 function _addInterfaceCmd
272 {
273     eval INTERFACE_CMDS_${INTERFACE_CMDS_IDX}='( "$@" )'
274     let ++INTERFACE_CMDS_IDX
275 }
276
277 ## Usage: _processSingleInterface <interface-directory>
278 function _processSingleInterface
279 {
280     local iface=$1
281
282     local ip
283     local dev
284     local prefix
285     local mask
286     local bcast
287     local name
288     local scope
289
290     _getInterfaceValue ip     '' "$iface"
291     _getInterfaceValue dev    '' "$iface" "$iface/.."
292     _getInterfaceValue prefix '' "$iface" "$iface/.."
293     _getInterfaceValue mask   '' "$iface" "$iface/.."
294     _getInterfaceValue bcast  '' "$iface" "$iface/.."
295     _getInterfaceValue name   '' "$iface"
296     _getInterfaceValue scope  '' "$iface" "$iface/.."
297
298     test "$ip" || { echo $"Can not read ip for '$iface'"; return 1; }
299     
300     test ! -e "$iface"/only_ip   || unset dev
301     
302     if test "$dev"; then
303         _transformMask2Prefix prefix "$prefix" "$mask"
304
305         case "$dev" in
306             *.*)
307                 test -d /proc/net/vlan || {
308                     echo -e $"VLAN device-name used, but vlan subsystem not enabled.\nTry to execute 'modprobe 8021q' before starting the vservers"
309                     return 1
310                 }
311                 test -f /proc/net/vlan || {
312                     _addInterfaceCmd VCONFIG "${dev/./ }"
313                     _addInterfaceCmd IP_ADDR 127.0.0.1/8 broadcast 127.255.255.255 dev "$dev"
314                     _addInterfaceCmd IP_LINK "$dev"
315                 }
316                 ;;
317         esac
318
319         INTERFACES=( "${INTERFACES[@]}" "$ip${prefix:+/$prefix}" )
320         _addInterfaceCmd IP_ADDR  "$ip${prefix:+/$prefix}" broadcast ${bcast:-+} ${name:+label "$dev:$name"} dev "$dev"
321         #_addInterfaceCmd IP_ROUTE "$ip${prefix:+/$prefix}" dev "$dev"
322         _addInterfaceCmd IP_LINK  "$dev"
323     fi
324 }
325
326 ## Usage: _generateInterfaceOptions <vserver-directory>
327 function _generateInterfaceOptions
328 {
329     local iface
330
331     for iface in "$1/interfaces/"*; do
332         test   -d "$iface"          || continue
333         test ! -e "$iface"/disabled || continue
334     
335         _processSingleInterface "$iface"
336     done
337 }
338
339 function enableInterfaces
340 {
341     local i=0
342     declare -a var
343
344     while test $i -lt $INTERFACE_CMDS_IDX; do
345         eval var='( "${INTERFACE_CMDS_'$i'[@]}" )'
346         local type=${var[0]}
347         unset var[0]
348         
349         case "$type" in
350             VCONFIG)    $_VCONFIG  add "${var[@]}";;
351             IP_ADDR)    $_IP addr  add "${var[@]}";;
352             IP_LINK)    $_IP link  set "${var[@]}" up;;
353             IP_ROUTE)   $_IP route add "${var[@]}";;
354             *)          echo "Unknown interface-command type '$type'" >&2; false;;
355         esac
356
357         let ++i
358     done
359 }
360
361 function disableInterfaces
362 {
363     local i=$INTERFACE_CMDS_IDX
364     declare -a var
365
366     while test $i -gt 0; do
367         let --i
368
369         eval var='( "${INTERFACE_CMDS_'$i'[@]}" )'
370         local type=${var[0]}
371         unset var[0]
372         
373         case "$type" in
374             VCONFIG)    $_VCONFIG  rem "${var[@]}";;
375             IP_ADDR)    $_IP addr  del "${var[@]}";;
376             IP_LINK)    ;; ## Ignore the link-down command for now
377             IP_ROUTE)   $_IP route del "${var[@]}";;
378             *)          echo "Unknown interface-command type '$type'" >&2; false;;
379         esac
380     done
381 }
382
383 ## Usage: prepareInit <vserver-directory>
384 function prepareInit
385 {
386     pushd "$1/vdir" >/dev/null
387     case "$INITSTYLE" in
388         sysv)
389             { find var/run  ! -type d -print0; \
390               find var/lock ! -type d -print0; } | xargs -0r $_CHROOT_RM
391             ;;
392         minit)
393             ;;
394     esac
395     "${INITCMD_PREPARE[@]}"
396     popd >/dev/null
397 }
398
399 function generateOptions
400 {
401     _generateInterfaceOptions  "$1"
402     _generateChbindOptions     "$1"
403     _generateNiceCommand       "$1"
404     _generateInitOptions       "$1"
405     _generateChcontextOptions  "$1"
406 }
407
408 function mountVserver
409 {
410     local cfgdir=$1
411     local vdir=$1/vdir
412
413     test -e "$cfgdir"/fstab || return 0
414     $_CHBIND "${CHBIND_OPTS[@]}" \
415         $_SECURE_MOUNT -a --chroot "$vdir" --fstab "$cfgdir"/fstab
416 }
417
418 function umountVserver
419 {
420     local cfgdir=$1
421     local vdir=$1/vdir
422
423     is_ok=1
424
425     test -e "$cfgdir"/fstab || return 0
426     
427     pushd "$vdir/" >/dev/null
428     tac "$cfgdir"/fstab | while read src dst tmp; do
429         test "$tmp" || continue
430         case x"$src" in
431             x#*)        continue;;
432         esac
433
434         $_CHBIND "${CHBIND_OPTS[@]}" \
435             $_SECURE_UMOUNT "$dst" || is_ok=
436     done
437     popd >/dev/null
438
439     test "$is_ok"
440 }
441
442 function _sourceWrap
443 {
444     local vdir name flavor start i
445     . "$@"
446 }
447
448 ## Usage: execScriptlets <vserver-cfgdir> <vserver-name> <script-flavor>
449 function execScriptlets
450 {
451     local vdir=$1
452     local name=$2
453     local flavor=$3
454     local start=
455     local i
456
457     for i in "$vdir/scripts/$flavor" "$vdir/scripts/$flavor.d"/*; do
458         test -f "$i" -a -r "$i" || continue
459         test -x "$i" || start=_sourceWrap
460         $start "$i" "$flavor" "$name"
461     done
462 }