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

set -e

umask 077

dir=`dirname "$0"`

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

tcpserver -HRDl0 127.0.0.1 10000 \
sh -c '
  exec 2>tlswrapper.log
  exec tlswrapper -vv -f ./cert.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 -f data.in data.out openssl.log cert.pem tlswrapper.log
  exit "${ex}"
}
trap "cleanup" EXIT TERM INT

SCLIENT_CMD="openssl s_client -nocommands -quiet -tls1_2 -verify_return_error -CAfile ca.pem"
(
  echo "ec prime256v1"
  echo "ec secp384r1"
  echo "ec secp521r1"
  echo "rsa 2048"
  echo "rsa 3072"
  echo "rsa 4096"
) | (
  while read type size; do

    # create CA + certificate
    umask 077
    "${dir}/ca.sh" "${type}" "${size}" >ca.pem
    "${dir}/server.sh" ca.pem "${type}" "${size}" 127.0.0.1 >cert.pem

    # 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} -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 key test: okkey: ${type} ${size}: failed:" >&2
        exit 1
      fi
      echo "TLS key test: okkey: ${type} ${size}: OK"
    )
  done
)

(
  echo "rsa 512"
) | (
  while read type size; do

    # create CA + certificate
    "${dir}/ca.sh" "${type}" "${size}" >ca.pem
    "${dir}/server.sh" ca.pem "${type}" "${size}" 127.0.0.1 >cert.pem

    # 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} -connect 127.0.0.1:10000 1>data.out 2>openssl.log; then
        echo "TLS key test: badkey: ${type} ${size}: failed: badkey accepted" >&2
        exit 1
      fi
      echo "TLS key test: badkey: ${type} ${size}: OK: badkey not accepted"
    )
  done
)
