cloudstack/scripts/storage/qcow2/managevolume.sh
2011-04-14 11:23:14 -07:00

185 lines
3.1 KiB
Bash

#!/usr/bin/env bash
#
# Copyright (C) 2010 Cloud.com, Inc. All rights reserved.
#
# This software is licensed under the GNU General Public License v3 or later.
#
# It 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, either version 3 of the License, or any later version.
# 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, see <http://www.gnu.org/licenses/>.
#
# $Id: managevolume.sh 9132 2010-06-04 20:17:43Z manuel $ $HeadURL: svn://svn.lab.vmops.com/repos/vmdev/java/scripts/storage/qcow2/managevolume.sh $
# modifyvolume.sh -- add or delete a disk volume
#
usage() {
echo "Usage: modifydisk.sh -f <folder of volume> -p <path of volume> -s <size of volume> [-d <delete>] \n"
}
check_params() {
# The folder, path, and size must be passed in
if [ "$fflag$pflag$sflag" != "111" ]
then
return 1
else
return 0
fi
}
cleanup_and_exit_if_error() {
local return_code=$1
local msg=$2
local path=$3
if [ $return_code -gt 0 ]
then
delete_disk $path
exit_if_error $return_code "$msg"
fi
}
exit_if_error() {
local return_code=$1
local msg=$2
if [ $return_code -gt 0 ]
then
printf "${msg}\n"
exit 1
fi
}
make_folder() {
local folder=$1
if [ ! -d ${folder} ]
then
mkdir -p ${folder}
fi
}
check_disk() {
local folder=$1
local path=$2
make_folder $folder
if [ -f $path ]
then
return 1
else
return 0
fi
}
delete_disk() {
local path=$1
if [ -f $path ]
then
rm $path
fi
return 0
}
create_disk() {
local path=$1
local size=$2
size=$(convert_size_to_gb $size)
if [ $? -gt 0 ]
then
return 1
fi
qemu-img create -f qcow2 $path $size
return $?
}
convert_size_to_gb() {
local size=$1
suffix=${size:(-1)}
case $suffix in
G)
;;
[0-9]) size=${size}G
;;
*) printf "Error in disk size: expect G as a suffix or no suffix\n"
return 1
;;
esac
echo $size
return 0
}
# set -x
fflag=0
pflag=0
sflag=0
dflag=0
while getopts 'f:p:s:d:' OPTION
do
case $OPTION in
f) fflag=1
folder="$OPTARG"
;;
p) pflag=1
path="$OPTARG"
;;
s) sflag=1
size="$OPTARG"
;;
d) dflag=1
;;
?) usage
exit 2
;;
esac
done
# Check all parameters
check_params
exit_if_error $? "$(usage)"
if [ "$dflag" == "0" ]
then
# Add the volume
# Create the folder if necessary, and make sure there is no existing disk there
check_disk $folder $path
cleanup_and_exit_if_error $? "Failed to create disk at $path; path already exists." $path
# Create the disk
create_disk $path $size
cleanup_and_exit_if_error $? "Failed to create disk at $path of size $datadisksize." $path
else
# Delete the volume
delete_disk $path
fi
exit 0