- getAllVservers: use _tmp instead of tmp to avoid clashes when caller
[util-vserver.git] / util-vserver / scripts / 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 _VS_NEWLINE='
19 '
20 declare -r _VS_NEWLINE=${_VS_NEWLINE:0-1}
21
22 function findObject
23 {
24     local mod=$1
25     local var=$2
26     local file=
27     local i=X
28     shift 2
29     
30     for i; do
31         test "$i"        || continue
32         test ! $mod "$i" || { file=$i; break; }
33     done
34
35     test -z "$i" -o "$file" || {
36         echo "Can not find file for '$var'; aborting"
37         exit 1
38     } >&2
39
40     eval "$var=\"$file\""
41 }
42
43 function findFile
44 {
45     findObject -f "$@"
46 }
47
48 function findDir
49 {
50     findObject -d "$@"
51 }
52
53 function findAndCopy
54 {
55     local dst=$1
56     test ! -s "$dst"         || return 0
57     
58     local tmp
59     shift
60     findFile tmp "$@"
61
62     test "$tmp" -a -s "$tmp" || return 0
63     cp -af "$tmp" "$dst"
64 }
65
66 function getPhysicalDir
67 {
68     ( set -P && cd "$1" && pwd )
69 }
70
71 function _pkgMountBindDir()
72 {
73     test "$1" != "$2" || return 0
74
75     mount -n --bind "$1" "$2"
76 }
77
78 function _pkgSetVarsBase
79 {
80     case "$vserver" in
81         /*)
82             if test -d "$vserver/vdir"; then
83                 BASEDIR=$vserver
84                 VDIR=$(getPhysicalDir "$vserver/vdir")
85                 
86                 PKGDIR=$BASEDIR/apps/pkgmgmt
87                 test -d "$PKGDIR" || {
88                     echo "Can not find configuration-directory for package-managment tools"
89                     exit 1
90                 }
91                 findDir EXECDIR      $PKGDIR/execdir     /
92             else
93                 VDIR=$(getPhysicalDir "$vserver")
94                 PKGDIR=
95             fi
96             ;;
97         *)
98             BASEDIR=$CONFDIR/$vserver
99             test -d "$BASEDIR" || {
100                 echo "Can not find configuration-directory"
101                 exit 1
102             }
103             
104             VDIR=$BASEDIR/vdir
105             test -d "$VDIR"   || VDIR=/vservers/$vserver
106             VDIR=$(getPhysicalDir "$VDIR")
107             
108             PKGDIR=$BASEDIR/apps/pkgmgmt
109             test -d "$PKGDIR" || {
110                 echo "Can not find configuration-directory for package-managment tools"
111                 exit 1
112             }
113
114             findDir EXECDIR      $PKGDIR/execdir     /
115
116             ;;
117     esac
118
119     if test -z "$WORKAROUND_106057"; then
120         _rpmdb_mntpoint=/dev
121     else
122         _rpmdb_mntpoint=/.rpmdb
123     fi
124 }
125
126 function _pkgSetVarsRPM
127 {
128     if test "$PKGDIR"; then
129         findDir RPMETCDIR    $PKGDIR/rpmetc      $PKGDIR/base/rpm/etc       /etc/rpm
130         findDir RPMSTATEDIR  $PKGDIR/rpmstate    $PKGDIR/base/rpm/state
131
132         findDir RPMLIBDIR    $PKGDIR/rpmlib      /
133
134     else
135         findDir RPMETCDIR    "$VDIR"/etc/rpm     /etc/rpm
136         findDir RPMSTATEDIR  "$VDIR"/var/lib/rpm
137         RPMLIBDIR=/
138     fi
139     
140     RPMSTATEDIR=$(getPhysicalDir "$RPMSTATEDIR")
141     RPMETCDIR=$(getPhysicalDir "$RPMETCDIR")
142 }
143
144 function _pkgSetVarsApt
145 {
146     if test "$PKGDIR"; then
147         findDir APTETCDIR    $PKGDIR/aptetc      $PKGDIR/base/apt/etc       /etc/apt
148         findDir APTSTATEDIR  $PKGDIR/aptstate    $PKGDIR/base/apt/state
149         findDir APTCACHEDIR  $PKGDIR/aptcache    $PKGDIR/base/apt/cache
150         findDir APTARCHIVDIR $PKGDIR/aptarchives $PKGDIR/base/apt/archives  /var/cache/apt/archives
151     else
152         findDir APTETCDIR    "$VDIR"/etc/apt            /etc/apt
153         findDir APTSTATEDIR  "$VDIR"/var/state/apt
154         findDir APTCACHEDIR  "$VDIR"/var/cache/apt
155         findDir APTARCHIVDIR "$VDIR"/var/cache/apt/archives /var/cache/apt/archives
156     fi
157 }
158
159 function _pkgMountBase
160 {
161     :
162 }
163
164 function _pkgMountApt
165 {
166     :
167 }
168
169 function _pkgMountRPM
170 {
171     _pkgMountBindDir "$RPMETCDIR" /etc/rpm
172     test "$RPMLIBDIR" = "/" || _pkgMountBindDir "$RPMLIBDIR" /usr/lib/rpm
173
174     "$_SECURE_MOUNT" --chroot "$VDIR" -n --secure --bind "$RPMSTATEDIR" "$_rpmdb_mntpoint"
175     test -z "$WORKAROUND_106057" || mount -n --bind "$RPMSTATEDIR" "$_rpmdb_mntpoint"
176 }
177
178 function _pkgSetEnvBase
179 {
180     test "$EXECDIR"   = "/" || {
181         PATH=$EXECDIR:$PATH
182         LD_LIBRARY_PATH=$EXECDIR${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
183     }
184
185     export PATH LD_LIBRARY_PATH
186 }
187
188 function _pkgSetEnvApt
189 {
190     :
191 }
192
193 function _pkgSetEnvRPM
194 {
195     CUR_VSERVER=$vserver
196     RPM_FAKE_NAMESPACE_MOUNTS=$_rpmdb_mntpoint
197     RPM_BINARY=$_VRPM_PRELOAD
198
199     export CUR_VSERVER RPM_FAKE_NAMESPACE_MOUNTS RPM_BINARY
200 }
201
202 function pkgInit
203 {
204     local i
205     local vserver=$1
206     shift
207     
208     _pkgSetVarsBase
209     for i; do
210         case "$i" in
211             rpm)        _pkgSetVarsRPM;;
212             apt)        _pkgSetVarsApt;;
213             *)          echo "Unknown packaging flavor"; exit 1;;
214         esac
215     done
216
217     _pkgMountBase
218     for i; do
219         case "$i" in
220             rpm)        _pkgMountRPM;;
221             apt)        _pkgMountApt;;
222         esac
223     done
224
225     _pkgSetEnvBase
226     for i; do
227         case "$i" in
228             rpm)        _pkgSetEnvRPM;;
229             apt)        _pkgSetEnvApt;;
230         esac
231     done
232
233     _PKG_FLAVORS="$@"
234     _PKG_VSERVER=$vserver
235 }
236
237 function getAllVservers
238 {
239     local i
240     declare -a _tmp=()
241
242     for i in $CONFDIR/*; do
243         test   -d "$i"          || continue
244         test ! -e "$i"/disabled || continue
245         test   -d "$i"/vdir     || continue
246
247         _tmp=( "${_tmp[@]}" "${i##$CONFDIR/}")
248     done
249
250     eval $1='( "${_tmp[@]}" )'
251 }
252
253 ## Usage: getVserverCtx <vdir> <result-varname> [<procnumber-varname> [<do-cleanup>]]
254 function getVserverStatus
255 {
256     test -r "$1"/run || return 1
257     eval read "$2" <"$1"/run
258
259     test "$3"        || return 0
260     local _tmp=$($_VPS ax | $_AWK '{print $2}' | $_GREP -x "$ctx" | $_WC -l )
261     eval "$3"=\$_tmp
262
263     test "$4" -a $_tmp = 0 || return 0
264     _tmp=$(readlink "$1/run")
265     test "$_tmp"           || return 1
266     rm -f "$_tmp"
267     return 0
268 }
269
270 ## Usage: isVserverRunning <vdir> [<ctx-varname>]
271 function isVserverRunning
272 {
273     local ctx procnum
274
275     getVserverStatus "$1" ctx procnum 1 || return 1
276     test $procnum != 0                  || return 1
277     test -z "$2" || eval "$2"=\$ctx
278     return 0
279 }
280
281 ## Called as 'getFileValue <varname> <filename>'
282 function getFileValue
283 {
284     test -r "$2" || return 0
285     eval read "$1" <"$2"
286 }
287
288 ## Called as 'getFileArray <varname> <filename>'
289 function getFileArray
290 {
291     test -r "$2" || return 0
292     
293     local IFS=$_VS_NEWLINE
294     eval "$1"='( $(< "$2") )'
295 }
296
297 function checkComponents
298 {
299     local       i
300     local       msg=$1
301     local       x_failed=
302
303     shift
304     
305     for i; do
306         local failed=
307         case "$i" in
308             core)       test -x "$_CHBIND"           || failed=1;;
309             build)      test -x "$_VSERVER_BUILD"    || failed=1;;
310             sysv)       test -x "$INITRDDIR/vserver" || failed=1;;
311             devel)      test -d "$INCLUDEDIR/vserver.h" || failed=1;;
312             *)          echo "Unknown component '$i'"
313                         return false
314                         ;;
315         esac
316
317         test -z "$failed" || {
318             echo "$msg: $i"
319             x_failed=1
320         }
321     done
322
323     test -z "$x_failed"
324 }