| #!/usr/bin/env bash |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| |
| |
| # $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() { |
| 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 |