Mercurial > hg > nginx-tests
view proxy_request_buffering_chunked.t @ 1320:8bd4b88fcac5
Tests: TODO js_return.t fail seen with older njs on some platforms.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Fri, 06 Apr 2018 17:21:23 +0300 |
parents | 882267679006 |
children |
line wrap: on
line source
#!/usr/bin/perl # (C) Maxim Dounin # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for unbuffered request body, chunked transfer-encoding. ############################################################################### 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(22); $t->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } http { %%TEST_GLOBALS_HTTP%% server { listen 127.0.0.1:8080; server_name localhost; client_header_buffer_size 1k; proxy_request_buffering off; proxy_http_version 1.1; location / { client_body_buffer_size 2k; add_header X-Body "$request_body"; proxy_pass http://127.0.0.1:8081; } location /small { client_body_in_file_only on; proxy_pass http://127.0.0.1:8080/; } location /single { client_body_in_single_buffer on; add_header X-Body "$request_body"; proxy_pass http://127.0.0.1:8081; } location /discard { return 200 "TEST\n"; } location /preread { proxy_pass http://127.0.0.1:8082/; } location /error_page { proxy_pass http://127.0.0.1:8081/404; error_page 404 /404; proxy_intercept_errors on; } location /404 { return 200 "$request_body\n"; } } server { listen 127.0.0.1:8081; server_name localhost; location / { proxy_pass http://127.0.0.1:8080/discard; } location /404 { } } } 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('/single', '0123456789' x 128), qr/X-Body: (0123456789){128}\x0d?$/ms, 'body in single buffer'); like(http_get_body('/error_page', '0123456789'), qr/^0123456789$/m, 'body in error page'); # 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 is disabled in unbuffered mode like(http_get_body('/small', '0123456789'), qr/X-Body: 0123456789\x0d?$/ms, 'small body in file only'); # interactive tests my $s = get_body('/preread', port(8082)); ok($s, 'no preread'); SKIP: { skip 'no preread failed', 3 unless $s; is($s->{upload}('01234'), '5' . CRLF . '01234' . CRLF, 'no preread - body part'); is($s->{upload}('56789', last => 1), '5' . CRLF . '56789' . CRLF . '0' . CRLF . CRLF, 'no preread - body part 2'); like($s->{http_end}(), qr/200 OK/, 'no preread - response'); } $s = get_body('/preread', port(8082), '01234'); ok($s, 'preread'); SKIP: { skip 'preread failed', 3 unless $s; is($s->{preread}, '5' . CRLF . '01234' . CRLF, 'preread - preread'); is($s->{upload}('56789', last => 1), '5' . CRLF . '56789' . CRLF . '0' . CRLF . CRLF, 'preread - body'); like($s->{http_end}(), qr/200 OK/, 'preread - response'); } $s = get_body('/preread', port(8082), '01234', many => 1); ok($s, 'chunks'); SKIP: { skip 'chunks failed', 3 unless $s; is($s->{preread}, '9' . CRLF . '01234many' . CRLF, 'chunks - preread'); is($s->{upload}('56789', many => 1, last => 1), '9' . CRLF . '56789many' . CRLF . '0' . CRLF . CRLF, 'chunks - body'); like($s->{http_end}(), qr/200 OK/, 'chunks - response'); } ############################################################################### 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 . "Transfer-Encoding: chunked" . CRLF . CRLF . sprintf("%x", length $body) . CRLF . $body . CRLF . "0" . CRLF . CRLF } @_), "GET $uri HTTP/1.1" . CRLF . "Host: localhost" . CRLF . "Connection: close" . CRLF . "Transfer-Encoding: chunked" . CRLF . CRLF . sprintf("%x", length $last) . CRLF . $last . CRLF . "0" . CRLF . CRLF ); } sub get_body { my ($url, $port, $body, %extra) = @_; my ($server, $client, $s); my ($last, $many) = (0, 0); $last = $extra{last} if defined $extra{last}; $many = $extra{many} if defined $extra{many}; $server = IO::Socket::INET->new( Proto => 'tcp', LocalHost => '127.0.0.1', LocalPort => $port, Listen => 5, Reuse => 1 ) or die "Can't create listening socket: $!\n"; my $r = <<EOF; GET $url HTTP/1.1 Host: localhost Connection: close Transfer-Encoding: chunked EOF if (defined $body) { $r .= sprintf("%x", length $body) . CRLF; $r .= $body . CRLF; } if (defined $body && $many) { $r .= sprintf("%x", length 'many') . CRLF; $r .= 'many' . CRLF; } if ($last) { $r .= "0" . CRLF . CRLF; } $s = http($r, start => 1); eval { local $SIG{ALRM} = sub { die "timeout\n" }; local $SIG{PIPE} = sub { die "sigpipe\n" }; alarm(5); $client = $server->accept(); log2c("(new connection $client)"); alarm(0); }; alarm(0); if ($@) { log_in("died: $@"); return undef; } $client->sysread(my $buf, 1024); log2i($buf); $buf =~ s/.*?\x0d\x0a?\x0d\x0a?(.*)/$1/ms; my $f = { preread => $buf }; $f->{upload} = sub { my ($body, %extra) = @_; my ($last, $many) = (0, 0); $last = $extra{last} if defined $extra{last}; $many = $extra{many} if defined $extra{many}; my $buf = sprintf("%x", length $body) . CRLF; $buf .= $body . CRLF; if ($many) { $buf .= sprintf("%x", length 'many') . CRLF; $buf .= 'many' . CRLF; } if ($last) { $buf .= "0" . CRLF . CRLF; } eval { local $SIG{ALRM} = sub { die "timeout\n" }; local $SIG{PIPE} = sub { die "sigpipe\n" }; alarm(5); log_out($buf); $s->write($buf); $client->sysread($buf, 1024); log2i($buf); alarm(0); }; alarm(0); if ($@) { log_in("died: $@"); return undef; } return $buf; }; $f->{http_end} = sub { my $buf = ''; $client->write(<<EOF); HTTP/1.1 200 OK Connection: close X-Port: $port OK EOF $client->close; eval { local $SIG{ALRM} = sub { die "timeout\n" }; local $SIG{PIPE} = sub { die "sigpipe\n" }; alarm(5); $s->sysread($buf, 1024); log_in($buf); $s->close(); alarm(0); }; alarm(0); if ($@) { log_in("died: $@"); return undef; } return $buf; }; return $f; } sub log2i { Test::Nginx::log_core('|| <<', @_); } sub log2o { Test::Nginx::log_core('|| >>', @_); } sub log2c { Test::Nginx::log_core('||', @_); } ###############################################################################