[dnssd] add support for service subtypes in DNS-SD server and client (#6773)

This commit updates `Dns::ServiceDiscovery::Server` to add support for
browsing a service subtype. In particular, it updates the code which
breaks a DNS name into its components such that it can correctly
parse and accept a service subtype name. It also updates how the
service name compression is done to handle the subtype names.

This commit also updates `Dns::Client` to remove the extra check of
service name (which is already verified) when trying to find a PTR
record for a given service instance name. This change enables browse
for service subtype (parsing the browse response).

This commit also updates and enhances the `test_dnssd.py` script
adding test-cases for browsing for service subtype.
This commit is contained in:
Abtin Keshavarzian
2021-07-02 11:42:55 -07:00
committed by GitHub
parent 0faa3fd4fe
commit 48b129e74b
5 changed files with 142 additions and 52 deletions
+97 -36
View File
@@ -32,20 +32,27 @@ import unittest
import thread_cert
# Test description:
#
# This test verifies DNS-SD server and DNS client behavior
# (browsing for services and/or subtype services, resolving an
# address, or resolving a service). It also indirectly covers the SRP
# client and server behavior and the interactions of DNS-SD server
# with SRP server).
#
# Topology:
# Four nodes, leader acting as SRP and DNS-SD servers, with 3 router
# nodes acting as SRP and DNS clients.
#
SERVER = 1
CLIENT1 = 2
CLIENT2 = 3
CLIENT3 = 4
DOMAIN = 'default.service.arpa.'
SERVICE = '_ipps._tcp'
#
# Topology:
# LEADER -- CLIENT1
# |
# CLIENT2
#
class TestDnssd(thread_cert.TestCase):
SUPPORT_NCP = False
@@ -61,39 +68,57 @@ class TestDnssd(thread_cert.TestCase):
CLIENT2: {
'mode': 'rdn',
},
CLIENT3: {
'mode': 'rdn',
}
}
def test(self):
self.nodes[SERVER].start()
server = self.nodes[SERVER]
client1 = self.nodes[CLIENT1]
client2 = self.nodes[CLIENT2]
client3 = self.nodes[CLIENT3]
#---------------------------------------------------------------
# Start the server & client devices.
server.start()
self.simulator.go(5)
self.assertEqual(self.nodes[SERVER].get_state(), 'leader')
self.nodes[SERVER].srp_server_set_enabled(True)
self.assertEqual(server.get_state(), 'leader')
server.srp_server_set_enabled(True)
self.nodes[CLIENT1].start()
client1.start()
client2.start()
client3.start()
self.simulator.go(5)
self.assertEqual(self.nodes[CLIENT1].get_state(), 'router')
self.assertEqual(client1.get_state(), 'router')
self.assertEqual(client2.get_state(), 'router')
self.assertEqual(client3.get_state(), 'router')
self.nodes[CLIENT2].start()
self.simulator.go(5)
self.assertEqual(self.nodes[CLIENT1].get_state(), 'router')
#---------------------------------------------------------------
# Register services on clients
client1_addrs = [self.nodes[CLIENT1].get_mleid(), self.nodes[CLIENT1].get_rloc()]
self._config_srp_client_services(CLIENT1, 'ins1', 'host1', 11111, 1, 1, client1_addrs)
client1_addrs = [client1.get_mleid(), client1.get_rloc()]
client2_addrs = [client2.get_mleid(), client2.get_rloc()]
client3_addrs = [client3.get_mleid(), client2.get_rloc()]
client2_addrs = [self.nodes[CLIENT2].get_mleid(), self.nodes[CLIENT2].get_rloc()]
self._config_srp_client_services(CLIENT2, 'ins2', 'host2', 22222, 2, 2, client2_addrs)
self._config_srp_client_services(client1, server, 'ins1', 'host1', 11111, 1, 1, client1_addrs, ",_s1,_s2")
self._config_srp_client_services(client2, server, 'ins2', 'host2', 22222, 2, 2, client2_addrs)
self._config_srp_client_services(client3, server, 'ins3', 'host3', 33333, 3, 3, client3_addrs, ",_s1")
# Test AAAA query using DNS client
answers = self.nodes[CLIENT1].dns_resolve(f"host1.{DOMAIN}", self.nodes[SERVER].get_mleid(), 53)
#---------------------------------------------------------------
# Resolve address (AAAA records)
answers = client1.dns_resolve(f"host1.{DOMAIN}", server.get_mleid(), 53)
self.assertEqual(set(ipaddress.IPv6Address(ip) for ip, _ in answers),
set(map(ipaddress.IPv6Address, client1_addrs)))
answers = self.nodes[CLIENT1].dns_resolve(f"host2.{DOMAIN}", self.nodes[SERVER].get_mleid(), 53)
answers = client1.dns_resolve(f"host2.{DOMAIN}", server.get_mleid(), 53)
self.assertEqual(set(ipaddress.IPv6Address(ip) for ip, _ in answers),
set(map(ipaddress.IPv6Address, client2_addrs)))
service_instances = self.nodes[CLIENT1].dns_browse(f'{SERVICE}.{DOMAIN}', self.nodes[SERVER].get_mleid(), 53)
self.assertEqual({'ins1', 'ins2'}, set(service_instances.keys()), service_instances)
#---------------------------------------------------------------
# Browsing for services
instance1_verify_info = {
'port': 11111,
@@ -119,15 +144,42 @@ class TestDnssd(thread_cert.TestCase):
'aaaa_ttl': lambda x: x > 0,
}
instance3_verify_info = {
'port': 33333,
'priority': 3,
'weight': 3,
'host': 'host3.default.service.arpa.',
'address': client3_addrs,
'txt_data': '',
'srv_ttl': lambda x: x > 0,
'txt_ttl': lambda x: x > 0,
'aaaa_ttl': lambda x: x > 0,
}
# Browse for main service
service_instances = client1.dns_browse(f'{SERVICE}.{DOMAIN}', server.get_mleid(), 53)
self.assertEqual({'ins1', 'ins2', 'ins3'}, set(service_instances.keys()))
self._assert_service_instance_equal(service_instances['ins1'], instance1_verify_info)
self._assert_service_instance_equal(service_instances['ins2'], instance2_verify_info)
self._assert_service_instance_equal(service_instances['ins3'], instance3_verify_info)
service_instance = self.nodes[CLIENT1].dns_resolve_service('ins1', f'{SERVICE}.{DOMAIN}',
self.nodes[SERVER].get_mleid(), 53)
# Browse for service sub-type _s1.
service_instances = client1.dns_browse(f'_s1._sub.{SERVICE}.{DOMAIN}', server.get_mleid(), 53)
self.assertEqual({'ins1', 'ins3'}, set(service_instances.keys()))
self._assert_service_instance_equal(service_instances['ins1'], instance1_verify_info)
# Browse for service sub-type _s2.
service_instances = client1.dns_browse(f'_s2._sub.{SERVICE}.{DOMAIN}', server.get_mleid(), 53)
self.assertEqual({'ins1'}, set(service_instances.keys()))
self._assert_service_instance_equal(service_instances['ins1'], instance1_verify_info)
#---------------------------------------------------------------
# Resolve service
service_instance = client1.dns_resolve_service('ins1', f'{SERVICE}.{DOMAIN}', server.get_mleid(), 53)
self._assert_service_instance_equal(service_instance, instance1_verify_info)
service_instance = self.nodes[CLIENT1].dns_resolve_service('ins2', f'{SERVICE}.{DOMAIN}',
self.nodes[SERVER].get_mleid(), 53)
service_instance = client1.dns_resolve_service('ins2', f'{SERVICE}.{DOMAIN}', server.get_mleid(), 53)
self._assert_service_instance_equal(service_instance, instance2_verify_info)
def _assert_service_instance_equal(self, instance, info):
@@ -147,17 +199,26 @@ class TestDnssd(thread_cert.TestCase):
self.assertTrue(check_ttl(instance[ttl_f]), instance)
def _config_srp_client_services(self, client, instancename, hostname, port, priority, weight, addrs):
self.nodes[client].netdata_show()
srp_server_port = self.nodes[client].get_srp_server_port()
def _config_srp_client_services(self,
client,
server,
instancename,
hostname,
port,
priority,
weight,
addrs,
subtypes=''):
client.netdata_show()
srp_server_port = client.get_srp_server_port()
self.nodes[client].srp_client_start(self.nodes[SERVER].get_mleid(), srp_server_port)
self.nodes[client].srp_client_set_host_name(hostname)
self.nodes[client].srp_client_set_host_address(*addrs)
self.nodes[client].srp_client_add_service(instancename, SERVICE, port, priority, weight)
client.srp_client_start(server.get_mleid(), srp_server_port)
client.srp_client_set_host_name(hostname)
client.srp_client_set_host_address(*addrs)
client.srp_client_add_service(instancename, SERVICE + subtypes, port, priority, weight)
self.simulator.go(5)
self.assertEqual(self.nodes[client].srp_client_get_host_state(), 'Registered')
self.assertEqual(client.srp_client_get_host_state(), 'Registered')
if __name__ == '__main__':