fix typo
[build.git] / vbuild-init-vserver.sh
1 #!/bin/bash
2 # -*-shell-*-
3 # $Id$
4
5 #shopt -s huponexit
6
7 COMMAND=$(basename $0)
8 DIRNAME=$(dirname $0)
9
10 # pkgs parsing utilities
11 PATH=$(dirname $0):$PATH . build.common
12
13 DEFAULT_FCDISTRO=f8
14 DEFAULT_PLDISTRO=planetlab
15 DEFAULT_PERSONALITY=linux32
16 DEFAULT_IFNAME=eth0
17
18 COMMAND_VBUILD="vbuild-init-vserver.sh"
19 COMMAND_MYPLC="vtest-init-vserver.sh"
20
21 function failure () {
22     echo "$COMMAND : Bailing out"
23     exit 1
24 }
25
26 # overwrite vserver's internal yum config from what is in
27 # .distributions/<distrib>/yum/yum.conf and /yum.repos.d 
28
29 function configure_yum_in_vserver () {
30     set -x 
31     set -e 
32     trap failure ERR INT
33
34     vserver=$1; shift
35     fcdistro=$1; shift
36
37     templates=/etc/vservers/.distributions/${fcdistro}
38     if [ -f ${templates}/yum/yum.conf ] ; then
39         echo "Initializing yum.conf in $vserver from ${templates}/yum"
40         sed -e "s!@YUMETCDIR@!/etc!g;
41                 s!@YUMCACHEDIR@!/var/cache/yum!g;
42                 s!@YUMLOGDIR@!/var/log!g;
43                 s!@YUMLOCKDIR@!/var/lock!g;
44                " ${templates}/yum/yum.conf > /vservers/$vserver/etc/yum.conf
45
46         # post process the various @...@ variables from this yum.conf file.
47     else
48         echo "Using $fcdistro default for yum.conf"
49     fi
50
51     if [ -d ${templates}/yum.repos.d ] ; then
52         echo "Initializing yum.repos.d in $vserver from ${templates}/yum.repos.d"
53         rm -rf /vservers/$vserver/etc/yum.repos.d
54         tar -C ${templates} -cf - yum.repos.d | tar -C /vservers/$vserver/etc -xvf -
55     else
56         echo "Cannot initialize yum.repos.d in $vserver"
57     fi
58
59     # for using vtest-init-vserver.sh as a general-purpose vserver creation wrapper
60     # just mention 'none' as the repo url
61     if [ -n "$MYPLC_MODE" -a "$REPO_URL" != "none" ] ; then
62         if [ ! -d /vservers/$vserver/etc/yum.repos.d ] ; then
63             echo "WARNING : cannot create myplc repo"
64         else
65             # exclude kernel from fedora repos 
66             for repo in /vservers/$vserver/etc/yum.repos.d/* ; do
67                 [ -f $repo ] && yumconf_exclude $repo "exclude=$pl_KEXCLUDES" 
68             done
69             # the build repo is not signed at this stage
70             cat > /vservers/$vserver/etc/yum.repos.d/myplc.repo <<EOF
71 [myplc]
72 name= MyPLC
73 baseurl=$REPO_URL
74 enabled=1
75 gpgcheck=0
76 EOF
77         fi
78     fi
79 }    
80
81 # return yum or debootstrap
82 function package_method () {
83     fcdistro=$1; shift
84     case $fcdistro in
85         f[0-9]*|centos[0-9]*) echo yum ;;
86         lenny|etch) echo debootstrap ;;
87         *) echo Unknown distro $fcdistro ;;
88     esac 
89 }
90
91 # return arch from debian distro and personality
92 function canonical_arch () {
93     personality=$1; shift
94     fcdistro=$1; shift
95     case $(package_method $fcdistro) in
96         yum)
97             case $personality in *32) echo i386 ;; *64) echo x86_64 ;; *) echo Unknown-arch-1 ;; esac ;;
98         debootstrap)
99             case $personality in *32) echo i386 ;; *64) echo amd64 ;; *) echo Unknown-arch-2 ;; esac ;;
100         *)
101             echo Unknown-arch-3 ;;
102     esac
103 }
104
105 function setup_vserver () {
106
107     set -x
108     set -e
109     trap failure ERR INT
110
111     vserver=$1; shift
112     fcdistro=$1; shift
113     personality=$1; shift
114
115     if [ -d /vservers/$vserver ] ; then
116         echo "$COMMAND : vserver $vserver seems to exist - bailing out"
117         exit 1
118     fi
119
120     pkg_method=$(package_method $fcdistro)
121     case $pkg_method in
122         yum)
123             build_options="-m yum -- -d $fcdistro" 
124             ;;
125         debootstrap)
126             arch=$(canonical_arch $personality $fcdistro)
127             build_options="-m debootstrap -- -d $fcdistro -- --arch $arch"
128             ;;
129         *)
130             build_options="something wrong" ;;
131     esac
132
133     # create it
134     # try to work around the vserver issue:
135     # vc_ctx_migrate: No such process
136     # rpm-fake.so: failed to initialize communication with resolver
137     for i in $(seq 20) ; do
138         $personality vserver $VERBOSE $vserver build $VSERVER_OPTIONS $build_options && break || true
139         echo "* ${i}-th attempt to 'vserver build' failed - waiting for 3 seconds"
140         sleep 3
141     done
142     # check success
143     [ -d /vservers/$vserver ] 
144
145     if [ ! -z "$personality" ] ; then
146         if [ -f "/etc/vservers/$vserver/personality" ] ; then
147             registered_personality=$(grep $personality /etc/vservers/$vserver/personality | wc -l)
148         else
149             registered_personality=0
150         fi
151         if [ $registered_personality -eq 0 -a "$personality" != "linux64" ] ; then
152             echo $personality >> /etc/vservers/$vserver/personality
153         fi
154     fi
155
156     if [ -n "$VBUILD_MODE" ] ; then 
157         ### capabilities required for a build vserver
158         # set up appropriate vserver capabilities to mount, mknod and IPC_LOCK
159         BCAPFILE=/etc/vservers/$vserver/bcapabilities
160         touch $BCAPFILE
161         cap=$(grep ^CAP_SYS_ADMIN /etc/vservers/$vserver/bcapabilities | wc -l)
162         [ $cap -eq 0 ] && echo 'CAP_SYS_ADMIN' >> /etc/vservers/$vserver/bcapabilities
163         cap=$(grep ^CAP_MKNOD /etc/vservers/$vserver/bcapabilities | wc -l)
164         [ $cap -eq 0 ] && echo 'CAP_MKNOD' >> /etc/vservers/$vserver/bcapabilities
165         cap=$(grep ^CAP_IPC_LOCK /etc/vservers/$vserver/bcapabilities | wc -l)
166         [ $cap -eq 0 ] && echo 'CAP_IPC_LOCK' >> /etc/vservers/$vserver/bcapabilities
167     else
168         ### capabilities required for a myplc vserver
169         # for /etc/plc.d/gpg - need to init /dev/random
170         cap=$(grep ^CAP_MKNOD /etc/vservers/$vserver/bcapabilities | wc -l)
171         [ $cap -eq 0 ] && echo 'CAP_MKNOD' >> /etc/vservers/$vserver/bcapabilities
172         cap=$(grep ^CAP_NET_BIND_SERVICE /etc/vservers/$vserver/bcapabilities | wc -l)
173         [ $cap -eq 0 ] && echo 'CAP_NET_BIND_SERVICE' >> /etc/vservers/$vserver/bcapabilities
174     fi
175
176     if [ "$pkg_method" = "yum" ] ; then
177         $personality vyum $vserver -- -y install yum
178         # ditto
179         for i in $(seq 20) ; do
180             $personality vserver $VERBOSE $vserver pkgmgmt internalize && break || true
181             echo "* ${i}-th attempt to 'vserver pkgmgmt internalize' failed - waiting for 3 seconds"
182             sleep 3
183         done
184     fi
185
186     # start the vserver so we can do the following operations
187     # redirect out/err to protect against the vserver's init sequence getting stalled 
188     # mostly used for f10 vservers created remotely through ssh
189     $personality vserver $VERBOSE $vserver start >& /dev/null
190     [ "$pkg_method" = "yum" ] && $personality vserver $VERBOSE $vserver exec sh -c "rm -f /var/lib/rpm/__db*"
191     [ "$pkg_method" = "yum" ] && $personality vserver $VERBOSE $vserver exec rpm --rebuilddb
192
193     # check if the vserver kernel is using VSERVER_DEVICE (vdevmap) support
194     need_vdevmap=$(grep "CONFIG_VSERVER_DEVICE=y" /boot/config-$(uname -r) | wc -l)
195
196     if [ $need_vdevmap -eq 1 ] ; then
197         ctx=$(cat /etc/vservers/$vserver/context)
198         vdevmap --set --xid $ctx --open --create --target /dev/null
199         vdevmap --set --xid $ctx --open --create --target /dev/root
200     fi
201             
202     # minimal config in the vserver for yum to work
203     [ "$pkg_method" = "yum" ] && configure_yum_in_vserver $vserver $fcdistro 
204
205     # set up resolv.conf
206     cp /etc/resolv.conf /vservers/$vserver/etc/resolv.conf
207     # and /etc/hosts for at least localhost
208     [ -f /vservers/$vserver/etc/hosts ] || echo "127.0.0.1 localhost localhost.localdomain" > /vservers/$vserver/etc/hosts
209
210 }
211
212 function devel_or_vtest_tools () {
213
214     set -x 
215     set -e 
216     trap failure ERR INT
217
218     vserver=$1; shift
219     fcdistro=$1; shift
220     pldistro=$1; shift
221     personality=$1; shift
222
223     pkg_method=$(package_method $fcdistro)
224
225     # check for .pkgs file based on pldistro
226     if [ -n "$VBUILD_MODE" ] ; then
227         pkgsname=devel.pkgs
228     else
229         pkgsname=vtest.pkgs
230     fi
231     pkgsfile=$(pl_locateDistroFile $DIRNAME $pldistro $pkgsname)
232
233     ### install individual packages, then groups
234     # get target arch - use uname -i here (we want either x86_64 or i386)
235     vserver_arch=$($personality vserver $vserver exec uname -i)
236     # on debian systems we get arch through the 'arch' command
237     [ "$vserver_arch" = "unknown" ] && vserver_arch=$($personality vserver $vserver exec arch)
238     
239     packages=$(pl_getPackages -a $vserver_arch $fcdistro $pldistro $pkgsfile)
240     groups=$(pl_getGroups -a $vserver_arch $fcdistro $pldistro $pkgsfile)
241
242     [ "$pkg_method" = yum ] && [ -n "$packages" ] && $personality vserver $vserver exec yum -y install $packages
243     [ "$pkg_method" = yum ] && for group_plus in $groups; do
244         group=$(echo $group_plus | sed -e "s,+++, ,g")
245         $personality vserver $vserver exec yum -y groupinstall "$group"
246     done
247
248     [ "$pkg_method" = debootstrap ] && $personality vserver $vserver exec apt-get update
249     [ "$pkg_method" = debootstrap ] && for package in $packages ; do 
250         $personality vserver $vserver exec apt-get install -y $package 
251     done
252     
253     return 0
254 }
255
256 function post_install () {
257     if [ -n "$VBUILD_MODE" ] ; then
258         post_install_vbuild "$@" 
259     else
260         post_install_myplc "$@"
261     fi
262     # setup localtime from the host
263     vserver=$1; shift 
264     cp /etc/localtime /vservers/$vserver/etc/localtime
265 }
266
267 function post_install_vbuild () {
268
269     set -x 
270     set -e 
271     trap failure ERR INT
272
273     vserver=$1; shift
274     personality=$1; shift
275
276 ### From myplc-devel-native.spec
277 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
278     cat << EOF | $personality vserver $VERBOSE $vserver exec bash -x
279     # set up /dev/loop* in vserver
280     for i in \$(seq 0 255) ; do
281         mknod -m 640 /dev/loop\$i b 7 \$i
282     done
283     
284     # create symlink for /dev/fd
285     [ ! -e "/dev/fd" ] && ln -s /proc/self/fd /dev/fd
286
287     # modify /etc/rpm/macros to not use /sbin/new-kernel-pkg
288     sed -i 's,/sbin/new-kernel-pkg:,,' /etc/rpm/macros
289     if [ -h "/sbin/new-kernel-pkg" ] ; then
290         filename=\$(readlink -f /sbin/new-kernel-pkg)
291         if [ "\$filename" == "/sbin/true" ] ; then
292                 echo "WARNING: /sbin/new-kernel-pkg symlinked to /sbin/true"
293                 echo "\tmost likely /etc/rpm/macros has /sbin/new-kernel-pkg declared in _netsharedpath."
294                 echo "\tPlease remove /sbin/new-kernel-pkg from _netsharedpath and reintall mkinitrd."
295                 exit 1
296         fi
297     fi
298     
299     # customize root's prompt
300     cat << PROFILE > /root/.profile
301 export PS1="[$vserver] \\w # "
302 PROFILE
303
304     uid=2000
305     gid=2000
306     
307     # add a "build" user to the system
308     builduser=\$(grep "^build:" /etc/passwd | wc -l)
309     if [ \$builduser -eq 0 ] ; then
310         groupadd -o -g \$gid build;
311         useradd -o -c 'Automated Build' -u \$uid -g \$gid -n -M -s /bin/bash build;
312     fi
313
314 # Allow build user to build certain RPMs as root
315     if [ -f /etc/sudoers ] ; then
316         buildsudo=\$(grep "^build.*ALL=(ALL).*NOPASSWD:.*ALL"  /etc/sudoers | wc -l)
317         if [ \$buildsudo -eq 0 ] ; then
318             echo "build   ALL=(ALL)       NOPASSWD: ALL" >> /etc/sudoers
319         fi
320         sed -i 's,^Defaults.*requiretty,#Defaults requiretty,' /etc/sudoers
321     fi
322 #
323 EOF
324
325 }
326
327 function post_install_myplc  () {
328     set -x 
329     set -e 
330     trap failure ERR INT
331
332     vserver=$1; shift
333     personality=$1; shift
334
335 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
336     cat << EOF | $personality vserver $VERBOSE $vserver exec bash -x
337
338     # create /etc/sysconfig/network if missing
339     [ -f /etc/sysconfig/network ] || echo NETWORKING=yes > /etc/sysconfig/network
340
341     # create symlink for /dev/fd
342     [ ! -e "/dev/fd" ] && ln -s /proc/self/fd /dev/fd
343
344     # turn off regular crond, as plc invokes plc_crond
345     chkconfig crond off
346
347     # take care of loginuid in /etc/pam.d 
348     sed -i "s,#*\(.*loginuid.*\),#\1," /etc/pam.d/*
349
350     # customize root's prompt
351     cat << PROFILE > /root/.profile
352 export PS1="[$vserver] \\w # "
353 PROFILE
354
355 EOF
356 }
357
358 # parses ifconfig's output to find out ip address and mask
359 # will then be passed to vserver as e.g. --interface 138.96.250.126/255.255.0.0
360 # default is to use lo, that's enough for local mirrors
361 # use -i eth0 in case your fedora mirror is on a separate box on the network
362 function vserverIfconfig () {
363     ifname=$1; shift
364     local result="" 
365     line=$(ifconfig $ifname 2> /dev/null | grep 'inet addr')
366     if [ -n "$line" ] ; then
367         set $line
368         for word in "$@" ; do
369             addr=$(echo $word | sed -e s,[aA][dD][dD][rR]:,,)
370             mask=$(echo $word | sed -e s,[mM][aA][sS][kK]:,,)
371             if [ "$word" != "$addr" ] ; then
372                 result="${addr}"
373             elif [ "$word" != "$mask" ] ; then
374                 result="${result}/${mask}"
375             fi
376         done
377     fi
378     if [ -z "$result" ] ; then 
379         echo "vserverIfconfig failed to locate $ifname"
380         exit 1
381     else
382         echo $result
383     fi
384 }
385
386 function usage () {
387     set +x 
388     echo "Usage: $COMMAND_VBUILD [options] vserver-name [ -- vserver-options ]"
389     echo "Usage: $COMMAND_MYPLC [options] vserver-name repo-url [ -- vserver-options ]"
390     echo "Requirements: you need to have a vserver-compliant kernel,"
391     echo "   as well as the util-vserver RPM installed"
392     echo "Description:"
393     echo "   This command creates a fresh vserver instance, for building, or running, myplc"
394     echo "Supported options"
395     echo " -f fcdistro - for creating the root filesystem - defaults to $DEFAULT_FCDISTRO"
396     echo " -d pldistro - defaults to $DEFAULT_PLDISTRO"
397     echo " -p personality - defaults to $DEFAULT_PERSONALITY"
398     echo " -i ifname: determines ip and netmask attached to ifname, and passes it to the vserver"
399     echo " -v : verbose - passes -v to calls to vserver"
400     echo "vserver-options"
401     echo "  all args after the optional -- are passed to vserver <name> build <options>"
402     echo "  typical usage is e.g. --interface eth0:200.150.100.10/24"
403     echo "With $COMMAND_MYPLC you can give 'none' as the URL, in which case"
404     echo "   myplc.repo does not get created"
405     exit 1
406 }
407
408 ### parse args and 
409 function main () {
410
411     set -e
412     trap failure ERR INT
413
414     case "$COMMAND" in
415         $COMMAND_VBUILD)
416             VBUILD_MODE=true ;;
417         $COMMAND_MYPLC)
418             MYPLC_MODE=true;;
419         *)
420             usage ;;
421     esac
422
423     VERBOSE=
424     IFNAME=""
425     VSERVER_OPTIONS=""
426     while getopts "f:d:p:i:v" opt ; do
427         case $opt in
428             f) fcdistro=$OPTARG;;
429             d) pldistro=$OPTARG;;
430             p) personality=$OPTARG;;
431             i) IFNAME=$OPTARG;;
432             v) VERBOSE="-v" ;;
433             *) usage ;;
434         esac
435     done
436         
437     shift $(($OPTIND - 1))
438
439     # parse fixed arguments
440     [[ -z "$@" ]] && usage
441     vserver=$1 ; shift
442     if [ -n "$MYPLC_MODE" ] ; then
443         [[ -z "$@" ]] && usage
444         REPO_URL=$1 ; shift
445     fi
446
447     # parse vserver options
448     if [[ -n "$@" ]] ; then
449         if [ "$1" == "--" ] ; then
450             shift
451             VSERVER_OPTIONS="$@"
452         else
453             usage
454         fi
455     fi
456
457     # with new util-vserver, it is mandatory to provide an IP even for building
458     if [ -n "$VBUILD_MODE" ] ; then
459         [ -z "$IFNAME" ] && IFNAME=$DEFAULT_IFNAME
460     fi
461     if [ -n "$IFNAME" ] ; then
462         localip=$(vserverIfconfig $IFNAME)
463         VSERVER_OPTIONS="$VSERVER_OPTIONS --interface $localip"
464     fi
465
466     [ -z "$fcdistro" ] && fcdistro=$DEFAULT_FCDISTRO
467     [ -z "$pldistro" ] && pldistro=$DEFAULT_PLDISTRO
468     [ -z "$personality" ] && personality=$DEFAULT_PERSONALITY
469
470     setup_vserver $vserver $fcdistro $personality 
471     devel_or_vtest_tools $vserver $fcdistro $pldistro $personality
472     post_install $vserver $personality
473
474 }
475
476 main "$@"