#!/bin/bash

ipstr="$1"

try() {
	cmd="$1"
	if [ $# -lt 2 ]; then
		variant=0
	else
		variant=$2
	fi
	echo "Executing $cmd with variant $variant.."
	output=$(enochecker_cli -A "http://localhost:8000/" -a "$ipstr" -v "$variant" -x 4000 "$cmd")
	echo $output
	[ -z "$(echo $output | grep OK)" ] && exit 1
}


if [ $# -lt 1 ]; then
	echo "USAGE: test.sh <IP>"
elif [ $# -eq 3 ]; then
	try "$2" "$3"
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