nginx/src/os/unix/ngx_udp_recv.c

   1
   2 /*
   3  * Copyright (C) Igor Sysoev
   4  */
   5
   6
   7 #include <ngx_config.h>
   8 #include <ngx_core.h>
   9 #include <ngx_event.h>
  10
  11
  12 #if (NGX_HAVE_KQUEUE)
  13
  14 ssize_t
  15 ngx_udp_unix_recv(ngx_connection_t *c, u_char *buf, size_t size)
  16 {
  17     ssize_t       n;
  18     ngx_err_t     err;
  19     ngx_event_t  *rev;
  20
  21     rev = c->read;
  22
  23     do {
  24         n = recv(c->fd, buf, size, 0);
  25
  26         ngx_log_debug3(NGX_LOG_DEBUG_EVENT, c->log, 0,
  27                        "recv: fd:%d %d of %d", c->fd, n, size);
  28
  29         if (n >= 0) {
  30             if (ngx_event_flags & NGX_USE_KQUEUE_EVENT) {
  31                 rev->available -= n;
  32
  33                 /*
  34                  * rev->available may be negative here because some additional
  35                  * bytes may be received between kevent() and recv()
  36                  */
  37
  38                 if (rev->available <= 0) {
  39                     rev->ready = 0;
  40                     rev->available = 0;
  41                 }
  42             }
  43
  44             return n;
  45         }
  46
  47         err = ngx_socket_errno;
  48
  49         if (err == NGX_EAGAIN || err == NGX_EINTR) {
  50             ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
  51                            "recv() not ready");
  52             n = NGX_AGAIN;
  53
  54         } else {
  55             n = ngx_connection_error(c, err, "recv() failed");
  56             break;
  57         }
  58
  59     } while (err == NGX_EINTR);
  60
  61     rev->ready = 0;
  62
  63     if (n == NGX_ERROR){
  64         rev->error = 1;
  65     }
  66
  67     return n;
  68 }
  69
  70 #else /* ! NGX_HAVE_KQUEUE */
  71
  72 ssize_t
  73 ngx_udp_unix_recv(ngx_connection_t *c, u_char *buf, size_t size)
  74 {
  75     ssize_t       n;
  76     ngx_err_t     err;
  77     ngx_event_t  *rev;
  78
  79     rev = c->read;
  80
  81     do {
  82         n = recv(c->fd, buf, size, 0);
  83
  84         ngx_log_debug3(NGX_LOG_DEBUG_EVENT, c->log, 0,
  85                        "recv: fd:%d %d of %d", c->fd, n, size);
  86
  87         if (n >= 0) {
  88             return n;
  89         }
  90
  91         err = ngx_socket_errno;
  92
  93         if (err == NGX_EAGAIN || err == NGX_EINTR) {
  94             ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
  95                            "recv() not ready");
  96             n = NGX_AGAIN;
  97
  98         } else {
  99             n = ngx_connection_error(c, err, "recv() failed");
 100             break;
 101         }
 102
 103     } while (err == NGX_EINTR);
 104
 105     rev->ready = 0;
 106
 107     if (n == NGX_ERROR){
 108         rev->error = 1;
 109     }
 110
 111     return n;
 112 }
 113
 114 #endif /* NGX_HAVE_KQUEUE */