#!/bin/sh
# 20210123
# Jan Mojzis
# Public domain.

set -e

umask 077

dir=`dirname "$0"`

# change directory to $AUTOPKGTEST_TMP
cd "${AUTOPKGTEST_TMP}"

# run tlswrapper port 10000
tcpserver -HRDl0 127.0.0.1 10000 \
sh -c '
  exec 2>tlswrapper.log
  exec tlswrapper -vv \
                  -e x25519 \
                  -e secp256r1 \
                  -e secp384r1 \
                  -e secp521r1 \
                  -f rsa.pem \
                  cat data.in
' &
tcpserverpid=$!
# Give some extra time for tcpserver to start,
# to avoid flaky test failures on slower testbeds
sleep 1

cleanup() {
  ex=$?
  #kill tcpserver
  kill -TERM "${tcpserverpid}" 1>/dev/null 2>/dev/null || :
  kill -KILL "${tcpserverpid}" 1>/dev/null 2>/dev/null || :
  if [ ${ex} -gt 0 ]; then
    (
      echo "tlswrapper.log:"
      cat tlswrapper.log || :
      echo "openssl.log:"
      cat openssl.log || :
    ) >&2
  fi
  rm -rf ec data.in data.out log rsa.pem tlswrapper.log
  exit "${ex}"
}
trap "cleanup" EXIT TERM INT

# create CA
"${dir}/ca.sh" ec prime256v1 >ca.pem

# create RSA certfile
"${dir}/server.sh" ca.pem rsa 2048 127.0.0.1 > rsa.pem

SCLIENT_CMD="openssl s_client -nocommands -quiet -tls1_2 -verify_return_error -CAfile ca.pem"
(
  echo X25519
  echo prime256v1
  echo secp384r1
  echo secp521r1
) | (
  while read curve name; do
    # create random datafile
    dd if=/dev/urandom of=data.in bs=1 count=32 2>/dev/null

    # run test
    (
      exec 0</dev/null
      ${SCLIENT_CMD} -curves "${curve}" -connect 127.0.0.1:10000 1>data.out 2>openssl.log || rm data.out
      if [ x"`sha512sum < data.in`" != x"`sha512sum < data.out`" ]; then
        echo "TLS ephemeral test: ${curve}: failed:" >&2
        exit 1
      fi
      echo "TLS ephemeral test: ${curve}: OK"
    )
  done
)

(
  echo secp192r1
  echo secp192k1
  echo secp224r1
  echo secp224k1
  echo secp256k1
) | (
  while read curve name; do
    # create random datafile
    dd if=/dev/urandom of=data.in bs=1 count=32 2>/dev/null

    # run test
    (
      exec 0</dev/null
      if ${SCLIENT_CMD} -curves "${curve}" -connect 127.0.0.1:10000 1>data.out 2>openssl.log; then
        echo "TLS ephemeral test: badalgo: ${curve}: failed: bad algorithm accepted" >&2
        exit 1
      fi
      echo "TLS ephemeral test: badalgo: ${curve}: OK: bad algorithm not accepted"
    )
  done
)
