#!/bin/bash

# Copyright (C) 2011 Red Hat, Inc. All rights reserved.
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions
# of the GNU General Public License v.2.
#
# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

. lib/test

# is_in_sync <VG/LV>
function is_in_sync()
{
	local dm_name
	local a
	local b
	local idx

	dm_name=`echo $1 | sed s:-:--: | sed s:/:-:`

	if ! a=(`dmsetup status $dm_name`); then
		echo "Unable to get sync status of $1"
		exit 1
	fi

	# 6th argument is the sync ratio for RAID and mirror
	echo ${a[@]}
	if [ ${a[2]} = "raid" ]; then
		# Last argument is the sync ratio for RAID
		idx=$((${#a[@]} - 1))
	elif [ ${a[2]} = "mirror" ]; then
		# 4th Arg tells us how far to the sync ratio
		idx=$((${a[3]} + 4))
	else
		echo "Unable to get sync ratio for target type '${a[2]}'"
		exit 1
	fi
	b=(`echo ${a[$idx]} | sed s:/:' ':`)

	if [ ${b[0]} != ${b[1]} ]; then
		echo "$dm_name (${a[3]}) is not in-sync"
		return 1
	fi

	if [[ ${a[$(($idx - 1))]} =~ a ]]; then
		echo "$dm_name in-sync, but 'a' characters in health status"
		exit 1
	fi

	if [ ${a[2]} = "raid" ]; then
		echo "$dm_name (${a[3]}) is in-sync"
	else
		echo "$dm_name (${a[2]}) is in-sync"
	fi

	return 0
}

# wait_for_sync <VG/LV>
function wait_for_sync()
{
	local i=0

	while ! is_in_sync $1; do
		sleep 2
		i=$(($i + 1))
		if [ $i -gt 500 ]; then
			echo "Sync is taking too long - assume stuck"
			exit 1
		fi
	done
}

function is_raid_available()
{
	local a

	modprobe dm-raid
	a=(`dmsetup targets | grep raid`)
	if [ -z $a ]; then
		echo "RAID target not available"
		return 1
	fi
	if [ ${a[1]} != "v1.1.0" ]; then
		echo "Bad RAID version"
		return 1
	fi

	return 0
}

########################################################
# MAIN
########################################################
is_raid_available || exit 200

aux prepare_vg 5 80

###########################################
# RAID1 convert tests
###########################################
for i in 1 2 3 4; do
	for j in 1 2 3 4; do
		if [ $i -eq 1 ]; then
			from="linear"
		else
			from="$i-way"
		fi
		if [ $j -eq 1 ]; then
			to="linear"
		else
			to="$j-way"
		fi
		echo "Converting from $from to $to"
		if [ $i -eq 1 ]; then
			# Shouldn't be able to create with just 1 image
			not lvcreate --type raid1 -m 0 -l 2 -n $lv1 $vg

			lvcreate -l 2 -n $lv1 $vg
		else
			lvcreate --type raid1 -m $(($i - 1)) -l 2 -n $lv1 $vg
			wait_for_sync $vg/$lv1
		fi
		lvconvert -m $((j - 1))  $vg/$lv1

		# FIXME: ensure no residual devices

		if [ $j -eq 1 ]; then
			check linear $vg $lv1
		fi
		lvremove -ff $vg
	done
done

#
# FIXME: Add tests that specify particular devices to be removed
#

###########################################
# RAID1 split tests
###########################################
# 3-way to 2-way/linear
lvcreate --type raid1 -m 2 -l 2 -n $lv1 $vg
wait_for_sync $vg/$lv1
lvconvert --splitmirrors 1 -n $lv2 $vg/$lv1
check lv_exists $vg $lv1
check linear $vg $lv2
# FIXME: ensure no residual devices
lvremove -ff $vg

# 2-way to linear/linear
lvcreate --type raid1 -m 1 -l 2 -n $lv1 $vg
wait_for_sync $vg/$lv1
lvconvert --splitmirrors 1 -n $lv2 $vg/$lv1
check linear $vg $lv1
check linear $vg $lv2
# FIXME: ensure no residual devices
lvremove -ff $vg

# 3-way to linear/2-way
lvcreate --type raid1 -m 2 -l 2 -n $lv1 $vg
wait_for_sync $vg/$lv1
# FIXME: Can't split off a RAID1 from a RAID1 yet
should lvconvert --splitmirrors 2 -n $lv2 $vg/$lv1
#check linear $vg $lv1
#check lv_exists $vg $lv2
# FIXME: ensure no residual devices
lvremove -ff $vg

###########################################
# RAID1 split + trackchanges / merge
###########################################
# 3-way to 2-way/linear
lvcreate --type raid1 -m 2 -l 2 -n $lv1 $vg
wait_for_sync $vg/$lv1
lvconvert --splitmirrors 1 --trackchanges $vg/$lv1
check lv_exists $vg $lv1
check linear $vg ${lv1}_rimage_2
lvconvert --merge $vg/${lv1}_rimage_2
# FIXME: ensure no residual devices
lvremove -ff $vg

###########################################
# Mirror to RAID1 conversion
###########################################
for i in 1 2 3 ; do
	lvcreate --type mirror -m $i -l 2 -n $lv1 $vg
	wait_for_sync $vg/$lv1
	lvconvert --type raid1 $vg/$lv1
	lvremove -ff $vg
done

exit 0