Mikaela
/
Limnoria
mirror of https://github.com/Mikaela/Limnoria.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add support for SRV records in domain name resolution.

This commit is contained in:
Valentin Lorentz 2014-03-01 09:22:14 +01:00
parent 108b0de0d1
commit dcb25db6e5
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/drivers/Socket.py
View File

@ -290,9 +290,11 @@ class SocketDriver(drivers.IrcDriver, drivers.ServersMixin):
drivers.log.connectError(self.currentServer, e)
self.scheduleReconnect()
return
port = server[1]
drivers.log.connect(self.currentServer)
try:
self.conn = utils.net.getSocket(address, socks_proxy)
self.conn = utils.net.getSocket(address, port=port,
socks_proxy=socks_proxy)
except socket.error as e:
drivers.log.connectError(self.currentServer, e)
self.scheduleReconnect()

8
src/utils/net.py
View File

@ -40,20 +40,20 @@ from .web import _ipAddr, _domain
emailRe = re.compile(r"^(\w&.+-]+!)*[\w&.+-]+@(%s|%s)$" % (_domain, _ipAddr),
re.I)
def getAddressFromHostname(host, attempt=0):
addrinfo = socket.getaddrinfo(host, None)
def getAddressFromHostname(host, port=None, attempt=0):
addrinfo = socket.getaddrinfo(host, port)
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, vhost=None, vhostv6=None):
def getSocket(host, port=None, socks_proxy=None, vhost=None, vhostv6=None):
"""Returns a socket of the correct AF_INET type (v4 or v6) in order to
communicate with host.
"""
if not socks_proxy:
addrinfo = socket.getaddrinfo(host, None)
addrinfo = socket.getaddrinfo(host, port)
host = addrinfo[0][4][0]
if socks_proxy:
import socks