Mercurial > hg > nginx-quic
view src/http/ngx_http_write_filter.c @ 70:e320bf51c4e3
nginx-0.0.1-2003-03-25-18:40:51 import
author | Igor Sysoev <igor@sysoev.ru> |
---|---|
date | Tue, 25 Mar 2003 15:40:51 +0000 |
parents | e43f406e4525 |
children | 59229033ae93 |
line wrap: on
line source
#include <ngx_config.h> #include <ngx_core.h> #include <ngx_hunk.h> #include <ngx_conf_file.h> #include <ngx_connection.h> #include <ngx_event_write.h> #include <ngx_http.h> #include <ngx_http_config.h> #include <ngx_http_write_filter.h> static void *ngx_http_write_filter_create_conf(ngx_pool_t *pool); static char *ngx_http_write_filter_merge_conf(ngx_pool_t *pool, void *parent, void *child); static void ngx_http_write_filter_init(ngx_pool_t *pool, ngx_http_conf_filter_t *cf); static ngx_command_t ngx_http_write_filter_commands[] = { {ngx_string("write_buffer"), NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1, ngx_conf_set_size_slot, NGX_HTTP_LOC_CONF_OFFSET, offsetof(ngx_http_write_filter_conf_t, buffer_output)}, {ngx_string(""), 0, NULL, 0, 0} }; ngx_http_module_t ngx_http_write_filter_module_ctx = { NULL, /* create server config */ NULL, /* init server config */ ngx_http_write_filter_create_conf, /* create location config */ ngx_http_write_filter_merge_conf, /* merge location config */ ngx_http_write_filter_init /* init filters */ }; ngx_module_t ngx_http_write_filter_module = { 0, /* module index */ &ngx_http_write_filter_module_ctx, /* module context */ ngx_http_write_filter_commands, /* module directives */ NGX_HTTP_MODULE_TYPE, /* module type */ NULL /* init module */ }; int ngx_http_write_filter(ngx_http_request_t *r, ngx_chain_t *in) { int last; off_t size, flush; ngx_chain_t *ce, **le, *chain; ngx_http_write_filter_ctx_t *ctx; ngx_http_write_filter_conf_t *conf; ctx = (ngx_http_write_filter_ctx_t *) ngx_http_get_module_ctx(r->main ? r->main : r, ngx_http_write_filter_module); if (ctx == NULL) { ngx_http_create_ctx(r, ctx, ngx_http_write_filter_module, sizeof(ngx_http_write_filter_ctx_t), NGX_ERROR); } size = flush = 0; last = 0; le = &ctx->out; /* find the size, the flush point and the last entry of saved chain */ for (ce = ctx->out; ce; ce = ce->next) { le = &ce->next; if (ce->hunk->type & NGX_HUNK_IN_MEMORY) { size += ce->hunk->last - ce->hunk->pos; } else { size += ce->hunk->file_last - ce->hunk->file_pos; } if (ce->hunk->type & (NGX_HUNK_FLUSH|NGX_HUNK_RECYCLED)) { flush = size; } if (ce->hunk->type & NGX_HUNK_LAST) { last = 1; } } /* add the new chain to the existent one */ for (/* void */; in; in = in->next) { ngx_test_null(ce, ngx_palloc(r->pool, sizeof(ngx_chain_t)), NGX_ERROR); ce->hunk = in->hunk; ce->next = NULL; *le = ce; le = &ce->next; if (ce->hunk->type & NGX_HUNK_IN_MEMORY) { size += ce->hunk->last - ce->hunk->pos; } else { size += ce->hunk->file_last - ce->hunk->file_pos; } if (ce->hunk->type & (NGX_HUNK_FLUSH|NGX_HUNK_RECYCLED)) { flush = size; } if (ce->hunk->type & NGX_HUNK_LAST) { last = 1; } } conf = (ngx_http_write_filter_conf_t *) ngx_http_get_module_loc_conf(r->main ? r->main : r, ngx_http_write_filter_module); #if (NGX_DEBUG_WRITE_FILTER) ngx_log_debug(r->connection->log, "write filter: last:%d flush:%d" _ last _ flush); #endif /* avoid the output if there is no last hunk, no flush point and size of the hunks is smaller then 'write_buffer' */ if (!last && flush == 0 && size < conf->buffer_output) { return NGX_OK; } chain = ngx_write_chain(r->connection, ctx->out, flush); #if (NGX_DEBUG_WRITE_FILTER) ngx_log_debug(r->connection->log, "write filter %x" _ chain); #endif if (chain == NGX_CHAIN_ERROR) { return NGX_ERROR; } ctx->out = chain; if (chain == NULL) { return NGX_OK; } else { return NGX_AGAIN; } } static void ngx_http_write_filter_init(ngx_pool_t *pool, ngx_http_conf_filter_t *cf) { cf->output_body_filter = ngx_http_write_filter; } static void *ngx_http_write_filter_create_conf(ngx_pool_t *pool) { ngx_http_write_filter_conf_t *conf; ngx_test_null(conf, ngx_palloc(pool, sizeof(ngx_http_write_filter_conf_t)), NULL); conf->buffer_output = NGX_CONF_UNSET; return conf; } static char *ngx_http_write_filter_merge_conf(ngx_pool_t *pool, void *parent, void *child) { ngx_http_write_filter_conf_t *prev = (ngx_http_write_filter_conf_t *) parent; ngx_http_write_filter_conf_t *conf = (ngx_http_write_filter_conf_t *) child; ngx_conf_size_merge(conf->buffer_output, prev->buffer_output, 1460); return NULL; }