Add support for static vserver IDs (needed for putting pl_conf in a slice)

[util-vserver.git] / scripts / vuseradd

diff --git a/scripts/vuseradd b/scripts/vuseradd
index 360b8ed..2be0ac9 100755 (executable)
--- a/scripts/vuseradd
+++ b/scripts/vuseradd
@@ -4,7 +4,7 @@
 #
 # Copyright (c) 2004  The Trustees of Princeton University (Trustees).
 #
-# $Id: vuseradd,v 1.12 2004/10/20 02:56:23 mef Exp $
+# $Id: vuseradd,v 1.21 2005/04/26 21:23:28 mlhuang Exp $
 #
 
 : ${UTIL_VSERVER_VARS:=$(dirname $0)/util-vserver-vars}
@@ -21,23 +21,21 @@ usage()
 }
 
 [ -z "$1" ] && usage
+[ "$1" == "--static" ] && { STATIC=yes; shift; }
 NAME=$1
 
 # add slices group if not already present
 groupadd slices 2>/dev/null || :
 
 # add user
-useradd -g slices -s /bin/vsh $NAME
+[ -z "$STATIC" ] && useradd -g slices -s /bin/vsh $NAME
 
-# automount keys
-if [ -d "/var/pl_sshd/keys/$NAME" ]; then
-    # (eval expands ~)
-    eval rm -rf ~$NAME/.ssh
-    eval ln -nsf /var/pl_sshd/keys/$NAME ~$NAME/.ssh
-fi
+# openssh-server 3.8 and above refuse login for "locked" accounts
+sed -i -e "s/$NAME:\!\!:\(.*\)/$NAME:*:\1/" /etc/shadow
 
-USERID=$(awk -F: "\$1 == \"$NAME\" { print \$3 }" < /etc/passwd)
-GROUPID=$(awk -F: "\$1 == \"slices\" { print \$3 }" < /etc/group)
+USERID=`id -u $NAME`
+GROUPID=`id -g $NAME`
+GROUPNAME=`id -gn $NAME`
 
 # create vserver configuration file
 if [ ! -f /etc/vservers/$NAME.conf ] ; then
@@ -61,14 +59,19 @@ if [ ! -d "$VROOTDIR/$NAME" ] ; then
        "$PKGLIBDIR/vbuild" "$VROOTDIR/vserver-reference" "$TMP"
        RETVAL=$?
        # move it to .vcache when complete
-       if [ $RETVAL -ne 0 ] || \
-          [ $(du -s "$TMP" | awk "{ print \$1 }") -lt \
-             $(du -s "$VROOTDIR/vserver-reference" | awk "{ print \$1 }") ] ; then
+       if [ $RETVAL -ne 0 ] ; then 
            echo "Error $RETVAL building $VROOTDIR/$NAME"
-           chattr -R -i "$TMP"
-           rm -rf "$TMP"
+           rm -rf "$TMP" /etc/vservers/$NAME.conf /var/run/vservers/$NAME.ctx
+           userdel -r $NAME
            exit $RETVAL
        else
+           # sanity check
+           vnewsize=$(du -s "$TMP" | awk "{ print \$1 }")
+           vrefsize=$(du -s "$VROOTDIR/vserver-reference" | awk "{ print \$1 }")
+           if [ $vnewsize -lt $vrefsize ] ; then
+               echo "WARNING: Unexpected for 'du -s $VROOTDIR/$NAME'=$vnewsize to be less than 'du -s $VROOTDIR/vserver-reference'=$vrefsize"
+           fi
+
            mv "$TMP" "$VROOTDIR/$NAME"
        fi
     fi
@@ -79,7 +82,7 @@ if [ -d "$VROOTDIR/$NAME" ] ; then
     chmod 755 "$VROOTDIR/$NAME"
 
     # add user in vserver
-    vserver $NAME suexec root groupadd -g $GROUPID slices
+    vserver $NAME suexec root groupadd -g $GROUPID $GROUPNAME
     vserver $NAME suexec root useradd -u $USERID -g $GROUPID -p '' $NAME
 
     # add an unrestricted entry to /etc/sudoers file
@@ -90,11 +93,6 @@ if [ -d "$VROOTDIR/$NAME" ] ; then
 fi
 
 # turn resource management on for vserver $NAME
-[ -x /usr/sbin/vs2ckrm ]  && /usr/sbin/vs2ckrm on $NAME
-[ -x /usr/sbin/cpulimit ]  && /usr/sbin/cpulimit on $NAME
-[ -x /usr/sbin/tasklimit ]  && /usr/sbin/tasklimit on $NAME
-[ -x /usr/sbin/iolimit ]  && /usr/sbin/iolimit on $NAME
-[ -x /usr/sbin/memlimit ]  && /usr/sbin/memlimit on $NAME
-[ -x /usr/sbin/bwlimit ]  && /usr/sbin/bwlimit on $NAME
-# PL2580 WAR: disklimit code is malfunctioning
-#[ -x /usr/sbin/disklimit ]  && /usr/sbin/disklimit on $NAME
+service resman start $NAME
+# XXX - resman doesn't print a trailing newline
+echo