| #!/usr/bin/perl |
| |
| # (C) Maxim Dounin |
| |
| # Tests for nginx request body reading. |
| |
| ############################################################################### |
| |
| use warnings; |
| use strict; |
| |
| use Test::More; |
| use Socket qw/ CRLF /; |
| |
| BEGIN { use FindBin; chdir($FindBin::Bin); } |
| |
| use lib 'lib'; |
| use Test::Nginx; |
| |
| ############################################################################### |
| |
| select STDERR; $| = 1; |
| select STDOUT; $| = 1; |
| |
| my $t = Test::Nginx->new()->has(qw/http proxy rewrite/)->plan(15); |
| |
| $t->write_file_expand('nginx.conf', <<'EOF'); |
| |
| %%TEST_GLOBALS%% |
| |
| daemon off; |
| |
| events { |
| } |
| |
| http { |
| %%TEST_GLOBALS_HTTP%% |
| |
| upstream u { |
| server 127.0.0.1:8082; |
| server 127.0.0.1:8080 backup; |
| } |
| |
| server { |
| listen 127.0.0.1:8080; |
| server_name localhost; |
| |
| client_header_buffer_size 1k; |
| |
| location / { |
| client_body_buffer_size 2k; |
| add_header X-Body "$request_body"; |
| add_header X-Body-File "$request_body_file"; |
| proxy_pass http://127.0.0.1:8081; |
| } |
| location /b { |
| client_body_buffer_size 2k; |
| client_body_in_file_only on; |
| add_header X-Body "$request_body"; |
| add_header X-Body-File "$request_body_file"; |
| proxy_pass http://127.0.0.1:8081; |
| } |
| location /small { |
| client_body_in_file_only on; |
| add_header X-Original-Uri "$request_uri"; |
| proxy_pass http://127.0.0.1:8080/; |
| } |
| location /single { |
| client_body_in_single_buffer on; |
| add_header X-Body "$request_body"; |
| add_header X-Body-File "$request_body_file"; |
| proxy_pass http://127.0.0.1:8081; |
| } |
| location /large { |
| client_max_body_size 1k; |
| proxy_pass http://127.0.0.1:8081; |
| } |
| location /discard { |
| return 200 "TEST\n"; |
| } |
| location /next { |
| proxy_pass http://u/; |
| } |
| location /redirect { |
| error_page 404 http://example.com/; |
| } |
| } |
| |
| server { |
| listen 127.0.0.1:8081; |
| server_name localhost; |
| |
| location / { |
| return 200 "TEST\n"; |
| } |
| } |
| |
| server { |
| listen 127.0.0.1:8082; |
| server_name localhost; |
| |
| location / { |
| return 444; |
| } |
| } |
| } |
| |
| EOF |
| |
| $t->run(); |
| |
| ############################################################################### |
| |
| unlike(http_get('/'), qr/X-Body:/ms, 'no body'); |
| |
| like(http_get_body('/', '0123456789'), |
| qr/X-Body: 0123456789\x0d?$/ms, 'body'); |
| |
| like(http_get_body('/', '0123456789' x 128), |
| qr/X-Body: (0123456789){128}\x0d?$/ms, 'body in two buffers'); |
| |
| like(http_get_body('/', '0123456789' x 512), |
| qr/X-Body-File/ms, 'body in file'); |
| |
| like(read_body_file(http_get_body('/b', '0123456789' x 512)), |
| qr/^(0123456789){512}$/s, 'body in file only'); |
| |
| like(http_get_body('/single', '0123456789' x 128), |
| qr/X-Body: (0123456789){128}\x0d?$/ms, 'body in single buffer'); |
| |
| like(http_get_body('/large', '0123456789' x 128), qr/ 413 /, 'body too large'); |
| |
| # pipelined requests |
| |
| like(http_get_body('/', '0123456789', '0123456789' x 128, '0123456789' x 512, |
| 'foobar'), qr/X-Body: foobar\x0d?$/ms, 'body pipelined'); |
| like(http_get_body('/', '0123456789' x 128, '0123456789' x 512, '0123456789', |
| 'foobar'), qr/X-Body: foobar\x0d?$/ms, 'body pipelined 2'); |
| |
| like(http_get_body('/discard', '0123456789', '0123456789' x 128, |
| '0123456789' x 512, 'foobar'), qr/(TEST.*){4}/ms, |
| 'body discard'); |
| like(http_get_body('/discard', '0123456789' x 128, '0123456789' x 512, |
| '0123456789', 'foobar'), qr/(TEST.*){4}/ms, |
| 'body discard 2'); |
| |
| # proxy with file only |
| |
| like(http_get_body('/small', '0123456789'), |
| qr/X-Body: 0123456789\x0d?$/ms, 'small body in file only'); |
| |
| # proxy with file only - reuse of r->header_in |
| |
| like( |
| http( |
| 'GET /small HTTP/1.0' . CRLF |
| . 'Content-Length: 10' . CRLF . CRLF |
| . '01234', |
| sleep => 0.1, |
| body => '56789' |
| ), |
| qr!X-Body: 0123456789\x0d?\x0a.*X-Original-Uri: /small!ms, |
| 'small body in file only, not preread' |
| ); |
| |
| # proxy_next_upstream |
| |
| like(http_get_body('/next', '0123456789'), |
| qr/X-Body: 0123456789\x0d?$/ms, 'body next upstream'); |
| |
| # discarded request body in redirect via error_page |
| |
| TODO: { |
| local $TODO = 'not yet' unless $t->has_version('1.17.7'); |
| |
| unlike( |
| http( |
| 'POST /redirect HTTP/1.1' . CRLF |
| . 'Host: localhost' . CRLF |
| . 'Content-Length: 10' . CRLF . CRLF |
| . '0123456789' . |
| 'GET /next HTTP/1.0' . CRLF . CRLF |
| ), |
| qr/400 Bad Request/ms, 'redirect - discard request body' |
| ); |
| |
| } |
| |
| ############################################################################### |
| |
| sub read_body_file { |
| my ($r) = @_; |
| return '' unless $r =~ m/X-Body-File: (.*)/; |
| open FILE, $1 |
| or return "$!"; |
| local $/; |
| my $content = <FILE>; |
| close FILE; |
| return $content; |
| } |
| |
| sub http_get_body { |
| my $uri = shift; |
| my $last = pop; |
| return http( join '', (map { |
| my $body = $_; |
| "GET $uri HTTP/1.1" . CRLF |
| . "Host: localhost" . CRLF |
| . "Content-Length: " . (length $body) . CRLF . CRLF |
| . $body |
| } @_), |
| "GET $uri HTTP/1.1" . CRLF |
| . "Host: localhost" . CRLF |
| . "Connection: close" . CRLF |
| . "Content-Length: " . (length $last) . CRLF . CRLF |
| . $last |
| ); |
| } |
| |
| ############################################################################### |