One Level Up
Top Level
src/event/ngx_event_pipe.c - nginx-1.7.10
Functions defined
Source code
- #include <ngx_config.h>
- #include <ngx_core.h>
- #include <ngx_event.h>
- #include <ngx_event_pipe.h>
- static ngx_int_t ngx_event_pipe_read_upstream(ngx_event_pipe_t *p);
- static ngx_int_t ngx_event_pipe_write_to_downstream(ngx_event_pipe_t *p);
- static ngx_int_t ngx_event_pipe_write_chain_to_temp_file(ngx_event_pipe_t *p);
- static ngx_inline void ngx_event_pipe_remove_shadow_links(ngx_buf_t *buf);
- static ngx_int_t ngx_event_pipe_drain_chains(ngx_event_pipe_t *p);
- ngx_int_t
- ngx_event_pipe(ngx_event_pipe_t *p, ngx_int_t do_write)
- {
- u_int flags;
- ngx_int_t rc;
- ngx_event_t *rev, *wev;
- for ( ;; ) {
- if (do_write) {
- p->log->action = "sending to client";
- rc = ngx_event_pipe_write_to_downstream(p);
- if (rc == NGX_ABORT) {
- return NGX_ABORT;
- }
- if (rc == NGX_BUSY) {
- return NGX_OK;
- }
- }
- p->read = 0;
- p->upstream_blocked = 0;
- p->log->action = "reading upstream";
- if (ngx_event_pipe_read_upstream(p) == NGX_ABORT) {
- return NGX_ABORT;
- }
- if (!p->read && !p->upstream_blocked) {
- break;
- }
- do_write = 1;
- }
- if (p->upstream->fd != (ngx_socket_t) -1) {
- rev = p->upstream->read;
- flags = (rev->eof || rev->error) ? NGX_CLOSE_EVENT : 0;
- if (ngx_handle_read_event(rev, flags) != NGX_OK) {
- return NGX_ABORT;
- }
- if (!rev->delayed) {
- if (rev->active && !rev->ready) {
- ngx_add_timer(rev, p->read_timeout);
- } else if (rev->timer_set) {
- ngx_del_timer(rev);
- }
- }
- }
- if (p->downstream->fd != (ngx_socket_t) -1
- && p->downstream->data == p->output_ctx)
- {
- wev = p->downstream->write;
- if (ngx_handle_write_event(wev, p->send_lowat) != NGX_OK) {
- return NGX_ABORT;
- }
- if (!wev->delayed) {
- if (wev->active && !wev->ready) {
- ngx_add_timer(wev, p->send_timeout);
- } else if (wev->timer_set) {
- ngx_del_timer(wev);
- }
- }
- }
- return NGX_OK;
- }
- static ngx_int_t
- ngx_event_pipe_read_upstream(ngx_event_pipe_t *p)
- {
- off_t limit;
- ssize_t n, size;
- ngx_int_t rc;
- ngx_buf_t *b;
- ngx_msec_t delay;
- ngx_chain_t *chain, *cl, *ln;
- if (p->upstream_eof || p->upstream_error || p->upstream_done) {
- return NGX_OK;
- }
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe read upstream: %d", p->upstream->read->ready);
- for ( ;; ) {
- if (p->upstream_eof || p->upstream_error || p->upstream_done) {
- break;
- }
- if (p->preread_bufs == NULL && !p->upstream->read->ready) {
- break;
- }
- if (p->preread_bufs) {
-
- chain = p->preread_bufs;
- p->preread_bufs = NULL;
- n = p->preread_size;
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe preread: %z", n);
- if (n) {
- p->read = 1;
- }
- } else {
- #if (NGX_HAVE_KQUEUE)
-
- if (p->upstream->read->available == 0
- && p->upstream->read->pending_eof)
- {
- p->upstream->read->ready = 0;
- p->upstream->read->eof = 1;
- p->upstream_eof = 1;
- p->read = 1;
- if (p->upstream->read->kq_errno) {
- p->upstream->read->error = 1;
- p->upstream_error = 1;
- p->upstream_eof = 0;
- ngx_log_error(NGX_LOG_ERR, p->log,
- p->upstream->read->kq_errno,
- "kevent() reported that upstream "
- "closed connection");
- }
- break;
- }
- #endif
- if (p->limit_rate) {
- if (p->upstream->read->delayed) {
- break;
- }
- limit = (off_t) p->limit_rate * (ngx_time() - p->start_sec + 1)
- - p->read_length;
- if (limit <= 0) {
- p->upstream->read->delayed = 1;
- delay = (ngx_msec_t) (- limit * 1000 / p->limit_rate + 1);
- ngx_add_timer(p->upstream->read, delay);
- break;
- }
- } else {
- limit = 0;
- }
- if (p->free_raw_bufs) {
-
- chain = p->free_raw_bufs;
- if (p->single_buf) {
- p->free_raw_bufs = p->free_raw_bufs->next;
- chain->next = NULL;
- } else {
- p->free_raw_bufs = NULL;
- }
- } else if (p->allocated < p->bufs.num) {
-
- b = ngx_create_temp_buf(p->pool, p->bufs.size);
- if (b == NULL) {
- return NGX_ABORT;
- }
- p->allocated++;
- chain = ngx_alloc_chain_link(p->pool);
- if (chain == NULL) {
- return NGX_ABORT;
- }
- chain->buf = b;
- chain->next = NULL;
- } else if (!p->cacheable
- && p->downstream->data == p->output_ctx
- && p->downstream->write->ready
- && !p->downstream->write->delayed)
- {
-
- p->upstream_blocked = 1;
- ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe downstream ready");
- break;
- } else if (p->cacheable
- || p->temp_file->offset < p->max_temp_file_size)
- {
-
- rc = ngx_event_pipe_write_chain_to_temp_file(p);
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe temp offset: %O", p->temp_file->offset);
- if (rc == NGX_BUSY) {
- break;
- }
- if (rc == NGX_AGAIN) {
- if (ngx_event_flags & NGX_USE_LEVEL_EVENT
- && p->upstream->read->active
- && p->upstream->read->ready)
- {
- if (ngx_del_event(p->upstream->read, NGX_READ_EVENT, 0)
- == NGX_ERROR)
- {
- return NGX_ABORT;
- }
- }
- }
- if (rc != NGX_OK) {
- return rc;
- }
- chain = p->free_raw_bufs;
- if (p->single_buf) {
- p->free_raw_bufs = p->free_raw_bufs->next;
- chain->next = NULL;
- } else {
- p->free_raw_bufs = NULL;
- }
- } else {
-
- ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "no pipe bufs to read in");
- break;
- }
- n = p->upstream->recv_chain(p->upstream, chain, limit);
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe recv chain: %z", n);
- if (p->free_raw_bufs) {
- chain->next = p->free_raw_bufs;
- }
- p->free_raw_bufs = chain;
- if (n == NGX_ERROR) {
- p->upstream_error = 1;
- return NGX_ERROR;
- }
- if (n == NGX_AGAIN) {
- if (p->single_buf) {
- ngx_event_pipe_remove_shadow_links(chain->buf);
- }
- break;
- }
- p->read = 1;
- if (n == 0) {
- p->upstream_eof = 1;
- break;
- }
- }
- delay = p->limit_rate ? (ngx_msec_t) n * 1000 / p->limit_rate : 0;
- p->read_length += n;
- cl = chain;
- p->free_raw_bufs = NULL;
- while (cl && n > 0) {
- ngx_event_pipe_remove_shadow_links(cl->buf);
- size = cl->buf->end - cl->buf->last;
- if (n >= size) {
- cl->buf->last = cl->buf->end;
- cl->buf->num = p->num++;
- if (p->input_filter(p, cl->buf) == NGX_ERROR) {
- return NGX_ABORT;
- }
- n -= size;
- ln = cl;
- cl = cl->next;
- ngx_free_chain(p->pool, ln);
- } else {
- cl->buf->last += n;
- n = 0;
- }
- }
- if (cl) {
- for (ln = cl; ln->next; ln = ln->next) { }
- ln->next = p->free_raw_bufs;
- p->free_raw_bufs = cl;
- }
- if (delay > 0) {
- p->upstream->read->delayed = 1;
- ngx_add_timer(p->upstream->read, delay);
- break;
- }
- }
- #if (NGX_DEBUG)
- for (cl = p->busy; cl; cl = cl->next) {
- ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe buf busy s:%d t:%d f:%d "
- "%p, pos %p, size: %z "
- "file: %O, size: %z",
- (cl->buf->shadow ? 1 : 0),
- cl->buf->temporary, cl->buf->in_file,
- cl->buf->start, cl->buf->pos,
- cl->buf->last - cl->buf->pos,
- cl->buf->file_pos,
- cl->buf->file_last - cl->buf->file_pos);
- }
- for (cl = p->out; cl; cl = cl->next) {
- ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe buf out s:%d t:%d f:%d "
- "%p, pos %p, size: %z "
- "file: %O, size: %z",
- (cl->buf->shadow ? 1 : 0),
- cl->buf->temporary, cl->buf->in_file,
- cl->buf->start, cl->buf->pos,
- cl->buf->last - cl->buf->pos,
- cl->buf->file_pos,
- cl->buf->file_last - cl->buf->file_pos);
- }
- for (cl = p->in; cl; cl = cl->next) {
- ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe buf in s:%d t:%d f:%d "
- "%p, pos %p, size: %z "
- "file: %O, size: %z",
- (cl->buf->shadow ? 1 : 0),
- cl->buf->temporary, cl->buf->in_file,
- cl->buf->start, cl->buf->pos,
- cl->buf->last - cl->buf->pos,
- cl->buf->file_pos,
- cl->buf->file_last - cl->buf->file_pos);
- }
- for (cl = p->free_raw_bufs; cl; cl = cl->next) {
- ngx_log_debug8(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe buf free s:%d t:%d f:%d "
- "%p, pos %p, size: %z "
- "file: %O, size: %z",
- (cl->buf->shadow ? 1 : 0),
- cl->buf->temporary, cl->buf->in_file,
- cl->buf->start, cl->buf->pos,
- cl->buf->last - cl->buf->pos,
- cl->buf->file_pos,
- cl->buf->file_last - cl->buf->file_pos);
- }
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe length: %O", p->length);
- #endif
- if (p->free_raw_bufs && p->length != -1) {
- cl = p->free_raw_bufs;
- if (cl->buf->last - cl->buf->pos >= p->length) {
- p->free_raw_bufs = cl->next;
- cl->buf->num = p->num++;
- if (p->input_filter(p, cl->buf) == NGX_ERROR) {
- return NGX_ABORT;
- }
- ngx_free_chain(p->pool, cl);
- }
- }
- if (p->length == 0) {
- p->upstream_done = 1;
- p->read = 1;
- }
- if ((p->upstream_eof || p->upstream_error) && p->free_raw_bufs) {
- p->free_raw_bufs->buf->num = p->num++;
- if (p->input_filter(p, p->free_raw_bufs->buf) == NGX_ERROR) {
- return NGX_ABORT;
- }
- p->free_raw_bufs = p->free_raw_bufs->next;
- if (p->free_bufs && p->buf_to_file == NULL) {
- for (cl = p->free_raw_bufs; cl; cl = cl->next) {
- if (cl->buf->shadow == NULL) {
- ngx_pfree(p->pool, cl->buf->start);
- }
- }
- }
- }
- if (p->cacheable && (p->in || p->buf_to_file)) {
- ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write chain");
- if (ngx_event_pipe_write_chain_to_temp_file(p) == NGX_ABORT) {
- return NGX_ABORT;
- }
- }
- return NGX_OK;
- }
- static ngx_int_t
- ngx_event_pipe_write_to_downstream(ngx_event_pipe_t *p)
- {
- u_char *prev;
- size_t bsize;
- ngx_int_t rc;
- ngx_uint_t flush, flushed, prev_last_shadow;
- ngx_chain_t *out, **ll, *cl;
- ngx_connection_t *downstream;
- downstream = p->downstream;
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write downstream: %d", downstream->write->ready);
- flushed = 0;
- for ( ;; ) {
- if (p->downstream_error) {
- return ngx_event_pipe_drain_chains(p);
- }
- if (p->upstream_eof || p->upstream_error || p->upstream_done) {
-
- for (cl = p->busy; cl; cl = cl->next) {
- cl->buf->recycled = 0;
- }
- if (p->out) {
- ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write downstream flush out");
- for (cl = p->out; cl; cl = cl->next) {
- cl->buf->recycled = 0;
- }
- rc = p->output_filter(p->output_ctx, p->out);
- if (rc == NGX_ERROR) {
- p->downstream_error = 1;
- return ngx_event_pipe_drain_chains(p);
- }
- p->out = NULL;
- }
- if (p->in) {
- ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write downstream flush in");
- for (cl = p->in; cl; cl = cl->next) {
- cl->buf->recycled = 0;
- }
- rc = p->output_filter(p->output_ctx, p->in);
- if (rc == NGX_ERROR) {
- p->downstream_error = 1;
- return ngx_event_pipe_drain_chains(p);
- }
- p->in = NULL;
- }
- ngx_log_debug0(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write downstream done");
- TODO
- p->downstream_done = 1;
- break;
- }
- if (downstream->data != p->output_ctx
- || !downstream->write->ready
- || downstream->write->delayed)
- {
- break;
- }
-
- prev = NULL;
- bsize = 0;
- for (cl = p->busy; cl; cl = cl->next) {
- if (cl->buf->recycled) {
- if (prev == cl->buf->start) {
- continue;
- }
- bsize += cl->buf->end - cl->buf->start;
- prev = cl->buf->start;
- }
- }
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write busy: %uz", bsize);
- out = NULL;
- if (bsize >= (size_t) p->busy_size) {
- flush = 1;
- goto flush;
- }
- flush = 0;
- ll = NULL;
- prev_last_shadow = 1;
- for ( ;; ) {
- if (p->out) {
- cl = p->out;
- if (cl->buf->recycled) {
- ngx_log_error(NGX_LOG_ALERT, p->log, 0,
- "recycled buffer in pipe out chain");
- }
- p->out = p->out->next;
- } else if (!p->cacheable && p->in) {
- cl = p->in;
- ngx_log_debug3(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write buf ls:%d %p %z",
- cl->buf->last_shadow,
- cl->buf->pos,
- cl->buf->last - cl->buf->pos);
- if (cl->buf->recycled && prev_last_shadow) {
- if (bsize + cl->buf->end - cl->buf->start > p->busy_size) {
- flush = 1;
- break;
- }
- bsize += cl->buf->end - cl->buf->start;
- }
- prev_last_shadow = cl->buf->last_shadow;
- p->in = p->in->next;
- } else {
- break;
- }
- cl->next = NULL;
- if (out) {
- *ll = cl;
- } else {
- out = cl;
- }
- ll = &cl->next;
- }
- flush:
- ngx_log_debug2(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe write: out:%p, f:%d", out, flush);
- if (out == NULL) {
- if (!flush) {
- break;
- }
-
- if (flushed++ > 10) {
- return NGX_BUSY;
- }
- }
- rc = p->output_filter(p->output_ctx, out);
- ngx_chain_update_chains(p->pool, &p->free, &p->busy, &out, p->tag);
- if (rc == NGX_ERROR) {
- p->downstream_error = 1;
- return ngx_event_pipe_drain_chains(p);
- }
- for (cl = p->free; cl; cl = cl->next) {
- if (cl->buf->temp_file) {
- if (p->cacheable || !p->cyclic_temp_file) {
- continue;
- }
-
- if (cl->buf->file_last == p->temp_file->offset) {
- p->temp_file->offset = 0;
- }
- }
- TODO
-
- if (cl->buf->last_shadow) {
- if (ngx_event_pipe_add_free_buf(p, cl->buf->shadow) != NGX_OK) {
- return NGX_ABORT;
- }
- cl->buf->last_shadow = 0;
- }
- cl->buf->shadow = NULL;
- }
- }
- return NGX_OK;
- }
- static ngx_int_t
- ngx_event_pipe_write_chain_to_temp_file(ngx_event_pipe_t *p)
- {
- ssize_t size, bsize, n;
- ngx_buf_t *b;
- ngx_uint_t prev_last_shadow;
- ngx_chain_t *cl, *tl, *next, *out, **ll, **last_out, **last_free, fl;
- if (p->buf_to_file) {
- fl.buf = p->buf_to_file;
- fl.next = p->in;
- out = &fl;
- } else {
- out = p->in;
- }
- if (!p->cacheable) {
- size = 0;
- cl = out;
- ll = NULL;
- prev_last_shadow = 1;
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe offset: %O", p->temp_file->offset);
- do {
- bsize = cl->buf->last - cl->buf->pos;
- ngx_log_debug4(NGX_LOG_DEBUG_EVENT, p->log, 0,
- "pipe buf ls:%d %p, pos %p, size: %z",
- cl->buf->last_shadow, cl->buf->start,
- cl->buf->pos, bsize);
- if (prev_last_shadow
- && ((size + bsize > p->temp_file_write_size)
- || (p->temp_file->offset + size + bsize
- > p->max_temp_file_size)))
- {
- break;
- }
- prev_last_shadow = cl->buf->last_shadow;
- size += bsize;
- ll = &cl->next;
- cl = cl->next;
- } while (cl);
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0, "size: %z", size);
- if (ll == NULL) {
- return NGX_BUSY;
- }
- if (cl) {
- p->in = cl;
- *ll = NULL;
- } else {
- p->in = NULL;
- p->last_in = &p->in;
- }
- } else {
- p->in = NULL;
- p->last_in = &p->in;
- }
- n = ngx_write_chain_to_temp_file(p->temp_file, out);
- if (n == NGX_ERROR) {
- return NGX_ABORT;
- }
- if (p->buf_to_file) {
- p->temp_file->offset = p->buf_to_file->last - p->buf_to_file->pos;
- n -= p->buf_to_file->last - p->buf_to_file->pos;
- p->buf_to_file = NULL;
- out = out->next;
- }
- if (n > 0) {
-
- if (p->out) {
- for (cl = p->out; cl->next; cl = cl->next) { }
- b = cl->buf;
- if (b->file_last == p->temp_file->offset) {
- p->temp_file->offset += n;
- b->file_last = p->temp_file->offset;
- goto free;
- }
- last_out = &cl->next;
- } else {
- last_out = &p->out;
- }
- cl = ngx_chain_get_free_buf(p->pool, &p->free);
- if (cl == NULL) {
- return NGX_ABORT;
- }
- b = cl->buf;
- ngx_memzero(b, sizeof(ngx_buf_t));
- b->tag = p->tag;
- b->file = &p->temp_file->file;
- b->file_pos = p->temp_file->offset;
- p->temp_file->offset += n;
- b->file_last = p->temp_file->offset;
- b->in_file = 1;
- b->temp_file = 1;
- *last_out = cl;
- }
- free:
- for (last_free = &p->free_raw_bufs;
- *last_free != NULL;
- last_free = &(*last_free)->next)
- {
-
- }
- for (cl = out; cl; cl = next) {
- next = cl->next;
- cl->next = p->free;
- p->free = cl;
- b = cl->buf;
- if (b->last_shadow) {
- tl = ngx_alloc_chain_link(p->pool);
- if (tl == NULL) {
- return NGX_ABORT;
- }
- tl->buf = b->shadow;
- tl->next = NULL;
- *last_free = tl;
- last_free = &tl->next;
- b->shadow->pos = b->shadow->start;
- b->shadow->last = b->shadow->start;
- ngx_event_pipe_remove_shadow_links(b->shadow);
- }
- }
- return NGX_OK;
- }
- ngx_int_t
- ngx_event_pipe_copy_input_filter(ngx_event_pipe_t *p, ngx_buf_t *buf)
- {
- ngx_buf_t *b;
- ngx_chain_t *cl;
- if (buf->pos == buf->last) {
- return NGX_OK;
- }
- cl = ngx_chain_get_free_buf(p->pool, &p->free);
- if (cl == NULL) {
- return NGX_ERROR;
- }
- b = cl->buf;
- ngx_memcpy(b, buf, sizeof(ngx_buf_t));
- b->shadow = buf;
- b->tag = p->tag;
- b->last_shadow = 1;
- b->recycled = 1;
- buf->shadow = b;
- ngx_log_debug1(NGX_LOG_DEBUG_EVENT, p->log, 0, "input buf #%d", b->num);
- if (p->in) {
- *p->last_in = cl;
- } else {
- p->in = cl;
- }
- p->last_in = &cl->next;
- if (p->length == -1) {
- return NGX_OK;
- }
- p->length -= b->last - b->pos;
- return NGX_OK;
- }
- static ngx_inline void
- ngx_event_pipe_remove_shadow_links(ngx_buf_t *buf)
- {
- ngx_buf_t *b, *next;
- b = buf->shadow;
- if (b == NULL) {
- return;
- }
- while (!b->last_shadow) {
- next = b->shadow;
- b->temporary = 0;
- b->recycled = 0;
- b->shadow = NULL;
- b = next;
- }
- b->temporary = 0;
- b->recycled = 0;
- b->last_shadow = 0;
- b->shadow = NULL;
- buf->shadow = NULL;
- }
- ngx_int_t
- ngx_event_pipe_add_free_buf(ngx_event_pipe_t *p, ngx_buf_t *b)
- {
- ngx_chain_t *cl;
- cl = ngx_alloc_chain_link(p->pool);
- if (cl == NULL) {
- return NGX_ERROR;
- }
- if (p->buf_to_file && b->start == p->buf_to_file->start) {
- b->pos = p->buf_to_file->last;
- b->last = p->buf_to_file->last;
- } else {
- b->pos = b->start;
- b->last = b->start;
- }
- b->shadow = NULL;
- cl->buf = b;
- if (p->free_raw_bufs == NULL) {
- p->free_raw_bufs = cl;
- cl->next = NULL;
- return NGX_OK;
- }
- if (p->free_raw_bufs->buf->pos == p->free_raw_bufs->buf->last) {
-
- cl->next = p->free_raw_bufs;
- p->free_raw_bufs = cl;
- return NGX_OK;
- }
-
- cl->next = p->free_raw_bufs->next;
- p->free_raw_bufs->next = cl;
- return NGX_OK;
- }
- static ngx_int_t
- ngx_event_pipe_drain_chains(ngx_event_pipe_t *p)
- {
- ngx_chain_t *cl, *tl;
- for ( ;; ) {
- if (p->busy) {
- cl = p->busy;
- p->busy = NULL;
- } else if (p->out) {
- cl = p->out;
- p->out = NULL;
- } else if (p->in) {
- cl = p->in;
- p->in = NULL;
- } else {
- return NGX_OK;
- }
- while (cl) {
- if (cl->buf->last_shadow) {
- if (ngx_event_pipe_add_free_buf(p, cl->buf->shadow) != NGX_OK) {
- return NGX_ABORT;
- }
- cl->buf->last_shadow = 0;
- }
- cl->buf->shadow = NULL;
- tl = cl->next;
- cl->next = p->free;
- p->free = cl;
- cl = tl;
- }
- }
- }
One Level Up
Top Level