mirror of
https://github.com/isc-projects/bind9.git
synced 2026-03-22 18:33:11 -04:00
168dba163c
The old name "common" clashes with the convention of system test directory naming. It appears as a system test directory, but it only contains helper files. To reduce confusion and to allow automatic detection of issues with possibly missing test files, rename the helper directory to "_common". The leading underscore indicates the directory is different and the its name can no longer be confused with regular system test directories.
83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
#!/usr/bin/python3
|
|
|
|
# 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.
|
|
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
pytest.importorskip("dns", minversion="2.0.0")
|
|
import dns.resolver
|
|
|
|
|
|
def run_rndc(server, rndc_command):
|
|
"""
|
|
Send the specified 'rndc_command' to 'server' with a timeout of 10 seconds
|
|
"""
|
|
rndc = os.getenv("RNDC")
|
|
port = os.getenv("CONTROLPORT")
|
|
|
|
cmdline = [rndc, "-c", "../_common/rndc.conf", "-p", port, "-s", server]
|
|
cmdline.extend(rndc_command)
|
|
|
|
subprocess.check_output(cmdline, stderr=subprocess.STDOUT, timeout=10)
|
|
|
|
|
|
def test_dnstap_dispatch_socket_addresses(named_port):
|
|
# Prepare for querying ns3.
|
|
resolver = dns.resolver.Resolver()
|
|
resolver.nameservers = ["10.53.0.3"]
|
|
resolver.port = named_port
|
|
|
|
# Send some query to ns3 so that it records something in its dnstap file.
|
|
ans = resolver.resolve("mail.example.", "A")
|
|
assert ans[0].address == "10.0.0.2"
|
|
|
|
# Before continuing, roll dnstap file to ensure it is flushed to disk.
|
|
run_rndc("10.53.0.3", ["dnstap", "-roll", "1"])
|
|
|
|
# Move the dnstap file aside so that it is retained for troubleshooting.
|
|
os.rename(os.path.join("ns3", "dnstap.out.0"), "dnstap.out.resolver_addresses")
|
|
|
|
# Read the contents of the dnstap file using dnstap-read.
|
|
output = subprocess.check_output(
|
|
[os.getenv("DNSTAPREAD"), "dnstap.out.resolver_addresses"], encoding="utf-8"
|
|
)
|
|
|
|
# Check whether all frames contain the expected addresses.
|
|
#
|
|
# Expected dnstap-read output format:
|
|
#
|
|
# 22-Jun-2022 12:09:06.168 RR 10.53.0.3:0 -> 10.53.0.1:7523 TCP ...
|
|
# 22-Jun-2022 12:09:06.168 RR 10.53.0.3:0 <- 10.53.0.1:7523 TCP ...
|
|
# 22-Jun-2022 12:09:06.168 RQ 10.53.0.3:56306 -> 10.53.0.2:7523 UDP ...
|
|
# 22-Jun-2022 12:09:06.168 RQ 10.53.0.3:56306 <- 10.53.0.2:7523 UDP ...
|
|
#
|
|
bad_frames = []
|
|
inspected_frames = 0
|
|
addr_regex = r"^10\.53\.0\.[0-9]+:[0-9]{1,5}$"
|
|
for line in output.splitlines():
|
|
_, _, frame_type, addr1, _, addr2, _ = line.split(" ", 6)
|
|
# Only inspect RESOLVER_QUERY and RESOLVER_RESPONSE frames.
|
|
if frame_type not in ("RQ", "RR"):
|
|
continue
|
|
inspected_frames += 1
|
|
if not re.match(addr_regex, addr1) or not re.match(addr_regex, addr2):
|
|
bad_frames.append(line)
|
|
|
|
assert (
|
|
len(bad_frames) == 0
|
|
), "{} out of {} inspected frames contain unexpected addresses:\n\n{}".format(
|
|
len(bad_frames), inspected_frames, "\n".join(bad_frames)
|
|
)
|