#!/bin/sh
# Sub-tests that require a mounted partition.
set -e
partition="$1"

. /usr/share/os-prober/common.sh

do_unmount() {
	if [ "$mounted" ]; then
		if ! umount "$tmpmnt"; then
			warn "failed to umount $tmpmnt"
		fi
	fi
	if [ -e "$dm_device" ]
	then
		debug "remove device mapper device $dm_device"
		dmsetup remove $dm_device
	fi
	rmdir "$tmpmnt" || true
}

if [ "x$1" = xbtrfs ]; then
	types=btrfs
	if [ -z "$2" -o -z "$3" ]; then
		debug "missing btrfs parameters, exiting"
		exit 1
	fi
	UUID="$2"
	BTRFSDEV="$3"
else
	partition="$1"
	types="$(fs_type "$partition")" || types=NOT-DETECTED
fi

if [ "$types" = NOT-DETECTED ]; then
	debug "$1 type not recognised; skipping"
	exit 1
elif [ "$types" = swap ]; then
	debug "$1 is a swap partition; skipping"
	exit 1
elif [ "$types" = crypto_LUKS ]; then
	debug "$1 is a LUKS partition; skipping"
	exit 1
elif [ "$types" = LVM2_member ]; then
	debug "$1 is an LVM member; skipping"
	exit 1
elif [ "$types" = ntfs ]; then
	if type ntfs-3g >/dev/null 2>&1; then
		types='ntfs-3g ntfs'
	fi
elif [ -z "$types" ]; then
	if type cryptsetup >/dev/null 2>&1 && \
	   cryptsetup luksDump "$partition" >/dev/null 2>&1; then
		debug "$1 is a LUKS partition; skipping"
		exit 1
	fi
	for type in $(grep -v nodev /proc/filesystems); do
		# hfsplus filesystems are mountable as hfs. Try hfs last so
		# that we can tell the difference.
		if [ "$type" = hfs ]; then
			delaytypes="${delaytypes:+$delaytypes }$type"
		elif [ "$type" = fuseblk ]; then
			if type ntfs-3g >/dev/null 2>&1; then
				types="${types:+$types }ntfs-3g"
			fi
		else
			types="${types:+$types }$type"
		fi
	done
fi

tmpmnt=/var/lib/os-prober/mount
if [ ! -d "$tmpmnt" ]; then
	mkdir "$tmpmnt"
fi

mounted=

# all btrfs processing here.  Handle both unmounted and
# mounted subvolumes.
if [ "$types" = btrfs ]; then
	partition="$BTRFSDEV"
	debug "begin btrfs processing for $UUID"
	# note that the btrfs volume must not be mounted ro
	if mount -t btrfs -U "$UUID" "$tmpmnt"  2>/dev/null; then
		debug "btrfs volume $UUID mounted"
	else
		warn "cannot mount btrfs volume $UUID, exiting"
		rmdir "$tmpmnt" || true
		exit 1
	fi
	# besides regular subvols, get ro and snapshot so thet can be excluded
        subvols=$(btrfs subvolume list "$tmpmnt" | cut -d ' ' -f 9)
        rosubvols=$(btrfs subvolume list -r "$tmpmnt" | cut -d ' ' -f 9)
        sssubvols=$(btrfs subvolume list -s "$tmpmnt" | cut -d ' ' -f 14)
        if ! umount "$tmpmnt"; then
            warn "failed to umount btrfs volume on $tmpmnt"
            rmdir "$tmpmnt" || true
            exit 1
        fi

	found=
	mounted=

	mpoint="$(grep btrfs /proc/self/mountinfo | grep "$partition " | cut -d ' ' -f 5)"
	if [ -n "$mpoint" -a "x$mpoint" = "x/" ]; then
		debug "This is the root for the running system" #running system must be done elsewhere
	else
	    #partition was not root of running system, so lets look for bootable subvols
	    if [ -n "$mpoint" ] ; then
		mounted=1  #partition was already mounted,so lets not unmount it when done
	    else
		# again, do not mount btrfs ro
		mount -t btrfs -U "$UUID" "$tmpmnt"
		mpoint="$tmpmnt"
	    fi

	    test="/usr/libexec/os-probes/mounted/90linux-distro"
	    if [ -f "$test" ] && [ -x "$test" ]; then
		debug "running subtest $test"
		if "$test" "$partition" "$mpoint" btrfs "UUID=$UUID"; then
		    debug "os found by subtest $test on $partition"
		    found=1
		fi
	    fi
	    if [ -z "$mounted" ]; then
		if ! umount "$tmpmnt"; then
		    warn "failed to umount $tmpmnt"
		fi
	    fi
	fi

	if [ -z "$subvols" ]; then
	        debug "no subvols found on btrfs volume $UUID"
	else
		found=
                for subvol in $subvols; do
			debug "begin btrfs processing for $UUID subvol=$subvol"
			if echo "$rosubvols" | grep -q -x "$subvol"; then
				continue
			fi
			if echo "$sssubvols" | grep -q -x "$subvol"; then
				continue
			fi
			mounted=
			mpoint="$(grep btrfs /proc/self/mountinfo | grep "$partition " | grep "/$subvol " | cut -d ' ' -f 5)"
			if [ -n "$mpoint" ]; then
				if [ "x$mpoint" = "x/" ]; then
					continue # this is the root for the running system
				fi
				mounted=1
			else
				# again, do not mount btrfs ro
				mount -t btrfs -o subvol="$subvol" -U "$UUID" "$tmpmnt"
				mpoint="$tmpmnt"
			fi
			test="/usr/libexec/os-probes/mounted/90linux-distro"
			if [ -f "$test" ] && [ -x "$test" ]; then
				debug "running subtest $test"
				if "$test" "$partition" "$mpoint" btrfs "UUID=$UUID" "subvol=$subvol"; then
					debug "os found by subtest $test on subvol $subvol"
					found=1
				fi
			fi
			if [ -z "$mounted" ]; then
				if ! umount "$tmpmnt"; then
				    warn "failed to umount $tmpmnt"
				fi
			fi
		done
	fi
	rmdir "$tmpmnt" || true
	if [ "$found" ]; then
		exit 0
	else
		exit 1
	fi
fi

if type grub-mount >/dev/null 2>&1 && \
   type grub2-probe >/dev/null 2>&1 && \
   grub-mount "$partition" "$tmpmnt" 2>/dev/null; then
	mounted=1
	type="$(grub2-probe -d "$partition" -t fs)" || true
	if [ "$type" ]; then
		debug "mounted using GRUB $type filesystem driver"
	else
		debug "mounted using GRUB, but unknown filesystem?"
		type=fuseblk
	fi
elif dm_device="$(do_dmsetup osprober "$partition")" && \
     [ "$dm_device" ]; then
	for type in $types $delaytypes; do
		if mountinfo=`mount -o ro -t "$type" "$dm_device" "$tmpmnt" 2>&1`; then
			debug "mounted as $type filesystem"
			mounted=1
			break
		else
			debug "mounting $dm_device ($partition) as $type failed: $mountinfo"
		fi
	done
fi

if [ "$mounted" ]; then
	for test in /usr/libexec/os-probes/mounted/*; do
		debug "running subtest $test"
		if [ -f "$test" ] && [ -x "$test" ]; then
			if "$test" "$partition" "$tmpmnt" "$type"; then
				debug "os found by subtest $test"
				do_unmount
				exit 0
			fi
		fi
	done
fi
do_unmount

# No tests found anything.
exit 1