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

fix some exceptions

Browse source
This commit is contained in:
breakwa11 2015-08-15 12:47:14 +08:00
parent 9defba2d36
commit 928c26e4f2
2 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
shadowsocks/encrypt.py
View file

@ -21,6 +21,7 @@ import os
import sys
import hashlib
import logging
import random
from shadowsocks import common
from shadowsocks.crypto import rc4_md5, openssl, sodium, table
@ -34,7 +35,10 @@ method_supported.update(table.ciphers)
def random_string(length):
return os.urandom(length)
try:
return os.urandom(length)
except (AttributeError, NotImplementedError):
return ''.join(chr(random.randrange(255)) for _ in range(length))
cached_keys = {}

7
shadowsocks/udprelay.py
View file

@ -1036,8 +1036,11 @@ class UDPRelay(object):
self.write_to_server_socket(data_to_send, r_addr)
elif data[0] > CMD_CONNECT_REMOTE and data[0] <= CMD_DISCONNECT:
if data[1] in self._reqid_to_hd:
self.update_activity(self._reqid_to_hd[data[1]])
self._reqid_to_hd[data[1]].handle_client(r_addr, *data)
if type(self._reqid_to_hd[data[1]]) is tuple:
pass
else:
self.update_activity(self._reqid_to_hd[data[1]])
self._reqid_to_hd[data[1]].handle_client(r_addr, *data)
else:
# disconnect
rsp_data = self._pack_rsp_data(CMD_DISCONNECT, data[1], RSP_STATE_EMPTY)