| #!/usr/bin/perl |
| |
| # (C) Sergey Kandaurov |
| # (C) Nginx, Inc. |
| |
| # Tests for stream upstream name resolved, proxy_next_upstream_tries. |
| |
| ############################################################################### |
| |
| use warnings; |
| use strict; |
| |
| use Test::More; |
| |
| BEGIN { use FindBin; chdir($FindBin::Bin); } |
| |
| use lib 'lib'; |
| use Test::Nginx; |
| use Test::Nginx::Stream qw/ stream /; |
| |
| ############################################################################### |
| |
| select STDERR; $| = 1; |
| select STDOUT; $| = 1; |
| |
| my $t = Test::Nginx->new()->has(qw/stream stream_map stream_return/) |
| ->write_file_expand('nginx.conf', <<'EOF'); |
| |
| %%TEST_GLOBALS%% |
| |
| daemon off; |
| |
| events { |
| } |
| |
| stream { |
| map $server_port $upstream { |
| %%PORT_8081%% a.example.com:%%PORT_8090%%; |
| %%PORT_8082%% a.example.com; |
| %%PORT_8083%% nx.example.com:%%PORT_8082%%; |
| } |
| |
| map $server_port $many { |
| default $server_port.many.example.com; |
| } |
| |
| resolver 127.0.0.1:%%PORT_8980_UDP%%; |
| |
| server { |
| listen 127.0.0.1:8081; |
| listen 127.0.0.1:8082; |
| listen 127.0.0.1:8083; |
| proxy_pass $upstream; |
| } |
| |
| server { |
| listen 127.0.0.1:8084; |
| proxy_pass $many:%%PORT_8090%%; |
| |
| proxy_next_upstream_tries 3; |
| proxy_connect_timeout 1s; |
| } |
| |
| server { |
| listen 127.0.0.1:8085; |
| proxy_pass $many:%%PORT_8090%%; |
| |
| proxy_next_upstream_tries 2; |
| proxy_connect_timeout 1s; |
| } |
| |
| server { |
| listen 127.0.0.1:8086; |
| proxy_pass $many:%%PORT_8090%%; |
| |
| proxy_next_upstream_tries 0; |
| proxy_connect_timeout 1s; |
| } |
| |
| server { |
| listen 127.0.0.1:8090; |
| return SEE-THIS; |
| } |
| } |
| |
| EOF |
| |
| $t->run_daemon(\&dns_daemon, port(8980), $t); |
| $t->run()->plan(8); |
| |
| $t->waitforfile($t->testdir . '/' . port(8980)); |
| |
| ############################################################################### |
| |
| ok(stream('127.0.0.1:' . port(8081))->read(), 'resolver'); |
| ok(!stream('127.0.0.1:' . port(8082))->read(), 'upstream no port'); |
| ok(!stream('127.0.0.1:' . port(8083))->read(), 'name not found'); |
| |
| ok(stream('127.0.0.1:' . port(8084))->read(), 'resolved tries'); |
| ok(!stream('127.0.0.1:' . port(8085))->read(), 'resolved tries limited'); |
| ok(stream('127.0.0.1:' . port(8086))->read(), 'resolved tries zero'); |
| |
| $t->stop(); |
| |
| SKIP: { |
| skip "relies on error log contents", 2 unless $ENV{TEST_NGINX_UNSAFE}; |
| |
| my $log = `grep -F '[error]' ${\($t->testdir())}/error.log`; |
| like($log, qr/no port in upstream "a.example.com"/, 'log - no port'); |
| like($log, qr/nx.example.com could not be resolved/, 'log - not found'); |
| |
| } |
| |
| ############################################################################### |
| |
| sub reply_handler { |
| my ($recv_data, $port) = @_; |
| |
| my (@name, @rdata); |
| |
| use constant NOERROR => 0; |
| use constant A => 1; |
| use constant IN => 1; |
| |
| # default values |
| |
| my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600); |
| |
| # decode name |
| |
| my ($len, $offset) = (undef, 12); |
| while (1) { |
| $len = unpack("\@$offset C", $recv_data); |
| last if $len == 0; |
| $offset++; |
| push @name, unpack("\@$offset A$len", $recv_data); |
| $offset += $len; |
| } |
| |
| $offset -= 1; |
| my ($id, $type, $class) = unpack("n x$offset n2", $recv_data); |
| |
| my $name = join('.', @name); |
| if ($name eq 'a.example.com' && $type == A) { |
| push @rdata, rd_addr($ttl, '127.0.0.1'); |
| |
| } elsif ($name =~ qr/many.example.com/ && $type == A) { |
| push @rdata, rd_addr($ttl, '127.0.0.2'); |
| push @rdata, rd_addr($ttl, '127.0.0.2'); |
| push @rdata, rd_addr($ttl, '127.0.0.1'); |
| } |
| |
| $len = @name; |
| pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata, |
| 0, 0, @name, $type, $class) . join('', @rdata); |
| } |
| |
| sub rd_addr { |
| my ($ttl, $addr) = @_; |
| |
| my $code = 'split(/\./, $addr)'; |
| |
| return pack 'n3N', 0xc00c, A, IN, $ttl if $addr eq ''; |
| |
| pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code); |
| } |
| |
| sub dns_daemon { |
| my ($port, $t) = @_; |
| |
| my ($data, $recv_data); |
| my $socket = IO::Socket::INET->new( |
| LocalAddr => '127.0.0.1', |
| LocalPort => $port, |
| Proto => 'udp', |
| ) |
| or die "Can't create listening socket: $!\n"; |
| |
| local $SIG{PIPE} = 'IGNORE'; |
| |
| # signal we are ready |
| |
| open my $fh, '>', $t->testdir() . '/' . $port; |
| close $fh; |
| |
| while (1) { |
| $socket->recv($recv_data, 65536); |
| $data = reply_handler($recv_data, $port); |
| $socket->send($data); |
| } |
| } |
| |
| ############################################################################### |