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

use sendall()

Browse source
This commit is contained in:
clowwindy 2012-12-30 14:01:41 +08:00
parent 12173a66d1
commit 65444c5b06
3 changed files with 12 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

6
README.md
View file

@ -40,8 +40,12 @@ You can use args to override settings from `config.json`.
You may want to install gevent for better performance.
$ apt-get install python-gevent
Or:
$ sudo easy_install gevent
troubleshooting
---------------

10
local.py
View file

@ -52,7 +52,7 @@ def get_table(key):
class ThreadingTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
pass
allow_reuse_address = True
class Socks5Server(SocketServer.StreamRequestHandler):
@ -65,14 +65,13 @@ class Socks5Server(SocketServer.StreamRequestHandler):
data = sock.recv(4096)
if data <= 0:
break
if remote.send(self.encrypt(data)) <= 0:
break
remote.sendall(self.encrypt(data))
if remote in r:
data = remote.recv(4096)
if data <= 0:
break
if sock.send(self.decrypt(data)) <= 0:
break
sock.sendall(self.decrypt(data))
finally:
sock.close()
remote.close()
@ -162,7 +161,6 @@ if __name__ == '__main__':
decrypt_table = string.maketrans(encrypt_table, string.maketrans('', ''))
try:
server = ThreadingTCPServer(('', PORT), Socks5Server)
server.allow_reuse_address = True
logging.info("starting server at port %d ..." % PORT)
server.serve_forever()
except socket.error, e:

9
server.py
View file

@ -52,7 +52,7 @@ def get_table(key):
class ThreadingTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
pass
allow_reuse_address = True
class Socks5Server(SocketServer.StreamRequestHandler):
@ -65,14 +65,12 @@ class Socks5Server(SocketServer.StreamRequestHandler):
data = sock.recv(4096)
if data <= 0:
break
if remote.send(self.decrypt(data)) <= 0:
break
remote.sendall(self.decrypt(data))
if remote in r:
data = remote.recv(4096)
if data <= 0:
break
if sock.send(self.encrypt(data)) <= 0:
break
sock.send(self.encrypt(data))
finally:
sock.close()
remote.close()
@ -137,7 +135,6 @@ if __name__ == '__main__':
ThreadingTCPServer.address_family = socket.AF_INET6
try:
server = ThreadingTCPServer(('', PORT), Socks5Server)
server.allow_reuse_address = True
logging.info("starting server at port %d ..." % PORT)
server.serve_forever()
except socket.error, e: