#!/bin/sh 
CLAMD_WRAPPER=${CLAMD_WRAPPER-}
CLAMD_TEST_UNIQ1=${CLAMD_TEST_UNIQ1-1}
CLAMD_TEST_UNIQ2=${CLAMD_TEST_UNIQ2-2}
TOP="../.."
LTEXEC="$TOP/libtool --mode=execute"
killclamd() {
	test -f clamd-test.pid || return
	pid=`cat clamd-test.pid 2>/dev/null`
	if test "X$pid" = "X"; then
		# file can be removed between the 'test' and 'cat',
		# it happened a few times for me
		return
	fi
	kill -0 $pid 2>/dev/null || return
	kill $pid
	pippo=0
	while kill -0 $pid 2>/dev/null; do
		sleep 1
		pippo=`expr $pippo + 1`
		if test $pippo -gt 9; then
			kill -KILL $pid
			echo "Clamd didn't quit";
			rm -f clamd-test.pid
			exit 4;
		fi
	done
	rm -f clamd-test.pid
}

die()
{
	killclamd
	exit $1
}

error()
{
	echo >&2
	echo "***" >&2
	echo "*** $1" >&2
	echo "***" >&2
}

scan_failed() {
	if test "X$unrar_disabled" = "X1" && test `grep -v '\.rar' $1 | grep OK | wc -l` -eq 0
	then
		error "UNRAR is disabled, won't be able to detect unrar files!";
	else
		error  $2;
		die 2;
	fi
}

start_clamd()
{
	rm -f clamd-test.log ../clamd-test1.log ../clamd-test2.log
	$LTEXEC $CLAMD_WRAPPER $TOP/clamd/clamd -c $1 --help >clamd-test.log 2>&1 || 
		{ error "Failed to start clamd --help!"; cat clamd-test.log; die 1; }
	grep "Clam AntiVirus Daemon" clamd-test.log >/dev/null ||
		{ error "Wrong --help reply from clamd!"; die 1; }
	$LTEXEC $CLAMD_WRAPPER $TOP/clamd/clamd -c $1 >clamd-test.log 2>&1 || 
		{ error "Failed to start clamd!"; cat clamd-test.log; die 1; }
}

run_clamdscan_fileonly() {
	rm -f clamdscan.log clamdscan-multiscan.log
	$TOP/clamdscan/clamdscan --version --config-file=test-clamd.conf 2>&1|grep "^ClamAV" >/dev/null || 
		{ error "clamdscan can't get version of clamd!"; die 1;}
	$TOP/clamdscan/clamdscan --quiet --config-file=test-clamd.conf $* --log=clamdscan.log
	if test $? = 2; then 
		error "Failed to run clamdscan!"
		cat clamdscan.log
		die 1
	fi
	$TOP/clamdscan/clamdscan --quiet --config-file=test-clamd.conf $* -m --log=clamdscan-multiscan.log
	if test $? = 2; then 
		error "Failed to run clamdscan (multiscan)!"
		cat clamdscan-multiscan.log
		die 1
	fi
}

run_clamdscan() {
	run_clamdscan_fileonly $*
	rm -f clamdscan-fdpass.log clamdscan-multiscan-fdpass.log
	$TOP/clamdscan/clamdscan --quiet --config-file=test-clamd.conf $* --fdpass --log=clamdscan-fdpass.log
	if test $? = 2; then 
		error "Failed to run clamdscan (fdpass)!"
		cat clamdscan-fdpass.log
		die 1
	fi
	$TOP/clamdscan/clamdscan --quiet --config-file=test-clamd.conf $* -m --fdpass --log=clamdscan-multiscan-fdpass.log
	if test $? = 2; then 
		error "Failed to run clamdscan (fdpass + multiscan)!"
		cat clamdscan-multiscan-fdpass.log
		die 1
	fi
	$TOP/clamdscan/clamdscan --quiet --config-file=test-clamd.conf $* --stream --log=clamdscan-stream.log
	if test $? = 2; then 
		error "Failed to run clamdscan (instream)!"
		cat clamdscan-stream.log
		die 1
	fi
	$TOP/clamdscan/clamdscan --quiet --config-file=test-clamd.conf $* -m --stream --log=clamdscan-multiscan-stream.log
	if test $? = 2; then 
		error "Failed to run clamdscan (instream + multiscan)!"
		cat clamdscan-multiscan-stream.log
		die 1
	fi
}

run_reload_test()
{
	rm -f reload-testfile
	echo "ClamAV-RELOAD-Test" >reload-testfile
	run_clamdscan reload-testfile
	grep "ClamAV-RELOAD-TestFile" clamdscan.log >/dev/null 2>/dev/null;
	if test $? -eq 0; then
		# it is not supposed to detect until we actually put the
		# signature there and reload!
		error "RELOAD test failed!"
		cat clamdscan.log
		die 10
	fi
	echo "ClamAV-RELOAD-TestFile:0:0:436c616d41562d52454c4f41442d54657374" >test-db/new.ndb
	$TOP/clamdscan/clamdscan --reload --config-file=test-clamd.conf
	if test $? -ne 0; then
		error "clamdscan says reload failed!"
		die 11
	fi
	run_clamdscan reload-testfile
	grep "ClamAV-RELOAD-TestFile" clamdscan.log >/dev/null 2>/dev/null;
	failed=0
	if test $? -ne 0; then
		error "RELOAD test failed! (after reload)"
		cat clamdscan.log
		failed=1
	fi
	grep "ClamAV-RELOAD-TestFile" clamdscan-multiscan.log >/dev/null 2>/dev/null;
	if test $? -ne 0; then
		error "RELOAD test failed! (after reload, multiscan)"
		cat clamdscan-multiscan.log
		failed=1
	fi
	if test "$failed" = "1"; then
		echo "RELOAD tests failed!"
		die 12
	fi
	rm -f reload-testfile
}

run_clamdscan_fdpass() {
	rm -f clamdscan.log
	$TOP/clamdscan/clamdscan --quiet --fdpass --config-file=test-clamd.conf - <$1 --log=clamdscan.log
	if test $? = 2; then
		error "Failed to run clamdscan (fdpass)!"
		cat clamdscan.log
		die 14
	fi
}

# We run multiple clamd tests in parallel, each in its own directory
prepare_clamd()
{
	cd clamdtest$1 
	# Set up test DBdir
	rm -rf test-db
	mkdir -p test-db
	cat <<EOF >test-db/test.hdb
aa15bcf478d165efd2065190eb473bcb:544:ClamAV-Test-File
EOF
	cp $abs_srcdir/input/daily.pdb test-db/
	$AWK "{ sub(/X/,\"$1\"); sub(/CWD/,\"`pwd`\"); print }" $abs_srcdir/test-clamd.conf >test-clamd.conf
}

rm -rf clamdtest$CLAMD_TEST_UNIQ1 clamdtest$CLAMD_TEST_UNIQ2
mkdir clamdtest$CLAMD_TEST_UNIQ1 clamdtest$CLAMD_TEST_UNIQ2 || 
	{ echo "Unable to create temporary directories!"; exit 1; }

# Prepare configuration for clamd #1 and #2
(prepare_clamd $CLAMD_TEST_UNIQ1)
(prepare_clamd $CLAMD_TEST_UNIQ2)
# Add clamd #2 specific configuration
echo "VirusEvent $abs_srcdir/virusaction-test.sh `pwd`/clamdtest$CLAMD_TEST_UNIQ2 \"Virus found: %v\"" >>clamdtest$CLAMD_TEST_UNIQ2/test-clamd.conf
echo "HeuristicScanPrecedence yes" >>clamdtest$CLAMD_TEST_UNIQ2/test-clamd.conf
grep -v LogFile clamdtest$CLAMD_TEST_UNIQ2/test-clamd.conf >tmp__
mv tmp__ clamdtest$CLAMD_TEST_UNIQ2/test-clamd.conf

# Start clamd #1 tests
(cd clamdtest$CLAMD_TEST_UNIQ1 
start_clamd test-clamd.conf

# Test that all testfiles are detected
FILES=$TOP/test/clam*
run_clamdscan $FILES
NFILES=`ls -1 $FILES | wc -l`
NINFECTED=`grep "Infected files" clamdscan.log | cut -f2 -d:|sed -e 's/ //g'`
NINFECTED_MULTI=`grep "Infected files" clamdscan-multiscan.log | cut -f2 -d:|sed -e 's/ //g'`
NINFECTED_FDPASS=`grep "Infected files" clamdscan-fdpass.log | cut -f2 -d:|sed -e 's/ //g'`
NINFECTED_MULTI_FDPASS=`grep "Infected files" clamdscan-multiscan-fdpass.log | cut -f2 -d:|sed -e 's/ //g'`
NINFECTED_STREAM=`grep "Infected files" clamdscan-stream.log | cut -f2 -d:|sed -e 's/ //g'`
NINFECTED_MULTI_STREAM=`grep "Infected files" clamdscan-multiscan-stream.log | cut -f2 -d:|sed -e 's/ //g'`
if test "$NFILES" -ne "0$NINFECTED"; then
	grep OK clamdscan.log
	scan_failed clamdscan.log "clamd did not detect all testfiles correctly!"
fi
if test "$NFILES" -ne "0$NINFECTED_MULTI"; then
	grep OK clamdscan-multiscan.log
	scan_failed clamdscan-multiscan.log "clamd did not detect all testfiles correctly in multiscan mode!"
fi
if test "$NFILES" -ne "0$NINFECTED_FDPASS"; then
	grep OK clamdscan-fdpass.log
	scan_failed clamdscan-multiscan.log "clamd did not detect all testfiles correctly in fdpass mode!"
fi
if test "$NFILES" -ne "0$NINFECTED_MULTI_FDPASS"; then
	grep OK clamdscan-multiscan-fdpass.log
	scan_failed clamdscan-multiscan.log "clamd did not detect all testfiles correctly in fdpass+multiscan mode!"
fi

$TOP/unit_tests/check_clamd
ecode=$?
if test $ecode -ne 77 && test $ecode -ne 0; then
    error "Failed clamd protocol test!"
    die 1
fi
# Test HeuristicScanPrecedence off feature
run_clamdscan ../clam-phish-exe
grep "ClamAV-Test-File" clamdscan.log >/dev/null 2>/dev/null;
if test $? -ne 0; then
	error "HeuristicScanPrecedence off test failed!"
	cat clamdscan.log
	die 4
fi
die 0
)&
pid1=$!

# Start clamd #2 tests
(cd clamdtest$CLAMD_TEST_UNIQ2
start_clamd test-clamd.conf

# Test VirusEvent feature
run_clamdscan_fileonly $TOP/test/clam.exe
grep "Virus found: ClamAV-Test-File.UNOFFICIAL" test-clamd.log >/dev/null 2>/dev/null; 
if test $? -ne 0; then
	error "Virusaction test failed!" 
	cat test-clamd.log
	die 2
fi

# Test HeuristicScanPrecedence feature
run_clamdscan ../clam-phish-exe
grep "Phishing.Heuristics.Email.SpoofedDomain" clamdscan.log >/dev/null 2>/dev/null;
if test $? -ne 0; then
	error "HeuristicScanPrecedence on test failed!"
	cat clamdscan.log
	die 3
fi

if grep "^#define HAVE_FD_PASSING 1" $TOP/clamav-config.h >/dev/null; then
	run_clamdscan_fdpass $TOP/test/clam.exe
	grep "ClamAV-Test-File" clamdscan.log >/dev/null 2>/dev/null;
	if test $? -ne 0; then
		error "FDpassing test failed!"
		cat clamdscan.log;
		die 4
	fi
else
	echo "*** No file descriptor passing support, skipping test"
fi

# Test RELOAD command
run_reload_test

die 0
)&

pid2=$!

wait $pid1
exitcode1=$?
wait $pid2
exitcode2=$?
rm -rf clamdtest$CLAMD_TEST_UNIQ1 clamdtest$CLAMD_TEST_UNIQ2 test-db accdenied
if (test $exitcode1 -ne 0 && test $exitcode1 -ne 127) || (test $exitcode2 -ne 0	&& test $exitcode2 -ne 127); then
	exit 1
fi
exit 0