#!/bin/sh
#
# Copyright © 2021 Daniel Lenski
#
# This file is part of openconnect.
#
# This is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public License
# as published by the Free Software Foundation; either version 2.1 of
# the License, or (at your option) any later version.
#
# This library 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>

# This test uses LD_PRELOAD
PRELOAD=1
srcdir=${srcdir:-.}
top_builddir=${top_builddir:-..}

. `dirname $0`/common.sh

FINGERPRINT="--servercert=pin-sha256:xp3scfzy3rO"
CERT=$certdir/server-cert.pem
KEY=$certdir/server-key.pem

echo "Testing Fortinet auth against fake server ..."

OCSERV=${srcdir}/fake-fortinet-server.py
launch_simple_sr_server $ADDRESS 443 $CERT $KEY > /dev/null 2>&1
PID=$!
wait_server $PID

SERVURL="https://$ADDRESS:443"
CLIENT="$OPENCONNECT --protocol=fortinet $FINGERPRINT -u test --passwd-on-stdin -q"
export LD_PRELOAD=libsocket_wrapper.so

echo -n "Authenticating with username/password and DEFAULT realm... "
( echo "test" | $CLIENT $SERVURL --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo -n "Authenticating with username/password and NON-DEFAULT path... "
( echo "test" | $CLIENT $SERVURL/fakeRealm --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo "Configuring fake server to require 1 round of tokeninfo-based 2FA."
curl -sk $SERVURL/CONFIGURE -d want_2fa=1 -d type_2fa=tokeninfo

echo -n "Authenticating with username/password/tokeninfo-2FA and DEFAULT path... "
( echo "test" | $CLIENT $SERVURL --token-mode=totp --token-secret=FAKE --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo -n "Authenticating with username/password/tokeninfo-2FA and NON-DEFAULT path... "
( echo "test" | $CLIENT $SERVURL/fake+Realm --token-mode=totp --token-secret=FAKE --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo "Configuring fake server to require 1 rounds of tokeninfo-based 2FA with optional \"FTM push\"."
curl -sk $SERVURL/CONFIGURE -d want_2fa=1 -d type_2fa=ftmpush

echo -n "Authenticating with username/password/FTM-push and DEFAULT path... "
( echo "test" | $CLIENT $SERVURL --form-entry=_challenge:code='' --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo "Configuring fake server to require 1 round of HTML-based 2FA."
curl -sk $SERVURL/CONFIGURE -d want_2fa=1 -d type_2fa=html

echo -n "Authenticating with username/password/HTML-2FA and DEFAULT path... "
( echo "test" | $CLIENT $SERVURL --token-mode=totp --token-secret=FAKE --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo "Configuring fake server to require 2 rounds of tokeninfo-based 2FA."
curl -sk $SERVURL/CONFIGURE -d want_2fa=2 -d type_2fa=tokeninfo

echo -n "Authenticating with username/password/(2 rounds of tokeninfo-2FA) and DEFAULT path... "
( echo "test" | $CLIENT $SERVURL --token-mode=totp --token-secret=FAKE --cookieonly >/dev/null 2>&1) ||
    fail $PID "Could not receive cookie from fake Fortinet server"

echo ok

echo "Resetting fake server to default configuration."
curl -sk $SERVURL/CONFIGURE -d ''

echo -n "Authenticating with username/password, then proceeding to tunnel stage... "
echo "test" | $CLIENT $SERVURL >/dev/null 2>&1
test $? = 2 || # what OpenConnect returns when server rejects cookie upon tunnel connection, as the fake server does
    fail $PID "Something went wrong in fake Fortinet server (other than the expected rejection of cookie)"

echo ok

cleanup

exit 0
