+++ /dev/null
-#!/usr/bin/perl -w
-
-# 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.
-
-#
-# Don't respond if the "norespond" file exists; otherwise respond to
-# any A or AAAA query.
-#
-
-use IO::File;
-use IO::Socket;
-use Net::DNS;
-use Net::DNS::Packet;
-
-my $localport = int($ENV{'PORT'});
-if (!$localport) { $localport = 5300; }
-
-my $sock = IO::Socket::INET->new(LocalAddr => "10.53.0.4",
- LocalPort => $localport, Proto => "udp") or die "$!";
-
-my $pidf = new IO::File "ans.pid", "w" or die "cannot open pid file: $!";
-print $pidf "$$\n" or die "cannot write pid file: $!";
-$pidf->close or die "cannot close pid file: $!";
-sub rmpid { unlink "ans.pid"; exit 1; };
-
-$SIG{INT} = \&rmpid;
-$SIG{TERM} = \&rmpid;
-
-for (;;) {
- $sock->recv($buf, 512);
-
- print "**** request from " , $sock->peerhost, " port ", $sock->peerport, "\n";
-
- my $packet;
-
- if ($Net::DNS::VERSION > 0.68) {
- $packet = new Net::DNS::Packet(\$buf, 0);
- $@ and die $@;
- } else {
- my $err;
- ($packet, $err) = new Net::DNS::Packet(\$buf, 0);
- $err and die $err;
- }
-
- print "REQUEST:\n";
- $packet->print;
-
- $packet->header->qr(1);
-
- my @questions = $packet->question;
- my $qname = $questions[0]->qname;
- my $qtype = $questions[0]->qtype;
-
- my $donotrespond = 0;
-
- if (-e 'norespond') {
- $donotrespond = 1;
- } else {
- $packet->header->aa(1);
- if ($qtype eq "A") {
- $packet->push("answer",
- new Net::DNS::RR($qname .
- " 300 A 192.0.2.1"));
- } elsif ($qtype eq "AAAA") {
- $packet->push("answer",
- new Net::DNS::RR($qname .
- " 300 AAAA 2001:db8:beef::1"));
- }
- }
-
- if ($donotrespond == 0) {
- if (index($qname, "latency") == 0) {
- # 50ms latency
- select(undef, undef, undef, 0.05);
- }
- $sock->send($packet->data);
- print "RESPONSE:\n";
- $packet->print;
- print "\n";
- }
-}
--- /dev/null
+"""
+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.
+"""
+
+from typing import AsyncGenerator
+
+import dns
+
+from isctest.asyncserver import (
+ ControllableAsyncDnsServer,
+ DnsResponseSend,
+ QueryContext,
+ ResponseHandler,
+ ToggleResponsesCommand,
+)
+
+
+class MaybeDelayedAddressAnswerHandler(ResponseHandler):
+ async def get_responses(
+ self, qctx: QueryContext
+ ) -> AsyncGenerator[DnsResponseSend, None]:
+ if qctx.qtype in (dns.rdatatype.A, dns.rdatatype.AAAA):
+ addr = "192.0.2.1" if qctx.qtype == dns.rdatatype.A else "2001:db8:beef::1"
+ rrset = dns.rrset.from_text(qctx.qname, 300, qctx.qclass, qctx.qtype, addr)
+ qctx.response.answer.append(rrset)
+
+ qctx.response.set_rcode(dns.rcode.NOERROR)
+ delay = 0.05 if qctx.qname.labels[0].startswith(b"latency") else 0.00
+ yield DnsResponseSend(qctx.response, delay=delay, authoritative=True)
+
+
+def main() -> None:
+ server = ControllableAsyncDnsServer([ToggleResponsesCommand])
+ server.install_response_handler(MaybeDelayedAddressAnswerHandler())
+ server.run()
+
+
+if __name__ == "__main__":
+ main()
"$RNDC" -c ../_common/rndc.conf -p "${CONTROLPORT}" -s "$@"
)
+dig_with_opts() (
+ "$DIG" -p "$PORT" "$@"
+)
+
+sendcmd() (
+ SERVER="${1}"
+ COMMAND="${2}"
+ COMMAND_ARGS="${3}"
+ dig_with_opts "@${SERVER}" "${COMMAND_ARGS}.${COMMAND}._control." TXT +time=5 +tries=1 +tcp >/dev/null 2>&1
+)
+
burst() {
server=${1}
num=${4:-20}
ret=0
# make the server lame and restart
rndccmd 10.53.0.3 flush
-touch ans4/norespond
+sendcmd 10.53.0.4 send-responses "disable"
for try in 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20; do
burst 10.53.0.3 a $try
# fetches-per-server is at 400, but at 20qps against a lame server,
n=$((n + 1))
echo_i "checking lame server recovery ($n)"
ret=0
-test -f ans4/norespond && rm -f ans4/norespond
+sendcmd 10.53.0.4 send-responses "enable"
for try in 1 2 3 4 5; do
burst 10.53.0.3 b $try
stat 10.53.0.3 0 200 || ret=1
ret=0
fail=0
success=0
-touch ans4/norespond
+sendcmd 10.53.0.4 send-responses "disable"
for try in 1 2 3 4 5; do
burst 10.53.0.3 d $try 300
$DIGCMD a ${try}.example >dig.out.ns3.$n.$try
fail=0
exceeded=0
success=0
-touch ans4/norespond
+sendcmd 10.53.0.4 send-responses "disable"
for try in 1 2 3 4 5; do
burst 10.53.0.3 b $try 400
$DIGCMD +time=2 a ${try}.example >dig.out.ns3.$n.$try
n=$((n + 1))
echo_i "checking clients are dropped at the clients-per-query limit ($n)"
ret=0
-test -f ans4/norespond && rm -f ans4/norespond
+sendcmd 10.53.0.4 send-responses "enable"
for try in 1 2 3 4 5; do
burst 10.53.0.5 latency $try 20 "dup"
sleep 1
n=$((n + 1))
echo_i "checking clients are dropped at the clients-per-query limit with stale-answer-client-timeout ($n)"
ret=0
-test -f ans4/norespond && rm -f ans4/norespond
+sendcmd 10.53.0.4 send-responses "enable"
for try in 1 2 3 4 5; do
burst 10.53.0.5 latency $try 20 "dup"
sleep 1
projects / thirdparty / bind9.git / commitdiff
