src/os/unix/ngx_writev_chain.c - nginx-1.7.10

Functions defined

Source code


  1. /*
  2. * Copyright (C) Igor Sysoev
  3. * Copyright (C) Nginx, Inc.
  4. */


  5. #include <ngx_config.h>
  6. #include <ngx_core.h>
  7. #include <ngx_event.h>


  8. ngx_chain_t *
  9. ngx_writev_chain(ngx_connection_t *c, ngx_chain_t *in, off_t limit)
  10. {
  11.     ssize_t        n, sent;
  12.     off_t          send, prev_send;
  13.     ngx_chain_t   *cl;
  14.     ngx_event_t   *wev;
  15.     ngx_iovec_t    vec;
  16.     struct iovec   iovs[NGX_IOVS_PREALLOCATE];

  17.     wev = c->write;

  18.     if (!wev->ready) {
  19.         return in;
  20.     }

  21. #if (NGX_HAVE_KQUEUE)

  22.     if ((ngx_event_flags & NGX_USE_KQUEUE_EVENT) && wev->pending_eof) {
  23.         (void) ngx_connection_error(c, wev->kq_errno,
  24.                                "kevent() reported about an closed connection");
  25.         wev->error = 1;
  26.         return NGX_CHAIN_ERROR;
  27.     }

  28. #endif

  29.     /* the maximum limit size is the maximum size_t value - the page size */

  30.     if (limit == 0 || limit > (off_t) (NGX_MAX_SIZE_T_VALUE - ngx_pagesize)) {
  31.         limit = NGX_MAX_SIZE_T_VALUE - ngx_pagesize;
  32.     }

  33.     send = 0;

  34.     vec.iovs = iovs;
  35.     vec.nalloc = NGX_IOVS_PREALLOCATE;

  36.     for ( ;; ) {
  37.         prev_send = send;

  38.         /* create the iovec and coalesce the neighbouring bufs */

  39.         cl = ngx_output_chain_to_iovec(&vec, in, limit - send, c->log);

  40.         if (cl == NGX_CHAIN_ERROR) {
  41.             return NGX_CHAIN_ERROR;
  42.         }

  43.         if (cl && cl->buf->in_file) {
  44.             ngx_log_error(NGX_LOG_ALERT, c->log, 0,
  45.                           "file buf in writev "
  46.                           "t:%d r:%d f:%d %p %p-%p %p %O-%O",
  47.                           cl->buf->temporary,
  48.                           cl->buf->recycled,
  49.                           cl->buf->in_file,
  50.                           cl->buf->start,
  51.                           cl->buf->pos,
  52.                           cl->buf->last,
  53.                           cl->buf->file,
  54.                           cl->buf->file_pos,
  55.                           cl->buf->file_last);

  56.             ngx_debug_point();

  57.             return NGX_CHAIN_ERROR;
  58.         }

  59.         send += vec.size;

  60.         n = ngx_writev(c, &vec);

  61.         if (n == NGX_ERROR) {
  62.             return NGX_CHAIN_ERROR;
  63.         }

  64.         sent = (n == NGX_AGAIN) ? 0 : n;

  65.         c->sent += sent;

  66.         in = ngx_chain_update_sent(in, sent);

  67.         if (send - prev_send != sent) {
  68.             wev->ready = 0;
  69.             return in;
  70.         }

  71.         if (send >= limit || in == NULL) {
  72.             return in;
  73.         }
  74.     }
  75. }


  76. ngx_chain_t *
  77. ngx_output_chain_to_iovec(ngx_iovec_t *vec, ngx_chain_t *in, size_t limit,
  78.     ngx_log_t *log)
  79. {
  80.     size_t         total, size;
  81.     u_char        *prev;
  82.     ngx_uint_t     n;
  83.     struct iovec  *iov;

  84.     iov = NULL;
  85.     prev = NULL;
  86.     total = 0;
  87.     n = 0;

  88.     for ( /* void */ ; in && total < limit; in = in->next) {

  89.         if (ngx_buf_special(in->buf)) {
  90.             continue;
  91.         }

  92.         if (in->buf->in_file) {
  93.             break;
  94.         }

  95.         if (!ngx_buf_in_memory(in->buf)) {
  96.             ngx_log_error(NGX_LOG_ALERT, log, 0,
  97.                           "bad buf in output chain "
  98.                           "t:%d r:%d f:%d %p %p-%p %p %O-%O",
  99.                           in->buf->temporary,
  100.                           in->buf->recycled,
  101.                           in->buf->in_file,
  102.                           in->buf->start,
  103.                           in->buf->pos,
  104.                           in->buf->last,
  105.                           in->buf->file,
  106.                           in->buf->file_pos,
  107.                           in->buf->file_last);

  108.             ngx_debug_point();

  109.             return NGX_CHAIN_ERROR;
  110.         }

  111.         size = in->buf->last - in->buf->pos;

  112.         if (size > limit - total) {
  113.             size = limit - total;
  114.         }

  115.         if (prev == in->buf->pos) {
  116.             iov->iov_len += size;

  117.         } else {
  118.             if (n == vec->nalloc) {
  119.                 break;
  120.             }

  121.             iov = &vec->iovs[n++];

  122.             iov->iov_base = (void *) in->buf->pos;
  123.             iov->iov_len = size;
  124.         }

  125.         prev = in->buf->pos + size;
  126.         total += size;
  127.     }

  128.     vec->count = n;
  129.     vec->size = total;

  130.     return in;
  131. }


  132. ssize_t
  133. ngx_writev(ngx_connection_t *c, ngx_iovec_t *vec)
  134. {
  135.     ssize_t    n;
  136.     ngx_err_t  err;

  137. eintr:

  138.     n = writev(c->fd, vec->iovs, vec->count);

  139.     ngx_log_debug2(NGX_LOG_DEBUG_EVENT, c->log, 0,
  140.                    "writev: %z of %uz", n, vec->size);

  141.     if (n == -1) {
  142.         err = ngx_errno;

  143.         switch (err) {
  144.         case NGX_EAGAIN:
  145.             ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
  146.                            "writev() not ready");
  147.             return NGX_AGAIN;

  148.         case NGX_EINTR:
  149.             ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
  150.                            "writev() was interrupted");
  151.             goto eintr;

  152.         default:
  153.             c->write->error = 1;
  154.             ngx_connection_error(c, err, "writev() failed");
  155.             return NGX_ERROR;
  156.         }
  157.     }

  158.     return n;
  159. }