mirror of
https://github.com/isc-projects/bind9.git
synced 2026-06-21 15:58:52 -04:00
9ffc833919
Replace the custom DNS server used in the "fetchlimit" system test with new code based on the isctest.asyncserver module.
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
"""
|
|
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()
|