nginx-0.0.1-2003-05-27-16:18:54 import

1 files changed, 79 insertions, 55 deletions

diff --git a/src/os/unix/ngx_recv.c b/src/os/unix/ngx_recv.c
index b2be905f7..98607d0c7 100644
--- a/src/os/unix/ngx_recv.c
+++ b/src/os/unix/ngx_recv.c
@@ -1,94 +1,118 @@
 
 #include <ngx_config.h>
 #include <ngx_core.h>
-#include <ngx_errno.h>
-#include <ngx_log.h>
-#include <ngx_recv.h>
-#include <ngx_connection.h>
 
 
+static int ngx_unix_recv_error(ngx_event_t *rev, ngx_err_t err);
+
+
+#if (HAVE_KQUEUE)
+
 ssize_t ngx_unix_recv(ngx_connection_t *c, char *buf, size_t size)
 {
     ssize_t       n;
-    ngx_err_t     err;
-    ngx_event_t  *ev;
+    ngx_event_t  *rev;
 
-    ev = c->read;
+    rev = c->read;
 
-#if (HAVE_KQUEUE) /* DEBUG */
     if (ngx_event_flags & NGX_HAVE_KQUEUE_EVENT) {
         ngx_log_debug(c->log, "recv: eof:%d, avail:%d, err:%d" _
-                      ev->eof _ ev->available _ ev->error);
+                      rev->eof _ rev->available _ rev->error);
+
+        if (rev->available == 0) {
+            if (rev->eof) {
+                if (rev->error) {
+                    rev->ready = 0;
+                    ngx_set_socket_errno(rev->error);
+                    return ngx_unix_recv_error(rev, rev->error);
+                }
+                return 0;
+
+            } else {
+                return NGX_AGAIN;
+            }
+        }
     }
-#endif
-
-#if (HAVE_KQUEUE)
 
-    if ((ngx_event_flags & NGX_HAVE_KQUEUE_EVENT)
-        && ev->eof && ev->available == 0) {
+    do {
+        n = recv(c->fd, buf, size, 0);
 
-        if (ev->error == 0) {
-            return 0;
-        }
+        ngx_log_debug(c->log, "recv: %d:%d" _ n _ size);
 
-        ngx_set_socket_errno(ev->error);
-        err = ev->error;
-        n = -1;
+        if (n >= 0) {
+            if (ngx_event_flags & NGX_HAVE_KQUEUE_EVENT) {
+                rev->available -= n;
+                if (rev->available == 0) {
+                    rev->ready = 0;
+                }
 
-    } else {
-        n = recv(c->fd, buf, size, 0);
+                return n;
+            }
 
-ngx_log_debug(c->log, "recv: read:%d:%d" _ n _ size);
+            if ((size_t) n < size) {
+                rev->ready = 0;
+            }
 
-        if (n == -1) {
-            err = ngx_socket_errno;
+            return n;
         }
-    }
 
-#else /* not kqueue */
+        rev->ready = 0;
+        n = ngx_unix_recv_error(rev, ngx_socket_errno);
+
+    } while (n == NGX_EINTR);
 
-    n = recv(c->fd, buf, size, 0);
+    return n;
+}
 
-ngx_log_debug(c->log, "recv: read:%d:%d" _ n _ size);
+#else /* ! NAVE_KQUEUE */
 
-    if (n == -1) {
-        err = ngx_socket_errno;
-    }
+ssize_t ngx_unix_recv(ngx_connection_t *c, char *buf, size_t size)
+{
+    ssize_t       n;
+    ngx_event_t  *rev;
 
-#endif
+    rev = c->read;
 
-    if (n == -1) {
-        ev->ready = 0;
+    do {
+        n = recv(c->fd, buf, size, 0);
 
-        if (err == NGX_ECONNRESET && ev->ignore_econnreset) {
-            return 0;
-        }
+        ngx_log_debug(c->log, "recv: %d:%d" _ n _ size);
 
-        if (err == NGX_EAGAIN) {
-            ngx_log_error(NGX_LOG_INFO, c->log, err, "recv() returned EAGAIN");
-            return NGX_AGAIN;
+        if (n >= 0) {
+            if ((size_t) n < size) {
+                rev->ready = 0;
+            }
+            return n;
         }
 
-        ngx_log_error(NGX_LOG_ERR, c->log, err, "recv() failed");
-        return NGX_ERROR;
-    }
+        rev->ready = 0;
+        n = ngx_unix_recv_error(rev, ngx_socket_errno);
 
-#if (HAVE_KQUEUE)
+    } while (n == NGX_EINTR);
+
+    return n;
+}
+
+#endif /* NAVE_KQUEUE */
 
-    if (ngx_event_flags & NGX_HAVE_KQUEUE_EVENT) {
-        ev->available -= n;
-        if (ev->available == 0) {
-            ev->ready = 0;
-        }
 
-        return n;
+static int ngx_unix_recv_error(ngx_event_t *rev, ngx_err_t err)
+{
+    if (err == NGX_ECONNRESET && rev->ignore_econnreset) {
+        return 0;
     }
 
-#endif
+    if (err == NGX_EAGAIN) {
+        ngx_log_error(NGX_LOG_INFO, rev->log, err, "recv() returned EAGAIN");
+        return NGX_AGAIN;
+    }
 
-    if ((size_t) n < size) {
-        ev->ready = 0;
+    if (err == NGX_EINTR) {
+        ngx_log_error(NGX_LOG_INFO, rev->log, err, "recv() returned EINTR");
+        return NGX_EINTR;
     }
 
-    return n;
+    ngx_log_error(NGX_LOG_ERR, rev->log, err, "recv() failed");
+
+    return NGX_ERROR;
 }