#!/bin/sh
# $Id: t-pmilter-14.sh,v 1.4 2007/01/11 02:00:26 ca Exp $
# Copyright (c) 2005 Sendmail, Inc. and its suppliers.
#	All rights reserved.
#
# By using this file, you agree to the terms and conditions set
# forth in the LICENSE file which can be found at the top level of
# the sendmail distribution.
#
# ----------------------------------------
# test pmilter restart
# uses smtpcnf.sh.
# ----------------------------------------
#
test -s stop && exit 1
#

if test X"${MTA_NO_SLOW_TEST}" != "X"
then
  echo "SKIPPED: $0: MTA_NO_SLOW_TEST is set"
  exit 0
fi

SD=`dirname $0`
if test "${SD}" = "."
then
   SD=../chkmts
fi
. ${SD}/common.sh
V=0

SCRIPTS=`cd ${SD} && pwd`

if ../smtps/smtps -VVV | ${GREP} MTA_USE_PMILTER >/dev/null
then
  :
else
  echo "SKIPPED: $0: smtps not compiled with MTA_USE_PMILTER (--enable-pmilter)"
  exit 0
fi

if test X"`uname`" = "XLinux"
then
  echo "SKIPPED: $0: on Linux, needs POSIX conformant pthread implementation"
  exit 0
fi

OUT="t-pmilter-14.out"
PMPRG="-p ../libpmilter/t-pmilter-1"
PMOPT=""

if test X"`uname`" != "XOSF1"
then
while getopts O:p:s:V FLAG
do
  case "${FLAG}" in
    O) PMOPT="${PMOPT} ${OPTARG}";;
    p) PMPRG="-p ${OPTARG}";;
    s) SKIP="${SKIP} ${OPTARG}";;
    V) VERBOSE=true;;
  esac
done
shift `expr ${OPTIND} - 1`
fi

# cd to right directory (where the programs will be executed).
cd ../qmgr || exit 1
rm -f ${MTMAP}

${CAT} > ${SMXCNF} <<EOF
qmgr { Log_Level = 12; wait_for_server=4; wait_for_client=4; }
smar { Log_Level = 12; nameserver = ${NS}; }
smtpc { Log_Level = 12; remote_port=${SNKPORT}; wait_for_server=4; }
smtps {
Log_Level = 14;
daemon_address = localhost:${SRVPORT};
wait_for_server=4;
policy_milter { socket {type=unix; path="pmilter.sock";} timeout=20;
 flags=accept_but_reconnect;}
}
EOF

# ----------------------------------------
# reject first MAIL command in pmilter
# then send two messages:
# one must wait before MAIL (in source), one must wait at MAIL (in pmilter)
# at that momemt kill pmilter so smtps has two open connections to it:
# one reading, one before writing
# send two more messages: first should go through (not yet reconnected),
# second should be stopped.
TEST=1
if echo "${SKIP} X" | ${GREP} -v " ${TEST} " > /dev/null
then
RCPT=x@y.z
rm -f ${PMIDS}
PMOPTS="-r m=550 -w m=10"
if ${SHELL} ${SCRIPTS}/smtpcnf.sh -V '-d 8' -z ${PMPRG} ${PMOPT} -P "${PMOPTS}" -Z 'smtpc2' -c '-E -O 20' -E 0 -A 0 -X 0 $@ -Sa0@b.c -R${RCPT}
then
  ERR=false

  # first mail rejected
  if ${GREP} 'MAIL=error' ${CLTL} >/dev/null
  then
    if ${GREP} -i 'err' ${PML} >/dev/null
    then
      echo "$0: test ${TEST} failed: err not found in ${PML}"
      ERR=true
      test X"${MTA_STOPONERROR}" != X && exit 1
    fi
  else
    echo "$0: test ${TEST} failed: MAIL=error not found in ${CLTL}"
    ERR=true
    test X"${MTA_STOPONERROR}" != X && exit 1
  fi

  if test -s ${PMIDS}
  then

    ${STE}/smtpc2 -E -W m=10 -O 20 -r ${H}:${SRVPORT} -fa1@b.c ${RCPT} > ${CLTL} 2>&1 &
    SRCID=$!
    sleep 1
    ${STE}/smtpc2 -E -O 20 -r ${H}:${SRVPORT} -fa2@b.c ${RCPT} > ${CLTL2} 2>&1 &
    SRC2ID=$!
    sleep 1

    # kill pmilter while in MAIL (hopefully)
    kill -HUP `${CAT} ${PMIDS}`

    # check first source: should go through (pmilter killed in MAIL)
    wait ${SRCID}
    if ${GREP} "total=1 " ${CLTL} >/dev/null
    then
     :
    else
      ERR=true
      echo "$0: test ${TEST} failed: total!=1 in ${CLTL}; 1"
      ${VERBOSE} && ${GREP} "total=" ${CLTL}
      test X"${MTA_STOPONERROR}" != X && exit 1
    fi

    # check second source: should go through (not yet reconnected)
    wait ${SRC2ID}
    if ${GREP} "total=1 " ${CLTL2} >/dev/null
    then
     :
    else
      ERR=true
      echo "$0: test ${TEST} failed: total!=1 in ${CLTL2}; 1"
      ${VERBOSE} && ${GREP} "total=" ${CLTL2}
      test X"${MTA_STOPONERROR}" != X && exit 1
    fi

    # start new pmilter
    ../libpmilter/t-pmilter-1 ${PMOPT} ${PMOPTS} > m${V}.log 2>&1 &
    PMID=$!
    echo ${PMID} >> ${PIDS}
    echo ${PMID} >> ${PMIDS}
    sleep 2

    ${STE}/smtpc2 -E -O 20 -r ${H}:${SRVPORT} -fa3@b.c ${RCPT} > ${CLTL} 2>&1 &
    SRCID=$!
    sleep 1
    ${STE}/smtpc2 -E -O 20 -r ${H}:${SRVPORT} -fa4@b.c ${RCPT} > ${CLTL2} 2>&1 &
    SRC2ID=$!
    sleep 1

    # check first source: should be blocked by pmilter in MAIL
    # (this session will do the reconnect)
    wait ${SRCID}
    if ${GREP} 'MAIL=error' ${CLTL} >/dev/null
    then
      if ${GREP} -i 'err' ${PML} >/dev/null
      then
        echo "$0: test ${TEST} failed: err not found in ${PML}"
        ERR=true
        test X"${MTA_STOPONERROR}" != X && exit 1
      fi
    else
      echo "$0: test ${TEST} failed: MAIL=error not found in ${CLTL}"
      ERR=true
      test X"${MTA_STOPONERROR}" != X && exit 1
    fi

    # check second source: should be blocked by pmilter in MAIL
    wait ${SRC2ID}
    if ${GREP} 'MAIL=error' ${CLTL2} >/dev/null
    then
      if ${GREP} -i 'err' ${PML} >/dev/null
      then
        echo "$0: test ${TEST} failed: err not found in ${PML}"
        ERR=true
        test X"${MTA_STOPONERROR}" != X && exit 1
      fi
    else
      echo "$0: test ${TEST} failed: MAIL=error not found in ${CLTL2}"
      ERR=true
      test X"${MTA_STOPONERROR}" != X && exit 1
    fi

  else
    echo "$0: test ${TEST}: cannot find pmilter pid"
  fi
else
    ERR=true
    echo "$0: test ${TEST} failed"
    test X"${MTA_STOPONERROR}" != X && exit 1
fi
rm -f ${PMIDS}

if ${ERR}
then
    echo "$0: test ${TEST} failed"
    ERRS=`expr ${ERRS} + 1 `
    test X"${MTA_STOPONERROR}" != X && exit 1
else
    ${VERBOSE} && echo "test ${TEST} succeeded"
fi
${SD}/stopmts.sh
fi


# ----------------------------------------
# end
if test "${ERRS}" = "0"
then
  exit 0
else
  echo "${ERRS} error(s)"
  exit 1
fi