nginx/src/os/unix/ngx_aio_write.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 #include <ngx_aio.h>
  11
  12 #if (NGX_HAVE_KQUEUE)
  13 #include <ngx_kqueue_module.h>
  14 #endif
  15
  16
  17 /*
  18  * the ready data requires 3 syscalls:
  19  *     aio_write(), aio_error(), aio_return()
  20  * the non-ready data requires 4 (kqueue) or 5 syscalls:
  21  *     aio_write(), aio_error(), notifiction, aio_error(), aio_return()
  22  *                               timeout, aio_cancel(), aio_error()
  23  */
  24
  25 ssize_t
  26 ngx_aio_write(ngx_connection_t *c, u_char *buf, size_t size)
  27 {
  28     int           n;
  29     ngx_event_t  *wev;
  30
  31     wev = c->write;
  32
  33     if (!wev->ready) {
  34         return NGX_AGAIN;
  35     }
  36
  37     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, wev->log, 0,
  38                    "aio: wev->complete: %d", wev->complete);
  39
  40     if (!wev->complete) {
  41         ngx_memzero(&wev->aiocb, sizeof(struct aiocb));
  42
  43         wev->aiocb.aio_fildes = c->fd;
  44         wev->aiocb.aio_buf = buf;
  45         wev->aiocb.aio_nbytes = size;
  46
  47 #if (NGX_HAVE_KQUEUE)
  48         wev->aiocb.aio_sigevent.sigev_notify_kqueue = ngx_kqueue;
  49         wev->aiocb.aio_sigevent.sigev_notify = SIGEV_KEVENT;
  50         wev->aiocb.aio_sigevent.sigev_value.sigval_ptr = wev;
  51 #endif
  52
  53         if (aio_write(&wev->aiocb) == -1) {
  54             ngx_log_error(NGX_LOG_CRIT, wev->log, ngx_errno,
  55                           "aio_write() failed");
  56             return NGX_ERROR;
  57         }
  58
  59         ngx_log_debug0(NGX_LOG_DEBUG_EVENT, wev->log, 0, "aio_write: OK");
  60
  61         wev->active = 1;
  62         wev->ready = 0;
  63     }
  64
  65     wev->complete = 0;
  66
  67     n = aio_error(&wev->aiocb);
  68     if (n == -1) {
  69         ngx_log_error(NGX_LOG_CRIT, wev->log, ngx_errno, "aio_error() failed");
  70         wev->error = 1;
  71         return NGX_ERROR;
  72     }
  73
  74     if (n != 0) {
  75         if (n == NGX_EINPROGRESS) {
  76             if (wev->ready) {
  77                 ngx_log_error(NGX_LOG_ALERT, wev->log, n,
  78                               "aio_write() still in progress");
  79                 wev->ready = 0;
  80             }
  81             return NGX_AGAIN;
  82         }
  83
  84         ngx_log_error(NGX_LOG_CRIT, wev->log, n, "aio_write() failed");
  85         wev->error = 1;
  86         wev->ready = 0;
  87
  88 #if 1
  89         n = aio_return(&wev->aiocb);
  90         if (n == -1) {
  91             ngx_log_error(NGX_LOG_ALERT, wev->log, ngx_errno,
  92                           "aio_return() failed");
  93         }
  94
  95         ngx_log_error(NGX_LOG_CRIT, wev->log, n, "aio_return() %d", n);
  96 #endif
  97
  98         return NGX_ERROR;
  99     }
 100
 101     n = aio_return(&wev->aiocb);
 102     if (n == -1) {
 103         ngx_log_error(NGX_LOG_ALERT, wev->log, ngx_errno,
 104                       "aio_return() failed");
 105
 106         wev->error = 1;
 107         wev->ready = 0;
 108         return NGX_ERROR;
 109     }
 110
 111
 112     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, wev->log, 0, "aio_write: %d", n);
 113
 114     wev->active = 0;
 115     wev->ready = 1;
 116
 117     return n;
 118 }