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

# touch dnsrps-off to not test with DNSRPS
# touch dnsrps-only to not test with classic RPZ

. ../conf.sh

status=0

t=0

DEBUG=
ARGS=

USAGE="$0: [-xS]"
while getopts "xS:" c; do
  case $c in
    x)
      set -x
      DEBUG=-x
      ARGS="$ARGS -x"
      ;;
    S)
      SAVE_RESULTS=-S
      ARGS="$ARGS -S"
      ;;
    *)
      echo "$USAGE" 1>&2
      exit 1
      ;;
  esac
done
shift $((OPTIND - 1))
if test "$#" -ne 0; then
  echo "$USAGE" 1>&2
  exit 1
fi
# really quit on control-C
trap 'exit 1' 1 2 15

DNSRPSCMD=../rpz/dnsrps
RNDCCMD="$RNDC -c ../_common/rndc.conf -p ${CONTROLPORT} -s"

# $1 = test name (such as 1a, 1b, etc. for which named.$1.conf exists)
run_server() {
  TESTNAME=$1

  echo_i "stopping resolver"
  stop_server --use-rndc --port ${CONTROLPORT} ns2

  sleep 1

  echo_i "starting resolver using named.$TESTNAME.conf"
  cp -f ns2/named.$TESTNAME.conf ns2/named.conf
  start_server --noclean --restart --port ${PORT} ns2
  sleep 3
}

run_query() {
  TESTNAME=$1
  LINE=$2

  NAME=$(sed -n -e "$LINE,"'$p' ns2/$TESTNAME.queries | head -n 1)
  $DIG $DIGOPTS $NAME a @10.53.0.2 -p ${PORT} -b 127.0.0.1 >dig.out.${t}
  grep "status: SERVFAIL" dig.out.${t} >/dev/null 2>&1 && return 1
  return 0
}

# $1 = test name (such as 1a, 1b, etc. for which $1.queries exists)
# $2 = line number in query file to test (the name to query is taken from this line)
expect_norecurse() {
  TESTNAME=$1
  LINE=$2

  NAME=$(sed -n -e "$LINE,"'$p' ns2/$TESTNAME.queries | head -n 1)
  t=$((t + 1))
  echo_i "testing $NAME doesn't recurse (${t})"
  add_test_marker 10.53.0.2
  run_query $TESTNAME $LINE || {
    echo_i "test ${t} failed"
    status=1
  }
}

# $1 = test name (such as 1a, 1b, etc. for which $1.queries exists)
# $2 = line number in query file to test (the name to query is taken from this line)
expect_recurse() {
  TESTNAME=$1
  LINE=$2

  NAME=$(sed -n -e "$LINE,"'$p' ns2/$TESTNAME.queries | head -n 1)
  t=$((t + 1))
  echo_i "testing $NAME recurses (${t})"
  add_test_marker 10.53.0.2
  run_query $TESTNAME $LINE && {
    echo_i "test ${t} failed"
    status=1
  }
  return 0
}

add_test_marker() {
  for ns in $@; do
    $RNDCCMD $ns null ---- test ${t} ----
  done
}

native=0
dnsrps=0
for mode in native dnsrps; do
  status=0
  case $mode in
    native)
      if [ -e dnsrps-only ]; then
        echo_i "'dnsrps-only' found: skipping native RPZ sub-test"
        continue
      else
        echo_i "running native RPZ sub-test"
      fi
      ;;
    dnsrps)
      if [ -e dnsrps-off ]; then
        echo_i "'dnsrps-off' found: skipping DNSRPS sub-test"
        continue
      fi
      echo_i "attempting to configure servers with DNSRPS..."
      stop_server --use-rndc --port ${CONTROLPORT}
      $SHELL ./setup.sh -N -D $DEBUG
      sed -n 's/^## //p' dnsrps.conf | cat_i
      if grep '^#fail' dnsrps.conf >/dev/null; then
        echo_i "exit status: 1"
        exit 1
      fi
      if grep '^#skip' dnsrps.conf >/dev/null; then
        echo_i "DNSRPS sub-test skipped"
        continue
      else
        echo_i "running DNSRPS sub-test"
        start_server --noclean --restart --port ${PORT}
        sleep 3
      fi
      ;;
  esac

  # show whether and why DNSRPS is enabled or disabled
  sed -n 's/^## //p' dnsrps.conf | cat_i

  t=$((t + 1))
  echo_i "testing that l1.l0 exists without RPZ (${t})"
  add_test_marker 10.53.0.2
  $DIG $DIGOPTS l1.l0 ns @10.53.0.2 -p ${PORT} >dig.out.${t}
  grep "status: NOERROR" dig.out.${t} >/dev/null 2>&1 || {
    echo_i "test ${t} failed"
    status=1
  }

  t=$((t + 1))
  echo_i "testing that l2.l1.l0 returns SERVFAIL without RPZ (${t})"
  add_test_marker 10.53.0.2
  $DIG $DIGOPTS l2.l1.l0 ns @10.53.0.2 -p ${PORT} >dig.out.${t}
  grep "status: SERVFAIL" dig.out.${t} >/dev/null 2>&1 || {
    echo_i "test ${t} failed"
    status=1
  }

  # Group 1
  run_server 1a
  expect_norecurse 1a 1
  run_server 1b
  expect_norecurse 1b 1
  expect_recurse 1b 2
  run_server 1c
  expect_norecurse 1c 1

  # Group 2
  run_server 2a
  for n in 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
    21 22 23 24 25 26 27 28 29 30 31 32; do
    expect_norecurse 2a $n
  done
  expect_recurse 2a 33

  # Group 3
  run_server 3a
  expect_recurse 3a 1
  run_server 3b
  expect_recurse 3b 1
  run_server 3c
  expect_recurse 3c 1
  run_server 3d
  expect_norecurse 3d 1
  expect_recurse 3d 2
  run_server 3e
  expect_norecurse 3e 1
  expect_recurse 3e 2
  run_server 3f
  expect_norecurse 3f 1
  expect_recurse 3f 2

  # Group 4
  testlist="aa ap bf"
  values="1 16 32"
  # Uncomment the following to test every skip value instead of
  # only a sample of values
  #
  #testlist="aa ab ac ad ae af ag ah ai aj ak al am an ao ap \
  #          aq ar as at au av aw ax ay az ba bb bc bd be bf"
  #values="1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 \
  #        21 22 23 24 25 26 27 28 29 30 31 32"
  set -- $values
  for n in $testlist; do
    run_server 4$n
    ni=$1
    t=$((t + 1))
    echo_i "testing that ${ni} of 33 queries skip recursion (${t})"
    add_test_marker 10.53.0.2
    c=0
    for i in 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 \
      17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33; do
      run_query 4$n $i || c=$((c + 1))
    done
    skipped=$((33 - c))
    if [ $skipped != $ni ]; then
      echo_i "test $t failed (actual=$skipped, expected=$ni)"
      status=1
    fi
    shift
  done

  # Group 5
  run_server 5a
  expect_norecurse 5a 1
  expect_norecurse 5a 2
  expect_recurse 5a 3
  expect_recurse 5a 4
  expect_recurse 5a 5
  expect_recurse 5a 6

  # Group 6
  echo_i "check recursive behavior consistency during policy update races"
  run_server 6a
  sleep 1
  t=$((t + 1))
  echo_i "running dig to cache CNAME record (${t})"
  add_test_marker 10.53.0.1 10.53.0.2
  $DIG $DIGOPTS @10.53.0.2 -p ${PORT} www.test.example.org CNAME >dig.out.${t}
  sleep 1
  echo_i "suspending authority server"
  PID=$(cat ns1/named.pid)
  kill -STOP $PID
  echo_i "adding an NSDNAME policy"
  cp ns2/db.6a.00.policy.local ns2/saved.policy.local
  cp ns2/db.6b.00.policy.local ns2/db.6a.00.policy.local
  $RNDC -c ../_common/rndc.conf -s 10.53.0.2 -p ${CONTROLPORT} reload 6a.00.policy.local 2>&1 | sed 's/^/ns2 /' | cat_i
  test -f dnsrpzd.pid && kill -USR1 $(cat dnsrpzd.pid) || true
  sleep 1
  t=$((t + 1))
  echo_i "running dig to follow CNAME (blocks, so runs in the background) (${t})"
  add_test_marker 10.53.0.2
  $DIG $DIGOPTS @10.53.0.2 -p ${PORT} www.test.example.org A +time=5 >dig.out.${t} &
  sleep 1
  echo_i "removing the NSDNAME policy"
  cp ns2/db.6c.00.policy.local ns2/db.6a.00.policy.local
  $RNDC -c ../_common/rndc.conf -s 10.53.0.2 -p ${CONTROLPORT} reload 6a.00.policy.local 2>&1 | sed 's/^/ns2 /' | cat_i
  test -f dnsrpzd.pid && kill -USR1 $(cat dnsrpzd.pid) || true
  sleep 1
  echo_i "resuming authority server"
  PID=$(cat ns1/named.pid)
  kill -CONT $PID
  add_test_marker 10.53.0.1
  for n in 1 2 3 4 5 6 7 8 9; do
    sleep 1
    [ -s dig.out.${t} ] || continue
    grep "status: .*," dig.out.${t} >/dev/null 2>&1 && break
  done
  grep "status: NOERROR" dig.out.${t} >/dev/null 2>&1 || {
    echo_i "test ${t} failed"
    status=1
  }

  echo_i "check recursive behavior consistency during policy removal races"
  cp ns2/saved.policy.local ns2/db.6a.00.policy.local
  run_server 6a
  sleep 1
  t=$((t + 1))
  echo_i "running dig to cache CNAME record (${t})"
  add_test_marker 10.53.0.1 10.53.0.2
  $DIG $DIGOPTS @10.53.0.2 -p ${PORT} www.test.example.org CNAME >dig.out.${t}
  sleep 1
  echo_i "suspending authority server"
  PID=$(cat ns1/named.pid)
  kill -STOP $PID
  echo_i "adding an NSDNAME policy"
  cp ns2/db.6b.00.policy.local ns2/db.6a.00.policy.local
  $RNDC -c ../_common/rndc.conf -s 10.53.0.2 -p ${CONTROLPORT} reload 6a.00.policy.local 2>&1 | sed 's/^/ns2 /' | cat_i
  test -f dnsrpzd.pid && kill -USR1 $(cat dnsrpzd.pid) || true
  sleep 1
  t=$((t + 1))
  echo_i "running dig to follow CNAME (blocks, so runs in the background) (${t})"
  add_test_marker 10.53.0.2
  $DIG $DIGOPTS @10.53.0.2 -p ${PORT} www.test.example.org A +time=5 >dig.out.${t} &
  sleep 1
  echo_i "removing the policy zone"
  cp ns2/named.default.conf ns2/named.conf
  rndc_reconfig ns2 10.53.0.2
  test -f dnsrpzd.pid && kill -USR1 $(cat dnsrpzd.pid) || true
  sleep 1
  echo_i "resuming authority server"
  PID=$(cat ns1/named.pid)
  kill -CONT $PID
  add_test_marker 10.53.0.1
  for n in 1 2 3 4 5 6 7 8 9; do
    sleep 1
    [ -s dig.out.${t} ] || continue
    grep "status: .*," dig.out.${t} >/dev/null 2>&1 && break
  done
  grep "status: NOERROR" dig.out.${t} >/dev/null 2>&1 || {
    echo_i "test ${t} failed"
    status=1
  }

  # Check maximum number of RPZ zones (64)
  t=$((t + 1))
  echo_i "testing maximum number of RPZ zones (${t})"
  add_test_marker 10.53.0.2
  run_server max
  i=1
  while test $i -le 64; do
    $DIG $DIGOPTS name$i a @10.53.0.2 -p ${PORT} -b 10.53.0.1 >dig.out.${t}.${i}
    grep "^name$i.[ 	]*[0-9]*[ 	]*IN[ 	]*A[ 	]*10.53.0.$i" dig.out.${t}.${i} >/dev/null 2>&1 || {
      echo_i "test $t failed: didn't get expected answer from policy zone $i"
      status=1
    }
    i=$((i + 1))
  done

  # Check CLIENT-IP behavior
  t=$((t + 1))
  echo_i "testing CLIENT-IP behavior (${t})"
  add_test_marker 10.53.0.2
  run_server clientip
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.4 >dig.out.${t}
  grep "status: NOERROR" dig.out.${t} >/dev/null 2>&1 || {
    echo_i "test $t failed: query failed"
    status=1
  }
  grep "^l2.l1.l0.[ 	]*[0-9]*[ 	]*IN[ 	]*A[ 	]*10.53.0.2" dig.out.${t} >/dev/null 2>&1 || {
    echo_i "test $t failed: didn't get expected answer"
    status=1
  }

  # Check CLIENT-IP behavior #2
  t=$((t + 1))
  echo_i "testing CLIENT-IP behavior #2 (${t})"
  add_test_marker 10.53.0.2
  run_server clientip2
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.1 >dig.out.${t}.1
  grep "status: SERVFAIL" dig.out.${t}.1 >/dev/null 2>&1 || {
    echo_i "test $t failed: query failed"
    status=1
  }
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.2 >dig.out.${t}.2
  grep "status: NXDOMAIN" dig.out.${t}.2 >/dev/null 2>&1 || {
    echo_i "test $t failed: query failed"
    status=1
  }
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.3 >dig.out.${t}.3
  grep "status: NOERROR" dig.out.${t}.3 >/dev/null 2>&1 || {
    echo_i "test $t failed: query failed"
    status=1
  }
  grep "^l2.l1.l0.[ 	]*[0-9]*[ 	]*IN[ 	]*A[ 	]*10.53.0.1" dig.out.${t}.3 >/dev/null 2>&1 || {
    echo_i "test $t failed: didn't get expected answer"
    status=1
  }
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.4 >dig.out.${t}.4
  grep "status: SERVFAIL" dig.out.${t}.4 >/dev/null 2>&1 || {
    echo_i "test $t failed: query failed"
    status=1
  }

  # Check RPZ log clause
  t=$((t + 1))
  echo_i "testing RPZ log clause (${t})"
  add_test_marker 10.53.0.2
  run_server log
  cur=$(awk 'BEGIN {l=0} /^/ {l++} END { print l }' ns2/named.run)
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.4 >dig.out.${t}
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.3 >>dig.out.${t}
  $DIG $DIGOPTS l2.l1.l0 a @10.53.0.2 -p ${PORT} -b 10.53.0.2 >>dig.out.${t}
  sed -n "$cur,"'$p' <ns2/named.run | grep "view recursive: rpz CLIENT-IP Local-Data rewrite l2.l1.l0/A/IN via 32.4.0.53.10.rpz-client-ip.log1" >/dev/null && {
    echo_ic "failed: unexpected rewrite message for policy zone log1 was logged"
    status=1
  }
  sed -n "$cur,"'$p' <ns2/named.run | grep "view recursive: rpz CLIENT-IP Local-Data rewrite l2.l1.l0/A/IN via 32.3.0.53.10.rpz-client-ip.log2" >/dev/null || {
    echo_ic "failed: expected rewrite message for policy zone log2 was not logged"
    status=1
  }
  sed -n "$cur,"'$p' <ns2/named.run | grep "view recursive: rpz CLIENT-IP Local-Data rewrite l2.l1.l0/A/IN via 32.2.0.53.10.rpz-client-ip.log3" >/dev/null || {
    echo_ic "failed: expected rewrite message for policy zone log3 was not logged"
    status=1
  }

  # Check wildcard behavior

  t=$((t + 1))
  echo_i "testing wildcard behavior with 1 RPZ zone (${t})"
  add_test_marker 10.53.0.2
  run_server wildcard1
  $DIG $DIGOPTS www.test1.example.net a @10.53.0.2 -p ${PORT} >dig.out.${t}.1
  grep "status: NXDOMAIN" dig.out.${t}.1 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }
  $DIG $DIGOPTS test1.example.net a @10.53.0.2 -p ${PORT} >dig.out.${t}.2
  grep "status: NXDOMAIN" dig.out.${t}.2 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }

  t=$((t + 1))
  echo_i "testing wildcard behavior with 2 RPZ zones (${t})"
  add_test_marker 10.53.0.2
  run_server wildcard2
  $DIG $DIGOPTS www.test1.example.net a @10.53.0.2 -p ${PORT} >dig.out.${t}.1
  grep "status: NXDOMAIN" dig.out.${t}.1 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }
  $DIG $DIGOPTS test1.example.net a @10.53.0.2 -p ${PORT} >dig.out.${t}.2
  grep "status: NXDOMAIN" dig.out.${t}.2 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }

  t=$((t + 1))
  echo_i "testing wildcard behavior with 1 RPZ zone and no non-wildcard triggers (${t})"
  add_test_marker 10.53.0.2
  run_server wildcard3
  $DIG $DIGOPTS www.test1.example.net a @10.53.0.2 -p ${PORT} >dig.out.${t}.1
  grep "status: NXDOMAIN" dig.out.${t}.1 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }
  $DIG $DIGOPTS test1.example.net a @10.53.0.2 -p ${PORT} >dig.out.${t}.2
  grep "status: NOERROR" dig.out.${t}.2 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }

  t=$((t + 1))
  echo_i "testing wildcard passthru before explicit drop (${t})"
  add_test_marker 10.53.0.2
  run_server wildcard4
  $DIG $DIGOPTS example.com a @10.53.0.2 -p ${PORT} >dig.out.${t}.1
  grep "status: NOERROR" dig.out.${t}.1 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }
  $DIG $DIGOPTS www.example.com a @10.53.0.2 -p ${PORT} >dig.out.${t}.2
  grep "status: NOERROR" dig.out.${t}.2 >/dev/null || {
    echo_i "test ${t} failed"
    status=1
  }

  if [ "$mode" = "native" ]; then
    # Check for invalid prefix length error
    t=$((t + 1))
    echo_i "testing for invalid prefix length error (${t})"
    add_test_marker 10.53.0.2
    run_server invalidprefixlength
    grep "invalid rpz IP address \"1000.4.0.53.10.rpz-client-ip.invalidprefixlength\"; invalid prefix length of 1000$" ns2/named.run >/dev/null || {
      echo_ic "failed: expected that invalid prefix length error would be logged"
      status=1
    }
  fi

  t=$((t + 1))
  echo_i "checking 'nsip-wait-recurse no' is faster than 'nsip-wait-recurse yes' ($t)"
  add_test_marker 10.53.0.2 10.53.0.3
  echo_i "timing 'nsip-wait-recurse yes' (default)"
  ret=0
  t1=$($PERL -e 'print time()."\n";')
  $DIG -p ${PORT} @10.53.0.3 foo.child.example.tld a >dig.out.yes.$t
  t2=$($PERL -e 'print time()."\n";')
  p1=$((t2 - t1))
  echo_i "elapsed time $p1 seconds"

  $RNDC -c ../_common/rndc.conf -s 10.53.0.3 -p ${CONTROLPORT} flush
  copy_setports ns3/named2.conf.in ns3/named.conf
  nextpart ns3/named.run >/dev/null
  $RNDC -c ../_common/rndc.conf -s 10.53.0.3 -p ${CONTROLPORT} reload >/dev/null
  wait_for_log 20 "rpz: policy: reload done" ns3/named.run || ret=1

  echo_i "timing 'nsip-wait-recurse no'"
  t3=$($PERL -e 'print time()."\n";')
  $DIG -p ${PORT} @10.53.0.3 foo.child.example.tld a >dig.out.no.$t
  t4=$($PERL -e 'print time()."\n";')
  p2=$((t4 - t3))
  echo_i "elapsed time $p2 seconds"

  if test $p1 -le $p2; then ret=1; fi
  if test $ret != 0; then echo_i "failed"; fi
  status=$((status + ret))

  $RNDC -c ../_common/rndc.conf -s 10.53.0.3 -p ${CONTROLPORT} flush
  # restore original named.conf
  copy_setports ns3/named1.conf.in ns3/named.conf
  nextpart ns3/named.run >/dev/null
  $RNDC -c ../_common/rndc.conf -s 10.53.0.3 -p ${CONTROLPORT} reload >/dev/null
  wait_for_log 20 "rpz: policy: reload done" ns3/named.run || ret=1

  t=$((t + 1))
  echo_i "checking 'nsdname-wait-recurse no' is faster than 'nsdname-wait-recurse yes' ($t)"
  add_test_marker 10.53.0.2 10.53.0.3
  echo_i "timing 'nsdname-wait-recurse yes' (default)"
  ret=0
  t1=$($PERL -e 'print time()."\n";')
  $DIG -p ${PORT} @10.53.0.3 foo.child.example.tld a >dig.out.yes.$t
  t2=$($PERL -e 'print time()."\n";')
  p1=$((t2 - t1))
  echo_i "elapsed time $p1 seconds"

  $RNDC -c ../_common/rndc.conf -s 10.53.0.3 -p ${CONTROLPORT} flush
  copy_setports ns3/named3.conf.in ns3/named.conf
  nextpart ns3/named.run >/dev/null
  $RNDC -c ../_common/rndc.conf -s 10.53.0.3 -p ${CONTROLPORT} reload >/dev/null
  wait_for_log 20 "rpz: policy: reload done" ns3/named.run || ret=1

  echo_i "timing 'nsdname-wait-recurse no'"
  t3=$($PERL -e 'print time()."\n";')
  $DIG -p ${PORT} @10.53.0.3 foo.child.example.tld a >dig.out.no.$t
  t4=$($PERL -e 'print time()."\n";')
  p2=$((t4 - t3))
  echo_i "elapsed time $p2 seconds"

  if test $p1 -le $p2; then ret=1; fi
  if test $ret != 0; then echo_i "failed"; fi
  status=$((status + ret))

  [ $status -ne 0 ] && pf=fail || pf=pass
  case $mode in
    native)
      native=$status
      echo_i "status (native RPZ sub-test): $status ($pf)"
      ;;
    dnsrps)
      dnsrps=$status
      echo_i "status (DNSRPS sub-test): $status ($pf)"
      ;;
    *) echo_i "invalid test mode" ;;
  esac
done
status=$((native + dnsrps))

[ $status -eq 0 ] || exit 1