#!/bin/sh
#
# Copyright (c) 2006 - 2008 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@"
confdir="@confdir@"
testdir="@testdir@"

. ${env_setup}

# If there is no useful db support compile in, disable test
${have_db} || exit 77

mkdir -p "${testdir}"
rm -rf "${testdir}/"*

R=TEST.H5L.SE

port=@port@

keytabfile=${testdir}/server.keytab
keytab="FILE:${keytabfile}"
cache="FILE:${testdir}/krb5ccfile"

kinit="${kinit} -c $cache ${afs_no_afslog}"
klist="${klist} -c $cache"
kgetcred="${kgetcred} -c $cache"
kdestroy="${kdestroy} -c $cache"
kadmin="${kadmin} -l -r $R"
kdc="${kdc} --addresses=localhost -P $port"

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

KRB5CCNAME=${cache}
export KRB5CCNAME

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

> ${testdir}/messages.log

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

# add both lucid and lucid.test.h5l.se to simulate aliases
${kadmin} add -p p1 --use-defaults host/lucid.test.h5l.se@${R} || exit 1
${kadmin} ext -k ${keytab} host/lucid.test.h5l.se@${R} || exit 1
${kadmin} add -p notp1 --use-defaults CIFS/lucid.test.h5l.se@${R} || exit 1
${kadmin} ext -k ${keytab} CIFS/lucid.test.h5l.se@${R} || exit 1
${kadmin} add -p p1 --use-defaults host/lucid@${R} || exit 1
${kadmin} ext -k ${keytab} host/lucid@${R} || exit 1

${kadmin} add -p p1 --use-defaults host/ok-delegate.test.h5l.se@${R} || exit 1
${kadmin} mod --attributes=+ok-as-delegate host/ok-delegate.test.h5l.se@${R} || exit 1
${kadmin} ext -k ${keytab} host/ok-delegate.test.h5l.se@${R} || exit 1


${kadmin} add -p p1 --use-defaults host/short@${R} || exit 1
${kadmin} mod --alias=host/long.test.h5l.se@${R} host/short@${R} || exit 1
# XXX ext should ext aliases too
${kadmin} ext -k ${keytab} host/short@${R} || exit 1
${ktutil} -k ${keytab} rename --no-delete host/short@${R} host/long.test.h5l.se@${R} || exit 1

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

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

# Create a server principal with no AES
${kadmin} add -p p1 --use-defaults host/no-aes.test.h5l.se@${R} || exit 1
${kadmin} get host/no-aes.test.h5l.se@${R} > /dev/null || exit 1
${kadmin} del_enctype host/no-aes.test.h5l.se@${R} \
    aes256-cts-hmac-sha1-96 aes128-cts-hmac-sha1-96 || exit 1
${kadmin} ext -k ${keytab} host/no-aes.test.h5l.se@${R} || exit 1

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

echo u1 > ${testdir}/foopassword

echo Starting kdc
env MallocStackLogging=1 MallocStackLoggingNoCompact=1 MallocErrorAbort=1 MallocLogFile=${testdir}/malloc-log \
${kdc} &
kdcpid=$!

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

sh ${wait_kdc} KDC ${testdir}/messages.log || exit 1

testfailed="echo test failed; cat ${testdir}/messages.log; exit 1"

echo "Getting client initial tickets" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword --forwardable user1@${R} || \
	{ eval "$testfailed"; }

echo "======test acceptor name is service only" ; > ${testdir}/messages.log
for mech in krb5 spnego; do

	echo "  testing mech $mech"

	${test_context} \
	    --mech-type=$mech \
	    --mutual-auth \
	    --acceptor-name=host \
	    --name-type=hostbased-service host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }

	echo "  testing mech $mech (check that it failes for wrong service used)"

	${test_context} \
	    --mech-type=$mech \
	    --mutual-auth \
	    --acceptor-name=CIFS \
	    --name-type=hostbased-service host@lucid.test.h5l.se && \
		{ eval "$testfailed"; }
done

echo "======test unreadable/non existant keytab and its error message" ; > ${testdir}/messages.log
${test_context} --mech-type=krb5 host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

mv ${keytabfile} ${keytabfile}.no

echo "checking non existant keytabfile (krb5)" ; > ${testdir}/messages.log
${test_context} --mech-type=krb5 host@lucid.test.h5l.se  > ${testdir}/test_context.log 2>&1 && \
	{ eval "$testfailed"; }
grep ${keytabfile} ${testdir}/test_context.log > /dev/null || \
	{ echo "string missing failed"; cat ${testdir}/test_context.log ; eval "$testfailed"; }
echo "checking non existant keytabfile (spengo)" ; > ${testdir}/messages.log
${test_context} --mech-type=spnego host@lucid.test.h5l.se > ${testdir}/test_context.log 2>&1 && \
	{ eval "$testfailed"; }
grep ${keytabfile} ${testdir}/test_context.log > /dev/null || \
	{ echo "string missing failed"; cat ${testdir}/test_context.log ; eval "$testfailed"; }

mv ${keytabfile}.no ${keytabfile}

echo "======test naming combinations"
echo "plain"  ; > ${testdir}/messages.log
${test_context} --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }
echo "plain (krb5)" ; > ${testdir}/messages.log
${test_context} --name-type=krb5-principal-name host/lucid.test.h5l.se@${R} || \
	{ eval "$testfailed"; }
echo "plain (krb5 realmless)" ; > ${testdir}/messages.log
${test_context} --name-type=krb5-principal-name host/lucid.test.h5l.se || \
	{ eval "$testfailed"; }
echo "dns canon on (long name) OFF, need dns_wrapper" ; > ${testdir}/messages.log
#${test_context} --dns-canon host@lucid.test.h5l.se || \
#	{ eval "$testfailed"; }
echo "dns canon off (long name)" ; > ${testdir}/messages.log
${test_context} --no-dns-canon host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }
echo "dns canon off (short name)" ; > ${testdir}/messages.log
${test_context} --no-dns-canon host@lucid || \
	{ eval "$testfailed"; }
echo "dns canon off (short name, krb5)" ; > ${testdir}/messages.log
${test_context}  --no-dns-canon --name-type=krb5-principal-name host/lucid@${R} || \
	{ eval "$testfailed"; }
echo "dns canon off (short name, krb5)" ; > ${testdir}/messages.log
${test_context}  --no-dns-canon --name-type=krb5-principal-name host/lucid || \
	{ eval "$testfailed"; }

echo "======test context building"
for mech in krb5 krb5iov spnego spnegoiov; do
	if [ "$mech" = "krb5iov" ] ; then
	    mech="krb5"
	    iov="--iov"
	fi
	if [ "$mech" = "spnegoiov" ] ; then
	    mech="spnego"
	    iov="--iov"
	fi

	echo "${mech} no-mutual ${iov}" ; > ${testdir}/messages.log
	${test_context} --mech-type=${mech} \
	    --wrapunwrap ${iov} \
	    --import-export \
	    --name-type=hostbased-service host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }

	echo "${mech} mutual ${iov}" ; > ${testdir}/messages.log
	${test_context} --mech-type=${mech} \
	    --mutual \
	    --import-export \
	    --wrapunwrap ${iov} \
	    --name-type=hostbased-service host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }

	echo "${mech} delegate ${iov}" ; > ${testdir}/messages.log
	${test_context} --mech-type=${mech} \
	    --delegate \
	    --wrapunwrap ${iov} \
	    --import-export \
	    --name-type=hostbased-service host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }

	echo "${mech} mutual delegate ${iov}" ; > ${testdir}/messages.log
	${test_context} --mech-type=${mech} \
	    --mutual --delegate \
	    --wrapunwrap ${iov} \
	    --import-export \
	    --name-type=hostbased-service host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }
done

echo "======dce-style"
for mech in krb5 krb5iov spnego; do
	iov=""
	if [ "$mech" = "krb5iov" ] ; then
	    mech="krb5"
	    iov="--iov"
	fi
	if [ "$mech" = "spnegoiov" ] ; then
	    mech="spnego"
	    iov="--iov"
	fi

	echo "${mech}: dce-style ${iov}" ; > ${testdir}/messages.log
	${test_context} \
	    --mech-type=${mech} \
	    --mutual \
	    --dce-style \
	    --wrapunwrap ${iov} \
	    --name-type=hostbased-service host@lucid.test.h5l.se || \
	    { eval "$testfailed"; }

done

echo "test gsskrb5_register_acceptor_identity (both positive and negative)"

cp ${keytabfile} ${keytabfile}.new
for mech in krb5 spnego; do
	echo "${mech}: acceptor_identity positive" ; > ${testdir}/messages.log
	${test_context} --gsskrb5-acceptor-identity=${keytabfile}.new \
		--mech-type=$mech host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }

	echo "${mech}: acceptor_identity positive (prefix)" ; > messages.log
	${test_context} --gsskrb5-acceptor-identity=FILE:${keytabfile}.new \
		--mech-type=$mech host@lucid.test.h5l.se || \
		{ eval "$testfailed"; }

	echo "${mech}: acceptor_identity negative" ; > messages.log
	${test_context} --gsskrb5-acceptor-identity=${keytabfile}.foo \
		--mech-type=$mech host@lucid.test.h5l.se 2>/dev/null && \
		{ eval "$testfailed"; }
done

rm ${keytabfile}.new


#echo "sasl-digest-md5"
#${test_context}  --mech-type=sasl-digest-md5 \
#    --name-type=hostbased-service \
#    host@lucid.test.h5l.se || \
#	{ eval "$testfailed"; }


echo "====== gss-api session key check"

# this will break when oneone invents a cooler enctype then aes256-cts-hmac-sha1-96
coolenctype="aes256-cts-hmac-sha1-96"
limit_enctype="des3-cbc-sha1"

echo "Getting client initial tickets" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }


echo "Building context on cred w/o aes, but still ${coolenctype} session key" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --mutual-auth \
    --session-enctype=${coolenctype} \
    --name-type=hostbased-service host@no-aes.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Building context on cred, check if its limited still to ${limit_enctype}" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --client-name=user1@${R} \
    --limit-enctype="${limit_enctype}" \
    --mutual-auth \
    --name-type=hostbased-service host@no-aes.test.h5l.se || \
	{ eval "$testfailed"; }


echo "====== ok-as-delegate"

echo "Getting client initial tickets" ; > ${testdir}/messages.log
${kinit} --forwardable \
    --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }

echo "ok-as-delegate not used" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --delegate \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "host without ok-as-delegate with policy-delegate" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --policy-delegate \
    --server-no-delegate \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "ok-as-delegate used by policy" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --policy-delegate \
    --name-type=hostbased-service host@ok-delegate.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Getting client initial tickets with --ok-as-delgate" ; > ${testdir}/messages.log
${kinit} --ok-as-delegate  --forwardable \
    --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }

echo "policy delegate to non delegate host" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --policy-delegate \
    --server-no-delegate \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "ok-as-delegate" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --delegate \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "======export/import cred"

echo "export-import cred (krb5)" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --delegate \
    --export-import-cred \
    --name-type=hostbased-service host@ok-delegate.test.h5l.se || \
	{ eval "$testfailed"; }

echo "export-import cred (spnego)" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=spnego \
    --delegate \
    --export-import-cred \
    --name-type=hostbased-service host@ok-delegate.test.h5l.se || \
	{ eval "$testfailed"; }


echo "======time diffs between client and server"

echo "Getting client initial ticket" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }

echo "No time offset" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Getting client initial ticket" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }

echo "Server time offset" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --mutual-auth \
    --server-time-offset=3600 \
    --max-loops=3 \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Server time offset (cached ?)" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --mutual-auth \
    --server-time-offset=3600 \
    --max-loops=2 \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Getting client initial ticket" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }
# Pre-poplute the cache since tgs-req will fail since our time is wrong
${kgetcred} host/lucid.test.h5l.se@${R} || \
	{ eval "$testfailed"; }

echo "Client time offset" ; > ${testdir}/messages.log
${test_context} \
    --mech-type=krb5 \
    --mutual-auth \
    --client-time-offset=3600 \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Getting client initial tickets (use-referrals)" ; > ${testdir}/messages.log
${kinit} \
    --password-file=${testdir}/foopassword \
    --use-referrals user1@${R} || \
	{ eval "$testfailed"; }

# XXX these tests really need to use somethat that resolve to something
${test_context} \
    --mech-type=krb5 \
    host@short || \
	{ eval "$testfailed"; }

${test_context} \
    --mech-type=krb5 \
    --name-type=krb5-principal-name host/short || \
	{ eval "$testfailed"; }

${test_context} \
    --mech-type=krb5 \
    host@long.test.h5l.se || \
	{ eval "$testfailed"; }

${test_context} \
    --mech-type=krb5 \
    --name-type=krb5-principal-name \
    host/long.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Make clients credentials cache stale (cpw)" ; > ${testdir}/messages.log

echo "Getting client initial ticket" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }
${kgetcred} host/lucid.test.h5l.se@${R} || \
	{ eval "$testfailed"; }

${kadmin} passwd -r host/lucid.test.h5l.se@${R} || exit 1
${ktutil} -k ${keytab} delete -p host/lucid.test.h5l.se@${R} || exit 1
${kadmin} ext -k ${keytab} host/lucid.test.h5l.se@${R} || exit 1

${test_context} \
    --mech-type=krb5 \
    --mutual-auth \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "Make clients credentials cache stale (delete/add)" ; > ${testdir}/messages.log

echo "Getting client initial ticket" ; > ${testdir}/messages.log
${kinit} --password-file=${testdir}/foopassword user1@${R} || \
	{ eval "$testfailed"; }
${kgetcred} host/lucid.test.h5l.se@${R} || \
	{ eval "$testfailed"; }

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

${test_context} \
    --mech-type=krb5 \
    --mutual-auth \
    --name-type=hostbased-service host@lucid.test.h5l.se || \
	{ eval "$testfailed"; }

echo "======fuzzer"

if [ X"$HEIMDAL_FUZZER" != X ] ; then

    for mech in krb5 spnego ; do
    
        echo "    $mech"
    
        rm -f ${testdir}/dumpdata.bin
    
        ${test_gss_fuzzer} \
            --mech-type=${mech} \
            --dump-data=${testdir}/dumpdata.bin \
            --name-type=hostbased-service host@lucid.test.h5l.se || \
                { \
                  if [ -f ${testdir}/dumpdata.bin ] ; then \
                      openssl base64 -e -in ${testdir}/dumpdata.bin; \
                  fi ; \
                  eval "$testfailed"; \
                }
    
    done
fi


trap "" EXIT

echo "killing kdc (${kdcpid})"
sh ${leaks_kill} kdc $kdcpid || exit 1

exit 0