#!/bin/sh

# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
#
# SPDX-License-Identifier: MPL-2.0
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0.  If a copy of the MPL was not distributed with this
# file, you can obtain one at https://mozilla.org/MPL/2.0/.
#
# See the COPYRIGHT file distributed with this work for additional
# information regarding copyright ownership.

set -e

#shellcheck source=conf.sh
. ../conf.sh

dig_with_opts() {
  "$DIG" -p "${PORT}" "$@"
}

wait_for_pid() (
  for pid in "$@"; do
    kill -0 "$pid" 2>/dev/null && return 1
  done
  return 0
)

status=0
n=0

n=$((n + 1))
echo_i "check lookups against TTL=0 records ($n)"
i=0
ret=0
passes=10
dig_with_opts @10.53.0.2 axfr example | grep -v "^ds0" \
  | awk '$2 == "0" { print "-q", $1, $4; print "-q", "zzz"$1, $4;}' >query.list

# add 1/5 second per query
timeout=$(($(wc -l <query.list) / 5))
while [ $i -lt $passes ]; do
  (dig_with_opts -d +qr @10.53.0.3 -f query.list >"dig.out$i.1.test$n" 2>&1) &
  pid1="$!"
  (dig_with_opts -d +qr @10.53.0.3 -f query.list >"dig.out$i.2.test$n" 2>&1) &
  pid2="$!"
  (dig_with_opts -d +qr @10.53.0.3 -f query.list >"dig.out$i.3.test$n" 2>&1) &
  pid3="$!"
  (dig_with_opts -d +qr @10.53.0.3 -f query.list >"dig.out$i.4.test$n" 2>&1) &
  pid4="$!"
  (dig_with_opts -d +qr @10.53.0.3 -f query.list >"dig.out$i.5.test$n" 2>&1) &
  pid5="$!"
  (dig_with_opts -d +qr @10.53.0.3 -f query.list >"dig.out$i.6.test$n" 2>&1) &
  pid6="$!"

  retry_quiet "$timeout" wait_for_pid "$pid1" "$pid2" "$pid3" "$pid4" "$pid5" "$pid6" || {
    echo_i "wait_for_pid failed"
    ret=1
  }
  kill -TERM "$pid1" "$pid2" "$pid3" "$pid4" "$pid5" "$pid6" 2>/dev/null || true

  wait "$pid1" || {
    echo_i "wait $pid1 (dig.out$i.1.test$n) failed with $?"
    ret=1
  }
  wait "$pid2" || {
    echo_i "wait $pid2 (dig.out$i.2.test$n) failed with $?"
    ret=1
  }
  wait "$pid3" || {
    echo_i "wait $pid3 (dig.out$i.3.test$n) failed with $?"
    ret=1
  }
  wait "$pid4" || {
    echo_i "wait $pid4 (dig.out$i.4.test$n) failed with $?"
    ret=1
  }
  wait "$pid5" || {
    echo_i "wait $pid5 (dig.out$i.5.test$n) failed with $?"
    ret=1
  }
  wait "$pid6" || {
    echo_i "wait $pid6 (dig.out$i.6.test$n) failed with $?"
    ret=1
  }

  grep "status: SERVFAIL" "dig.out$i.1.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.2.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.3.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.4.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.5.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.6.test$n" >/dev/null && ret=1
  [ $ret = 1 ] && break
  i=$((i + 1))
  echo_i "successfully completed pass $i of $passes"
done
if [ $ret != 0 ]; then echo_i "failed"; fi
status=$((status + ret))

repeat_query() (
  i=0
  while [ "$i" -lt "$1" ]; do
    dig_with_opts +short "@$2" "$3" | tee "dig.out$i.test$n" || return 1
    i=$((i + 1))
  done
)

count_unique() (
  repeat_query "$@" | sort -u | wc -l
)

n=$((n + 1))
echo_i "check repeated recursive lookups of non recurring TTL=0 responses get new values ($n)"
ret=0
repeats=9
count=$(count_unique "$repeats" 10.53.0.3 foo.increment)
if [ "$count" -ne "$repeats" ]; then
  echo_i "failed (count=$count, repeats=$repeats)"
  ret=1
fi
status=$((status + ret))

n=$((n + 1))
echo_i "check lookups against TTL=1 records ($n)"
i=0
passes=10
ret=0
while [ $i -lt $passes ]; do
  dig_with_opts @10.53.0.3 www.one.tld >"dig.out$i.1.test$n" || ret=1
  dig_with_opts @10.53.0.3 www.one.tld >"dig.out$i.2.test$n" || ret=1
  dig_with_opts @10.53.0.3 www.one.tld >"dig.out$i.3.test$n" || ret=1
  dig_with_opts @10.53.0.3 www.one.tld >"dig.out$i.4.test$n" || ret=1
  dig_with_opts @10.53.0.3 www.one.tld >"dig.out$i.5.test$n" || ret=1
  dig_with_opts @10.53.0.3 www.one.tld >"dig.out$i.6.test$n" || ret=1
  grep "status: SERVFAIL" "dig.out$i.1.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.2.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.3.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.4.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.5.test$n" >/dev/null && ret=1
  grep "status: SERVFAIL" "dig.out$i.6.test$n" >/dev/null && ret=1
  [ $ret = 1 ] && break
  i=$((i + 1))
  echo_i "successfully completed pass $i of $passes"
  sleep 1
done
if [ $ret != 0 ]; then echo_i "failed"; fi
status=$((status + ret))

echo_i "exit status: $status"
[ "$status" -eq 0 ] || exit 1