#!/bin/bash -eu

# Run the set of "canonical" benchmarks that are shown on
# https://nuetzlich.net/gocryptfs/comparison/

cd "$(dirname "$0")"
MYNAME=$(basename "$0")
source tests/fuse-unmount.bash

function usage {
	echo "Usage: $MYNAME [-encfs] [-openssl=true] [-openssl=false] [-dd] [DIR]"
}

OPT_ENCFS=0
OPT_LOOPBACK=0
OPT_OPENSSL=""
OPT_DIR=""
DD_ONLY=""

while [[ $# -gt 0 ]] ; do
	case $1 in
		-h)
			usage
			exit 1
			;;
		-encfs)
			OPT_ENCFS=1
			;;
		-openssl=true)
			OPT_OPENSSL="-openssl=true"
			;;
		-openssl=false)
			OPT_OPENSSL="-openssl=false"
			;;
		-dd)
			DD_ONLY=1
			;;
		-loopback)
			OPT_LOOPBACK=1
			;;
		-*)
			echo "Invalid option: $1"
			usage
			exit 2
			;;
		*)
			if [[ ! -z $OPT_DIR ]] ; then
				echo "Duplicate DIR argument: $1"
				usage
				exit 3
			fi
			OPT_DIR=$1
			;;
	esac
	shift
done

if [[ -z $OPT_DIR ]] ; then
	OPT_DIR=/tmp
fi

# Create directories
CRYPT=$(mktemp -d "$OPT_DIR/$MYNAME.XXX")
MNT=$CRYPT.mnt
mkdir $MNT

# Mount
if [[ $OPT_ENCFS -eq 1 ]]; then
	if [[ ! -z $OPT_OPENSSL ]] ; then
		echo "The option $OPT_OPENSSL only works with gocryptfs"
		exit 1
	fi
	echo -n "Testing EncFS at $CRYPT: "
	encfs --version
	encfs --extpass="echo test" --standard $CRYPT $MNT > /dev/null
elif [[ $OPT_LOOPBACK -eq 1 ]]; then
	echo "Testing go-fuse loopback"
	$HOME/go/src/github.com/hanwen/go-fuse/example/loopback/loopback $MNT $CRYPT &
	sleep 0.5
else
	echo -n "Testing gocryptfs at $CRYPT: "
	gocryptfs -version
	gocryptfs -q -init -extpass="echo test" -scryptn=10 $CRYPT
	gocryptfs -q -extpass="echo test" $OPT_OPENSSL $CRYPT $MNT
fi

# Make sure we have actually mounted something
if ! mountpoint $MNT ; then
	exit 1
fi

# Cleanup trap
trap "cd /; fuse-unmount -z $MNT; rm -rf $CRYPT $MNT" EXIT

# Benchmarks
if [[ $DD_ONLY -eq 1 ]]; then
	echo -n "WRITE: "
	dd if=/dev/zero of=$MNT/zero bs=131072 count=20000 2>&1 | tail -n 1
	rm $MNT/zero
else
	./tests/canonical-benchmarks.bash $MNT
fi