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/cgroup $DIR/dlimits/0 $DIR/sysctl/0
  91     echo default > $DIR/apps/init/mark
  92
  93     # Set persistent for the network context
  94     echo persistent,lback_allow > $DIR/nflags
  95
  96     # Set default capabilities
  97     echo "CAP_NET_RAW" > $DIR/bcapabilities
  98     touch $DIR/ccapabilities
  99
 100     # Set up the scheduler
 101     echo 100 > $DIR/sched/interval
 102     echo 1000 > $DIR/sched/interval2
 103     echo 0 > $DIR/sched/fill-rate
 104     echo 1 > $DIR/sched/fill-rate2
 105     touch $DIR/sched/idle-time
 106     echo 100 > $DIR/sched/tokens
 107     echo 50 > $DIR/sched/tokens-min
 108     echo 100 > $DIR/sched/tokens-max
 109
 110     echo 1024 > $DIR/cgroup/cpu.shares
 111
 112     # Set up disk limits (10 GB)
 113     echo `$_READLINK $DIR/vdir` > $DIR/dlimits/0/directory
 114     echo 2 > $DIR/dlimits/0/reserved
 115     echo -1 > $DIR/dlimits/0/inodes_total
 116     echo 10000000 > $DIR/dlimits/0/space_total
 117
 118     # Set up sysctl variables
 119     echo net.ipv4.ip_forward > $DIR/sysctl/0/setting
 120     echo 1 > $DIR/sysctl/0/value
 121
 122     # Add spaces directory
 123     mkdir -p $DIR/spaces
 124
 125     # Remove the basically empty guest directory
 126     rm -rf $__DEFAULT_VSERVERDIR/$NAME
 127     # Move the guest back
 128     if [ "$HAS_VSERVERDIR" = 1 ] ; then
 129         mv "$TMP/$NAME" $__DEFAULT_VSERVERDIR/$NAME
 130         rm -rf "$TMP"
 131     fi
 132 fi
 133
 134 if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
 135     # Check the cache
 136     if [ "$TYPE" = "default" ] ; then
 137         for i in "$__DEFAULT_VSERVERDIR/.vcache/"* ; do
 138             [ -d "$i" ] && mv "$i" "$__DEFAULT_VSERVERDIR/$NAME" && break
 139         done
 140     fi
 141
 142     # Build slice from reference image
 143     if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
 144         REF="$__DEFAULT_VSERVERDIR/.vref/$TYPE"
 145
 146         # Build in temporary directory
 147         mkdir -p "$__DEFAULT_VSERVERDIR/.vtmp"
 148         TMP=$(mktemp -d "$__DEFAULT_VSERVERDIR/.vtmp/$NAME.XXXXXX")
 149         "$_VCLONE" "$REF"/ "$TMP"/
 150         RETVAL=$?
 151
 152         # Move it to its permanent location when complete
 153         if [ $RETVAL -eq 0 ] ; then
 154             mv "$TMP" "$__DEFAULT_VSERVERDIR/$NAME"
 155         else
 156             echo "Error $RETVAL building $__DEFAULT_VSERVERDIR/$NAME"
 157             rm -rf "$TMP" $__CONFDIR/$NAME $__PKGSTATEDIR/$NAME.ctx
 158             userdel -r $NAME
 159             exit $RETVAL
 160         fi
 161     fi
 162 fi
 163
 164 if [ -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
 165     # Fix permissions
 166     chmod 755 "$__DEFAULT_VSERVERDIR/$NAME"
 167
 168     # Add user in vserver
 169     $_VSERVER $NAME start --rescue sh -c \
 170         "groupadd -g $GROUPID $GROUPNAME ; useradd -m -u $USERID -g $GROUPID -p '' $NAME"
 171
 172     # Stop the guest (since it's persistent)
 173     $_VSERVER $NAME stop
 174
 175     # Add an unrestricted entry to /etc/sudoers file
 176     if [ -f "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ] && \
 177        ! grep -q "^$NAME" "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ; then
 178         echo "$NAME     ALL=(ALL)       ALL" >> "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers"
 179     fi
 180
 181     cp -a /dev/fuse $__DEFAULT_VSERVERDIR/$NAME/dev/
 182
 183 fi
 184
 185 exit 0