#!/bin/bash

ipstr="$1"

SCRIPTPATH="$(dirname $(readlink -f "$0"))"
export REVHASH_PATH="$SCRIPTPATH/src/revhash/revhash"

try() {
	cmd="$1"
	if [ $# -lt 2 ]; then
		variant=0
	else
		variant=$2
	fi
	echo "Executing $cmd with variant $variant.."
	python3 src/checker.py -j run -v "$variant" -x 4000 \
		--flag ENOTESTFLAG123= --flag_regex 'ENO.*=' \
		${@:3} "$cmd" | tee /tmp/checker-log
	[ -z "$(cat /tmp/checker-log | grep OK)" ] && exit 1
}


if [ $# -eq 2 ]; then
	try $@
elif [ "$1" == "test-exploits" ]; then
	try exploit 0
	try exploit 1
else
	try putflag 0
	try getflag 0

	try putflag 1
	try getflag 1

	try putnoise 0
	try getnoise 0

	try putflag 1
	try getflag 1

	try havoc 0
	try havoc 1
	try havoc 2
	try havoc 3

	try exploit 0
	try exploit 1
fi

exit 0