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

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

. ${env_setup}
. ${top_srcdir}/tests/bin/test-lib.sh

R=TEST.H5L.SE

port=@port@

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

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

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

KRB5_KTNAME="${keytab}-no"
export KRB5_KTNAME
KRB5CCNAME="${cache}-no"
export KRB5CCNAME

KRB5_TRACE="0-/FILE:${objdir}/messages.log"

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

GSS_MECH_CONFIG="${objdir}/mech"
export GSS_MECH_CONFIG

test_init

test_section "context building for negoex"

for HOPS in 1 2 3 4 5
do
    test_section "test_negoex_1 $HOPS hops"
    test_run env HOPS=$HOPS ${context} \
	--mech-type=spnego --ret-mech-type=test_negoex_1 \
	--name-type=hostbased-service \
	host@host.test.h5l.se
done

for HOPS in 1 2 3 4 5
do
    test_section "test_negoex_1 $HOPS hops early keys"
    test_run env HOPS=$HOPS KEY=always ${context} \
	--mech-type=spnego --ret-mech-type=test_negoex_1 \
	--name-type=hostbased-service \
	host@host.test.h5l.se
done

test_section "test_negoex_1 no keys (expected failure)"
test_run not env HOPS=1 KEY=never ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 no optimistic token"
test_run env HOPS=1 NEGOEX_NO_OPTIMISTIC_TOKEN=1 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 initiator query fail, test_negoex_2 pass"
test_run env INIT_QUERY_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_2 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 acceptor query fail, test_negoex_2 pass"
test_run env ACCEPT_QUERY_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_2 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 acceptor exchange fail, test_negoex_2 pass"
test_run env ACCEPT_EXCHANGE_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_2 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 first mech initiator exchange fail (expected failure)"
test_run not env INIT_EXCHANGE_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 first mech initiator exchange fail, two hops (expected failure)"
test_run not env HOPS=2 INIT_EXCHANGE_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 first mech initiator exchange fail, two hops, early keys (expected failure)"
test_run not env HOPS=2 KEY=always INIT_EXCHANGE_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 first mech init_sec_context fail (expected failure)"
test_run not env INIT_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 first mech accept_sec_context fail (expected failure)"
test_run not env HOPS=2 ACCEPT_FAIL=102 ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 alert from acceptor to initiator"
test_run env HOPS=3 KEY=init-always ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

test_section "test_negoex_1 alert from initiator to acceptor"
test_run env HOPS=4 KEY=accept-always ${context} \
    --mech-type=spnego --ret-mech-type=test_negoex_1 \
    --name-type=hostbased-service \
    host@host.test.h5l.se

unset GSS_MECH_CONFIG

test_section "test context building for sanon-x25519"
for mech in sanon-x25519 sanon-x25519iov spnego spnegoiov; do
	iov=""
	if [ "$mech" = "sanon-x25519iov" ] ; then
	    mech="sanon-x25519"
	    iov="--iov"
	fi
	if [ "$mech" = "spnegoiov" ] ; then
	    mech="spnego"
	    iov="--iov"
	fi

	test_section "${mech} anon-flag ${iov}"
	test_run ${context} --mech-type=${mech} \
	    --anonymous \
	    --ret-mech-type=sanon-x25519 \
	    --i-channel-bindings=negoex_sanon_test_h5l_se \
	    --a-channel-bindings=negoex_sanon_test_h5l_se \
	    --wrapunwrap ${iov} \
	    host@lucid.test.h5l.se

	test_section "${mech} anon-initiator ${iov}"
	test_run env KRB5_TRACE=0-/FILE:/dev/stderr ${context} --mech-type=${mech} \
	    --anonymous \
	    --client-name=WELLKNOWN/ANONYMOUS@WELLKNOWN:ANONYMOUS \
	    --ret-mech-type=sanon-x25519 \
	    --i-channel-bindings=negoex_sanon_test_h5l_se \
	    --a-channel-bindings=negoex_sanon_test_h5l_se \
	    --wrapunwrap ${iov} \
	    host@lucid.test.h5l.se

	test_section "${mech} anon-acceptor ${iov}"
	test_run env KRB5_TRACE=0-/FILE:/dev/stderr ${context} --mech-type=${mech} \
	    --anonymous \
	    --ret-mech-type=sanon-x25519 \
	    --i-channel-bindings=negoex_sanon_test_h5l_se \
	    --a-channel-bindings=negoex_sanon_test_h5l_se \
	    --wrapunwrap ${iov} \
	    WELLKNOWN@ANONYMOUS
done

test_section "export-import-context for sanon-x25519"
for mech in sanon-x25519 sanon-x25519iov spnego spnegoiov; do
	iov=""
	if [ "$mech" = "sanon-x25519iov" ] ; then
	    mech="sanon-x25519"
	    iov="--iov"
	fi
	if [ "$mech" = "spnegoiov" ] ; then
	    mech="spnego"
	    iov="--iov"
	fi

	test_section "${mech}: export-import-context ${iov}"
	test_run ${context} \
	    --mech-type=${mech} \
	    --anonymous \
	    --export-import-context \
	    --wrapunwrap ${iov} \
	    --name-type=hostbased-service host@lucid.test.h5l.se

	test_section "${mech}: export-import-context ${iov} (split tokens)"
	test_run ${context} \
	    --mech-type=${mech} \
	    --anonymous \
	    --export-import-context \
	    --wrapunwrap ${iov} \
	    --token-split=128 \
	    --name-type=hostbased-service host@lucid.test.h5l.se

done

test_section "dce-style for sanon-x25519"
for mech in spnego spnegoiov; do
	iov=""
	if [ "$mech" = "spnegoiov" ] ; then
	    mech="spnego"
	    iov="--iov"
	fi

	test_section "${mech}: dce-style ${iov}"
	test_run ${context} \
	    --mech-type=${mech} \
	    --anonymous --dce-style \
	    --wrapunwrap ${iov} \
	    --name-type=hostbased-service host@lucid.test.h5l.se

done

test_finish
exit $?