check.sh

#!/bin/sh

if test -t 1
then
    colors=$(tput colors)
    if test -n "$colors" && test $colors -ge 8
    then
        reset="$(tput sgr0)"
        red="$(tput setaf 1)"
        green="$(tput setaf 2)"
    fi
fi

lacc="$1"
file="$2"
comp="$3"
bdir="$4"
if [ -z "$file" ] || [ ! -f "$file" ]; then
	echo "Usage: $0 <compiler> <file> [<reference compiler>]";
	exit 1
fi

if [ -z "$comp" ]; then
	comp="gcc -std=c89 -Wno-psabi"
fi

if [ -z "$bdir" ]; then
	bdir="."
fi

$comp -v 2>&1 >/dev/null | grep "enable-default-pie" > /dev/null
if [ "$?" -eq "0" ]; then
	lacc="${lacc} -fPIC"
fi

i=`dirname ${file}`
d=`dirname ${bdir}/${file}`
f=`basename ${file} .c`
mkdir -p ${d}
$comp $file -o ${d}/${f}.out
if [ "$?" -ne "0" ]; then
	echo "${file}: ${red}Invalid input file!${reset}";
	exit 1
fi

${d}/${f}.out > ${d}/${f}.ans.txt; answer="$?"

check()
{
	$lacc $1 $file -o ${d}/${f}.s
	if [ "$?" -ne "0" ]; then
		echo "${red}Compilation failed!${reset}";
		return 1
	fi
	if [ "$1" = "-E" ]; then
		mv ${d}/${f}.s ${d}/${f}.i
		$lacc -c ${d}/${f}.i -o ${d}/${f}.o
		if [ "$?" -ne "0" ]; then
			echo "${red}Compilation failed!${reset}";
			return 1
		fi
	elif [ "$1" = "-S" ]; then
		$comp -c ${d}/${f}.s -o ${d}/${f}.o
		if [ "$?" -ne "0" ]; then
			echo "${red}Assembly failed!${reset}";
			return 1
		fi
	else
		mv ${d}/${f}.s ${d}/${f}.o
	fi
	$comp ${d}/${f}.o -o ${d}/${f}.out -lm
	if [ "$?" -ne "0" ]; then
		echo "${red}Linking failed!${reset}";
		return 1
	fi

	${d}/${f}.out > ${d}/${f}.txt
	result="$?"
	difference=`diff ${d}/${f}.ans.txt ${d}/${f}.txt`
	diffres="$?"

	if [ "$result" -eq "$answer" ] && [ "$diffres" -eq "0" ]; then
		echo "${green}Ok!${reset}"
		return 0
	else
		echo "${red}Wrong result!${reset}"
		if [ "$result" -ne "$answer" ]; then
			echo "Result differ: was ${result}, expected ${answer}." >&2
		fi
		if [ "$diffres" -ne "0" ]; then
			echo "Output differ:" >&2
			echo "$difference" >&2
		fi
		return 1
	fi
}

prp=$(check -E); result="$?"; retval=$((retval + result))
asm=$(check -S); result="$?"; retval=$((retval + result))
elf=$(check -c); result="$?"; retval=$((retval + result))
opt=$(check "-c -O1"); result="$?"; retval=$((retval + result))
echo "[-E: ${prp}] [-S: ${asm}] [-c: ${elf}] [-c -O1: ${opt}] :: ${file}"

if [ $retval -eq 0 ] && [ -f "${i}/${f}.sh" ]
then
	./${i}/${f}.sh "$lacc" "${i}/${f}" "${bdir}"
	retval=$?
	if [ $retval -ne 0 ]
	then
		echo "${red}Failed validation script!${reset}"
	fi
fi

rm -f ${d}/${f}.out ${d}/${f}.ans.txt ${d}/${f}.txt ${d}/${f}.s ${d}/${f}.i ${d}/${f}.o
exit $retval