Mercurial > hg > nginx-tests

comparison http_resolver_ipv4.t @ 1797:0b06942f0b8b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: http resolver with ipv4/ipv6 parameters.
author Eugene Grebenschikov <e.grebenshchikov@f5.com>
date Wed, 12 Oct 2022 11:55:32 -0700
parents
children 1786f49cca17
comparison
equal deleted inserted replaced
1796:e1fd234baac0 1797:0b06942f0b8b
1 #!/usr/bin/perl
2
3 # (C) Sergey Kandaurov
4 # (C) Eugene Grebenschikov
5 # (C) Nginx, Inc.
6
7 # Tests for http resolver with ipv4/ipv6 parameters.
8
9 ###############################################################################
10
11 use warnings;
12 use strict;
13
14 use Test::More;
15
16 BEGIN { use FindBin; chdir($FindBin::Bin); }
17
18 use lib 'lib';
19 use Test::Nginx;
20
21 ###############################################################################
22
23 select STDERR; $| = 1;
24 select STDOUT; $| = 1;
25
26 my $t = Test::Nginx->new()->has(qw/http proxy rewrite/);
27
28 $t->write_file_expand('nginx.conf', <<'EOF');
29
30 %%TEST_GLOBALS%%
31
32 daemon off;
33
34 events {
35 }
36
37 http {
38 %%TEST_GLOBALS_HTTP%%
39
40 server {
41 listen 127.0.0.1:8080;
42 server_name localhost;
43
44 location / {
45 proxy_pass http://$arg_h:%%PORT_8081%%/;
46 resolver 127.0.0.1:%%PORT_8980_UDP%% ipv4=on ipv6=on;
47 }
48
49 location /ipv4 {
50 proxy_pass http://$arg_h:%%PORT_8081%%/;
51 resolver 127.0.0.1:%%PORT_8980_UDP%% ipv4=on ipv6=off;
52 }
53
54 location /ipv6 {
55 proxy_pass http://$arg_h:%%PORT_8081%%/;
56 resolver 127.0.0.1:%%PORT_8980_UDP%% ipv4=off ipv6=on;
57 }
58 }
59
60 server {
61 listen 127.0.0.1:8081;
62 server_name localhost;
63
64 location / {
65 return 200 "ipv4";
66 }
67 }
68
69 server {
70 listen [::1]:%%PORT_8081%%;
71 server_name localhost;
72
73 location / {
74 return 200 "ipv6";
75 }
76 }
77 }
78
79 EOF
80
81 $t->try_run('no resolver ipv4')->plan(3);
82
83 $t->run_daemon(\&dns_daemon, port(8980), $t);
84 $t->waitforfile($t->testdir . '/' . port(8980));
85
86 ###############################################################################
87
88 like(many('/', 10), qr/ipv4: \d+, ipv6: \d+/, 'ipv4 ipv6');
89 is(many('/ipv4', 10), 'ipv4: 10', 'ipv4 only');
90 is(many('/ipv6', 10), 'ipv6: 10', 'ipv6 only');
91
92 ###############################################################################
93
94 sub many {
95 my ($uri, $count) = @_;
96 my %hits;
97
98 for (1 .. $count) {
99 if (http_get("$uri?h=example.com") =~ /(ipv(4|6))/) {;
100 $hits{$1} = 0 unless defined $hits{$1};
101 $hits{$1}++;
102 }
103 }
104
105 return join ', ', map { $_ . ": " . $hits{$_} } sort keys %hits;
106 }
107
108 ###############################################################################
109
110 sub reply_handler {
111 my ($recv_data, $port) = @_;
112
113 my (@name, @rdata);
114
115 use constant NOERROR => 0;
116
117 use constant A => 1;
118 use constant AAAA => 28;
119
120 use constant IN => 1;
121
122 # default values
123
124 my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600);
125
126 # decode name
127
128 my ($len, $offset) = (undef, 12);
129 while (1) {
130 $len = unpack("\@$offset C", $recv_data);
131 last if $len == 0;
132 $offset++;
133 push @name, unpack("\@$offset A$len", $recv_data);
134 $offset += $len;
135 }
136
137 $offset -= 1;
138 my ($id, $type, $class) = unpack("n x$offset n2", $recv_data);
139
140 my $name = join('.', @name);
141 if ($name eq 'example.com') {
142 if ($type == A) {
143 push @rdata, rd_addr($ttl, '127.0.0.1');
144 }
145 if ($type == AAAA) {
146 push @rdata, rd_addr6($ttl, "::1");
147 }
148 }
149
150 $len = @name;
151 pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata,
152 0, 0, @name, $type, $class) . join('', @rdata);
153 }
154
155 sub rd_addr {
156 my ($ttl, $addr) = @_;
157
158 my $code = 'split(/\./, $addr)';
159
160 pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code);
161 }
162
163 sub expand_ip6 {
164 my ($addr) = @_;
165
166 substr ($addr, index($addr, "::"), 2) =
167 join "0", map { ":" } (0 .. 8 - (split /:/, $addr) + 1);
168 map { hex "0" x (4 - length $_) . "$_" } split /:/, $addr;
169 }
170
171 sub rd_addr6 {
172 my ($ttl, $addr) = @_;
173
174 pack 'n3N nn8', 0xc00c, AAAA, IN, $ttl, 16, expand_ip6($addr);
175 }
176
177 sub dns_daemon {
178 my ($port, $t) = @_;
179
180 my ($data, $recv_data);
181 my $socket = IO::Socket::INET->new(
182 LocalAddr => '127.0.0.1',
183 LocalPort => $port,
184 Proto => 'udp',
185 )
186 or die "Can't create listening socket: $!\n";
187
188 # signal we are ready
189
190 open my $fh, '>', $t->testdir() . '/' . $port;
191 close $fh;
192
193 while (1) {
194 $socket->recv($recv_data, 65536);
195 $data = reply_handler($recv_data, $port);
196 $socket->send($data);
197 }
198 }
199
200 ###############################################################################