| |
| /* |
| * Copyright (C) Igor Sysoev |
| */ |
| |
| |
| #include <ngx_config.h> |
| #include <ngx_core.h> |
| #include <ngx_http.h> |
| |
| |
| static ngx_int_t ngx_http_chunked_filter_init(ngx_conf_t *cf); |
| |
| |
| static ngx_http_module_t ngx_http_chunked_filter_module_ctx = { |
| NULL, /* preconfiguration */ |
| ngx_http_chunked_filter_init, /* postconfiguration */ |
| |
| NULL, /* create main configuration */ |
| NULL, /* init main configuration */ |
| |
| NULL, /* create server configuration */ |
| NULL, /* merge server configuration */ |
| |
| NULL, /* create location configuration */ |
| NULL /* merge location configuration */ |
| }; |
| |
| |
| ngx_module_t ngx_http_chunked_filter_module = { |
| NGX_MODULE_V1, |
| &ngx_http_chunked_filter_module_ctx, /* module context */ |
| NULL, /* module directives */ |
| NGX_HTTP_MODULE, /* module type */ |
| NULL, /* init master */ |
| NULL, /* init module */ |
| NULL, /* init process */ |
| NULL, /* init thread */ |
| NULL, /* exit thread */ |
| NULL, /* exit process */ |
| NULL, /* exit master */ |
| NGX_MODULE_V1_PADDING |
| }; |
| |
| |
| static ngx_http_output_header_filter_pt ngx_http_next_header_filter; |
| static ngx_http_output_body_filter_pt ngx_http_next_body_filter; |
| |
| |
| static ngx_int_t |
| ngx_http_chunked_header_filter(ngx_http_request_t *r) |
| { |
| if (r->headers_out.status == NGX_HTTP_NOT_MODIFIED |
| || r->headers_out.status == NGX_HTTP_NO_CONTENT |
| || r->headers_out.status == NGX_HTTP_CREATED |
| || r != r->main |
| || (r->method & NGX_HTTP_HEAD)) |
| { |
| return ngx_http_next_header_filter(r); |
| } |
| |
| if (r->headers_out.content_length_n == -1) { |
| if (r->http_version < NGX_HTTP_VERSION_11) { |
| r->keepalive = 0; |
| |
| } else { |
| r->chunked = 1; |
| } |
| } |
| |
| return ngx_http_next_header_filter(r); |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_chunked_body_filter(ngx_http_request_t *r, ngx_chain_t *in) |
| { |
| u_char *chunk; |
| off_t size; |
| ngx_buf_t *b; |
| ngx_chain_t out, tail, *cl, *tl, **ll; |
| |
| if (in == NULL || !r->chunked || r->header_only) { |
| return ngx_http_next_body_filter(r, in); |
| } |
| |
| out.buf = NULL; |
| ll = &out.next; |
| |
| size = 0; |
| cl = in; |
| |
| for ( ;; ) { |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http chunk: %d", ngx_buf_size(cl->buf)); |
| |
| size += ngx_buf_size(cl->buf); |
| |
| if (cl->buf->flush |
| || cl->buf->sync |
| || ngx_buf_in_memory(cl->buf) |
| || cl->buf->in_file) |
| { |
| tl = ngx_alloc_chain_link(r->pool); |
| if (tl == NULL) { |
| return NGX_ERROR; |
| } |
| |
| tl->buf = cl->buf; |
| *ll = tl; |
| ll = &tl->next; |
| } |
| |
| if (cl->next == NULL) { |
| break; |
| } |
| |
| cl = cl->next; |
| } |
| |
| if (size) { |
| b = ngx_calloc_buf(r->pool); |
| if (b == NULL) { |
| return NGX_ERROR; |
| } |
| |
| /* the "0000000000000000" is 64-bit hexadimal string */ |
| |
| chunk = ngx_palloc(r->pool, sizeof("0000000000000000" CRLF) - 1); |
| if (chunk == NULL) { |
| return NGX_ERROR; |
| } |
| |
| b->temporary = 1; |
| b->pos = chunk; |
| b->last = ngx_sprintf(chunk, "%xO" CRLF, size); |
| |
| out.buf = b; |
| } |
| |
| if (cl->buf->last_buf) { |
| b = ngx_calloc_buf(r->pool); |
| if (b == NULL) { |
| return NGX_ERROR; |
| } |
| |
| b->memory = 1; |
| b->last_buf = 1; |
| b->pos = (u_char *) CRLF "0" CRLF CRLF; |
| b->last = b->pos + 7; |
| |
| cl->buf->last_buf = 0; |
| |
| if (size == 0) { |
| b->pos += 2; |
| out.buf = b; |
| out.next = NULL; |
| |
| return ngx_http_next_body_filter(r, &out); |
| } |
| |
| } else { |
| if (size == 0) { |
| *ll = NULL; |
| return ngx_http_next_body_filter(r, out.next); |
| } |
| |
| b = ngx_calloc_buf(r->pool); |
| if (b == NULL) { |
| return NGX_ERROR; |
| } |
| |
| b->memory = 1; |
| b->pos = (u_char *) CRLF; |
| b->last = b->pos + 2; |
| } |
| |
| tail.buf = b; |
| tail.next = NULL; |
| *ll = &tail; |
| |
| return ngx_http_next_body_filter(r, &out); |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_chunked_filter_init(ngx_conf_t *cf) |
| { |
| ngx_http_next_header_filter = ngx_http_top_header_filter; |
| ngx_http_top_header_filter = ngx_http_chunked_header_filter; |
| |
| ngx_http_next_body_filter = ngx_http_top_body_filter; |
| ngx_http_top_body_filter = ngx_http_chunked_body_filter; |
| |
| return NGX_OK; |
| } |