#!/bin/bash
# copyright (C) 2013 FUJITSU LIMITED All Rights Reserved

# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; version 2
# of the License.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  
# 02110-1301, USA.

# check root
if [ ${EUID:-${UID}} != 0 ]; then
    echo "error: Because you are not root, you cannot execute this command. "
    exit 1
fi

FLG_C=0
FLG_N=0
FLG_I=0
FLG_M=0
FLG_S=0

# check options
while getopts c:n:i:m:s: OPT
do
  case $OPT in
    "c" ) FLG_C=1; CPURATE=$OPTARG  ;;
    "i" ) FLG_I=1; CPUN=$OPTARG  ;;
    "n" ) FLG_N=1; NUMA=$OPTARG  ;;
    "m" ) FLG_M=1; MEMLIMIT=$OPTARG  ;;
    "s" ) FLG_S=1; MEMSWLIMIT=$OPTARG  ;;
  esac
done

shift `expr $OPTIND - 1`

if [ $# -lt 1 ] ; then
  echo "usage : set [ -c CPURATE ] [ -i CPUN ] [ -n NUMA ] [ -m MEMLIMIT ] [ -s MEMSWLIMIT ] LXCNAME ..."
  exit -1
fi

lsdir() {
  ls -f --ind=none $1 | sed '/^\.\{1,2\}$/d'
}

# When uuid is specified, the container name is returned. 
uuid2name() {
  NAME=$1
  if [ -d /opt/lxcf/${NAME} ]; then
    echo -n $NAME
    return
  fi
  for i in `lsdir /etc/lxcf/rsc`
  do
     if [ x$NAME == x`cat /etc/lxcf/rsc/${i}/uuid` ]; then
       echo -n $i
       return
     fi
  done
  echo -n $NAME
}

for i in $*
do
  LXCNAME=`uuid2name ${i}`

  if [ $FLG_C -eq 1 ] ; then
    /usr/sbin/lxcf cpurate $LXCNAME $CPURATE
  fi
  if [ $FLG_I -eq 1 ] ; then
    /usr/sbin/lxcf cpun $LXCNAME $CPUN
  fi
  if [ $FLG_N -eq 1 ] ; then
    /usr/sbin/lxcf numa $LXCNAME $NUMA
  fi
  if [ $FLG_M -eq 1 ] ; then
    /usr/sbin/lxcf memlimit $LXCNAME $MEMLIMIT
  fi
  if [ $FLG_S -eq 1 ] ; then
    /usr/sbin/lxcf memswlimit $LXCNAME $MEMSWLIMIT
  fi
done

exit 0

