Mercurial > hg > nginx-quic
view src/http/ngx_http_request_body.c @ 146:5ac79e574285
nginx-0.0.1-2003-10-14-09:26:00 import
author | Igor Sysoev <igor@sysoev.ru> |
---|---|
date | Tue, 14 Oct 2003 05:26:00 +0000 |
parents | a7e45c45a95c |
children | e7e094d34162 |
line wrap: on
line source
#include <ngx_config.h> #include <ngx_core.h> #include <ngx_http.h> int ngx_http_init_client_request_body(ngx_http_request_t *r, int size) { int header_in_part, len; ngx_hunk_t *h; ngx_http_request_body_t *rb; ngx_test_null(rb, ngx_pcalloc(r->pool, sizeof(ngx_http_request_body_t)), NGX_HTTP_INTERNAL_SERVER_ERROR); header_in_part = r->header_in->end - r->header_in->pos; if (header_in_part) { rb->header_in_pos = r->header_in->pos; } if (header_in_part > r->headers_in.content_length_n) { header_in_part = r->headers_in.content_length_n; } else { len = r->headers_in.content_length_n - header_in_part; if (len > size) { len = size; } else if (len > NGX_PAGE_SIZE) { len = ((len + NGX_PAGE_SIZE - 1) / NGX_PAGE_SIZE) * NGX_PAGE_SIZE; } if (len) { ngx_test_null(rb->hunk, ngx_create_temp_hunk(r->pool, len, 0, 0), NGX_HTTP_INTERNAL_SERVER_ERROR); } } r->request_body = rb; return NGX_OK; } int ngx_http_read_client_request_body(ngx_http_request_t *r) { int size, n, rc; ngx_chain_t *entry; ngx_http_request_body_t *rb; rb = r->request_body; do { if (r->header_in->last < r->header_in->end) { rb->chain[0].hunk = r->header_in; if (rb->hunk) { rb->chain[0].next = &rb->chain[1]; rb->chain[1].hunk = rb->hunk; rb->chain[1].next = NULL; } else { rb->chain[0].next = NULL; } } else { rb->chain[0].hunk = rb->hunk; rb->chain[0].next = NULL; } n = ngx_recv_chain(r->connection, rb->chain); if (n == NGX_ERROR) { return NGX_ERROR; } if (n == NGX_AGAIN) { return NGX_AGAIN; } for (entry = rb->chain; entry; entry = entry->next) { size = entry->hunk->end - entry->hunk->last; if (n >= size) { n -= size; entry->hunk->last = entry->hunk->end; continue; } entry->hunk->last += n; break; } if (rb->hunk && rb->hunk->last == rb->hunk->end) { if (rb->temp_file.fd == NGX_INVALID_FILE) { rc = ngx_create_temp_file(&rb->temp_file, rb->temp_path, r->pool, 0); if (rc == NGX_ERROR) { return NGX_ERROR; } if (rc == NGX_AGAIN) { return NGX_AGAIN; } } n = ngx_write_file(&rb->temp_file, rb->hunk->pos, rb->hunk->last - rb->hunk->pos, rb->offset); if (rc == NGX_ERROR) { return NGX_ERROR; } rb->offset += n; rb->hunk->last = rb->hunk->pos; } } while (r->connection->read->ready); return NGX_OK; } int ngx_http_init_client_request_body_chain(ngx_http_request_t *r) { int i; ngx_hunk_t *h; ngx_http_request_body_t *rb; rb = r->request_body; rb->chain[0].hunk = rb->header_out; i = 0; if (r->header_in->pos < r->header_in->last) { rb->chain[i].next = &rb->chain[i + 1]; i++; rb->chain[i].hunk = r->header_in; } if (rb->temp_file.fd != NGX_INVALID_FILE) { if (rb->file_hunk == NULL) { ngx_test_null(h, ngx_alloc_hunk(r->pool), NGX_ERROR); h->type = NGX_HUNK_FILE; h->pos = h->start = h->pre_start = 0; h->last = h->end = h->post_end = 0; h->file_pos = 0; h->file_last = rb->offset; h->file = &rb->temp_file; h->shadow = NULL; h->tag = 0; rb->file_hunk = h; } rb->chain[i].next = &rb->chain[i + 1]; i++; rb->chain[i].hunk = rb->file_hunk; } if (rb->hunk && rb->hunk->pos < rb->hunk->last) { rb->chain[i].next = &rb->chain[i + 1]; i++; rb->chain[i].hunk = h; } rb->chain[i].next = NULL; return NGX_OK; } void ngx_http_reinit_client_request_body_hunks(ngx_http_request_t *r) { ngx_http_request_body_t *rb; rb = r->request_body; if (rb->header_in_pos) { r->header_in->pos = rb->header_in_pos; } if (rb->file_hunk) { rb->file_hunk->file_pos = 0; } if (rb->hunk) { rb->hunk->pos = rb->hunk->start; } }