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