From 004e9292f4ee2fdadd2511154c2042aba5978382 Mon Sep 17 00:00:00 2001
From: clowwindy <clowwindy42@gmail.com>
Date: Sun, 30 Dec 2012 14:12:15 +0800
Subject: [PATCH] implement send_all myself

---
 local.py  | 13 +++++++++++--
 server.py | 14 ++++++++++++--
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/local.py b/local.py
index d0f9452..aaa4a4d 100755
--- a/local.py
+++ b/local.py
@@ -50,6 +50,15 @@ def get_table(key):
         table.sort(lambda x, y: int(a % (ord(x) + i) - a % (ord(y) + i)))
     return table
 
+def send_all(sock, data):
+    bytes_sent = 0
+    while True:
+        r = sock.send(data[bytes_sent:])
+        if r < 0:
+            return r
+        bytes_sent += r
+        if bytes_sent == len(data):
+            return bytes_sent
 
 class ThreadingTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
     allow_reuse_address = True
@@ -65,13 +74,13 @@ class Socks5Server(SocketServer.StreamRequestHandler):
                     data = sock.recv(4096)
                     if data <= 0:
                         break
-                    remote.sendall(self.encrypt(data))
+                    send_all(remote, self.encrypt(data))
 
                 if remote in r:
                     data = remote.recv(4096)
                     if data <= 0:
                         break
-                    sock.sendall(self.decrypt(data))
+                    send_all(sock, self.decrypt(data))
         finally:
             sock.close()
             remote.close()
diff --git a/server.py b/server.py
index 552599e..b608c24 100755
--- a/server.py
+++ b/server.py
@@ -50,6 +50,16 @@ def get_table(key):
         table.sort(lambda x, y: int(a % (ord(x) + i) - a % (ord(y) + i)))
     return table
 
+def send_all(sock, data):
+    bytes_sent = 0
+    while True:
+        r = sock.send(data[bytes_sent:])
+        if r < 0:
+            return r
+        bytes_sent += r
+        if bytes_sent == len(data):
+            return bytes_sent
+
 
 class ThreadingTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
     allow_reuse_address = True
@@ -65,12 +75,12 @@ class Socks5Server(SocketServer.StreamRequestHandler):
                     data = sock.recv(4096)
                     if data <= 0:
                         break
-                    remote.sendall(self.decrypt(data))
+                    send_all(remote, self.decrypt(data))
                 if remote in r:
                     data = remote.recv(4096)
                     if data <= 0:
                         break
-                    sock.send(self.encrypt(data))
+                    send_all(sock, self.encrypt(data))
         finally:
             sock.close()
             remote.close()