Setting tag util-vserver-pl-0.4-9
[util-vserver-pl.git] / scripts / vuseradd
1 #!/bin/bash
2 #
3 # useradd(8) wrapper for vservers
4 #
5 # Mark Huang <mlhuang@cs.princeton.edu>
6 # Copyright (C) 2004-2006 The Trustees of Princeton University
7 #
8 # $Id$
9 #
10
11 : ${UTIL_VSERVER_VARS:=/usr/lib/util-vserver/util-vserver-vars}
12 test -e "$UTIL_VSERVER_VARS" || {
13     echo "Can not find util-vserver installation; aborting..."
14     exit 1
15 }
16 . "$UTIL_VSERVER_VARS"
17
18 shopt -s nullglob
19
20 # Defaults
21 TYPE="default"
22
23 usage()
24 {
25     TYPES=
26     pushd "$__DEFAULT_VSERVERDIR/.vref" >/dev/null
27     for ref in * ; do
28         if [ -z "$TYPES" ] ; then
29             TYPES=$ref
30         else
31             TYPES="$TYPES, $ref"
32         fi
33     done
34     popd >/dev/null
35
36     echo "Usage: vuseradd [OPTION]... [NAME]"
37     echo "      -t      Reference image type ($TYPES)"
38     exit 1
39 }
40
41 # Get options
42 while getopts "t:" opt ; do
43     case $opt in
44         t)
45             TYPE="$OPTARG"
46             ;;
47         *)
48             usage
49             ;;
50     esac
51 done
52 shift $(($OPTIND - 1))
53
54 # Get slice name
55 [ -z "$1" ] && usage
56 NAME=$1
57
58 # Add slices group to /etc/group if not already present
59 groupadd slices 2>/dev/null || :
60
61 # Add slice name to /etc/passwd
62 useradd -g slices -s /bin/vsh $NAME -p '*'
63
64 USERID=`id -u $NAME`
65 GROUPID=`id -g $NAME`
66 GROUPNAME=`id -gn $NAME`
67
68 # Create /etc/vservers configuration files
69 if [ ! -d $__CONFDIR/$NAME ] ; then
70     # Move away the guest contents for now
71     if [ -d $__DEFAULT_VSERVERDIR/$NAME ] ; then
72         mkdir -p "$__DEFAULT_VSERVERDIR/.vtmp"
73         TMP=$(mktemp -d "$__DEFAULT_VSERVERDIR/.vtmp/$NAME.XXXXXX")
74         mv $__DEFAULT_VSERVERDIR/$NAME "$TMP"
75         HAS_VSERVERDIR=1
76     else
77         HAS_VSERVERDIR=0
78     fi
79
80     $_VSERVER $NAME build -m skeleton --context $USERID \
81                 --interface nodev:`hostname -i` \
82                 --interface nodev:127.0.0.1 \
83                 --flags persistent,~info_init
84     RETVAL=$?
85     DIR=$__CONFDIR/$NAME
86     if [ $RETVAL -ne 0 ] ; then
87         echo "Error $RETVAL building $DIR"
88         rm -rf $DIR $__DEFAULT_VSERVERDIR/$NAME
89     fi
90     mkdir -p $DIR/apps/init $DIR/rlimits $DIR/sched $DIR/dlimits/0 $DIR/sysctl/0
91     echo default > $DIR/apps/init/mark
92     echo 1000 > $DIR/rlimits/nproc.hard
93
94     # Set persistent for the network context
95     echo persistent,lback_allow > $DIR/nflags
96
97     # Set up the scheduler
98     echo 100 > $DIR/sched/interval
99     echo 1000 > $DIR/sched/interval2
100     echo 0 > $DIR/sched/fill-rate
101     echo 1 > $DIR/sched/fill-rate2
102     touch $DIR/sched/idle-time
103     echo 100 > $DIR/sched/tokens
104     echo 50 > $DIR/sched/tokens-min
105     echo 100 > $DIR/sched/tokens-max
106
107     # Set up disk limits (unlimited)
108     echo `$_READLINK $DIR/vdir` > $DIR/dlimits/0/directory
109     echo 2 > $DIR/dlimits/0/reserved
110     echo -1 > $DIR/dlimits/0/inodes_total
111     echo -1 > $DIR/dlimits/0/space_total
112
113     # Set up sysctl variables
114     echo net.ipv4.ip_forward > $DIR/sysctl/0/setting
115     echo 1 > $DIR/sysctl/0/value
116     
117     # Disable mount namespaces
118     touch $DIR/nonamespace
119
120     # Add spaces directory
121     mkdir -p $DIR/spaces
122
123     # Remove the basically empty guest directory
124     rm -rf $__DEFAULT_VSERVERDIR/$NAME
125     # Move the guest back
126     if [ "$HAS_VSERVERDIR" = 1 ] ; then
127         mv "$TMP/$NAME" $__DEFAULT_VSERVERDIR/$NAME
128         rm -rf "$TMP"
129     fi
130 fi
131
132 if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
133     # Check the cache
134     if [ "$TYPE" = "default" ] ; then
135         for i in "$__DEFAULT_VSERVERDIR/.vcache/"* ; do
136             [ -d "$i" ] && mv "$i" "$__DEFAULT_VSERVERDIR/$NAME" && break
137         done
138     fi
139
140     # Build slice from reference image
141     if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
142         REF="$__DEFAULT_VSERVERDIR/.vref/$TYPE"
143
144         # Build in temporary directory
145         mkdir -p "$__DEFAULT_VSERVERDIR/.vtmp"
146         TMP=$(mktemp -d "$__DEFAULT_VSERVERDIR/.vtmp/$NAME.XXXXXX")
147         "$_VCLONE" "$REF"/ "$TMP"/
148         RETVAL=$?
149
150         # Move it to its permanent location when complete
151         if [ $RETVAL -eq 0 ] ; then 
152             mv "$TMP" "$__DEFAULT_VSERVERDIR/$NAME"
153         else
154             echo "Error $RETVAL building $__DEFAULT_VSERVERDIR/$NAME"
155             rm -rf "$TMP" $__CONFDIR/$NAME $__PKGSTATEDIR/$NAME.ctx
156             userdel -r $NAME
157             exit $RETVAL
158         fi
159     fi
160 fi
161
162 if [ -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
163     # Fix permissions
164     chmod 755 "$__DEFAULT_VSERVERDIR/$NAME"
165
166     # Add user in vserver
167     $_VSERVER $NAME start --rescue sh -c \
168         "groupadd -g $GROUPID $GROUPNAME ; useradd -m -u $USERID -g $GROUPID -p '' $NAME"
169
170     # Stop the guest (since it's persistent)
171     $_VSERVER $NAME stop
172
173     # Add an unrestricted entry to /etc/sudoers file
174     if [ -f "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ] && \
175        ! grep -q "^$NAME" "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ; then
176         echo "$NAME     ALL=(ALL)       ALL" >> "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers"
177     fi
178
179     cp -a /dev/fuse $__DEFAULT_VSERVERDIR/$NAME/dev/
180
181 fi
182
183 exit 0