#!/bin/sh
#
# Copyright (c) 2006 Kungliga Tekniska Högskolan
# (Royal Institute of Technology, Stockholm, Sweden). 
# All rights reserved. 
#
# Redistribution and use in source and binary forms, with or without 
# modification, are permitted provided that the following conditions 
# are met: 
#
# 1. Redistributions of source code must retain the above copyright 
#    notice, this list of conditions and the following disclaimer. 
#
# 2. Redistributions in binary form must reproduce the above copyright 
#    notice, this list of conditions and the following disclaimer in the 
#    documentation and/or other materials provided with the distribution. 
#
# 3. Neither the name of the Institute nor the names of its contributors 
#    may be used to endorse or promote products derived from this software 
#    without specific prior written permission. 
#
# THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 
# ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 
# SUCH DAMAGE. 
#
# Id
#

env_setup="@env_setup@"
srcdir="@srcdir@"
objdir="@objdir@"

. ${env_setup}

# If there is no useful db support compiled in, disable test
../db/have-db || exit 77

R=TEST.H5L.SE

port=@port@

keytabfile=${objdir}/server.keytab
keytab="FILE:${keytabfile}"
cache="FILE:krb5ccfile"
cacheds="FILE:krb5ccfile-ds"

kinit="${TESTS_ENVIRONMENT} ../../kuser/kinit -c $cache ${afs_no_afslog} --forwardable"
kinitds="${TESTS_ENVIRONMENT} ../../kuser/kinit -c $cacheds ${afs_no_afslog}"
kadmin="${TESTS_ENVIRONMENT} ../../kadmin/kadmin -l -r $R"
kdc="${TESTS_ENVIRONMENT} ../../kdc/kdc --addresses=localhost -P $port"

context="${TESTS_ENVIRONMENT} ../../lib/gssapi/test_context"

KRB5_CONFIG="${objdir}/krb5.conf"
export KRB5_CONFIG

KRB5_KTNAME="${keytab}"
export KRB5_KTNAME
KRB5CCNAME="${cache}"
export KRB5CCNAME
NTLM_ACCEPTOR_CCACHE="${cacheds}"
export NTLM_ACCEPTOR_CCACHE
NTLM_USER_FILE="${srcdir}/ntlm-user-file.txt"
export NTLM_USER_FILE

GSSAPI_SPNEGO_NAME=host@host.test.h5l.se
export GSSAPI_SPNEGO_NAME

rm -f ${keytabfile}
rm -f current-db*
rm -f out-*
rm -f mkey.file*

> messages.log

echo Creating database
${kadmin} \
    init \
    --realm-max-ticket-life=1day \
    --realm-max-renewable-life=1month \
    ${R} || exit 1

${kadmin} add -p p1 --use-defaults host/host.test.h5l.se@${R} || exit 1
${kadmin} ext -k ${keytab} host/host.test.h5l.se@${R} || exit 1

${kadmin} add -p kaka --use-defaults digest/${R}@${R} || exit 1

${kadmin} add -p ds --use-defaults digestserver@${R} || exit 1
${kadmin} modify --attributes=+allow-digest digestserver@${R} || exit 1

${kadmin} add -p u1 --use-defaults user1@${R} || exit 1

echo "Doing database check"
${kadmin} check ${R} || exit 1

echo u1 > ${objdir}/foopassword
echo ds > ${objdir}/barpassword

echo Starting kdc
${kdc} --detach --testing || { echo "kdc failed to start"; exit 1; }
kdcpid=`getpid kdc`

trap "kill ${kdcpid}; echo signal killing kdc; exit 1;" EXIT

exitcode=0

echo "Getting client initial tickets"
${kinit} --password-file=${objdir}/foopassword user1@${R} || exitcode=1
echo "Getting digestserver initial tickets"
${kinitds} --password-file=${objdir}/barpassword digestserver@${R} || exitcode=1

echo "======context building for each mech"

for mech in ntlm krb5 ; do 
    echo "${mech}"
    ${context} --mech-type=${mech} --ret-mech-type=${mech} \
        --name-type=hostbased-service host@host.test.h5l.se || \
	{ exitcode=1 ; echo test failed; }
done

echo "spnego"
${context} \
    --mech-type=spnego \
    --ret-mech-type=krb5 \
    --name-type=hostbased-service \
    host@host.test.h5l.se || \
    { exitcode=1 ; echo test failed; }

echo "test failure cases"
${context} --mech-type=ntlm --ret-mech-type=krb5 \
    --name-type=hostbased-service host@host.test.h5l.se 2> /dev/null && \
    { exitcode=1 ; echo test failed; }

${context} --mech-type=krb5 --ret-mech-type=ntlm \
    --name-type=hostbased-service host@host.test.h5l.se 2> /dev/null && \
    { exitcode=1 ; echo test failed; }

echo "======spnego variants context building"

for arg in \
     "" \
     "--mutual" \
     "--delegate" \
     "--mutual --delegate" \
     "--getverifymic --wrapunwrap" \
     "--mutual --getverifymic --wrapunwrap" \
    ; do

    echo "no NTLM acceptor cred ${arg}"
    NTLM_ACCEPTOR_CCACHE="${cacheds}-no"
    ${context} --mech-type=spnego \
        $arg \
        --name-type=hostbased-service \
        --ret-mech-type=krb5  \
        host@host.test.h5l.se || \
        { exitcode=1 ; echo test failed; }
    NTLM_ACCEPTOR_CCACHE="${cacheds}"

    echo "no NTLM initiator cred ${arg}"
    NTLM_USER_FILE="${srcdir}/ntlm-user-file.txt-no"
    ${context} --mech-type=spnego \
        $arg \
        --name-type=hostbased-service \
        --ret-mech-type=krb5 \
        host@host.test.h5l.se || \
        { exitcode=1 ; echo test failed; }
    NTLM_USER_FILE="${srcdir}/ntlm-user-file.txt"

    echo "no krb5 acceptor cred ${arg}"
    KRB5_KTNAME="${keytab}-no"
    ${context} --mech-type=spnego \
        $arg \
	--server-no-delegate \
        --name-type=hostbased-service \
        --ret-mech-type=ntlm \
        host@host.test.h5l.se || \
        { exitcode=1 ; echo test failed; }
    KRB5_KTNAME="${keytab}"

    echo "no krb5 initiator cred ${arg}"
    KRB5CCNAME="${cache}-no"
    ${context} --mech-type=spnego \
        $arg \
	--server-no-delegate \
        --name-type=hostbased-service \
        --ret-mech-type=ntlm \
        host@host.test.h5l.se || \
        { exitcode=1 ; echo test failed; }
    KRB5CCNAME="${cache}"

done

trap "" EXIT

echo "killing kdc (${kdcpid})"
kill ${kdcpid} 2> /dev/null

exit $exitcode