Mercurial > hg > nginx-tests
view http_resolver_aaaa.t @ 1990:85d88cd5091c
Tests: SSI include stub tests.
author | Maxim Dounin <mdounin@mdounin.ru> |
---|---|
date | Thu, 04 Jul 2024 17:42:55 +0300 |
parents | e4974af3fb12 |
children |
line wrap: on
line source
#!/usr/bin/perl # (C) Sergey Kandaurov # (C) Nginx, Inc. # Tests for AAAA capable http resolver. ############################################################################### use warnings; use strict; use Test::More; 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/); $t->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } http { %%TEST_GLOBALS_HTTP%% server { listen 127.0.0.1:8080; listen [::1]:%%PORT_8080%%; server_name localhost; location / { resolver 127.0.0.1:%%PORT_8981_UDP%%; proxy_pass http://$host:%%PORT_8080%%/backend; proxy_next_upstream http_504 timeout error; proxy_intercept_errors on; proxy_connect_timeout 50ms; error_page 504 502 /50x; add_header X-Host $upstream_addr; } location /two { resolver 127.0.0.1:%%PORT_8981_UDP%% 127.0.0.1:%%PORT_8982_UDP%%; proxy_pass http://$host:%%PORT_8080%%/backend; } location /backend { return 200; } location /50x { return 200 $upstream_addr; } } } EOF $t->try_run('no inet6 support')->plan(72); $t->run_daemon(\&dns_daemon, port(8981), $t); $t->run_daemon(\&dns_daemon, port(8982), $t); $t->waitforfile($t->testdir . '/' . port(8981)); $t->waitforfile($t->testdir . '/' . port(8982)); ############################################################################### my (@n, $response); my $p0 = port(8080); like(http_host_header('aaaa.example.net', '/'), qr/\[fe80::1\]/, 'AAAA'); like(http_host_header('cname.example.net', '/'), qr/\[fe80::1\]/, 'CNAME'); like(http_host_header('cname.example.net', '/'), qr/\[fe80::1\]/, 'CNAME cached'); # CNAME + AAAA combined answer # demonstrates the name in answer section different from what is asked like(http_host_header('cname_a.example.net', '/'), qr/\[::1\]/, 'CNAME + AAAA'); # many AAAA records in round robin # nonexisting IPs enumerated with proxy_next_upstream like(http_host_header('many.example.net', '/'), qr/^\[fe80::(1\]:$p0, \[fe80::2\]:$p0|2\]:$p0, \[fe80::1\]:$p0)$/m, 'AAAA many'); like(http_host_header('many.example.net', '/'), qr/^\[fe80::(1\]:$p0, \[fe80::2\]:$p0|2\]:$p0, \[fe80::1\]:$p0)$/m, 'AAAA many cached'); # tests for several resolvers specified in directive # query bad ns, make sure that error responses are not cached like(http_host_header('2.example.net', '/two'), qr/502 Bad/, 'two ns bad'); # now get correct response like(http_host_header('2.example.net', '/two'), qr/200 OK/, 'two ns good'); # response is cached, actual request would get error like(http_host_header('2.example.net', '/two'), qr/200 OK/, 'two ns cached'); # various ipv4/ipv6 combinations $response = http_host_header('z_z.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'zero zero responses'); like($response, qr/502 Bad/, 'zero zero'); like(http_host_header('z_n.example.net', '/'), qr/^\[fe80::1\]:$p0$/ms, 'zero AAAA'); $response = http_host_header('z_c.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'zero CNAME responses'); like($response, qr/127.0.0.201:$p0/, 'zero CNAME 1'); like($response, qr/\[fe80::1\]:$p0/, 'zero CNAME 2'); $response = http_host_header('z_cn.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'zero CNAME+AAAA responses'); like($response, qr/\[fe80::1\]:$p0/, 'zero CNAME+AAAA 1'); like($response, qr/\[fe80::2\]:$p0/, 'zero CNAME+AAAA 2'); $response = http_host_header('z_e.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'zero error responses'); like($response, qr/502 Bad/, 'zero error'); like(http_host_header('n_z.example.net', '/'), qr/^127.0.0.201:$p0$/ms, 'A zero'); $response = http_host_header('n_n.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'A AAAA responses'); like($response, qr/127.0.0.201:$p0/, 'A AAAA 1'); like($response, qr/\[fe80::1\]:$p0/, 'A AAAA 2'); like(http_host_header('n_c.example.net', '/'), qr/^127.0.0.201:$p0$/ms, 'A CNAME'); $response = http_host_header('n_cn.example.net', '/'); is(@n = $response =~ /$p0/g, 4, 'A CNAME+AAAA responses'); like($response, qr/127.0.0.201:$p0/, 'A CNAME+AAAA 1'); like($response, qr/127.0.0.202:$p0/, 'A CNAME+AAAA 2'); like($response, qr/\[fe80::1\]:$p0/, 'A CNAME+AAAA 3'); like($response, qr/\[fe80::2\]:$p0/, 'A CNAME+AAAA 4'); $response = http_host_header('n_e.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'A error responses'); like($response, qr/502 Bad/, 'A error'); $response = http_host_header('c_z.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'CNAME zero responses'); like($response, qr/502 Bad/, 'CNAME zero'); like(http_host_header('c_n.example.net', '/'), qr/^\[fe80::1\]:$p0$/ms, 'CNAME AAAA'); $response = http_host_header('c_c.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'CNAME CNAME responses'); like($response, qr/127.0.0.201:$p0/, 'CNAME CNAME 1'); like($response, qr/\[fe80::1\]:$p0/, 'CNAME CNAME 2'); like(http_host_header('c1_c2.example.net', '/'), qr/^\[fe80::1\]:$p0$/ms, 'CNAME1 CNAME2'); $response = http_host_header('c_cn.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'CNAME CNAME+AAAA responses'); like($response, qr/\[fe80::1\]:$p0/, 'CNAME CNAME+AAAA 1'); like($response, qr/\[fe80::2\]:$p0/, 'CNAME CNAME+AAAA 1'); $response = http_host_header('c_e.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'CNAME error responses'); like($response, qr/502 Bad/, 'CNAME error'); $response = http_host_header('cn_z.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'CNAME+A zero responses'); like($response, qr/127.0.0.201:$p0/, 'CNAME+A zero 1'); like($response, qr/127.0.0.202:$p0/, 'CNAME+A zero 2'); $response = http_host_header('cn_n.example.net', '/'); is(@n = $response =~ /$p0/g, 4, 'CNAME+A AAAA responses'); like($response, qr/127.0.0.201:$p0/, 'CNAME+A AAAA 1'); like($response, qr/127.0.0.202:$p0/, 'CNAME+A AAAA 2'); like($response, qr/\[fe80::1\]:$p0/, 'CNAME+A AAAA 3'); like($response, qr/\[fe80::2\]:$p0/, 'CNAME+A AAAA 4'); $response = http_host_header('cn_c.example.net', '/'); is(@n = $response =~ /$p0/g, 2, 'CNAME+A CNAME responses'); like($response, qr/127.0.0.201:$p0/, 'CNAME+A CNAME 1'); like($response, qr/127.0.0.202:$p0/, 'CNAME+A CNAME 2'); $response = http_host_header('cn_cn.example.net', '/'); is(@n = $response =~ /$p0/g, 4, 'CNAME+A CNAME+AAAA responses'); like($response, qr/127.0.0.201:$p0/, 'CNAME+A CNAME+AAAA 1'); like($response, qr/127.0.0.202:$p0/, 'CNAME+A CNAME+AAAA 2'); like($response, qr/\[fe80::1\]:$p0/, 'CNAME+A CNAME+AAAA 3'); like($response, qr/\[fe80::2\]:$p0/, 'CNAME+A CNAME+AAAA 4'); $response = http_host_header('cn_e.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'CNAME+A error responses'); like($response, qr/502 Bad/, 'CNAME+A error'); $response = http_host_header('e_z.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'error zero responses'); like($response, qr/502 Bad/, 'error zero'); $response = http_host_header('e_n.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'error AAAA responses'); like($response, qr/502 Bad/, 'error AAAA'); $response = http_host_header('e_c.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'error CNAME responses'); like($response, qr/502 Bad/, 'error CNAME'); $response = http_host_header('e_cn.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'error CNAME+AAAA responses'); like($response, qr/502 Bad/, 'error CNAME+AAAA'); $response = http_host_header('e_e.example.net', '/'); is(@n = $response =~ /$p0/g, 0, 'error error responses'); like($response, qr/502 Bad/, 'error error'); ############################################################################### sub http_host_header { my ($host, $uri) = @_; return http(<<EOF); GET $uri HTTP/1.0 Host: $host EOF } ############################################################################### sub reply_handler { my ($recv_data, $port, $state) = @_; my (@name, @rdata); use constant NOERROR => 0; use constant SERVFAIL => 2; use constant NXDOMAIN => 3; use constant A => 1; use constant CNAME => 5; use constant AAAA => 28; use constant DNAME => 39; 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 'aaaa.example.net') || ($name eq 'alias.example.net')) { if ($type == AAAA) { push @rdata, rd_addr6($ttl, "fe80::1"); } } elsif ($name eq 'alias2.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); } if ($type == AAAA) { push @rdata, rd_addr6($ttl, "fe80::1"); } } elsif ($name eq 'alias4.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); } } elsif ($name eq 'alias6.example.net') { if ($type == AAAA) { push @rdata, rd_addr6($ttl, "fe80::1"); } } elsif (($name eq 'many.example.net') && $type == AAAA) { $state->{manycnt}++; if ($state->{manycnt} > 1) { $rcode = SERVFAIL; } push @rdata, rd_addr6($ttl, 'fe80::1'); push @rdata, rd_addr6($ttl, 'fe80::2'); } elsif ($name eq 'cname.example.net') { $state->{cnamecnt}++; if ($state->{cnamecnt} > 2) { $rcode = SERVFAIL; } push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc012); } elsif ($name eq 'cname_a.example.net') { push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc014); # points to "alias" set in previous rdata if ($type == AAAA) { push @rdata, pack('n3N nn8', 0xc031, AAAA, IN, $ttl, 16, expand_ip6("::1")); } } elsif ($name eq '2.example.net') { if ($port == port(8981)) { $state->{twocnt}++; } if ($state->{twocnt} & 1) { $rcode = SERVFAIL; } if ($type == AAAA) { push @rdata, rd_addr6($ttl, '::1'); } } elsif ($name eq 'z_z.example.net') { # assume no answers given } elsif ($name eq 'z_n.example.net') { if ($type == AAAA) { push @rdata, rd_addr6($ttl, 'fe80::1'); } } elsif ($name eq 'z_c.example.net') { if ($type == AAAA) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc010); } } elsif ($name eq 'z_cn.example.net') { if ($type == AAAA) { push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc011); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::1")); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::2")); } } elsif ($name eq 'z_e.example.net') { if ($type == AAAA) { $rcode = SERVFAIL; } } elsif ($name eq 'n_z.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); } } elsif ($name eq 'n_n.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); } if ($type == AAAA) { push @rdata, rd_addr6($ttl, 'fe80::1'); } } elsif ($name eq 'n_c.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); } if ($type == AAAA) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc010); } } elsif ($name eq 'n_cn.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); push @rdata, rd_addr($ttl, '127.0.0.202'); } if ($type == AAAA) { push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc011); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::1")); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::2")); } } elsif ($name eq 'n_e.example.net') { if ($type == A) { push @rdata, rd_addr($ttl, '127.0.0.201'); } if ($type == AAAA) { $rcode = SERVFAIL; } } elsif ($name eq 'c_z.example.net') { if ($type == A) { push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc010); } } elsif ($name eq 'c_n.example.net') { if ($type == A) { push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc010); } if ($type == AAAA) { push @rdata, rd_addr6($ttl, "fe80::1"); } } elsif ($name eq 'c_c.example.net') { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc010); } elsif ($name eq 'c1_c2.example.net') { if ($type == A) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias4', 0xc012); } if ($type == AAAA) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias6', 0xc012); } } elsif ($name eq 'c_cn.example.net') { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc011); if ($type == AAAA) { push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::1")); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::2")); } } elsif ($name eq 'cn_z.example.net') { if ($type == A) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc011); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.201')); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.202')); } } elsif ($name eq 'cn_n.example.net') { if ($type == A) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc011); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.201')); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.202')); } if ($type == AAAA) { push @rdata, pack('n3N nn8', 0xc00c, AAAA, IN, $ttl, 16, expand_ip6("fe80::1")); push @rdata, pack('n3N nn8', 0xc00c, AAAA, IN, $ttl, 16, expand_ip6("fe80::2")); } } elsif ($name eq 'cn_c.example.net') { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc011); if ($type == A) { push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.201')); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.202')); } } elsif ($name eq 'cn_cn.example.net') { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc012); if ($type == A) { push @rdata, pack("n3N nC4", 0xc02f, A, IN, $ttl, 4, split('\.', '127.0.0.201')); push @rdata, pack("n3N nC4", 0xc02f, A, IN, $ttl, 4, split('\.', '127.0.0.202')); } if ($type == AAAA) { push @rdata, pack('n3N nn8', 0xc02f, AAAA, IN, $ttl, 16, expand_ip6("fe80::1")); push @rdata, pack('n3N nn8', 0xc02f, AAAA, IN, $ttl, 16, expand_ip6("fe80::2")); } } elsif ($name eq 'cn_e.example.net') { if ($type == A) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc011); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.201')); push @rdata, pack("n3N nC4", 0xc02e, A, IN, $ttl, 4, split('\.', '127.0.0.202')); } if ($type == AAAA) { $rcode = SERVFAIL; } } elsif ($name eq 'e_z.example.net') { if ($type == A) { $rcode = SERVFAIL; } } elsif ($name eq 'e_n.example.net') { if ($type == A) { $rcode = SERVFAIL; } if ($type == AAAA) { push @rdata, rd_addr6($ttl, 'fe80::1'); } } elsif ($name eq 'e_c.example.net') { if ($type == A) { $rcode = SERVFAIL; } if ($type == AAAA) { push @rdata, pack("n3N nCa6n", 0xc00c, CNAME, IN, $ttl, 9, 6, 'alias2', 0xc010); } } elsif ($name eq 'e_cn.example.net') { if ($type == A) { $rcode = SERVFAIL; } if ($type == AAAA) { push @rdata, pack("n3N nCa5n", 0xc00c, CNAME, IN, $ttl, 8, 5, 'alias', 0xc011); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::1")); push @rdata, pack('n3N nn8', 0xc02e, AAAA, IN, $ttl, 16, expand_ip6("fe80::2")); } } elsif ($name eq 'e_e.example.net') { if ($type == A) { $rcode = SERVFAIL; } if ($type == AAAA) { $rcode = NXDOMAIN; } } $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)'; pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code); } sub expand_ip6 { my ($addr) = @_; substr ($addr, index($addr, "::"), 2) = join "0", map { ":" } (0 .. 8 - (split /:/, $addr) + 1); map { hex "0" x (4 - length $_) . "$_" } split /:/, $addr; } sub rd_addr6 { my ($ttl, $addr) = @_; pack 'n3N nn8', 0xc00c, AAAA, IN, $ttl, 16, expand_ip6($addr); } 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"; # track number of relevant queries my %state = ( cnamecnt => 0, twocnt => 0, manycnt => 0, ); # 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, \%state); $socket->send($data); } } ###############################################################################