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

Handle cipher exceptions #783 (#791)

Browse source
This commit is contained in:
Zou Yong 2017-03-15 00:35:17 +08:00 committed by mengskysama
parent 74f8f8cb85
commit 4e21f83bd6
3 changed files with 26 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
shadowsocks/tcprelay.py
View file

@ -641,6 +641,7 @@ class TCPRelayHandler(object):
logging.error(eventloop.get_sock_error(self._remote_sock))
self.destroy()
@shell.exception_handle(self_=True, destroy=True)
def handle_event(self, sock, event):
# handle all events in this handler and dispatch them to methods
if self._stage == STAGE_DESTROYED:

0
shadowsocks/tunnel.py Normal file → Executable file
View file

35
shadowsocks/udprelay.py
View file

@ -170,14 +170,16 @@ class UDPRelay(object):
else:
data = data[3:]
else:
data, key, iv = cryptor.decrypt_all(self._password,
self._method,
data)
# decrypt data
try:
data, key, iv = cryptor.decrypt_all(self._password,
self._method,
data)
except Exception:
logging.debug('UDP handle_server: decrypt data failed')
return
if not data:
logging.debug(
'UDP handle_server: data is empty after decrypt'
)
logging.debug('UDP handle_server: data is empty after decrypt')
return
header_result = parse_header(data)
if header_result is None:
@ -238,7 +240,11 @@ class UDPRelay(object):
# spec https://shadowsocks.org/en/spec/one-time-auth.html
if self._ota_enable_session:
data = self._ota_chunk_data_gen(key, iv, data)
data = cryptor.encrypt_all_m(key, iv, m, self._method, data)
try:
data = cryptor.encrypt_all_m(key, iv, m, self._method, data)
except Exception:
logging.debug("UDP handle_server: encrypt data failed")
return
if not data:
return
else:
@ -267,12 +273,21 @@ class UDPRelay(object):
# drop
return
data = pack_addr(r_addr[0]) + struct.pack('>H', r_addr[1]) + data
response = cryptor.encrypt_all(self._password, self._method, data)
try:
response = cryptor.encrypt_all(self._password,
self._method, data)
except Exception:
logging.debug("UDP handle_client: encrypt data failed")
return
if not response:
return
else:
data, key, iv = cryptor.decrypt_all(self._password,
self._method, data)
try:
data, key, iv = cryptor.decrypt_all(self._password,
self._method, data)
except Exception:
logging.debug('UDP handle_client: decrypt data failed')
return
if not data:
return
header_result = parse_header(data)