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

add gevent

Browse source
This commit is contained in:
clowwindy 2012-12-30 13:28:51 +08:00
parent 8c5c40915e
commit 12173a66d1
3 changed files with 25 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

13
server.py
View file

@ -20,6 +20,14 @@
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
import sys
try:
import gevent, gevent.monkey
gevent.monkey.patch_all(dns=gevent.version_info[0]>=1)
except ImportError:
gevent = None
print >>sys.stderr, 'warning: gevent not found, using threading instead'
import socket
import select
@ -27,7 +35,6 @@ import SocketServer
import struct
import string
import hashlib
import sys
import os
import json
import logging
@ -58,13 +65,13 @@ class Socks5Server(SocketServer.StreamRequestHandler):
data = sock.recv(4096)
if data <= 0:
break
if remote.sendall(self.decrypt(data)) is not None:
if remote.send(self.decrypt(data)) <= 0:
break
if remote in r:
data = remote.recv(4096)
if data <= 0:
break
if sock.sendall(self.encrypt(data)) is not None:
if sock.send(self.encrypt(data)) <= 0:
break
finally:
sock.close()