- add -t option for specifying reference type (e.g. "default" or

[util-vserver.git] / scripts / vuseradd

diff --git a/scripts/vuseradd b/scripts/vuseradd
index 177466a..23118bc 100755 (executable)
--- a/scripts/vuseradd
+++ b/scripts/vuseradd
@@ -2,91 +2,126 @@
 #
 # useradd(8) wrapper for vservers
 #
-# Copyright (c) 2004  The Trustees of Princeton University (Trustees).
+# Mark Huang <mlhuang@cs.princeton.edu>
+# Copyright (C) 2004-2006 The Trustees of Princeton University
 #
-# $Id: vuseradd,v 1.17 2004/11/17 19:35:47 mef Exp $
+# $Id: vuseradd,v 1.25 2005/11/08 00:22:59 smuir Exp $
 #
 
-: ${UTIL_VSERVER_VARS:=$(dirname $0)/util-vserver-vars}
+: ${UTIL_VSERVER_VARS:=/usr/lib/util-vserver/util-vserver-vars}
 test -e "$UTIL_VSERVER_VARS" || {
     echo "Can not find util-vserver installation; aborting..."
     exit 1
 }
 . "$UTIL_VSERVER_VARS"
 
+shopt -s nullglob
+
+# Defaults
+TYPE="default"
+
 usage()
 {
-    echo "usage: $0 name"
+    TYPES=
+    pushd "$__DEFAULT_VSERVERDIR/.vref" >/dev/null
+    for ref in * ; do
+       if [ -z "$TYPES" ] ; then
+           TYPES=$ref
+       else
+           TYPES="$TYPES, $ref"
+       fi
+    done
+    popd >/dev/null
+
+    echo "Usage: vuseradd [OPTION]... [NAME]"
+    echo "     -t      Reference image type ($TYPES)"
     exit 1
 }
 
+# Get options
+while getopts "t:" opt ; do
+    case $opt in
+       t)
+           TYPE="$OPTARG"
+           ;;
+       *)
+           usage
+           ;;
+    esac
+done
+shift $(($OPTIND - 1))
+
+# Get slice name
 [ -z "$1" ] && usage
 NAME=$1
 
-# add slices group if not already present
+# Add slices group to /etc/group if not already present
 groupadd slices 2>/dev/null || :
 
-# add user
+# Add slice name to /etc/passwd
 useradd -g slices -s /bin/vsh $NAME
 
-USERID=$(awk -F: "\$1 == \"$NAME\" { print \$3 }" < /etc/passwd)
-GROUPID=$(awk -F: "\$1 == \"slices\" { print \$3 }" < /etc/group)
+# openssh-server 3.8 and above refuse login for "locked" accounts
+for file in /etc/passwd /etc/shadow ; do
+    [ -f $file ] && sed -i -e "s/$NAME:\!\!:\(.*\)/$NAME:*:\1/" $file
+done
+
+USERID=`id -u $NAME`
+GROUPID=`id -g $NAME`
+GROUPNAME=`id -gn $NAME`
 
-# create vserver configuration file
-if [ ! -f /etc/vservers/$NAME.conf ] ; then
+# Create /etc/vservers configuration file
+if [ ! -f $__CONFDIR/$NAME.conf ] ; then
     sed \
        -e "s/.*S_CONTEXT=.*/S_CONTEXT=$USERID/" \
        -e "s/.*ONBOOT=.*/ONBOOT=yes/" \
-       < $PKGLIBDIR/sample.conf \
-       > /etc/vservers/$NAME.conf
+       < $__PKGLIBDIR/defaults/sample.conf \
+       > $__CONFDIR/$NAME.conf
 fi
 
-if [ ! -d "$VROOTDIR/$NAME" ] ; then
-    # check the cache
-    shopt -s nullglob
-    for i in "$VROOTDIR/.vcache/"* ; do
-       [ -d "$i" ] && mv "$i" "$VROOTDIR/$NAME" && break
-    done
-    # build vserver
-    if [ ! -d "$VROOTDIR/$NAME" ] ; then
-       # build image in .vtmp
-       TMP=$(mktemp -d "$VROOTDIR/.vtmp/$NAME.XXXXXX")
-       "$PKGLIBDIR/vbuild" "$VROOTDIR/vserver-reference" "$TMP"
+if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
+    # Check the cache
+    if [ "$TYPE" = "default" ] ; then
+       for i in "$__DEFAULT_VSERVERDIR/.vcache/"* ; do
+           [ -d "$i" ] && mv "$i" "$__DEFAULT_VSERVERDIR/$NAME" && break
+       done
+    fi
+
+    # Build slice from reference image
+    if [ ! -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
+       REF="$__DEFAULT_VSERVERDIR/.vref/$TYPE"
+
+       # Build in temporary directory
+       mkdir -p "$__DEFAULT_VSERVERDIR/.vtmp"
+       TMP=$(mktemp -d "$__DEFAULT_VSERVERDIR/.vtmp/$NAME.XXXXXX")
+       "$__PKGLIBDIR/vbuild" "$REF" "$TMP"
        RETVAL=$?
-       # move it to .vcache when complete
-       if [ $RETVAL -ne 0 ] ; then 
-           echo "Error $RETVAL building $VROOTDIR/$NAME"
-           chattr -R -i "$TMP"
-           rm -rf "$TMP"
-           userdel $NAME
-           exit $RETVAL
+
+       # Move it to its permanent location when complete
+       if [ $RETVAL -eq 0 ] ; then 
+           mv "$TMP" "$__DEFAULT_VSERVERDIR/$NAME"
        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"
+           echo "Error $RETVAL building $__DEFAULT_VSERVERDIR/$NAME"
+           rm -rf "$TMP" $__CONFDIR/$NAME.conf $__PKGSTATEDIR/$NAME.ctx
+           userdel -r $NAME
+           exit $RETVAL
        fi
     fi
 fi
 
-if [ -d "$VROOTDIR/$NAME" ] ; then
-    # fix permissions
-    chmod 755 "$VROOTDIR/$NAME"
+if [ -d "$__DEFAULT_VSERVERDIR/$NAME" ] ; then
+    # Fix permissions
+    chmod 755 "$__DEFAULT_VSERVERDIR/$NAME"
 
-    # add user in vserver
-    vserver $NAME suexec root groupadd -g $GROUPID slices
-    vserver $NAME suexec root useradd -u $USERID -g $GROUPID -p '' $NAME
+    # Add user in vserver
+    $_VSERVER_LEGACY $NAME suexec root sh -c \
+       "groupadd -g $GROUPID $GROUPNAME ; useradd -u $USERID -g $GROUPID -p '' $NAME"
 
-    # add an unrestricted entry to /etc/sudoers file
-    if [ -f "$VROOTDIR/$NAME/etc/sudoers" ] && \
-       ! grep -q "^$NAME" "$VROOTDIR/$NAME/etc/sudoers" ; then
-       echo "$NAME     ALL=(ALL)       ALL" >> "$VROOTDIR/$NAME/etc/sudoers"
+    # Add an unrestricted entry to /etc/sudoers file
+    if [ -f "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ] && \
+       ! grep -q "^$NAME" "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers" ; then
+       echo "$NAME     ALL=(ALL)       ALL" >> "$__DEFAULT_VSERVERDIR/$NAME/etc/sudoers"
     fi
 fi
 
-# turn resource management on for vserver $NAME
-service resman start $NAME
+exit 0