mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
316 lines
6.6 KiB
Bash
Executable File
316 lines
6.6 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Copyright (C) 2011 Citrix Systems, 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: createvm.sh 10292 2010-07-07 00:24:04Z edison $ $HeadURL: svn://svn.lab.vmops.com/repos/vmdev/java/scripts/storage/qcow2/createvm.sh $
|
|
# createvm.sh -- create a vm image
|
|
|
|
usage() {
|
|
echo "Usage (clone VM from template): createvm.sh -t <template dir> -i <rootdisk dir> -f <datadisk folder> -n <datadisk name> -s <datadisk size in GB>\n\
|
|
Usage (create blank rootdisk): createvm.sh -i <rootdisk dir> -S <rootdisk size in GB> \n"
|
|
}
|
|
|
|
check_params() {
|
|
# Either this VM is being cloned from a template, or it is being booted from an ISO and the rootdisk size will be specified
|
|
if [ "$tflag$Sflag" != "10" ] && [ "$tflag$Sflag" != "01" ]
|
|
then
|
|
return 1
|
|
fi
|
|
|
|
# A rootdisk folder must be specified
|
|
if [ "$iflag" != "1" ]
|
|
then
|
|
return 1
|
|
fi
|
|
|
|
# If a datadisk folder is specified, then a datadisk size and a datadisk name must be specified
|
|
if [ "$fflag" == "1" ]
|
|
then
|
|
if [ "$sflag$nflag" != "11" ]
|
|
then
|
|
return 1
|
|
fi
|
|
fi
|
|
|
|
return 0
|
|
}
|
|
|
|
cleanup_and_exit_if_error() {
|
|
local return_code=$1
|
|
local msg=$2
|
|
local rootdiskfolder=$3
|
|
local datadiskfolder=$4
|
|
local datadiskname=$5
|
|
|
|
if [ $return_code -gt 0 ]
|
|
then
|
|
cleanup_disks $rootdiskfolder $datadiskfolder $datadiskname
|
|
exit_if_error $return_code "$msg"
|
|
fi
|
|
}
|
|
|
|
cleanup_disks() {
|
|
local rootdiskfolder=$1
|
|
local datadiskfolder=$2
|
|
local datadiskname=$3
|
|
|
|
local datadiskpath=""
|
|
if [ "$datadiskfolder" != "" ] && [ "$datadiskname" != "" ]
|
|
then
|
|
datadiskpath="${datadiskfolder}/${datadiskname}"
|
|
fi
|
|
|
|
if [ "$rootdiskfolder" != "" ] && [ -d $rootdiskfolder ]
|
|
then
|
|
rm -rf $rootdiskfolder
|
|
fi
|
|
|
|
if [ "$datadiskpath" != "" ] && [ -f $datadiskpath ]
|
|
then
|
|
rm $datadiskpath
|
|
fi
|
|
|
|
return 0
|
|
}
|
|
|
|
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_rootdisk() {
|
|
local rootdiskfolder=$1
|
|
|
|
make_folder $rootdiskfolder
|
|
|
|
if [ -f ${rootdiskfolder}/rootdisk ]
|
|
then
|
|
return 1
|
|
else
|
|
return 0
|
|
fi
|
|
}
|
|
|
|
check_datadisk() {
|
|
local datadiskfolder=$1
|
|
local datadiskname=$2
|
|
|
|
make_folder $datadiskfolder
|
|
|
|
if [ -f ${datadiskfolder}/${datadiskname} ]
|
|
then
|
|
return 1
|
|
else
|
|
return 0
|
|
fi
|
|
}
|
|
|
|
strip_leading_slash() {
|
|
local folder=$1
|
|
|
|
if [ ${folder:0:1} != / ]
|
|
then
|
|
folder=/$folder
|
|
fi
|
|
|
|
echo $folder
|
|
}
|
|
|
|
clone_template_to_rootdisk() {
|
|
local rootdiskfolder=$1
|
|
local templatepath=$2
|
|
|
|
curDir=$(pwd)
|
|
cd $rootdiskfolder
|
|
qemu-img create -f qcow2 -b $templatepath ${rootdiskfolder}/rootdisk
|
|
cd $curDir
|
|
|
|
return $?
|
|
}
|
|
|
|
create_blank_rootdisk() {
|
|
local rootdiskfolder=$1
|
|
local rootdisksize=$2
|
|
|
|
rootdisksize=$(convert_size_to_gb $rootdisksize)
|
|
|
|
if [ $? -gt 0 ]
|
|
then
|
|
return 1
|
|
fi
|
|
|
|
qemu-img create -f qcow2 ${rootdiskfolder}/rootdisk $rootdisksize
|
|
|
|
return $?
|
|
}
|
|
|
|
create_datadisk() {
|
|
local datadiskfolder=$1
|
|
local datadiskname=$2
|
|
local datadisksize=$3
|
|
local diskfmt=$4
|
|
|
|
datadisksize=$(convert_size_to_gb $datadisksize)
|
|
|
|
if [ $? -gt 0 ]
|
|
then
|
|
return 1
|
|
fi
|
|
|
|
qemu-img create -f $diskfmt ${datadiskfolder}/${datadiskname} $datadisksize
|
|
|
|
return $?
|
|
}
|
|
|
|
convert_size_to_gb() {
|
|
local size=$1
|
|
|
|
suffix=${size:(-1)}
|
|
case $suffix in
|
|
M)
|
|
;;
|
|
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
|
|
|
|
tflag=0
|
|
iflag=0
|
|
Sflag=0
|
|
fflag=0
|
|
sflag=0
|
|
nflag=0
|
|
|
|
while getopts 't:i:S:f:s:n:u:' OPTION
|
|
do
|
|
case $OPTION in
|
|
t) tflag=1
|
|
templatepath="$OPTARG"
|
|
;;
|
|
i) iflag=1
|
|
rootdiskfolder="$OPTARG"
|
|
;;
|
|
S) Sflag=1
|
|
rootdisksize="$OPTARG"
|
|
;;
|
|
f) fflag=1
|
|
datadiskfolder="$OPTARG"
|
|
;;
|
|
s) sflag=1
|
|
datadisksize="$OPTARG"
|
|
;;
|
|
n) nflag=1
|
|
datadiskname="$OPTARG"
|
|
;;
|
|
?) usage
|
|
exit 2
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Check all parameters
|
|
#check_params
|
|
#exit_if_error $? "$(usage)"
|
|
|
|
if [ -n "$rootdiskfolder" ]
|
|
then
|
|
# Create the rootdisk folder if necessary, and make sure there is no existing rootdisk there
|
|
check_rootdisk $rootdiskfolder
|
|
exit_if_error $? "Failed to create rootdisk; a rootdisk already exists at $rootdiskfolder."
|
|
|
|
if [ "$tflag" == "1" ]
|
|
then
|
|
# A template path was passed in, so clone the template to a new rootdisk
|
|
clone_template_to_rootdisk $rootdiskfolder $templatepath
|
|
exit_if_error $? "Failed to clone template $templatepath to $rootdiskfolder/rootdisk."
|
|
else
|
|
# A template path was not passed in, so create a blank rootdisk at the rootdisk folder
|
|
create_blank_rootdisk $rootdiskfolder $rootdisksize
|
|
exit_if_error $? "Failed to create a blank rootdisk at $rootdiskfolder/rootdisk."
|
|
fi
|
|
fi
|
|
|
|
if [ -n "$datadisksize" ]
|
|
then
|
|
# Create the datadisk folder if necessary, and make sure there is no existing datadisk there
|
|
check_datadisk $datadiskfolder $datadiskname
|
|
cleanup_and_exit_if_error $? "Failed to create datadisk in $datadiskfolder; datadisk with $datadiskname already exists." $rootdiskfolder
|
|
|
|
# Create the datadisk
|
|
create_datadisk $datadiskfolder $datadiskname $datadisksize qcow2
|
|
cleanup_and_exit_if_error $? "Failed to create datadisk in $datadiskfolder of size $datadisksize." $rootdiskfolder $datadiskfolder $datadiskname
|
|
else
|
|
# Create a datadisk for domr/domp
|
|
#create_datadisk $rootdiskfolder datadisk 10M raw
|
|
#exit_if_error $? "Failed to create datadisk"
|
|
loopdev=$(losetup -f)
|
|
losetup $loopdev $datadiskfolder
|
|
retry=10
|
|
while [ $retry -gt 0 ]
|
|
do
|
|
success=$(losetup -a |grep $loopdev)
|
|
if [ $? -eq 0 ]
|
|
then
|
|
break
|
|
fi
|
|
retry=$(($retry-1))
|
|
sleep 1
|
|
done
|
|
mkfs -t ext3 $loopdev &>/dev/null
|
|
retry=10
|
|
while [ $retry -gt 0 ]
|
|
do
|
|
losetup -d $loopdev
|
|
if [ $? -eq 0 ]
|
|
then
|
|
break
|
|
fi
|
|
retry=$(($retry-1))
|
|
sleep 1
|
|
done
|
|
fi
|
|
|
|
exit 0
|