nginx/src/os/unix/ngx_aio_read.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_read(ngx_connection_t *c, u_char *buf, size_t size)
  27 {
  28     int           n;
  29     ngx_event_t  *rev;
  30
  31     rev = c->read;
  32
  33     if (!rev->ready) {
  34         ngx_log_error(NGX_LOG_ALERT, c->log, 0, "second aio post");
  35         return NGX_AGAIN;
  36     }
  37
  38     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0,
  39                    "rev->complete: %d", rev->complete);
  40     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0,
  41                    "aio size: %d", size);
  42
  43     if (!rev->complete) {
  44         ngx_memzero(&rev->aiocb, sizeof(struct aiocb));
  45
  46         rev->aiocb.aio_fildes = c->fd;
  47         rev->aiocb.aio_buf = buf;
  48         rev->aiocb.aio_nbytes = size;
  49
  50 #if (NGX_HAVE_KQUEUE)
  51         rev->aiocb.aio_sigevent.sigev_notify_kqueue = ngx_kqueue;
  52         rev->aiocb.aio_sigevent.sigev_notify = SIGEV_KEVENT;
  53         rev->aiocb.aio_sigevent.sigev_value.sigval_ptr = rev;
  54 #endif
  55
  56         if (aio_read(&rev->aiocb) == -1) {
  57             ngx_log_error(NGX_LOG_CRIT, rev->log, ngx_errno,
  58                           "aio_read() failed");
  59             rev->error = 1;
  60             return NGX_ERROR;
  61         }
  62
  63         ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0,
  64                        "aio_read: #%d OK", c->fd);
  65
  66         rev->active = 1;
  67         rev->ready = 0;
  68     }
  69
  70     rev->complete = 0;
  71
  72     n = aio_error(&rev->aiocb);
  73     if (n == -1) {
  74         ngx_log_error(NGX_LOG_ALERT, c->log, ngx_errno, "aio_error() failed");
  75         rev->error = 1;
  76         return NGX_ERROR;
  77     }
  78
  79     if (n != 0) {
  80         if (n == NGX_EINPROGRESS) {
  81             if (rev->ready) {
  82                 ngx_log_error(NGX_LOG_ALERT, c->log, n,
  83                               "aio_read() still in progress");
  84                 rev->ready = 0;
  85             }
  86             return NGX_AGAIN;
  87         }
  88
  89         ngx_log_error(NGX_LOG_CRIT, c->log, n, "aio_read() failed");
  90         rev->error = 1;
  91         rev->ready = 0;
  92         return NGX_ERROR;
  93     }
  94
  95     n = aio_return(&rev->aiocb);
  96     if (n == -1) {
  97         ngx_log_error(NGX_LOG_ALERT, c->log, ngx_errno,
  98                       "aio_return() failed");
  99
 100         rev->error = 1;
 101         rev->ready = 0;
 102         return NGX_ERROR;
 103     }
 104
 105     ngx_log_debug2(NGX_LOG_DEBUG_EVENT, rev->log, 0,
 106                    "aio_read: #%d %d", c->fd, n);
 107
 108     if (n == 0) {
 109         rev->eof = 1;
 110         rev->ready = 0;
 111     } else {
 112         rev->ready = 1;
 113     }
 114
 115     rev->active = 0;
 116
 117     return n;
 118 }