addition for adding the universe feed to ubuntu for python-rpm that is crucially...
[build.git] / lbuild-initvm.sh
1 #!/bin/bash
2 # -*-shell-*-
3
4 #shopt -s huponexit
5
6 COMMAND=$(basename $0)
7 DIRNAME=$(dirname $0)
8 BUILD_DIR=$(pwd)
9
10 # pkgs parsing utilities
11 export PATH=$(dirname $0):$PATH
12
13 . build.common
14
15 DEFAULT_FCDISTRO=f20
16 DEFAULT_PLDISTRO=lxc
17 DEFAULT_PERSONALITY=linux64
18
19 COMMAND_LBUILD="lbuild-initvm.sh"
20 COMMAND_LTEST="ltest-initvm.sh"
21
22 ##########
23 # when creating build boxes we use private NAT'ed addresses for the VMs
24 # as per virbr0 that is taken care of by libvirt at startup
25 PRIVATE_BRIDGE="virbr0"
26 PRIVATE_PREFIX="192.168.122."
27 PRIVATE_GATEWAY="192.168.122.1"
28 # beware that changing this would break the logic of random_private_byte...
29 PRIVATE_MASKLEN=24
30
31 # we just try randomly in that range until a free IP address shows up
32 PRIVATE_ATTEMPTS=20
33
34 # constant
35 PUBLIC_BRIDGE=br0
36
37 # the network interface name as seen from the container
38 VIF_GUEST=eth0
39
40 ##############################
41 ## stolen from tests/system/template-qemu/qemu-bridge-init
42 # use /proc/net/dev instead of a hard-wired list
43 function gather_interfaces () {
44     python <<EOF
45 for line in file("/proc/net/dev"):
46     if ':' not in line: continue
47     ifname=line.replace(" ","").split(":")[0]
48     if ifname.find("lo")==0: continue
49     if ifname.find("br")==0: continue
50     if ifname.find("virbr")==0: continue
51     if ifname.find("tap")==0: continue
52     print ifname
53 EOF
54 }
55
56 function discover_interface () {
57     for ifname in $(gather_interfaces); do
58         ip link show $ifname | grep -qi 'state UP' && { echo $ifname; return; }
59     done
60     # still not found ? that's bad
61     echo unknown
62 }
63 ########## check for a free IP
64 function ip_is_busy () {
65     target=$1; shift
66     ping -c 1 -W 1 $target >& /dev/null
67 }
68
69 function random_private_byte () {
70     for attempt in $(seq $PRIVATE_ATTEMPTS); do
71         byte=$(($RANDOM % 256))
72         if [ "$byte" == 0 -o "$byte" == 1 ] ; then continue; fi
73         ip=${PRIVATE_PREFIX}${byte}
74         ip_is_busy $ip || { echo $byte; return; }
75     done
76     echo "Cannot seem to find a free IP address in range ${PRIVATE_PREFIX}.xx/24 after $PRIVATE_ATTEMPTS attempts - exiting"
77     exit 1
78 }
79
80 ########## networking -- ctd
81 function gethostbyname () {
82     hostname=$1
83     python -c "import socket; print socket.gethostbyname('"$hostname"')" 2> /dev/null
84 }
85
86 # e.g. 21 -> 255.255.248.0
87 function masklen_to_netmask () {
88     masklen=$1; shift
89     python <<EOF
90 import sys
91 masklen=$masklen
92 if not (masklen>=1 and masklen<=32): 
93   print "Wrong masklen",masklen
94   exit(1)
95 result=[]
96 for i in range(4):
97     if masklen>=8:
98        result.append(8)
99        masklen-=8
100     else:
101        result.append(masklen)
102        masklen=0
103 print ".".join([ str(256-2**(8-i)) for i in result ])
104   
105 EOF
106 }
107
108 #################### bridge initialization
109 function create_bridge_if_needed() {
110    
111     # turn on verbosity
112     set -x
113
114     # already created ? - we're done
115     ip addr show $PUBLIC_BRIDGE >& /dev/null && {
116         echo "Bridge already set up - skipping create_bridge_if_needed"
117         return 0
118     }
119
120     # find out the physical interface to bridge onto
121     if_lan=$(discover_interface)
122
123     ip addr show $if_lan &>/dev/null || {
124         echo "Cannot use interface $if_lan - exiting"
125         exit 1
126     }
127
128     #################### bride initialization
129     check_yum_installed bridge-utils
130
131     echo "========== $COMMAND: entering create_bridge - beg"
132     hostname
133     uname -a
134     ip addr show
135     ip route
136     echo "========== $COMMAND: entering create_bridge - end"
137
138     # disable netfilter calls for bridge interface (they cause panick on 2.6.35 anyway)
139     #
140     # another option would be to accept the all forward packages for
141     # bridged interface like: -A FORWARD -m physdev --physdev-is-bridged -j ACCEPT
142     sysctl net.bridge.bridge-nf-call-iptables=0
143     sysctl net.bridge.bridge-nf-call-ip6tables=0
144     sysctl net.bridge.bridge-nf-call-arptables=0
145
146     
147     #Getting host IP/masklen
148     address=$(ip addr show $if_lan | grep -v inet6 | grep inet | head --lines=1 | awk '{print $2;}')
149     [ -z "$address" ] && { echo "ERROR: Could not determine IP address for $if_lan" ; exit 1 ; }
150
151     broadcast=$(ip addr show $if_lan | grep -v inet6 | grep inet | head --lines=1 | awk '{print $4;}')
152     [ -z "$broadcast" ] && echo "WARNING: Could not determine broadcast address for $if_lan"
153
154     gateway=$(ip route show | grep default | awk '{print $3;}')
155     [ -z "$gateway" ] && echo "WARNING: Could not determine gateway IP"
156
157
158     # creating the bridge
159     echo "Creating bridge PUBLIC_BRIDGE=$PUBLIC_BRIDGE"
160     brctl addbr $PUBLIC_BRIDGE
161     brctl addif $PUBLIC_BRIDGE $if_lan
162     echo "Activating promiscuous mode if_lan=$if_lan"
163     ip link set $if_lan up promisc on
164     sleep 2
165     # rely on dhcp to re assign IP.. 
166     echo "Starting dhclient on $PUBLIC_BRIDGE"
167     dhclient $PUBLIC_BRIDGE
168     sleep 1
169
170     #Reconfigure the routing table
171     echo "Configuring gateway=$gateway"
172     ip route add default via $gateway dev $PUBLIC_BRIDGE
173     ip route del default via $gateway dev $if_lan
174     # at this point we have an extra route like e.g.
175     ## ip route show
176     #default via 138.96.112.250 dev br0
177     #138.96.112.0/21 dev em1  proto kernel  scope link  src 138.96.112.57
178     #138.96.112.0/21 dev br0  proto kernel  scope link  src 138.96.112.57
179     #192.168.122.0/24 dev virbr0  proto kernel  scope link  src 192.168.122.1
180     route_dest=$(ip route show | grep -v default | grep "dev $PUBLIC_BRIDGE" | awk '{print $1;}')
181     ip route del $route_dest dev $if_lan
182
183     echo "========== $COMMAND: exiting create_bridge - beg"
184     ip addr show
185     ip route show
186     echo "========== $COMMAND: exiting create_bridge - end"
187
188     # for safety
189     sleep 3
190     return 0
191
192 }
193
194 ##############################
195 # return yum or debootstrap
196 function package_method () {
197     fcdistro=$1; shift
198     case $fcdistro in
199         f[0-9]*|centos[0-9]*|sl[0-9]*) echo yum ;;
200         squeeze|wheezy|oneiric|precise|quantal|raring|saucy) echo debootstrap ;;
201         *) echo Unknown distro $fcdistro ;;
202     esac 
203 }
204
205 # return arch from debian distro and personality
206 function canonical_arch () {
207     personality=$1; shift
208     fcdistro=$1; shift
209     case $(package_method $fcdistro) in
210         yum)
211             case $personality in *32) echo i386 ;; *64) echo x86_64 ;; *) echo Unknown-arch-1 ;; esac ;;
212         debootstrap)
213             case $personality in *32) echo i386 ;; *64) echo amd64 ;; *) echo Unknown-arch-2 ;; esac ;;
214         *)
215             echo Unknown-arch-3 ;;
216     esac
217 }
218
219 # the new test framework creates /timestamp in /vservers/<name> *before* populating it
220 function almost_empty () { 
221     dir="$1"; shift ; 
222     # non existing is fine
223     [ ! -d $dir ] && return 0; 
224     # need to have at most one file
225     count=$(cd $dir; ls | wc -l); [ $count -le 1 ]; 
226 }
227
228 ##############################
229 function check_yum_installed () {
230     package=$1; shift
231     rpm -q $package >& /dev/null || yum -y install $package
232 }
233
234 function check_yumgroup_installed () {
235     group="$1"; shift
236     yum grouplist "$group" | grep -q Installed || { yum -y groupinstall "$group" ; }
237 }
238
239 ##############################
240 function fedora_install() {
241     set -x
242     set -e
243
244     mkdir -p /var/lock/subsys/
245     (
246         flock -n -x 200 || { echo "Cache repository is busy." ; return 1 ; }
247
248         if [ ! -e "$cache/rootfs" ]; then
249             echo "Getting cache download in $cache/rootfs ... "
250             fedora_download || { echo "Failed to download 'fedora base'"; return 1; }
251         else
252             echo "Updating cache $cache/rootfs ..."
253             if ! yum --installroot $cache/rootfs -y --nogpgcheck update ; then
254                 echo "Failed to update 'fedora base', continuing with last known good cache"
255             else
256                 echo "Update finished"
257             fi
258         fi
259
260         echo "Copy $cache/rootfs to $rootfs_path ... "
261         mkdir -p $rootfs_path
262         rsync -a $cache/rootfs/ $rootfs_path/
263         
264         return 0
265
266         ) 200>/var/lock/subsys/lxc
267
268     return $?
269 }
270
271 function fedora_download() {
272     set -x
273     # check the mini fedora was not already downloaded
274     INSTALL_ROOT=$cache/partial
275     echo $INSTALL_ROOT
276
277     # download a mini fedora into a cache
278     echo "Downloading fedora minimal ..."
279
280     mkdir -p $INSTALL_ROOT || { echo "Failed to create '$INSTALL_ROOT' directory" ; return 1; }
281
282     mkdir -p $INSTALL_ROOT/etc/yum.repos.d   
283     mkdir -p $INSTALL_ROOT/dev
284     mknod -m 0444 $INSTALL_ROOT/dev/random c 1 8
285     mknod -m 0444 $INSTALL_ROOT/dev/urandom c 1 9
286
287     # copy yum config and repo files
288     cp /etc/yum.conf $INSTALL_ROOT/etc/
289     cp /etc/yum.repos.d/fedora* $INSTALL_ROOT/etc/yum.repos.d/
290
291     # append fedora repo files with desired $release and $basearch
292     for f in $INSTALL_ROOT/etc/yum.repos.d/* ; do
293       sed -i "s/\$basearch/$arch/g; s/\$releasever/$release/g;" $f
294     done 
295
296     MIRROR_URL=http://mirror.onelab.eu/fedora/releases/$release/Everything/$arch/os
297     RELEASE_URL1="$MIRROR_URL/Packages/fedora-release-$release-1.noarch.rpm"
298     # with fedora18 the rpms are scattered by first name
299     RELEASE_URL2="$MIRROR_URL/Packages/f/fedora-release-$release-1.noarch.rpm"
300     RELEASE_TARGET=$INSTALL_ROOT/fedora-release-$release.noarch.rpm
301     found=""
302     for attempt in $RELEASE_URL1 $RELEASE_URL2; do
303         if curl -f $attempt -o $RELEASE_TARGET ; then
304             echo "Retrieved $attempt"
305             found=true
306             break
307         else
308             echo "Failed attempt $attempt"
309         fi
310     done
311     [ -n "$found" ] || { echo "Could not retrieve fedora-release rpm - exiting" ; exit 1; }
312     
313     mkdir -p $INSTALL_ROOT/var/lib/rpm
314     rpm --root $INSTALL_ROOT  --initdb
315     # when installing f12 this apparently is already present, so ignore result
316     rpm --root $INSTALL_ROOT -ivh $INSTALL_ROOT/fedora-release-$release.noarch.rpm || :
317     # however f12 root images won't get created on a f18 host
318     # (the issue here is the same as the one we ran into when dealing with a vs-box)
319     # in a nutshell, in f12 the glibc-common and filesystem rpms have an apparent conflict
320     # >>> file /usr/lib/locale from install of glibc-common-2.11.2-3.x86_64 conflicts 
321     #          with file from package filesystem-2.4.30-2.fc12.x86_64
322     # in fact this was - of course - allowed by f12's rpm but later on a fix was made 
323     #   http://rpm.org/gitweb?p=rpm.git;a=commitdiff;h=cf1095648194104a81a58abead05974a5bfa3b9a
324     # So ideally if we want to be able to build f12 images from f18 we need an rpm that has
325     # this patch undone, like we have in place on our f14 boxes (our f14 boxes need a f18-like rpm)
326
327     YUM="yum --installroot=$INSTALL_ROOT --nogpgcheck -y"
328     PKG_LIST="yum initscripts passwd rsyslog vim-minimal dhclient chkconfig rootfiles policycoreutils openssh-server openssh-clients"
329     echo "$YUM install $PKG_LIST"
330     $YUM install $PKG_LIST || { echo "Failed to download rootfs, aborting." ; return 1; }
331
332     mv "$INSTALL_ROOT" "$cache/rootfs"
333     echo "Download complete."
334
335     return 0
336 }
337
338 ##############################
339 function fedora_configure() {
340
341     set -x
342     set -e
343
344     # disable selinux in fedora
345     mkdir -p $rootfs_path/selinux
346     echo 0 > $rootfs_path/selinux/enforce
347
348     # set the hostname
349     case "$fcdistro" in 
350         f18|f2?)
351             cat <<EOF > ${rootfs_path}/etc/hostname
352 $GUEST_HOSTNAME
353 EOF
354             echo ;;
355         *)
356             cat <<EOF > ${rootfs_path}/etc/sysconfig/network
357 NETWORKING=yes
358 HOSTNAME=$GUEST_HOSTNAME
359 EOF
360             # set minimal hosts
361             cat <<EOF > $rootfs_path/etc/hosts
362 127.0.0.1 localhost $GUEST_HOSTNAME
363 EOF
364             echo ;;
365     esac
366
367     dev_path="${rootfs_path}/dev"
368     rm -rf $dev_path
369     mkdir -p $dev_path
370     mknod -m 666 ${dev_path}/null c 1 3
371     mknod -m 666 ${dev_path}/zero c 1 5
372     mknod -m 666 ${dev_path}/random c 1 8
373     mknod -m 666 ${dev_path}/urandom c 1 9
374     mkdir -m 755 ${dev_path}/pts
375     mkdir -m 1777 ${dev_path}/shm
376     mknod -m 666 ${dev_path}/tty c 5 0
377     mknod -m 666 ${dev_path}/tty0 c 4 0
378     mknod -m 666 ${dev_path}/tty1 c 4 1
379     mknod -m 666 ${dev_path}/tty2 c 4 2
380     mknod -m 666 ${dev_path}/tty3 c 4 3
381     mknod -m 666 ${dev_path}/tty4 c 4 4
382     mknod -m 600 ${dev_path}/console c 5 1
383     mknod -m 666 ${dev_path}/full c 1 7
384     mknod -m 600 ${dev_path}/initctl p
385     mknod -m 666 ${dev_path}/ptmx c 5 2
386
387     if [ "$(echo $fcdistro | cut -d"f" -f2)" -le "14" ]; then
388         fedora_configure_init
389     else
390         fedora_configure_systemd
391     fi
392
393     guest_ifcfg=${rootfs_path}/etc/sysconfig/network-scripts/ifcfg-$VIF_GUEST
394     ( [ -n "$BUILD_MODE" ] && write_guest_ifcfg_build || write_guest_ifcfg_test ) > $guest_ifcfg
395
396     fedora_configure_yum $lxc $fcdistro $pldistro
397
398     return 0
399 }
400
401 function fedora_configure_init() {
402     set -e
403     set -x
404     sed -i 's|.sbin.start_udev||' ${rootfs_path}/etc/rc.sysinit
405     sed -i 's|.sbin.start_udev||' ${rootfs_path}/etc/rc.d/rc.sysinit
406     # don't mount devpts, for pete's sake
407     sed -i 's/^.*dev.pts.*$/#\0/' ${rootfs_path}/etc/rc.sysinit
408     sed -i 's/^.*dev.pts.*$/#\0/' ${rootfs_path}/etc/rc.d/rc.sysinit
409     chroot ${rootfs_path} chkconfig udev-post off
410     chroot ${rootfs_path} chkconfig network on
411 }
412
413 # this code of course is for guests that do run on systemd
414 function fedora_configure_systemd() {
415     set -e
416     set -x
417     # so ignore if we can't find /etc/systemd at all 
418     [ -d ${rootfs_path}/etc/systemd ] || return 0
419     # otherwise let's proceed
420     ln -sf /lib/systemd/system/multi-user.target ${rootfs_path}/etc/systemd/system/default.target
421     touch ${rootfs_path}/etc/fstab
422     ln -sf /dev/null ${rootfs_path}/etc/systemd/system/udev.service
423 # Thierry - Feb 2013
424 # this was intended for f16 initially, in order to enable getty that otherwise would not start
425 # having a getty running is helpful only if ssh won't start though, and we see a correlation between
426 # VM's that refuse to lxc-stop and VM's that run crazy getty's
427 # so, turning getty off for now instead
428 #   #dependency on a device unit fails it specially that we disabled udev
429 #    sed -i 's/After=dev-%i.device/After=/' ${rootfs_path}/lib/systemd/system/getty\@.service
430     ln -sf /dev/null ${rootfs_path}/etc/systemd/system/"getty@.service"
431     rm -f ${rootfs_path}/etc/systemd/system/getty.target.wants/*service || :
432 # can't seem to handle this one with systemctl
433     chroot ${rootfs_path} chkconfig network on
434 }
435
436 # overwrite container yum config
437 function fedora_configure_yum () {
438     set -x 
439     set -e 
440     trap failure ERR INT
441
442     lxc=$1; shift
443     fcdistro=$1; shift
444     pldistro=$1; shift
445
446     # rpm --rebuilddb
447     chroot $rootfs_path rpm --rebuilddb
448
449     echo "Initializing yum.repos.d in $lxc"
450     rm -f $rootfs_path/etc/yum.repos.d/*
451
452     cat > $rootfs_path/etc/yum.repos.d/building.repo <<EOF
453 [fedora]
454 name=Fedora $release - $arch
455 baseurl=http://mirror.onelab.eu/fedora/releases/$release/Everything/$arch/os/
456 enabled=1
457 metadata_expire=7d
458 gpgcheck=1
459 gpgkey=http://mirror.onelab.eu/keys/RPM-GPG-KEY-fedora-$release-primary
460
461 [updates]
462 name=Fedora $release - $arch - Updates
463 baseurl=http://mirror.onelab.eu/fedora/updates/$release/$arch/
464 enabled=1
465 metadata_expire=7d
466 gpgcheck=1
467 gpgkey=http://mirror.onelab.eu/keys/RPM-GPG-KEY-fedora-$release-primary
468 EOF
469     
470     # for using vtest-init-lxc.sh as a general-purpose lxc creation wrapper
471     # just mention 'none' as the repo url
472     if [ -n "$REPO_URL" ] ; then
473         if [ ! -d $rootfs_path/etc/yum.repos.d ] ; then
474             echo "WARNING : cannot create myplc repo"
475         else
476             # exclude kernel from fedora repos 
477             yumexclude=$(pl_plcyumexclude $fcdistro $pldistro $DIRNAME)
478             for repo in $rootfs_path/etc/yum.repos.d/* ; do
479                 [ -f $repo ] && yumconf_exclude $repo "exclude=$yumexclude" 
480             done
481             # the build repo is not signed at this stage
482             cat > $rootfs_path/etc/yum.repos.d/myplc.repo <<EOF
483 [myplc]
484 name= MyPLC
485 baseurl=$REPO_URL
486 enabled=1
487 gpgcheck=0
488 EOF
489         fi
490     fi
491 }    
492
493 ##############################
494 # need to specify the right mirror for debian variants like ubuntu and the like
495 function debian_mirror () {
496     fcdistro=$1; shift
497     case $fcdistro in
498         squeeze|wheezy) 
499             echo http://ftp2.fr.debian.org/debian/ ;;
500         oneiric|precise|quantal|raring|saucy) 
501             echo http://mir1.ovh.net/ubuntu/ubuntu/ ;;
502         *) echo unknown distro $fcdistro; exit 1;;
503     esac
504 }
505
506 function debian_install () {
507     set -e
508     set -x
509     mkdir -p $rootfs_path
510     arch=$(canonical_arch $personality $fcdistro)
511     mirror=$(debian_mirror $fcdistro)
512     debootstrap --arch $arch $fcdistro $rootfs_path $mirror
513 }
514
515 function debian_configure () {
516     guest_interfaces=${rootfs_path}/etc/network/interfaces
517     ( [ -n "$BUILD_MODE" ] && write_guest_interfaces_build || write_guest_interfaces_test ) > $guest_interfaces
518 }
519
520 function write_guest_interfaces_build () {
521     cat <<EOF
522 auto $VIF_GUEST
523 iface $VIF_GUEST inet dhcp
524 EOF
525 }
526
527 function write_guest_interfaces_test () {
528     cat <<EOF
529 auto $VIF_GUEST
530 iface $VIF_GUEST
531     address $IP
532     netmask $NETMASK
533     gateway $GATEWAY
534 EOF
535 }
536 ##############################
537 function setup_lxc() {
538
539     set -x
540     set -e
541     #trap failure ERR INT
542
543     lxc=$1; shift
544     fcdistro=$1; shift
545     pldistro=$1; shift
546     personality=$1; shift
547
548     # create lxc container 
549     
550     pkg_method=$(package_method $fcdistro)
551     case $pkg_method in
552         yum)
553             fedora_install || { echo "failed to install fedora root image"; exit 1 ; }
554             fedora_configure || { echo "failed to configure fedora for a container"; exit 1 ; }
555             ;;
556         debootstrap)
557             debian_install || { echo "failed to install debian/ubuntu root image"; exit 1 ; }
558             debian_configure || { echo "failed to configure debian/ubuntu for a container"; exit 1 ; }
559             ;;
560         *)
561             echo "$COMMAND:: unknown package_method - exiting"
562             exit 1
563             ;;
564     esac
565
566     # Enable cgroup -- xxx -- is this really useful ?
567     mkdir $rootfs_path/cgroup
568     
569     # set up resolv.conf
570     cp /etc/resolv.conf $rootfs_path/etc/resolv.conf
571     # and /etc/hosts for at least localhost
572     [ -f $rootfs_path/etc/hosts ] || echo "127.0.0.1 localhost localhost.localdomain" > $rootfs_path/etc/hosts
573     
574     # grant ssh access from host to guest
575     mkdir $rootfs_path/root/.ssh
576     cat /root/.ssh/id_rsa.pub >> $rootfs_path/root/.ssh/authorized_keys
577     
578     config_xml=$config_path/"lxc.xml"
579     ( [ -n "$BUILD_MODE" ] && write_lxc_xml_build $lxc || write_lxc_xml_test $lxc ) > $config_xml
580     
581     # define lxc container for libvirt
582     virsh -c lxc:/// define $config_xml
583
584     return 0
585 }
586
587 function write_lxc_xml_test () {
588     lxc=$1; shift
589     cat <<EOF
590 <domain type='lxc'>
591   <name>$lxc</name>
592   <memory>524288</memory>
593   <os>
594     <type arch='$arch2'>exe</type>
595     <init>/sbin/init</init>
596   </os>
597   <features>
598     <acpi/>
599   </features>
600   <vcpu>1</vcpu>
601   <clock offset='utc'/>
602   <on_poweroff>destroy</on_poweroff>
603   <on_reboot>restart</on_reboot>
604   <on_crash>destroy</on_crash>
605   <devices>
606     <emulator>/usr/libexec/libvirt_lxc</emulator>
607     <filesystem type='mount'>
608       <source dir='$rootfs_path'/>
609       <target dir='/'/>
610     </filesystem>
611     <interface type="bridge">
612       <source bridge="$PUBLIC_BRIDGE"/>
613       <target dev='$VIF_HOST'/>
614     </interface>
615     <console type='pty' />
616   </devices>
617   <network>
618     <name>host-bridge</name>
619     <forward mode="bridge"/>
620     <bridge name="$PUBLIC_BRIDGE"/>
621   </network>
622 </domain>
623 EOF
624 }
625
626 function write_lxc_xml_build () { 
627     lxc=$1; shift
628     cat <<EOF
629 <domain type='lxc'>
630   <name>$lxc</name>
631   <memory>524288</memory>
632   <os>
633     <type arch='$arch2'>exe</type>
634     <init>/sbin/init</init>
635   </os>
636   <features>
637     <acpi/>
638   </features>
639   <vcpu>1</vcpu>
640   <clock offset='utc'/>
641   <on_poweroff>destroy</on_poweroff>
642   <on_reboot>restart</on_reboot>
643   <on_crash>destroy</on_crash>
644   <devices>
645     <emulator>/usr/libexec/libvirt_lxc</emulator>
646     <filesystem type='mount'>
647       <source dir='$rootfs_path'/>
648       <target dir='/'/>
649     </filesystem>
650     <interface type="network">
651       <source network="default"/>
652     </interface>
653     <console type='pty' />
654   </devices>
655 </domain>
656 EOF
657 }
658
659 # this one is dhcp-based
660 function write_guest_ifcfg_build () {
661     cat <<EOF
662 DEVICE=$VIF_GUEST
663 BOOTPROTO=dhcp
664 ONBOOT=yes
665 NM_CONTROLLED=no
666 TYPE=Ethernet
667 MTU=1500
668 EOF
669 }
670
671 # use fixed IP as specified by GUEST_HOSTNAME
672 function write_guest_ifcfg_test () {
673     cat <<EOF
674 DEVICE=$VIF_GUEST
675 BOOTPROTO=static
676 ONBOOT=yes
677 HOSTNAME=$GUEST_HOSTNAME
678 IPADDR=$IP
679 NETMASK=$NETMASK
680 GATEWAY=$GATEWAY
681 NM_CONTROLLED=no
682 TYPE=Ethernet
683 MTU=1500
684 EOF
685 }
686
687 function devel_or_vtest_tools () {
688
689     set -x 
690     set -e 
691     trap failure ERR INT
692
693     lxc=$1; shift
694     fcdistro=$1; shift
695     pldistro=$1; shift
696     personality=$1; shift
697
698     pkg_method=$(package_method $fcdistro)
699
700     pkgsfile=$(pl_locateDistroFile $DIRNAME $pldistro $PREINSTALLED)
701
702     ### install individual packages, then groups
703     # get target arch - use uname -i here (we want either x86_64 or i386)
704    
705     lxc_arch=$(chroot $rootfs_path uname -i)
706     # on debian systems we get arch through the 'arch' command
707     [ "$lxc_arch" = "unknown" ] && lxc_arch=$(chroot $rootfs_path arch)
708
709     packages=$(pl_getPackages -a $lxc_arch $fcdistro $pldistro $pkgsfile)
710     groups=$(pl_getGroups -a $lxc_arch $fcdistro $pldistro $pkgsfile)
711
712     case "$pkg_method" in
713         yum)
714             [ -n "$packages" ] && chroot $rootfs_path yum -y install $packages
715             for group_plus in $groups; do
716                 group=$(echo $group_plus | sed -e "s,+++, ,g")
717                 chroot $rootfs_path yum -y groupinstall "$group"
718             done
719             # store current rpm list in /init-lxc.rpms in case we need to check the contents
720             chroot $rootfs_path rpm -aq > $rootfs_path/init-lxc.rpms
721             ;;
722         debootstrap)
723             # for ubuntu
724             if grep -iq ubuntu /vservers/$vserver/rootfs/etc/lsb-release 2> /dev/null; then
725                 # on ubuntu, at this point we end up with a single feed in /etc/apt/sources.list
726                 # we need at least to add the 'universe' feed for python-rpm
727                 ( cd /vservers/$vserver/rootfs/etc/apt ; head -1 sources.list | sed -e s,main,universe, > sources.list.d/universe.list )
728                 # also adding a link to updates sounds about right
729                 ( cd /vservers/$vserver/rootfs/etc/apt ; head -1 sources.list | sed -e 's, main,-updates main,' > sources.list.d/updates.list )
730             fi
731             chroot $rootfs_path apt-get update
732             for package in $packages ; do
733                 # close stdin in an attempt to avoid this hanging
734                 # xxx also we ignore result for now, not sure if the kind of errors like below
735                 # truly is serious or not
736 #Setting up at (3.1.13-2ubuntu2) ...
737 #initctl: Unable to connect to Upstart: Failed to connect to socket /com/ubuntu/upstart: Connection refused
738 #initctl: Unable to connect to Upstart: Failed to connect to socket /com/ubuntu/upstart: Connection refused
739 #start: Unable to connect to Upstart: Failed to connect to socket /com/ubuntu/upstart: Connection refused
740
741                 chroot $rootfs_path apt-get install -y $package < /dev/null || :
742             done
743             ### xxx todo install groups with apt..
744             ;;
745         *)
746             echo "unknown pkg_method $pkg_method"
747             ;;
748     esac
749
750     return 0
751 }
752
753 function post_install () {
754     lxc=$1; shift 
755     personality=$1; shift
756     if [ -n "$BUILD_MODE" ] ; then
757         post_install_build $lxc $personality
758         lxc_start $lxc
759         # manually run dhclient in guest - somehow this network won't start on its own
760         virsh -c lxc:/// lxc-enter-namespace $lxc $(bin_in_container $lxc dhclient) $VIF_GUEST
761     else
762         post_install_myplc $lxc $personality
763         lxc_start $lxc
764         wait_for_ssh $lxc
765     fi
766     # setup localtime from the host
767     cp /etc/localtime $rootfs_path/etc/localtime
768 }
769
770 function post_install_build () {
771
772     set -x 
773     set -e 
774     trap failure ERR INT
775
776     lxc=$1; shift
777     personality=$1; shift
778
779 ### From myplc-devel-native.spec
780 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
781     cat << EOF | chroot $rootfs_path bash -x
782     # set up /dev/loop* in lxc
783     for i in \$(seq 0 255) ; do
784         /bin/mknod -m 640 /dev/loop\$i b 7 \$i
785     done
786     
787     # create symlink for /dev/fd
788     [ ! -e "/dev/fd" ] && /bin/ln -s /proc/self/fd /dev/fd
789
790     # modify /etc/rpm/macros to not use /sbin/new-kernel-pkg
791     /bin/sed -i 's,/sbin/new-kernel-pkg:,,' /etc/rpm/macros
792     if [ -h "/sbin/new-kernel-pkg" ] ; then
793         filename=\$(/bin/readlink -f /sbin/new-kernel-pkg)
794         if [ "\$filename" == "/sbin/true" ] ; then
795                 /bin/echo "WARNING: /sbin/new-kernel-pkg symlinked to /sbin/true"
796                 /bin/echo "\tmost likely /etc/rpm/macros has /sbin/new-kernel-pkg declared in _netsharedpath."
797                 /bin/echo "\tPlease remove /sbin/new-kernel-pkg from _netsharedpath and reintall mkinitrd."
798                 exit 1
799         fi
800     fi
801     
802     # customize root's prompt
803     /bin/cat << PROFILE > /root/.profile
804 export PS1="[$lxc] \\w # "
805 PROFILE
806
807     uid=2000
808     gid=2000
809     
810     # add a "build" user to the system
811     builduser=\$(grep "^build:" /etc/passwd | wc -l)
812     if [ \$builduser -eq 0 ] ; then
813         groupadd -o -g \$gid build;
814         useradd -o -c 'Automated Build' -u \$uid -g \$gid -n -M -s /bin/bash build;
815     fi
816
817 # Allow build user to build certain RPMs as root
818     if [ -f /etc/sudoers ] ; then
819         buildsudo=\$(grep "^build.*ALL=(ALL).*NOPASSWD:.*ALL"  /etc/sudoers | wc -l)
820         if [ \$buildsudo -eq 0 ] ; then
821             echo "build   ALL=(ALL)       NOPASSWD: ALL" >> /etc/sudoers
822         fi
823         sed -i 's,^Defaults.*requiretty,#Defaults requiretty,' /etc/sudoers
824     fi
825 #
826 EOF
827         
828 }
829
830 function post_install_myplc  () {
831     set -x 
832     set -e 
833     trap failure ERR INT
834
835     lxc=$1; shift
836     personality=$1; shift
837
838 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
839     cat << EOF | chroot $rootfs_path bash -x
840
841     # create /etc/sysconfig/network if missing
842     [ -f /etc/sysconfig/network ] || /bin/echo NETWORKING=yes > /etc/sysconfig/network
843
844     # create symlink for /dev/fd
845     [ ! -e "/dev/fd" ] && /bin/ln -s /proc/self/fd /dev/fd
846
847     # turn off regular crond, as plc invokes plc_crond
848     /sbin/chkconfig crond off
849
850     # take care of loginuid in /etc/pam.d 
851     /bin/sed -i "s,#*\(.*loginuid.*\),#\1," /etc/pam.d/*
852
853     # customize root's prompt
854     /bin/cat << PROFILE > /root/.profile
855 export PS1="[$lxc] \\w # "
856 PROFILE
857
858 EOF
859 }
860
861 function lxc_start() {
862
863     set -x
864     set -e
865     #trap failure ERR INT
866
867     lxc=$1; shift
868   
869     virsh -c lxc:/// start $lxc
870   
871     return 0
872 }
873
874 function wait_for_ssh () {
875     set -x
876     set -e
877     #trap failure ERR INT
878
879     lxc=$1; shift
880   
881     echo $IP is up, waiting for ssh...
882
883     #wait max 5 min for sshd to start 
884     ssh_up=""
885     stop_time=$(($(date +%s) + 300))
886     current_time=$(date +%s)
887     
888     counter=1
889     while [ "$current_time" -lt "$stop_time" ] ; do
890          echo "$counter-th attempt to reach sshd in container $lxc ..."
891          ssh -o "StrictHostKeyChecking no" $IP 'uname -i' && { ssh_up=true; echo "SSHD in container $lxc is UP"; break ; } || :
892          sleep 10
893          current_time=$(($current_time + 10))
894          counter=$(($counter+1))
895     done
896
897     # Thierry: this is fatal, let's just exit with a failure here
898     [ -z $ssh_up ] && { echo "SSHD in container $lxc is not running" ; exit 1 ; } 
899     return 0
900 }
901
902 ####################
903 function failure () {
904     echo "$COMMAND : Bailing out"
905     exit 1
906 }
907
908 function usage () {
909     set +x 
910     echo "Usage: $COMMAND_LBUILD [options] lxc-name"
911     echo "Usage: $COMMAND_LTEST [options] lxc-name"
912     echo "Description:"
913     echo "   This command creates a fresh lxc instance, for building, or running a test myplc"
914     echo "Supported options"
915     echo " -f fcdistro - for creating the root filesystem - defaults to $DEFAULT_FCDISTRO"
916     echo " -d pldistro - defaults to $DEFAULT_PLDISTRO"
917     echo " -p personality - defaults to $DEFAULT_PERSONALITY"
918     echo " -n hostname - the hostname to use in container - required with $COMMAND_LTEST"
919     echo " -r repo-url - used to populate yum.repos.d - required with $COMMAND_LTEST"
920     echo " -P pkgs_file - defines the set of extra pacakges"
921     echo "    by default we use vtest.pkgs or devel.pkgs according to $COMMAND"
922     echo " -v be verbose"
923     exit 1
924 }
925
926 ### parse args and 
927 function main () {
928
929     #set -e
930     #trap failure ERR INT
931
932     if [ "$(id -u)" != "0" ]; then
933           echo "This script should be run as 'root'"
934           exit 1
935     fi
936
937     case "$COMMAND" in
938         $COMMAND_LBUILD)
939             BUILD_MODE=true ;;
940         $COMMAND_LTEST)
941             TEST_MODE=true;;
942         *)
943             usage ;;
944     esac
945
946     echo 'build mode=' $BUILD_MODE 'test mode=' $TEST_MODE
947
948     # the set of preinstalled packages - depends on vbuild or vtest
949     if [ -n "$BUILD_MODE" ] ; then
950         PREINSTALLED=devel.pkgs
951     else
952         PREINSTALLED=vtest.pkgs
953     fi
954     while getopts "f:d:p:n:r:P:v" opt ; do
955         case $opt in
956             f) fcdistro=$OPTARG;;
957             d) pldistro=$OPTARG;;
958             p) personality=$OPTARG;;
959             n) GUEST_HOSTNAME=$OPTARG;;
960             r) REPO_URL=$OPTARG;;
961             P) PREINSTALLED=$OPTARG;;
962             v) VERBOSE=true; set -x;;
963             *) usage ;;
964         esac
965     done
966         
967     shift $(($OPTIND - 1))
968
969     # parse fixed arguments
970     [[ -z "$@" ]] && usage
971     lxc=$1 ; shift
972
973     # check we've exhausted the arguments
974     [[ -n "$@" ]] && usage
975
976     [ -z "$fcdistro" ] && fcdistro=$DEFAULT_FCDISTRO
977     [ -z "$pldistro" ] && pldistro=$DEFAULT_PLDISTRO
978     [ -z "$personality" ] && personality=$DEFAULT_PERSONALITY
979     
980     if [ -n "$BUILD_MODE" ] ; then
981         [ -z "$GUEST_HOSTNAME" ] && GUEST_HOSTNAME=$lxc
982     else
983         [[ -z "$GUEST_HOSTNAME" ]] && usage
984         # use -r none to get rid of this warning
985         if [ "$REPO_URL" == "none" ] ; then
986             REPO_URL=""
987         elif [ -z "$REPO_URL" ] ; then
988             echo "WARNING -- setting up a yum repo is recommended" 
989         fi
990     fi
991
992     ##########
993     release=$(echo $fcdistro | cut -df -f2)
994
995     if [ "$personality" == "linux32" ]; then
996         arch=i386
997         arch2=i686
998     elif [ "$personality" == "linux64" ]; then
999         arch=x86_64
1000         arch2=x86_64
1001     else
1002         echo "Unknown personality: $personality"
1003     fi
1004
1005     if [ -n "$BUILD_MODE" ] ; then
1006
1007         # Bridge IP affectation
1008         byte=$(random_private_byte)
1009         IP=${PRIVATE_PREFIX}$byte
1010         NETMASK=$(masklen_to_netmask $PRIVATE_MASKLEN)
1011         GATEWAY=$PRIVATE_GATEWAY
1012         VIF_HOST="i$byte"
1013     else
1014         [[ -z "GUEST_HOSTNAME" ]] && usage
1015        
1016         create_bridge_if_needed
1017
1018         IP=$(gethostbyname $GUEST_HOSTNAME)
1019         # use same NETMASK as bridge interface br0
1020         MASKLEN=$(ip addr show $PUBLIC_BRIDGE | grep -v inet6 | grep inet | awk '{print $2;}' | cut -d/ -f2)
1021         NETMASK=$(masklen_to_netmask $MASKLEN)
1022         GATEWAY=$(ip route show | grep default | awk '{print $3}')
1023         VIF_HOST="i$(echo $GUEST_HOSTNAME | cut -d. -f1)"
1024     fi
1025
1026     echo "the IP address of container $lxc is $IP, host virtual interface is $VIF_HOST"
1027
1028     path=/vservers
1029     [ ! -d $path ] && mkdir $path
1030     rootfs_path=$path/$lxc/rootfs
1031     config_path=$path/$lxc
1032     cache_base=/var/cache/lxc/fedora/$arch
1033     cache=$cache_base/$release
1034     
1035     # check whether the rootfs directory is created to know if the container exists
1036     # bacause /var/lib/lxc/$lxc is already created while putting $lxc.timestamp
1037     [ -d $rootfs_path ] && \
1038         { echo "container $lxc already exists in filesystem - exiting" ; exit 1 ; }
1039     virsh -c lxc:/// domuuid $lxc >& /dev/null && \
1040         { echo "container $lxc already exists in libvirt - exiting" ; exit 1 ; }
1041
1042     setup_lxc $lxc $fcdistro $pldistro $personality 
1043
1044     devel_or_vtest_tools $lxc $fcdistro $pldistro $personality
1045
1046     post_install $lxc $personality
1047     
1048     echo $COMMAND Done
1049 }
1050
1051 main "$@"