Alawn/ShadowsockR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

removefd

Browse Source
This commit is contained in:
破娃酱
2017-05-10 20:10:23 +08:00
parent a1f56d1233
commit 5e14119b41
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
shadowsocks/eventloop.py
+4
View File
@@ -176,6 +176,10 @@ class EventLoop(object):
del self._fdmap[fd]
self._impl.unregister(fd)
def removefd(self, fd):
del self._fdmap[fd]
self._impl.unregister(fd)
def add_periodic(self, callback):
self._periodic_callbacks.append(callback)
shadowsocks/tcprelay.py
+5 -5
View File
@@ -1061,7 +1061,7 @@ class TCPRelayHandler(object):
else:
logging.warn('unknown socket from %s:%d' % (self._client_address[0], self._client_address[1]))
try:
self._loop.remove(sock)
self._loop.removefd(fd)
except Exception as e:
shell.print_exception(e)
try:
@@ -1100,7 +1100,7 @@ class TCPRelayHandler(object):
if self._remote_sock:
logging.debug('destroying remote')
try:
self._loop.remove(self._remote_sock)
self._loop.removefd(self._remote_sock_fd)
except Exception as e:
shell.print_exception(e)
try:
@@ -1113,7 +1113,7 @@ class TCPRelayHandler(object):
if self._remote_sock_v6:
logging.debug('destroying remote_v6')
try:
self._loop.remove(self._remote_sock_v6)
self._loop.removefd(self._remotev6_sock_fd)
except Exception as e:
shell.print_exception(e)
try:
@@ -1126,7 +1126,7 @@ class TCPRelayHandler(object):
if self._local_sock:
logging.debug('destroying local')
try:
self._loop.remove(self._local_sock)
self._loop.removefd(self._local_sock_fd)
except Exception as e:
shell.print_exception(e)
try:
@@ -1403,7 +1403,7 @@ class TCPRelay(object):
logging.warn('unknown fd')
handle = True
try:
self._eventloop.remove(sock)
self._eventloop.removefd(fd)
except Exception as e:
shell.print_exception(e)
sock.close()