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: vuseradd,v 1.28 2007/07/05 19:05:14 dhozac Exp $
   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:0.0.0.0/0 \
  82                 --flags persistent,~info_init,sched_hard
  83     RETVAL=$?
  84     DIR=$__CONFDIR/$NAME
  85     if [ $RETVAL -ne 0 ] ; then
  86         echo "Error $RETVAL building $DIR"
  87         rm -rf $DIR $__DEFAULT_VSERVERDIR/$NAME
  88     fi
  89     mkdir -p $DIR/apps/init $DIR/rlimits $DIR/sched $DIR/dlimits/0
  90     echo default > $DIR/apps/init/mark
  91     echo 1000 > $DIR/rlimits/nproc
  92
  93     # Set persistent for the network context
  94     echo persistent > $DIR/nflags
  95
  96     # Set up the scheduler
  97     echo 1000 > $DIR/sched/interval
  98     echo 1000 > $DIR/sched/interval2
  99     echo 0 > $DIR/sched/fill-rate
 100     echo 32 > $DIR/sched/fill-rate2
 101     touch $DIR/sched/idle-time
 102     echo 100 > $DIR/sched/tokens
 103     echo 50 > $DIR/sched/tokens-min
 104     echo 100 > $DIR/sched/tokens-max
 105
 106     # Set up disk limits (unlimited)
 107     echo `$_READLINK $DIR/vdir` > $DIR/dlimits/0/directory
 108     echo 2 > $DIR/dlimits/0/reserved
 109     echo -1 > $DIR/dlimits/0/inodes_total
 110     echo -1 > $DIR/dlimits/0/space_total
 111
 112     # Remove the basically empty guest directory
 113     rm -rf $__DEFAULT_VSERVERDIR/$NAME
 114     # Move the guest back
 115     if [ "$HAS_VSERVERDIR" = 1 ] ; then
 116         mv "$TMP/$NAME" $__DEFAULT_VSERVERDIR/$NAME
 117         rm -rf "$TMP"
 118     fi
 119 fi
 120
 121 if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
 122     # Check the cache
 123     if [ "$TYPE" = "default" ] ; then
 124         for i in "$__DEFAULT_VSERVERDIR/.vcache/"* ; do
 125             [ -d "$i" ] && mv "$i" "$__DEFAULT_VSERVERDIR/$NAME" && break
 126         done
 127     fi
 128
 129     # Build slice from reference image
 130     if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
 131         REF="$__DEFAULT_VSERVERDIR/.vref/$TYPE"
 132
 133         # Build in temporary directory
 134         mkdir -p "$__DEFAULT_VSERVERDIR/.vtmp"
 135         TMP=$(mktemp -d "$__DEFAULT_VSERVERDIR/.vtmp/$NAME.XXXXXX")
 136         "$_VCLONE" "$REF"/ "$TMP"/
 137         RETVAL=$?
 138
 139         # Move it to its permanent location when complete
 140         if [ $RETVAL -eq 0 ] ; then
 141             mv "$TMP" "$__DEFAULT_VSERVERDIR/$NAME"
 142         else
 143             echo "Error $RETVAL building $__DEFAULT_VSERVERDIR/$NAME"
 144             rm -rf "$TMP" $__CONFDIR/$NAME $__PKGSTATEDIR/$NAME.ctx
 145             userdel -r $NAME
 146             exit $RETVAL
 147         fi
 148     fi
 149 fi
 150
 151 if [ -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
 152     # Fix permissions
 153     chmod 755 "$__DEFAULT_VSERVERDIR/$NAME"
 154
 155     # Add user in vserver
 156     $_VSERVER ----insecure $NAME suexec root sh -c \
 157         "groupadd -g $GROUPID $GROUPNAME ; useradd -u $USERID -g $GROUPID -p '' $NAME"
 158
 159     # Add an unrestricted entry to /etc/sudoers file
 160     if [ -f "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ] && \
 161        ! grep -q "^$NAME" "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ; then
 162         echo "$NAME     ALL=(ALL)       ALL" >> "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers"
 163     fi
 164 fi
 165
 166 exit 0