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

fix UDPRelay _forbidden_portset

Browse source
This commit is contained in:
BreakWa11 2016-06-16 01:35:18 +08:00
parent 2f8517190c
commit 8556fd3c18
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
shadowsocks/udprelay.py
View file

@ -926,6 +926,10 @@ class UDPRelay(object):
self._forbidden_iplist = config['forbidden_ip'] self._forbidden_iplist = config['forbidden_ip']
else: else:
self._forbidden_iplist = None self._forbidden_iplist = None
if 'forbidden_port' in config:
self._forbidden_portset = config['forbidden_port']
else:
self._forbidden_portset = None
addrs = socket.getaddrinfo(self._listen_addr, self._listen_port, 0, addrs = socket.getaddrinfo(self._listen_addr, self._listen_port, 0,
socket.SOCK_DGRAM, socket.SOL_UDP) socket.SOCK_DGRAM, socket.SOL_UDP)
@ -1083,8 +1087,10 @@ class UDPRelay(object):
return return
if self._forbidden_portset: if self._forbidden_portset:
if sa[1] in self._forbidden_portset: if sa[1] in self._forbidden_portset:
raise Exception('Port %d is in forbidden list, reject' % logging.debug('Port %d is in forbidden list, reject' %
sa[1]) sa[1])
# drop
return
client = socket.socket(af, socktype, proto) client = socket.socket(af, socktype, proto)
client.setblocking(False) client.setblocking(False)
is_dns = False is_dns = False