#!/bin/sh

# Copyright (C) 2014 Nikos Mavrogiannopoulos
#
# This file is part of GnuTLS.
#
# GnuTLS is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 3 of the License, or (at
# your option) any later version.
#
# GnuTLS is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GnuTLS; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

srcdir="${srcdir:-.}"

if ! test -x "/usr/bin/dieharder"; then
	exit 77
fi

VERSION=`dieharder -l|grep version|cut -d ' ' -f 6`

if test "$1" = "full"; then
	OPTIONS="-a"
else
	if test "${VERSION}" = "2.28.1"; then
		OPTIONS="-d 5"
		OPTIONS2="-d 10"
	else
		OPTIONS="-d 202"
		OPTIONS2="-d 10"
	fi
fi

OUTFILE=rng.log
RNGFILE=rng.out
RNGFILE2=rng2.out

rm -f "${OUTFILE}"
rm -f "${RNGFILE}"
rm -f "${RNGFILE2}"

. "${srcdir}/../scripts/common.sh"


RINPUTNO=`dieharder -g -1|grep file_input_raw|cut -d '|' -f 2|cut -d ' ' -f 1`

if test -z "${RINPUTNO}"; then
	echo "Cannot determine dieharder option for raw file input, assuming 201"
	RINPUTNO=201
fi

echo ""
echo "Testing nonce PRNG"

./rng nonce 64 "${RNGFILE}"
./rng nonce 64 "${RNGFILE2}"
cmp "${RNGFILE}" "${RNGFILE2}"  >/dev/null 2>&1
ret=$?

if test ${ret} = 0; then
	echo "numbers are repeated in nonce!"
	exit 1
fi

./rng nonce 100000000 "${RNGFILE}"

dieharder -f "${RNGFILE}" -g ${RINPUTNO} ${OPTIONS} >"${OUTFILE}" 2>&1
if ! test -z "${OPTIONS2}"; then
	dieharder -f "${RNGFILE}" -g ${RINPUTNO} ${OPTIONS2} >>"${OUTFILE}" 2>&1
fi
grep FAILED "${OUTFILE}" >/dev/null 2>&1
ret=$?

if test "${ret}" = "0"; then
	echo "test failed for nonce"
	exit 1
fi

grep PASSED "${OUTFILE}" >/dev/null 2>&1
ret=$?

if test "${ret}" != "0"; then
	echo "could not run dieharder test?"
	exit 1
fi

cat "${OUTFILE}"
rm -f "${OUTFILE}"
echo ""
echo "Testing key PRNG"

./rng key 64 "${RNGFILE}"
./rng key 64 "${RNGFILE2}"
cmp "${RNGFILE}" "${RNGFILE2}" >/dev/null 2>&1
ret=$?

if test ${ret} = 0; then
	echo "numbers are repeated in nonce!"
	exit 1
fi

./rng key 100000000 "${RNGFILE}"

dieharder -f "${RNGFILE}" -g ${RINPUTNO} ${OPTIONS} >"${OUTFILE}" 2>&1
if ! test -z "${OPTIONS2}"; then
	dieharder -f "${RNGFILE}" -g ${RINPUTNO} ${OPTIONS2} >>"${OUTFILE}" 2>&1
fi
grep FAILED "${OUTFILE}" >/dev/null 2>&1 
ret=$?


if test "${ret}" = "0"; then
	echo "test failed for key"
	exit 1
fi

grep PASSED "${OUTFILE}" >/dev/null 2>&1
ret=$?

if test "${ret}" != "0"; then
	echo "could not run dieharder test?"
	exit 1
fi

cat "${OUTFILE}"
rm -f "${OUTFILE}"
echo ""
echo "Testing /dev/zero PRNG"
dd if=/dev/zero of="${RNGFILE}" bs=4 count=10000000 >/dev/null 2>&1

dieharder -f "${RNGFILE}" -g ${RINPUTNO} ${OPTIONS} >"${OUTFILE}" 2>&1
if ! test -z "${OPTIONS2}"; then
	dieharder -f "${RNGFILE}" -g ${RINPUTNO} ${OPTIONS2} >>"${OUTFILE}" 2>&1
fi
grep PASSED "${OUTFILE}" >/dev/null 2>&1 
ret=$?

if test "${ret}" = "0"; then
	echo "test succeeded for /dev/zero!!!"
	exit 1
fi

grep FAILED "${OUTFILE}" >/dev/null 2>&1
ret=$?

if test "${ret}" != "0"; then
	echo "could not run dieharder test?"
	exit 1
fi

cat "${OUTFILE}"
rm -f "${OUTFILE}"
rm -f "${RNGFILE}"
rm -f "${RNGFILE2}"

exit 0