#!/bin/sh
#
# Copyright (C) 2015 Red Hat
#
# This file is part of ocserv.
#
# ocserv 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 2 of the License, or (at
# your option) any later version.
#
# ocserv 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.

SERV="${SERV:-../src/ocserv}"
srcdir=${srcdir:-.}

. `dirname $0`/common.sh

eval "${GETPORT}"

echo "Testing whether cookies will expire as expected... "

PIDFILE="${srcdir}/ct$$.pid.tmp"

update_config test-cookie-timeout.config
launch_server -d 1 -f -c ${CONFIG} & PID=$!
wait_server $PID

echo "Connecting to obtain cookie... "
eval `echo "test" | $OPENCONNECT -q localhost:$PORT -u test --authenticate --servercert=pin-sha256:xp3scfzy3rOQsv9NcOve/8YVVv+pHr4qNCXEXrNl5s8=`

if [ -z "$COOKIE" ];then
	fail $PID "Could not obtain cookie"
fi

#echo "Cookie: $COOKIE"
sleep 16
echo ""
echo "Connecting with cookie... "
echo "test" | $OPENCONNECT -q -b localhost:$PORT -u test -C "$COOKIE" --servercert=pin-sha256:xp3scfzy3rOQsv9NcOve/8YVVv+pHr4qNCXEXrNl5s8= --script=/bin/true --verbose --pid-file "${PIDFILE}" --background

sleep 4

if [ ! -f "${PIDFILE}" ];then
	fail $PID "It was not possible to establish session!"
fi

CPID=`cat "${PIDFILE}"`
kill -9 $CPID
rm -f "${PIDFILE}"

sleep 16
echo ""
echo "Connecting again with cookie... "
echo "test" | $OPENCONNECT -b -q localhost:$PORT -u test -C "$COOKIE" --servercert=pin-sha256:xp3scfzy3rOQsv9NcOve/8YVVv+pHr4qNCXEXrNl5s8= --script=/bin/true --verbose --pid-file "${PIDFILE}" --background

sleep 4

if [ ! -f "${PIDFILE}" ];then
	fail $PID "It was not possible to establish second session!"
fi

CPID=`cat "${PIDFILE}"`
kill -9 $CPID
rm -f "${PIDFILE}"

sleep 16
echo ""
echo "Connecting after forced kill with cookie... "
echo "test" | $OPENCONNECT -b -q localhost:$PORT -u test -C "$COOKIE" --servercert=pin-sha256:xp3scfzy3rOQsv9NcOve/8YVVv+pHr4qNCXEXrNl5s8= --script=/bin/true --verbose --pid-file "${PIDFILE}" --background

sleep 4

if [ ! -f "${PIDFILE}" ];then
	fail $PID "It was not possible to establish third session!"
fi

CPID=`cat "${PIDFILE}"`
kill -9 $CPID
rm -f "${PIDFILE}"


sleep 45
echo ""
echo "Connecting with cookie after expiration... "
echo "test" | $OPENCONNECT -q -b localhost:$PORT -u test -C "$COOKIE" --servercert=pin-sha256:xp3scfzy3rOQsv9NcOve/8YVVv+pHr4qNCXEXrNl5s8= --script=/bin/true --verbose --pid-file "${PIDFILE}" --background

sleep 4

if [ -f "${PIDFILE}" ];then
	fail $PID "Session was established with expired cookie!"
	CPID=`cat "${PIDFILE}"`
	kill $CPID
	rm -f "${PIDFILE}"
fi

# test cookie verification after cookie verification failure. That is to verify whether
# the channel between main and sec-mod is in consistent state.
echo "Connecting (again) to obtain cookie... "
echo "test" | $OPENCONNECT -q localhost:$PORT -u test --authenticate --servercert=pin-sha256:xp3scfzy3rOQsv9NcOve/8YVVv+pHr4qNCXEXrNl5s8=

if test $? != 0;then
	fail $PID "Could not obtain cookie"
fi

kill $PID
wait

rm -f "${PIDFILE}" ${CONFIG}

exit 0