#!/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$
#

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

R=EXAMPLE.ORG

port=8888

kadmin="../../kadmin/kadmin -l -r $R"
kdc="../../kdc/kdc --addresses=localhost -P $port"

server=host/datan.example.org
cache="FILE:${objdir}/cache.krb5"
keytabfile=${objdir}/server.keytab
keytab="FILE:${keytabfile}"

kinit="../../kuser/kinit -c $cache --no-afslog"
klist="../../kuser/klist -c $cache"
kgetcred="../../kuser/kgetcred -c $cache"
kdestroy="../../kuser/kdestroy -c $cache"
ktutil="../../admin/ktutil"

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

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

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

${kadmin} add -p foo --use-defaults foo@${R} || exit 1
${kadmin} add -p bar --use-defaults bar@${R} || exit 1
${kadmin} add -p kaka --use-defaults ${server}@${R} || exit 1
${kadmin} ext -k ${keytab} ${server}@${R} || exit 1

${ktutil} -k ${keytab} list > tempfile || exit 1
grep -ve '^FILE:' tempfile | grep -ve '^Vno' | \
    awk '/1/ !~ $1 { exit 1 }' || exit 1

${kadmin} get foo@${R} > tempfile || exit 1
enctypes=`grep Keytypes: tempfile | sed 's/(pw-salt)//g' | sed 's/,//g' | sed 's/Keytypes://'`

enctype_sans_aes=`echo $enctypes | sed 's/aes[^ ]*//g'`

echo foo > ${objdir}/foopassword

echo Starting kdc
${kdc} &
kdcpid=$!
sleep 2

if tail -1 messages.log | grep "No sockets" ; then
    echo "The KDC failed to bind to any sockets, another KDC running ?"
    exit 1
fi

exitcode=0

echo "Getting client initial tickets"
${kinit} --password-file=${objdir}/foopassword foo@$R || exitcode=1
echo "Getting tickets"
${kgetcred} ${server}@${R} || exitcode=1
echo "Listing tickets"
${klist} > /dev/null || exitcode=1
./ap-req ${server}@${R} ${keytab} ${cache} || exitcode=1
${kdestroy}

for a in $enctypes; do
	echo "Getting client initial tickets ($a)"
	${kinit} --enctype=$a --password-file=${objdir}/foopassword foo@$R || exitcode=1
	echo "Getting tickets"
	${kgetcred} ${server}@${R} || exitcode=1
	./ap-req ${server}@${R} ${keytab} ${cache} || exitcode=1
	${kdestroy}
done


echo "Getting client initial tickets"
${kinit} --password-file=${objdir}/foopassword foo@$R || exitcode=1
for a in $enctypes; do
	echo "Getting tickets ($a)"
	${kgetcred} -e $a ${server}@${R} || exitcode=1
done
${kdestroy}

echo "try all permutations"
for a in $enctypes; do
	echo "Getting client initial tickets ($a)"
	${kinit} --enctype=$a --password-file=${objdir}/foopassword foo@$R || exitcode=1
	for b in $enctypes; do
		echo "Getting tickets ($a ->  $b)"
		${kgetcred} -e $b ${server}@${R} || exitcode=1
	done
	${kdestroy}
done

echo "Getting server initial tickets"
${kinit} --keytab=${keytab} ${server}@$R || exitcode=1
echo "Listing tickets"
${klist} | grep "Principal: ${server}" > /dev/null || exitcode=1
${kdestroy}

#echo deleting all but aes enctypes on krbtgt
#${kadmin} del_enctype krbtgt/${R}@${R} ${enctype_sans_aes} || exit 1
#
#echo "try all permutations (only aes)"
#for a in $enctypes; do
#	echo "Getting client initial tickets ($a)"
#	${kinit} --enctype=$a --password-file=${objdir}/foopassword foo@$R || exitcode=1
#	for b in $enctypes; do
#		echo "Getting tickets ($a ->  $b)"
#		${kgetcred} -e $b ${server}@${R} || exitcode=1
#	done
#	${kdestroy}
#done

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

exit $exitcode