Add supybot.utils.net.getAddressFromHostname() and improve Socket dri…

…ver to try successively all IP addresses.

src/drivers/Socket.py

@@ -72,6 +72,7 @@ def __init__(self, irc):
         drivers.IrcDriver.__init__(self, irc)
         drivers.ServersMixin.__init__(self, irc)
         self.conn = None
+        self._attempt = -1
         self.servers = ()
         self.eagains = 0
         self.inbuffer = b''
@@ -239,6 +240,7 @@ def connect(self, **kwargs):
         self.reconnect(reset=False, **kwargs)
 
     def reconnect(self, reset=True):
+        self._attempt += 1
         self.nextReconnectTime = None
         if self.connected:
             drivers.log.reconnect(self.irc.network)
@@ -256,6 +258,8 @@ def reconnect(self, reset=True):
         else:
             drivers.log.debug('Not resetting %s.', self.irc)
         server = self._getNextServer()
+        address = utils.net.getAddressFromHostname(server[0],
+                attempt=self._attempt)
         drivers.log.connect(self.currentServer)
         try:
             socks_proxy = getattr(conf.supybot.networks, self.irc.network) \
@@ -267,7 +271,7 @@ def reconnect(self, reset=True):
                 log.error('Cannot use socks proxy (SocksiPy not installed), '
                         'using direct connection instead.')
                 socks_proxy = ''
-            self.conn = utils.net.getSocket(server[0], socks_proxy)
+            self.conn = utils.net.getSocket(address, socks_proxy)
             vhost = conf.supybot.protocols.irc.vhost()
             self.conn.bind((vhost, 0))
         except socket.error, e:
@@ -281,7 +285,7 @@ def reconnect(self, reset=True):
             if getattr(conf.supybot.networks, self.irc.network).ssl():
                 assert globals().has_key('ssl')
                 self.conn = ssl.wrap_socket(self.conn)
-            self.conn.connect(server)
+            self.conn.connect((address, server[1]))
             def setTimeout():
                 self.conn.settimeout(conf.supybot.drivers.poll())
             conf.supybot.drivers.poll.addCallback(setTimeout)

src/utils/net.py

@@ -80,6 +80,14 @@ def match(self, addr):
         return count >= 1
 emailRe = EmailRe()
 
+def getAddressFromHostname(host, attempt=0):
+    addrinfo = socket.getaddrinfo(host, None)
+    addresses = []
+    for (family, socktype, proto, canonname, sockaddr) in addrinfo:
+        if sockaddr[0] not in addresses:
+            addresses.append(sockaddr[0])
+    return addresses[attempt % len(addresses)]
+
 def getSocket(host, socks_proxy=None):
     """Returns a socket of the correct AF_INET type (v4 or v6) in order to
     communicate with host.