vbatts/shadowsocks
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

add async dns

Browse source
This commit is contained in:
clowwindy 2014-06-08 15:58:59 +08:00
parent fbc4906445
commit 71c3759c12
5 changed files with 30 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
shadowsocks/asyncdns.py
View file

@ -244,9 +244,7 @@ class DNSResolver(object):
self._hostname_to_cb = {}
self._cb_to_hostname = {}
self._cache = lru_cache.LRUCache(timeout=300)
self._sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
socket.SOL_UDP)
self._sock.setblocking(False)
self._sock = None
self._parse_config()
def _parse_config(self):
@ -272,7 +270,12 @@ class DNSResolver(object):
self._dns_server = ('8.8.8.8', 53)
def add_to_loop(self, loop):
if self._loop:
raise Exception('already add to loop')
self._loop = loop
self._sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
socket.SOL_UDP)
self._sock.setblocking(False)
loop.add(self._sock, eventloop.POLL_IN)
loop.add_handler(self.handle_events)
@ -362,6 +365,9 @@ class DNSResolver(object):
else:
arr.append(callback)
def close(self):
self._sock.close()
def test():
logging.getLogger('').handlers = []