Mercurial > hg > nginx-tests
view http_resolver_cname.t @ 1236:93f749c1d5c5
Tests: fixed parallel tests execution with UDP.
Previously, when checking ports availability, a UDP socket was always created
first, then a TCP socket was created. On success, one of UDP and TCP sockets
was closed (depending on the "udp" option) and the second one was used to busy
this port in other scripts. This lead to the following problem: in an attempt
to reopen a UDP socket used in a given testing script it could be stolen by
another script as part of checking ports availability.
To solve this problem, UDP and TCP ports were split into two non-overlapping
ranges: TCP ports are only used in the range 8000-8499, and UDP ports - in
the range 8500-8999. In addition, the order of creating sockets in UDP tests
has been reversed: now a TCP socket used as a lock precedes a UDP socket.
author | Andrey Zelenkov <zelenkov@nginx.com> |
---|---|
date | Thu, 26 Oct 2017 18:00:21 +0300 |
parents | 882267679006 |
children | e4974af3fb12 |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for http resolver with CNAME. ############################################################################### use warnings; use strict; use Test::More; BEGIN { use FindBin; chdir($FindBin::Bin); } use lib 'lib'; use Test::Nginx qw/ :DEFAULT http_end /; ############################################################################### select STDERR; $| = 1; select STDOUT; $| = 1; my $t = Test::Nginx->new()->has(qw/http proxy/)->plan(11); $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; location /short { resolver 127.0.0.1:%%PORT_8081_UDP%%; resolver_timeout 2s; proxy_pass http://$host:%%PORT_8080%%/t; } location /long { resolver 127.0.0.1:%%PORT_8081_UDP%%; resolver_timeout 5s; proxy_pass http://$host:%%PORT_8080%%/t; } location / { } } } EOF $t->run_daemon(\&dns_daemon, port(8081), $t); $t->write_file('t', ''); $t->run(); $t->waitforfile($t->testdir . '/' . port(8081)); ############################################################################### # CNAME pointing to name which times out like(http_host('cn01.example.net', '/short'), qr/502 Bad/, 'CNAME timeout'); # several requests on CNAME pointing to invalid name my @s; push @s, http_host('cn03.example.net', '/long', start => 1); push @s, http_host('cn03.example.net', '/long', start => 1); like(http_end(pop @s), qr/502 Bad/, 'invalid CNAME - first'); like(http_end(pop @s), qr/502 Bad/, 'invalid CNAME - last'); # several requests on CNAME pointing to cached name @s = (); http_host('a.example.net', '/long'); push @s, http_host('cn04.example.net', '/long', start => 1); push @s, http_host('cn04.example.net', '/long', start => 1); like(http_end(pop @s), qr/200 OK/, 'cached CNAME - first'); like(http_end(pop @s), qr/200 OK/, 'cached CNAME - last'); # several requests on CNAME pointing to name being resolved @s = (); my $s = http_host('cn06.example.net', '/long', start => 1); sleep 1; push @s, http_host('cn05.example.net', '/long', start => 1); push @s, http_host('cn05.example.net', '/long', start => 1); like(http_end(pop @s), qr/502 Bad/, 'CNAME in progress - first'); like(http_end(pop @s), qr/502 Bad/, 'CNAME in progress - last'); # several requests on CNAME pointing to name which times out # 1st request receives CNAME with short ttl # 2nd request replaces expired CNAME @s = (); push @s, http_host('cn07.example.net', '/long', start => 1); sleep 2; push @s, http_host('cn07.example.net', '/long', start => 1); like(http_end(pop @s), qr/502 Bad/, 'CNAME ttl - first'); like(http_end(pop @s), qr/502 Bad/, 'CNAME ttl - last'); # several requests on CNAME pointing to name # 1st request aborts before name is resolved # 2nd request finishes with name resolved @s = (); push @s, http_host('cn09.example.net', '/long', start => 1); push @s, http_host('cn09.example.net', '/long', start => 1); select undef, undef, undef, 0.4; # let resolver hang on CNAME close(shift @s); like(http_end(pop @s), qr/200 OK/, 'abort on CNAME'); like(http_host('cn001.example.net', '/short'), qr/502 Bad/, 'recurse uncached'); ############################################################################### sub http_host { my ($host, $uri, %extra) = @_; return http(<<EOF, %extra); GET $uri HTTP/1.0 Host: $host EOF } ############################################################################### sub reply_handler { my ($recv_data) = @_; my (@name, @rdata); use constant NOERROR => 0; use constant A => 1; use constant CNAME => 5; 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.net' && $type == A) { push @rdata, rd_addr($ttl, '127.0.0.1'); } elsif ($name eq 'b.example.net' && $type == A) { sleep 2; push @rdata, rd_addr($ttl, '127.0.0.1'); } elsif ($name eq 'cn01.example.net') { $ttl = 1; push @rdata, pack("n3N nCa4n", 0xc00c, CNAME, IN, $ttl, 7, 4, "cn02", 0xc011); } elsif ($name =~ /cn0[268].example.net/) { # resolver timeout return; } elsif ($name eq 'cn03.example.net') { select undef, undef, undef, 1.1; push @rdata, pack("n3N nC", 0xc00c, CNAME, IN, $ttl, 0); } elsif ($name eq 'cn04.example.net') { select undef, undef, undef, 1.1; push @rdata, pack("n3N nCa1n", 0xc00c, CNAME, IN, $ttl, 4, 1, "a", 0xc011); } elsif ($name eq 'cn05.example.net') { select undef, undef, undef, 1.1; push @rdata, pack("n3N nCa4n", 0xc00c, CNAME, IN, $ttl, 7, 4, "cn06", 0xc011); } elsif ($name eq 'cn07.example.net') { $ttl = 1; push @rdata, pack("n3N nCa4n", 0xc00c, CNAME, IN, $ttl, 7, 4, "cn08", 0xc011); } elsif ($name eq 'cn09.example.net') { if ($type == A) { # await both HTTP requests select undef, undef, undef, 0.2; } push @rdata, pack("n3N nCa1n", 0xc00c, CNAME, IN, $ttl, 4, 1, "b", 0xc011); } elsif ($name eq 'cn052.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.1'); } } elsif ($name =~ /cn0\d+.example.net/) { my ($id) = $name =~ /cn(\d+)/; $id++; push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, "cn$id", 0xc012); } $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"; # signal we are ready open my $fh, '>', $t->testdir() . '/' . $port; close $fh; while (1) { $socket->recv($recv_data, 65536); $data = reply_handler($recv_data); $socket->send($data); } } ###############################################################################