From 8eda09c8cc876b6545ce84b58a9554fc88755555 Mon Sep 17 00:00:00 2001
From: Max Lv <max.c.lv@gmail.com>
Date: Sat, 12 Jan 2013 16:12:40 +0800
Subject: [PATCH] fix a serious bug for uploading

---
 local.c | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/local.c b/local.c
index 010dbe93..f6a9528d 100755
--- a/local.c
+++ b/local.c
@@ -26,13 +26,17 @@
 #endif
 
 #ifndef MSG_NOSIGNAL
-#define MSG_NOSIGNAL MSG_HAVEMORE
+#define MSG_NOSIGNAL SO_NOSIGPIPE
 #endif
 
 #ifndef EAGAIN
 #define EAGAIN EWOULDBLOCK
 #endif
 
+#ifndef EWOULDBLOCK
+#define EWOULDBLOCK EAGAIN
+#endif
+
 #define min(a,b) (((a)<(b))?(a):(b))
 
 static char *_server;
@@ -119,7 +123,7 @@ static void server_recv_cb (EV_P_ ev_io *w, int revents) {
             close_and_free_server(EV_A_ server);
             return;
         } else if(r < 0) {
-            if (errno == EAGAIN) {
+            if (errno == EAGAIN || errno == EWOULDBLOCK) {
                 // no data
                 // continue to wait for recv
                 break;
@@ -136,8 +140,9 @@ static void server_recv_cb (EV_P_ ev_io *w, int revents) {
             encrypt_ctx(remote->buf, r, server->e_ctx);
             int w = send(remote->fd, remote->buf, r, 0);
             if(w == -1) {
-                if (errno == EAGAIN) {
+            if (errno == EAGAIN || errno == EWOULDBLOCK) {
                     // no data, wait for send
+                    remote->buf_len = r;
                     ev_io_stop(EV_A_ &server_recv_ctx->io);
                     ev_io_start(EV_A_ &remote->send_ctx->io);
                     break;
@@ -260,11 +265,10 @@ static void server_send_cb (EV_P_ ev_io *w, int revents) {
         ssize_t r = send(server->fd, server->buf,
                 server->buf_len, 0);
         if (r < 0) {
-            if (errno != EAGAIN) {
+            if (errno == EAGAIN || errno == EWOULDBLOCK) {
                 perror("send");
                 close_and_free_remote(EV_A_ remote);
                 close_and_free_server(EV_A_ server);
-                return;
             }
             return;
         }
@@ -330,7 +334,7 @@ static void remote_recv_cb (EV_P_ ev_io *w, int revents) {
             close_and_free_server(EV_A_ server);
             return;
         } else if(r < 0) {
-            if (errno == EAGAIN) {
+            if (errno == EAGAIN || errno == EWOULDBLOCK) {
                 // no data
                 // continue to wait for recv
                 break;
@@ -344,8 +348,9 @@ static void remote_recv_cb (EV_P_ ev_io *w, int revents) {
         decrypt_ctx(server->buf, r, server->d_ctx);
         int w = send(server->fd, server->buf, r, MSG_NOSIGNAL);
         if(w == -1) {
-            if (errno == EAGAIN) {
+            if (errno == EAGAIN || errno == EWOULDBLOCK) {
                 // no data, wait for send
+                server->buf_len = r;
                 ev_io_stop(EV_A_ &remote_recv_ctx->io);
                 ev_io_start(EV_A_ &server->send_ctx->io);
                 break;
@@ -406,12 +411,11 @@ static void remote_send_cb (EV_P_ ev_io *w, int revents) {
             ssize_t r = send(remote->fd, remote->buf,
                     remote->buf_len, 0);
             if (r < 0) {
-                if (errno != EAGAIN) {
+                if (errno != EAGAIN && errno != EWOULDBLOCK) {
                     perror("send");
                     // close and free
                     close_and_free_remote(EV_A_ remote);
                     close_and_free_server(EV_A_ server);
-                    return;
                 }
                 return;
             }