src/event/ngx_event_pipe.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. #include <ngx_event_pipe.h>


  9. static ngx_int_t ngx_event_pipe_read_upstream(ngx_event_pipe_t *p);
  10. static ngx_int_t ngx_event_pipe_write_to_downstream(ngx_event_pipe_t *p);

  11. static ngx_int_t ngx_event_pipe_write_chain_to_temp_file(ngx_event_pipe_t *p);
  12. static ngx_inline void ngx_event_pipe_remove_shadow_links(ngx_buf_t *buf);
  13. static ngx_int_t ngx_event_pipe_drain_chains(ngx_event_pipe_t *p);


  14. ngx_int_t
  15. ngx_event_pipe(ngx_event_pipe_t *p, ngx_int_t do_write)
  16. {
  17.     u_int         flags;
  18.     ngx_int_t     rc;
  19.     ngx_event_t  *rev, *wev;

  20.     for ( ;; ) {
  21.         if (do_write) {
  22.             p->log->action = "sending to client";

  23.             rc = ngx_event_pipe_write_to_downstream(p);

  24.             if (rc == NGX_ABORT) {
  25.                 return NGX_ABORT;
  26.             }

  27.             if (rc == NGX_BUSY) {
  28.                 return NGX_OK;
  29.             }
  30.         }

  31.         p->read = 0;
  32.         p->upstream_blocked = 0;

  33.         p->log->action = "reading upstream";

  34.         if (ngx_event_pipe_read_upstream(p) == NGX_ABORT) {
  35.             return NGX_ABORT;
  36.         }

  37.         if (!p->read && !p->upstream_blocked) {
  38.             break;
  39.         }

  40.         do_write = 1;
  41.     }

  42.     if (p->upstream->fd != (ngx_socket_t) -1) {
  43.         rev = p->upstream->read;

  44.         flags = (rev->eof || rev->error) ? NGX_CLOSE_EVENT : 0;

  45.         if (ngx_handle_read_event(rev, flags) != NGX_OK) {
  46.             return NGX_ABORT;
  47.         }

  48.         if (!rev->delayed) {
  49.             if (rev->active && !rev->ready) {
  50.                 ngx_add_timer(rev, p->read_timeout);

  51.             } else if (rev->timer_set) {
  52.                 ngx_del_timer(rev);
  53.             }
  54.         }
  55.     }

  56.     if (p->downstream->fd != (ngx_socket_t) -1
  57.         && p->downstream->data == p->output_ctx)
  58.     {
  59.         wev = p->downstream->write;
  60.         if (ngx_handle_write_event(wev, p->send_lowat) != NGX_OK) {
  61.             return NGX_ABORT;
  62.         }

  63.         if (!wev->delayed) {
  64.             if (wev->active && !wev->ready) {
  65.                 ngx_add_timer(wev, p->send_timeout);

  66.             } else if (wev->timer_set) {
  67.                 ngx_del_timer(wev);
  68.             }
  69.         }
  70.     }

  71.     return NGX_OK;
  72. }


  73. static ngx_int_t
  74. ngx_event_pipe_read_upstream(ngx_event_pipe_t *p)
  75. {
  76.     off_t         limit;
  77.     ssize_t       n, size;
  78.     ngx_int_t     rc;
  79.     ngx_buf_t    *b;
  80.     ngx_msec_t    delay;
  81.     ngx_chain_t  *chain, *cl, *ln;

  82.     if (p->upstream_eof || p->upstream_error || p->upstream_done) {
  83.         return NGX_OK;
  84.     }

  85.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  86.                    "pipe read upstream: %d", p->upstream->read->ready);

  87.     for ( ;; ) {

  88.         if (p->upstream_eof || p->upstream_error || p->upstream_done) {
  89.             break;
  90.         }

  91.         if (p->preread_bufs == NULL && !p->upstream->read->ready) {
  92.             break;
  93.         }

  94.         if (p->preread_bufs) {

  95.             /* use the pre-read bufs if they exist */

  96.             chain = p->preread_bufs;
  97.             p->preread_bufs = NULL;
  98.             n = p->preread_size;

  99.             ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  100.                            "pipe preread: %z", n);

  101.             if (n) {
  102.                 p->read = 1;
  103.             }

  104.         } else {

  105. #if (NGX_HAVE_KQUEUE)

  106.             /*
  107.              * kqueue notifies about the end of file or a pending error.
  108.              * This test allows not to allocate a buf on these conditions
  109.              * and not to call c->recv_chain().
  110.              */

  111.             if (p->upstream->read->available == 0
  112.                 && p->upstream->read->pending_eof)
  113.             {
  114.                 p->upstream->read->ready = 0;
  115.                 p->upstream->read->eof = 1;
  116.                 p->upstream_eof = 1;
  117.                 p->read = 1;

  118.                 if (p->upstream->read->kq_errno) {
  119.                     p->upstream->read->error = 1;
  120.                     p->upstream_error = 1;
  121.                     p->upstream_eof = 0;

  122.                     ngx_log_error(NGX_LOG_ERR, p->log,
  123.                                   p->upstream->read->kq_errno,
  124.                                   "kevent() reported that upstream "
  125.                                   "closed connection");
  126.                 }

  127.                 break;
  128.             }
  129. #endif

  130.             if (p->limit_rate) {
  131.                 if (p->upstream->read->delayed) {
  132.                     break;
  133.                 }

  134.                 limit = (off_t) p->limit_rate * (ngx_time() - p->start_sec + 1)
  135.                         - p->read_length;

  136.                 if (limit <= 0) {
  137.                     p->upstream->read->delayed = 1;
  138.                     delay = (ngx_msec_t) (- limit * 1000 / p->limit_rate + 1);
  139.                     ngx_add_timer(p->upstream->read, delay);
  140.                     break;
  141.                 }

  142.             } else {
  143.                 limit = 0;
  144.             }

  145.             if (p->free_raw_bufs) {

  146.                 /* use the free bufs if they exist */

  147.                 chain = p->free_raw_bufs;
  148.                 if (p->single_buf) {
  149.                     p->free_raw_bufs = p->free_raw_bufs->next;
  150.                     chain->next = NULL;
  151.                 } else {
  152.                     p->free_raw_bufs = NULL;
  153.                 }

  154.             } else if (p->allocated < p->bufs.num) {

  155.                 /* allocate a new buf if it's still allowed */

  156.                 b = ngx_create_temp_buf(p->pool, p->bufs.size);
  157.                 if (b == NULL) {
  158.                     return NGX_ABORT;
  159.                 }

  160.                 p->allocated++;

  161.                 chain = ngx_alloc_chain_link(p->pool);
  162.                 if (chain == NULL) {
  163.                     return NGX_ABORT;
  164.                 }

  165.                 chain->buf = b;
  166.                 chain->next = NULL;

  167.             } else if (!p->cacheable
  168.                        && p->downstream->data == p->output_ctx
  169.                        && p->downstream->write->ready
  170.                        && !p->downstream->write->delayed)
  171.             {
  172.                 /*
  173.                  * if the bufs are not needed to be saved in a cache and
  174.                  * a downstream is ready then write the bufs to a downstream
  175.                  */

  176.                 p->upstream_blocked = 1;

  177.                 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
  178.                                "pipe downstream ready");

  179.                 break;

  180.             } else if (p->cacheable
  181.                        || p->temp_file->offset < p->max_temp_file_size)
  182.             {

  183.                 /*
  184.                  * if it is allowed, then save some bufs from p->in
  185.                  * to a temporary file, and add them to a p->out chain
  186.                  */

  187.                 rc = ngx_event_pipe_write_chain_to_temp_file(p);

  188.                 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  189.                                "pipe temp offset: %O", p->temp_file->offset);

  190.                 if (rc == NGX_BUSY) {
  191.                     break;
  192.                 }

  193.                 if (rc == NGX_AGAIN) {
  194.                     if (ngx_event_flags & NGX_USE_LEVEL_EVENT
  195.                         && p->upstream->read->active
  196.                         && p->upstream->read->ready)
  197.                     {
  198.                         if (ngx_del_event(p->upstream->read, NGX_READ_EVENT, 0)
  199.                             == NGX_ERROR)
  200.                         {
  201.                             return NGX_ABORT;
  202.                         }
  203.                     }
  204.                 }

  205.                 if (rc != NGX_OK) {
  206.                     return rc;
  207.                 }

  208.                 chain = p->free_raw_bufs;
  209.                 if (p->single_buf) {
  210.                     p->free_raw_bufs = p->free_raw_bufs->next;
  211.                     chain->next = NULL;
  212.                 } else {
  213.                     p->free_raw_bufs = NULL;
  214.                 }

  215.             } else {

  216.                 /* there are no bufs to read in */

  217.                 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
  218.                                "no pipe bufs to read in");

  219.                 break;
  220.             }

  221.             n = p->upstream->recv_chain(p->upstream, chain, limit);

  222.             ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  223.                            "pipe recv chain: %z", n);

  224.             if (p->free_raw_bufs) {
  225.                 chain->next = p->free_raw_bufs;
  226.             }
  227.             p->free_raw_bufs = chain;

  228.             if (n == NGX_ERROR) {
  229.                 p->upstream_error = 1;
  230.                 return NGX_ERROR;
  231.             }

  232.             if (n == NGX_AGAIN) {
  233.                 if (p->single_buf) {
  234.                     ngx_event_pipe_remove_shadow_links(chain->buf);
  235.                 }

  236.                 break;
  237.             }

  238.             p->read = 1;

  239.             if (n == 0) {
  240.                 p->upstream_eof = 1;
  241.                 break;
  242.             }
  243.         }

  244.         delay = p->limit_rate ? (ngx_msec_t) n * 1000 / p->limit_rate : 0;

  245.         p->read_length += n;
  246.         cl = chain;
  247.         p->free_raw_bufs = NULL;

  248.         while (cl && n > 0) {

  249.             ngx_event_pipe_remove_shadow_links(cl->buf);

  250.             size = cl->buf->end - cl->buf->last;

  251.             if (n >= size) {
  252.                 cl->buf->last = cl->buf->end;

  253.                 /* STUB */ cl->buf->num = p->num++;

  254.                 if (p->input_filter(p, cl->buf) == NGX_ERROR) {
  255.                     return NGX_ABORT;
  256.                 }

  257.                 n -= size;
  258.                 ln = cl;
  259.                 cl = cl->next;
  260.                 ngx_free_chain(p->pool, ln);

  261.             } else {
  262.                 cl->buf->last += n;
  263.                 n = 0;
  264.             }
  265.         }

  266.         if (cl) {
  267.             for (ln = cl; ln->next; ln = ln->next) { /* void */ }

  268.             ln->next = p->free_raw_bufs;
  269.             p->free_raw_bufs = cl;
  270.         }

  271.         if (delay > 0) {
  272.             p->upstream->read->delayed = 1;
  273.             ngx_add_timer(p->upstream->read, delay);
  274.             break;
  275.         }
  276.     }

  277. #if (NGX_DEBUG)

  278.     for (cl = p->busy; cl; cl = cl->next) {
  279.         ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
  280.                        "pipe buf busy s:%d t:%d f:%d "
  281.                        "%p, pos %p, size: %z "
  282.                        "file: %O, size: %z",
  283.                        (cl->buf->shadow ? 1 : 0),
  284.                        cl->buf->temporary, cl->buf->in_file,
  285.                        cl->buf->start, cl->buf->pos,
  286.                        cl->buf->last - cl->buf->pos,
  287.                        cl->buf->file_pos,
  288.                        cl->buf->file_last - cl->buf->file_pos);
  289.     }

  290.     for (cl = p->out; cl; cl = cl->next) {
  291.         ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
  292.                        "pipe buf out  s:%d t:%d f:%d "
  293.                        "%p, pos %p, size: %z "
  294.                        "file: %O, size: %z",
  295.                        (cl->buf->shadow ? 1 : 0),
  296.                        cl->buf->temporary, cl->buf->in_file,
  297.                        cl->buf->start, cl->buf->pos,
  298.                        cl->buf->last - cl->buf->pos,
  299.                        cl->buf->file_pos,
  300.                        cl->buf->file_last - cl->buf->file_pos);
  301.     }

  302.     for (cl = p->in; cl; cl = cl->next) {
  303.         ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
  304.                        "pipe buf in   s:%d t:%d f:%d "
  305.                        "%p, pos %p, size: %z "
  306.                        "file: %O, size: %z",
  307.                        (cl->buf->shadow ? 1 : 0),
  308.                        cl->buf->temporary, cl->buf->in_file,
  309.                        cl->buf->start, cl->buf->pos,
  310.                        cl->buf->last - cl->buf->pos,
  311.                        cl->buf->file_pos,
  312.                        cl->buf->file_last - cl->buf->file_pos);
  313.     }

  314.     for (cl = p->free_raw_bufs; cl; cl = cl->next) {
  315.         ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
  316.                        "pipe buf free s:%d t:%d f:%d "
  317.                        "%p, pos %p, size: %z "
  318.                        "file: %O, size: %z",
  319.                        (cl->buf->shadow ? 1 : 0),
  320.                        cl->buf->temporary, cl->buf->in_file,
  321.                        cl->buf->start, cl->buf->pos,
  322.                        cl->buf->last - cl->buf->pos,
  323.                        cl->buf->file_pos,
  324.                        cl->buf->file_last - cl->buf->file_pos);
  325.     }

  326.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  327.                    "pipe length: %O", p->length);

  328. #endif

  329.     if (p->free_raw_bufs && p->length != -1) {
  330.         cl = p->free_raw_bufs;

  331.         if (cl->buf->last - cl->buf->pos >= p->length) {

  332.             p->free_raw_bufs = cl->next;

  333.             /* STUB */ cl->buf->num = p->num++;

  334.             if (p->input_filter(p, cl->buf) == NGX_ERROR) {
  335.                  return NGX_ABORT;
  336.             }

  337.             ngx_free_chain(p->pool, cl);
  338.         }
  339.     }

  340.     if (p->length == 0) {
  341.         p->upstream_done = 1;
  342.         p->read = 1;
  343.     }

  344.     if ((p->upstream_eof || p->upstream_error) && p->free_raw_bufs) {

  345.         /* STUB */ p->free_raw_bufs->buf->num = p->num++;

  346.         if (p->input_filter(p, p->free_raw_bufs->buf) == NGX_ERROR) {
  347.             return NGX_ABORT;
  348.         }

  349.         p->free_raw_bufs = p->free_raw_bufs->next;

  350.         if (p->free_bufs && p->buf_to_file == NULL) {
  351.             for (cl = p->free_raw_bufs; cl; cl = cl->next) {
  352.                 if (cl->buf->shadow == NULL) {
  353.                     ngx_pfree(p->pool, cl->buf->start);
  354.                 }
  355.             }
  356.         }
  357.     }

  358.     if (p->cacheable && (p->in || p->buf_to_file)) {

  359.         ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
  360.                        "pipe write chain");

  361.         if (ngx_event_pipe_write_chain_to_temp_file(p) == NGX_ABORT) {
  362.             return NGX_ABORT;
  363.         }
  364.     }

  365.     return NGX_OK;
  366. }


  367. static ngx_int_t
  368. ngx_event_pipe_write_to_downstream(ngx_event_pipe_t *p)
  369. {
  370.     u_char            *prev;
  371.     size_t             bsize;
  372.     ngx_int_t          rc;
  373.     ngx_uint_t         flush, flushed, prev_last_shadow;
  374.     ngx_chain_t       *out, **ll, *cl;
  375.     ngx_connection_t  *downstream;

  376.     downstream = p->downstream;

  377.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  378.                    "pipe write downstream: %d", downstream->write->ready);

  379.     flushed = 0;

  380.     for ( ;; ) {
  381.         if (p->downstream_error) {
  382.             return ngx_event_pipe_drain_chains(p);
  383.         }

  384.         if (p->upstream_eof || p->upstream_error || p->upstream_done) {

  385.             /* pass the p->out and p->in chains to the output filter */

  386.             for (cl = p->busy; cl; cl = cl->next) {
  387.                 cl->buf->recycled = 0;
  388.             }

  389.             if (p->out) {
  390.                 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
  391.                                "pipe write downstream flush out");

  392.                 for (cl = p->out; cl; cl = cl->next) {
  393.                     cl->buf->recycled = 0;
  394.                 }

  395.                 rc = p->output_filter(p->output_ctx, p->out);

  396.                 if (rc == NGX_ERROR) {
  397.                     p->downstream_error = 1;
  398.                     return ngx_event_pipe_drain_chains(p);
  399.                 }

  400.                 p->out = NULL;
  401.             }

  402.             if (p->in) {
  403.                 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
  404.                                "pipe write downstream flush in");

  405.                 for (cl = p->in; cl; cl = cl->next) {
  406.                     cl->buf->recycled = 0;
  407.                 }

  408.                 rc = p->output_filter(p->output_ctx, p->in);

  409.                 if (rc == NGX_ERROR) {
  410.                     p->downstream_error = 1;
  411.                     return ngx_event_pipe_drain_chains(p);
  412.                 }

  413.                 p->in = NULL;
  414.             }

  415.             ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
  416.                            "pipe write downstream done");

  417.             /* TODO: free unused bufs */

  418.             p->downstream_done = 1;
  419.             break;
  420.         }

  421.         if (downstream->data != p->output_ctx
  422.             || !downstream->write->ready
  423.             || downstream->write->delayed)
  424.         {
  425.             break;
  426.         }

  427.         /* bsize is the size of the busy recycled bufs */

  428.         prev = NULL;
  429.         bsize = 0;

  430.         for (cl = p->busy; cl; cl = cl->next) {

  431.             if (cl->buf->recycled) {
  432.                 if (prev == cl->buf->start) {
  433.                     continue;
  434.                 }

  435.                 bsize += cl->buf->end - cl->buf->start;
  436.                 prev = cl->buf->start;
  437.             }
  438.         }

  439.         ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  440.                        "pipe write busy: %uz", bsize);

  441.         out = NULL;

  442.         if (bsize >= (size_t) p->busy_size) {
  443.             flush = 1;
  444.             goto flush;
  445.         }

  446.         flush = 0;
  447.         ll = NULL;
  448.         prev_last_shadow = 1;

  449.         for ( ;; ) {
  450.             if (p->out) {
  451.                 cl = p->out;

  452.                 if (cl->buf->recycled) {
  453.                     ngx_log_error(NGX_LOG_ALERT, p->log, 0,
  454.                                   "recycled buffer in pipe out chain");
  455.                 }

  456.                 p->out = p->out->next;

  457.             } else if (!p->cacheable && p->in) {
  458.                 cl = p->in;

  459.                 ngx_log_debug3(NGX_LOG_DEBUG_EVENT, p->log, 0,
  460.                                "pipe write buf ls:%d %p %z",
  461.                                cl->buf->last_shadow,
  462.                                cl->buf->pos,
  463.                                cl->buf->last - cl->buf->pos);

  464.                 if (cl->buf->recycled && prev_last_shadow) {
  465.                     if (bsize + cl->buf->end - cl->buf->start > p->busy_size) {
  466.                         flush = 1;
  467.                         break;
  468.                     }

  469.                     bsize += cl->buf->end - cl->buf->start;
  470.                 }

  471.                 prev_last_shadow = cl->buf->last_shadow;

  472.                 p->in = p->in->next;

  473.             } else {
  474.                 break;
  475.             }

  476.             cl->next = NULL;

  477.             if (out) {
  478.                 *ll = cl;
  479.             } else {
  480.                 out = cl;
  481.             }
  482.             ll = &cl->next;
  483.         }

  484.     flush:

  485.         ngx_log_debug2(NGX_LOG_DEBUG_EVENT, p->log, 0,
  486.                        "pipe write: out:%p, f:%d", out, flush);

  487.         if (out == NULL) {

  488.             if (!flush) {
  489.                 break;
  490.             }

  491.             /* a workaround for AIO */
  492.             if (flushed++ > 10) {
  493.                 return NGX_BUSY;
  494.             }
  495.         }

  496.         rc = p->output_filter(p->output_ctx, out);

  497.         ngx_chain_update_chains(p->pool, &p->free, &p->busy, &out, p->tag);

  498.         if (rc == NGX_ERROR) {
  499.             p->downstream_error = 1;
  500.             return ngx_event_pipe_drain_chains(p);
  501.         }

  502.         for (cl = p->free; cl; cl = cl->next) {

  503.             if (cl->buf->temp_file) {
  504.                 if (p->cacheable || !p->cyclic_temp_file) {
  505.                     continue;
  506.                 }

  507.                 /* reset p->temp_offset if all bufs had been sent */

  508.                 if (cl->buf->file_last == p->temp_file->offset) {
  509.                     p->temp_file->offset = 0;
  510.                 }
  511.             }

  512.             /* TODO: free buf if p->free_bufs && upstream done */

  513.             /* add the free shadow raw buf to p->free_raw_bufs */

  514.             if (cl->buf->last_shadow) {
  515.                 if (ngx_event_pipe_add_free_buf(p, cl->buf->shadow) != NGX_OK) {
  516.                     return NGX_ABORT;
  517.                 }

  518.                 cl->buf->last_shadow = 0;
  519.             }

  520.             cl->buf->shadow = NULL;
  521.         }
  522.     }

  523.     return NGX_OK;
  524. }


  525. static ngx_int_t
  526. ngx_event_pipe_write_chain_to_temp_file(ngx_event_pipe_t *p)
  527. {
  528.     ssize_t       size, bsize, n;
  529.     ngx_buf_t    *b;
  530.     ngx_uint_t    prev_last_shadow;
  531.     ngx_chain_t  *cl, *tl, *next, *out, **ll, **last_out, **last_free, fl;

  532.     if (p->buf_to_file) {
  533.         fl.buf = p->buf_to_file;
  534.         fl.next = p->in;
  535.         out = &fl;

  536.     } else {
  537.         out = p->in;
  538.     }

  539.     if (!p->cacheable) {

  540.         size = 0;
  541.         cl = out;
  542.         ll = NULL;
  543.         prev_last_shadow = 1;

  544.         ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
  545.                        "pipe offset: %O", p->temp_file->offset);

  546.         do {
  547.             bsize = cl->buf->last - cl->buf->pos;

  548.             ngx_log_debug4(NGX_LOG_DEBUG_EVENT, p->log, 0,
  549.                            "pipe buf ls:%d %p, pos %p, size: %z",
  550.                            cl->buf->last_shadow, cl->buf->start,
  551.                            cl->buf->pos, bsize);

  552.             if (prev_last_shadow
  553.                 && ((size + bsize > p->temp_file_write_size)
  554.                     || (p->temp_file->offset + size + bsize
  555.                         > p->max_temp_file_size)))
  556.             {
  557.                 break;
  558.             }

  559.             prev_last_shadow = cl->buf->last_shadow;

  560.             size += bsize;
  561.             ll = &cl->next;
  562.             cl = cl->next;

  563.         } while (cl);

  564.         ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0, "size: %z", size);

  565.         if (ll == NULL) {
  566.             return NGX_BUSY;
  567.         }

  568.         if (cl) {
  569.            p->in = cl;
  570.            *ll = NULL;

  571.         } else {
  572.            p->in = NULL;
  573.            p->last_in = &p->in;
  574.         }

  575.     } else {
  576.         p->in = NULL;
  577.         p->last_in = &p->in;
  578.     }

  579.     n = ngx_write_chain_to_temp_file(p->temp_file, out);

  580.     if (n == NGX_ERROR) {
  581.         return NGX_ABORT;
  582.     }

  583.     if (p->buf_to_file) {
  584.         p->temp_file->offset = p->buf_to_file->last - p->buf_to_file->pos;
  585.         n -= p->buf_to_file->last - p->buf_to_file->pos;
  586.         p->buf_to_file = NULL;
  587.         out = out->next;
  588.     }

  589.     if (n > 0) {
  590.         /* update previous buffer or add new buffer */

  591.         if (p->out) {
  592.             for (cl = p->out; cl->next; cl = cl->next) { /* void */ }

  593.             b = cl->buf;

  594.             if (b->file_last == p->temp_file->offset) {
  595.                 p->temp_file->offset += n;
  596.                 b->file_last = p->temp_file->offset;
  597.                 goto free;
  598.             }

  599.             last_out = &cl->next;

  600.         } else {
  601.             last_out = &p->out;
  602.         }

  603.         cl = ngx_chain_get_free_buf(p->pool, &p->free);
  604.         if (cl == NULL) {
  605.             return NGX_ABORT;
  606.         }

  607.         b = cl->buf;

  608.         ngx_memzero(b, sizeof(ngx_buf_t));

  609.         b->tag = p->tag;

  610.         b->file = &p->temp_file->file;
  611.         b->file_pos = p->temp_file->offset;
  612.         p->temp_file->offset += n;
  613.         b->file_last = p->temp_file->offset;

  614.         b->in_file = 1;
  615.         b->temp_file = 1;

  616.         *last_out = cl;
  617.     }

  618. free:

  619.     for (last_free = &p->free_raw_bufs;
  620.          *last_free != NULL;
  621.          last_free = &(*last_free)->next)
  622.     {
  623.         /* void */
  624.     }

  625.     for (cl = out; cl; cl = next) {
  626.         next = cl->next;

  627.         cl->next = p->free;
  628.         p->free = cl;

  629.         b = cl->buf;

  630.         if (b->last_shadow) {

  631.             tl = ngx_alloc_chain_link(p->pool);
  632.             if (tl == NULL) {
  633.                 return NGX_ABORT;
  634.             }

  635.             tl->buf = b->shadow;
  636.             tl->next = NULL;

  637.             *last_free = tl;
  638.             last_free = &tl->next;

  639.             b->shadow->pos = b->shadow->start;
  640.             b->shadow->last = b->shadow->start;

  641.             ngx_event_pipe_remove_shadow_links(b->shadow);
  642.         }
  643.     }

  644.     return NGX_OK;
  645. }


  646. /* the copy input filter */

  647. ngx_int_t
  648. ngx_event_pipe_copy_input_filter(ngx_event_pipe_t *p, ngx_buf_t *buf)
  649. {
  650.     ngx_buf_t    *b;
  651.     ngx_chain_t  *cl;

  652.     if (buf->pos == buf->last) {
  653.         return NGX_OK;
  654.     }

  655.     cl = ngx_chain_get_free_buf(p->pool, &p->free);
  656.     if (cl == NULL) {
  657.         return NGX_ERROR;
  658.     }

  659.     b = cl->buf;

  660.     ngx_memcpy(b, buf, sizeof(ngx_buf_t));
  661.     b->shadow = buf;
  662.     b->tag = p->tag;
  663.     b->last_shadow = 1;
  664.     b->recycled = 1;
  665.     buf->shadow = b;

  666.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0, "input buf #%d", b->num);

  667.     if (p->in) {
  668.         *p->last_in = cl;
  669.     } else {
  670.         p->in = cl;
  671.     }
  672.     p->last_in = &cl->next;

  673.     if (p->length == -1) {
  674.         return NGX_OK;
  675.     }

  676.     p->length -= b->last - b->pos;

  677.     return NGX_OK;
  678. }


  679. static ngx_inline void
  680. ngx_event_pipe_remove_shadow_links(ngx_buf_t *buf)
  681. {
  682.     ngx_buf_t  *b, *next;

  683.     b = buf->shadow;

  684.     if (b == NULL) {
  685.         return;
  686.     }

  687.     while (!b->last_shadow) {
  688.         next = b->shadow;

  689.         b->temporary = 0;
  690.         b->recycled = 0;

  691.         b->shadow = NULL;
  692.         b = next;
  693.     }

  694.     b->temporary = 0;
  695.     b->recycled = 0;
  696.     b->last_shadow = 0;

  697.     b->shadow = NULL;

  698.     buf->shadow = NULL;
  699. }


  700. ngx_int_t
  701. ngx_event_pipe_add_free_buf(ngx_event_pipe_t *p, ngx_buf_t *b)
  702. {
  703.     ngx_chain_t  *cl;

  704.     cl = ngx_alloc_chain_link(p->pool);
  705.     if (cl == NULL) {
  706.         return NGX_ERROR;
  707.     }

  708.     if (p->buf_to_file && b->start == p->buf_to_file->start) {
  709.         b->pos = p->buf_to_file->last;
  710.         b->last = p->buf_to_file->last;

  711.     } else {
  712.         b->pos = b->start;
  713.         b->last = b->start;
  714.     }

  715.     b->shadow = NULL;

  716.     cl->buf = b;

  717.     if (p->free_raw_bufs == NULL) {
  718.         p->free_raw_bufs = cl;
  719.         cl->next = NULL;

  720.         return NGX_OK;
  721.     }

  722.     if (p->free_raw_bufs->buf->pos == p->free_raw_bufs->buf->last) {

  723.         /* add the free buf to the list start */

  724.         cl->next = p->free_raw_bufs;
  725.         p->free_raw_bufs = cl;

  726.         return NGX_OK;
  727.     }

  728.     /* the first free buf is partially filled, thus add the free buf after it */

  729.     cl->next = p->free_raw_bufs->next;
  730.     p->free_raw_bufs->next = cl;

  731.     return NGX_OK;
  732. }


  733. static ngx_int_t
  734. ngx_event_pipe_drain_chains(ngx_event_pipe_t *p)
  735. {
  736.     ngx_chain_t  *cl, *tl;

  737.     for ( ;; ) {
  738.         if (p->busy) {
  739.             cl = p->busy;
  740.             p->busy = NULL;

  741.         } else if (p->out) {
  742.             cl = p->out;
  743.             p->out = NULL;

  744.         } else if (p->in) {
  745.             cl = p->in;
  746.             p->in = NULL;

  747.         } else {
  748.             return NGX_OK;
  749.         }

  750.         while (cl) {
  751.             if (cl->buf->last_shadow) {
  752.                 if (ngx_event_pipe_add_free_buf(p, cl->buf->shadow) != NGX_OK) {
  753.                     return NGX_ABORT;
  754.                 }

  755.                 cl->buf->last_shadow = 0;
  756.             }

  757.             cl->buf->shadow = NULL;
  758.             tl = cl->next;
  759.             cl->next = p->free;
  760.             p->free = cl;
  761.             cl = tl;
  762.         }
  763.     }
  764. }