customizes root bash prompt
[build.git] / vbuild-init-vserver.sh
1 #!/bin/bash
2 # -*-shell-*-
3
4 COMMAND=$(basename $0)
5
6 # lst parsing utilities
7 PATH=$(dirname $0):$PATH . build.common
8
9 function failure () {
10     echo "$COMMAND : Bailing out"
11     exit 1
12 }
13
14 # overwrite vserver's internal yum config from what is in
15 # .distributions/<distrib>/yum/yum.conf and /yum.repos.d 
16
17 function configure_yum_in_vserver () {
18     set -x 
19     set -e 
20     trap failure ERR INT
21
22     vserver=$1; shift
23     fcdistro=$1; shift
24
25     cd /etc/vservers/.distributions/${fcdistro}
26     if [ -f yum/yum.conf ] ; then
27         echo "Initializing yum.conf in $vserver from $(pwd)/yum"
28         cp yum/yum.conf /vservers/$vserver/etc/yum.conf
29     else
30         echo "Cannot initialize yum.conf in $vserver - using $fcdistro default"
31     fi
32
33     if [ -d yum.repos.d ] ; then
34         echo "Initializing yum.repos.d in $vserver from $(pwd)/yum.repos.d"
35         rm -rf /vservers/$vserver/etc/yum.repos.d
36         tar cf - yum.repos.d | tar -C /vservers/$vserver/etc -xvf -
37     else
38         echo "Cannot initialize yum.repos.d in $vserver"
39     fi
40     cd -
41 }    
42
43 function setup_vserver () {
44
45     set -x
46     set -e
47     trap failure ERR INT
48
49     vserver=$1; shift
50     fcdistro=$1; shift
51     personality=$1; shift
52
53     CLONED=0
54     # create the new vserver
55     if [ ! -d /etc/vservers/$vserver ] ; then
56         # check if we can create the vserver from a reference vserver
57         #if [ -d /vservers/${fcdistro}_reference ] ; then
58         if [ 0 -ne 0 ] ; then
59             $personality vserver $VERBOSE $vserver build -m clone -- --source /vservers/${fcdistro}_reference
60             CLONED=1
61         else
62             $personality vserver $VERBOSE $vserver build -m yum -- -d $fcdistro
63         fi
64     fi
65
66     if [ ! -z "$personality" ] ; then
67         l32=$(grep $personality /etc/vservers/$vserver/personality | wc -l)
68         [ $l32 -eq 0 ] && echo $personality >> /etc/vservers/$vserver/personality
69     fi
70
71     # set up appropriate vserver capabilities to mount, mknod and IPC_LOCK
72     BCAPFILE=/etc/vservers/$vserver/bcapabilities
73     touch $BCAPFILE
74     cap=$(grep ^CAP_SYS_ADMIN /etc/vservers/$vserver/bcapabilities | wc -l)
75     [ $cap -eq 0 ] && echo 'CAP_SYS_ADMIN' >> /etc/vservers/$vserver/bcapabilities
76     cap=$(grep ^CAP_MKNOD /etc/vservers/$vserver/bcapabilities | wc -l)
77     [ $cap -eq 0 ] && echo 'CAP_MKNOD' >> /etc/vservers/$vserver/bcapabilities
78     cap=$(grep ^CAP_IPC_LOCK /etc/vservers/$vserver/bcapabilities | wc -l)
79     [ $cap -eq 0 ] && echo 'CAP_IPC_LOCK' >> /etc/vservers/$vserver/bcapabilities
80
81     # start the vserver so we can do the following operations
82     if [ $CLONED -eq 0 ] ; then
83         $personality vyum $vserver -- -y install yum
84         $personality vserver $VERBOSE $vserver pkgmgmt internalize
85     fi
86     $personality vserver $VERBOSE $vserver start
87     $personality vserver $VERBOSE $vserver exec rm -f /var/lib/rpm/__db*
88     $personality vserver $VERBOSE $vserver exec rpm --rebuilddb
89
90     # minimal config in the vserver for yum to work
91     configure_yum_in_vserver $vserver $fcdistro
92
93     # set up resolv.conf
94     cp /etc/resolv.conf /vservers/$vserver/etc/resolv.conf
95 }
96
97 function devel_tools () {
98
99     set -x 
100     set -e 
101     trap failure ERR INT
102
103     vserver=$1; shift
104     fcdistro=$1; shift
105     pldistro=$1; shift
106     personality=$1; shift
107
108     # check for .lst file based on pldistro
109     lst=${pldistro}-devel.lst
110     if [ -f $lst ] ; then
111         echo "$COMMAND: Using $lst"
112     else
113         echo "$COMMAND : Cannot locate $lst - exiting"
114         usage
115     fi
116
117     # install individual packages, then groups
118     packages=$(pl_getPackages2 ${fcdistro} $lst)
119     groups=$(pl_getGroups2 ${fcdistro} $lst)
120
121     [ -n "$packages" ] && $personality vserver $vserver exec yum -y install $packages
122     [ -n "$groups" ] && $personality vserver $vserver exec yum -y groupinstall $groups
123     return 0
124 }
125
126 function post_install () {
127
128     set -x 
129     set -e 
130     trap failure ERR INT
131
132     vserver=$1; shift
133     personality=$1; shift
134
135 ### From myplc-devel-native.spec
136 # be careful to backslash $ in this, otherwise it's the root context that's going to do the evaluation
137     cat << EOF | $personality vserver $VERBOSE $vserver exec bash -x
138     # set up /dev/loop* in vserver
139     for i in \$(seq 0 255) ; do
140         mknod -m 640 /dev/loop\$i b 7 \$i
141     done
142     
143     # create symlink for /dev/fd
144     ln -fs /proc/self/fd /dev/fd
145
146     # modify /etc/rpm/macros to not use /sbin/new-kernel-pkg
147     sed -i 's,/sbin/new-kernel-pkg:,,' /etc/rpm/macros
148     if [ -h "/sbin/new-kernel-pkg" ] ; then
149         filename=\$(readlink -f /sbin/new-kernel-pkg)
150         if [ "\$filename" == "/sbin/true" ] ; then
151                 echo "WARNING: /sbin/new-kernel-pkg symlinked to /sbin/true"
152                 echo "\tmost likely /etc/rpm/macros has /sbin/new-kernel-pkg declared in _netsharedpath."
153                 echo "\tPlease remove /sbin/new-kernel-pkg from _netsharedpath and reintall mkinitrd."
154                 exit 1
155         fi
156     fi
157     
158     # customize root's prompt
159     cat << PROFILE > /root/.profile
160 export PS1="[$vserver] \\w # "
161 PROFILE
162
163     uid=2000
164     gid=2000
165     
166     # add a "build" user to the system
167     builduser=\$(grep "^build:" /etc/passwd | wc -l)
168     if [ \$builduser -eq 0 ] ; then
169         groupadd -o -g \$gid build;
170         useradd -o -c 'Automated Build' -u \$uid -g \$gid -n -M -s /bin/bash build;
171     fi
172
173 # Allow build user to build certain RPMs as root
174     if [ -f /etc/sudoers ] ; then
175         buildsudo=\$(grep "^build.*ALL=(ALL).*NOPASSWD:.*ALL"  /etc/sudoers | wc -l)
176         if [ \$buildsudo -eq 0 ] ; then
177             echo "build   ALL=(ALL)       NOPASSWD: ALL" >> /etc/sudoers
178         fi
179         sed -i 's,^Defaults.*requiretty,#Defaults requiretty,' /etc/sudoers
180     fi
181 #
182 EOF
183
184 }
185
186 function usage () {
187     set +x 
188     echo "Usage: $COMMAND [-s] [-p] [-v] vserver-name distribution pldistro [personality]"
189     echo "Requirements: you need to have a vserver-compliant kernel,"
190     echo "  as well as the util-vserver RPM installed"
191     echo "Description:"
192     echo "  This command creates a fresh vserver instance, with the specified name"
193     echo "  The root filesystem is created from the specified distribution, e.g. fc6"
194     echo "  The third argument denotes a pldistro, e.g. onelab"
195     echo "  The last, optional, argument defaults to linux32"
196     echo "This is done in three steps"
197     echo " (*) setup phase : vserver creation, yum internalization and config (from /etc/vservers)"
198     echo " (*) tools install : the tools required for building are installed"
199     echo "    to this end we search for a .lst file that specifies the pkgs & groups"
200     echo "    assuming the above that pldistro is onelab:"
201     echo "    (*) we first check for onelab-devel-fc6.lst"
202     echo "    (*) and then for onelab-devel.lst"
203     echo " (*) post-install : create a build user, + various tunings required"
204     echo "Options:"
205     echo " -s : skips the setup phase"
206     echo " -t : skips the tools phase"
207     echo " -p : skips the post-install"
208     echo " -v : passes -v to calls to vserver"
209     exit 1
210 }
211
212 ### parse args and 
213 function main () {
214
215     set -e
216     trap failure ERR INT
217
218     DO_SETUP=true
219     DO_TOOLS=true
220     DO_POST=true
221     VERBOSE=
222     while getopts "stpvh" opt ; do
223         case $opt in
224             s) DO_SETUP="" ;;
225             t) DO_TOOLS="" ;;
226             p) DO_POST="" ;;
227             v) VERBOSE="-v" ;;
228             h|*) usage ;;
229         esac
230     done
231         
232     shift $(($OPTIND - 1))
233     
234     [[ -z "$@" ]] && usage
235     vserver=$1 ; shift
236     [[ -z "$@" ]] && usage
237     fcdistro=$1 ; shift
238     [[ -z "$@" ]] && usage
239     pldistro=$1 ; shift
240     if [[ -z "$@" ]] ; then
241         personality=linux32
242     else
243         personality=$1; shift
244     fi
245     [[ -n "$@" ]] && usage
246
247     [ -n "$DO_SETUP" ] && setup_vserver $vserver $fcdistro $personality
248     [ -n "$DO_TOOLS" ] && devel_tools $vserver $fcdistro $pldistro $personality
249     [ -n "$DO_POST" ] && post_install $vserver $personality
250
251 }
252
253 main "$@"