Replaced IPv6 resolver.

1 files changed, 12 insertions, 2 deletions

diff --git a/ipv6.py b/ipv6.py
index eb6ce2e..12562c9 100644
--- a/ipv6.py
+++ b/ipv6.py
@@ -5,6 +5,9 @@ from twisted.internet import default
 from twisted.internet import protocol
 from twisted.internet import reactor
 
+from twisted.names.client import Resolver
+resolver = Resolver('/etc/resolv.conf')
+
 class IPv6Address(object):
     def __init__(self, type, host, port, flowInfo, scope):
         self.type = type
@@ -38,10 +41,17 @@ class Client(tcp.Client):
         if isIPv6Address(self.addr[0]):
             self._setRealAddress(self.addr[0])
         else:
-            reactor.resolve(self.addr[0]).addCallbacks(
-                self._setRealAddress, self.failIfNotConnected
+            resolver.lookupIPV6Address(self.addr[0]).addCallbacks(
+                self.resolvedAddress, self.failIfNotConnected
             )
 
+    def resolvedAddress(self, addr):
+        aaaa = addr[0][-1].payload
+        if aaaa.fancybasename != 'AAAA':
+            self.failIfNotConnected(Exception('Unable to resolve IPv6 address.'))
+            return
+        self._setRealAddress(aaaa._address)
+
     def getHost(self):
         return IPv6Address('TCP', *self.socket.getsockname())